slideshowApp/env/enviroment.go

package env

import (
	"os"
	"strings"

	"github.com/joho/godotenv"
)

const (
	Env             EnvKey = "ENV"
	PhotoDir        EnvKey = "PHOTO_DIR"
	Host            EnvKey = "HOST"
	Port            EnvKey = "PORT"
	IntervalDefault EnvKey = "INTERVAL_DEFAULT"
	LogPath         EnvKey = "LOG_PATH"
)

type EnvKey string

func Load(path string) error {
	if path == "" {
		path = ".env"
	}
	return godotenv.Load(path)
}

func (key EnvKey) GetValue() string {
	keyStr := string(key)
	value := os.Getenv(keyStr)
	if key == Port && value == "" {
		value = "8080"
	}

	return value
}

func (key EnvKey) SetValue(value string) {
	os.Setenv(string(key), value)
}

func (key EnvKey) GetBoolValue() bool {
	value := strings.ToLower(os.Getenv(string(key)))
	return value == "true" || value == "1"
}