// Config: parse hosts.yaml // SPDX-License-Identifier: GPL-3.0-or-later // Copyright (C) 2021-2024 Simon Ruderich package config import ( "fmt" "os" "strings" "gopkg.in/yaml.v2" ) type Hosts struct { List []*Host Map map[string]*Host } type Host struct { Name string `yaml:"name"` SshUser string `yaml:"ssh_user"` } func LoadHosts() (*Hosts, error) { const path = "hosts.yaml" var hostList []*Host x, err := os.ReadFile(path) if err != nil { return nil, err } err = yaml.UnmarshalStrict(x, &hostList) if err != nil { return nil, fmt.Errorf("%s: failed to load: %v", path, err) } hostMap := make(map[string]*Host) for _, x := range hostList { errPrefix := fmt.Sprintf("%s: host %q:", path, x.Name) if x.Name == GroupAll { return nil, fmt.Errorf( "%s conflict with pre-defined group %q", errPrefix, x.Name) } if strings.HasPrefix(x.Name, GroupDetectedPrefix) { return nil, fmt.Errorf( "%s name must not start with %q "+ "(reserved for detected groups)", errPrefix, GroupDetectedPrefix) } if strings.Contains(x.Name, GroupSpecialSeparator) { return nil, fmt.Errorf( "%s name must not contain %q", errPrefix, GroupSpecialSeparator) } if hostMap[x.Name] != nil { return nil, fmt.Errorf("%s host name already exists", errPrefix) } hostMap[x.Name] = x } return &Hosts{ List: hostList, Map: hostMap, }, nil }