#pragma once

#include "Entity/Scope.h"
#include "Sema/ASTVisitor.h"

/// 引用消解器
class ReferenceResolver final : public ASTVisitorWithScope {
public:
    void visit(const SharedPtr<ModuleNode> &module) override;

    void visit(const SharedPtr<VarDeclNode> &varDecl) override;

    void visit(const SharedPtr<ParmVarDeclNode> &paramVarDecl) override;

    void visit(const SharedPtr<FunctionDeclNode> &funcDecl) override;

    void visit(const SharedPtr<IdentifierExprNode> &varExpr) override;

    void visit(const SharedPtr<CallExprNode> &callExpr) override;

    void visit(const SharedPtr<CompoundStmtNode> &compStmt) override;

    void visit(const SharedPtr<ForStmtNode> &forStmt) override;

    void visit(const SharedPtr<ContinueStmtNode> &continueStmt) override;

    void visit(const SharedPtr<BreakStmtNode> &breakStmt) override;

    void visit(const SharedPtr<ReturnStmtNode> &returnStmt) override;

private:
    /**
     * @brief 查找变量
     * @details 沿着当前作用域递归向上查找变量
     *
     * @param name 变量名
     * @return 变量定义节点
     */
    inline SharedPtr<VarDeclNode> resolveVariable(const String &name) {
        return getCurrentScope()->resolveVariable(name);
    }

    /**
     * @brief 查找函数
     * @details 在顶级作用域中查找函数
     *
     * @param name 函数名
     * @return 函数定义节点
     */
    inline SharedPtr<FunctionDeclNode> resolveFunction(const String &name) {
        return getCurrentScope()->getTopLevel()->resolveFunction(name);
    }

    static SharedPtr<StmtNode> resolveRefIterationStmt(const SharedPtr<Node> &node);
};