ast.cpp

#include "ast.h"
#include <sstream>

namespace AR437 {

// NumberLiteral
void NumberLiteral::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string NumberLiteral::toString() const {
    return std::to_string(value);
}

// StringLiteral
void StringLiteral::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string StringLiteral::toString() const {
    return "\"" + value + "\"";
}

// BooleanLiteral
void BooleanLiteral::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string BooleanLiteral::toString() const {
    return value ? "true" : "false";
}

// NullLiteral
void NullLiteral::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string NullLiteral::toString() const {
    return "null";
}

// Identifier
void Identifier::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string Identifier::toString() const {
    return name;
}

// BinaryOperation
void BinaryOperation::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string BinaryOperation::toString() const {
    std::ostringstream oss;
    oss << "(" << left->toString() << " " << TokenUtil::tokenTypeToString(operator_) 
        << " " << right->toString() << ")";
    return oss.str();
}

// UnaryOperation
void UnaryOperation::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string UnaryOperation::toString() const {
    return TokenUtil::tokenTypeToString(operator_) + operand->toString();
}

// Assignment
void Assignment::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string Assignment::toString() const {
    return variable + " = " + value->toString();
}

// FunctionCall
void FunctionCall::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string FunctionCall::toString() const {
    std::ostringstream oss;
    oss << name << "(";
    for (size_t i = 0; i < arguments.size(); ++i) {
        if (i > 0) oss << ", ";
        oss << arguments[i]->toString();
    }
    oss << ")";
    return oss.str();
}

// ExpressionStatement
void ExpressionStatement::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string ExpressionStatement::toString() const {
    return expression->toString() + ";";
}

// VariableDeclaration
void VariableDeclaration::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string VariableDeclaration::toString() const {
    std::string result = "let " + name;
    if (initializer) {
        result += " = " + initializer->toString();
    }
    return result + ";";
}

// Block
void Block::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string Block::toString() const {
    std::ostringstream oss;
    oss << "{\n";
    for (const auto& stmt : statements) {
        oss << "  " << stmt->toString() << "\n";
    }
    oss << "}";
    return oss.str();
}

// IfStatement
void IfStatement::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string IfStatement::toString() const {
    std::string result = "if (" + condition->toString() + ") " + thenBranch->toString();
    if (elseBranch) {
        result += " else " + elseBranch->toString();
    }
    return result;
}

// WhileStatement
void WhileStatement::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string WhileStatement::toString() const {
    return "while (" + condition->toString() + ") " + body->toString();
}

// ReturnStatement
void ReturnStatement::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string ReturnStatement::toString() const {
    std::string result = "return";
    if (value) {
        result += " " + value->toString();
    }
    return result + ";";
}

// DisplayStatement
void DisplayStatement::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string DisplayStatement::toString() const {
    return "display " + expression->toString() + ";";
}

// ReadStatement
void ReadStatement::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string ReadStatement::toString() const {
    return "read " + variableName + ";";
}

// FunctionDeclaration
void FunctionDeclaration::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string FunctionDeclaration::toString() const {
    std::ostringstream oss;
    oss << "func " << name << "(";
    for (size_t i = 0; i < parameters.size(); ++i) {
        if (i > 0) oss << ", ";
        oss << parameters[i];
    }
    oss << ") " << body->toString();
    return oss.str();
}

// Program
void Program::accept(ASTVisitor& visitor) {
    visitor.visit(*this);
}

std::string Program::toString() const {
    std::ostringstream oss;
    for (const auto& stmt : statements) {
        oss << stmt->toString() << "\n";
    }
    return oss.str();
}

} // namespace AR437