mirror of https://github.com/qwc/backive.git
82 lines
2.4 KiB
Go
82 lines
2.4 KiB
Go
package backive
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/fsnotify/fsnotify"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// Configuration struct holding the settings and config items of devices and backups
|
|
type Configuration struct {
|
|
Settings Settings `mapstructure:"settings"`
|
|
Devices Devices `mapstructure:"devices"`
|
|
Backups Backups `mapstructure:"backups"`
|
|
Vconfig *viper.Viper
|
|
}
|
|
|
|
// Settings struct holds the global configuration items
|
|
type Settings struct {
|
|
SystemMountPoint string `mapstructure:"systemMountPoint"`
|
|
UserMountPoint string `mapstructure:"userMountPoint"`
|
|
UnixSocketLocation string `mapstructure:"unixSocketLocation"`
|
|
LogLocation string `mapstructure:"logLocation"`
|
|
DbLocation string `mapstructure:"dbLocation"`
|
|
}
|
|
|
|
// CreateViper creates a viper instance for usage later
|
|
func (c *Configuration) CreateViper() {
|
|
if c.Vconfig == nil {
|
|
vconfig := viper.New()
|
|
// vconfig.Debug()
|
|
vconfig.SetConfigName("backive")
|
|
// do not set config file explicitly or viper doesnt search for it, and /etc search fails
|
|
//vconfig.SetConfigFile("backive.yml")
|
|
//vconfig.SetConfigFile("backive.yaml")
|
|
vconfig.SetConfigType("yaml")
|
|
//vconfig.AddConfigPath("$HOME/.backive/")
|
|
vconfig.AddConfigPath(".") // backup config in local dir
|
|
vconfig.AddConfigPath("/etc/backive/") // system config
|
|
vconfig.OnConfigChange(func(e fsnotify.Event) {
|
|
log.Printf("Event: %s", e)
|
|
if e.Op == fsnotify.Write {
|
|
log.Printf("Reloading %s", e.Name)
|
|
c.Load()
|
|
}
|
|
})
|
|
vconfig.WatchConfig()
|
|
c.Vconfig = vconfig
|
|
}
|
|
}
|
|
|
|
// Load loads the configuration from the disk
|
|
func (c *Configuration) Load() {
|
|
c.CreateViper()
|
|
vc := c.Vconfig
|
|
if err := vc.ReadInConfig(); err != nil {
|
|
if _, ok := err.(viper.ConfigFileNotFoundError); ok {
|
|
panic(fmt.Errorf("fatal: No config file could be found: %w", err))
|
|
}
|
|
panic(fmt.Errorf("fatal error config file: %w ", err))
|
|
}
|
|
log.Printf("Configuration file used: %s", vc.ConfigFileUsed())
|
|
|
|
//Unmarshal all into Configuration type
|
|
err := vc.Unmarshal(c)
|
|
if err != nil {
|
|
fmt.Printf("Error occured when loading config: %v\n", err)
|
|
panic("No configuration available!")
|
|
}
|
|
for k, v := range c.Backups {
|
|
log.Printf("Initializing backup '%s'\n", k)
|
|
v.Name = k
|
|
log.Printf("Initialized backup '%s'\n", v.Name)
|
|
}
|
|
for k, v := range c.Devices {
|
|
log.Printf("Initializing device '%s'\n", k)
|
|
v.Name = k
|
|
log.Printf("Initialized device '%s'\n", v.Name)
|
|
}
|
|
}
|