coder/enterprise/cli/prebuilds.go

package cli

import (
	"fmt"

	"golang.org/x/xerrors"

	"github.com/coder/serpent"

	"github.com/coder/coder/v2/cli"
	"github.com/coder/coder/v2/codersdk"
)

func (r *RootCmd) prebuilds() *serpent.Command {
	cmd := &serpent.Command{
		Use:   "prebuilds",
		Short: "Manage Coder prebuilds",
		Long: "Administrators can use these commands to manage prebuilt workspace settings.\n" + cli.FormatExamples(
			cli.Example{
				Description: "Pause Coder prebuilt workspace reconciliation.",
				Command:     "coder prebuilds pause",
			},
			cli.Example{
				Description: "Resume Coder prebuilt workspace reconciliation if it has been paused.",
				Command:     "coder prebuilds resume",
			},
		),
		Aliases: []string{"prebuild"},
		Handler: func(inv *serpent.Invocation) error {
			return inv.Command.HelpHandler(inv)
		},
		Children: []*serpent.Command{
			r.pausePrebuilds(),
			r.resumePrebuilds(),
		},
	}
	return cmd
}

func (r *RootCmd) pausePrebuilds() *serpent.Command {
	client := new(codersdk.Client)
	cmd := &serpent.Command{
		Use:   "pause",
		Short: "Pause prebuilds",
		Middleware: serpent.Chain(
			serpent.RequireNArgs(0),
			r.InitClient(client),
		),
		Handler: func(inv *serpent.Invocation) error {
			err := client.PutPrebuildsSettings(inv.Context(), codersdk.PrebuildsSettings{
				ReconciliationPaused: true,
			})
			if err != nil {
				return xerrors.Errorf("unable to pause prebuilds: %w", err)
			}

			_, _ = fmt.Fprintln(inv.Stderr, "Prebuilds are now paused.")
			return nil
		},
	}
	return cmd
}

func (r *RootCmd) resumePrebuilds() *serpent.Command {
	client := new(codersdk.Client)
	cmd := &serpent.Command{
		Use:   "resume",
		Short: "Resume prebuilds",
		Middleware: serpent.Chain(
			serpent.RequireNArgs(0),
			r.InitClient(client),
		),
		Handler: func(inv *serpent.Invocation) error {
			err := client.PutPrebuildsSettings(inv.Context(), codersdk.PrebuildsSettings{
				ReconciliationPaused: false,
			})
			if err != nil {
				return xerrors.Errorf("unable to resume prebuilds: %w", err)
			}

			_, _ = fmt.Fprintln(inv.Stderr, "Prebuilds are now resumed.")
			return nil
		},
	}
	return cmd
}