Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings
This repository was archived by the owner on Aug 30, 2024. It is now read-only.
/coder-v1-cliPublic archive

Commit74c6417

Browse files
committed
Add ability to create new environments and images
1 parent2d52bb3 commit74c6417

File tree

2 files changed

+181
-0
lines changed

2 files changed

+181
-0
lines changed

‎coder-sdk/env.go

Lines changed: 126 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,9 @@ package coder
22

33
import (
44
"context"
5+
"fmt"
56
"net/http"
7+
"nhooyr.io/websocket/wsjson"
68
"time"
79

810
"cdr.dev/coder-cli/internal/x/xjson"
@@ -75,3 +77,127 @@ func (c Client) DialWsep(ctx context.Context, env *Environment) (*websocket.Conn
7577
}
7678
returnconn,nil
7779
}
80+
81+
// CreateEnvironmentRequest is used to configure a new environment
82+
typeCreateEnvironmentRequeststruct {
83+
Namestring`json:"name"`
84+
ImageIDstring`json:"image_id"`
85+
ImageTagstring`json:"image_tag"`
86+
CPUCoresfloat32`json:"cpu_cores"`
87+
MemoryGBint`json:"memory_gb"`
88+
DiskGBint`json:"disk_gb"`
89+
GPUsint`json:"gpus"`
90+
Services []string`json:"services"`
91+
}
92+
93+
// CreateEnvironment sends a request to create an environment.
94+
func (cClient)CreateEnvironment(ctx context.Context,orgIDstring,reqCreateEnvironmentRequest) (Environment,error) {
95+
varenvEnvironment
96+
err:=c.requestBody(
97+
ctx,
98+
http.MethodPost,"/api/orgs/"+orgID+"/environments",
99+
req,
100+
&env,
101+
)
102+
returnenv,err
103+
}
104+
105+
typeenvUpdatestruct {
106+
Typestring`json:"type"`
107+
}
108+
109+
// WaitForEnvironmentReady watches the environment update websocket and waits for the "done" message type before returning.
110+
func (cClient)WaitForEnvironmentReady(ctx context.Context,envIDstring)error {
111+
u:=c.copyURL()
112+
ifc.BaseURL.Scheme=="https" {
113+
u.Scheme="wss"
114+
}else {
115+
u.Scheme="ws"
116+
}
117+
u.Path="/api/environments/"+envID+"/watch-update"
118+
119+
conn,resp,err:=websocket.Dial(ctx,u.String(),
120+
&websocket.DialOptions{
121+
HTTPHeader:map[string][]string{
122+
"Cookie": {"session_token="+c.Token},
123+
},
124+
},
125+
)
126+
iferr!=nil {
127+
ifresp!=nil {
128+
returnbodyError(resp)
129+
}
130+
returnerr
131+
}
132+
133+
for {
134+
m:=envUpdate{}
135+
err=wsjson.Read(ctx,conn,&m)
136+
iferr!=nil {
137+
returnfmt.Errorf("read ws json msg: %w",err)
138+
}
139+
ifm.Type=="done" {
140+
break
141+
}
142+
}
143+
144+
returnnil
145+
}
146+
147+
typestatsstruct {
148+
ContainerStatusstring`json:"container_status"`
149+
StatErrorstring`json:"stat_error"`
150+
Timestring`json:"time"`
151+
}
152+
153+
// WatchEnvironmentStats watches the environment update websocket for a given duration.
154+
func (cClient)WatchEnvironmentStats(ctx context.Context,envIDstring,duration time.Duration)error {
155+
u:=c.copyURL()
156+
ifc.BaseURL.Scheme=="https" {
157+
u.Scheme="wss"
158+
}else {
159+
u.Scheme="ws"
160+
}
161+
u.Path="/api/environments/"+envID+"/watch-stats"
162+
163+
conn,resp,err:=websocket.Dial(ctx,u.String(),
164+
&websocket.DialOptions{
165+
HTTPHeader:map[string][]string{
166+
"Cookie": {"session_token="+c.Token},
167+
},
168+
},
169+
)
170+
iferr!=nil {
171+
ifresp!=nil {
172+
returnbodyError(resp)
173+
}
174+
returnerr
175+
}
176+
177+
statsCtx,statsCancel:=context.WithTimeout(ctx,duration)
178+
deferstatsCancel()
179+
180+
for {
181+
select {
182+
case<-statsCtx.Done():
183+
returnnil
184+
default:
185+
m:=stats{}
186+
err=wsjson.Read(ctx,conn,&m)
187+
iferr!=nil {
188+
returnfmt.Errorf("read ws json msg: %w",err)
189+
}
190+
}
191+
}
192+
}
193+
194+
// DeleteEnvironment deletes the environment.
195+
func (cClient)DeleteEnvironment(ctx context.Context,envIDstring)error {
196+
err:=c.requestBody(
197+
ctx,
198+
http.MethodDelete,"/api/environments/"+envID,
199+
nil,
200+
nil,
201+
)
202+
returnerr
203+
}

‎coder-sdk/image.go

Lines changed: 55 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,55 @@
1+
package coder
2+
3+
import (
4+
"context"
5+
"net/http"
6+
)
7+
8+
// Image describes a Coder Image
9+
typeImagestruct {
10+
IDstring`json:"id"`
11+
OrganizationIDstring`json:"organization_id"`
12+
Repositorystring`json:"repository"`
13+
Descriptionstring`json:"description"`
14+
URLstring`json:"url"`// user-supplied URL for image
15+
DefaultCPUCoresfloat32`json:"default_cpu_cores"`
16+
DefaultMemoryGBint`json:"default_memory_gb"`
17+
DefaultDiskGBint`json:"default_disk_gb"`
18+
Deprecatedbool`json:"deprecated"`
19+
}
20+
21+
// NewRegistryRequest describes a docker registry used in importing an image
22+
typeNewRegistryRequeststruct {
23+
FriendlyNamestring`json:"friendly_name"`
24+
Registrystring`json:"registry"`
25+
Usernamestring`json:"username"`
26+
Passwordstring`json:"password"`
27+
}
28+
29+
// ImportImageRequest is used to import new images and registries into Coder
30+
typeImportImageRequeststruct {
31+
// RegistryID is used to import images to existing registries.
32+
RegistryID*string`json:"registry_id"`
33+
// NewRegistry is used when adding a new registry.
34+
NewRegistry*NewRegistryRequest`json:"new_registry"`
35+
// Repository refers to the image. For example: "codercom/ubuntu".
36+
Repositorystring`json:"repository"`
37+
Tagstring`json:"tag"`
38+
DefaultCPUCoresfloat32`json:"default_cpu_cores"`
39+
DefaultMemoryGBint`json:"default_memory_gb"`
40+
DefaultDiskGBint`json:"default_disk_gb"`
41+
Descriptionstring`json:"description"`
42+
URLstring`json:"url"`
43+
}
44+
45+
// ImportImage creates a new image and optionally a new registry
46+
func (cClient)ImportImage(ctx context.Context,orgIDstring,reqImportImageRequest) (Image,error) {
47+
varimgImage
48+
err:=c.requestBody(
49+
ctx,
50+
http.MethodPost,"/api/orgs/"+orgID+"/images",
51+
req,
52+
&img,
53+
)
54+
returnimg,err
55+
}

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp