status-go/protocol/messenger_response.go

package protocol

import (
	"github.com/status-im/status-go/protocol/common"
	"github.com/status-im/status-go/protocol/communities"
	"github.com/status-im/status-go/protocol/encryption/multidevice"
	"github.com/status-im/status-go/protocol/transport"
	"github.com/status-im/status-go/services/mailservers"
)

type MessengerResponse struct {
	Chats            []*Chat                         `json:"chats,omitempty"`
	RemovedChats     []string                        `json:"removedChats,omitempty"`
	Messages         []*common.Message               `json:"messages,omitempty"`
	Contacts         []*Contact                      `json:"contacts,omitempty"`
	Installations    []*multidevice.Installation     `json:"installations,omitempty"`
	EmojiReactions   []*EmojiReaction                `json:"emojiReactions,omitempty"`
	Invitations      []*GroupChatInvitation          `json:"invitations,omitempty"`
	Communities      []*communities.Community        `json:"communities,omitempty"`
	CommunityChanges []*communities.CommunityChanges `json:"communitiesChanges,omitempty"`
	Filters          []*transport.Filter             `json:"filters,omitempty"`
	RemovedFilters   []*transport.Filter             `json:"removedFilters,omitempty"`
	Mailservers      []mailservers.Mailserver        `json:"mailservers,omitempty"`
	MailserverTopics []mailservers.MailserverTopic   `json:"mailserverTopics,omitempty"`
	MailserverRanges []mailservers.ChatRequestRange  `json:"mailserverRanges,omitempty"`
}

func (m *MessengerResponse) IsEmpty() bool {
	return len(m.Chats)+len(m.Messages)+len(m.Contacts)+len(m.Installations)+len(m.Invitations)+len(m.EmojiReactions)+len(m.Communities)+len(m.CommunityChanges)+len(m.Filters)+len(m.RemovedFilters)+len(m.RemovedChats)+len(m.MailserverTopics)+len(m.Mailservers)+len(m.MailserverRanges) == 0
}

// Merge takes another response and appends the new Chats & new Messages and replaces
// the existing Messages & Chats if they have the same ID
func (m *MessengerResponse) Merge(response *MessengerResponse) error {
	if len(response.Contacts)+len(response.Installations)+len(response.EmojiReactions)+len(response.Invitations) != 0 {
		return ErrNotImplemented
	}

	m.overrideChats(response.Chats)
	m.overrideMessages(response.Messages)

	return nil
}

// overrideChats append new chats and override existing ones in response.Chats
func (m *MessengerResponse) overrideChats(chats []*Chat) {
	for _, overrideChat := range chats {
		var found = false
		for idx, chat := range m.Chats {
			if chat.ID == overrideChat.ID {
				m.Chats[idx] = overrideChat
				found = true
			}
		}
		if !found {
			m.Chats = append(m.Chats, overrideChat)
		}
	}
}

// overrideMessages append new messages and override existing ones in response.Messages
func (m *MessengerResponse) overrideMessages(messages []*common.Message) {
	for _, overrideMessage := range messages {
		var found = false
		for idx, chat := range m.Messages {
			if chat.ID == overrideMessage.ID {
				m.Messages[idx] = overrideMessage
				found = true
			}
		}
		if !found {
			m.Messages = append(m.Messages, overrideMessage)
		}
	}
}