forked from railwayapp/cli
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathenvs.go
More file actions
110 lines (93 loc) · 2.3 KB
/
Copy pathenvs.go
File metadata and controls
110 lines (93 loc) · 2.3 KB
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
package controller
import (
"context"
"encoding/json"
"fmt"
"io/ioutil"
"os"
"github.com/railwayapp/cli/entity"
"github.com/railwayapp/cli/ui"
)
func (c *Controller) GetEnvs(ctx context.Context) (*entity.Envs, error) {
// Get envs through production token if it exists
if c.cfg.RailwayProductionToken != "" {
envs, err := c.gtwy.GetEnvsWithProjectToken(ctx)
if err != nil {
return nil, err
}
return envs, err
}
projectCfg, err := c.cfg.GetProjectConfigs()
if err != nil {
return nil, err
}
if val, ok := projectCfg.LockedEnvsNames[projectCfg.Environment]; ok && val {
fmt.Println(ui.Bold(ui.RedText("Protected Environment Detected!").String()))
confirm, err := ui.PromptYesNo("Continue fetching variables?")
if err != nil {
return nil, err
}
if !confirm {
return nil, nil
}
}
return c.gtwy.GetEnvs(ctx, &entity.GetEnvsRequest{
ProjectID: projectCfg.Project,
EnvironmentID: projectCfg.Environment,
})
}
func (c *Controller) SaveEnvsToFile(ctx context.Context) error {
envs, err := c.GetEnvs(ctx)
if err != nil {
return err
}
err = c.cfg.CreatePathIfNotExist(c.cfg.RailwayEnvFilePath)
if err != nil {
return err
}
encoded, err := json.MarshalIndent(envs, "", " ")
if err != nil {
return err
}
err = ioutil.WriteFile(c.cfg.RailwayEnvFilePath, encoded, os.ModePerm)
if err != nil {
return err
}
return nil
}
func (c *Controller) UpdateEnvs(ctx context.Context, envs *entity.Envs) (*entity.Envs, error) {
projectCfg, err := c.cfg.GetProjectConfigs()
if err != nil {
return nil, err
}
if val, ok := projectCfg.LockedEnvsNames[projectCfg.Environment]; ok && val {
fmt.Println(ui.Bold(ui.RedText("Protected Environment Detected!").String()))
confirm, err := ui.PromptYesNo("Continue updating variables?")
if err != nil {
return nil, err
}
if !confirm {
return nil, nil
}
}
project, err := c.GetProject(ctx, projectCfg.Project)
if err != nil {
return nil, err
}
pluginID := ""
for _, p := range project.Plugins {
if p.Name == "env" {
pluginID = p.ID
}
}
environment, err := c.cfg.GetEnvironment()
if err != nil {
return nil, err
}
return c.gtwy.UpdateEnvsForPlugin(ctx, &entity.UpdateEnvsRequest{
ProjectID: projectCfg.Project,
EnvironmentID: environment,
PluginID: pluginID,
Envs: envs,
})
}