mirror of https://github.com/coder/coder.git
101 lines
2.5 KiB
Go
101 lines
2.5 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
"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", cliui.DefaultStyles.Keyword.Render(workspace.Name), cliui.DefaultStyles.DateTimeStamp.Render(time.Now().Format(time.Stamp)))
|
|
return nil
|
|
},
|
|
}
|
|
return cmd
|
|
}
|