mirror of
https://github.com/cupcakearmy/autorestic.git
synced 2024-08-28 23:27:18 +02:00
116 lines
2.4 KiB
Go
116 lines
2.4 KiB
Go
package internal
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"path"
|
|
"strings"
|
|
"sync"
|
|
|
|
"github.com/mitchellh/go-homedir"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
const VERSION = "1.0.0"
|
|
|
|
type Config struct {
|
|
Locations map[string]Location `mapstructure:"locations"`
|
|
Backends map[string]Backend `mapstructure:"backends"`
|
|
}
|
|
|
|
var once sync.Once
|
|
var config *Config
|
|
|
|
func GetConfig() *Config {
|
|
if config == nil {
|
|
once.Do(func() {
|
|
config = &Config{}
|
|
if err := viper.UnmarshalExact(config); err != nil {
|
|
log.Fatal("Nope ", err)
|
|
}
|
|
})
|
|
}
|
|
return config
|
|
}
|
|
|
|
func GetPathRelativeToConfig(p string) string {
|
|
if path.IsAbs(p) {
|
|
return p
|
|
} else if strings.HasPrefix(p, "~") {
|
|
home, err := homedir.Dir()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return path.Join(home, strings.TrimPrefix(p, "~"))
|
|
} else {
|
|
return path.Join(path.Dir(viper.ConfigFileUsed()), p)
|
|
}
|
|
}
|
|
|
|
func (c Config) CheckConfig() error {
|
|
for name, backend := range c.Backends {
|
|
if err := backend.validate(); err != nil {
|
|
return fmt.Errorf("backend \"%s\": %s", name, err)
|
|
}
|
|
}
|
|
for name, location := range c.Locations {
|
|
if err := location.validate(c); err != nil {
|
|
return fmt.Errorf("location \"%s\": %s", name, err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func GetAllOrSelected(cmd *cobra.Command, backends bool) ([]string, error) {
|
|
var list []string
|
|
if backends {
|
|
for key := range config.Backends {
|
|
list = append(list, key)
|
|
}
|
|
} else {
|
|
for key := range config.Locations {
|
|
list = append(list, key)
|
|
}
|
|
}
|
|
all, _ := cmd.Flags().GetBool("all")
|
|
if all {
|
|
return list, nil
|
|
} else {
|
|
var selected []string
|
|
if backends {
|
|
tmp, _ := cmd.Flags().GetStringSlice("backend")
|
|
selected = tmp
|
|
} else {
|
|
tmp, _ := cmd.Flags().GetStringSlice("location")
|
|
selected = tmp
|
|
}
|
|
for _, s := range selected {
|
|
found := false
|
|
for _, l := range list {
|
|
if l == s {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
if !found {
|
|
if backends {
|
|
return nil, fmt.Errorf("invalid backend \"%s\"", s)
|
|
} else {
|
|
return nil, fmt.Errorf("invalid location \"%s\"", s)
|
|
}
|
|
}
|
|
}
|
|
return selected, nil
|
|
}
|
|
}
|
|
|
|
func AddFlagsToCommand(cmd *cobra.Command, backend bool) {
|
|
cmd.PersistentFlags().BoolP("all", "a", false, "Backup all locations")
|
|
if backend {
|
|
cmd.PersistentFlags().StringSliceP("backend", "b", []string{}, "backends")
|
|
} else {
|
|
cmd.PersistentFlags().StringSliceP("location", "l", []string{}, "Locations")
|
|
}
|
|
}
|