theater/tendrils
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add structured node config with names/macs/ips and avoid flag

Browse Source 
Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Ian Gulliver
2026-01-31 09:52:51 -08:00
parent 3b8005a28c
commit bb22e16460
8 changed files with 276 additions and 135 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
config.go
View File

@@ -12,10 +12,17 @@ type Config struct {
}
type Location struct {
Name string `yaml:"name" json:"name"`
Direction string `yaml:"direction,omitempty" json:"direction,omitempty"`
Nodes []string `yaml:"nodes,omitempty" json:"nodes,omitempty"`
Children []*Location `yaml:"children,omitempty" json:"children,omitempty"`
Name string `yaml:"name" json:"name"`
Direction string `yaml:"direction,omitempty" json:"direction,omitempty"`
Nodes []*NodeConfig `yaml:"nodes,omitempty" json:"nodes,omitempty"`
Children []*Location `yaml:"children,omitempty" json:"children,omitempty"`
}
type NodeConfig struct {
Names []string `yaml:"names,omitempty" json:"names,omitempty"`
MACs []string `yaml:"macs,omitempty" json:"macs,omitempty"`
IPs []string `yaml:"ips,omitempty" json:"ips,omitempty"`
Avoid bool `yaml:"avoid,omitempty" json:"avoid,omitempty"`
}
func LoadConfig(path string) (*Config, error) {
@@ -39,3 +46,16 @@ func LoadConfig(path string) (*Config, error) {
func (c *Config) ToJSON() ([]byte, error) {
return json.Marshal(c)
}
func (c *Config) AllNodeConfigs() []*NodeConfig {
var result []*NodeConfig
var visit func([]*Location)
visit = func(locs []*Location) {
for _, loc := range locs {
result = append(result, loc.Nodes...)
visit(loc.Children)
}
}
visit(c.Locations)
return result
}