// Copyright (C) 2020-2024 Free Software Foundation, Inc.
// This file is part of GCC.
// GCC is free software; you can redistribute it and/or modify it under
// the terms of the GNU General Public License as published by the Free
// Software Foundation; either version 3, or (at your option) any later
// version.
// GCC is distributed in the hope that it will be useful, but WITHOUT ANY
// WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
// for more details.
// You should have received a copy of the GNU General Public License
// along with GCC; see the file COPYING3. If not see
// .
#include "rust-early-name-resolver.h"
#include "rust-ast-full.h"
#include "rust-name-resolver.h"
#include "rust-macro-builtins.h"
namespace Rust {
namespace Resolver {
// Check if a module contains the `#[macro_use]` attribute
static bool
is_macro_use_module (const AST::Module &mod)
{
for (const auto &attr : mod.get_outer_attrs ())
if (attr.get_path ().as_string () == "macro_use")
return true;
return false;
}
std::vector>
EarlyNameResolver::accumulate_escaped_macros (AST::Module &module)
{
if (!is_macro_use_module (module))
return {};
// Parse the module's items if they haven't been expanded and the file
// should be parsed (i.e isn't hidden behind an untrue or impossible cfg
// directive)
if (module.get_kind () == AST::Module::UNLOADED)
module.load_items ();
std::vector> escaped_macros;
scoped (module.get_node_id (), [&module, &escaped_macros, this] {
for (auto &item : module.get_items ())
{
if (item->get_ast_kind () == AST::Kind::MODULE)
{
auto &module = *static_cast (item.get ());
auto new_macros = accumulate_escaped_macros (module);
std::move (new_macros.begin (), new_macros.end (),
std::back_inserter (escaped_macros));
continue;
}
if (item->get_ast_kind () == AST::Kind::MACRO_RULES_DEFINITION)
escaped_macros.emplace_back (item->clone_item ());
}
});
return escaped_macros;
}
EarlyNameResolver::EarlyNameResolver ()
: current_scope (UNKNOWN_NODEID), resolver (*Resolver::get ()),
mappings (*Analysis::Mappings::get ())
{}
void
EarlyNameResolver::go (AST::Crate &crate)
{
std::vector> new_items;
auto items = crate.take_items ();
scoped (crate.get_node_id (), [&items, &new_items, this] {
for (auto &&item : items)
{
auto new_macros = std::vector> ();
if (item->get_ast_kind () == AST::Kind::MODULE)
new_macros = accumulate_escaped_macros (
*static_cast (item.get ()));
new_items.emplace_back (std::move (item));
std::move (new_macros.begin (), new_macros.end (),
std::back_inserter (new_items));
}
});
crate.set_items (std::move (new_items));
scoped (crate.get_node_id (), [&crate, this] () {
for (auto &item : crate.items)
item->accept_vis (*this);
});
}
void
EarlyNameResolver::resolve_generic_args (AST::GenericArgs &generic_args)
{
for (auto &arg : generic_args.get_generic_args ())
arg.accept_vis (*this);
for (auto &arg : generic_args.get_binding_args ())
arg.get_type ()->accept_vis (*this);
}
void
EarlyNameResolver::resolve_qualified_path_type (AST::QualifiedPathType &path)
{
path.get_type ()->accept_vis (*this);
if (path.has_as_clause ())
path.get_as_type_path ().accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::Token &)
{}
void
EarlyNameResolver::visit (AST::DelimTokenTree &)
{}
void
EarlyNameResolver::visit (AST::AttrInputMetaItemContainer &)
{}
void
EarlyNameResolver::visit (AST::IdentifierExpr &)
{}
void
EarlyNameResolver::visit (AST::Lifetime &)
{}
void
EarlyNameResolver::visit (AST::LifetimeParam &)
{}
void
EarlyNameResolver::visit (AST::ConstGenericParam &)
{}
// FIXME: ARTHUR: Do we need to perform macro resolution for paths as well?
// std::arch::asm!()?
void
EarlyNameResolver::visit (AST::PathInExpression &path)
{
for (auto &segment : path.get_segments ())
if (segment.has_generic_args ())
resolve_generic_args (segment.get_generic_args ());
}
void
EarlyNameResolver::visit (AST::TypePathSegment &)
{}
void
EarlyNameResolver::visit (AST::TypePathSegmentGeneric &segment)
{
if (segment.has_generic_args ())
resolve_generic_args (segment.get_generic_args ());
}
void
EarlyNameResolver::visit (AST::TypePathSegmentFunction &segment)
{
for (auto &type : segment.get_type_path_function ().get_params ())
type->accept_vis (*this);
segment.get_type_path_function ().get_return_type ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TypePath &path)
{
for (auto &seg : path.get_segments ())
seg->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::QualifiedPathInExpression &path)
{
resolve_qualified_path_type (path.get_qualified_path_type ());
for (auto &segment : path.get_segments ())
if (segment.has_generic_args ())
resolve_generic_args (segment.get_generic_args ());
}
void
EarlyNameResolver::visit (AST::QualifiedPathInType &path)
{
resolve_qualified_path_type (path.get_qualified_path_type ());
for (auto &segment : path.get_segments ())
segment->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::LiteralExpr &)
{}
void
EarlyNameResolver::visit (AST::AttrInputLiteral &)
{}
void
EarlyNameResolver::visit (AST::AttrInputMacro &)
{}
void
EarlyNameResolver::visit (AST::MetaItemLitExpr &)
{}
void
EarlyNameResolver::visit (AST::MetaItemPathLit &)
{}
void
EarlyNameResolver::visit (AST::BorrowExpr &expr)
{
expr.get_borrowed_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::DereferenceExpr &expr)
{
expr.get_dereferenced_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ErrorPropagationExpr &expr)
{
expr.get_propagating_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::NegationExpr &expr)
{
expr.get_negated_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ArithmeticOrLogicalExpr &expr)
{
expr.get_left_expr ()->accept_vis (*this);
expr.get_right_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ComparisonExpr &expr)
{
expr.get_left_expr ()->accept_vis (*this);
expr.get_right_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::LazyBooleanExpr &expr)
{
expr.get_left_expr ()->accept_vis (*this);
expr.get_right_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TypeCastExpr &expr)
{
expr.get_casted_expr ()->accept_vis (*this);
expr.get_type_to_cast_to ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::AssignmentExpr &expr)
{
expr.get_left_expr ()->accept_vis (*this);
expr.get_right_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::CompoundAssignmentExpr &expr)
{
expr.get_left_expr ()->accept_vis (*this);
expr.get_right_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::GroupedExpr &expr)
{
expr.get_expr_in_parens ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ArrayElemsValues &elems)
{
for (auto &expr : elems.get_values ())
expr->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ArrayElemsCopied &elems)
{
elems.get_elem_to_copy ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ArrayExpr &expr)
{
expr.get_array_elems ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ArrayIndexExpr &expr)
{
expr.get_array_expr ()->accept_vis (*this);
expr.get_index_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TupleExpr &expr)
{
for (auto &elem : expr.get_tuple_elems ())
elem->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TupleIndexExpr &expr)
{
expr.get_tuple_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::StructExprStruct &)
{}
void
EarlyNameResolver::visit (AST::StructExprFieldIdentifier &)
{}
void
EarlyNameResolver::visit (AST::StructExprFieldIdentifierValue &field)
{
field.get_value ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::StructExprFieldIndexValue &field)
{
field.get_value ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::StructExprStructFields &expr)
{
for (auto &field : expr.get_fields ())
field->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::StructExprStructBase &)
{}
void
EarlyNameResolver::visit (AST::CallExpr &expr)
{
expr.get_function_expr ()->accept_vis (*this);
for (auto ¶m : expr.get_params ())
param->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::MethodCallExpr &expr)
{
expr.get_receiver_expr ()->accept_vis (*this);
for (auto ¶m : expr.get_params ())
param->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::FieldAccessExpr &expr)
{
expr.get_receiver_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ClosureExprInner &expr)
{
expr.get_definition_expr ()->accept_vis (*this);
for (auto ¶m : expr.get_params ())
if (param.has_type_given ())
param.get_type ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::BlockExpr &expr)
{
scoped (expr.get_node_id (), [&expr, this] () {
for (auto &stmt : expr.get_statements ())
stmt->accept_vis (*this);
if (expr.has_tail_expr ())
expr.get_tail_expr ()->accept_vis (*this);
});
}
void
EarlyNameResolver::visit (AST::ClosureExprInnerTyped &expr)
{
expr.get_definition_block ()->accept_vis (*this);
for (auto ¶m : expr.get_params ())
param.get_type ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ContinueExpr &)
{}
void
EarlyNameResolver::visit (AST::BreakExpr &expr)
{
if (expr.has_break_expr ())
expr.get_break_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::RangeFromToExpr &expr)
{
expr.get_from_expr ()->accept_vis (*this);
expr.get_to_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::RangeFromExpr &expr)
{
expr.get_from_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::RangeToExpr &expr)
{
expr.get_to_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::RangeFullExpr &)
{}
void
EarlyNameResolver::visit (AST::RangeFromToInclExpr &expr)
{
expr.get_from_expr ()->accept_vis (*this);
expr.get_to_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::RangeToInclExpr &expr)
{
expr.get_to_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ReturnExpr &expr)
{
if (expr.has_returned_expr ())
expr.get_returned_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::UnsafeBlockExpr &expr)
{
expr.get_block_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::LoopExpr &expr)
{
expr.get_loop_block ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::WhileLoopExpr &expr)
{
expr.get_predicate_expr ()->accept_vis (*this);
expr.get_loop_block ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::WhileLetLoopExpr &expr)
{
expr.get_scrutinee_expr ()->accept_vis (*this);
expr.get_loop_block ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ForLoopExpr &expr)
{
scoped (expr.get_node_id (), [&expr, this] () {
expr.get_pattern ()->accept_vis (*this);
expr.get_iterator_expr ()->accept_vis (*this);
expr.get_loop_block ()->accept_vis (*this);
});
}
void
EarlyNameResolver::visit (AST::IfExpr &expr)
{
expr.get_condition_expr ()->accept_vis (*this);
expr.get_if_block ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::IfExprConseqElse &expr)
{
expr.get_condition_expr ()->accept_vis (*this);
expr.get_if_block ()->accept_vis (*this);
expr.get_else_block ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::IfLetExpr &expr)
{
expr.get_value_expr ()->accept_vis (*this);
scoped (expr.get_node_id (),
[&expr, this] () { expr.get_if_block ()->accept_vis (*this); });
}
void
EarlyNameResolver::visit (AST::IfLetExprConseqElse &expr)
{
expr.get_value_expr ()->accept_vis (*this);
expr.get_if_block ()->accept_vis (*this);
expr.get_else_block ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::MatchExpr &expr)
{
expr.get_scrutinee_expr ()->accept_vis (*this);
scoped (expr.get_node_id (), [&expr, this] () {
for (auto &arm : expr.get_match_cases ())
{
scoped (arm.get_node_id (), [&arm, this] () {
if (arm.get_arm ().has_match_arm_guard ())
arm.get_arm ().get_guard_expr ()->accept_vis (*this);
for (auto &pattern : arm.get_arm ().get_patterns ())
pattern->accept_vis (*this);
arm.get_expr ()->accept_vis (*this);
});
}
});
}
void
EarlyNameResolver::visit (AST::AwaitExpr &expr)
{
expr.get_awaited_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::AsyncBlockExpr &expr)
{
expr.get_block_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TypeParam ¶m)
{
for (auto &bound : param.get_type_param_bounds ())
bound->accept_vis (*this);
if (param.has_type ())
param.get_type ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::LifetimeWhereClauseItem &)
{}
void
EarlyNameResolver::visit (AST::TypeBoundWhereClauseItem &item)
{
for (auto &bound : item.get_type_param_bounds ())
bound->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::Method &method)
{
if (method.has_generics ())
for (auto &generic : method.get_generic_params ())
generic->accept_vis (*this);
if (method.get_self_param ().has_type ())
method.get_self_param ().get_type ()->accept_vis (*this);
for (auto ¶m : method.get_function_params ())
param.get_type ()->accept_vis (*this);
if (method.has_return_type ())
method.get_return_type ()->accept_vis (*this);
method.get_definition ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::Module &module)
{
if (module.get_kind () == AST::Module::UNLOADED)
module.load_items ();
// so we need to only go "one scope down" for fetching macros. Macros within
// functions are still scoped only within that function. But we have to be
// careful because nested modules with #[macro_use] actually works!
std::vector> new_items;
auto items = module.take_items ();
scoped (module.get_node_id (), [&items, &new_items, this] {
for (auto &&item : items)
{
auto new_macros = std::vector> ();
if (item->get_ast_kind () == AST::Kind::MODULE)
new_macros = accumulate_escaped_macros (
*static_cast (item.get ()));
new_items.emplace_back (std::move (item));
std::move (new_macros.begin (), new_macros.end (),
std::back_inserter (new_items));
}
});
module.set_items (std::move (new_items));
scoped (module.get_node_id (), [&module, this] () {
for (auto &item : module.get_items ())
item->accept_vis (*this);
});
}
void
EarlyNameResolver::visit (AST::ExternCrate &)
{}
void
EarlyNameResolver::visit (AST::UseTreeGlob &)
{}
void
EarlyNameResolver::visit (AST::UseTreeList &)
{}
void
EarlyNameResolver::visit (AST::UseTreeRebind &)
{}
void
EarlyNameResolver::visit (AST::UseDeclaration &)
{}
void
EarlyNameResolver::visit (AST::Function &function)
{
if (function.has_generics ())
for (auto &generic : function.get_generic_params ())
generic->accept_vis (*this);
for (auto ¶m : function.get_function_params ())
param.get_type ()->accept_vis (*this);
if (function.has_return_type ())
function.get_return_type ()->accept_vis (*this);
function.get_definition ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TypeAlias &type_alias)
{
type_alias.get_type_aliased ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::StructStruct &struct_item)
{
for (auto &field : struct_item.get_fields ())
field.get_field_type ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TupleStruct &tuple_struct)
{
for (auto &field : tuple_struct.get_fields ())
field.get_field_type ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::EnumItem &)
{}
void
EarlyNameResolver::visit (AST::EnumItemTuple &item)
{
for (auto &field : item.get_tuple_fields ())
field.get_field_type ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::EnumItemStruct &item)
{
for (auto &field : item.get_struct_fields ())
field.get_field_type ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::EnumItemDiscriminant &item)
{
item.get_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::Enum &enum_item)
{
for (auto &generic : enum_item.get_generic_params ())
generic->accept_vis (*this);
for (auto &variant : enum_item.get_variants ())
variant->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::Union &)
{}
void
EarlyNameResolver::visit (AST::ConstantItem &const_item)
{
const_item.get_type ()->accept_vis (*this);
const_item.get_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::StaticItem &static_item)
{
static_item.get_type ()->accept_vis (*this);
static_item.get_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TraitItemFunc &item)
{
auto &decl = item.get_trait_function_decl ();
if (decl.has_return_type ())
decl.get_return_type ()->accept_vis (*this);
for (auto &generic : decl.get_generic_params ())
generic->accept_vis (*this);
for (auto ¶m : decl.get_function_params ())
param.get_type ()->accept_vis (*this);
if (item.has_definition ())
item.get_definition ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TraitItemMethod &item)
{
// FIXME: Can we factor this with the above function?
auto &decl = item.get_trait_method_decl ();
if (decl.has_return_type ())
decl.get_return_type ()->accept_vis (*this);
for (auto &generic : decl.get_generic_params ())
generic->accept_vis (*this);
for (auto ¶m : decl.get_function_params ())
param.get_type ()->accept_vis (*this);
if (item.has_definition ())
item.get_definition ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TraitItemConst &item)
{
item.get_type ()->accept_vis (*this);
if (item.has_expr ())
item.get_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TraitItemType &)
{}
void
EarlyNameResolver::visit (AST::Trait &trait)
{
for (auto &generic : trait.get_generic_params ())
generic->accept_vis (*this);
scoped (trait.get_node_id (), [&trait, this] () {
for (auto &item : trait.get_trait_items ())
item->accept_vis (*this);
});
}
void
EarlyNameResolver::visit (AST::InherentImpl &impl)
{
impl.get_type ()->accept_vis (*this);
for (auto &generic : impl.get_generic_params ())
generic->accept_vis (*this);
scoped (impl.get_node_id (), [&impl, this] () {
for (auto &item : impl.get_impl_items ())
item->accept_vis (*this);
});
}
void
EarlyNameResolver::visit (AST::TraitImpl &impl)
{
impl.get_type ()->accept_vis (*this);
for (auto &generic : impl.get_generic_params ())
generic->accept_vis (*this);
scoped (impl.get_node_id (), [&impl, this] () {
for (auto &item : impl.get_impl_items ())
item->accept_vis (*this);
});
}
void
EarlyNameResolver::visit (AST::ExternalTypeItem &item)
{
// nothing to do?
}
void
EarlyNameResolver::visit (AST::ExternalStaticItem &item)
{
item.get_type ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ExternalFunctionItem &item)
{
for (auto &generic : item.get_generic_params ())
generic->accept_vis (*this);
for (auto ¶m : item.get_function_params ())
param.get_type ()->accept_vis (*this);
if (item.has_return_type ())
item.get_return_type ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ExternBlock &block)
{
scoped (block.get_node_id (), [&block, this] () {
for (auto &item : block.get_extern_items ())
item->accept_vis (*this);
});
}
void
EarlyNameResolver::visit (AST::MacroMatchFragment &)
{}
void
EarlyNameResolver::visit (AST::MacroMatchRepetition &)
{}
void
EarlyNameResolver::visit (AST::MacroMatcher &)
{}
void
EarlyNameResolver::visit (AST::MacroRulesDefinition &rules_def)
{
auto path = CanonicalPath::new_seg (rules_def.get_node_id (),
rules_def.get_rule_name ().as_string ());
resolver.get_macro_scope ().insert (path, rules_def.get_node_id (),
rules_def.get_locus ());
/* Since the EarlyNameResolver runs multiple time (fixed point algorithm)
* we could be inserting the same macro def over and over again until we
* implement some optimizations */
// FIXME: ARTHUR: Remove that lookup and add proper optimizations instead
AST::MacroRulesDefinition *tmp = nullptr;
if (mappings.lookup_macro_def (rules_def.get_node_id (), &tmp))
return;
mappings.insert_macro_def (&rules_def);
rust_debug_loc (rules_def.get_locus (), "inserting macro def: [%s]",
path.get ().c_str ());
}
void
EarlyNameResolver::visit (AST::MacroInvocation &invoc)
{
auto &invoc_data = invoc.get_invoc_data ();
auto has_semicolon = invoc.has_semicolon ();
if (invoc.get_kind () == AST::MacroInvocation::InvocKind::Builtin)
for (auto &pending_invoc : invoc.get_pending_eager_invocations ())
pending_invoc->accept_vis (*this);
// ??
// switch on type of macro:
// - '!' syntax macro (inner switch)
// - procedural macro - "A token-based function-like macro"
// - 'macro_rules' (by example/pattern-match) macro? or not? "an
// AST-based function-like macro"
// - else is unreachable
// - attribute syntax macro (inner switch)
// - procedural macro attribute syntax - "A token-based attribute
// macro"
// - legacy macro attribute syntax? - "an AST-based attribute macro"
// - non-macro attribute: mark known
// - else is unreachable
// - derive macro (inner switch)
// - derive or legacy derive - "token-based" vs "AST-based"
// - else is unreachable
// - derive container macro - unreachable
// lookup the rules for this macro
NodeId resolved_node = UNKNOWN_NODEID;
NodeId source_node = UNKNOWN_NODEID;
if (has_semicolon)
source_node = invoc.get_macro_node_id ();
else
source_node = invoc.get_pattern_node_id ();
auto seg
= CanonicalPath::new_seg (source_node, invoc_data.get_path ().as_string ());
bool found = resolver.get_macro_scope ().lookup (seg, &resolved_node);
if (!found)
{
rust_error_at (invoc.get_locus (), "unknown macro: [%s]",
seg.get ().c_str ());
return;
}
// lookup the rules
AST::MacroRulesDefinition *rules_def = nullptr;
bool ok = mappings.lookup_macro_def (resolved_node, &rules_def);
rust_assert (ok);
auto &outer_attrs = rules_def->get_outer_attrs ();
bool is_builtin
= std::any_of (outer_attrs.begin (), outer_attrs.end (),
[] (AST::Attribute attr) {
return attr.get_path () == "rustc_builtin_macro";
});
if (is_builtin)
{
auto builtin_kind
= builtin_macro_from_string (rules_def->get_rule_name ().as_string ());
invoc.map_to_builtin (builtin_kind);
}
auto attributes = rules_def->get_outer_attrs ();
/* Since the EarlyNameResolver runs multiple time (fixed point algorithm)
* we could be inserting the same macro def over and over again until we
* implement some optimizations */
// FIXME: ARTHUR: Remove that lookup and add proper optimizations instead
AST::MacroRulesDefinition *tmp_def = nullptr;
if (mappings.lookup_macro_invocation (invoc, &tmp_def))
return;
mappings.insert_macro_invocation (invoc, rules_def);
}
// FIXME: ARTHUR: Do we need to resolve these as well here?
void
EarlyNameResolver::visit (AST::MetaItemPath &)
{}
void
EarlyNameResolver::visit (AST::MetaItemSeq &)
{}
void
EarlyNameResolver::visit (AST::MetaWord &)
{}
void
EarlyNameResolver::visit (AST::MetaNameValueStr &)
{}
void
EarlyNameResolver::visit (AST::MetaListPaths &)
{}
void
EarlyNameResolver::visit (AST::MetaListNameValueStr &)
{}
void
EarlyNameResolver::visit (AST::LiteralPattern &)
{}
void
EarlyNameResolver::visit (AST::IdentifierPattern &pattern)
{
if (pattern.has_pattern_to_bind ())
pattern.get_pattern_to_bind ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::WildcardPattern &)
{}
void
EarlyNameResolver::visit (AST::RestPattern &)
{}
void
EarlyNameResolver::visit (AST::RangePatternBoundLiteral &)
{}
void
EarlyNameResolver::visit (AST::RangePatternBoundPath &)
{}
void
EarlyNameResolver::visit (AST::RangePatternBoundQualPath &)
{}
void
EarlyNameResolver::visit (AST::RangePattern &pattern)
{
pattern.get_lower_bound ()->accept_vis (*this);
pattern.get_upper_bound ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ReferencePattern &pattern)
{
pattern.get_referenced_pattern ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::StructPatternFieldTuplePat &field)
{
field.get_index_pattern ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::StructPatternFieldIdentPat &field)
{
field.get_ident_pattern ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::StructPatternFieldIdent &)
{}
void
EarlyNameResolver::visit (AST::StructPattern &)
{}
void
EarlyNameResolver::visit (AST::TupleStructItemsNoRange &tuple_items)
{
for (auto &pattern : tuple_items.get_patterns ())
pattern->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TupleStructItemsRange &tuple_items)
{
for (auto &pattern : tuple_items.get_lower_patterns ())
pattern->accept_vis (*this);
for (auto &pattern : tuple_items.get_upper_patterns ())
pattern->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TupleStructPattern &pattern)
{
pattern.get_items ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TuplePatternItemsMultiple &tuple_items)
{
for (auto &pattern : tuple_items.get_patterns ())
pattern->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TuplePatternItemsRanged &tuple_items)
{
for (auto &pattern : tuple_items.get_lower_patterns ())
pattern->accept_vis (*this);
for (auto &pattern : tuple_items.get_upper_patterns ())
pattern->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TuplePattern &pattern)
{
pattern.get_items ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::GroupedPattern &pattern)
{
pattern.get_pattern_in_parens ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::SlicePattern &pattern)
{
for (auto &item : pattern.get_items ())
item->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::AltPattern &pattern)
{
for (auto &alt : pattern.get_alts ())
alt->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::EmptyStmt &)
{}
void
EarlyNameResolver::visit (AST::LetStmt &stmt)
{
if (stmt.has_type ())
stmt.get_type ()->accept_vis (*this);
if (stmt.has_init_expr ())
stmt.get_init_expr ()->accept_vis (*this);
stmt.get_pattern ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::ExprStmt &stmt)
{
stmt.get_expr ()->accept_vis (*this);
}
void
EarlyNameResolver::visit (AST::TraitBound &)
{}
void
EarlyNameResolver::visit (AST::ImplTraitType &)
{}
void
EarlyNameResolver::visit (AST::TraitObjectType &)
{}
void
EarlyNameResolver::visit (AST::ParenthesisedType &)
{}
void
EarlyNameResolver::visit (AST::ImplTraitTypeOneBound &)
{}
void
EarlyNameResolver::visit (AST::TraitObjectTypeOneBound &)
{}
void
EarlyNameResolver::visit (AST::TupleType &)
{}
void
EarlyNameResolver::visit (AST::NeverType &)
{}
void
EarlyNameResolver::visit (AST::RawPointerType &)
{}
void
EarlyNameResolver::visit (AST::ReferenceType &)
{}
void
EarlyNameResolver::visit (AST::ArrayType &)
{}
void
EarlyNameResolver::visit (AST::SliceType &)
{}
void
EarlyNameResolver::visit (AST::InferredType &)
{}
void
EarlyNameResolver::visit (AST::BareFunctionType &type)
{
for (auto ¶m : type.get_function_params ())
param.get_type ()->accept_vis (*this);
if (type.has_return_type ())
type.get_return_type ()->accept_vis (*this);
}
} // namespace Resolver
} // namespace Rust