pulumi/cmd/stack_select.go
2017-11-02 11:19:00 -07:00

65 lines
1.7 KiB
Go

// Copyright 2016-2017, Pulumi Corporation. All rights reserved.
package cmd
import (
"fmt"
"github.com/pulumi/pulumi/pkg/tokens"
"github.com/spf13/cobra"
"github.com/pulumi/pulumi/pkg/util/cmdutil"
)
// newStackSelectCmd handles both the "local" and "cloud" scenarios in its implementation.
func newStackSelectCmd() *cobra.Command {
return &cobra.Command{
Use: "select [<stack>]",
Short: "Switch the current workspace to the given stack",
Long: "Switch the current workspace to the given stack. This allows you to use\n" +
"other commands like `config`, `preview`, and `push` without needing to specify the\n" +
"stack name each and every time.\n" +
"\n" +
"If no <stack> argument is supplied, the current stack is printed.",
Run: cmdutil.RunFunc(func(cmd *cobra.Command, args []string) error {
// Display the name of the current stack if a new one isn't specified.
if len(args) == 0 {
name, err := getCurrentStack()
if err != nil {
return err
}
fmt.Printf("%v\n", name)
return nil
}
// Gather the list of possible stack names.
var allStacks []tokens.QName
var err error
if !usePulumiCloudCommands() {
allStacks, err = getStacks()
if err != nil {
return err
}
} else {
cloudStacks, err := getCloudStacks()
if err != nil {
return err
}
for _, cloudStack := range cloudStacks {
allStacks = append(allStacks, cloudStack.StackName)
}
}
// Confirm the stack name is valid.
selectedStack := tokens.QName(args[0])
for _, stack := range allStacks {
if stack == selectedStack {
return setCurrentStack(selectedStack)
}
}
return fmt.Errorf("no stack with name '%v' found", selectedStack)
}),
}
}