feat: improve cli interation

cli/args.go

@@ -0,0 +1,41 @@
+package cli
+
+import (
+	"flag"
+	"fmt"
+	"os"
+)
+
+type Options struct {
+	PrintAst bool
+}
+
+type Args struct {
+	Options   Options
+	Input     string
+	Arguments []string
+}
+
+const HELP_HEADER = `Usage: umbra [options] [[file] [arguments]]
+
+Options:`
+
+func Parse() Args {
+	parsedArgs := Args{}
+
+	flag.BoolVar(&parsedArgs.Options.PrintAst, "ast", false, "Prints the AST of the program")
+	flag.Usage = func() {
+		fmt.Fprintln(os.Stderr, HELP_HEADER)
+		flag.PrintDefaults()
+	}
+	flag.Parse()
+
+	args := flag.Args()
+
+	if len(args) > 0 {
+		parsedArgs.Input = args[0]
+		parsedArgs.Arguments = args[1:]
+	}
+
+	return parsedArgs
+}

cli/print-ast.go

@@ -0,0 +1,10 @@
+package cli
+
+import (
+	"github.com/pmqueiroz/umbra/ast"
+	"github.com/sanity-io/litter"
+)
+
+func PrintAst(module ast.ModuleStatement) {
+	litter.Dump(module)
+}

go.mod

@@ -1,3 +1,5 @@
 module github.com/pmqueiroz/umbra
 
 go 1.21.6
+
+require github.com/sanity-io/litter v1.5.5 // indirect

go.sum

@@ -0,0 +1,5 @@
+github.com/davecgh/go-spew v0.0.0-20161028175848-04cdfd42973b/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/pmezard/go-difflib v0.0.0-20151028094244-d8ed2627bdf0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/sanity-io/litter v1.5.5 h1:iE+sBxPBzoK6uaEP5Lt3fHNgpKcHXc/A2HGETy0uJQo=
+github.com/sanity-io/litter v1.5.5/go.mod h1:9gzJgR2i4ZpjZHsKvUXIRQVk7P+yM3e+jAF7bU2UI5U=
+github.com/stretchr/testify v0.0.0-20161117074351-18a02ba4a312/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=

main.go

@@ -2,26 +2,17 @@ package main
 
 import (
 	"bufio"
-	"encoding/json"
 	"fmt"
 	"os"
 	"strings"
 
 	"github.com/pmqueiroz/umbra/ast"
+	"github.com/pmqueiroz/umbra/cli"
+	"github.com/pmqueiroz/umbra/interpreter"
 	"github.com/pmqueiroz/umbra/tokens"
 )
 
-func jsonASTPrint(module ast.ModuleStatement) {
-	astJson, err := json.MarshalIndent(module, "", "  ")
-	if err != nil {
-		fmt.Println("Failed to format ast JSON:", err)
-		return
-	}
-
-	fmt.Println(string(astJson))
-}
-
-func run(contents ...string) {
+func run(options cli.Options, contents ...string) {
 	content := strings.Join(contents, "\n")
 	tokens, err := tokens.Tokenize(content)
 
@@ -31,22 +22,28 @@ func run(contents ...string) {
 
 	module := ast.Parse(tokens)
 
-	fmt.Printf("%#v\n", module)
+	if options.PrintAst {
+		cli.PrintAst(module)
+	}
+
+	env := interpreter.NewEnvironment(nil)
 
-	jsonASTPrint(module)
+	if err := interpreter.Interpret(module, env); err != nil {
+		fmt.Println("Erro:", err)
+	}
 }
 
-func runFile(path string) {
+func runFile(path string, options cli.Options) {
 	fileContent, err := readFile(path)
 
 	if err != nil {
 		fmt.Printf("%s\n", err.Error())
 	}
 
-	run(fileContent)
+	run(options, fileContent)
 }
 
-func runPrompt() {
+func runPrompt(options cli.Options) {
 	reader := bufio.NewReader(os.Stdin)
 
 	fmt.Print("Welcome to Umbra REPL!\nEnter :q to exit.\n")
@@ -59,21 +56,19 @@ func runPrompt() {
 		}
 
 		run(
+			options,
 			"module main",
 			line,
 		)
 	}
 }
 
 func main() {
-	args := os.Args[1:]
+	args := cli.Parse()
 
-	if len(args) > 1 {
-		fmt.Println("Usage: umbra [script]")
-		os.Exit(64)
-	} else if len(args) == 1 {
-		runFile(args[0])
+	if args.Input != "" {
+		runFile(args.Input, args.Options)
 	} else {
-		runPrompt()
+		runPrompt(args.Options)
 	}
 }