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
|
package env
import (
"fmt"
"github.com/bdw/ship/internal/ssh"
"github.com/bdw/ship/internal/state"
"github.com/spf13/cobra"
)
var unsetCmd = &cobra.Command{
Use: "unset <app> KEY...",
Short: "Unset environment variable(s)",
Long: "Remove one or more environment variables from an app.",
Args: cobra.MinimumNArgs(2),
RunE: runUnset,
}
func runUnset(cmd *cobra.Command, args []string) error {
name := args[0]
if err := state.ValidateName(name); err != nil {
return err
}
keys := 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 {
return fmt.Errorf("no environment variables set")
}
changed := false
for _, key := range keys {
if _, exists := app.Env[key]; exists {
delete(app.Env, key)
changed = true
fmt.Printf("Unset %s\n", key)
} else {
fmt.Printf("Warning: %s not found\n", key)
}
}
if !changed {
return nil
}
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
}
|
