status-go/protocol/messenger_pin_messages.go

191 lines
4.7 KiB
Go
Raw Normal View History

package protocol
import (
"bytes"
"context"
"crypto/ecdsa"
"crypto/sha256"
"encoding/binary"
"errors"
"fmt"
gethcommon "github.com/ethereum/go-ethereum/common"
"github.com/status-im/status-go/eth-node/crypto"
"github.com/status-im/status-go/protocol/common"
"github.com/status-im/status-go/protocol/protobuf"
)
// SendPinMessage sends the PinMessage to the corresponding chat
func (m *Messenger) SendPinMessage(ctx context.Context, message *common.PinMessage) (*MessengerResponse, error) {
m.mutex.Lock()
defer m.mutex.Unlock()
return m.sendPinMessage(ctx, message)
}
func (m *Messenger) sendPinMessage(ctx context.Context, message *common.PinMessage) (*MessengerResponse, error) {
var response MessengerResponse
// A valid added chat is required.
chat, ok := m.allChats.Load(message.ChatId)
if !ok {
return nil, errors.New("chat not found")
}
err := m.handleStandaloneChatIdentity(chat)
if err != nil {
return nil, err
}
err = extendPinMessageFromChat(message, chat, &m.identity.PublicKey, m.getTimesource())
if err != nil {
return nil, err
}
message.ID, err = generatePinMessageID(&m.identity.PublicKey, message, chat)
if err != nil {
return nil, err
}
encodedMessage, err := m.encodeChatEntity(chat, message)
if err != nil {
return nil, err
}
rawMessage := common.RawMessage{
LocalChatID: chat.ID,
Payload: encodedMessage,
MessageType: protobuf.ApplicationMetadataMessage_PIN_MESSAGE,
SkipGroupMessageWrap: true,
ResendAutomatically: true,
}
_, err = m.dispatchMessage(ctx, rawMessage)
if err != nil {
return nil, err
}
err = m.persistence.SavePinMessages([]*common.PinMessage{message})
if err != nil {
return nil, err
}
if message.Pinned {
id, err := generatePinMessageNotificationID(&m.identity.PublicKey, message, chat)
if err != nil {
return nil, err
}
chatMessage := &common.Message{
ChatMessage: &protobuf.ChatMessage{
Clock: message.Clock,
Timestamp: m.getTimesource().GetCurrentTime(),
ChatId: chat.ID,
MessageType: message.MessageType,
ResponseTo: message.MessageId,
ContentType: protobuf.ChatMessage_SYSTEM_MESSAGE_PINNED_MESSAGE,
},
WhisperTimestamp: m.getTimesource().GetCurrentTime(),
ID: id,
LocalChatID: chat.ID,
From: m.myHexIdentity(),
}
msg := []*common.Message{chatMessage}
err = m.persistence.SaveMessages(msg)
if err != nil {
return nil, err
}
msg, err = m.pullMessagesAndResponsesFromDB(msg)
if err != nil {
return nil, err
}
response.SetMessages(msg)
err = m.prepareMessages(response.messages)
if err != nil {
return nil, err
}
}
response.AddPinMessage(message)
response.AddChat(chat)
return &response, m.saveChat(chat)
}
func (m *Messenger) PinnedMessageByChatID(chatID, cursor string, limit int) ([]*common.PinnedMessage, string, error) {
pinnedMsgs, cursor, err := m.persistence.PinnedMessageByChatID(chatID, cursor, limit)
if err != nil {
return nil, "", err
}
if m.httpServer != nil {
for idx := range pinnedMsgs {
msg := pinnedMsgs[idx].Message
err = m.prepareMessage(msg, m.httpServer)
if err != nil {
return nil, "", err
}
pinnedMsgs[idx].Message = msg
}
}
return pinnedMsgs, cursor, nil
}
func (m *Messenger) SavePinMessages(messages []*common.PinMessage) error {
return m.persistence.SavePinMessages(messages)
}
func generatePinMessageID(pubKey *ecdsa.PublicKey, pm *common.PinMessage, chat *Chat) (string, error) {
data, err := pinMessageBaseID(pubKey, pm, chat)
if err != nil {
return "", err
}
id := sha256.Sum256(data)
idString := fmt.Sprintf("%x", id)
return idString, nil
}
func pinMessageBaseID(pubKey *ecdsa.PublicKey, pm *common.PinMessage, chat *Chat) ([]byte, error) {
data := gethcommon.FromHex(pm.MessageId)
switch {
case chat.ChatType == ChatTypeOneToOne:
ourPubKey := crypto.FromECDSAPub(pubKey)
tmpPubKey, err := chat.PublicKey()
if err != nil {
return nil, err
}
theirPubKey := crypto.FromECDSAPub(tmpPubKey)
if bytes.Compare(ourPubKey, theirPubKey) < 0 {
data = append(data, ourPubKey...) // our key
data = append(data, theirPubKey...) // their key
} else {
data = append(data, theirPubKey...) // their key
data = append(data, ourPubKey...) // our key
}
default:
data = append(data, []byte(chat.ID)...)
}
return data, nil
}
func generatePinMessageNotificationID(pubKey *ecdsa.PublicKey, pm *common.PinMessage, chat *Chat) (string, error) {
data, err := pinMessageBaseID(pubKey, pm, chat)
if err != nil {
return "", err
}
clockBytes := make([]byte, 8)
binary.LittleEndian.PutUint64(clockBytes, pm.Clock)
data = append(data, clockBytes...)
id := sha256.Sum256(data)
idString := fmt.Sprintf("%x", id)
return idString, nil
}