add Flags to core command for commands without flags

Signed-off-by: Jason McCallister <jason@craftcms.com>
craftcms · Mar 30, 2020 · a6a5e98 · a6a5e98
1 parent 7ecacc7
commit a6a5e98
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/command/core.go b/command/core.go
@@ -1,6 +1,8 @@
 package command
 
 import (
+	"flag"
+
 	"github.com/mitchellh/cli"
 	"github.com/spf13/viper"
 )
@@ -14,3 +16,10 @@ type CoreCommand struct {
 	flagName   string
 	flagDryRun bool
 }
+
+func (c *CoreCommand) Flags() *flag.FlagSet {
+	s := flag.NewFlagSet("init", 128)
+	s.StringVar(&c.flagName, "name", "", "name of the machine")
+	s.BoolVar(&c.flagDryRun, "dry-run", false, "skip executing the command")
+	return s
+}