You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
7 years ago
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"github.com/caarlos0/env"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
func DefaultConfig() Config {
|
||
|
|
return Config{
|
||
|
|
DB: "./test.db",
|
||
|
|
APIkey: "thisIsNotASecretTokenNow",
|
||
|
|
Host: "localhost",
|
||
|
|
Port: "6889",
|
||
|
|
Title: "Sgot",
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type Config struct {
|
||
|
|
DB string `env:"SHRT_DB_FILE"`
|
||
|
|
APIkey string `env:"SHRT_API_KEY"`
|
||
|
|
Host string `env:"SHRT_HOST"`
|
||
|
|
Port string `env:"SHRT_PORT"`
|
||
|
|
Domain string `env:"SHRT_DOMAIN"`
|
||
|
|
Title string `env:"SHRT_TITLE"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c Config) HostPort() string {
|
||
|
|
return c.Host + ":" + c.Port
|
||
|
|
}
|
||
|
|
|
||
|
|
func ReadConfig() *Config {
|
||
|
|
c, err := readConfig(flag.NewFlagSet(os.Args[0], flag.ExitOnError), os.Args[1:])
|
||
|
|
if err != nil {
|
||
|
|
// sould never happen, because of flag default policy ExitOnError
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
return c
|
||
|
|
}
|
||
|
|
|
||
|
|
func readConfig(f *flag.FlagSet, args []string) (*Config, error){
|
||
|
|
config := DefaultConfig()
|
||
|
|
|
||
|
|
// Environment variables
|
||
|
|
err := env.Parse(&config)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
f.StringVar(&config.Host, "host", config.Host, "The host to listen on")
|
||
|
|
f.StringVar(&config.Port, "port", config.Port, "The port to listen on")
|
||
|
|
f.StringVar(&config.APIkey, "apikey", config.APIkey, "The Key to connect to the API")
|
||
|
|
f.StringVar(&config.DB, "db-file", config.DB, "The db file to use")
|
||
|
|
f.StringVar(&config.Domain, "domain", config.Domain, "The domain for redirect links")
|
||
|
|
f.StringVar(&config.Title, "title", config.Title, "The title on the Front")
|
||
|
|
|
||
|
|
// Arguments variables
|
||
|
|
err = f.Parse(args)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return &config, err
|
||
|
|
}
|