vgo-support - re-working code generator

cobra/cmd/init.go

@@ -15,15 +15,20 @@ package cmd
 
 import (
 	"fmt"
+	"github.com/spf13/cobra/cobra/tpl"
 	"os"
 	"path"
 	"path/filepath"
+	"text/template"
 
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
 )
 
-var initCmd = &cobra.Command{
+var (
+	pkgName string
+
+	initCmd = &cobra.Command{
 		Use:     "init [name]",
 		Aliases: []string{"initialize", "initialise", "create"},
 		Short:   "Initialize a Cobra Application",
@@ -40,6 +45,32 @@ and the appropriate structure for a Cobra-based CLI application.
 Init will not use an existing directory with contents.`,
 
 		Run: func(cmd *cobra.Command, args []string) {
+
+			wd, err := os.Getwd()
+			if err != nil {
+				er(err)
+			}
+
+			project := &Project{
+				AbsolutePath: wd,
+				PkgName:      pkgName,
+				Legal:        getLicense(),
+				Copyright:    copyrightLine(),
+			}
+
+			// open file for writing
+			f, err := os.Create(fmt.Sprintf("%s/main.go", project.AbsolutePath))
+			if err != nil {
+				er(err)
+			}
+			defer f.Close()
+
+			t := template.Must(template.New("init").Parse(string(tpl.MainTemplate())))
+			err = t.Execute(f, project)
+			if err != nil {
+				er(err)
+			}
+			/*
 				wd, err := os.Getwd()
 				if err != nil {
 					er(err)
@@ -63,14 +94,17 @@ Init will not use an existing directory with contents.`,
 				}
 
 				initializeProject(project)
+			*/
 
-		fmt.Fprintln(cmd.OutOrStdout(), `Your Cobra application is ready at
+			fmt.Printf("Your Cobra applicaton is ready at\n%s\n", project.AbsolutePath)
-`+project.AbsPath()+`
-
-Give it a try by going there and running `+"`go run main.go`."+`
-Add commands to it by running `+"`cobra add [cmdname]`.")
 		},
 	}
+)
+
+func init() {
+	initCmd.Flags().StringVar(&pkgName, "pkg-name", "", "fully qualified pkg name")
+	initCmd.MarkFlagRequired("pkg-name")
+}
 
 func initializeProject(project *Project) {
 	if !exists(project.AbsPath()) { // If path doesn't yet exist, create it

cobra/cmd/project.go

@@ -9,6 +9,13 @@ import (
 
 // Project contains name, license and paths to projects.
 type Project struct {
+	// v2
+	PkgName      string
+	Copyright    string
+	AbsolutePath string
+	Legal        License
+
+	// v1
 	absPath string
 	cmdPath string
 	srcPath string

cobra/cmd/root.go

@@ -23,7 +23,8 @@ import (
 
 var (
 	// Used for flags.
-	cfgFile, userLicense string
+	cfgFile     string
+	userLicense string
 
 	rootCmd = &cobra.Command{
 		Use:   "cobra",

cobra/tpl/main.go

@@ -0,0 +1,17 @@
+package tpl
+
+func MainTemplate() []byte {
+	return []byte(`
+/*
+{{ .Copyright }}
+{{if .Legal.Header}}{{ .Legal.Header }}{{end}}
+*/
+package main
+
+import "{{ .PkgName }}/cmd"
+
+func main() {
+	cmd.Execute()
+}
+`)
+}