From 23dedf2eff8d8c814834765ef16dd743525446d9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rafa=C5=82=20Grodzi=C5=84ski?= <rafal.grodzinski@gmail.com>
Date: Sun, 1 Jun 2025 11:43:49 +0900
Subject: [PATCH] Nullable token in statement

---
 src/Parser.cpp    | 8 ++++----
 src/Parser.h      | 1 -
 src/Statement.cpp | 4 ++--
 src/Statement.h   | 6 +++---
 4 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/src/Parser.cpp b/src/Parser.cpp
index 27b1f09..8ecf66c 100644
--- a/src/Parser.cpp
+++ b/src/Parser.cpp
@@ -10,7 +10,7 @@ vector<shared_ptr<Statement>> Parser::getStatements() {
         shared_ptr<Statement> statement = nextStatement();
         // Abort parsing if we got an error
         if (!statement->isValid()) {
-            cerr << "Unexpected token '" << statement->getToken().getLexme() << "' at " << statement->getToken().getLine() << ":" << statement->getToken().getColumn() << endl;
+            cerr << "Unexpected token '" << statement->getToken()->getLexme() << "' at " << statement->getToken()->getLine() << ":" << statement->getToken()->getColumn() << endl;
             return vector<shared_ptr<Statement>>();
         }
         statements.push_back(statement);
@@ -33,7 +33,7 @@ shared_ptr<Statement> Parser::nextStatement() {
 }
 
 shared_ptr<Statement> Parser::matchInvalidStatement() {
-    return make_shared<Statement>(Statement::Kind::INVALID, tokens.at(currentIndex), nullptr);
+    return make_shared<Statement>(Statement::Kind::INVALID, make_shared<Token>(tokens.at(currentIndex)), nullptr);
 }
 
 //
@@ -42,9 +42,9 @@ shared_ptr<Statement> Parser::matchInvalidStatement() {
 shared_ptr<Statement> Parser::matchExpressionStatement() {
     shared_ptr<Expression> expression = term();
     if (expression->isValid()) {
-        return make_shared<Statement>(Statement::Kind::EXPRESSION, tokens.at(currentIndex), expression);
+        return make_shared<Statement>(Statement::Kind::EXPRESSION, nullptr, expression);
     } else {
-        return make_shared<Statement>(Statement::Kind::INVALID, tokens.at(currentIndex), expression);
+        return make_shared<Statement>(Statement::Kind::INVALID, make_shared<Token>(tokens.at(currentIndex)), expression);
     }
 }
 
diff --git a/src/Parser.h b/src/Parser.h
index f82c118..afa50fa 100644
--- a/src/Parser.h
+++ b/src/Parser.h
@@ -28,7 +28,6 @@ private:
 
 public:
     Parser(vector<Token> tokens);
-    //shared_ptr<Expression> getExpression();
     vector<shared_ptr<Statement>> getStatements();
 };
 
diff --git a/src/Statement.cpp b/src/Statement.cpp
index ff9c525..f612eea 100644
--- a/src/Statement.cpp
+++ b/src/Statement.cpp
@@ -1,13 +1,13 @@
 #include "Statement.h"
 
-Statement::Statement(Kind kind, Token token, shared_ptr<Expression> expression): kind(kind), token(token), expression(expression) {
+Statement::Statement(Kind kind, shared_ptr<Token> token, shared_ptr<Expression> expression): kind(kind), token(token), expression(expression) {
 }
 
 shared_ptr<Expression> Statement::getExpression() {
     return expression;
 }
 
-Token Statement::getToken() {
+shared_ptr<Token> Statement::getToken() {
     return token;
 }
 
diff --git a/src/Statement.h b/src/Statement.h
index 1188df8..aeef197 100644
--- a/src/Statement.h
+++ b/src/Statement.h
@@ -17,12 +17,12 @@ public:
 
 private:
     Kind kind;
-    Token token;
+    shared_ptr<Token> token;
     shared_ptr<Expression> expression;
 
 public:
-    Statement(Kind kind, Token token, shared_ptr<Expression> expression);
-    Token getToken();
+    Statement(Kind kind, shared_ptr<Token> token, shared_ptr<Expression> expression);
+    shared_ptr<Token> getToken();
     shared_ptr<Expression> getExpression();
     bool isValid();
     string toString();