From d84c0f36afcde4e5af2907d70931de75957d5277 Mon Sep 17 00:00:00 2001
From: Mel <einebeere@gmail.com>
Date: Tue, 16 Aug 2022 20:33:41 +0000
Subject: Implement setting values at index in an array

---
 pkg/lang/vm/exec.go | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

(limited to 'pkg/lang/vm/exec.go')

diff --git a/pkg/lang/vm/exec.go b/pkg/lang/vm/exec.go
index 34cf444..a64751a 100644
--- a/pkg/lang/vm/exec.go
+++ b/pkg/lang/vm/exec.go
@@ -832,6 +832,48 @@ func (vm *VM) execIndex() error {
 	return nil
 }
 
+func (vm *VM) execSetAtIndex() error {
+	e, err := vm.popAndDrop()
+	if err != nil {
+		return err
+	}
+
+	i, err := vm.popAndDrop()
+	if err != nil {
+		return err
+	}
+
+	v, err := vm.popAndDrop()
+	if err != nil {
+		return err
+	}
+
+	if i.Type() != value.IntType || v.Type() != value.ArrayType {
+		return ErrInvalidOperandTypes{
+			Op: code.OpSetAtIndex,
+			X:  v.Type(),
+			Y:  i.Type(),
+		}
+	}
+
+	arr := v.Data().(value.ArrayData)
+	idx := i.Data().(value.IntData).Get()
+
+	len, err := arr.Len(vm.memory)
+	if err != nil {
+		return err
+	}
+
+	if idx < 0 || idx >= int64(len) {
+		return ErrArrayIndexOutOfBounds{
+			Index: int(idx),
+			Len:   int(len),
+		}
+	}
+
+	return arr.Set(vm.memory, int(idx), e)
+}
+
 func (vm *VM) execCall(argCount uint) error {
 	var err error
 
-- 
cgit 1.4.1