Saving progress: trying to test actual controller generation

Author: ehutchllew

.gitignore

@@ -19,3 +19,5 @@
 
 # Go workspace file
 go.work
+
+test/

cmd/generate.cmd.go

@@ -1,13 +1,71 @@
 package cmd
 
 import (
+	"fmt"
+	"log"
+	"os"
 	"sync"
+	"text/template"
 
 	"github.com/ev-the-dev/bootmaker/models"
+	ct "github.com/ev-the-dev/bootmaker/templates"
 )
 
 var wg sync.WaitGroup
 
+var tempFuncs = template.FuncMap{
+	"formatModuleName": formatModuleName,
+}
+
 func generateFiles(answers *models.WizardAnswers) {
+	defer wg.Wait()
+
+	cwd, cwdErr := os.Getwd()
+	if cwdErr != nil {
+		log.Fatalf("Unable to get Current Working Directory: %v", cwdErr)
+	}
+
+	if answers.Controller {
+		wg.Add(1)
+		go generateController(cwd, answers.ModuleName)
+	}
+}
+
+func generateController(cwd string, moduleName string) error {
+	fmt.Printf("\n#####\n%s\n#####\n", yellowText("Generating Controller"))
+	mkDirErr := os.MkdirAll(fmt.Sprintf("%s/controllers/", cwd), 0744)
+	if mkDirErr != nil {
+		fmt.Printf("\n!!!!!\n%s\n!!!!!\nerr: %v\n", redText("Unable to create controller directory"), mkDirErr)
+		os.Exit(1)
+	}
+
+	w, err := os.Create(fmt.Sprintf("%s/controllers/%s.controller.ts", cwd, moduleName))
+	if err != nil {
+		fmt.Printf("\n!!!!!\n%s\n!!!!!\nerr: %v\n", redText("Unable to create controller file"), err)
+		os.Exit(1)
+	}
+	defer w.Close()
+
+	cTmpl := template.Must(template.New("controller").Funcs(tempFuncs).Parse(ct.ControllerTemplate))
+	cTmpl.Execute(w, "")
+
+	fmt.Printf("\n#####\n%s\n#####\n", greenText("Done Generating Controller"))
+	wg.Done()
+	return nil
+}
+
+func greenText(text string) string {
+	return fmt.Sprintf("\x1b[32m%s\x1b[0m", text)
+}
+
+func redText(text string) string {
+	return fmt.Sprintf("\x1b[31m%s\x1b[0m", text)
+}
+
+func yellowText(text string) string {
+	return fmt.Sprintf("\x1b[33m%s\x1b[0m", text)
+}
 
+func formatModuleName(name string) string {
+	return "TEST"
 }

cmd/root.cmd.go

@@ -63,7 +63,6 @@ func Execute() {
 		Service:       updatedWS.selected[models.SERVICE],
 	}
 
-	// pass wA to write stream
 	fmt.Printf("wizard answers::: %+v: ", wA)
 	generateFiles(wA)
 }

templates/controller.template.go

@@ -3,7 +3,7 @@ package templates
 // NOTE: if module contains hyphen (-) need to consolidate that
 // to PascalCase
 var ControllerTemplate = `
-{{with $pMN := call formatModuleName .ModuleName}}
+{{with $pMN := call .formatModuleName .ModuleName}}
 import { Controller } from "@nestjs/common"
 
 import { {{.ModuleName}}ControllerAdapters } from "@{{.ModuleName}}/adapters/controller.adapters"
@@ -16,7 +16,3 @@ export class {{.ModuleName}}Controller {
 }
 {{end}}
 `
-
-func formatModuleName(name string) string {
-	return "TEST"
-}