Prepare SourceAnalyzer for macros

This commit is contained in:
Aleksey Kladov 2019-11-16 00:40:54 +03:00
parent bd8af6a413
commit 0404e647e6
2 changed files with 14 additions and 9 deletions

View File

@ -91,7 +91,7 @@ fn def_with_body_from_child_node(
#[derive(Debug)]
pub struct SourceAnalyzer {
// FIXME: this doesn't handle macros at all
file_id: FileId,
file_id: HirFileId,
resolver: Resolver,
body_owner: Option<DefWithBody>,
body_source_map: Option<Arc<BodySourceMap>>,
@ -159,7 +159,7 @@ pub fn new(
body_source_map: Some(source_map),
infer: Some(def.infer(db)),
scopes: Some(scopes),
file_id,
file_id: file_id.into(),
}
} else {
SourceAnalyzer {
@ -171,18 +171,18 @@ pub fn new(
body_source_map: None,
infer: None,
scopes: None,
file_id,
file_id: file_id.into(),
}
}
}
fn expr_id(&self, expr: &ast::Expr) -> Option<ExprId> {
let src = Source { file_id: self.file_id.into(), ast: expr };
let src = Source { file_id: self.file_id, ast: expr };
self.body_source_map.as_ref()?.node_expr(src)
}
fn pat_id(&self, pat: &ast::Pat) -> Option<PatId> {
let src = Source { file_id: self.file_id.into(), ast: pat };
let src = Source { file_id: self.file_id, ast: pat };
self.body_source_map.as_ref()?.node_pat(src)
}
@ -290,8 +290,7 @@ fn resolve_local_name(&self, name_ref: &ast::NameRef) -> Option<ScopeEntryWithSy
let name = name_ref.as_name();
let source_map = self.body_source_map.as_ref()?;
let scopes = self.scopes.as_ref()?;
let scope =
scope_for(scopes, source_map, Source::new(self.file_id.into(), name_ref.syntax()))?;
let scope = scope_for(scopes, source_map, Source::new(self.file_id, name_ref.syntax()))?;
let entry = scopes.resolve_name_in_scope(scope, &name)?;
Some(ScopeEntryWithSyntax {
name: entry.name().clone(),

View File

@ -223,6 +223,7 @@ pub fn to_node(&self, db: &dyn db::AstDatabase) -> N {
}
}
/// FIXME: https://github.com/matklad/with ?
#[derive(Debug, PartialEq, Eq, Clone, Copy, Hash)]
pub struct Source<T> {
pub file_id: HirFileId,
@ -236,11 +237,16 @@ pub fn new(file_id: HirFileId, ast: T) -> Source<T> {
Source { file_id, ast }
}
// Similarly, naming here is stupid...
pub fn with_ast<U>(&self, ast: U) -> Source<U> {
Source::new(self.file_id, ast)
}
pub fn map<F: FnOnce(T) -> U, U>(self, f: F) -> Source<U> {
Source { file_id: self.file_id, ast: f(self.ast) }
Source::new(self.file_id, f(self.ast))
}
pub fn as_ref(&self) -> Source<&T> {
Source { file_id: self.file_id, ast: &self.ast }
self.with_ast(&self.ast)
}
pub fn file_syntax(&self, db: &impl db::AstDatabase) -> SyntaxNode {
db.parse_or_expand(self.file_id).expect("source created from invalid file")