#include <Poco/String.h>
#include <Interpreters/misc.h>
#include <Interpreters/MarkTableIdentifiersVisitor.h>
#include <Interpreters/IdentifierSemantic.h>
#include <Parsers/ASTFunction.h>
#include <Parsers/ASTSelectQuery.h>
#include <Parsers/ASTTablesInSelectQuery.h>

namespace DB
{

bool MarkTableIdentifiersMatcher::needChildVisit(ASTPtr & node, const ASTPtr & child)
{
    if (child->as<ASTSelectQuery>())
        return false;
    if (node->as<ASTTableExpression>())
        return false;
    return true;
}

void MarkTableIdentifiersMatcher::visit(ASTPtr & ast, Data & data)
{
    if (auto * node_func = ast->as<ASTFunction>())
        visit(*node_func, ast, data);
    else if (auto * node_table = ast->as<ASTTableExpression>())
        visit(*node_table, ast, data);
}

void MarkTableIdentifiersMatcher::visit(ASTTableExpression & table, ASTPtr &, Data &)
{
    if (table.database_and_table_name)
        setIdentifierSpecial(table.database_and_table_name);
}

void MarkTableIdentifiersMatcher::visit(const ASTFunction & func, ASTPtr &, Data & data)
{
    /// `IN t` can be specified, where t is a table, which is equivalent to `IN (SELECT * FROM t)`.
    if (functionIsInOrGlobalInOperator(func.name))
    {
        auto & ast = func.arguments->children.at(1);
        if (auto opt_name = tryGetIdentifierName(ast))
            if (!data.aliases.count(*opt_name))
                setIdentifierSpecial(ast);
    }

    // first argument of joinGet can be a table identifier
    if (func.name == "joinGet")
    {
        auto & ast = func.arguments->children.at(0);
        if (auto opt_name = tryGetIdentifierName(ast))
            setIdentifierSpecial(ast);
    }
}

}