/
package_add_plan.go
73 lines (61 loc) · 1.55 KB
/
package_add_plan.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package cmd
import (
"io"
"github.com/spf13/afero"
"github.com/spf13/cobra"
"github.com/kudobuilder/kudo/pkg/kudoctl/cmd/generate"
"github.com/kudobuilder/kudo/pkg/kudoctl/cmd/prompt"
)
const (
pkgAddPlanDesc = `Adds a plan to existing operator package files.
`
pkgAddPlanExample = ` kubectl kudo package add plan
`
)
type packageAddPlanCmd struct {
path string
interactive bool
out io.Writer
fs afero.Fs
}
// newPackageAddPlanCmd adds a plan to an exist operator package
func newPackageAddPlanCmd(fs afero.Fs, out io.Writer) *cobra.Command {
pkg := &packageAddPlanCmd{out: out, fs: fs}
cmd := &cobra.Command{
Use: "plan",
Short: "adds a plan to the operator.yaml file",
Long: pkgAddPlanDesc,
Example: pkgAddPlanExample,
RunE: func(cmd *cobra.Command, args []string) error {
path, err := generate.OperatorPath(fs)
if err != nil {
return err
}
pkg.path = path
if err := pkg.run(); err != nil {
return err
}
return nil
},
}
f := cmd.Flags()
f.BoolVarP(&pkg.interactive, "interactive", "i", false, "Interactive mode.")
return cmd
}
func (pkg *packageAddPlanCmd) run() error {
planNames, err := generate.PlanNameList(pkg.fs, pkg.path)
if err != nil {
return err
}
// get list of tasks
tasks, err := generate.TaskList(pkg.fs, pkg.path)
if err != nil {
return err
}
// interactive mode
planName, plan, err := prompt.ForPlan(planNames, tasks, pkg.fs, pkg.path, createTaskFromPrompts)
if err != nil {
return err
}
return generate.AddPlan(pkg.fs, pkg.path, planName, plan)
}