Merge pull request #5 from looplanguage/feat/assign_index

Feat/assign index

Author: kanersps

code/code.go

@@ -35,6 +35,8 @@ const (
 
 	OpSetVar
 	OpGetVar
+
+	OpSetIndex
 )
 
 var definitions = map[OpCode]*Definition{
@@ -68,4 +70,6 @@ var definitions = map[OpCode]*Definition{
 
 	OpSetVar: {"OpSetVar", []int{2}},
 	OpGetVar: {"OpGetVar", []int{2}},
+
+	OpSetIndex: {"OpSetIndex", []int{}},
 }

compiler/compile.go

@@ -207,18 +207,40 @@ func (c *Compiler) Compile(node ast.Node, root, identifier, previous string) err
 				}
 
 				c.emit(code.OpSetLocal, s.Index)
-				return nil
+			} else {
+				return fmt.Errorf("undefined variable %s", node.Identifier.Value)
+			}
+		} else {
+			err := c.Compile(node.Value, root, "", previous)
+			if err != nil {
+				return err
 			}
 
-			return fmt.Errorf("undefined variable %s", node.Identifier.Value)
+			c.emit(code.OpSetVar, variable.Index)
 		}
+	case *ast.IndexAssign:
+		// Put the new value on the stack
+		err := c.Compile(node.Value, root, "", "")
+
+		if err != nil {
+			return err
+		}
+
+		// Put the index on the array
+		err = c.Compile(node.Index, root, "", "")
+
+		if err != nil {
+			return err
+		}
+
+		// Put the array on the stack
+		err = c.Compile(node.Object, root, "", "")
 
-		err := c.Compile(node.Value, root, "", previous)
 		if err != nil {
 			return err
 		}
 
-		c.emit(code.OpSetVar, variable.Index)
+		c.emit(code.OpSetIndex)
 	case *ast.Identifier:
 		variable := c.currentScope.FindByName(node.Value, root)
 

go.mod

@@ -2,4 +2,4 @@ module github.com/looplanguage/compiler
 
 go 1.17
 
-require github.com/looplanguage/loop v0.5.2
+require github.com/looplanguage/loop v0.7.0

go.sum

@@ -1,4 +1,4 @@
 github.com/looplanguage/loop v0.5.2 h1:ZukH3aO9Khf5PgwQF3L33ehOP7PBhZt+FKmyxe1MB7o=
 github.com/looplanguage/loop v0.5.2/go.mod h1:H1ENscrP1l2BSqSe4azQ6owr6TkBd7HgaUwDZDeUT6I=
-github.com/looplanguage/loop v0.6.0 h1:TWJlbo81V/yWDq7ZW2HZl4bHnApfO6Rs7qjbKI6pNoE=
-github.com/looplanguage/loop v0.6.0/go.mod h1:H1ENscrP1l2BSqSe4azQ6owr6TkBd7HgaUwDZDeUT6I=
+github.com/looplanguage/loop v0.7.0 h1:60hMnGPXJ0O6xJ/dcrhAlUyplcmZWuTGjW0TTFnTw0M=
+github.com/looplanguage/loop v0.7.0/go.mod h1:H1ENscrP1l2BSqSe4azQ6owr6TkBd7HgaUwDZDeUT6I=