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 ", 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 }