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

57 lines
1.5 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"`
User string `yaml:"user"`
Log struct {
ToStdout bool `yaml:"stdout"`
Error string `yaml:"error"`
Upload string `yaml:"upload"`
Download string `yaml:"download"`
Clean string `yaml:"clean"`
CleanError string `yaml:"clean_error"`
} `yaml:"log"`
Uploads struct {
2022-02-06 02:22:23 +04:00
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.
func (c *Configuration) SplitNetworkAddress() (string, string) {
2022-02-06 02:22:23 +04:00
s := strings.Split(c.ListenOn, " ")
return s[0], s[1]
2022-02-06 02:22:23 +04:00
}