logik2021/codego/core/environment/environment.go

72 lines
1.6 KiB
Go
Raw Normal View History

2021-05-09 18:22:12 +02:00
package environment
import (
"errors"
"fmt"
"log"
"os"
"github.com/joho/godotenv"
)
var ENV_FILE_NAME string
/* ---------------------------------------------------------------- *
* EXPORTS
* ---------------------------------------------------------------- */
func ReadEnvKey(key string, optional ...string) string {
value, err := readEnvKey(key, optional...)
if err != nil {
log.Fatal(err)
}
return value
}
func ReadEnvKeyAllowMissing(key string, optional ...string) string {
value, _ := readEnvKey(key, optional...)
return value
}
func ReadEnvKeyDefault(key string, valueDefault string, optional ...string) string {
value, _ := readEnvKey(key, optional...)
if value == "" {
return valueDefault
}
return value
}
/* ---------------------------------------------------------------- *
* PRIVATE
* ---------------------------------------------------------------- */
func loadEnvFile(path string) error {
if path == "" {
log.Fatal("Path name to environment file is not allowed to be empty!")
}
return godotenv.Load(path)
}
func getEnvKey(key string) string {
return os.Getenv(key)
}
func readEnvKey(key string, optional ...string) (string, error) {
var path string = ENV_FILE_NAME
var err error
var value string
if len(optional) > 0 {
path = optional[0]
}
err = loadEnvFile(path)
if err != nil {
log.Fatalf("Could not read environment file, \033[1m%s\033[0m!\n", path)
}
err = nil
value = getEnvKey(key)
if value == "" {
err = errors.New(fmt.Sprintf("Environment key \033[1m%s\033[0m missing or empty in environment file \033[1m%s\033[0m!", key, path))
}
return value, err
}