Allow env vars to be passed in (#5)

abe-winter · Apr 8, 2024 · 925cde4 · 925cde4
1 parent 1846c71
commit 925cde4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/web/cmd/droid/droid.go b/web/cmd/droid/droid.go
@@ -4,6 +4,7 @@ package droid
 
 import (
 	"os"
+	"strings"
 
 	"go.viam.com/utils"
 
@@ -23,7 +24,11 @@ func DroidStopHook() {
 	server.ForceRestart = true
 }
 
-func MainEntry(configPath, writeablePath string) {
+func MainEntry(configPath, writeablePath, osEnv string) {
 	os.Args = append(os.Args, "-config", configPath)
+	for _, envEntry := range strings.Split(osEnv, "\n") {
+		entryParts := strings.SplitN(envEntry, "=", 2)
+		os.Setenv(entryParts[0], entryParts[1])
+	}
 	utils.ContextualMain(server.RunServer, logger)
 }