diff --git a/cli/server.go b/cli/server.go index 68722f1a5e133..dd5460a9c83d3 100644 --- a/cli/server.go +++ b/cli/server.go @@ -361,18 +361,7 @@ func server() *cobra.Command { return xerrors.Errorf("delete workspace: %w", err) } - err = cliui.ProvisionerJob(cmd.Context(), cmd.OutOrStdout(), cliui.ProvisionerJobOptions{ - Fetch: func() (codersdk.ProvisionerJob, error) { - build, err := client.WorkspaceBuild(cmd.Context(), build.ID) - return build.Job, err - }, - Cancel: func() error { - return client.CancelWorkspaceBuild(cmd.Context(), build.ID) - }, - Logs: func() (<-chan codersdk.ProvisionerJobLog, error) { - return client.WorkspaceBuildLogsAfter(cmd.Context(), build.ID, before) - }, - }) + err = cliui.WorkspaceBuild(cmd.Context(), cmd.OutOrStdout(), client, build.ID, before) if err != nil { return xerrors.Errorf("delete workspace %s: %w", workspace.Name, err) } diff --git a/cli/workspacestart.go b/cli/workspacestart.go index 5b985f7d88f0d..abb1bdcf3a01e 100644 --- a/cli/workspacestart.go +++ b/cli/workspacestart.go @@ -35,19 +35,7 @@ func workspaceStart() *cobra.Command { if err != nil { return err } - err = cliui.ProvisionerJob(cmd.Context(), cmd.OutOrStdout(), cliui.ProvisionerJobOptions{ - Fetch: func() (codersdk.ProvisionerJob, error) { - build, err := client.WorkspaceBuild(cmd.Context(), build.ID) - return build.Job, err - }, - Cancel: func() error { - return client.CancelWorkspaceBuild(cmd.Context(), build.ID) - }, - Logs: func() (<-chan codersdk.ProvisionerJobLog, error) { - return client.WorkspaceBuildLogsAfter(cmd.Context(), build.ID, before) - }, - }) - return err + return cliui.WorkspaceBuild(cmd.Context(), cmd.OutOrStdout(), client, build.ID, before) }, } } diff --git a/cli/workspacestop.go b/cli/workspacestop.go index cbf90c987c52b..e7cb32336eb93 100644 --- a/cli/workspacestop.go +++ b/cli/workspacestop.go @@ -35,22 +35,7 @@ func workspaceStop() *cobra.Command { if err != nil { return err } - err = cliui.ProvisionerJob(cmd.Context(), cmd.OutOrStdout(), cliui.ProvisionerJobOptions{ - Fetch: func() (codersdk.ProvisionerJob, error) { - build, err := client.WorkspaceBuild(cmd.Context(), build.ID) - return build.Job, err - }, - Cancel: func() error { - return client.CancelWorkspaceBuild(cmd.Context(), build.ID) - }, - Logs: func() (<-chan codersdk.ProvisionerJobLog, error) { - return client.WorkspaceBuildLogsAfter(cmd.Context(), build.ID, before) - }, - }) - if err != nil { - return err - } - return nil + return cliui.WorkspaceBuild(cmd.Context(), cmd.OutOrStdout(), client, build.ID, before) }, } }