mirror of https://github.com/coder/coder.git
90 lines
2.9 KiB
Go
90 lines
2.9 KiB
Go
package codersdk
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
const (
|
|
WorkspaceAgentPortShareLevelOwner WorkspaceAgentPortShareLevel = "owner"
|
|
WorkspaceAgentPortShareLevelAuthenticated WorkspaceAgentPortShareLevel = "authenticated"
|
|
WorkspaceAgentPortShareLevelPublic WorkspaceAgentPortShareLevel = "public"
|
|
)
|
|
|
|
type (
|
|
WorkspaceAgentPortShareLevel string
|
|
UpsertWorkspaceAgentPortShareRequest struct {
|
|
AgentName string `json:"agent_name"`
|
|
Port int32 `json:"port"`
|
|
ShareLevel WorkspaceAgentPortShareLevel `json:"share_level"`
|
|
}
|
|
WorkspaceAgentPortShares struct {
|
|
Shares []WorkspaceAgentPortShare `json:"shares"`
|
|
}
|
|
WorkspaceAgentPortShare struct {
|
|
WorkspaceID uuid.UUID `json:"workspace_id" format:"uuid"`
|
|
AgentName string `json:"agent_name"`
|
|
Port int32 `json:"port"`
|
|
ShareLevel WorkspaceAgentPortShareLevel `json:"share_level"`
|
|
}
|
|
DeleteWorkspaceAgentPortShareRequest struct {
|
|
AgentName string `json:"agent_name"`
|
|
Port int32 `json:"port"`
|
|
}
|
|
)
|
|
|
|
func (l WorkspaceAgentPortShareLevel) ValidMaxLevel() bool {
|
|
return l == WorkspaceAgentPortShareLevelOwner ||
|
|
l == WorkspaceAgentPortShareLevelAuthenticated ||
|
|
l == WorkspaceAgentPortShareLevelPublic
|
|
}
|
|
|
|
func (l WorkspaceAgentPortShareLevel) ValidPortShareLevel() bool {
|
|
return l == WorkspaceAgentPortShareLevelAuthenticated ||
|
|
l == WorkspaceAgentPortShareLevelPublic
|
|
}
|
|
|
|
func (c *Client) GetWorkspaceAgentPortShares(ctx context.Context, workspaceID uuid.UUID) (WorkspaceAgentPortShares, error) {
|
|
var shares WorkspaceAgentPortShares
|
|
res, err := c.Request(ctx, http.MethodGet, fmt.Sprintf("/api/v2/workspaces/%s/port-share", workspaceID), nil)
|
|
if err != nil {
|
|
return shares, err
|
|
}
|
|
defer res.Body.Close()
|
|
if res.StatusCode != http.StatusOK {
|
|
return shares, ReadBodyAsError(res)
|
|
}
|
|
|
|
return shares, json.NewDecoder(res.Body).Decode(&shares)
|
|
}
|
|
|
|
func (c *Client) UpsertWorkspaceAgentPortShare(ctx context.Context, workspaceID uuid.UUID, req UpsertWorkspaceAgentPortShareRequest) (WorkspaceAgentPortShare, error) {
|
|
var share WorkspaceAgentPortShare
|
|
res, err := c.Request(ctx, http.MethodPost, fmt.Sprintf("/api/v2/workspaces/%s/port-share", workspaceID), req)
|
|
if err != nil {
|
|
return share, err
|
|
}
|
|
defer res.Body.Close()
|
|
if res.StatusCode != http.StatusOK {
|
|
return share, ReadBodyAsError(res)
|
|
}
|
|
|
|
return share, json.NewDecoder(res.Body).Decode(&share)
|
|
}
|
|
|
|
func (c *Client) DeleteWorkspaceAgentPortShare(ctx context.Context, workspaceID uuid.UUID, req DeleteWorkspaceAgentPortShareRequest) error {
|
|
res, err := c.Request(ctx, http.MethodDelete, fmt.Sprintf("/api/v2/workspaces/%s/port-share", workspaceID), req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer res.Body.Close()
|
|
if res.StatusCode != http.StatusOK {
|
|
return ReadBodyAsError(res)
|
|
}
|
|
return nil
|
|
}
|