forked from Kispi/Core
34 lines
677 B
Go
34 lines
677 B
Go
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/joho/godotenv"
|
||
|
|
"github.com/pterm/pterm"
|
||
|
|
)
|
||
|
|
|
||
|
|
// GetEnv returns the value of the environment variable named by the key.
|
||
|
|
// If the env key is not present, it returns the default value.
|
||
|
|
func GetEnv(key, def string) string {
|
||
|
|
if value, ok := os.LookupEnv(key); ok {
|
||
|
|
return value
|
||
|
|
}
|
||
|
|
return def
|
||
|
|
}
|
||
|
|
|
||
|
|
func LoadEnv() {
|
||
|
|
err := godotenv.Load("../.env")
|
||
|
|
if err != nil {
|
||
|
|
pterm.Info.Println("No .env file found. Proceeding...")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Add env variables without "VITE_" prefix.
|
||
|
|
for _, e := range os.Environ() {
|
||
|
|
pair := strings.Split(e, "=")
|
||
|
|
if strings.HasPrefix(pair[0], "VITE_") {
|
||
|
|
os.Setenv(pair[0][5:], pair[1])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|