Improved implementation of assign operators

src/core/CompileAssign.go

@@ -1,8 +1,12 @@
 package core
 
 import (
+	"fmt"
+
+	"git.urbach.dev/cli/q/src/asm"
 	"git.urbach.dev/cli/q/src/ast"
 	"git.urbach.dev/cli/q/src/errors"
+	"git.urbach.dev/cli/q/src/eval"
 	"git.urbach.dev/cli/q/src/token"
 )
 
@@ -11,33 +15,54 @@ func (f *Function) CompileAssign(node *ast.Assign) error {
 	left := node.Expression.Children[0]
 	right := node.Expression.Children[1]
 
-	if left.IsLeaf() {
-		name := left.Token.Text(f.File.Bytes)
-		variable := f.VariableByName(name)
-
-		if variable == nil {
-			return errors.New(&errors.UnknownIdentifier{Name: name}, f.File, left.Token.Position)
+	if left.Token.Kind == token.Separator {
+		if right.Token.Kind == token.Div {
+			return f.CompileAssignDivision(node.Expression)
 		}
 
-		defer f.UseVariable(variable)
-		return f.Execute(node.Expression.Token, variable.Value.Register, right)
-	}
+		if right.Token.Kind == token.Call {
+			return f.MultiAssign(left, right)
+		}
 
-	if left.Token.Kind == token.Dot {
-		return f.CompileAssignField(node)
-	}
-
-	if left.Token.Kind == token.Array {
-		return f.CompileAssignArray(node)
-	}
-
-	if left.Token.Kind == token.Separator && right.Token.Kind == token.Div {
-		return f.CompileAssignDivision(node.Expression)
-	}
-
-	if !ast.IsFunctionCall(right) {
 		return errors.New(errors.NotImplemented, f.File, node.Expression.Token.Position)
 	}
 
-	return f.MultiAssign(left, right)
+	leftValue, err := f.Evaluate(left)
+
+	if err != nil {
+		return err
+	}
+
+	operation := node.Expression.Token
+
+	switch leftValue := leftValue.(type) {
+	case eval.Register:
+		f.Execute(operation, leftValue.Register, right)
+	case eval.Memory:
+		if operation.Kind == token.Assign {
+			rightValue, err := f.Evaluate(right)
+
+			if err != nil {
+				return err
+			}
+
+			f.ValueToMemory(rightValue, leftValue.Memory)
+			return nil
+		}
+
+		tmp := f.NewRegister()
+		f.ValueToRegister(leftValue, tmp)
+		err := f.Execute(operation, tmp, right)
+
+		if err != nil {
+			return err
+		}
+
+		f.MemoryRegister(asm.STORE, leftValue.Memory, tmp)
+		f.FreeRegister(tmp)
+	default:
+		panic(fmt.Errorf("%s: not implemented: %v", f.UniqueName, leftValue))
+	}
+
+	return nil
 }