mirror of
https://github.com/coder/coder.git
synced 2026-06-04 13:38:21 +00:00
Spike Curtis
bddb808b25
Fixes all our Go file imports to match the preferred spec that we've _mostly_ been using. For example: ``` import ( "context" "time" "github.com/prometheus/client_golang/prometheus" "golang.org/x/xerrors" "gopkg.in/natefinch/lumberjack.v2" "cdr.dev/slog/v3" "github.com/coder/coder/v2/codersdk/agentsdk" "github.com/coder/serpent" ) ``` 3 groups: standard library, 3rd partly libs, Coder libs. This PR makes the change across the codebase. The PR in the stack above modifies our formatting to maintain this state of affairs, and is a separate PR so it's possible to review that one in detail.
92 lines
2.1 KiB
Go
92 lines
2.1 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/coder/coder/v2/cli"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
"github.com/coder/serpent"
|
|
)
|
|
|
|
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 {
|
|
cmd := &serpent.Command{
|
|
Use: "pause",
|
|
Short: "Pause prebuilds",
|
|
Middleware: serpent.Chain(
|
|
serpent.RequireNArgs(0),
|
|
),
|
|
Handler: func(inv *serpent.Invocation) error {
|
|
client, err := r.InitClient(inv)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
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 {
|
|
cmd := &serpent.Command{
|
|
Use: "resume",
|
|
Short: "Resume prebuilds",
|
|
Middleware: serpent.Chain(
|
|
serpent.RequireNArgs(0),
|
|
),
|
|
Handler: func(inv *serpent.Invocation) error {
|
|
client, err := r.InitClient(inv)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
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
|
|
}
|