summaryrefslogtreecommitdiffstats
path: root/commands/gen.go
diff options
context:
space:
mode:
Diffstat (limited to 'commands/gen.go')
-rw-r--r--commands/gen.go5
1 files changed, 3 insertions, 2 deletions
diff --git a/commands/gen.go b/commands/gen.go
index 7ff75372a..fde5ae3bc 100644
--- a/commands/gen.go
+++ b/commands/gen.go
@@ -202,12 +202,13 @@ func (c *genCommand) Run(ctx context.Context, cd *simplecobra.Commandeer, args [
return nil
}
-func (c *genCommand) WithCobraCommand(cmd *cobra.Command) error {
+func (c *genCommand) Init(cd *simplecobra.Commandeer) error {
+ cmd := cd.CobraCommand
cmd.Short = "A collection of several useful generators."
return nil
}
-func (c *genCommand) Init(cd, runner *simplecobra.Commandeer) error {
+func (c *genCommand) PreRun(cd, runner *simplecobra.Commandeer) error {
c.rootCmd = cd.Root.Command.(*rootCommand)
return nil
}