1
0
httpprocprobed/configuration.go

109 lines
2.6 KiB
Go

package main
import (
"bufio"
"io/ioutil"
"log"
"os"
"strconv"
"strings"
)
// Configuration holds a list of process names to be tracked and a listen address.
type Configuration struct {
ListenAddress string
IndentedOutput bool
Processes []string
}
// LoadConfiguration loads configuration from a file.
func LoadConfiguration(path string) (conf *Configuration, err error) {
conf = &Configuration{}
file, err := os.Open(path)
if err != nil {
return nil, err
}
defer file.Close()
s := bufio.NewScanner(file)
s.Split(bufio.ScanLines)
for s.Scan() {
kv := strings.Split(s.Text(), " = ")
switch kv[0] {
case "listen_address":
conf.ListenAddress = kv[1]
case "indented_output":
v, err := strconv.ParseBool(kv[1])
if err != nil {
log.Printf("WARN: could not parse \"indented_output\", valid values are true or false. Defaulted to false.\n")
}
conf.IndentedOutput = v
case "processes":
if kv[1] != "" {
conf.Processes = strings.Split(kv[1], " ")
} else {
log.Printf("WARN: \"processes\" list is empty.\n")
conf.Processes = []string{}
}
}
}
return conf, nil
}
// StoreConfiguration writes Configuration into a file.
func (conf *Configuration) StoreConfiguration(path string) (err error) {
var config strings.Builder
config.WriteString("listen_address = ")
config.WriteString(conf.ListenAddress)
config.WriteByte('\n')
config.WriteString("indented_output = ")
config.WriteString(strconv.FormatBool(conf.IndentedOutput))
config.WriteByte('\n')
config.WriteString("processes = ")
config.WriteString(strings.Join(conf.Processes, " "))
if err := ioutil.WriteFile(path, []byte(config.String()), 0644); err != nil {
return err
}
return nil
}
// AddProcessToList appends a new given process into a configuration file.
func (conf *Configuration) AddProcessToList(process string, configPath string) error {
for _, v := range conf.Processes {
if v == process {
return ErrIsOnList
}
}
conf.Processes = append(conf.Processes, process)
if err := conf.StoreConfiguration(configPath); err != nil {
return err
}
return nil
}
// RemoveProcessFromList removes a given process from a configuration file.
func (conf *Configuration) RemoveProcessFromList(process string, configPath string) error {
for k, v := range conf.Processes {
if v == process {
newlist := make([]string, len(conf.Processes)-1)
newlist = append(conf.Processes[:k], conf.Processes[k+1:]...)
conf.Processes = newlist
if err := conf.StoreConfiguration(configPath); err != nil {
return err
}
return nil
}
}
return ErrNotFound
}