Fixed incorrect division results

src/core/CompileAssignDivision.go

@@ -2,30 +2,60 @@ package core
 
 import (
 	"git.akyoto.dev/cli/q/src/asm"
-	"git.akyoto.dev/cli/q/src/ast"
 	"git.akyoto.dev/cli/q/src/errors"
+	"git.akyoto.dev/cli/q/src/expression"
+	"git.akyoto.dev/cli/q/src/scope"
+	"git.akyoto.dev/cli/q/src/token"
+	"git.akyoto.dev/cli/q/src/types"
 	"git.akyoto.dev/cli/q/src/x86"
 )
 
 // CompileAssignDivision compiles an assign statement that has quotient and remainder on the left side and division on the right.
-func (f *Function) CompileAssignDivision(node *ast.Assign) error {
-	left := node.Expression.Children[0]
-	right := node.Expression.Children[1]
+func (f *Function) CompileAssignDivision(expr *expression.Expression) error {
+	var (
+		left              = expr.Children[0]
+		right             = expr.Children[1]
+		quotientVariable  *scope.Variable
+		remainderVariable *scope.Variable
+		err               error
+	)
 
-	quotient := left.Children[0]
-	name := quotient.Token.Text(f.File.Bytes)
-	quotientVariable := f.VariableByName(name)
+	if expr.Token.Kind == token.Define {
+		quotientVariable, err = f.Define(left.Children[0])
 
-	if quotientVariable == nil {
-		return errors.New(&errors.UnknownIdentifier{Name: name}, f.File, quotient.Token.Position)
-	}
+		if err != nil {
+			return err
+		}
 
-	remainder := left.Children[1]
-	name = remainder.Token.Text(f.File.Bytes)
-	remainderVariable := f.VariableByName(name)
+		remainderVariable, err = f.Define(left.Children[1])
 
-	if remainderVariable == nil {
-		return errors.New(&errors.UnknownIdentifier{Name: name}, f.File, remainder.Token.Position)
+		if err != nil {
+			return err
+		}
+
+		quotientVariable.Type = types.Int
+		remainderVariable.Type = types.Int
+		f.AddVariable(quotientVariable)
+		f.AddVariable(remainderVariable)
+	} else {
+		quotient := left.Children[0]
+		name := quotient.Token.Text(f.File.Bytes)
+		quotientVariable = f.VariableByName(name)
+
+		if quotientVariable == nil {
+			return errors.New(&errors.UnknownIdentifier{Name: name}, f.File, quotient.Token.Position)
+		}
+
+		remainder := left.Children[1]
+		name = remainder.Token.Text(f.File.Bytes)
+		remainderVariable = f.VariableByName(name)
+
+		if remainderVariable == nil {
+			return errors.New(&errors.UnknownIdentifier{Name: name}, f.File, remainder.Token.Position)
+		}
+
+		defer f.UseVariable(quotientVariable)
+		defer f.UseVariable(remainderVariable)
 	}
 
 	dividend := right.Children[0]
@@ -38,9 +68,7 @@ func (f *Function) CompileAssignDivision(node *ast.Assign) error {
 	divisor := right.Children[1]
 	err = f.Execute(right.Token, dividendRegister, divisor)
 	f.RegisterRegister(asm.MOVE, quotientVariable.Register, x86.RAX)
-	f.UseVariable(quotientVariable)
 	f.RegisterRegister(asm.MOVE, remainderVariable.Register, x86.RDX)
-	f.UseVariable(remainderVariable)
 
 	if isTemporary {
 		f.FreeRegister(dividendRegister)