Move child_by_source from hir-def to hir
This commit is contained in:
parent
2d5b758439
commit
bc862b897d
@ -21,7 +21,79 @@
|
|||||||
//!
|
//!
|
||||||
//! This is a work of fiction. Any similarities to Kotlin's `BindingContext` are
|
//! This is a work of fiction. Any similarities to Kotlin's `BindingContext` are
|
||||||
//! a coincidence.
|
//! a coincidence.
|
||||||
pub mod keys;
|
|
||||||
|
pub mod keys {
|
||||||
|
use std::marker::PhantomData;
|
||||||
|
|
||||||
|
use hir_expand::{attrs::AttrId, MacroCallId};
|
||||||
|
use rustc_hash::FxHashMap;
|
||||||
|
use syntax::{ast, AstNode, AstPtr};
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
dyn_map::{DynMap, Policy},
|
||||||
|
BlockId, ConstId, EnumId, EnumVariantId, ExternCrateId, FieldId, FunctionId, ImplId,
|
||||||
|
LifetimeParamId, Macro2Id, MacroRulesId, ProcMacroId, StaticId, StructId, TraitAliasId,
|
||||||
|
TraitId, TypeAliasId, TypeOrConstParamId, UnionId, UseId,
|
||||||
|
};
|
||||||
|
|
||||||
|
pub type Key<K, V> = crate::dyn_map::Key<AstPtr<K>, V, AstPtrPolicy<K, V>>;
|
||||||
|
|
||||||
|
pub const BLOCK: Key<ast::BlockExpr, BlockId> = Key::new();
|
||||||
|
pub const FUNCTION: Key<ast::Fn, FunctionId> = Key::new();
|
||||||
|
pub const CONST: Key<ast::Const, ConstId> = Key::new();
|
||||||
|
pub const STATIC: Key<ast::Static, StaticId> = Key::new();
|
||||||
|
pub const TYPE_ALIAS: Key<ast::TypeAlias, TypeAliasId> = Key::new();
|
||||||
|
pub const IMPL: Key<ast::Impl, ImplId> = Key::new();
|
||||||
|
pub const TRAIT: Key<ast::Trait, TraitId> = Key::new();
|
||||||
|
pub const TRAIT_ALIAS: Key<ast::TraitAlias, TraitAliasId> = Key::new();
|
||||||
|
pub const STRUCT: Key<ast::Struct, StructId> = Key::new();
|
||||||
|
pub const UNION: Key<ast::Union, UnionId> = Key::new();
|
||||||
|
pub const ENUM: Key<ast::Enum, EnumId> = Key::new();
|
||||||
|
pub const EXTERN_CRATE: Key<ast::ExternCrate, ExternCrateId> = Key::new();
|
||||||
|
pub const USE: Key<ast::Use, UseId> = Key::new();
|
||||||
|
|
||||||
|
pub const ENUM_VARIANT: Key<ast::Variant, EnumVariantId> = Key::new();
|
||||||
|
pub const TUPLE_FIELD: Key<ast::TupleField, FieldId> = Key::new();
|
||||||
|
pub const RECORD_FIELD: Key<ast::RecordField, FieldId> = Key::new();
|
||||||
|
pub const TYPE_PARAM: Key<ast::TypeParam, TypeOrConstParamId> = Key::new();
|
||||||
|
pub const CONST_PARAM: Key<ast::ConstParam, TypeOrConstParamId> = Key::new();
|
||||||
|
pub const LIFETIME_PARAM: Key<ast::LifetimeParam, LifetimeParamId> = Key::new();
|
||||||
|
|
||||||
|
pub const MACRO_RULES: Key<ast::MacroRules, MacroRulesId> = Key::new();
|
||||||
|
pub const MACRO2: Key<ast::MacroDef, Macro2Id> = Key::new();
|
||||||
|
pub const PROC_MACRO: Key<ast::Fn, ProcMacroId> = Key::new();
|
||||||
|
pub const MACRO_CALL: Key<ast::MacroCall, MacroCallId> = Key::new();
|
||||||
|
pub const ATTR_MACRO_CALL: Key<ast::Item, MacroCallId> = Key::new();
|
||||||
|
pub const DERIVE_MACRO_CALL: Key<ast::Attr, (AttrId, MacroCallId, Box<[Option<MacroCallId>]>)> =
|
||||||
|
Key::new();
|
||||||
|
|
||||||
|
/// XXX: AST Nodes and SyntaxNodes have identity equality semantics: nodes are
|
||||||
|
/// equal if they point to exactly the same object.
|
||||||
|
///
|
||||||
|
/// In general, we do not guarantee that we have exactly one instance of a
|
||||||
|
/// syntax tree for each file. We probably should add such guarantee, but, for
|
||||||
|
/// the time being, we will use identity-less AstPtr comparison.
|
||||||
|
pub struct AstPtrPolicy<AST, ID> {
|
||||||
|
_phantom: PhantomData<(AST, ID)>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<AST: AstNode + 'static, ID: 'static> Policy for AstPtrPolicy<AST, ID> {
|
||||||
|
type K = AstPtr<AST>;
|
||||||
|
type V = ID;
|
||||||
|
fn insert(map: &mut DynMap, key: AstPtr<AST>, value: ID) {
|
||||||
|
map.map
|
||||||
|
.entry::<FxHashMap<AstPtr<AST>, ID>>()
|
||||||
|
.or_insert_with(Default::default)
|
||||||
|
.insert(key, value);
|
||||||
|
}
|
||||||
|
fn get<'a>(map: &'a DynMap, key: &AstPtr<AST>) -> Option<&'a ID> {
|
||||||
|
map.map.get::<FxHashMap<AstPtr<AST>, ID>>()?.get(key)
|
||||||
|
}
|
||||||
|
fn is_empty(map: &DynMap) -> bool {
|
||||||
|
map.map.get::<FxHashMap<AstPtr<AST>, ID>>().map_or(true, |it| it.is_empty())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
use std::{
|
use std::{
|
||||||
hash::Hash,
|
hash::Hash,
|
||||||
|
@ -1,72 +0,0 @@
|
|||||||
//! keys to be used with `DynMap`
|
|
||||||
|
|
||||||
use std::marker::PhantomData;
|
|
||||||
|
|
||||||
use hir_expand::{attrs::AttrId, MacroCallId};
|
|
||||||
use rustc_hash::FxHashMap;
|
|
||||||
use syntax::{ast, AstNode, AstPtr};
|
|
||||||
|
|
||||||
use crate::{
|
|
||||||
dyn_map::{DynMap, Policy},
|
|
||||||
BlockId, ConstId, EnumId, EnumVariantId, ExternCrateId, FieldId, FunctionId, ImplId,
|
|
||||||
LifetimeParamId, Macro2Id, MacroRulesId, ProcMacroId, StaticId, StructId, TraitAliasId,
|
|
||||||
TraitId, TypeAliasId, TypeOrConstParamId, UnionId, UseId,
|
|
||||||
};
|
|
||||||
|
|
||||||
pub type Key<K, V> = crate::dyn_map::Key<AstPtr<K>, V, AstPtrPolicy<K, V>>;
|
|
||||||
|
|
||||||
pub const BLOCK: Key<ast::BlockExpr, BlockId> = Key::new();
|
|
||||||
pub const FUNCTION: Key<ast::Fn, FunctionId> = Key::new();
|
|
||||||
pub const CONST: Key<ast::Const, ConstId> = Key::new();
|
|
||||||
pub const STATIC: Key<ast::Static, StaticId> = Key::new();
|
|
||||||
pub const TYPE_ALIAS: Key<ast::TypeAlias, TypeAliasId> = Key::new();
|
|
||||||
pub const IMPL: Key<ast::Impl, ImplId> = Key::new();
|
|
||||||
pub const TRAIT: Key<ast::Trait, TraitId> = Key::new();
|
|
||||||
pub const TRAIT_ALIAS: Key<ast::TraitAlias, TraitAliasId> = Key::new();
|
|
||||||
pub const STRUCT: Key<ast::Struct, StructId> = Key::new();
|
|
||||||
pub const UNION: Key<ast::Union, UnionId> = Key::new();
|
|
||||||
pub const ENUM: Key<ast::Enum, EnumId> = Key::new();
|
|
||||||
pub const EXTERN_CRATE: Key<ast::ExternCrate, ExternCrateId> = Key::new();
|
|
||||||
pub const USE: Key<ast::Use, UseId> = Key::new();
|
|
||||||
|
|
||||||
pub const ENUM_VARIANT: Key<ast::Variant, EnumVariantId> = Key::new();
|
|
||||||
pub const TUPLE_FIELD: Key<ast::TupleField, FieldId> = Key::new();
|
|
||||||
pub const RECORD_FIELD: Key<ast::RecordField, FieldId> = Key::new();
|
|
||||||
pub const TYPE_PARAM: Key<ast::TypeParam, TypeOrConstParamId> = Key::new();
|
|
||||||
pub const CONST_PARAM: Key<ast::ConstParam, TypeOrConstParamId> = Key::new();
|
|
||||||
pub const LIFETIME_PARAM: Key<ast::LifetimeParam, LifetimeParamId> = Key::new();
|
|
||||||
|
|
||||||
pub const MACRO_RULES: Key<ast::MacroRules, MacroRulesId> = Key::new();
|
|
||||||
pub const MACRO2: Key<ast::MacroDef, Macro2Id> = Key::new();
|
|
||||||
pub const PROC_MACRO: Key<ast::Fn, ProcMacroId> = Key::new();
|
|
||||||
pub const MACRO_CALL: Key<ast::MacroCall, MacroCallId> = Key::new();
|
|
||||||
pub const ATTR_MACRO_CALL: Key<ast::Item, MacroCallId> = Key::new();
|
|
||||||
pub const DERIVE_MACRO_CALL: Key<ast::Attr, (AttrId, MacroCallId, Box<[Option<MacroCallId>]>)> =
|
|
||||||
Key::new();
|
|
||||||
|
|
||||||
/// XXX: AST Nodes and SyntaxNodes have identity equality semantics: nodes are
|
|
||||||
/// equal if they point to exactly the same object.
|
|
||||||
///
|
|
||||||
/// In general, we do not guarantee that we have exactly one instance of a
|
|
||||||
/// syntax tree for each file. We probably should add such guarantee, but, for
|
|
||||||
/// the time being, we will use identity-less AstPtr comparison.
|
|
||||||
pub struct AstPtrPolicy<AST, ID> {
|
|
||||||
_phantom: PhantomData<(AST, ID)>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<AST: AstNode + 'static, ID: 'static> Policy for AstPtrPolicy<AST, ID> {
|
|
||||||
type K = AstPtr<AST>;
|
|
||||||
type V = ID;
|
|
||||||
fn insert(map: &mut DynMap, key: AstPtr<AST>, value: ID) {
|
|
||||||
map.map
|
|
||||||
.entry::<FxHashMap<AstPtr<AST>, ID>>()
|
|
||||||
.or_insert_with(Default::default)
|
|
||||||
.insert(key, value);
|
|
||||||
}
|
|
||||||
fn get<'a>(map: &'a DynMap, key: &AstPtr<AST>) -> Option<&'a ID> {
|
|
||||||
map.map.get::<FxHashMap<AstPtr<AST>, ID>>()?.get(key)
|
|
||||||
}
|
|
||||||
fn is_empty(map: &DynMap) -> bool {
|
|
||||||
map.map.get::<FxHashMap<AstPtr<AST>, ID>>().map_or(true, |it| it.is_empty())
|
|
||||||
}
|
|
||||||
}
|
|
@ -361,9 +361,7 @@ impl ItemScope {
|
|||||||
self.macro_invocations.get(&call).copied()
|
self.macro_invocations.get(&call).copied()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn iter_macro_invoc(
|
pub fn iter_macro_invoc(&self) -> impl Iterator<Item = (&AstId<ast::MacroCall>, &MacroCallId)> {
|
||||||
&self,
|
|
||||||
) -> impl Iterator<Item = (&AstId<ast::MacroCall>, &MacroCallId)> {
|
|
||||||
self.macro_invocations.iter()
|
self.macro_invocations.iter()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -401,9 +399,7 @@ impl ItemScope {
|
|||||||
self.macro_invocations.insert(call, call_id);
|
self.macro_invocations.insert(call, call_id);
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn attr_macro_invocs(
|
pub fn attr_macro_invocs(&self) -> impl Iterator<Item = (AstId<ast::Item>, MacroCallId)> + '_ {
|
||||||
&self,
|
|
||||||
) -> impl Iterator<Item = (AstId<ast::Item>, MacroCallId)> + '_ {
|
|
||||||
self.attr_macros.iter().map(|(k, v)| (*k, *v))
|
self.attr_macros.iter().map(|(k, v)| (*k, *v))
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -440,7 +436,7 @@ impl ItemScope {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn derive_macro_invocs(
|
pub fn derive_macro_invocs(
|
||||||
&self,
|
&self,
|
||||||
) -> impl Iterator<
|
) -> impl Iterator<
|
||||||
Item = (
|
Item = (
|
||||||
|
@ -47,7 +47,6 @@ pub mod resolver;
|
|||||||
|
|
||||||
pub mod nameres;
|
pub mod nameres;
|
||||||
|
|
||||||
pub mod child_by_source;
|
|
||||||
pub mod src;
|
pub mod src;
|
||||||
|
|
||||||
pub mod find_path;
|
pub mod find_path;
|
||||||
@ -354,9 +353,9 @@ impl_loc!(ProcMacroLoc, id: Function, container: CrateRootModuleId);
|
|||||||
pub struct BlockId(ra_salsa::InternId);
|
pub struct BlockId(ra_salsa::InternId);
|
||||||
#[derive(Debug, Hash, PartialEq, Eq, Clone)]
|
#[derive(Debug, Hash, PartialEq, Eq, Clone)]
|
||||||
pub struct BlockLoc {
|
pub struct BlockLoc {
|
||||||
ast_id: AstId<ast::BlockExpr>,
|
pub ast_id: AstId<ast::BlockExpr>,
|
||||||
/// The containing module.
|
/// The containing module.
|
||||||
module: ModuleId,
|
pub module: ModuleId,
|
||||||
}
|
}
|
||||||
impl_intern!(BlockId, BlockLoc, intern_block, lookup_intern_block);
|
impl_intern!(BlockId, BlockLoc, intern_block, lookup_intern_block);
|
||||||
|
|
||||||
@ -935,7 +934,7 @@ impl_from!(
|
|||||||
);
|
);
|
||||||
|
|
||||||
impl GenericDefId {
|
impl GenericDefId {
|
||||||
fn file_id_and_params_of(
|
pub fn file_id_and_params_of(
|
||||||
self,
|
self,
|
||||||
db: &dyn DefDatabase,
|
db: &dyn DefDatabase,
|
||||||
) -> (HirFileId, Option<ast::GenericParamList>) {
|
) -> (HirFileId, Option<ast::GenericParamList>) {
|
||||||
|
@ -1,5 +1,6 @@
|
|||||||
//! See `Semantics`.
|
//! See `Semantics`.
|
||||||
|
|
||||||
|
mod child_by_source;
|
||||||
mod source_to_def;
|
mod source_to_def;
|
||||||
|
|
||||||
use std::{
|
use std::{
|
||||||
|
@ -8,7 +8,7 @@ use either::Either;
|
|||||||
use hir_expand::{attrs::collect_attrs, HirFileId};
|
use hir_expand::{attrs::collect_attrs, HirFileId};
|
||||||
use syntax::{ast, AstPtr};
|
use syntax::{ast, AstPtr};
|
||||||
|
|
||||||
use crate::{
|
use hir_def::{
|
||||||
db::DefDatabase,
|
db::DefDatabase,
|
||||||
dyn_map::{
|
dyn_map::{
|
||||||
keys::{self, Key},
|
keys::{self, Key},
|
||||||
@ -23,7 +23,7 @@ use crate::{
|
|||||||
VariantId,
|
VariantId,
|
||||||
};
|
};
|
||||||
|
|
||||||
pub trait ChildBySource {
|
pub(crate) trait ChildBySource {
|
||||||
fn child_by_source(&self, db: &dyn DefDatabase, file_id: HirFileId) -> DynMap {
|
fn child_by_source(&self, db: &dyn DefDatabase, file_id: HirFileId) -> DynMap {
|
||||||
let mut res = DynMap::default();
|
let mut res = DynMap::default();
|
||||||
self.child_by_source_to(db, &mut res, file_id);
|
self.child_by_source_to(db, &mut res, file_id);
|
@ -87,7 +87,6 @@
|
|||||||
|
|
||||||
use either::Either;
|
use either::Either;
|
||||||
use hir_def::{
|
use hir_def::{
|
||||||
child_by_source::ChildBySource,
|
|
||||||
dyn_map::{
|
dyn_map::{
|
||||||
keys::{self, Key},
|
keys::{self, Key},
|
||||||
DynMap,
|
DynMap,
|
||||||
@ -111,7 +110,10 @@ use syntax::{
|
|||||||
AstNode, AstPtr, SyntaxNode,
|
AstNode, AstPtr, SyntaxNode,
|
||||||
};
|
};
|
||||||
|
|
||||||
use crate::{db::HirDatabase, InFile, InlineAsmOperand, SemanticsImpl};
|
use crate::{
|
||||||
|
db::HirDatabase, semantics::child_by_source::ChildBySource, InFile, InlineAsmOperand,
|
||||||
|
SemanticsImpl,
|
||||||
|
};
|
||||||
|
|
||||||
#[derive(Default)]
|
#[derive(Default)]
|
||||||
pub(super) struct SourceToDefCache {
|
pub(super) struct SourceToDefCache {
|
||||||
|
@ -104,7 +104,7 @@ fn edit_struct_def(
|
|||||||
ast::make::tokens::single_newline().text(),
|
ast::make::tokens::single_newline().text(),
|
||||||
);
|
);
|
||||||
edit.insert(tuple_fields_text_range.start(), w.syntax().text());
|
edit.insert(tuple_fields_text_range.start(), w.syntax().text());
|
||||||
if !w.syntax().last_token().is_some_and(|t| t.kind() == SyntaxKind::COMMA) {
|
if w.syntax().last_token().is_none_or(|t| t.kind() != SyntaxKind::COMMA) {
|
||||||
edit.insert(tuple_fields_text_range.start(), ",");
|
edit.insert(tuple_fields_text_range.start(), ",");
|
||||||
}
|
}
|
||||||
edit.insert(
|
edit.insert(
|
||||||
|
@ -382,7 +382,7 @@ pub fn semantic_diagnostics(
|
|||||||
// A bunch of parse errors in a file indicate some bigger structural parse changes in the
|
// A bunch of parse errors in a file indicate some bigger structural parse changes in the
|
||||||
// file, so we skip semantic diagnostics so we can show these faster.
|
// file, so we skip semantic diagnostics so we can show these faster.
|
||||||
Some(m) => {
|
Some(m) => {
|
||||||
if !db.parse_errors(file_id).as_deref().is_some_and(|es| es.len() >= 16) {
|
if db.parse_errors(file_id).as_deref().is_none_or(|es| es.len() < 16) {
|
||||||
m.diagnostics(db, &mut diags, config.style_lints);
|
m.diagnostics(db, &mut diags, config.style_lints);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user