Merge pull request #5 from sagar290/RegisterCommand

Hotfix: Register command issue
Shipu · Mar 7, 2023 · 72de70c · 72de70c
2 parents b15d8ac + 2c9c4ed
commit 72de70c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/cmd/generate/cmd.go b/cmd/generate/cmd.go
@@ -1,7 +1,6 @@
 package generate
 
 import (
-	"fmt"
 	"github.com/spf13/afero"
 	"github.com/spf13/cobra"
 	"log"
@@ -35,7 +34,6 @@ func makeCommand(cmd *cobra.Command, args []string) error {
 	return nil
 }
 
-
 func createCMDFolders(fs afero.Fs, name string) {
 	fs.Mkdir(name, 0755)
 	fs.Mkdir(name+"/commands", 0755)

diff --git a/cmd/main.go b/cmd/main.go
@@ -17,13 +17,14 @@ func hello(cmd *cobra.Command, args []string) {
 	log.Println("art command")
 }
 
-// add command if necessary
-func AddCommand(cmd *cobra.Command)  {
+// AddCommand add custom command function if necessary
+func AddCommand(cmd *cobra.Command) {
 	RootCmd.AddCommand(cmd)
 }
 
 func init() {
 	RootCmd.AddCommand(generate.CrudCmd)
+	RootCmd.AddCommand(generate.MakeCommand)
 }
 
 func Execute() {