Improved instruction parser

src/build/Function.go

@@ -14,6 +14,7 @@ import (
 // Function represents a function.
 type Function struct {
 	Name      string
+	File      *File
 	Head      token.List
 	Body      token.List
 	Variables map[string]*Variable
@@ -24,32 +25,60 @@ type Function struct {
 // Compile turns a function into machine code.
 func (f *Function) Compile() {
 	if config.Verbose {
-		ansi.Underline.Println(f.Name)
+		ansi.Bold.Println(f.Name)
 	}
 
-	for _, line := range f.Lines() {
-		if len(line) == 0 {
+	start := 0
+	groupLevel := 0
+
+	for i, t := range f.Body {
+		if start == i && (t.Kind == token.NewLine || t.Kind == token.BlockStart || t.Kind == token.BlockEnd) {
+			start = i + 1
 			continue
 		}
 
-		if config.Verbose {
-			ansi.Dim.Print("[   ] ")
-			fmt.Println(line)
-		}
+		switch t.Kind {
+		case token.NewLine:
+			if groupLevel > 0 {
+				continue
+			}
 
-		err := f.compileInstruction(line)
+			if start != -1 {
+				instruction := f.Body[start:i]
+				err := f.CompileInstruction(instruction)
 
-		if err != nil {
-			ansi.Red.Println(err)
-			return
+				if err != nil {
+					f.Error = err
+					return
+				}
+
+				start = -1
+			}
+
+			start = i + 1
+
+		case token.GroupStart:
+			groupLevel++
+
+		case token.GroupEnd:
+			groupLevel--
+
+		case token.BlockStart:
+			// Add scope
+
+		case token.BlockEnd:
+			// Remove scope
 		}
 	}
 
 	f.Assembler.Return()
 
 	if config.Verbose {
+		fmt.Println()
+		ansi.Bold.Println(f.Name + ".asm")
+
 		for _, x := range f.Assembler.Instructions {
-			ansi.Dim.Print("[asm] ")
+			ansi.Dim.Print("│ ")
 			fmt.Print(x.Mnemonic.String())
 
 			if x.Data != nil {
@@ -61,27 +90,44 @@ func (f *Function) Compile() {
 	}
 }
 
-// compileInstruction compiles a single instruction.
-func (f *Function) compileInstruction(line token.List) error {
-	switch line[0].Kind {
-	case token.Keyword:
+// CompileInstruction compiles a single instruction.
+func (f *Function) CompileInstruction(line token.List) error {
+	if config.Verbose {
+		ansi.Dim.Print("│ ")
+		fmt.Println(line)
+	}
+
+	if line[0].Kind == token.Keyword {
 		switch line[0].Text() {
 		case "return":
 			f.Assembler.Return()
 		}
 
-	case token.Identifier:
-		if len(line) >= 2 && line[1].Kind == token.Define {
+		return nil
+	}
+
+	if line[0].Kind == token.Identifier {
+		if len(line) < 2 {
+			return fmt.Errorf("error to be implemented")
+		}
+
+		if line[1].Kind == token.Define {
 			name := line[0].Text()
 			value := line[2:]
 
+			if len(value) == 0 {
+				return fmt.Errorf("error to be implemented")
+				// return errors.New(errors.MissingAssignmentValue, f.File.Path, f.File.Tokens, f.Cursor)
+			}
+
 			if config.Verbose {
-				ansi.Dim.Printf("[var] ")
-				fmt.Println(name, value)
+				ansi.Dim.Printf("├── var ")
+				fmt.Println(name, "=", value)
 			}
 
 			f.Variables[name] = &Variable{
-				Value:   line[2:],
+				Name:    name,
+				Value:   value,
 				IsConst: true,
 			}
 
@@ -115,7 +161,7 @@ func (f *Function) compileInstruction(line token.List) error {
 					variable, exists := f.Variables[name]
 
 					if !exists {
-						return fmt.Errorf("Unknown identifier '%s'", name)
+						panic("Unknown identifier " + name)
 					}
 
 					if !variable.IsConst {
@@ -142,30 +188,6 @@ func (f *Function) compileInstruction(line token.List) error {
 	return nil
 }
 
-// Lines returns the lines in the function body.
-func (f *Function) Lines() []token.List {
-	var (
-		lines []token.List
-		start = 0
-		i     = 0
-	)
-
-	for i < len(f.Body) {
-		if f.Body[i].Kind == token.NewLine {
-			lines = append(lines, f.Body[start:i])
-			start = i + 1
-		}
-
-		i++
-	}
-
-	if i != start {
-		lines = append(lines, f.Body[start:i])
-	}
-
-	return lines
-}
-
 // String returns the function name.
 func (f *Function) String() string {
 	return f.Name