1
0
dwelling-upload/internal/configuration/configuration.go

51 lines
1.3 KiB
Go
Raw Normal View History

2022-02-06 02:22:23 +04:00
package configuration
import (
"os"
"strings"
"github.com/pkg/errors"
"gopkg.in/yaml.v3"
)
// Configuration holds a list of process names to be tracked and a listen address.
type Configuration struct {
ListenOn string `yaml:"listen_on"`
2022-02-07 04:47:26 +04:00
HashSalt string `yaml:"hash_salt"`
WebDir string `yaml:"web_dir"`
2022-02-07 18:17:12 +04:00
LogDir string `yaml:"log_dir"`
2022-02-06 02:22:23 +04:00
Uploads struct {
Directory string `yaml:"directory"`
Limits struct {
2022-02-07 04:47:26 +04:00
FileSize int64 `yaml:"file_size"`
KeepForHours int `yaml:"keep_for_hours"`
Storage int64 `yaml:"storage"`
2022-02-06 02:22:23 +04:00
} `yaml:"limits"`
} `yaml:"uploads"`
}
func LoadConfiguration(path string) (*Configuration, error) {
2022-02-06 03:01:16 +04:00
configFile, err := os.Open(path)
2022-02-06 02:22:23 +04:00
if err != nil {
return nil, errors.Wrap(err, "failed to open configuration file")
}
defer configFile.Close()
2022-02-06 02:22:23 +04:00
config := &Configuration{}
2022-02-06 03:01:16 +04:00
if err := yaml.NewDecoder(configFile).Decode(config); err != nil {
return nil, errors.Wrap(err, "failed to parse configuration file")
2022-02-06 02:22:23 +04:00
}
return config, nil
}
// SplitNetworkAddress splits ListenOn option and returns as two strings
// network type (e.g. tcp, unix, udp) and address:port or /path/to/prog.socket
// to listen on.
2022-02-06 02:22:23 +04:00
func (c *Configuration) SplitNetworkAddress() (n string, a string) {
s := strings.Split(c.ListenOn, " ")
n, a = s[0], s[1]
return n, a
}