1 files changed, 132 insertions, 0 deletions

diff --git a/cmd/ship/env/set.go b/cmd/ship/env/set.go
new file mode 100644
index 0000000..e11d2c9
--- /dev/null
+++ b/cmd/ship/env/set.go
@@ -0,0 +1,132 @@
+package env
+
+import (
+        "bufio"
+        "fmt"
+        "os"
+        "strings"
+
+        "github.com/bdw/ship/internal/ssh"
+        "github.com/bdw/ship/internal/state"
+        "github.com/spf13/cobra"
+)
+
+var setCmd = &cobra.Command{
+        Use:   "set <app> KEY=VALUE...",
+        Short: "Set environment variable(s)",
+        Long:  "Set one or more environment variables for an app. Variables are specified as KEY=VALUE pairs.",
+        Args:  cobra.MinimumNArgs(2),
+        RunE:  runSet,
+}
+
+func init() {
+        setCmd.Flags().StringP("file", "f", "", "Load environment from file")
+}
+
+func runSet(cmd *cobra.Command, args []string) error {
+        name := args[0]
+        envVars := args[1:]
+
+        st, err := state.Load()
+        if err != nil {
+                return fmt.Errorf("error loading state: %w", err)
+        }
+
+        host, _ := cmd.Flags().GetString("host")
+        if host == "" {
+                host = st.GetDefaultHost()
+        }
+
+        if host == "" {
+                return fmt.Errorf("--host is required")
+        }
+
+        app, err := st.GetApp(host, name)
+        if err != nil {
+                return err
+        }
+
+        if app.Type != "app" {
+                return fmt.Errorf("env is only available for apps, not static sites")
+        }
+
+        if app.Env == nil {
+                app.Env = make(map[string]string)
+        }
+
+        // Set variables from args
+        for _, e := range envVars {
+                parts := strings.SplitN(e, "=", 2)
+                if len(parts) == 2 {
+                        app.Env[parts[0]] = parts[1]
+                        fmt.Printf("Set %s\n", parts[0])
+                } else {
+                        return fmt.Errorf("invalid format: %s (expected KEY=VALUE)", e)
+                }
+        }
+
+        // Set variables from file if provided
+        envFile, _ := cmd.Flags().GetString("file")
+        if envFile != "" {
+                fileEnv, err := parseEnvFile(envFile)
+                if err != nil {
+                        return fmt.Errorf("error reading env file: %w", err)
+                }
+                for k, v := range fileEnv {
+                        app.Env[k] = v
+                        fmt.Printf("Set %s\n", k)
+                }
+        }
+
+        if err := st.Save(); err != nil {
+                return fmt.Errorf("error saving state: %w", err)
+        }
+
+        client, err := ssh.Connect(host)
+        if err != nil {
+                return fmt.Errorf("error connecting to VPS: %w", err)
+        }
+        defer client.Close()
+
+        fmt.Println("-> Updating environment file on VPS...")
+        envFilePath := fmt.Sprintf("/etc/ship/env/%s.env", name)
+        envContent := ""
+        for k, v := range app.Env {
+                envContent += fmt.Sprintf("%s=%s\n", k, v)
+        }
+        if err := client.WriteSudoFile(envFilePath, envContent); err != nil {
+                return fmt.Errorf("error updating env file: %w", err)
+        }
+
+        fmt.Println("-> Restarting service...")
+        if _, err := client.RunSudo(fmt.Sprintf("systemctl restart %s", name)); err != nil {
+                return fmt.Errorf("error restarting service: %w", err)
+        }
+
+        fmt.Println("Environment variables updated successfully")
+        return nil
+}
+
+func parseEnvFile(path string) (map[string]string, error) {
+        file, err := os.Open(path)
+        if err != nil {
+                return nil, err
+        }
+        defer file.Close()
+
+        env := make(map[string]string)
+        scanner := bufio.NewScanner(file)
+        for scanner.Scan() {
+                line := strings.TrimSpace(scanner.Text())
+                if line == "" || strings.HasPrefix(line, "#") {
+                        continue
+                }
+
+                parts := strings.SplitN(line, "=", 2)
+                if len(parts) == 2 {
+                        env[parts[0]] = parts[1]
+                }
+        }
+
+        return env, scanner.Err()
+}