61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
package state
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net/netip"
|
|
|
|
"github.com/google/uuid"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
// Config contains all MTD services and cloud provider configs
|
|
type Config struct {
|
|
MTD mtdconf `yaml:"mtd"`
|
|
AWS aws `yaml:"aws"`
|
|
}
|
|
|
|
type mtdconf struct {
|
|
Services []service `yaml:"services"`
|
|
}
|
|
|
|
type service struct {
|
|
ID customUUID `yaml:"id"`
|
|
ServiceID string `yaml:"cloud_id"`
|
|
EntryIP netip.Addr `yaml:"entry_ip"`
|
|
EntryPort uint16 `yaml:"entry_port"`
|
|
ServiceIP netip.Addr `yaml:"service_ip"`
|
|
ServicePort uint16 `yaml:"service_port"`
|
|
}
|
|
|
|
type customUUID uuid.UUID
|
|
|
|
type aws struct {
|
|
Regions []string `yaml:"regions"`
|
|
CredentialsPath string `yaml:"credentials_path"`
|
|
}
|
|
|
|
func (u *customUUID) UnmarshalYAML(value *yaml.Node) error {
|
|
id, err := uuid.Parse(value.Value)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*u = customUUID(id)
|
|
return nil
|
|
}
|
|
|
|
// LoadConf loads config from a yaml file
|
|
func LoadConf(filename string) (Config, error) {
|
|
var config Config
|
|
|
|
data, err := ioutil.ReadFile(filename)
|
|
if err != nil {
|
|
return config, err
|
|
}
|
|
|
|
err = yaml.Unmarshal([]byte(data), &config)
|
|
if err != nil {
|
|
return config, err
|
|
}
|
|
|
|
return config, nil
|
|
} |