feat(cmd): add output option

Author: coffeemakingtoaster

.gitignore

@@ -73,3 +73,4 @@ tags
 # End of https://www.toptal.com/developers/gitignore/api/go,linux,vim,executable
 
 *.benchmark
+out

cmd/dockerfile-parser/dockerfile-parser.go

@@ -12,7 +12,8 @@ import (
 func main() {
 	startTime := time.Now()
 	recursive := slices.Contains(os.Args, "-r")
-	count := wrapper.ParsePath(os.Args[len(os.Args)-1], recursive)
+	output := slices.Contains(os.Args, "-o")
+	count := wrapper.ParsePath(os.Args[len(os.Args)-1], recursive, output)
 	diff := time.Now().Sub(startTime)
 	fmt.Printf("Parsing %d files finished in %v\n", count, diff)
 }

internal/pkg/wrapper/wrapper.go

@@ -7,21 +7,22 @@ import (
 	"strings"
 
 	"github.com/coffeemakingtoaster/dockerfile-parser/internal/pkg/display"
+	"github.com/coffeemakingtoaster/dockerfile-parser/pkg/ast"
 	"github.com/coffeemakingtoaster/dockerfile-parser/pkg/lexer"
 	"github.com/coffeemakingtoaster/dockerfile-parser/pkg/parser"
 )
 
-func ParsePath(path string, recursive bool) int {
+func ParsePath(path string, recursive, output bool) int {
 	isFile, err := isFile(path)
 	if err != nil {
 		panic(err)
 	}
 	if isFile {
-		parseAndDisplayFileList([]string{path})
+		parseAndDisplayFileList([]string{path}, output)
 		return 1
 	} else {
 		paths := buildDirPathList(path, recursive)
-		parseAndDisplayFileList(paths)
+		parseAndDisplayFileList(paths, output)
 		return len(paths)
 	}
 }
@@ -66,7 +67,7 @@ func isFile(path string) (bool, error) {
 	}
 }
 
-func parseAndDisplayFileList(paths []string) {
+func parseAndDisplayFileList(paths []string, output bool) {
 	for _, path := range paths {
 		fmt.Printf("---\t%s\t---\n", path)
 		l, err := lexer.NewFromFile(path)
@@ -85,5 +86,18 @@ func parseAndDisplayFileList(paths []string) {
 			continue
 		}
 		display.DisplayAst(root)
+		if output {
+			outputReconstructed(root, filepath.Base(path))
+		}
+	}
+}
+
+func outputReconstructed(root *ast.StageNode, filename string) {
+	os.MkdirAll("./out", 0755)
+	content := root.Reconstruct()
+	data := strings.Join(content, "\n")
+	err := os.WriteFile(filepath.Join("./out", filename), []byte(data), 0755)
+	if err != nil {
+		fmt.Printf("Something went wrong: %s", err.Error())
 	}
 }

pkg/ast/reconstruct.go

@@ -129,7 +129,7 @@ func (oi *OnbuildInstructionNode) Reconstruct() []string {
 func (ri *RunInstructionNode) Reconstruct() []string {
 	reconstructed := fmt.Sprintf("%s ", ri.Instruction())
 	if !ri.ShellForm && !ri.IsHeredoc {
-		return []string{reconstructed + escapeSlice(strings.Split(ri.Cmd[0], " "))}
+		return []string{reconstructed + escapeSlice(ri.Cmd)}
 	}
 	if ri.IsHeredoc {
 		reconstructed += "<< "

pkg/ast/reconstruct_test.go

@@ -218,7 +218,7 @@ func TestReconstructInstruction(t *testing.T) {
 			Input: ast.StageNode{
 				Instructions: []ast.InstructionNode{
 					&ast.RunInstructionNode{
-						Cmd:       []string{"curl google.com"},
+						Cmd:       []string{"curl", "google.com"},
 						ShellForm: false,
 					},
 				},