57c73e8239
* Add --repo flag to commands * Example of using common args * Add -r to stop * Add validation to the add helper * Skip prompt for single option * Migrate (mostly) everything to addGetOrChooseCodespaceCommandArgs * Fix typo in logsCmd * Use R instead of r * Update a couple -r usages * Refactor to codespace_selector * Clean up selector, apply it in a couple examples * Use apiClient instead in constructor * Restore addDeprecatedRepoShorthand * Finish implementing the commands * Update existing tests to use the selector * Add tests for selector * linter * Catch case where there's no conflict * Make the flag persistent for ports * Add support to stop
106 lines
3 KiB
Go
106 lines
3 KiB
Go
package codespace
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/cli/cli/v2/internal/codespaces/api"
|
|
"github.com/cli/cli/v2/pkg/cmdutil"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type stopOptions struct {
|
|
selector *CodespaceSelector
|
|
orgName string
|
|
userName string
|
|
}
|
|
|
|
func newStopCmd(app *App) *cobra.Command {
|
|
opts := &stopOptions{}
|
|
|
|
stopCmd := &cobra.Command{
|
|
Use: "stop",
|
|
Short: "Stop a running codespace",
|
|
Args: noArgsConstraint,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
if opts.orgName != "" && opts.selector.codespaceName != "" && opts.userName == "" {
|
|
return cmdutil.FlagErrorf("using `--org` with `--codespace` requires `--user`")
|
|
}
|
|
return app.StopCodespace(cmd.Context(), opts)
|
|
},
|
|
}
|
|
opts.selector = AddCodespaceSelector(stopCmd, app.apiClient)
|
|
stopCmd.Flags().StringVarP(&opts.orgName, "org", "o", "", "The `login` handle of the organization (admin-only)")
|
|
stopCmd.Flags().StringVarP(&opts.userName, "user", "u", "", "The `username` to stop codespace for (used with --org)")
|
|
|
|
return stopCmd
|
|
}
|
|
|
|
func (a *App) StopCodespace(ctx context.Context, opts *stopOptions) error {
|
|
var (
|
|
codespaceName = opts.selector.codespaceName
|
|
repoName = opts.selector.repoName
|
|
ownerName = opts.userName
|
|
)
|
|
|
|
if codespaceName == "" {
|
|
a.StartProgressIndicatorWithLabel("Fetching codespaces")
|
|
codespaces, err := a.apiClient.ListCodespaces(ctx, api.ListCodespacesOptions{
|
|
RepoName: repoName,
|
|
OrgName: opts.orgName,
|
|
UserName: ownerName,
|
|
})
|
|
a.StopProgressIndicator()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to list codespaces: %w", err)
|
|
}
|
|
|
|
var runningCodespaces []*api.Codespace
|
|
for _, c := range codespaces {
|
|
cs := codespace{c}
|
|
if cs.running() {
|
|
runningCodespaces = append(runningCodespaces, c)
|
|
}
|
|
}
|
|
if len(runningCodespaces) == 0 {
|
|
return errors.New("no running codespaces")
|
|
}
|
|
|
|
includeOwner := opts.orgName != ""
|
|
skipPromptForSingleOption := repoName != ""
|
|
codespace, err := chooseCodespaceFromList(ctx, runningCodespaces, includeOwner, skipPromptForSingleOption)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to choose codespace: %w", err)
|
|
}
|
|
codespaceName = codespace.Name
|
|
ownerName = codespace.Owner.Login
|
|
} else {
|
|
a.StartProgressIndicatorWithLabel("Fetching codespace")
|
|
|
|
var c *api.Codespace
|
|
var err error
|
|
|
|
if opts.orgName == "" {
|
|
c, err = a.apiClient.GetCodespace(ctx, codespaceName, false)
|
|
} else {
|
|
c, err = a.apiClient.GetOrgMemberCodespace(ctx, opts.orgName, ownerName, codespaceName)
|
|
}
|
|
a.StopProgressIndicator()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get codespace: %q: %w", codespaceName, err)
|
|
}
|
|
cs := codespace{c}
|
|
if !cs.running() {
|
|
return fmt.Errorf("codespace %q is not running", codespaceName)
|
|
}
|
|
}
|
|
|
|
a.StartProgressIndicatorWithLabel("Stopping codespace")
|
|
defer a.StopProgressIndicator()
|
|
if err := a.apiClient.StopCodespace(ctx, codespaceName, opts.orgName, ownerName); err != nil {
|
|
return fmt.Errorf("failed to stop codespace: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|