Max Schmitt
8 years ago
14 changed files with 166 additions and 201 deletions
@ -1,17 +0,0 @@ |
|||||
{ |
|
||||
"Store": { |
|
||||
"DBPath": "main.db", |
|
||||
"ShortedIDLength": 4 |
|
||||
}, |
|
||||
"Handlers": { |
|
||||
"ListenAddr": ":8080", |
|
||||
"BaseURL": "http://localhost:3000", |
|
||||
"EnableGinDebugMode": false, |
|
||||
"OAuth": { |
|
||||
"Google": { |
|
||||
"ClientID": "", |
|
||||
"ClientSecret": "" |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
@ -0,0 +1,11 @@ |
|||||
|
http: |
||||
|
ListenAddr: ':8080' |
||||
|
BaseURL: 'http://localhost:3000' |
||||
|
General: |
||||
|
DBPath: main.db |
||||
|
EnableDebugMode: true |
||||
|
ShortedIDLength: 4 |
||||
|
oAuth: |
||||
|
Google: |
||||
|
ClientID: replace me |
||||
|
ClientSecret: replace me |
||||
@ -1,92 +0,0 @@ |
|||||
package config |
|
||||
|
|
||||
import ( |
|
||||
"encoding/json" |
|
||||
"io/ioutil" |
|
||||
"os" |
|
||||
"path/filepath" |
|
||||
|
|
||||
"github.com/pkg/errors" |
|
||||
) |
|
||||
|
|
||||
// Configuration holds all the needed parameters use
|
|
||||
// the URL Shortener
|
|
||||
type Configuration struct { |
|
||||
Store Store |
|
||||
Handlers Handlers |
|
||||
} |
|
||||
|
|
||||
// Store contains the needed fields for the Store package
|
|
||||
type Store struct { |
|
||||
DBPath string |
|
||||
ShortedIDLength uint |
|
||||
} |
|
||||
|
|
||||
// Handlers contains the needed fields for the Handlers package
|
|
||||
type Handlers struct { |
|
||||
ListenAddr string |
|
||||
BaseURL string |
|
||||
EnableDebugMode bool |
|
||||
Secret []byte |
|
||||
OAuth struct { |
|
||||
Google struct { |
|
||||
ClientID string |
|
||||
ClientSecret string |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
var ( |
|
||||
config *Configuration |
|
||||
configPath string |
|
||||
) |
|
||||
|
|
||||
// Get returns the configuration from a given file
|
|
||||
func Get() *Configuration { |
|
||||
return config |
|
||||
} |
|
||||
|
|
||||
// Preload loads the configuration file into the memory for further usage
|
|
||||
func Preload() error { |
|
||||
var err error |
|
||||
configPath, err = getConfigPath() |
|
||||
if err != nil { |
|
||||
return errors.Wrap(err, "could not get configuration path") |
|
||||
} |
|
||||
if err = updateConfig(); err != nil { |
|
||||
return errors.Wrap(err, "could not update config") |
|
||||
} |
|
||||
return nil |
|
||||
} |
|
||||
|
|
||||
func updateConfig() error { |
|
||||
file, err := ioutil.ReadFile(configPath) |
|
||||
if err != nil { |
|
||||
return errors.Wrap(err, "could not read configuration file") |
|
||||
} |
|
||||
if err = json.Unmarshal(file, &config); err != nil { |
|
||||
return errors.Wrap(err, "could not unmarshal configuration file") |
|
||||
} |
|
||||
return nil |
|
||||
} |
|
||||
|
|
||||
func getConfigPath() (string, error) { |
|
||||
ex, err := os.Executable() |
|
||||
if err != nil { |
|
||||
return "", errors.Wrap(err, "could not get executable path") |
|
||||
} |
|
||||
return filepath.Join(filepath.Dir(ex), "config.json"), nil |
|
||||
} |
|
||||
|
|
||||
// Set replaces the current configuration with the given one
|
|
||||
func Set(conf *Configuration) error { |
|
||||
data, err := json.MarshalIndent(conf, "", " ") |
|
||||
if err != nil { |
|
||||
return err |
|
||||
} |
|
||||
if err = ioutil.WriteFile(configPath, data, 0644); err != nil { |
|
||||
return err |
|
||||
} |
|
||||
config = conf |
|
||||
return nil |
|
||||
} |
|
||||
@ -0,0 +1,54 @@ |
|||||
|
package util |
||||
|
|
||||
|
import ( |
||||
|
"os" |
||||
|
"path/filepath" |
||||
|
"strings" |
||||
|
|
||||
|
"github.com/pkg/errors" |
||||
|
"github.com/spf13/viper" |
||||
|
) |
||||
|
|
||||
|
var dataDirPath string |
||||
|
|
||||
|
func ReadInConfig() error { |
||||
|
viper.AutomaticEnv() |
||||
|
viper.SetEnvPrefix("gus") |
||||
|
viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_")) |
||||
|
viper.SetConfigName("config") |
||||
|
viper.AddConfigPath(".") |
||||
|
SetConfigDefaults() |
||||
|
err := viper.ReadInConfig() |
||||
|
if err != nil { |
||||
|
return errors.Wrap(err, "could not reload config file") |
||||
|
} |
||||
|
return CheckForDatadir() |
||||
|
} |
||||
|
|
||||
|
func SetConfigDefaults() { |
||||
|
viper.SetDefault("http.ListenAddr", ":8080") |
||||
|
viper.SetDefault("http.BaseURL", "http://localhost:3000") |
||||
|
|
||||
|
viper.SetDefault("General.DataDir", "data") |
||||
|
viper.SetDefault("General.EnableDebugMode", true) |
||||
|
viper.SetDefault("General.ShortedIDLength", 4) |
||||
|
} |
||||
|
|
||||
|
func GetDataDir() string { |
||||
|
return dataDirPath |
||||
|
} |
||||
|
|
||||
|
func CheckForDatadir() error { |
||||
|
var err error |
||||
|
dataDirPath, err = filepath.Abs(viper.GetString("General.DataDir")) |
||||
|
if err != nil { |
||||
|
return errors.Wrap(err, "could not get relative data dir path") |
||||
|
} |
||||
|
if _, err = os.Stat(dataDirPath); os.IsNotExist(err) { |
||||
|
err = os.MkdirAll(dataDirPath, 0755) |
||||
|
if err != nil { |
||||
|
return errors.Wrap(err, "could not create config directory") |
||||
|
} |
||||
|
} |
||||
|
return nil |
||||
|
} |
||||
@ -0,0 +1,36 @@ |
|||||
|
package util |
||||
|
|
||||
|
import ( |
||||
|
"crypto/rand" |
||||
|
"io/ioutil" |
||||
|
"os" |
||||
|
"path/filepath" |
||||
|
|
||||
|
"github.com/pkg/errors" |
||||
|
) |
||||
|
|
||||
|
var privateKey []byte |
||||
|
|
||||
|
func CheckForPrivateKey() error { |
||||
|
privateDat := filepath.Join(GetDataDir(), "private.dat") |
||||
|
d, err := ioutil.ReadFile(privateDat) |
||||
|
if err == nil { |
||||
|
privateKey = d |
||||
|
} else if os.IsNotExist(err) { |
||||
|
b := make([]byte, 256) |
||||
|
if _, err := rand.Read(b); err != nil { |
||||
|
return errors.Wrap(err, "could not read random bytes") |
||||
|
} |
||||
|
if err = ioutil.WriteFile(privateDat, b, 0644); err != nil { |
||||
|
return errors.Wrap(err, "could not write private key") |
||||
|
} |
||||
|
privateKey = b |
||||
|
} else if err != nil { |
||||
|
return errors.Wrap(err, "could not read private key") |
||||
|
} |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
func GetPrivateKey() []byte { |
||||
|
return privateKey |
||||
|
} |
||||
Loading…
Reference in new issue