mirror of
https://github.com/coder/coder.git
synced 2026-06-02 20:48:20 +00:00
Spike Curtis
bddb808b25
Fixes all our Go file imports to match the preferred spec that we've _mostly_ been using. For example: ``` import ( "context" "time" "github.com/prometheus/client_golang/prometheus" "golang.org/x/xerrors" "gopkg.in/natefinch/lumberjack.v2" "cdr.dev/slog/v3" "github.com/coder/coder/v2/codersdk/agentsdk" "github.com/coder/serpent" ) ``` 3 groups: standard library, 3rd partly libs, Coder libs. This PR makes the change across the codebase. The PR in the stack above modifies our formatting to maintain this state of affairs, and is a separate PR so it's possible to review that one in detail.
109 lines
3.1 KiB
Go
109 lines
3.1 KiB
Go
package cli
|
|
|
|
import (
|
|
"sort"
|
|
"sync"
|
|
|
|
"github.com/google/uuid"
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/coder/coder/v2/agent/agentcontainers"
|
|
"github.com/coder/coder/v2/cli/cliui"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
"github.com/coder/serpent"
|
|
)
|
|
|
|
func (r *RootCmd) show() *serpent.Command {
|
|
var details bool
|
|
return &serpent.Command{
|
|
Use: "show <workspace>",
|
|
Short: "Display details of a workspace's resources and agents",
|
|
Options: serpent.OptionSet{
|
|
{
|
|
Flag: "details",
|
|
Description: "Show full error messages and additional details.",
|
|
Default: "false",
|
|
Value: serpent.BoolOf(&details),
|
|
},
|
|
},
|
|
Middleware: serpent.Chain(
|
|
serpent.RequireNArgs(1),
|
|
),
|
|
Handler: func(inv *serpent.Invocation) error {
|
|
client, err := r.InitClient(inv)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
buildInfo, err := client.BuildInfo(inv.Context())
|
|
if err != nil {
|
|
return xerrors.Errorf("get server version: %w", err)
|
|
}
|
|
workspace, err := namedWorkspace(inv.Context(), client, inv.Args[0])
|
|
if err != nil {
|
|
return xerrors.Errorf("get workspace: %w", err)
|
|
}
|
|
|
|
options := cliui.WorkspaceResourcesOptions{
|
|
WorkspaceName: workspace.Name,
|
|
ServerVersion: buildInfo.Version,
|
|
ShowDetails: details,
|
|
}
|
|
if workspace.LatestBuild.Status == codersdk.WorkspaceStatusRunning {
|
|
// Get listening ports for each agent.
|
|
ports, devcontainers := fetchRuntimeResources(inv, client, workspace.LatestBuild.Resources...)
|
|
options.ListeningPorts = ports
|
|
options.Devcontainers = devcontainers
|
|
}
|
|
|
|
return cliui.WorkspaceResources(inv.Stdout, workspace.LatestBuild.Resources, options)
|
|
},
|
|
}
|
|
}
|
|
|
|
func fetchRuntimeResources(inv *serpent.Invocation, client *codersdk.Client, resources ...codersdk.WorkspaceResource) (map[uuid.UUID]codersdk.WorkspaceAgentListeningPortsResponse, map[uuid.UUID]codersdk.WorkspaceAgentListContainersResponse) {
|
|
ports := make(map[uuid.UUID]codersdk.WorkspaceAgentListeningPortsResponse)
|
|
devcontainers := make(map[uuid.UUID]codersdk.WorkspaceAgentListContainersResponse)
|
|
var wg sync.WaitGroup
|
|
var mu sync.Mutex
|
|
for _, res := range resources {
|
|
for _, agent := range res.Agents {
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
lp, err := client.WorkspaceAgentListeningPorts(inv.Context(), agent.ID)
|
|
if err != nil {
|
|
cliui.Warnf(inv.Stderr, "Failed to get listening ports for agent %s: %v", agent.Name, err)
|
|
}
|
|
sort.Slice(lp.Ports, func(i, j int) bool {
|
|
return lp.Ports[i].Port < lp.Ports[j].Port
|
|
})
|
|
mu.Lock()
|
|
ports[agent.ID] = lp
|
|
mu.Unlock()
|
|
}()
|
|
|
|
if agent.ParentID.Valid {
|
|
continue
|
|
}
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
dc, err := client.WorkspaceAgentListContainers(inv.Context(), agent.ID, map[string]string{
|
|
// Labels set by VSCode Remote Containers and @devcontainers/cli.
|
|
agentcontainers.DevcontainerConfigFileLabel: "",
|
|
agentcontainers.DevcontainerLocalFolderLabel: "",
|
|
})
|
|
if err != nil {
|
|
cliui.Warnf(inv.Stderr, "Failed to get devcontainers for agent %s: %v", agent.Name, err)
|
|
}
|
|
mu.Lock()
|
|
devcontainers[agent.ID] = dc
|
|
mu.Unlock()
|
|
}()
|
|
}
|
|
}
|
|
wg.Wait()
|
|
return ports, devcontainers
|
|
}
|