Add . and -> operator overloads

sand-lang · Aug 3, 2020 · f230e6b · f230e6b
1 parent 71c3b0a
commit f230e6b
Show file tree

Hide file tree

Showing 6 changed files with 994 additions and 948 deletions.
diff --git a/src/grammar/SandLexer.g4 b/src/grammar/SandLexer.g4
@@ -129,9 +129,9 @@ fragment CharChar: ~ ['\r\n] | Escape;
 DecimalLiteral: NONZERODIGIT (DIGITSEPARATOR? DIGIT)*;
 
 FloatingLiteral:
-	(DecimalLiteral | ZeroLiteral)? '.' (
-		DIGIT (DIGITSEPARATOR? DIGIT)*
-	)?;
+	(DecimalLiteral | ZeroLiteral)? '.' DIGIT (
+		DIGITSEPARATOR? DIGIT
+	)*;
 
 ZeroLiteral: '0';
 

diff --git a/src/grammar/SandParser.g4 b/src/grammar/SandParser.g4
@@ -172,6 +172,8 @@ overloadableOperator:
 	| '>'
 	| greaterThanOrEqualToOperator
 	| '[' ']'
+	| '->'
+	| '.'
 	| shiftOperator
 	| shiftEqualOperator;
 

diff --git a/src/grammar/Visitor.hpp b/src/grammar/Visitor.hpp
@@ -2884,6 +2884,23 @@ class Visitor
         auto scope = this->scopes.top();
         auto expr = this->valueFromExpression(context->expression());
 
+        if (context->Arrow())
+        {
+            std::vector<Value *> args = {expr};
+            if (auto overload = this->getOperatorOverload("->", args))
+            {
+                expr = overload->call(scope->builder(), scope->module(), args);
+            }
+        }
+        else
+        {
+            std::vector<Value *> args = {expr};
+            if (auto overload = this->getOperatorOverload(".", args))
+            {
+                expr = overload->call(scope->builder(), scope->module(), args);
+            }
+        }
+
         if (context->Arrow())
         {
             if (!Type::behind_reference(expr->type)->is_pointer())