Skip to content
This repository was archived by the owner on Aug 30, 2024. It is now read-only.

Commit b76e49d

Browse files
author
Russtopia
committed
Add subcommands 'create' and 'del' to manage devurls
1 parent 6450687 commit b76e49d

File tree

6 files changed

+178
-8
lines changed

6 files changed

+178
-8
lines changed

‎cmd/coder/main.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,8 @@ func (r *rootCmd) Subcommands() []cli.Command {
3737
&shellCmd{},
3838
&syncCmd{},
3939
&urlsCmd{},
40+
&createURLCmd{},
41+
&delURLCmd{},
4042
&versionCmd{},
4143
&configSSHCmd{},
4244
}

‎cmd/coder/sync.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@ func (cmd *syncCmd) RegisterFlags(fl *pflag.FlagSet) {
3232
}
3333

3434
// See https://lxadm.com/Rsync_exit_codes#List_of_standard_rsync_exit_codes.
35-
var IncompatRsync = errors.New("rsync: exit status 2")
36-
var StreamErrRsync = errors.New("rsync: exit status 12")
35+
var errIncompatRsync = errors.New("rsync: exit status 2")
36+
var errStreamErrRsync = errors.New("rsync: exit status 12")
3737

3838
func (cmd *syncCmd) Run(fl *pflag.FlagSet) {
3939
var (
@@ -81,9 +81,9 @@ func (cmd *syncCmd) Run(fl *pflag.FlagSet) {
8181
err = s.Run()
8282
}
8383

84-
if fmt.Sprintf("%v", err) == fmt.Sprintf("%v", IncompatRsync) {
84+
if fmt.Sprintf("%v", err) == fmt.Sprintf("%v", errIncompatRsync) {
8585
flog.Fatal("no compatible rsync present on remote machine")
86-
} else if fmt.Sprintf("%v", err) == fmt.Sprintf("%v", StreamErrRsync) {
86+
} else if fmt.Sprintf("%v", err) == fmt.Sprintf("%v", errStreamErrRsync) {
8787
flog.Fatal("error in rsync protocol datastream (no installed remote rsync?)")
8888
} else {
8989
flog.Fatal("sync: %v", err)

‎cmd/coder/urls.go

Lines changed: 118 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,13 +14,118 @@ import (
1414
)
1515

1616
type urlsCmd struct{}
17+
type createURLCmd struct{}
18+
type delURLCmd struct{}
1719

20+
// DevURL is the parsed json response record for a devURL from cemanager
1821
type DevURL struct {
22+
ID string `json:"id"`
1923
URL string `json:"url"`
2024
Port string `json:"port"`
2125
Access string `json:"access"`
2226
}
2327

28+
var urlAccessLevel = map[string]string{
29+
"PRIVATE": "Only you can access",
30+
"ORG": "All members of your organization can access",
31+
"AUTHED": "Authenticated users can access",
32+
"PUBLIC": "Anyone on the internet can access this link",
33+
}
34+
35+
func isAccessLevelValid(level string) bool {
36+
_, ok := urlAccessLevel[level]
37+
return ok
38+
}
39+
40+
func (cmd createURLCmd) Spec() cli.CommandSpec {
41+
return cli.CommandSpec{
42+
Name: "create",
43+
Usage: "<env name> <port> <access>",
44+
Desc: `create/update a devurl for external access
45+
<access> is one of [PRIVATE | ORG | AUTHED | PUBLIC]`,
46+
}
47+
}
48+
49+
// Run() creates or updates a devURL, specified by env ID and port
50+
// (fl.Arg(0) and fl.Arg(1)), with access level (fl.Arg(2)) on
51+
// the cemanager.
52+
func (cmd createURLCmd) Run(fl *pflag.FlagSet) {
53+
var envName = fl.Arg(0)
54+
var port = fl.Arg(1)
55+
var access = fl.Arg(2)
56+
57+
if envName == "" {
58+
exitUsage(fl)
59+
}
60+
61+
if !isAccessLevelValid(access) {
62+
fmt.Printf("Invalid access level '%v'\n", access)
63+
exitUsage(fl)
64+
}
65+
66+
entClient := requireAuth()
67+
68+
env := findEnv(entClient, envName)
69+
70+
urlID, found := devURLID(port, urlList(fl))
71+
if found {
72+
fmt.Printf("Updating devurl %v\n", urlID)
73+
} else {
74+
fmt.Printf("Adding devurl for port %v\n", port)
75+
}
76+
77+
err := entClient.UpsertDevURL(env.ID, port, access)
78+
if err != nil {
79+
flog.Error("upsert devurl: %s", err.Error())
80+
}
81+
}
82+
83+
func (cmd delURLCmd) Spec() cli.CommandSpec {
84+
return cli.CommandSpec{
85+
Name: "del",
86+
Usage: "<env name> <port>",
87+
Desc: "delete a devurl",
88+
}
89+
}
90+
91+
// Return the ID of a devURL, given the env name and port.
92+
// ("", false) is returned if no match is found.
93+
94+
func devURLID(port string, urls []DevURL) (string, bool) {
95+
for _, url := range urls {
96+
if url.Port == port {
97+
return url.ID, true
98+
}
99+
}
100+
return "", false
101+
}
102+
103+
// Run() deletes a devURL, specified by env ID and port, from the cemanager.
104+
func (cmd delURLCmd) Run(fl *pflag.FlagSet) {
105+
var envName = fl.Arg(0)
106+
var port = fl.Arg(1)
107+
108+
if envName == "" {
109+
exitUsage(fl)
110+
}
111+
112+
entClient := requireAuth()
113+
114+
env := findEnv(entClient, envName)
115+
116+
urlID, found := devURLID(port, urlList(fl))
117+
if found {
118+
fmt.Printf("Found devurl id:%v\n", urlID)
119+
} else {
120+
flog.Fatal("Not found: %v", port)
121+
}
122+
123+
err := entClient.DelDevURL(env.ID, urlID)
124+
if err != nil {
125+
flog.Error("delete devurl: %s", err.Error())
126+
}
127+
}
128+
24129
func (cmd urlsCmd) Spec() cli.CommandSpec {
25130
return cli.CommandSpec{
26131
Name: "urls",
@@ -29,7 +134,9 @@ func (cmd urlsCmd) Spec() cli.CommandSpec {
29134
}
30135
}
31136

32-
func (cmd urlsCmd) Run(fl *pflag.FlagSet) {
137+
// urlList() returns the list of active devURLs from the cemanager server.
138+
139+
func urlList(fl *pflag.FlagSet) []DevURL {
33140
var envName = fl.Arg(0)
34141

35142
if envName == "" {
@@ -41,9 +148,9 @@ func (cmd urlsCmd) Run(fl *pflag.FlagSet) {
41148
env := findEnv(entClient, envName)
42149

43150
reqString := "%s/api/environments/%s/devurls?session_token=%s"
44-
reqUrl := fmt.Sprintf(reqString, entClient.BaseURL, env.ID, entClient.Token)
151+
reqURL := fmt.Sprintf(reqString, entClient.BaseURL, env.ID, entClient.Token)
45152

46-
resp, err := http.Get(reqUrl)
153+
resp, err := http.Get(reqURL)
47154
if err != nil {
48155
flog.Fatal("%v", err)
49156
}
@@ -65,6 +172,14 @@ func (cmd urlsCmd) Run(fl *pflag.FlagSet) {
65172
fmt.Printf("no dev urls were found for environment: %s\n", envName)
66173
}
67174

175+
return devURLs
176+
}
177+
178+
// Run() gets the list of active devURLs from the cemanager for the
179+
// specified environment and outputs info to stdout.
180+
func (cmd urlsCmd) Run(fl *pflag.FlagSet) {
181+
devURLs := urlList(fl)
182+
68183
w := tabwriter.NewWriter(os.Stdout, 0, 0, 1, ' ', tabwriter.TabIndent)
69184
for _, devURL := range devURLs {
70185
fmt.Fprintf(w, "%s\t%s\t%s\n", devURL.URL, devURL.Port, devURL.Access)

‎internal/entclient/activity.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
package entclient
22

3-
import "net/http"
3+
import (
4+
"net/http"
5+
)
46

57
func (c Client) PushActivity(source string, envID string) error {
68
res, err := c.request("POST", "/api/metrics/usage/push", map[string]string{

‎internal/entclient/request.go

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,9 @@ import (
88
"golang.org/x/xerrors"
99
)
1010

11+
// request() makes an HTTP request of the specified method, returning the
12+
// *http.Response and any error.
13+
1114
func (c Client) request(
1215
method string, path string,
1316
request interface{},
@@ -30,6 +33,9 @@ func (c Client) request(
3033
return client.Do(req)
3134
}
3235

36+
// requestBody() makes an HTTP request of the specified method, storing any
37+
// returned body in the supplied response.
38+
3339
func (c Client) requestBody(
3440
method string, path string, request interface{}, response interface{},
3541
) error {

‎internal/entclient/url.go

Lines changed: 45 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,45 @@
1+
package entclient
2+
3+
import (
4+
"fmt"
5+
"net/http"
6+
)
7+
8+
func (c Client) DelDevURL(envID, urlID string) error {
9+
reqString := "/api/environments/%s/devurls/%s"
10+
reqUrl := fmt.Sprintf(reqString, envID, urlID)
11+
12+
res, err := c.request("DELETE", reqUrl, map[string]string{
13+
"environment_id": envID,
14+
"url_id": urlID,
15+
})
16+
if err != nil {
17+
return err
18+
}
19+
20+
if res.StatusCode != http.StatusOK {
21+
return bodyError(res)
22+
}
23+
24+
return nil
25+
}
26+
27+
func (c Client) UpsertDevURL(envID, port, access string) error {
28+
reqString := "/api/environments/%s/devurls"
29+
reqUrl := fmt.Sprintf(reqString, envID)
30+
31+
res, err := c.request("POST", reqUrl, map[string]string{
32+
"environment_id": envID,
33+
"port": port,
34+
"access": access,
35+
})
36+
if err != nil {
37+
return err
38+
}
39+
40+
if res.StatusCode != http.StatusOK {
41+
return bodyError(res)
42+
}
43+
44+
return nil
45+
}

0 commit comments

Comments
 (0)