package databag

import (
  "time"
  "errors"
  "net/http"
  "encoding/hex"
  "gorm.io/gorm"
  "github.com/google/uuid"
  "databag/internal/store"
  "github.com/theckman/go-securerandom"
)

func SetOpenMessage(w http.ResponseWriter, r *http.Request) {

  var message DataMessage
  if err := ParseRequest(r, w, &message); err != nil {
    ErrResponse(w, http.StatusBadRequest, err)
    return
  }

  var connect Connect
  guid, messageType, ts, err := ReadDataMessage(&message, &connect)
  if messageType != APP_MSGCONNECT || err != nil {
    ErrResponse(w, http.StatusBadRequest, err)
    return
  }
  if ts + APP_CONNECTEXPIRE < time.Now().Unix() {
    ErrResponse(w, http.StatusBadRequest, errors.New("message has expired"))
    return
  }

  // load referenced account
  var account store.Account
  if err := store.DB.Where("guid = ?", connect.Contact).First(&account).Error; err != nil {
    if errors.Is(err, gorm.ErrRecordNotFound) {
      ErrResponse(w, http.StatusNotFound, err)
    } else {
      ErrResponse(w, http.StatusInternalServerError, err)
    }
    return
  }

  // see if card exists
  slot := &store.CardSlot{}
  card := &store.Card{}
  if err := store.DB.Preload("CardSlot").Where("account_id = ? AND guid = ?", account.Guid, guid).First(card).Error; err != nil {
    if !errors.Is(err, gorm.ErrRecordNotFound) {
      ErrResponse(w, http.StatusInternalServerError, err)
      return
    }

    // create new card
    data, res := securerandom.Bytes(APP_TOKENSIZE)
    if res != nil {
      ErrResponse(w, http.StatusInternalServerError, res)
      return
    }
    card.Guid = guid
    card.Username = connect.Handle
    card.Name = connect.Name
    card.Description = connect.Description
    card.Location = connect.Location
    card.Image = connect.Image
    card.Version = connect.Version
    card.Node = connect.Node
    card.ProfileRevision = connect.ProfileRevision
    card.Status = APP_CARDPENDING
    card.NotifiedProfile = connect.ProfileRevision
    card.NotifiedArticle = connect.ArticleRevision
    card.NotifiedView = connect.ViewRevision
    card.NotifiedChannel = connect.ChannelRevision
    card.OutToken = connect.Token
    card.InToken = hex.EncodeToString(data)
    card.AccountID = account.Guid

    // create slot
    err = store.DB.Transaction(func(tx *gorm.DB) error {
      if res := tx.Save(card).Error; res != nil {
        return res
      }
      slot.CardSlotId = uuid.New().String()
      slot.AccountID = account.ID
      slot.Revision = account.CardRevision + 1
      slot.CardID = card.ID
      slot.Card = card
      if res := tx.Save(slot).Error; res != nil {
        return res
      }
      if res := tx.Model(&account).Update("card_revision", account.CardRevision + 1).Error; res != nil {
        return res
      }
      return nil
    })
    if err != nil {
      ErrResponse(w, http.StatusInternalServerError, err)
      return
    }
  } else {

    // update profile if revision changed
    if connect.ProfileRevision > card.ProfileRevision {
      card.Username = connect.Handle
      card.Name = connect.Name
      card.Description = connect.Description
      card.Location = connect.Location
      card.Image = connect.Image
      card.Version = connect.Version
      card.Node = connect.Node
      card.ProfileRevision = connect.ProfileRevision
    }
    if connect.ArticleRevision > card.NotifiedArticle {
      card.NotifiedArticle = connect.ArticleRevision
    }
    if connect.ViewRevision > card.NotifiedView {
      card.NotifiedView = connect.ViewRevision
    }
    if connect.ChannelRevision > card.NotifiedChannel {
      card.NotifiedChannel = connect.ChannelRevision
    }
    if connect.ProfileRevision > card.NotifiedProfile {
      card.NotifiedProfile = connect.ProfileRevision
    }
    if card.Status == APP_CARDCONFIRMED {
      card.Status = APP_CARDREQUESTED
    }
    if card.Status == APP_CARDCONNECTING {
      card.Status = APP_CARDCONNECTED
    }
    card.OutToken = connect.Token
    card.DetailRevision = account.CardRevision + 1

    // save contact card
    err  = store.DB.Transaction(func(tx *gorm.DB) error {
      if res := tx.Save(&card).Error; res != nil {
        return res
      }
      slot = &card.CardSlot
      slot.Revision = account.CardRevision + 1
      if res := tx.Preload("Card").Save(slot).Error; res != nil {
        return res
      }
      if res := tx.Model(&account).Update("card_revision", account.CardRevision + 1).Error; res != nil {
        return res
      }
      return nil
    })
    if err != nil {
      ErrResponse(w, http.StatusInternalServerError, err)
      return
    }
  }

  status := &ContactStatus{
    Token: card.InToken,
    Status: card.Status,
    ViewRevision: card.ViewRevision,
    ChannelRevision: account.ChannelRevision,
    ProfileRevision: account.ProfileRevision,
    ArticleRevision: account.ArticleRevision,
  }
  SetStatus(&account)
  WriteResponse(w, &status)
}