Implement setting values at index in an array

1 files changed, 42 insertions, 0 deletions

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