89 lines
2.4 KiB
Go
89 lines
2.4 KiB
Go
package cmds
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/BurntSushi/toml"
|
|
)
|
|
|
|
// cfg holds the configuration file data from config.toml
|
|
var cfg = struct {
|
|
Postgres struct {
|
|
User string
|
|
Pass string
|
|
Host string
|
|
Port int
|
|
DBName string
|
|
}
|
|
}{}
|
|
|
|
// testCfg holds the test configuration file data from config.toml
|
|
var testCfg = struct {
|
|
Found bool
|
|
Postgres struct {
|
|
// Test details for template test file generation
|
|
TestUser string
|
|
TestPass string
|
|
TestHost string
|
|
TestPort int
|
|
TestDBName string
|
|
}
|
|
}{}
|
|
|
|
func LoadConfigFile(filename string) {
|
|
var tmpCfg = struct {
|
|
Postgres struct {
|
|
User string `toml:"user"`
|
|
Pass string `toml:"pass"`
|
|
Host string `toml:"host"`
|
|
Port int `toml:"port"`
|
|
DBName string `toml:"dbname"`
|
|
// Test details for template test file generation
|
|
TestUser string `toml:"test_user"`
|
|
TestPass string `toml:"test_pass"`
|
|
TestHost string `toml:"test_host"`
|
|
TestPort int `toml:"test_port"`
|
|
TestDBName string `toml:"test_dbname"`
|
|
} `toml:"postgres"`
|
|
}{}
|
|
|
|
_, err := toml.DecodeFile(filename, &tmpCfg)
|
|
|
|
if os.IsNotExist(err) {
|
|
fmt.Printf("Failed to find the toml configuration file %s: %s", filename, err)
|
|
return
|
|
}
|
|
|
|
if err != nil {
|
|
fmt.Println("Failed to decode toml configuration file:", err)
|
|
}
|
|
|
|
cfg.Postgres.User = tmpCfg.Postgres.User
|
|
cfg.Postgres.Pass = tmpCfg.Postgres.Pass
|
|
cfg.Postgres.Host = tmpCfg.Postgres.Host
|
|
cfg.Postgres.Port = tmpCfg.Postgres.Port
|
|
cfg.Postgres.DBName = tmpCfg.Postgres.DBName
|
|
|
|
testCfg.Postgres.TestUser = tmpCfg.Postgres.TestUser
|
|
testCfg.Postgres.TestPass = tmpCfg.Postgres.TestPass
|
|
testCfg.Postgres.TestHost = tmpCfg.Postgres.TestHost
|
|
testCfg.Postgres.TestPort = tmpCfg.Postgres.TestPort
|
|
testCfg.Postgres.TestDBName = tmpCfg.Postgres.TestDBName
|
|
|
|
// If all test cfg variables are present set found flag to true
|
|
if testCfg.Postgres.TestUser != "" && testCfg.Postgres.TestPass != "" &&
|
|
testCfg.Postgres.TestHost != "" && testCfg.Postgres.TestPort != 0 &&
|
|
testCfg.Postgres.TestDBName != "" {
|
|
testCfg.Found = true
|
|
}
|
|
|
|
// As a safety precaution, set found to false if
|
|
// the dbname is the same as the cfg dbname. This will prevent the test
|
|
// from erasing the production database tables if someone accidently
|
|
// configures the config.toml incorrectly.
|
|
if testCfg.Postgres.TestDBName == cfg.Postgres.DBName {
|
|
testCfg.Found = false
|
|
testCfg.Postgres.TestDBName = ""
|
|
}
|
|
}
|