generated from brevdev/seed
-
Notifications
You must be signed in to change notification settings - Fork 9
/
delete.go
101 lines (87 loc) · 2.67 KB
/
delete.go
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
package delete
import (
_ "embed"
"fmt"
"strings"
"github.com/brevdev/brev-cli/pkg/cmd/completions"
"github.com/brevdev/brev-cli/pkg/cmd/util"
"github.com/brevdev/brev-cli/pkg/entity"
breverrors "github.com/brevdev/brev-cli/pkg/errors"
"github.com/brevdev/brev-cli/pkg/terminal"
"github.com/hashicorp/go-multierror"
"github.com/spf13/cobra"
stripmd "github.com/writeas/go-strip-markdown"
)
var (
//go:embed doc.md
deleteLong string
deleteExample = "brev delete <ws_name>"
)
type DeleteStore interface {
completions.CompletionStore
DeleteWorkspace(workspaceID string) (*entity.Workspace, error)
GetWorkspaceByNameOrID(orgID string, nameOrID string) ([]entity.Workspace, error)
}
func NewCmdDelete(t *terminal.Terminal, loginDeleteStore DeleteStore, noLoginDeleteStore DeleteStore) *cobra.Command {
cmd := &cobra.Command{
Annotations: map[string]string{"workspace": ""},
Use: "delete",
DisableFlagsInUseLine: true,
Short: "Delete a Brev instance",
Long: stripmd.Strip(deleteLong),
Example: deleteExample,
ValidArgsFunction: completions.GetAllWorkspaceNameCompletionHandler(noLoginDeleteStore, t),
RunE: func(cmd *cobra.Command, args []string) error {
var allError error
for _, workspace := range args {
err := deleteWorkspace(workspace, t, loginDeleteStore)
if err != nil {
allError = multierror.Append(allError, err)
}
}
if allError != nil {
return breverrors.WrapAndTrace(allError)
}
return nil
},
}
return cmd
}
func deleteWorkspace(workspaceName string, t *terminal.Terminal, deleteStore DeleteStore) error {
workspace, err := util.GetUserWorkspaceByNameOrIDErr(deleteStore, workspaceName)
if err != nil {
err1 := handleAdminUser(err, deleteStore)
if err1 != nil {
return breverrors.WrapAndTrace(err1)
}
}
var workspaceID string
if workspace != nil {
workspaceID = workspace.ID
} else {
workspaceID = workspaceName
}
deletedWorkspace, err := deleteStore.DeleteWorkspace(workspaceID)
if err != nil {
return breverrors.WrapAndTrace(err)
}
t.Vprintf("Deleting instance %s. This can take a few minutes. Run 'brev ls' to check status\n", deletedWorkspace.Name)
return nil
}
func handleAdminUser(err error, deleteStore DeleteStore) error {
if strings.Contains(err.Error(), "not found") {
user, err1 := deleteStore.GetCurrentUser()
if err1 != nil {
return breverrors.WrapAndTrace(err1)
}
if user.GlobalUserType != "Admin" {
return breverrors.WrapAndTrace(err)
}
fmt.Println("attempting to delete a workspace you don't own as admin")
return nil
}
if err != nil {
return breverrors.WrapAndTrace(err)
}
return nil
}