func addOrSetCmd(r ring.Ring, b *ring.Builder, args []string, n ring.BuilderNode) error {
if r != nil {
return fmt.Errorf("cannot add a node to ring; use with a builder instead")
}
active := true
capacity := uint32(1)
var tiers []string
var addresses []string
meta := ""
for _, arg := range args {
sarg := strings.SplitN(arg, "=", 2)
if len(sarg) != 2 {
return fmt.Errorf(`invalid expression %#v; needs "="`, arg)
}
if sarg[0] == "" {
return fmt.Errorf(`invalid expression %#v; nothing was left of "="`, arg)
}
if sarg[1] == "" {
return fmt.Errorf(`invalid expression %#v; nothing was right of "="`, arg)
}
switch sarg[0] {
case "active":
switch sarg[1] {
case "true":
active = true
case "false":
active = false
default:
return fmt.Errorf(`invalid expression %#v; use "true" or "false" for the value of active`, arg)
}
if n != nil {
n.SetActive(active)
}
case "capacity":
c, err := strconv.Atoi(sarg[1])
if err != nil {
return fmt.Errorf("invalid expression %#v; %s", arg, err)
}
if c < 0 {
return fmt.Errorf("invalid expression %#v; min is 0", arg)
}
if c > math.MaxUint32 {
return fmt.Errorf("invalid expression %#v; max is %d", arg, math.MaxUint32)
}
capacity = uint32(c)
if n != nil {
n.SetCapacity(capacity)
}
case "meta":
meta = sarg[1]
if n != nil {
n.SetMeta(meta)
}
default:
if strings.HasPrefix(sarg[0], "tier") {
level, err := strconv.Atoi(sarg[0][4:])
if err != nil {
return fmt.Errorf("invalid expression %#v; %#v doesn't specify a number", arg, sarg[0][4:])
}
if level < 0 {
return fmt.Errorf("invalid expression %#v; minimum level is 0", arg)
}
if len(tiers) <= level {
t := make([]string, level+1)
copy(t, tiers)
tiers = t
}
tiers[level] = sarg[1]
if n != nil {
n.SetTier(level, tiers[level])
}
} else if strings.HasPrefix(sarg[0], "address") {
index, err := strconv.Atoi(sarg[0][7:])
if err != nil {
return fmt.Errorf("invalid expression %#v; %#v doesn't specify a number", arg, sarg[0][4:])
}
if index < 0 {
return fmt.Errorf("invalid expression %#v; minimum index is 0", arg)
}
if len(addresses) <= index {
a := make([]string, index+1)
copy(a, addresses)
addresses = a
}
addresses[index] = sarg[1]
if n != nil {
n.SetAddress(index, addresses[index])
}
}
}
}
if n == nil {
n = b.AddNode(active, capacity, tiers, addresses, meta)
report := [][]string{
[]string{"ID:", fmt.Sprintf("%016x", n.ID())},
[]string{"Active:", fmt.Sprintf("%v", n.Active())},
[]string{"Capacity:", fmt.Sprintf("%d", n.Capacity())},
[]string{"Tiers:", strings.Join(n.Tiers(), "\n")},
[]string{"Addresses:", strings.Join(n.Addresses(), "\n")},
[]string{"Meta:", n.Meta()},
}
fmt.Print(brimtext.Align(report, nil))
}
return nil
}
