pasty/internal/env/env.go

package env

import (
	"os"
	"strconv"
	"time"

	"github.com/joho/godotenv"
	"github.com/lus/pasty/internal/static"
)

// Load loads an optional .env file
func Load() {
	godotenv.Load()
}

// MustString returns the content of the environment variable with the given key or the given fallback
func MustString(key, fallback string) string {
	found := os.Getenv(static.EnvironmentVariablePrefix + key)
	if found == "" {
		return fallback
	}
	return found
}

// MustBool uses MustString and parses it into a boolean
func MustBool(key string, fallback bool) bool {
	parsed, _ := strconv.ParseBool(MustString(key, strconv.FormatBool(fallback)))
	return parsed
}

// MustInt uses MustString and parses it into an integer
func MustInt(key string, fallback int) int {
	parsed, _ := strconv.Atoi(MustString(key, strconv.Itoa(fallback)))
	return parsed
}

// MustDuration uses MustString and parses it into a duration
func MustDuration(key string, fallback time.Duration) time.Duration {
	parsed, _ := time.ParseDuration(MustString(key, fallback.String()))
	return parsed
}