lnd+server: Allow configurable Node Alias and Color
This commit sets the Node's Alias and Color to the values set in config.
This commit is contained in:
parent
a40ee8fe8a
commit
92ac81b3b0
43
server.go
43
server.go
@ -220,14 +220,19 @@ func newServer(listenAddrs []string, chanDB *channeldb.DB, cc *chainControl,
|
|||||||
|
|
||||||
chanGraph := chanDB.ChannelGraph()
|
chanGraph := chanDB.ChannelGraph()
|
||||||
|
|
||||||
defaultColor := color.RGBA{ // #3399FF
|
// Parse node color from configuration.
|
||||||
R: 51,
|
color, err := parseHexColor(cfg.Color)
|
||||||
G: 153,
|
if err != nil {
|
||||||
B: 255,
|
srvrLog.Errorf("unable to parse color: %v\n", err)
|
||||||
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO(roasbeef): make alias configurable
|
// If no alias is provided, default to first 10 characters of public key
|
||||||
alias, err := lnwire.NewNodeAlias(hex.EncodeToString(serializedPubKey[:10]))
|
alias := cfg.Alias
|
||||||
|
if alias == "" {
|
||||||
|
alias = hex.EncodeToString(serializedPubKey[:10])
|
||||||
|
}
|
||||||
|
nodeAlias, err := lnwire.NewNodeAlias(alias)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@ -236,9 +241,9 @@ func newServer(listenAddrs []string, chanDB *channeldb.DB, cc *chainControl,
|
|||||||
LastUpdate: time.Now(),
|
LastUpdate: time.Now(),
|
||||||
Addresses: selfAddrs,
|
Addresses: selfAddrs,
|
||||||
PubKey: privKey.PubKey(),
|
PubKey: privKey.PubKey(),
|
||||||
Alias: alias.String(),
|
Alias: nodeAlias.String(),
|
||||||
Features: s.globalFeatures,
|
Features: s.globalFeatures,
|
||||||
Color: defaultColor,
|
Color: color,
|
||||||
}
|
}
|
||||||
|
|
||||||
// If our information has changed since our last boot, then we'll
|
// If our information has changed since our last boot, then we'll
|
||||||
@ -250,9 +255,9 @@ func newServer(listenAddrs []string, chanDB *channeldb.DB, cc *chainControl,
|
|||||||
Timestamp: uint32(selfNode.LastUpdate.Unix()),
|
Timestamp: uint32(selfNode.LastUpdate.Unix()),
|
||||||
Addresses: selfNode.Addresses,
|
Addresses: selfNode.Addresses,
|
||||||
NodeID: selfNode.PubKey,
|
NodeID: selfNode.PubKey,
|
||||||
Alias: alias,
|
Alias: nodeAlias,
|
||||||
Features: selfNode.Features.RawFeatureVector,
|
Features: selfNode.Features.RawFeatureVector,
|
||||||
RGBColor: defaultColor,
|
RGBColor: color,
|
||||||
}
|
}
|
||||||
selfNode.AuthSig, err = discovery.SignAnnouncement(s.nodeSigner,
|
selfNode.AuthSig, err = discovery.SignAnnouncement(s.nodeSigner,
|
||||||
s.identityPriv.PubKey(), nodeAnn,
|
s.identityPriv.PubKey(), nodeAnn,
|
||||||
@ -1703,3 +1708,21 @@ func (s *server) Peers() []*peer {
|
|||||||
|
|
||||||
return peers
|
return peers
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// parseHexColor takes a hex string representation of a color in the
|
||||||
|
// form "#RRGGBB", parses the hex color values, and returns a color.RGBA
|
||||||
|
// struct of the same color.
|
||||||
|
func parseHexColor(colorStr string) (color.RGBA, error) {
|
||||||
|
if len(colorStr) != 7 || colorStr[0] != '#' {
|
||||||
|
return color.RGBA{}, errors.New("Color must be in format #RRGGBB")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Decode the hex color string to bytes.
|
||||||
|
// The resulting byte array is in the form [R, G, B].
|
||||||
|
colorBytes, err := hex.DecodeString(colorStr[1:])
|
||||||
|
if err != nil {
|
||||||
|
return color.RGBA{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return color.RGBA{R: colorBytes[0], G: colorBytes[1], B: colorBytes[2]}, nil
|
||||||
|
}
|
||||||
|
34
server_test.go
Normal file
34
server_test.go
Normal file
@ -0,0 +1,34 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
func TestParseHexColor(t *testing.T) {
|
||||||
|
empty := ""
|
||||||
|
color, err := parseHexColor(empty)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatalf("Empty color string should return error, but did not")
|
||||||
|
}
|
||||||
|
|
||||||
|
tooLong := "#1234567"
|
||||||
|
color, err = parseHexColor(tooLong)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatalf("Invalid color string %s should return error, but did not",
|
||||||
|
tooLong)
|
||||||
|
}
|
||||||
|
|
||||||
|
invalidFormat := "$123456"
|
||||||
|
color, err = parseHexColor(invalidFormat)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatalf("Invalid color string %s should return error, but did not",
|
||||||
|
invalidFormat)
|
||||||
|
}
|
||||||
|
|
||||||
|
valid := "#C0FfeE"
|
||||||
|
color, err = parseHexColor(valid)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Color %s valid to parse: %s", valid, err)
|
||||||
|
}
|
||||||
|
if color.R != 0xc0 || color.G != 0xff || color.B != 0xee {
|
||||||
|
t.Fatalf("Color %s incorrectly parsed as %v", valid, color)
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user