mirror of https://github.com/status-im/go-waku.git
56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
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, ", ")
|
|
}
|