59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
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"`
|
|
HashSalt string `yaml:"hash_salt"`
|
|
User string `yaml:"user"`
|
|
Chroot string `yaml:"chroot"`
|
|
Log struct {
|
|
ToStdout bool `yaml:"stdout"`
|
|
Error string `yaml:"error"`
|
|
Upload string `yaml:"upload"`
|
|
Download string `yaml:"download"`
|
|
Delete string `yaml:"delete"`
|
|
Clean string `yaml:"clean"`
|
|
CleanError string `yaml:"clean_error"`
|
|
} `yaml:"log"`
|
|
Uploads struct {
|
|
Directory string `yaml:"directory"`
|
|
Limits struct {
|
|
FileSize int64 `yaml:"file_size"`
|
|
KeepForHours int `yaml:"keep_for_hours"`
|
|
Storage int64 `yaml:"storage"`
|
|
} `yaml:"limits"`
|
|
} `yaml:"uploads"`
|
|
}
|
|
|
|
func LoadConfiguration(path string) (*Configuration, error) {
|
|
configFile, err := os.Open(path)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to open configuration file")
|
|
}
|
|
defer configFile.Close()
|
|
|
|
config := &Configuration{}
|
|
|
|
if err := yaml.NewDecoder(configFile).Decode(config); err != nil {
|
|
return nil, errors.Wrap(err, "failed to parse configuration file")
|
|
}
|
|
|
|
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) {
|
|
s := strings.Split(c.ListenOn, " ")
|
|
return s[0], s[1]
|
|
}
|