databag/net/server/internal/api_setOpenMessage.go

package databag

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

//SetOpenMessage delivers connection message to contact
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 != APPMsgConnect || err != nil {
		ErrResponse(w, http.StatusBadRequest, err)
		return
	}
	if ts+APPConnectExpire < 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(APPTokenSize)
		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 = APPCardPending
    card.StatusUpdated = time.Now().Unix()
		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
		card.DetailRevision = account.CardRevision + 1

		// 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 == APPCardConfirmed {
			card.Status = APPCardRequested
      card.StatusUpdated = time.Now().Unix()
		}
		if card.Status == APPCardConnecting {
			card.Status = APPCardConnected
      card.StatusUpdated = time.Now().Unix()
		}
		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)
}