autorestic/internal/config.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")
	}
}