status-im/status-go

package json

import (
    "reflect"

    "github.com/status-im/status-go/api/multiformat"
    "github.com/status-im/status-go/protocol/identity/emojihash"
)

type PublicKeyData struct {
    CompressedKey string   `json:"compressedKey"`
    EmojiHash     []string `json:"emojiHash"`
}

func getPubKeyData(publicKey string) (*PublicKeyData, error) {
    compressedKey, err := multiformat.SerializeLegacyKey(publicKey)
    if err != nil {
        return nil, err
    }

    emojiHash, err := emojihash.GenerateFor(publicKey)
    if err != nil {
        return nil, err
    }

    return &PublicKeyData{compressedKey, emojiHash}, nil

}

func ExtendStructWithPubKeyData(publicKey string, item any) (any, error) {
    // If the public key is empty, do not attempt to extend the incoming item
    if publicKey == "" {
        return item, nil
    }

    pkd, err := getPubKeyData(publicKey)
    if err != nil {
        return nil, err
    }

    // Create a struct with 2 embedded substruct fields in order to circumvent
    // "embedded field type cannot be a (pointer to a) type parameter"
    // compiler error that arises if we were to use a generic function instead
    typ := reflect.StructOf([]reflect.StructField{
        {
            Name:      "Item",
            Anonymous: true,
            Type:      reflect.TypeOf(item),
        },
        {
            Name:      "Pkd",
            Anonymous: true,
            Type:      reflect.TypeOf(pkd),
        },
    })

    v := reflect.New(typ).Elem()
    v.Field(0).Set(reflect.ValueOf(item))
    v.Field(1).Set(reflect.ValueOf(pkd))
    s := v.Addr().Interface()

    return s, nil
}