status-im/status-go

package protocol

import (
    "crypto/ecdsa"

    "github.com/status-im/status-go/eth-node/crypto"
    "github.com/status-im/status-go/eth-node/types"
    "github.com/status-im/status-go/protocol/common"
    "github.com/status-im/status-go/protocol/identity/alias"
    "github.com/status-im/status-go/protocol/identity/identicon"
)

func extendMessageFromChat(message *common.Message, chat *Chat, key *ecdsa.PublicKey, timesource common.TimeSource) error {
    clock, timestamp := chat.NextClockAndTimestamp(timesource)

    message.LocalChatID = chat.ID
    message.Clock = clock
    message.Timestamp = timestamp
    message.From = types.EncodeHex(crypto.FromECDSAPub(key))
    message.SigPubKey = key
    message.WhisperTimestamp = timestamp
    message.Seen = true
    message.OutgoingStatus = common.OutgoingStatusSending

    identicon, err := identicon.GenerateBase64(message.From)
    if err != nil {
        return err
    }

    message.Identicon = identicon

    alias, err := alias.GenerateFromPublicKeyString(message.From)
    if err != nil {
        return err
    }

    message.Alias = alias
    return nil

}

func extendPinMessageFromChat(message *common.PinMessage, chat *Chat, key *ecdsa.PublicKey, timesource common.TimeSource) error {
    clock, timestamp := chat.NextClockAndTimestamp(timesource)

    message.LocalChatID = chat.ID
    message.Clock = clock
    message.From = types.EncodeHex(crypto.FromECDSAPub(key))
    message.SigPubKey = key
    message.WhisperTimestamp = timestamp

    identicon, err := identicon.GenerateBase64(message.From)
    if err != nil {
        return err
    }

    message.Identicon = identicon

    alias, err := alias.GenerateFromPublicKeyString(message.From)
    if err != nil {
        return err
    }

    message.Alias = alias
    return nil

}