Implement setting values at index in an array

1 files changed, 15 insertions, 0 deletions

diff --git a/pkg/lang/compiler/compiler.go b/pkg/lang/compiler/compiler.go
index d1b98da..193cd90 100644
--- a/pkg/lang/compiler/compiler.go
+++ b/pkg/lang/compiler/compiler.go
@@ -759,6 +759,21 @@ func (comp *Compiler) compileAssignExpr(t *code.Builder, expr ast.ExprBinary) er
 
 		t.AppendOp(code.OpSetMember)
 		t.AppendString(name)
+	case ast.ExprKindSubscription:
+		subscriptionExpr := expr.Left.Value.(ast.ExprSubscription)
+		if err := comp.compileExpr(t, subscriptionExpr.Obj); err != nil {
+			return err
+		}
+
+		if err := comp.compileExpr(t, subscriptionExpr.Key); err != nil {
+			return err
+		}
+
+		if err := comp.compileExpr(t, expr.Right); err != nil {
+			return err
+		}
+
+		t.AppendOp(code.OpSetAtIndex)
 	default:
 		return fmt.Errorf("invalid left-hand side of assignment to %v", expr.Left.Kind)
 	}