go-waku/waku/cliutils/protected_topic.go

56 lines
1.1 KiB
Go
Raw Normal View History

package cliutils
import (
"crypto/ecdsa"
"encoding/hex"
"errors"
"fmt"
"strings"
"github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/crypto"
)
type ProtectedTopic struct {
Topic string
PublicKey *ecdsa.PublicKey
}
func (p ProtectedTopic) String() string {
pubKBytes := crypto.FromECDSAPub(p.PublicKey)
return fmt.Sprintf("%s:%s", p.Topic, hex.EncodeToString(pubKBytes))
}
type ProtectedTopicSlice struct {
Values *[]ProtectedTopic
}
func (k *ProtectedTopicSlice) Set(value string) error {
protectedTopicParts := strings.Split(value, ":")
if len(protectedTopicParts) != 2 {
return errors.New("expected topic_name:hex_encoded_public_key")
}
pubk, err := crypto.UnmarshalPubkey(common.FromHex(protectedTopicParts[1]))
if err != nil {
return err
}
*k.Values = append(*k.Values, ProtectedTopic{
Topic: protectedTopicParts[0],
PublicKey: pubk,
})
return nil
}
func (v *ProtectedTopicSlice) String() string {
if v.Values == nil {
return ""
}
var output []string
for _, v := range *v.Values {
output = append(output, v.String())
}
return strings.Join(output, ", ")
}