Added more tests

src/core/ExpressionToRegister.go

@@ -1,6 +1,8 @@
 package core
 
 import (
+	"math"
+
 	"git.akyoto.dev/cli/q/src/asm"
 	"git.akyoto.dev/cli/q/src/ast"
 	"git.akyoto.dev/cli/q/src/cpu"
@@ -40,10 +42,46 @@ func (f *Function) ExpressionToRegister(node *expression.Expression, register cp
 	}
 
 	if node.Token.Kind == token.Array {
-		array := f.VariableByName(node.Children[0].Token.Text(f.File.Bytes))
-		offset, err := f.Number(node.Children[1].Token)
-		f.MemoryRegister(asm.LOAD, asm.Memory{Base: array.Register, Offset: int8(offset), Length: 1}, register)
-		return types.Int, err
+		name := node.Children[0].Token.Text(f.File.Bytes)
+		array := f.VariableByName(name)
+
+		if array == nil {
+			return nil, errors.New(&errors.UnknownIdentifier{Name: name}, f.File, node.Children[0].Token.Position)
+		}
+
+		index := node.Children[1]
+
+		memory := asm.Memory{
+			Base:           array.Register,
+			Offset:         0,
+			OffsetRegister: cpu.Register(math.MaxUint8),
+			Length:         byte(1),
+		}
+
+		if index.Token.IsNumeric() {
+			offset, err := f.Number(index.Token)
+
+			if err != nil {
+				return nil, err
+			}
+
+			memory.Offset = int8(offset)
+		} else {
+			typ, err := f.ExpressionToRegister(index, register)
+
+			if err != nil {
+				return nil, err
+			}
+
+			if !types.Is(typ, types.Int) {
+				return nil, errors.New(&errors.TypeMismatch{Encountered: typ.Name(), Expected: types.Int.Name()}, f.File, index.Token.Position)
+			}
+
+			memory.OffsetRegister = register
+		}
+
+		f.MemoryRegister(asm.LOAD, memory, register)
+		return types.Int, nil
 	}
 
 	if node.Token.Kind == token.Period {