opensource/coder
3
0
Fork 0
mirror of https://github.com/coder/coder.git synced 2026-09-24 15:04:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: convert entire CLI to clibase (#6491)

Browse Source 
I'm sorry.
This commit is contained in:
Ammar Bandukwala
2023-03-23 17:42:20 -05:00
committed by GitHub
parent b71b8daa21
commit 2bd6d2908e
345 changed files with 9965 additions and 9082 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cli/state.go
+56 -40
View File
@@ -6,78 +6,92 @@ import (
"os"
"strconv"
"github.com/spf13/cobra"
"github.com/coder/coder/cli/clibase"
"github.com/coder/coder/cli/cliui"
"github.com/coder/coder/codersdk"
)
func state() *cobra.Command {
cmd := &cobra.Command{
func (r *RootCmd) state() *clibase.Cmd {
cmd := &clibase.Cmd{
Use: "state",
Short: "Manually manage Terraform state to fix broken workspaces",
RunE: func(cmd *cobra.Command, args []string) error {
return cmd.Help()
Handler: func(inv *clibase.Invocation) error {
return inv.Command.HelpHandler(inv)
},
Children: []*clibase.Cmd{
r.statePull(),
r.statePush(),
},
}
cmd.AddCommand(statePull(), statePush())
return cmd
}
func statePull() *cobra.Command {
var buildNumber int
cmd := &cobra.Command{
func (r *RootCmd) statePull() *clibase.Cmd {
var buildNumber int64
client := new(codersdk.Client)
cmd := &clibase.Cmd{
Use: "pull <workspace> [file]",
Short: "Pull a Terraform state file from a workspace.",
Args: cobra.MinimumNArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
client, err := CreateClient(cmd)
if err != nil {
return err
}
Middleware: clibase.Chain(
clibase.RequireRangeArgs(1, 2),
r.InitClient(client),
),
Handler: func(inv *clibase.Invocation) error {
var err error
var build codersdk.WorkspaceBuild
if buildNumber == 0 {
workspace, err := namedWorkspace(cmd, client, args[0])
workspace, err := namedWorkspace(inv.Context(), client, inv.Args[0])
if err != nil {
return err
}
build = workspace.LatestBuild
} else {
build, err = client.WorkspaceBuildByUsernameAndWorkspaceNameAndBuildNumber(cmd.Context(), codersdk.Me, args[0], strconv.Itoa(buildNumber))
build, err = client.WorkspaceBuildByUsernameAndWorkspaceNameAndBuildNumber(inv.Context(), codersdk.Me, inv.Args[0], strconv.FormatInt(buildNumber, 10))
if err != nil {
return err
}
}
state, err := client.WorkspaceBuildState(cmd.Context(), build.ID)
state, err := client.WorkspaceBuildState(inv.Context(), build.ID)
if err != nil {
return err
}
if len(args) < 2 {
_, _ = fmt.Fprintln(cmd.OutOrStdout(), string(state))
if len(inv.Args) < 2 {
_, _ = fmt.Fprintln(inv.Stdout, string(state))
return nil
}
return os.WriteFile(args[1], state, 0o600)
return os.WriteFile(inv.Args[1], state, 0o600)
},
}
cmd.Flags().IntVarP(&buildNumber, "build", "b", 0, "Specify a workspace build to target by name.")
cmd.Options = clibase.OptionSet{
buildNumberOption(&buildNumber),
}
return cmd
}
func statePush() *cobra.Command {
var buildNumber int
cmd := &cobra.Command{
func buildNumberOption(n *int64) clibase.Option {
return clibase.Option{
Flag: "build",
FlagShorthand: "b",
Description: "Specify a workspace build to target by name. Defaults to latest.",
Value: clibase.Int64Of(n),
}
}
func (r *RootCmd) statePush() *clibase.Cmd {
var buildNumber int64
client := new(codersdk.Client)
cmd := &clibase.Cmd{
Use: "push <workspace> <file>",
Args: cobra.ExactArgs(2),
Short: "Push a Terraform state file to a workspace.",
RunE: func(cmd *cobra.Command, args []string) error {
client, err := CreateClient(cmd)
if err != nil {
return err
}
workspace, err := namedWorkspace(cmd, client, args[0])
Middleware: clibase.Chain(
clibase.RequireNArgs(2),
r.InitClient(client),
),
Handler: func(inv *clibase.Invocation) error {
workspace, err := namedWorkspace(inv.Context(), client, inv.Args[0])
if err != nil {
return err
}
@@ -85,23 +99,23 @@ func statePush() *cobra.Command {
if buildNumber == 0 {
build = workspace.LatestBuild
} else {
build, err = client.WorkspaceBuildByUsernameAndWorkspaceNameAndBuildNumber(cmd.Context(), codersdk.Me, args[0], strconv.Itoa(buildNumber))
build, err = client.WorkspaceBuildByUsernameAndWorkspaceNameAndBuildNumber(inv.Context(), codersdk.Me, inv.Args[0], strconv.FormatInt((buildNumber), 10))
if err != nil {
return err
}
}
var state []byte
if args[1] == "-" {
state, err = io.ReadAll(cmd.InOrStdin())
if inv.Args[1] == "-" {
state, err = io.ReadAll(inv.Stdin)
} else {
state, err = os.ReadFile(args[1])
state, err = os.ReadFile(inv.Args[1])
}
if err != nil {
return err
}
build, err = client.CreateWorkspaceBuild(cmd.Context(), workspace.ID, codersdk.CreateWorkspaceBuildRequest{
build, err = client.CreateWorkspaceBuild(inv.Context(), workspace.ID, codersdk.CreateWorkspaceBuildRequest{
TemplateVersionID: build.TemplateVersionID,
Transition: build.Transition,
ProvisionerState: state,
@@ -109,9 +123,11 @@ func statePush() *cobra.Command {
if err != nil {
return err
}
return cliui.WorkspaceBuild(cmd.Context(), cmd.OutOrStderr(), client, build.ID)
return cliui.WorkspaceBuild(inv.Context(), inv.Stderr, client, build.ID)
},
}
cmd.Flags().IntVarP(&buildNumber, "build", "b", 0, "Specify a workspace build to target by name.")
cmd.Options = clibase.OptionSet{
buildNumberOption(&buildNumber),
}
return cmd
}