2016-11-01 09:56:12 +00:00
|
|
|
package settings
|
|
|
|
|
|
|
|
import (
|
2016-12-15 06:06:24 +00:00
|
|
|
"os"
|
|
|
|
|
2016-11-01 09:56:12 +00:00
|
|
|
"gopkg.in/yaml.v2"
|
|
|
|
)
|
|
|
|
|
|
|
|
type Settings struct {
|
|
|
|
Colors `yaml:"-"`
|
2016-11-20 17:19:09 +00:00
|
|
|
Resources `yaml:"resources" json:"resources"`
|
|
|
|
Server `yaml:"server" json:"server"`
|
|
|
|
Config `yaml:"config" json:"config"`
|
2016-11-01 09:56:12 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
type Config struct {
|
2016-11-20 16:58:52 +00:00
|
|
|
Flimit uint64 `yaml:"flimit" json:"flimit"`
|
2016-11-01 09:56:12 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
type Server struct {
|
2016-11-20 16:58:52 +00:00
|
|
|
Enabled bool `yaml:"enable" json:"enable"`
|
|
|
|
Open bool `yaml:"open" json:"open"`
|
|
|
|
Host string `yaml:"host" json:"host"`
|
|
|
|
Port int `yaml:"port" json:"port"`
|
2016-11-01 09:56:12 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
type Resources struct {
|
2016-11-20 16:58:52 +00:00
|
|
|
Config string `yaml:"-" json:"-"`
|
|
|
|
Output string `yaml:"output" json:"output"`
|
|
|
|
Log string `yaml:"log" json:"log"`
|
2016-11-01 09:56:12 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Read from the configuration file
|
|
|
|
func (s *Settings) Read(out interface{}) error {
|
2016-12-15 06:06:24 +00:00
|
|
|
localConfigPath := s.Resources.Config
|
|
|
|
if _, err := os.Stat(".realize/"); err == nil {
|
|
|
|
localConfigPath = ".realize/" + s.Resources.Config
|
|
|
|
}
|
|
|
|
content, err := s.Stream(localConfigPath)
|
2016-11-01 09:56:12 +00:00
|
|
|
if err == nil {
|
|
|
|
err = yaml.Unmarshal(content, out)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
// Record create and unmarshal the yaml config file
|
2016-12-15 06:06:24 +00:00
|
|
|
func (s *Settings) Record(out interface{}) error {
|
2016-11-01 09:56:12 +00:00
|
|
|
y, err := yaml.Marshal(out)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2016-12-15 06:06:24 +00:00
|
|
|
if _, err := os.Stat(".realize/"); os.IsNotExist(err) {
|
|
|
|
os.Mkdir(".realize", 0770)
|
|
|
|
}
|
|
|
|
return s.Write(".realize/"+s.Resources.Config, y)
|
2016-11-01 09:56:12 +00:00
|
|
|
}
|