feat: cli argv support (wip)

Author: soulteary

internal/updater/updater.go

@@ -92,34 +92,43 @@ func ScanFiles(rootDir string) ([]string, error) {
 	return files, nil
 }
 
-func UpdateConfInDir(rootDir string, fn func(s string) (string, error)) error {
-
+func UpdateConfInDir(rootDir string, outputDir string, indent int, fn func(s string) (string, error)) error {
 	files, err := ScanFiles(rootDir)
 	if err != nil {
 		return err
 	}
 	for _, file := range files {
-		data, err := os.ReadFile(file)
+		buf, err := os.ReadFile(file)
 		if err != nil {
+			fmt.Printf("Formatter Nginx Conf %s failed, can not open the file\n", err)
 			return err
 		}
-		modifiedData, err := fn(FixVars(FixReturn(EncodeEscapeChars(string(data)))))
+		modifiedData, err := fn(FixVars(FixReturn(EncodeEscapeChars(string(buf)))))
 		if err != nil {
+			fmt.Printf("Formatter Nginx Conf %s failed, can not format the file\n", err)
 			return err
 		}
 
-		err = os.WriteFile(file, []byte(DecodeEscapeChars(modifiedData)), 0644)
+		output := ""
+		relPath, err := filepath.Rel(rootDir, file)
+		if err != nil {
+			output = filepath.Join(outputDir, file)
+		} else {
+			output = filepath.Join(outputDir, relPath)
+		}
+
+		err = os.WriteFile(output, []byte(DecodeEscapeChars(modifiedData)), 0644)
 		if err != nil {
+			fmt.Printf("Formatter Nginx Conf %s failed, can not save the file\n", output)
 			return err
 		}
 
-		rel, err := filepath.Rel(rootDir, file)
+		relPath, err = filepath.Rel(rootDir, output)
 		if err != nil {
-			fmt.Printf("Formatter Nginx Conf %s Successed\n", file)
+			fmt.Printf("Formatter Nginx Conf %s Successed\n", output)
 		} else {
-			fmt.Printf("Formatter Nginx Conf %s Successed\n", rel)
+			fmt.Printf("Formatter Nginx Conf %s Successed\n", relPath)
 		}
 	}
-
 	return nil
 }

main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"log"
 	"os"
@@ -10,15 +11,59 @@ import (
 	"github.com/soulteary/nginx-formatter/internal/version"
 )
 
+var FORMATTER_SRC = ""
+var FORMATTER_DEST = ""
+var FORMATTER_INDENT = 2
+
+func InitArgv() {
+	var inputDir string
+	var outputDir string
+	var indent int
+	flag.StringVar(&inputDir, "input", "", "Input directory")
+	flag.StringVar(&outputDir, "output", "", "Output directory")
+	flag.IntVar(&indent, "indent", 2, "Indent size")
+	flag.Parse()
+
+	if inputDir == "" {
+		dir, err := os.Getwd()
+		if err != nil {
+			log.Fatal(err)
+		}
+		fmt.Println("No input directory specified, use the current working directory:", dir)
+		FORMATTER_SRC = dir
+	} else {
+		fmt.Println("Specify the working directory as:", inputDir)
+		FORMATTER_SRC = inputDir
+	}
+
+	if outputDir == "" {
+		dir, err := os.Getwd()
+		if err != nil {
+			log.Fatal(err)
+		}
+		fmt.Println("No output directory specified, use the current working directory:", dir)
+		FORMATTER_DEST = dir
+	} else {
+		err := os.MkdirAll(outputDir, 0755)
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+	}
+
+	if indent <= 0 {
+		FORMATTER_INDENT = 2
+	} else {
+		FORMATTER_INDENT = indent
+	}
+}
+
 func main() {
 	fmt.Printf("Nginx Formatter v%s\n\n", version.Version)
 
-	dir, err := os.Getwd()
-	if err != nil {
-		log.Fatal(err)
-	}
+	InitArgv()
 
-	err = updater.UpdateConfInDir(dir, formatter.Formatter)
+	err := updater.UpdateConfInDir(FORMATTER_SRC, FORMATTER_DEST, FORMATTER_INDENT, formatter.Formatter)
 	if err != nil {
 		log.Fatal(err)
 	}