Merge pull request #57 from dispatchrun/dotenv_support

Add dotenv support to `dispatch run` and `dispatch login` commands

Author: chicoxyzzy

cli/config.go

@@ -5,9 +5,11 @@ import (
 	"errors"
 	"fmt"
 	"io"
+	"log/slog"
 	"os"
 	"path/filepath"
 
+	"github.com/joho/godotenv"
 	"github.com/pelletier/go-toml/v2"
 	"golang.org/x/term"
 )
@@ -23,9 +25,15 @@ var (
 	DispatchConsoleUrl       string
 
 	DispatchConfigPath string
+
+	DotEnvFilePath string
 )
 
 func init() {
+	setVariables()
+}
+
+func setVariables() {
 	DispatchApiUrl = os.Getenv("DISPATCH_API_URL")
 	if DispatchApiUrl == "" {
 		DispatchApiUrl = "https://api.dispatch.run"
@@ -144,3 +152,18 @@ func runConfigFlow() error {
 	}
 	return nil
 }
+
+func loadEnvFromFile(path string) error {
+	if path != "" {
+		absolutePath, err := filepath.Abs(path)
+		if err != nil {
+			return fmt.Errorf("failed to get absolute path for %s: %v", path, err)
+		}
+		if err := godotenv.Load(path); err != nil {
+			return fmt.Errorf("failed to load env file from %s: %v", absolutePath, err)
+		}
+		slog.Info("loading environment variables from file", "path", absolutePath)
+	}
+	setVariables()
+	return nil
+}

cli/main.go

@@ -19,11 +19,15 @@ Documentation: https://docs.dispatch.run
 Discord: https://dispatch.run/discord
 Support: support@dispatch.run
 `,
+		PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
+			return loadEnvFromFile(DotEnvFilePath)
+		},
 		RunE: func(cmd *cobra.Command, args []string) error {
 			return cmd.Help()
 		},
 	}
 	cmd.PersistentFlags().StringVarP(&DispatchApiKeyCli, "api-key", "k", "", "Dispatch API key (env: DISPATCH_API_KEY)")
+	cmd.PersistentFlags().StringVarP(&DotEnvFilePath, "env-file", "", "", "Path to .env file")
 
 	cmd.AddGroup(&cobra.Group{
 		ID:    "management",

go.mod

@@ -21,6 +21,7 @@ require (
 	github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
 	github.com/containerd/console v1.0.4-0.20230313162750-1ae8d489ac81 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
+	github.com/joho/godotenv v1.5.1 // indirect
 	github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
 	github.com/mattn/go-isatty v0.0.18 // indirect
 	github.com/mattn/go-localereader v0.0.1 // indirect

go.sum

@@ -23,6 +23,8 @@ github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
 github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
 github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
+github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
+github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
 github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY=
 github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0=
 github.com/mattn/go-isatty v0.0.18 h1:DOKFKCQ7FNG2L1rbrmstDN4QVRdS89Nkh85u68Uwp98=