Function implementation for parsing lock statements in Parser class.

n8lang · Dec 21, 2024 · 5d211b6 · 5d211b6
1 parent 9a6e77b
commit 5d211b6
Showing 1 changed file with 18 additions and 22 deletions.
diff --git a/src/n8/parser/Parser.cpp b/src/n8/parser/Parser.cpp
@@ -26,6 +26,7 @@
 #include <n8/ast/expression/FunctionCallExpression.hpp>
 #include <n8/ast/expression/GroupedExpression.hpp>
 #include <n8/ast/expression/IfElseExpression.hpp>
+#include <n8/ast/expression/LockExpression.hpp>
 #include <n8/ast/expression/LoopExpression.hpp>
 #include <n8/ast/expression/MaybeExpression.hpp>
 #include <n8/ast/expression/NilCoalescingExpression.hpp>
@@ -92,9 +93,6 @@ Token Parser::previous() {
 
 Token Parser::peek() {
     if(this->isAtEnd())
-        #ifdef _MSC_VER
-        #   pragma warning(disable : 5272)
-        #endif
         throw ParserException(
             std::make_shared<Token>(this->previous()),
             "Encountered end-of-file."
@@ -121,9 +119,6 @@ Token Parser::current() {
 
 Token Parser::consume(const std::string& image) {
     if(this->isAtEnd())
-        #ifdef _MSC_VER
-        #   pragma warning(disable : 5272)
-        #endif
         throw ParserException(
             std::make_shared<Token>(this->previous()),
             "Expecting \"" + image +
@@ -132,9 +127,6 @@ Token Parser::consume(const std::string& image) {
 
     Token token = this->peek();
     if(token.getImage() != image)
-        #ifdef _MSC_VER
-        #   pragma warning(disable : 5272)
-        #endif
         throw ParserException(
             std::make_shared<Token>(this->previous()),
             "Expecting \"" + image +
@@ -147,19 +139,13 @@ Token Parser::consume(const std::string& image) {
 
 Token Parser::consume(TokenType type) {
     if(this->isAtEnd())
-        #ifdef _MSC_VER
-        #   pragma warning(disable : 5272)
-        #endif
         throw ParserException(
             std::make_shared<Token>(this->previous()),
             "Expecting token type, encountered end-of-code."
         );
 
     Token token = this->peek();
     if(token.getType() != type)
-        #ifdef _MSC_VER
-        #   pragma warning(disable : 5272)
-        #endif
         throw ParserException(
             std::make_shared<Token>(this->current()),
             "Expecting " + tokenTypeToString(type) +
@@ -373,10 +359,6 @@ std::shared_ptr<ASTNode> Parser::exprLiteral() {
 
     if(!expr) {
         auto address = this->current();
-
-        #ifdef _MSC_VER
-        #   pragma warning(disable : 5272)
-        #endif
         throw ParserException(
             std::make_shared<Token>(address),
             "Expecting expression, encountered " +
@@ -447,6 +429,21 @@ std::shared_ptr<ASTNode> Parser::exprSize() {
     );
 }
 
+std::shared_ptr<ASTNode> Parser::exprLock() {
+    Token address = this->consume("lock");
+    this->consume("(");
+
+    Token variable = this->consume(TokenType::IDENTIFIER);
+    this->consume(")");
+
+    std::shared_ptr<ASTNode> expression = this->expression();
+    return std::make_shared<LockExpression>(
+        std::make_shared<Token>(address),
+        std::make_shared<Token>(variable),
+        std::move(expression)
+    );
+}
+
 std::shared_ptr<ASTNode> Parser::exprType() {
     Token address = this->consume("type");
     std::shared_ptr<ASTNode> expression = this->expression();
@@ -510,9 +507,6 @@ std::shared_ptr<ASTNode> Parser::exprWhen() {
         }
         else if(this->isNext("else", TokenType::KEYWORD)) {
             if(defaultCase)
-                #ifdef _MSC_VER
-                #   pragma warning(disable : 5272)
-                #endif
                 throw ParserException(
                     std::make_shared<Token>(address),
                     "Cannot have more than one (1) else for when expression."
@@ -610,6 +604,8 @@ std::shared_ptr<ASTNode> Parser::exprPrimary() {
         expression = this->exprSize();
     else if(this->isNext("parallel", TokenType::KEYWORD))
         expression = this->exprParallel();
+    else if(this->isNext("lock", TokenType::KEYWORD))
+        expression = this->exprLock();
     else if(this->isNext("val", TokenType::KEYWORD))
         expression = this->exprVal();
     else if(this->isNext("[", TokenType::OPERATOR))