mirror of
https://github.com/coder/coder.git
synced 2025-07-09 11:45:56 +00:00
dd97fe2bce
This change will improve over CLI performance and "snappiness" as well as substantially reduce our test times. Preliminary benchmarks show `coder server --help` times cut from 300ms to 120ms on my dogfood instance. The inefficiency of lipgloss disproportionately impacts our system, as all help text for every command is generated whenever any command is invoked. The `pretty` API could clean up a lot of the code (e.g., by replacing complex string concatenations with Printf), but this commit is too expansive as is so that work will be done in a follow up.
106 lines
2.5 KiB
Go
106 lines
2.5 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/coder/pretty"
|
|
|
|
"github.com/coder/coder/v2/cli/clibase"
|
|
"github.com/coder/coder/v2/cli/cliui"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
)
|
|
|
|
func (r *RootCmd) restart() *clibase.Cmd {
|
|
var parameterFlags workspaceParameterFlags
|
|
|
|
client := new(codersdk.Client)
|
|
cmd := &clibase.Cmd{
|
|
Annotations: workspaceCommand,
|
|
Use: "restart <workspace>",
|
|
Short: "Restart a workspace",
|
|
Middleware: clibase.Chain(
|
|
clibase.RequireNArgs(1),
|
|
r.InitClient(client),
|
|
),
|
|
Options: append(parameterFlags.cliBuildOptions(), cliui.SkipPromptOption()),
|
|
Handler: func(inv *clibase.Invocation) error {
|
|
ctx := inv.Context()
|
|
out := inv.Stdout
|
|
|
|
workspace, err := namedWorkspace(inv.Context(), client, inv.Args[0])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
lastBuildParameters, err := client.WorkspaceBuildParameters(inv.Context(), workspace.LatestBuild.ID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
template, err := client.Template(inv.Context(), workspace.TemplateID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
buildOptions, err := asWorkspaceBuildParameters(parameterFlags.buildOptions)
|
|
if err != nil {
|
|
return xerrors.Errorf("can't parse build options: %w", err)
|
|
}
|
|
|
|
buildParameters, err := prepStartWorkspace(inv, client, prepStartWorkspaceArgs{
|
|
Action: WorkspaceRestart,
|
|
Template: template,
|
|
|
|
LastBuildParameters: lastBuildParameters,
|
|
|
|
PromptBuildOptions: parameterFlags.promptBuildOptions,
|
|
BuildOptions: buildOptions,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = cliui.Prompt(inv, cliui.PromptOptions{
|
|
Text: "Confirm restart workspace?",
|
|
IsConfirm: true,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
build, err := client.CreateWorkspaceBuild(ctx, workspace.ID, codersdk.CreateWorkspaceBuildRequest{
|
|
Transition: codersdk.WorkspaceTransitionStop,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = cliui.WorkspaceBuild(ctx, out, client, build.ID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
build, err = client.CreateWorkspaceBuild(ctx, workspace.ID, codersdk.CreateWorkspaceBuildRequest{
|
|
Transition: codersdk.WorkspaceTransitionStart,
|
|
RichParameterValues: buildParameters,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = cliui.WorkspaceBuild(ctx, out, client, build.ID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, _ = fmt.Fprintf(out,
|
|
"\nThe %s workspace has been restarted at %s!\n",
|
|
pretty.Sprint(cliui.DefaultStyles.Keyword, workspace.Name), cliui.Timestamp(time.Now()),
|
|
)
|
|
return nil
|
|
},
|
|
}
|
|
return cmd
|
|
}
|