status-go/protocol/local_notifications.go

package protocol

import (
	"crypto/ecdsa"
	"encoding/json"

	gethcommon "github.com/ethereum/go-ethereum/common"
	"github.com/status-im/status-go/protocol/common"
	"github.com/status-im/status-go/protocol/communities"
	localnotifications "github.com/status-im/status-go/services/local-notifications"
)

type NotificationBody struct {
	Message   *common.Message        `json:"message"`
	Contact   *Contact               `json:"contact"`
	Chat      *Chat                  `json:"chat"`
	Community *communities.Community `json:"community"`
}

func showMessageNotification(publicKey ecdsa.PublicKey, message *common.Message, chat *Chat, responseTo *common.Message) bool {
	if chat != nil && !chat.Active {
		return false
	}

	if chat != nil && (chat.OneToOne() || chat.PrivateGroupChat()) {
		return true
	}

	if message.Mentioned {
		return true
	}

	if responseTo != nil {
		publicKeyString := common.PubkeyToHex(&publicKey)
		return responseTo.From == publicKeyString
	}

	return false
}

func (n NotificationBody) MarshalJSON() ([]byte, error) {
	type Alias NotificationBody
	item := struct{ *Alias }{Alias: (*Alias)(&n)}
	return json.Marshal(item)
}

func NewMessageNotification(id string, message *common.Message, chat *Chat, contact *Contact, contacts *contactMap) (*localnotifications.Notification, error) {
	body := &NotificationBody{
		Message: message,
		Chat:    chat,
		Contact: contact,
	}

	return body.toMessageNotification(id, contacts)
}

func DeletedMessageNotification(id string, chat *Chat) *localnotifications.Notification {
	return &localnotifications.Notification{
		BodyType:       localnotifications.TypeMessage,
		ID:             gethcommon.HexToHash(id),
		IsConversation: true,
		ConversationID: chat.ID,
		Deeplink:       chat.DeepLink(),
		Deleted:        true,
	}
}

func NewCommunityRequestToJoinNotification(id string, community *communities.Community, contact *Contact) *localnotifications.Notification {
	body := &NotificationBody{
		Community: community,
		Contact:   contact,
	}

	return body.toCommunityRequestToJoinNotification(id)
}

func NewPrivateGroupInviteNotification(id string, chat *Chat, contact *Contact) *localnotifications.Notification {
	body := &NotificationBody{
		Chat:    chat,
		Contact: contact,
	}

	return body.toPrivateGroupInviteNotification(id)
}

func (n NotificationBody) toMessageNotification(id string, contacts *contactMap) (*localnotifications.Notification, error) {
	var title string
	if n.Chat.PrivateGroupChat() || n.Chat.Public() || n.Chat.CommunityChat() {
		title = n.Chat.Name
	} else if n.Chat.OneToOne() {
		title = n.Contact.CanonicalName()

	}

	canonicalNames := make(map[string]string)
	for _, mentionID := range n.Message.Mentions {
		contact, ok := contacts.Load(mentionID)
		if !ok {
			var err error
			contact, err = buildContactFromPkString(mentionID)
			if err != nil {
				return nil, err
			}
		}
		canonicalNames[mentionID] = contact.CanonicalName()
	}

	// We don't pass idenity as only interested in the simplified text
	simplifiedText, err := n.Message.GetSimplifiedText("", canonicalNames)
	if err != nil {
		return nil, err
	}

	return &localnotifications.Notification{
		Body:                n,
		ID:                  gethcommon.HexToHash(id),
		BodyType:            localnotifications.TypeMessage,
		Category:            localnotifications.CategoryMessage,
		Deeplink:            n.Chat.DeepLink(),
		Title:               title,
		Message:             simplifiedText,
		IsConversation:      true,
		IsGroupConversation: true,
		Author: localnotifications.NotificationAuthor{
			Name: n.Contact.CanonicalName(),
			Icon: n.Contact.CanonicalImage(),
			ID:   n.Contact.ID,
		},
		Timestamp:      n.Message.WhisperTimestamp,
		ConversationID: n.Chat.ID,
		Image:          "",
	}, nil
}

func (n NotificationBody) toPrivateGroupInviteNotification(id string) *localnotifications.Notification {
	return &localnotifications.Notification{
		ID:       gethcommon.HexToHash(id),
		Body:     n,
		Title:    n.Contact.CanonicalName() + " invited you to " + n.Chat.Name,
		Message:  n.Contact.CanonicalName() + " wants you to join group " + n.Chat.Name,
		BodyType: localnotifications.TypeMessage,
		Category: localnotifications.CategoryGroupInvite,
		Deeplink: n.Chat.DeepLink(),
		Author: localnotifications.NotificationAuthor{
			Name: n.Contact.CanonicalName(),
			Icon: n.Contact.CanonicalImage(),
			ID:   n.Contact.ID,
		},
		Image: "",
	}
}

func (n NotificationBody) toCommunityRequestToJoinNotification(id string) *localnotifications.Notification {
	return &localnotifications.Notification{
		ID:       gethcommon.HexToHash(id),
		Body:     n,
		Title:    n.Contact.CanonicalName() + " wants to join " + n.Community.Name(),
		Message:  n.Contact.CanonicalName() + " wants to join  message " + n.Community.Name(),
		BodyType: localnotifications.TypeMessage,
		Category: localnotifications.CategoryCommunityRequestToJoin,
		Deeplink: "status-im://cr/" + n.Community.IDString(),
		Image:    "",
	}
}