config.go

// Package config provides a type that represent configuration
package main

import (
	"errors"
	flag "github.com/spf13/pflag"
	"github.com/spf13/viper"
	"log"
	"os"
)

// Config defines configuration
type Config struct {
	Fields     []string `yaml:"fields" mapstructure:"fields"`          // Fields to print
	Skips      []string `yaml:"skips"  mapstructure:"skips"`           // Fields to skip
	JsonFields []string `yaml:"jsonFields"  mapstructure:"jsonFields"` // Fields to print in JSON format
	Color      bool     `yaml:"color"  mapstructure:"color"`           // Enable color
}

// LoadConfig loads configuration from file, env and flags and return compiled and validated config
func LoadConfig() (*Config, error) {
	v := viper.New()

	dirname, err := os.UserHomeDir()
	if err != nil {
		log.Fatal(err)
	}

	f := flag.CommandLine
	f.StringP("config", "c", dirname+"/.logparse/config.yaml", "The configuration file to use to configure this application")
	flag.Parse()

	configFile, err := f.GetString("config")
	if err != nil {
		exit(err, 1)
	}
	if configFile == "" {
		return nil, errors.New("missing config")
	}

	v.SetConfigFile(configFile)
	if err := v.ReadInConfig(); err != nil {
		return nil, err
	}

	var config Config
	if err := v.Unmarshal(&config); err != nil {
		exit(err, 1)
	}

	return &config, nil
}