mirror of https://github.com/lus/pasty.git
43 lines
1.0 KiB
Go
43 lines
1.0 KiB
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
|
|
}
|