Refactor to separate packages

Fix linter errors

Author: gordonbondon

.gitignore

@@ -7,3 +7,4 @@
 !.vscode/tasks.json
 !.vscode/launch.json
 !.vscode/extensions.json
+terraform-variables-generator

CHANGELOG.md

@@ -0,0 +1,4 @@
+## (Unreleased)
+
+ENHANCEMENTS:
+* Optimize project structure, split to separate packages and add cobra for cli

cmd/cmd.go

@@ -0,0 +1,63 @@
+package cmd
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/spf13/cobra"
+
+	log "github.com/sirupsen/logrus"
+
+	"github.com/alexandrst88/terraform-variables-generator/pkg/generator"
+	"github.com/alexandrst88/terraform-variables-generator/pkg/utils"
+)
+
+const (
+	tfFileExt = "*.tf"
+)
+
+var (
+	generatorVersion string
+
+	vars     bool
+	varsFile string
+)
+
+// Execute will run main logic
+func Execute(version string) {
+	generatorVersion = version
+
+	cmd := &cobra.Command{
+		Use:     "generator",
+		Short:   "CLI for generating terraform variables",
+		Example: "  terraform-variable-generator",
+		Version: generatorVersion,
+		Run:     runGenerator,
+	}
+
+	cmd.PersistentFlags().BoolVar(&vars, "vars", true, "generate variables")
+	cmd.PersistentFlags().StringVar(&varsFile, "vars-file", "./variables.tf", "path to generated variables file")
+
+	if err := cmd.Execute(); err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+	os.Exit(0)
+}
+
+func runGenerator(cmd *cobra.Command, args []string) {
+	if vars {
+		if utils.FileExists(varsFile) {
+			utils.UserPromt(varsFile)
+		}
+
+		tfFiles, err := utils.GetAllFiles(tfFileExt)
+		utils.CheckError(err)
+		if len(tfFiles) == 0 {
+			log.Warn("No terraform files to proceed, exiting")
+			return
+		}
+
+		generator.GenerateVars(tfFiles, varsFile)
+	}
+}

helpers.go

@@ -0,0 +1,10 @@
+package main
+
+import "github.com/alexandrst88/terraform-variables-generator/cmd"
+
+// Version is updated by linker flags during build time
+var Version = "dev"
+
+func main() {
+	cmd.Execute(Version)
+}

main.go

@@ -1,9 +1,11 @@
-package main
+package generator
 
 import (
 	"regexp"
 	"sort"
 	"strings"
+
+	"github.com/alexandrst88/terraform-variables-generator/pkg/utils"
 )
 
 type terraformVars struct {
@@ -16,13 +18,13 @@ func (t *terraformVars) matchVarPref(row, varPrefix string) {
 		match := pattern.FindAllStringSubmatch(row, -1)
 		for _, m := range match {
 			res := replacer.Replace(m[0])
-			if !containsElement(t.Variables, res) {
+			if !utils.ContainsElement(t.Variables, res) {
 				t.Variables = append(t.Variables, res)
 			}
 		}
 	}
 }
 
 func (t *terraformVars) sortVars() {
-	sort.Sort(sort.StringSlice(t.Variables))
+	sort.Strings(t.Variables)
 }

configuration.mock renamed to pkg/generator/configuration.mock

@@ -1,4 +1,4 @@
-package main
+package generator
 
 import (
 	"bufio"
@@ -8,13 +8,13 @@ import (
 	"sync"
 
 	log "github.com/sirupsen/logrus"
+
+	"github.com/alexandrst88/terraform-variables-generator/pkg/utils"
 )
 
 var replacer *strings.Replacer
 var varPrefix = "var."
-var tfFileExt = "*.tf"
 
-var dstFile = "./variables.tf"
 var varTemplate = template.Must(template.New("var_file").Parse(`{{range .}}
 variable "{{ . }}" {
   description = ""
@@ -35,17 +35,9 @@ func init() {
 		" ", "",
 	)
 }
-func main() {
-	if fileExists(dstFile) {
-		userPromt()
-	}
 
-	tfFiles, err := getAllFiles(tfFileExt)
-	if len(tfFiles) == 0 {
-		log.Warn("No terraform files to proceed, exiting")
-		os.Exit(0)
-	}
-	checkError(err)
+// GenerateVars will write generated vars to file
+func GenerateVars(tfFiles []string, dstFile string) {
 	var wg sync.WaitGroup
 	messages := make(chan string)
 	wg.Add(len(tfFiles))
@@ -69,11 +61,10 @@ func main() {
 	}()
 	wg.Wait()
 	f, err := os.Create(dstFile)
-	checkError(err)
+	utils.CheckError(err)
 
 	t.sortVars()
 	err = varTemplate.Execute(f, t.Variables)
-	checkError(err)
+	utils.CheckError(err)
 	log.Infof("Variables are generated to %q file", dstFile)
-
 }

terraform.go renamed to pkg/generator/terraform.go

@@ -1,24 +1,25 @@
-package main
+package generator
 
 import (
 	"bufio"
 	"os"
 	"testing"
+
+	"github.com/alexandrst88/terraform-variables-generator/pkg/utils"
 )
 
-var mockFile = "tf_configuration.mock"
 var testExtFile = "*.mock"
 
 func TestContainsElement(t *testing.T) {
 	testSlice := []string{"Terraform", "Puppet", "Ansible"}
-	if containsElement(testSlice, "Chef") {
+	if utils.ContainsElement(testSlice, "Chef") {
 		t.Error("Should return false, but return true")
 	}
 }
 
 func TestGetAllFiles(t *testing.T) {
-	files, err := getAllFiles(testExtFile)
-	checkError(err)
+	files, err := utils.GetAllFiles(testExtFile)
+	utils.CheckError(err)
 	if len(files) == 0 {
 		t.Error("Should found at least one file")
 	}
@@ -28,8 +29,8 @@ func TestMatchVariable(t *testing.T) {
 	ter := &terraformVars{}
 	var messages []string
 
-	file, err := getAllFiles(testExtFile)
-	checkError(err)
+	file, err := utils.GetAllFiles(testExtFile)
+	utils.CheckError(err)
 
 	fileHandle, _ := os.Open(file[0])
 	defer fileHandle.Close()

generator.go renamed to pkg/generator/vars.go

@@ -1,4 +1,4 @@
-package main
+package utils
 
 import (
 	"os"
@@ -7,21 +7,24 @@ import (
 	log "github.com/sirupsen/logrus"
 )
 
-func getAllFiles(ext string) ([]string, error) {
+// GetAllFiles will get all files in current directory
+func GetAllFiles(ext string) ([]string, error) {
 	dir, err := os.Getwd()
-	checkError(err)
+	CheckError(err)
+
 	var files []string
 	log.Infof("Finding files in %q directory", dir)
 	files, err = filepath.Glob(ext)
-	checkError(err)
+	CheckError(err)
 
 	if len(files) == 0 {
 		log.Infof("No files with %q extensions found in %q", ext, dir)
 	}
 	return files, nil
 }
 
-func fileExists(name string) bool {
+// FileExists checks if file exists
+func FileExists(name string) bool {
 	if _, err := os.Stat(name); err == nil {
 		return true
 	}

generator_test.go renamed to pkg/generator/vars_test.go

@@ -0,0 +1,40 @@
+package utils
+
+import (
+	"fmt"
+	"os"
+
+	log "github.com/sirupsen/logrus"
+)
+
+// CheckError is a convenient wrapper for error check
+func CheckError(e error) {
+	if e != nil {
+		log.Fatal(e)
+	}
+}
+
+// ContainsElement check if slice of strings contains provided string
+func ContainsElement(slice []string, value string) bool {
+	if len(slice) == 0 {
+		return false
+	}
+	for _, s := range slice {
+		if value == s {
+			return true
+		}
+	}
+	return false
+}
+
+// UserPromt will ask user if file needs to be overridden
+func UserPromt(dstFile string) {
+	var response string
+	log.Warnf("File %q already exists, type 'yes' if you want replace it", dstFile)
+	fmt.Print("-> ")
+	_, err := fmt.Scanln(&response)
+	CheckError(err)
+	if response != "yes" {
+		os.Exit(0)
+	}
+}