mirror of
https://github.com/Open-Argon/argon-v3.git
synced 2025-12-06 08:56:07 +00:00
add environment variable support
This commit is contained in:
34
src/env.go
Normal file
34
src/env.go
Normal file
@@ -0,0 +1,34 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
|
||||
"github.com/joho/godotenv"
|
||||
)
|
||||
|
||||
// get the environment variables
|
||||
func getEnv() ArObject {
|
||||
env := make(anymap)
|
||||
for _, e := range os.Environ() {
|
||||
pair := strings.Split(e, "=")
|
||||
env[pair[0]] = ArString(pair[1])
|
||||
}
|
||||
cwd, err := os.Getwd()
|
||||
if err == nil {
|
||||
envfile := filepath.Join(cwd, ".env")
|
||||
err := godotenv.Load(envfile)
|
||||
if err == nil {
|
||||
values, err := godotenv.Read()
|
||||
if err == nil {
|
||||
for k, v := range values {
|
||||
env[k] = ArString(v)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return Map(env)
|
||||
}
|
||||
|
||||
var env = getEnv()
|
||||
Reference in New Issue
Block a user