1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
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()
}
|
