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.
67 lines
1.3 KiB
Go
67 lines
1.3 KiB
Go
3 years ago
|
package config
|
||
|
|
|
||
|
|
// package serves to store/load config files for server
|
||
|
|
|
||
|
|
import (
|
||
|
|
_ "fmt"
|
||
|
|
"github.com/spf13/viper"
|
||
|
|
"log"
|
||
|
|
"os/exec"
|
||
|
|
"bytes"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type serverconfig struct {
|
||
|
|
URL string
|
||
|
|
Token string
|
||
|
|
Bucket string
|
||
|
|
Orginization string
|
||
|
|
}
|
||
|
|
|
||
|
|
func ReadServerConfig() *serverconfig {
|
||
|
|
|
||
|
|
viper.SetConfigName("database")
|
||
|
|
viper.SetConfigType("yaml")
|
||
|
|
viper.AddConfigPath("../../internal/configs")
|
||
|
|
viper.SetDefault("Orginization","ForeLight")
|
||
|
|
viper.SetDefault("URL","http://localhost:8086")
|
||
|
|
var C serverconfig
|
||
|
|
err := viper.Unmarshal(&C)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
if C.Token == "" {
|
||
|
|
// token unset
|
||
|
|
cmd := exec.Command("cat","admin_token")
|
||
|
|
var out bytes.Buffer
|
||
|
|
var stderr bytes.Buffer
|
||
|
|
cmd.Stdout = &out
|
||
|
|
cmd.Stderr = &stderr
|
||
|
|
if err := cmd.Run(); err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
outstring := out.String()
|
||
|
|
C.Token = strings.Trim(outstring," \n")
|
||
|
|
viper.Set("token",C.Token)
|
||
|
|
viper.WriteConfig()
|
||
|
|
}
|
||
|
|
return &C
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *serverconfig) GetUrl() string {
|
||
|
|
return s.URL
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
func (s *serverconfig) GetOrg() string {
|
||
|
|
return s.Orginization
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *serverconfig) GetBucket() string {
|
||
|
|
return s.Bucket
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *serverconfig) GetToken() string {
|
||
|
|
return s.Token
|
||
|
|
}
|