166 lines
2.5 KiB
Go
166 lines
2.5 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"os"
|
|
"time"
|
|
|
|
yaml "gopkg.in/yaml.v2"
|
|
)
|
|
|
|
// Load a config from a file.
|
|
func Load(file string) (config *Config, err error) {
|
|
f, err := os.Open(file)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
defer f.Close()
|
|
|
|
config, err = Read(f)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to parse %s: %v", file, err)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// Read a config from a reader.
|
|
func Read(reader io.Reader) (config *Config, err error) {
|
|
config = &Config{}
|
|
|
|
err = yaml.NewDecoder(reader).Decode(config)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// Parse the config in data.
|
|
func Parse(data []byte) (config *Config, err error) {
|
|
config = &Config{}
|
|
|
|
err = yaml.Unmarshal(data, config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// Config represent this system's configuration
|
|
type Config struct {
|
|
Vars []VarDefault
|
|
|
|
Layers []string
|
|
Modules []string
|
|
|
|
RootUser struct {
|
|
PasswordHash string `yaml:"password_hash"`
|
|
AuthorizedKeys []string `yaml:"authorized_keys"`
|
|
} `yaml:"root_user"`
|
|
|
|
Storage StorageConfig
|
|
|
|
Mounts []MountDef
|
|
VPNs []VPNDef
|
|
|
|
Groups []GroupDef
|
|
Users []UserDef
|
|
|
|
Files []FileDef
|
|
|
|
Networks []NetworkDef
|
|
}
|
|
|
|
type VarDefault struct {
|
|
Name string
|
|
Default string
|
|
}
|
|
|
|
type StorageConfig struct {
|
|
UdevMatch string `yaml:"udev_match"`
|
|
RemoveVolumes []string `yaml:"remove_volumes"`
|
|
Volumes []VolumeDef
|
|
}
|
|
|
|
type VolumeDef struct {
|
|
Name string
|
|
Size string
|
|
Extents string
|
|
FS string
|
|
Mount struct {
|
|
Path string
|
|
Options string
|
|
}
|
|
}
|
|
|
|
type MountDef struct {
|
|
Type string
|
|
Dev string
|
|
Path string
|
|
Options string
|
|
}
|
|
|
|
type VPNDef struct {
|
|
Name string `yaml:"name"`
|
|
ListenPort *int `yaml:"port"`
|
|
IPs []string
|
|
Peers []VPNPeer
|
|
}
|
|
|
|
type VPNPeer struct {
|
|
PublicKey string `yaml:"public_key"`
|
|
WithPreSharedKey bool `yaml:"with_preshared_key"`
|
|
Endpoint *net.UDPAddr
|
|
KeepAlive time.Duration `yaml:"keepalive"`
|
|
AllowedIPs []string `yaml:"allowed_ips"`
|
|
}
|
|
|
|
type GroupDef struct {
|
|
Name string
|
|
Gid int
|
|
}
|
|
|
|
type UserDef struct {
|
|
Name string
|
|
Gid int
|
|
Uid int
|
|
}
|
|
|
|
type FileDef struct {
|
|
Path string
|
|
Mode os.FileMode
|
|
Content string
|
|
Symlink string
|
|
Dir bool
|
|
}
|
|
|
|
type NetworkDef struct {
|
|
Match struct {
|
|
All bool
|
|
Name string
|
|
Ping *struct {
|
|
Source string
|
|
Target string
|
|
Count int
|
|
Timeout int
|
|
}
|
|
}
|
|
Optional bool
|
|
Script string
|
|
}
|
|
|
|
func (c *Config) FileContent(filePath string) []byte {
|
|
for _, f := range c.Files {
|
|
if f.Path == filePath {
|
|
return []byte(f.Content)
|
|
}
|
|
}
|
|
return nil
|
|
}
|