You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
2.4 KiB
82 lines
2.4 KiB
package util
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
envstruct "github.com/mxschmitt/golang-env-struct"
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/pkg/errors"
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
// Configuration are the available config values
|
|
type Configuration struct {
|
|
ListenAddr string `yaml:"ListenAddr" env:"LISTEN_ADDR"`
|
|
BaseURL string `yaml:"BaseURL" env:"BASE_URL"`
|
|
DataDir string `yaml:"DataDir" env:"DATA_DIR"`
|
|
UseSSL bool `yaml:"EnableSSL" env:"USE_SSL"`
|
|
EnableDebugMode bool `yaml:"EnableDebugMode" env:"ENABLE_DEBUG_MODE"`
|
|
ShortedIDLength int `yaml:"ShortedIDLength" env:"SHORTED_ID_LENGTH"`
|
|
Google oAuthConf `yaml:"Google" env:"GOOGLE"`
|
|
GitHub oAuthConf `yaml:"GitHub" env:"GITHUB"`
|
|
Microsoft oAuthConf `yaml:"Microsoft" env:"MICROSOFT"`
|
|
}
|
|
|
|
type oAuthConf struct {
|
|
ClientID string `yaml:"ClientID" env:"CLIENT_ID"`
|
|
ClientSecret string `yaml:"ClientSecret" env:"CLIENT_SECRET"`
|
|
}
|
|
|
|
// config contains the default values
|
|
var config = Configuration{
|
|
ListenAddr: ":8080",
|
|
BaseURL: "http://localhost:3000",
|
|
DataDir: "data",
|
|
EnableDebugMode: false,
|
|
UseSSL: false,
|
|
ShortedIDLength: 4,
|
|
}
|
|
|
|
// ReadInConfig loads the Configuration and other needed folders for further usage
|
|
func ReadInConfig() error {
|
|
file, err := ioutil.ReadFile("config.yaml")
|
|
if err == nil {
|
|
if err := yaml.Unmarshal(file, &config); err != nil {
|
|
return errors.Wrap(err, "could not unmarshal yaml file")
|
|
}
|
|
} else if !os.IsNotExist(err) {
|
|
return errors.Wrap(err, "could not read config file")
|
|
} else {
|
|
logrus.Info("No configuration file found, using defaults with environment variable overrides.")
|
|
}
|
|
if err := envstruct.ApplyEnvVars(&config, "GUS"); err != nil {
|
|
return errors.Wrap(err, "could not apply environment configuration")
|
|
}
|
|
config.DataDir, err = filepath.Abs(config.DataDir)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not get relative data dir path")
|
|
}
|
|
if _, err = os.Stat(config.DataDir); os.IsNotExist(err) {
|
|
if err = os.MkdirAll(config.DataDir, 0755); err != nil {
|
|
return errors.Wrap(err, "could not create config directory")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (o oAuthConf) Enabled() bool {
|
|
return o.ClientSecret != ""
|
|
}
|
|
|
|
// GetConfig returns the configuration from the memory
|
|
func GetConfig() Configuration {
|
|
return config
|
|
}
|
|
|
|
// SetConfig sets the configuration into the memory
|
|
func SetConfig(c Configuration) {
|
|
config = c
|
|
}
|
|
|