version = "0.1.0"
dependencies = [
"fst 0.3.2 (registry+https://github.com/rust-lang/crates.io-index)",
- "id-arena 1.0.2 (git+https://github.com/fitzgen/id-arena/?rev=43ecd67)",
"log 0.4.6 (registry+https://github.com/rust-lang/crates.io-index)",
"parking_lot 0.6.4 (registry+https://github.com/rust-lang/crates.io-index)",
"ra_db 0.1.0",
"ra_editor 0.1.0",
+ "ra_hir 0.1.0",
"ra_syntax 0.1.0",
"rayon 1.0.3 (registry+https://github.com/rust-lang/crates.io-index)",
"relative-path 0.4.0 (registry+https://github.com/rust-lang/crates.io-index)",
salsa = "0.8.0"
rustc-hash = "1.0"
parking_lot = "0.6.4"
-id-arena = { git = "https://github.com/fitzgen/id-arena/", rev = "43ecd67" }
ra_syntax = { path = "../ra_syntax" }
ra_editor = { path = "../ra_editor" }
ra_db = { path = "../ra_db" }
+hir = { path = "../ra_hir", package = "ra_hir" }
test_utils = { path = "../test_utils" }
+++ /dev/null
-//! A simple id-based arena, similar to https://github.com/fitzgen/id-arena.
-//! We use our own version for more compact id's and to allow inherent impls
-//! on Ids.
-
-use std::{
- fmt,
- hash::{Hash, Hasher},
- marker::PhantomData,
-};
-
-pub(crate) struct Id<T> {
- idx: u32,
- _ty: PhantomData<fn() -> T>,
-}
-
-impl<T> fmt::Debug for Id<T> {
- fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
- f.debug_tuple("Id").field(&self.idx).finish()
- }
-}
-impl<T> Copy for Id<T> {}
-impl<T> Clone for Id<T> {
- fn clone(&self) -> Id<T> {
- *self
- }
-}
-
-impl<T> PartialEq for Id<T> {
- fn eq(&self, other: &Id<T>) -> bool {
- self.idx == other.idx
- }
-}
-
-impl<T> Eq for Id<T> {}
-
-impl<T> Hash for Id<T> {
- fn hash<H: Hasher>(&self, h: &mut H) {
- self.idx.hash(h);
- }
-}
-
-#[derive(Debug, PartialEq, Eq)]
-pub(crate) struct ArenaBehavior<T> {
- _ty: PhantomData<T>,
-}
-
-impl<T> id_arena::ArenaBehavior for ArenaBehavior<T> {
- type Id = Id<T>;
- fn new_arena_id() -> usize {
- 0
- }
- fn new_id(_arena_id: usize, index: usize) -> Id<T> {
- Id {
- idx: index as u32,
- _ty: PhantomData,
- }
- }
- fn index(id: Id<T>) -> usize {
- id.idx as usize
- }
- fn arena_id(_id: Id<T>) -> usize {
- 0
- }
-}
-
-pub(crate) type Arena<T> = id_arena::Arena<T, ArenaBehavior<T>>;
use crate::{
db,
- hir,
Cancelable, FilePosition
};
ast::{self, LoopBodyOwner},
SyntaxKind::*,
};
-
-use crate::{
- db::RootDatabase,
- completion::CompletionItem,
- hir::{
+use hir::{
self,
FnScopes,
Def,
Path,
- },
+};
+
+use crate::{
+ db::RootDatabase,
+ completion::CompletionItem,
Cancelable
};
use parking_lot::Mutex;
use salsa::{self, Database};
use ra_db::{LocationIntener, BaseDatabase};
+use hir::{self, DefId, DefLoc, FnId, SourceItemId};
use crate::{
- hir::{self, DefId, DefLoc, FnId, SourceItemId},
symbol_index,
};
+++ /dev/null
-use std::sync::Arc;
-
-use ra_syntax::{
- SyntaxNode,
- ast::FnDefNode,
-};
-use ra_db::{SourceRootId, LocationIntener, SyntaxDatabase};
-
-use crate::{
- FileId,
- hir::{
- DefLoc, DefId, FnId,
- SourceFileItems, SourceItemId,
- query_definitions,
- function::{FnScopes},
- module::{ModuleId, ModuleTree, ModuleSource,
- nameres::{ItemMap, InputModuleItems}},
- },
- Cancelable,
-};
-
-salsa::query_group! {
-
-pub(crate) trait HirDatabase: SyntaxDatabase
- + AsRef<LocationIntener<DefLoc, DefId>>
- + AsRef<LocationIntener<SourceItemId, FnId>>
-{
- fn fn_scopes(fn_id: FnId) -> Arc<FnScopes> {
- type FnScopesQuery;
- use fn query_definitions::fn_scopes;
- }
- fn fn_syntax(fn_id: FnId) -> FnDefNode {
- type FnSyntaxQuery;
- // Don't retain syntax trees in memory
- storage dependencies;
- use fn query_definitions::fn_syntax;
- }
-
- fn file_items(file_id: FileId) -> Arc<SourceFileItems> {
- type SourceFileItemsQuery;
- storage dependencies;
- use fn query_definitions::file_items;
- }
-
- fn file_item(source_item_id: SourceItemId) -> SyntaxNode {
- type FileItemQuery;
- storage dependencies;
- use fn query_definitions::file_item;
- }
-
- fn submodules(source: ModuleSource) -> Cancelable<Arc<Vec<crate::hir::module::imp::Submodule>>> {
- type SubmodulesQuery;
- use fn query_definitions::submodules;
- }
-
- fn input_module_items(source_root_id: SourceRootId, module_id: ModuleId) -> Cancelable<Arc<InputModuleItems>> {
- type InputModuleItemsQuery;
- use fn query_definitions::input_module_items;
- }
- fn item_map(source_root_id: SourceRootId) -> Cancelable<Arc<ItemMap>> {
- type ItemMapQuery;
- use fn query_definitions::item_map;
- }
- fn module_tree(source_root_id: SourceRootId) -> Cancelable<Arc<ModuleTree>> {
- type ModuleTreeQuery;
- use fn crate::hir::module::imp::module_tree;
- }
-}
-
-}
+++ /dev/null
-mod scope;
-
-use std::{
- cmp::{max, min},
- sync::Arc,
-};
-
-use ra_syntax::{
- TextRange, TextUnit, SyntaxNodeRef,
- ast::{self, AstNode, DocCommentsOwner, NameOwner},
-};
-
-use crate::{
- hir::{FnId, HirDatabase, SourceItemId},
- FileId,
-};
-
-pub(crate) use self::scope::FnScopes;
-
-impl FnId {
- pub(crate) fn get(db: &impl HirDatabase, file_id: FileId, fn_def: ast::FnDef) -> FnId {
- let file_items = db.file_items(file_id);
- let item_id = file_items.id_of(fn_def.syntax());
- let item_id = SourceItemId { file_id, item_id };
- FnId::from_loc(db, &item_id)
- }
-}
-
-pub(crate) struct Function {
- fn_id: FnId,
-}
-
-impl Function {
- pub(crate) fn guess_from_source(
- db: &impl HirDatabase,
- file_id: FileId,
- fn_def: ast::FnDef,
- ) -> Function {
- let fn_id = FnId::get(db, file_id, fn_def);
- Function { fn_id }
- }
-
- pub(crate) fn guess_for_name_ref(
- db: &impl HirDatabase,
- file_id: FileId,
- name_ref: ast::NameRef,
- ) -> Option<Function> {
- Function::guess_for_node(db, file_id, name_ref.syntax())
- }
-
- pub(crate) fn guess_for_bind_pat(
- db: &impl HirDatabase,
- file_id: FileId,
- bind_pat: ast::BindPat,
- ) -> Option<Function> {
- Function::guess_for_node(db, file_id, bind_pat.syntax())
- }
-
- fn guess_for_node(
- db: &impl HirDatabase,
- file_id: FileId,
- node: SyntaxNodeRef,
- ) -> Option<Function> {
- let fn_def = node.ancestors().find_map(ast::FnDef::cast)?;
- let res = Function::guess_from_source(db, file_id, fn_def);
- Some(res)
- }
-
- pub(crate) fn scope(&self, db: &impl HirDatabase) -> Arc<FnScopes> {
- db.fn_scopes(self.fn_id)
- }
-
- pub(crate) fn signature_info(&self, db: &impl HirDatabase) -> Option<FnSignatureInfo> {
- let syntax = db.fn_syntax(self.fn_id);
- FnSignatureInfo::new(syntax.borrowed())
- }
-}
-
-#[derive(Debug, Clone)]
-pub struct FnSignatureInfo {
- pub name: String,
- pub label: String,
- pub ret_type: Option<String>,
- pub params: Vec<String>,
- pub doc: Option<String>,
-}
-
-impl FnSignatureInfo {
- fn new(node: ast::FnDef) -> Option<Self> {
- let name = node.name()?.text().to_string();
-
- let mut doc = None;
-
- // Strip the body out for the label.
- let mut label: String = if let Some(body) = node.body() {
- let body_range = body.syntax().range();
- let label: String = node
- .syntax()
- .children()
- .filter(|child| !child.range().is_subrange(&body_range))
- .map(|node| node.text().to_string())
- .collect();
- label
- } else {
- node.syntax().text().to_string()
- };
-
- if let Some((comment_range, docs)) = FnSignatureInfo::extract_doc_comments(node) {
- let comment_range = comment_range
- .checked_sub(node.syntax().range().start())
- .unwrap();
- let start = comment_range.start().to_usize();
- let end = comment_range.end().to_usize();
-
- // Remove the comment from the label
- label.replace_range(start..end, "");
-
- // Massage markdown
- let mut processed_lines = Vec::new();
- let mut in_code_block = false;
- for line in docs.lines() {
- if line.starts_with("```") {
- in_code_block = !in_code_block;
- }
-
- let line = if in_code_block && line.starts_with("```") && !line.contains("rust") {
- "```rust".into()
- } else {
- line.to_string()
- };
-
- processed_lines.push(line);
- }
-
- if !processed_lines.is_empty() {
- doc = Some(processed_lines.join("\n"));
- }
- }
-
- let params = FnSignatureInfo::param_list(node);
- let ret_type = node.ret_type().map(|r| r.syntax().text().to_string());
-
- Some(FnSignatureInfo {
- name,
- ret_type,
- params,
- label: label.trim().to_owned(),
- doc,
- })
- }
-
- fn extract_doc_comments(node: ast::FnDef) -> Option<(TextRange, String)> {
- if node.doc_comments().count() == 0 {
- return None;
- }
-
- let comment_text = node.doc_comment_text();
-
- let (begin, end) = node
- .doc_comments()
- .map(|comment| comment.syntax().range())
- .map(|range| (range.start().to_usize(), range.end().to_usize()))
- .fold((std::usize::MAX, std::usize::MIN), |acc, range| {
- (min(acc.0, range.0), max(acc.1, range.1))
- });
-
- let range = TextRange::from_to(TextUnit::from_usize(begin), TextUnit::from_usize(end));
-
- Some((range, comment_text))
- }
-
- fn param_list(node: ast::FnDef) -> Vec<String> {
- let mut res = vec![];
- if let Some(param_list) = node.param_list() {
- if let Some(self_param) = param_list.self_param() {
- res.push(self_param.syntax().text().to_string())
- }
-
- // Maybe use param.pat here? See if we can just extract the name?
- //res.extend(param_list.params().map(|p| p.syntax().text().to_string()));
- res.extend(
- param_list
- .params()
- .filter_map(|p| p.pat())
- .map(|pat| pat.syntax().text().to_string()),
- );
- }
- res
- }
-}
+++ /dev/null
-use rustc_hash::{FxHashMap, FxHashSet};
-
-use ra_syntax::{
- AstNode, SmolStr, SyntaxNodeRef, TextRange,
- algo::generate,
- ast::{self, ArgListOwner, LoopBodyOwner, NameOwner},
-};
-use ra_db::LocalSyntaxPtr;
-
-use crate::{
-
- arena::{Arena, Id},
-};
-
-pub(crate) type ScopeId = Id<ScopeData>;
-
-#[derive(Debug, PartialEq, Eq)]
-pub struct FnScopes {
- pub(crate) self_param: Option<LocalSyntaxPtr>,
- scopes: Arena<ScopeData>,
- scope_for: FxHashMap<LocalSyntaxPtr, ScopeId>,
-}
-
-#[derive(Debug, PartialEq, Eq)]
-pub struct ScopeEntry {
- name: SmolStr,
- ptr: LocalSyntaxPtr,
-}
-
-#[derive(Debug, PartialEq, Eq)]
-pub(crate) struct ScopeData {
- parent: Option<ScopeId>,
- entries: Vec<ScopeEntry>,
-}
-
-impl FnScopes {
- pub(crate) fn new(fn_def: ast::FnDef) -> FnScopes {
- let mut scopes = FnScopes {
- self_param: fn_def
- .param_list()
- .and_then(|it| it.self_param())
- .map(|it| LocalSyntaxPtr::new(it.syntax())),
- scopes: Arena::default(),
- scope_for: FxHashMap::default(),
- };
- let root = scopes.root_scope();
- scopes.add_params_bindings(root, fn_def.param_list());
- if let Some(body) = fn_def.body() {
- compute_block_scopes(body, &mut scopes, root)
- }
- scopes
- }
- pub(crate) fn entries(&self, scope: ScopeId) -> &[ScopeEntry] {
- &self.scopes[scope].entries
- }
- pub fn scope_chain<'a>(&'a self, node: SyntaxNodeRef) -> impl Iterator<Item = ScopeId> + 'a {
- generate(self.scope_for(node), move |&scope| {
- self.scopes[scope].parent
- })
- }
- pub(crate) fn resolve_local_name<'a>(
- &'a self,
- name_ref: ast::NameRef,
- ) -> Option<&'a ScopeEntry> {
- let mut shadowed = FxHashSet::default();
- let ret = self
- .scope_chain(name_ref.syntax())
- .flat_map(|scope| self.entries(scope).iter())
- .filter(|entry| shadowed.insert(entry.name()))
- .filter(|entry| entry.name() == &name_ref.text())
- .nth(0);
- ret
- }
-
- pub fn find_all_refs(&self, pat: ast::BindPat) -> Vec<ReferenceDescriptor> {
- let fn_def = pat.syntax().ancestors().find_map(ast::FnDef::cast).unwrap();
- let name_ptr = LocalSyntaxPtr::new(pat.syntax());
- let refs: Vec<_> = fn_def
- .syntax()
- .descendants()
- .filter_map(ast::NameRef::cast)
- .filter(|name_ref| match self.resolve_local_name(*name_ref) {
- None => false,
- Some(entry) => entry.ptr() == name_ptr,
- })
- .map(|name_ref| ReferenceDescriptor {
- name: name_ref.syntax().text().to_string(),
- range: name_ref.syntax().range(),
- })
- .collect();
-
- refs
- }
-
- fn root_scope(&mut self) -> ScopeId {
- self.scopes.alloc(ScopeData {
- parent: None,
- entries: vec![],
- })
- }
- fn new_scope(&mut self, parent: ScopeId) -> ScopeId {
- self.scopes.alloc(ScopeData {
- parent: Some(parent),
- entries: vec![],
- })
- }
- fn add_bindings(&mut self, scope: ScopeId, pat: ast::Pat) {
- let entries = pat
- .syntax()
- .descendants()
- .filter_map(ast::BindPat::cast)
- .filter_map(ScopeEntry::new);
- self.scopes[scope].entries.extend(entries);
- }
- fn add_params_bindings(&mut self, scope: ScopeId, params: Option<ast::ParamList>) {
- params
- .into_iter()
- .flat_map(|it| it.params())
- .filter_map(|it| it.pat())
- .for_each(|it| self.add_bindings(scope, it));
- }
- fn set_scope(&mut self, node: SyntaxNodeRef, scope: ScopeId) {
- self.scope_for.insert(LocalSyntaxPtr::new(node), scope);
- }
- fn scope_for(&self, node: SyntaxNodeRef) -> Option<ScopeId> {
- node.ancestors()
- .map(LocalSyntaxPtr::new)
- .filter_map(|it| self.scope_for.get(&it).map(|&scope| scope))
- .next()
- }
-}
-
-impl ScopeEntry {
- fn new(pat: ast::BindPat) -> Option<ScopeEntry> {
- let name = pat.name()?;
- let res = ScopeEntry {
- name: name.text(),
- ptr: LocalSyntaxPtr::new(pat.syntax()),
- };
- Some(res)
- }
- pub(crate) fn name(&self) -> &SmolStr {
- &self.name
- }
- pub(crate) fn ptr(&self) -> LocalSyntaxPtr {
- self.ptr
- }
-}
-
-fn compute_block_scopes(block: ast::Block, scopes: &mut FnScopes, mut scope: ScopeId) {
- for stmt in block.statements() {
- match stmt {
- ast::Stmt::LetStmt(stmt) => {
- if let Some(expr) = stmt.initializer() {
- scopes.set_scope(expr.syntax(), scope);
- compute_expr_scopes(expr, scopes, scope);
- }
- scope = scopes.new_scope(scope);
- if let Some(pat) = stmt.pat() {
- scopes.add_bindings(scope, pat);
- }
- }
- ast::Stmt::ExprStmt(expr_stmt) => {
- if let Some(expr) = expr_stmt.expr() {
- scopes.set_scope(expr.syntax(), scope);
- compute_expr_scopes(expr, scopes, scope);
- }
- }
- }
- }
- if let Some(expr) = block.expr() {
- scopes.set_scope(expr.syntax(), scope);
- compute_expr_scopes(expr, scopes, scope);
- }
-}
-
-fn compute_expr_scopes(expr: ast::Expr, scopes: &mut FnScopes, scope: ScopeId) {
- match expr {
- ast::Expr::IfExpr(e) => {
- let cond_scope = e
- .condition()
- .and_then(|cond| compute_cond_scopes(cond, scopes, scope));
- if let Some(block) = e.then_branch() {
- compute_block_scopes(block, scopes, cond_scope.unwrap_or(scope));
- }
- if let Some(block) = e.else_branch() {
- compute_block_scopes(block, scopes, scope);
- }
- }
- ast::Expr::BlockExpr(e) => {
- if let Some(block) = e.block() {
- compute_block_scopes(block, scopes, scope);
- }
- }
- ast::Expr::LoopExpr(e) => {
- if let Some(block) = e.loop_body() {
- compute_block_scopes(block, scopes, scope);
- }
- }
- ast::Expr::WhileExpr(e) => {
- let cond_scope = e
- .condition()
- .and_then(|cond| compute_cond_scopes(cond, scopes, scope));
- if let Some(block) = e.loop_body() {
- compute_block_scopes(block, scopes, cond_scope.unwrap_or(scope));
- }
- }
- ast::Expr::ForExpr(e) => {
- if let Some(expr) = e.iterable() {
- compute_expr_scopes(expr, scopes, scope);
- }
- let mut scope = scope;
- if let Some(pat) = e.pat() {
- scope = scopes.new_scope(scope);
- scopes.add_bindings(scope, pat);
- }
- if let Some(block) = e.loop_body() {
- compute_block_scopes(block, scopes, scope);
- }
- }
- ast::Expr::LambdaExpr(e) => {
- let scope = scopes.new_scope(scope);
- scopes.add_params_bindings(scope, e.param_list());
- if let Some(body) = e.body() {
- scopes.set_scope(body.syntax(), scope);
- compute_expr_scopes(body, scopes, scope);
- }
- }
- ast::Expr::CallExpr(e) => {
- compute_call_scopes(e.expr(), e.arg_list(), scopes, scope);
- }
- ast::Expr::MethodCallExpr(e) => {
- compute_call_scopes(e.expr(), e.arg_list(), scopes, scope);
- }
- ast::Expr::MatchExpr(e) => {
- if let Some(expr) = e.expr() {
- compute_expr_scopes(expr, scopes, scope);
- }
- for arm in e.match_arm_list().into_iter().flat_map(|it| it.arms()) {
- let scope = scopes.new_scope(scope);
- for pat in arm.pats() {
- scopes.add_bindings(scope, pat);
- }
- if let Some(expr) = arm.expr() {
- compute_expr_scopes(expr, scopes, scope);
- }
- }
- }
- _ => expr
- .syntax()
- .children()
- .filter_map(ast::Expr::cast)
- .for_each(|expr| compute_expr_scopes(expr, scopes, scope)),
- };
-
- fn compute_call_scopes(
- receiver: Option<ast::Expr>,
- arg_list: Option<ast::ArgList>,
- scopes: &mut FnScopes,
- scope: ScopeId,
- ) {
- arg_list
- .into_iter()
- .flat_map(|it| it.args())
- .chain(receiver)
- .for_each(|expr| compute_expr_scopes(expr, scopes, scope));
- }
-
- fn compute_cond_scopes(
- cond: ast::Condition,
- scopes: &mut FnScopes,
- scope: ScopeId,
- ) -> Option<ScopeId> {
- if let Some(expr) = cond.expr() {
- compute_expr_scopes(expr, scopes, scope);
- }
- if let Some(pat) = cond.pat() {
- let s = scopes.new_scope(scope);
- scopes.add_bindings(s, pat);
- Some(s)
- } else {
- None
- }
- }
-}
-
-#[derive(Debug)]
-pub struct ReferenceDescriptor {
- pub range: TextRange,
- pub name: String,
-}
-
-#[cfg(test)]
-mod tests {
- use ra_editor::find_node_at_offset;
- use ra_syntax::SourceFileNode;
- use test_utils::extract_offset;
-
- use super::*;
-
- fn do_check(code: &str, expected: &[&str]) {
- let (off, code) = extract_offset(code);
- let code = {
- let mut buf = String::new();
- let off = u32::from(off) as usize;
- buf.push_str(&code[..off]);
- buf.push_str("marker");
- buf.push_str(&code[off..]);
- buf
- };
- let file = SourceFileNode::parse(&code);
- let marker: ast::PathExpr = find_node_at_offset(file.syntax(), off).unwrap();
- let fn_def: ast::FnDef = find_node_at_offset(file.syntax(), off).unwrap();
- let scopes = FnScopes::new(fn_def);
- let actual = scopes
- .scope_chain(marker.syntax())
- .flat_map(|scope| scopes.entries(scope))
- .map(|it| it.name())
- .collect::<Vec<_>>();
- assert_eq!(actual.as_slice(), expected);
- }
-
- #[test]
- fn test_lambda_scope() {
- do_check(
- r"
- fn quux(foo: i32) {
- let f = |bar, baz: i32| {
- <|>
- };
- }",
- &["bar", "baz", "foo"],
- );
- }
-
- #[test]
- fn test_call_scope() {
- do_check(
- r"
- fn quux() {
- f(|x| <|> );
- }",
- &["x"],
- );
- }
-
- #[test]
- fn test_metod_call_scope() {
- do_check(
- r"
- fn quux() {
- z.f(|x| <|> );
- }",
- &["x"],
- );
- }
-
- #[test]
- fn test_loop_scope() {
- do_check(
- r"
- fn quux() {
- loop {
- let x = ();
- <|>
- };
- }",
- &["x"],
- );
- }
-
- #[test]
- fn test_match() {
- do_check(
- r"
- fn quux() {
- match () {
- Some(x) => {
- <|>
- }
- };
- }",
- &["x"],
- );
- }
-
- #[test]
- fn test_shadow_variable() {
- do_check(
- r"
- fn foo(x: String) {
- let x : &str = &x<|>;
- }",
- &["x"],
- );
- }
-
- fn do_check_local_name(code: &str, expected_offset: u32) {
- let (off, code) = extract_offset(code);
- let file = SourceFileNode::parse(&code);
- let fn_def: ast::FnDef = find_node_at_offset(file.syntax(), off).unwrap();
- let name_ref: ast::NameRef = find_node_at_offset(file.syntax(), off).unwrap();
-
- let scopes = FnScopes::new(fn_def);
-
- let local_name_entry = scopes.resolve_local_name(name_ref).unwrap();
- let local_name = local_name_entry.ptr().resolve(&file);
- let expected_name =
- find_node_at_offset::<ast::Name>(file.syntax(), expected_offset.into()).unwrap();
- assert_eq!(local_name.range(), expected_name.syntax().range());
- }
-
- #[test]
- fn test_resolve_local_name() {
- do_check_local_name(
- r#"
- fn foo(x: i32, y: u32) {
- {
- let z = x * 2;
- }
- {
- let t = x<|> * 3;
- }
- }"#,
- 21,
- );
- }
-
- #[test]
- fn test_resolve_local_name_declaration() {
- do_check_local_name(
- r#"
- fn foo(x: String) {
- let x : &str = &x<|>;
- }"#,
- 21,
- );
- }
-
- #[test]
- fn test_resolve_local_name_shadow() {
- do_check_local_name(
- r"
- fn foo(x: String) {
- let x : &str = &x;
- x<|>
- }",
- 46,
- );
- }
-}
+++ /dev/null
-//! HIR (previsouly known as descriptors) provides a high-level OO acess to Rust
-//! code.
-//!
-//! The principal difference between HIR and syntax trees is that HIR is bound
-//! to a particular crate instance. That is, it has cfg flags and features
-//! applied. So, there relation between syntax and HIR is many-to-one.
-
-pub(crate) mod db;
-mod query_definitions;
-mod function;
-mod module;
-mod path;
-
-use std::ops::Index;
-
-use ra_syntax::{SyntaxNodeRef, SyntaxNode};
-use ra_db::{LocationIntener, SourceRootId};
-
-use crate::{
- FileId,
- hir::db::HirDatabase,
- Cancelable,
- arena::{Arena, Id},
-};
-
-pub(crate) use self::{
- path::{Path, PathKind},
- module::{Module, ModuleId, Problem},
- function::{Function, FnScopes},
-};
-
-pub use self::function::FnSignatureInfo;
-
-#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
-pub(crate) struct FnId(u32);
-ra_db::impl_numeric_id!(FnId);
-
-impl FnId {
- pub(crate) fn from_loc(
- db: &impl AsRef<LocationIntener<SourceItemId, FnId>>,
- loc: &SourceItemId,
- ) -> FnId {
- db.as_ref().loc2id(loc)
- }
- pub(crate) fn loc(self, db: &impl AsRef<LocationIntener<SourceItemId, FnId>>) -> SourceItemId {
- db.as_ref().id2loc(self)
- }
-}
-
-#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
-pub(crate) struct DefId(u32);
-ra_db::impl_numeric_id!(DefId);
-
-#[derive(Clone, Debug, PartialEq, Eq, Hash)]
-pub(crate) enum DefLoc {
- Module {
- id: ModuleId,
- source_root: SourceRootId,
- },
- Item {
- source_item_id: SourceItemId,
- },
-}
-
-impl DefId {
- pub(crate) fn loc(self, db: &impl AsRef<LocationIntener<DefLoc, DefId>>) -> DefLoc {
- db.as_ref().id2loc(self)
- }
-}
-
-impl DefLoc {
- pub(crate) fn id(&self, db: &impl AsRef<LocationIntener<DefLoc, DefId>>) -> DefId {
- db.as_ref().loc2id(&self)
- }
-}
-
-pub(crate) enum Def {
- Module(Module),
- Item,
-}
-
-impl DefId {
- pub(crate) fn resolve(self, db: &impl HirDatabase) -> Cancelable<Def> {
- let loc = self.loc(db);
- let res = match loc {
- DefLoc::Module { id, source_root } => {
- let descr = Module::new(db, source_root, id)?;
- Def::Module(descr)
- }
- DefLoc::Item { .. } => Def::Item,
- };
- Ok(res)
- }
-}
-
-/// Identifier of item within a specific file. This is stable over reparses, so
-/// it's OK to use it as a salsa key/value.
-pub(crate) type SourceFileItemId = Id<SyntaxNode>;
-
-#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
-pub(crate) struct SourceItemId {
- file_id: FileId,
- item_id: SourceFileItemId,
-}
-
-/// Maps item's `SyntaxNode`s to `SourceFileItemId` and back.
-#[derive(Debug, PartialEq, Eq, Default)]
-pub(crate) struct SourceFileItems {
- arena: Arena<SyntaxNode>,
-}
-
-impl SourceFileItems {
- fn alloc(&mut self, item: SyntaxNode) -> SourceFileItemId {
- self.arena.alloc(item)
- }
- fn id_of(&self, item: SyntaxNodeRef) -> SourceFileItemId {
- let (id, _item) = self
- .arena
- .iter()
- .find(|(_id, i)| i.borrowed() == item)
- .unwrap();
- id
- }
-}
-
-impl Index<SourceFileItemId> for SourceFileItems {
- type Output = SyntaxNode;
- fn index(&self, idx: SourceFileItemId) -> &SyntaxNode {
- &self.arena[idx]
- }
-}
+++ /dev/null
-use std::sync::Arc;
-
-use ra_syntax::{
- ast::{self, NameOwner},
- SmolStr,
-};
-use relative_path::RelativePathBuf;
-use rustc_hash::{FxHashMap, FxHashSet};
-use ra_db::{SourceRoot, SourceRootId, FileResolverImp};
-
-use crate::{
- hir::HirDatabase,
- Cancelable, FileId,
-};
-
-use super::{
- LinkData, LinkId, ModuleData, ModuleId, ModuleSource,
- ModuleTree, Problem,
-};
-
-#[derive(Clone, Hash, PartialEq, Eq, Debug)]
-pub(crate) enum Submodule {
- Declaration(SmolStr),
- Definition(SmolStr, ModuleSource),
-}
-
-impl Submodule {
- fn name(&self) -> &SmolStr {
- match self {
- Submodule::Declaration(name) => name,
- Submodule::Definition(name, _) => name,
- }
- }
-}
-
-pub(crate) fn modules<'a>(
- root: impl ast::ModuleItemOwner<'a>,
-) -> impl Iterator<Item = (SmolStr, ast::Module<'a>)> {
- root.items()
- .filter_map(|item| match item {
- ast::ModuleItem::Module(m) => Some(m),
- _ => None,
- })
- .filter_map(|module| {
- let name = module.name()?.text();
- Some((name, module))
- })
-}
-
-pub(crate) fn module_tree(
- db: &impl HirDatabase,
- source_root: SourceRootId,
-) -> Cancelable<Arc<ModuleTree>> {
- db.check_canceled()?;
- let res = create_module_tree(db, source_root)?;
- Ok(Arc::new(res))
-}
-
-fn create_module_tree<'a>(
- db: &impl HirDatabase,
- source_root: SourceRootId,
-) -> Cancelable<ModuleTree> {
- let mut tree = ModuleTree::default();
-
- let mut roots = FxHashMap::default();
- let mut visited = FxHashSet::default();
-
- let source_root = db.source_root(source_root);
- for &file_id in source_root.files.iter() {
- let source = ModuleSource::SourceFile(file_id);
- if visited.contains(&source) {
- continue; // TODO: use explicit crate_roots here
- }
- assert!(!roots.contains_key(&file_id));
- let module_id = build_subtree(
- db,
- &source_root,
- &mut tree,
- &mut visited,
- &mut roots,
- None,
- source,
- )?;
- roots.insert(file_id, module_id);
- }
- Ok(tree)
-}
-
-fn build_subtree(
- db: &impl HirDatabase,
- source_root: &SourceRoot,
- tree: &mut ModuleTree,
- visited: &mut FxHashSet<ModuleSource>,
- roots: &mut FxHashMap<FileId, ModuleId>,
- parent: Option<LinkId>,
- source: ModuleSource,
-) -> Cancelable<ModuleId> {
- visited.insert(source);
- let id = tree.push_mod(ModuleData {
- source,
- parent,
- children: Vec::new(),
- });
- for sub in db.submodules(source)?.iter() {
- let link = tree.push_link(LinkData {
- name: sub.name().clone(),
- owner: id,
- points_to: Vec::new(),
- problem: None,
- });
-
- let (points_to, problem) = match sub {
- Submodule::Declaration(name) => {
- let (points_to, problem) =
- resolve_submodule(source, &name, &source_root.file_resolver);
- let points_to = points_to
- .into_iter()
- .map(|file_id| match roots.remove(&file_id) {
- Some(module_id) => {
- tree.mods[module_id].parent = Some(link);
- Ok(module_id)
- }
- None => build_subtree(
- db,
- source_root,
- tree,
- visited,
- roots,
- Some(link),
- ModuleSource::SourceFile(file_id),
- ),
- })
- .collect::<Cancelable<Vec<_>>>()?;
- (points_to, problem)
- }
- Submodule::Definition(_name, submodule_source) => {
- let points_to = build_subtree(
- db,
- source_root,
- tree,
- visited,
- roots,
- Some(link),
- *submodule_source,
- )?;
- (vec![points_to], None)
- }
- };
-
- tree.links[link].points_to = points_to;
- tree.links[link].problem = problem;
- }
- Ok(id)
-}
-
-fn resolve_submodule(
- source: ModuleSource,
- name: &SmolStr,
- file_resolver: &FileResolverImp,
-) -> (Vec<FileId>, Option<Problem>) {
- let file_id = match source {
- ModuleSource::SourceFile(it) => it,
- ModuleSource::Module(..) => {
- // TODO
- return (Vec::new(), None);
- }
- };
- let mod_name = file_resolver.file_stem(file_id);
- let is_dir_owner = mod_name == "mod" || mod_name == "lib" || mod_name == "main";
-
- let file_mod = RelativePathBuf::from(format!("../{}.rs", name));
- let dir_mod = RelativePathBuf::from(format!("../{}/mod.rs", name));
- let points_to: Vec<FileId>;
- let problem: Option<Problem>;
- if is_dir_owner {
- points_to = [&file_mod, &dir_mod]
- .iter()
- .filter_map(|path| file_resolver.resolve(file_id, path))
- .collect();
- problem = if points_to.is_empty() {
- Some(Problem::UnresolvedModule {
- candidate: file_mod,
- })
- } else {
- None
- }
- } else {
- points_to = Vec::new();
- problem = Some(Problem::NotDirOwner {
- move_to: RelativePathBuf::from(format!("../{}/mod.rs", mod_name)),
- candidate: file_mod,
- });
- }
- (points_to, problem)
-}
+++ /dev/null
-pub(super) mod imp;
-pub(super) mod nameres;
-
-use std::sync::Arc;
-
-use ra_editor::find_node_at_offset;
-
-use ra_syntax::{
- algo::generate,
- ast::{self, AstNode, NameOwner},
- SmolStr, SyntaxNode,
-};
-use ra_db::SourceRootId;
-use relative_path::RelativePathBuf;
-
-use crate::{
- FileId, FilePosition, Cancelable,
- hir::{DefLoc, DefId, Path, PathKind, HirDatabase, SourceItemId},
- arena::{Arena, Id},
-};
-
-pub(crate) use self::nameres::ModuleScope;
-
-/// `Module` is API entry point to get all the information
-/// about a particular module.
-#[derive(Debug, Clone)]
-pub(crate) struct Module {
- tree: Arc<ModuleTree>,
- source_root_id: SourceRootId,
- module_id: ModuleId,
-}
-
-impl Module {
- /// Lookup `Module` by `FileId`. Note that this is inherently
- /// lossy transformation: in general, a single source might correspond to
- /// several modules.
- pub fn guess_from_file_id(
- db: &impl HirDatabase,
- file_id: FileId,
- ) -> Cancelable<Option<Module>> {
- Module::guess_from_source(db, file_id, ModuleSource::SourceFile(file_id))
- }
-
- /// Lookup `Module` by position in the source code. Note that this
- /// is inherently lossy transformation: in general, a single source might
- /// correspond to several modules.
- pub fn guess_from_position(
- db: &impl HirDatabase,
- position: FilePosition,
- ) -> Cancelable<Option<Module>> {
- let file = db.source_file(position.file_id);
- let module_source = match find_node_at_offset::<ast::Module>(file.syntax(), position.offset)
- {
- Some(m) if !m.has_semi() => ModuleSource::new_inline(db, position.file_id, m),
- _ => ModuleSource::SourceFile(position.file_id),
- };
- Module::guess_from_source(db, position.file_id, module_source)
- }
-
- fn guess_from_source(
- db: &impl HirDatabase,
- file_id: FileId,
- module_source: ModuleSource,
- ) -> Cancelable<Option<Module>> {
- let source_root_id = db.file_source_root(file_id);
- let module_tree = db.module_tree(source_root_id)?;
-
- let res = match module_tree.any_module_for_source(module_source) {
- None => None,
- Some(module_id) => Some(Module {
- tree: module_tree,
- source_root_id,
- module_id,
- }),
- };
- Ok(res)
- }
-
- pub(super) fn new(
- db: &impl HirDatabase,
- source_root_id: SourceRootId,
- module_id: ModuleId,
- ) -> Cancelable<Module> {
- let module_tree = db.module_tree(source_root_id)?;
- let res = Module {
- tree: module_tree,
- source_root_id,
- module_id,
- };
- Ok(res)
- }
-
- /// Returns `mod foo;` or `mod foo {}` node whihc declared this module.
- /// Returns `None` for the root module
- pub fn parent_link_source(&self, db: &impl HirDatabase) -> Option<(FileId, ast::ModuleNode)> {
- let link = self.module_id.parent_link(&self.tree)?;
- let file_id = link.owner(&self.tree).source(&self.tree).file_id();
- let src = link.bind_source(&self.tree, db);
- Some((file_id, src))
- }
-
- pub fn source(&self) -> ModuleSource {
- self.module_id.source(&self.tree)
- }
-
- /// Parent module. Returns `None` if this is a root module.
- pub fn parent(&self) -> Option<Module> {
- let parent_id = self.module_id.parent(&self.tree)?;
- Some(Module {
- module_id: parent_id,
- ..self.clone()
- })
- }
-
- /// The root of the tree this module is part of
- pub fn crate_root(&self) -> Module {
- let root_id = self.module_id.crate_root(&self.tree);
- Module {
- module_id: root_id,
- ..self.clone()
- }
- }
-
- /// `name` is `None` for the crate's root module
- #[allow(unused)]
- pub fn name(&self) -> Option<SmolStr> {
- let link = self.module_id.parent_link(&self.tree)?;
- Some(link.name(&self.tree))
- }
-
- pub fn def_id(&self, db: &impl HirDatabase) -> DefId {
- let def_loc = DefLoc::Module {
- id: self.module_id,
- source_root: self.source_root_id,
- };
- def_loc.id(db)
- }
-
- /// Finds a child module with the specified name.
- pub fn child(&self, name: &str) -> Option<Module> {
- let child_id = self.module_id.child(&self.tree, name)?;
- Some(Module {
- module_id: child_id,
- ..self.clone()
- })
- }
-
- /// Returns a `ModuleScope`: a set of items, visible in this module.
- pub(crate) fn scope(&self, db: &impl HirDatabase) -> Cancelable<ModuleScope> {
- let item_map = db.item_map(self.source_root_id)?;
- let res = item_map.per_module[&self.module_id].clone();
- Ok(res)
- }
-
- pub(crate) fn resolve_path(
- &self,
- db: &impl HirDatabase,
- path: Path,
- ) -> Cancelable<Option<DefId>> {
- let mut curr = match path.kind {
- PathKind::Crate => self.crate_root(),
- PathKind::Self_ | PathKind::Plain => self.clone(),
- PathKind::Super => ctry!(self.parent()),
- }
- .def_id(db);
-
- let segments = path.segments;
- for name in segments.iter() {
- let module = match curr.loc(db) {
- DefLoc::Module { id, source_root } => Module::new(db, source_root, id)?,
- _ => return Ok(None),
- };
- let scope = module.scope(db)?;
- curr = ctry!(ctry!(scope.get(&name)).def_id);
- }
- Ok(Some(curr))
- }
-
- pub fn problems(&self, db: &impl HirDatabase) -> Vec<(SyntaxNode, Problem)> {
- self.module_id.problems(&self.tree, db)
- }
-}
-
-/// Phisically, rust source is organized as a set of files, but logically it is
-/// organized as a tree of modules. Usually, a single file corresponds to a
-/// single module, but it is not nessary the case.
-///
-/// Module encapsulate the logic of transitioning from the fuzzy world of files
-/// (which can have multiple parents) to the precise world of modules (which
-/// always have one parent).
-#[derive(Default, Debug, PartialEq, Eq)]
-pub(crate) struct ModuleTree {
- mods: Arena<ModuleData>,
- links: Arena<LinkData>,
-}
-
-impl ModuleTree {
- pub(in crate::hir) fn modules<'a>(&'a self) -> impl Iterator<Item = ModuleId> + 'a {
- self.mods.iter().map(|(id, _)| id)
- }
-
- fn modules_for_source(&self, source: ModuleSource) -> Vec<ModuleId> {
- self.mods
- .iter()
- .filter(|(_idx, it)| it.source == source)
- .map(|(idx, _)| idx)
- .collect()
- }
-
- fn any_module_for_source(&self, source: ModuleSource) -> Option<ModuleId> {
- self.modules_for_source(source).pop()
- }
-}
-
-/// `ModuleSource` is the syntax tree element that produced this module:
-/// either a file, or an inlinde module.
-#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
-pub(crate) enum ModuleSource {
- SourceFile(FileId),
- Module(SourceItemId),
-}
-
-/// An owned syntax node for a module. Unlike `ModuleSource`,
-/// this holds onto the AST for the whole file.
-pub(crate) enum ModuleSourceNode {
- SourceFile(ast::SourceFileNode),
- Module(ast::ModuleNode),
-}
-
-pub(crate) type ModuleId = Id<ModuleData>;
-type LinkId = Id<LinkData>;
-
-#[derive(Clone, Debug, Hash, PartialEq, Eq)]
-pub enum Problem {
- UnresolvedModule {
- candidate: RelativePathBuf,
- },
- NotDirOwner {
- move_to: RelativePathBuf,
- candidate: RelativePathBuf,
- },
-}
-
-impl ModuleId {
- pub(in crate::hir) fn source(self, tree: &ModuleTree) -> ModuleSource {
- tree.mods[self].source
- }
- fn parent_link(self, tree: &ModuleTree) -> Option<LinkId> {
- tree.mods[self].parent
- }
- fn parent(self, tree: &ModuleTree) -> Option<ModuleId> {
- let link = self.parent_link(tree)?;
- Some(tree.links[link].owner)
- }
- fn crate_root(self, tree: &ModuleTree) -> ModuleId {
- generate(Some(self), move |it| it.parent(tree))
- .last()
- .unwrap()
- }
- fn child(self, tree: &ModuleTree, name: &str) -> Option<ModuleId> {
- let link = tree.mods[self]
- .children
- .iter()
- .map(|&it| &tree.links[it])
- .find(|it| it.name == name)?;
- Some(*link.points_to.first()?)
- }
- fn children<'a>(self, tree: &'a ModuleTree) -> impl Iterator<Item = (SmolStr, ModuleId)> + 'a {
- tree.mods[self].children.iter().filter_map(move |&it| {
- let link = &tree.links[it];
- let module = *link.points_to.first()?;
- Some((link.name.clone(), module))
- })
- }
- fn problems(self, tree: &ModuleTree, db: &impl HirDatabase) -> Vec<(SyntaxNode, Problem)> {
- tree.mods[self]
- .children
- .iter()
- .filter_map(|&it| {
- let p = tree.links[it].problem.clone()?;
- let s = it.bind_source(tree, db);
- let s = s.borrowed().name().unwrap().syntax().owned();
- Some((s, p))
- })
- .collect()
- }
-}
-
-impl LinkId {
- fn owner(self, tree: &ModuleTree) -> ModuleId {
- tree.links[self].owner
- }
- fn name(self, tree: &ModuleTree) -> SmolStr {
- tree.links[self].name.clone()
- }
- fn bind_source<'a>(self, tree: &ModuleTree, db: &impl HirDatabase) -> ast::ModuleNode {
- let owner = self.owner(tree);
- match owner.source(tree).resolve(db) {
- ModuleSourceNode::SourceFile(root) => {
- let ast = imp::modules(root.borrowed())
- .find(|(name, _)| name == &tree.links[self].name)
- .unwrap()
- .1;
- ast.owned()
- }
- ModuleSourceNode::Module(it) => it,
- }
- }
-}
-
-#[derive(Debug, PartialEq, Eq, Hash)]
-pub(crate) struct ModuleData {
- source: ModuleSource,
- parent: Option<LinkId>,
- children: Vec<LinkId>,
-}
-
-impl ModuleSource {
- pub(crate) fn new_inline(
- db: &impl HirDatabase,
- file_id: FileId,
- module: ast::Module,
- ) -> ModuleSource {
- assert!(!module.has_semi());
- let items = db.file_items(file_id);
- let item_id = items.id_of(module.syntax());
- let id = SourceItemId { file_id, item_id };
- ModuleSource::Module(id)
- }
-
- pub(crate) fn as_file(self) -> Option<FileId> {
- match self {
- ModuleSource::SourceFile(f) => Some(f),
- ModuleSource::Module(..) => None,
- }
- }
-
- pub(crate) fn file_id(self) -> FileId {
- match self {
- ModuleSource::SourceFile(f) => f,
- ModuleSource::Module(source_item_id) => source_item_id.file_id,
- }
- }
-
- pub(crate) fn resolve(self, db: &impl HirDatabase) -> ModuleSourceNode {
- match self {
- ModuleSource::SourceFile(file_id) => {
- let syntax = db.source_file(file_id);
- ModuleSourceNode::SourceFile(syntax.ast().owned())
- }
- ModuleSource::Module(item_id) => {
- let syntax = db.file_item(item_id);
- let syntax = syntax.borrowed();
- let module = ast::Module::cast(syntax).unwrap();
- ModuleSourceNode::Module(module.owned())
- }
- }
- }
-}
-
-#[derive(Hash, Debug, PartialEq, Eq)]
-struct LinkData {
- owner: ModuleId,
- name: SmolStr,
- points_to: Vec<ModuleId>,
- problem: Option<Problem>,
-}
-
-impl ModuleTree {
- fn push_mod(&mut self, data: ModuleData) -> ModuleId {
- self.mods.alloc(data)
- }
- fn push_link(&mut self, data: LinkData) -> LinkId {
- let owner = data.owner;
- let id = self.links.alloc(data);
- self.mods[owner].children.push(id);
- id
- }
-}
+++ /dev/null
-//! Name resolution algorithm. The end result of the algorithm is `ItemMap`: a
-//! map with maps each module to it's scope: the set of items, visible in the
-//! module. That is, we only resolve imports here, name resolution of item
-//! bodies will be done in a separate step.
-//!
-//! Like Rustc, we use an interative per-crate algorithm: we start with scopes
-//! containing only directly defined items, and then iteratively resolve
-//! imports.
-//!
-//! To make this work nicely in the IDE scenarios, we place `InputModuleItems`
-//! in between raw syntax and name resolution. `InputModuleItems` are computed
-//! using only the module's syntax, and it is all directly defined items plus
-//! imports. The plain is to make `InputModuleItems` independent of local
-//! modifications (that is, typing inside a function shold not change IMIs),
-//! such that the results of name resolution can be preserved unless the module
-//! structure itself is modified.
-use std::{
- sync::Arc,
-};
-
-use rustc_hash::FxHashMap;
-use ra_syntax::{
- TextRange,
- SmolStr, SyntaxKind::{self, *},
- ast::{self, AstNode}
-};
-use ra_db::SourceRootId;
-
-use crate::{
- Cancelable, FileId,
- hir::{
- DefId, DefLoc,
- SourceItemId, SourceFileItemId, SourceFileItems,
- Path, PathKind,
- HirDatabase,
- module::{ModuleId, ModuleTree},
- },
-};
-
-/// Item map is the result of the name resolution. Item map contains, for each
-/// module, the set of visible items.
-#[derive(Default, Debug, PartialEq, Eq)]
-pub(crate) struct ItemMap {
- pub(crate) per_module: FxHashMap<ModuleId, ModuleScope>,
-}
-
-#[derive(Debug, Default, PartialEq, Eq, Clone)]
-pub(crate) struct ModuleScope {
- items: FxHashMap<SmolStr, Resolution>,
-}
-
-impl ModuleScope {
- pub(crate) fn entries<'a>(&'a self) -> impl Iterator<Item = (&'a SmolStr, &Resolution)> + 'a {
- self.items.iter()
- }
- pub(crate) fn get(&self, name: &SmolStr) -> Option<&Resolution> {
- self.items.get(name)
- }
-}
-
-/// A set of items and imports declared inside a module, without relation to
-/// other modules.
-///
-/// This stands in-between raw syntax and name resolution and alow us to avoid
-/// recomputing name res: if `InputModuleItems` are the same, we can avoid
-/// running name resolution.
-#[derive(Debug, Default, PartialEq, Eq)]
-pub(crate) struct InputModuleItems {
- items: Vec<ModuleItem>,
- imports: Vec<Import>,
-}
-
-#[derive(Debug, PartialEq, Eq)]
-struct ModuleItem {
- id: SourceFileItemId,
- name: SmolStr,
- kind: SyntaxKind,
- vis: Vis,
-}
-
-#[derive(Debug, PartialEq, Eq)]
-enum Vis {
- // Priv,
- Other,
-}
-
-#[derive(Debug, Clone, PartialEq, Eq)]
-struct Import {
- path: Path,
- kind: ImportKind,
-}
-
-#[derive(Debug, Clone, Copy, PartialEq, Eq)]
-pub(crate) struct NamedImport {
- file_item_id: SourceFileItemId,
- relative_range: TextRange,
-}
-
-impl NamedImport {
- pub(crate) fn range(&self, db: &impl HirDatabase, file_id: FileId) -> TextRange {
- let source_item_id = SourceItemId {
- file_id,
- item_id: self.file_item_id,
- };
- let syntax = db.file_item(source_item_id);
- let offset = syntax.borrowed().range().start();
- self.relative_range + offset
- }
-}
-
-#[derive(Debug, Clone, PartialEq, Eq)]
-enum ImportKind {
- Glob,
- Named(NamedImport),
-}
-
-/// Resolution is basically `DefId` atm, but it should account for stuff like
-/// multiple namespaces, ambiguity and errors.
-#[derive(Debug, Clone, PartialEq, Eq)]
-pub(crate) struct Resolution {
- /// None for unresolved
- pub(crate) def_id: Option<DefId>,
- /// ident by whitch this is imported into local scope.
- pub(crate) import: Option<NamedImport>,
-}
-
-// #[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
-// enum Namespace {
-// Types,
-// Values,
-// }
-
-// #[derive(Debug)]
-// struct PerNs<T> {
-// types: Option<T>,
-// values: Option<T>,
-// }
-
-impl InputModuleItems {
- pub(in crate::hir) fn new<'a>(
- file_items: &SourceFileItems,
- items: impl Iterator<Item = ast::ModuleItem<'a>>,
- ) -> InputModuleItems {
- let mut res = InputModuleItems::default();
- for item in items {
- res.add_item(file_items, item);
- }
- res
- }
-
- fn add_item(&mut self, file_items: &SourceFileItems, item: ast::ModuleItem) -> Option<()> {
- match item {
- ast::ModuleItem::StructDef(it) => self.items.push(ModuleItem::new(file_items, it)?),
- ast::ModuleItem::EnumDef(it) => self.items.push(ModuleItem::new(file_items, it)?),
- ast::ModuleItem::FnDef(it) => self.items.push(ModuleItem::new(file_items, it)?),
- ast::ModuleItem::TraitDef(it) => self.items.push(ModuleItem::new(file_items, it)?),
- ast::ModuleItem::TypeDef(it) => self.items.push(ModuleItem::new(file_items, it)?),
- ast::ModuleItem::ImplItem(_) => {
- // impls don't define items
- }
- ast::ModuleItem::UseItem(it) => self.add_use_item(file_items, it),
- ast::ModuleItem::ExternCrateItem(_) => {
- // TODO
- }
- ast::ModuleItem::ConstDef(it) => self.items.push(ModuleItem::new(file_items, it)?),
- ast::ModuleItem::StaticDef(it) => self.items.push(ModuleItem::new(file_items, it)?),
- ast::ModuleItem::Module(it) => self.items.push(ModuleItem::new(file_items, it)?),
- }
- Some(())
- }
-
- fn add_use_item(&mut self, file_items: &SourceFileItems, item: ast::UseItem) {
- let file_item_id = file_items.id_of(item.syntax());
- let start_offset = item.syntax().range().start();
- Path::expand_use_item(item, |path, range| {
- let kind = match range {
- None => ImportKind::Glob,
- Some(range) => ImportKind::Named(NamedImport {
- file_item_id,
- relative_range: range - start_offset,
- }),
- };
- self.imports.push(Import { kind, path })
- })
- }
-}
-
-impl ModuleItem {
- fn new<'a>(file_items: &SourceFileItems, item: impl ast::NameOwner<'a>) -> Option<ModuleItem> {
- let name = item.name()?.text();
- let kind = item.syntax().kind();
- let vis = Vis::Other;
- let id = file_items.id_of(item.syntax());
- let res = ModuleItem {
- id,
- name,
- kind,
- vis,
- };
- Some(res)
- }
-}
-
-pub(in crate::hir) struct Resolver<'a, DB> {
- pub db: &'a DB,
- pub input: &'a FxHashMap<ModuleId, Arc<InputModuleItems>>,
- pub source_root: SourceRootId,
- pub module_tree: Arc<ModuleTree>,
- pub result: ItemMap,
-}
-
-impl<'a, DB> Resolver<'a, DB>
-where
- DB: HirDatabase,
-{
- pub(in crate::hir) fn resolve(mut self) -> Cancelable<ItemMap> {
- for (&module_id, items) in self.input.iter() {
- self.populate_module(module_id, items)
- }
-
- for &module_id in self.input.keys() {
- self.db.check_canceled()?;
- self.resolve_imports(module_id);
- }
- Ok(self.result)
- }
-
- fn populate_module(&mut self, module_id: ModuleId, input: &InputModuleItems) {
- let file_id = module_id.source(&self.module_tree).file_id();
-
- let mut module_items = ModuleScope::default();
-
- for import in input.imports.iter() {
- if let Some(name) = import.path.segments.iter().last() {
- if let ImportKind::Named(import) = import.kind {
- module_items.items.insert(
- name.clone(),
- Resolution {
- def_id: None,
- import: Some(import),
- },
- );
- }
- }
- }
-
- for item in input.items.iter() {
- if item.kind == MODULE {
- // handle submodules separatelly
- continue;
- }
- let def_loc = DefLoc::Item {
- source_item_id: SourceItemId {
- file_id,
- item_id: item.id,
- },
- };
- let def_id = def_loc.id(self.db);
- let resolution = Resolution {
- def_id: Some(def_id),
- import: None,
- };
- module_items.items.insert(item.name.clone(), resolution);
- }
-
- for (name, mod_id) in module_id.children(&self.module_tree) {
- let def_loc = DefLoc::Module {
- id: mod_id,
- source_root: self.source_root,
- };
- let def_id = def_loc.id(self.db);
- let resolution = Resolution {
- def_id: Some(def_id),
- import: None,
- };
- module_items.items.insert(name, resolution);
- }
-
- self.result.per_module.insert(module_id, module_items);
- }
-
- fn resolve_imports(&mut self, module_id: ModuleId) {
- for import in self.input[&module_id].imports.iter() {
- self.resolve_import(module_id, import);
- }
- }
-
- fn resolve_import(&mut self, module_id: ModuleId, import: &Import) {
- let ptr = match import.kind {
- ImportKind::Glob => return,
- ImportKind::Named(ptr) => ptr,
- };
-
- let mut curr = match import.path.kind {
- // TODO: handle extern crates
- PathKind::Plain => return,
- PathKind::Self_ => module_id,
- PathKind::Super => {
- match module_id.parent(&self.module_tree) {
- Some(it) => it,
- // TODO: error
- None => return,
- }
- }
- PathKind::Crate => module_id.crate_root(&self.module_tree),
- };
-
- for (i, name) in import.path.segments.iter().enumerate() {
- let is_last = i == import.path.segments.len() - 1;
-
- let def_id = match self.result.per_module[&curr].items.get(name) {
- None => return,
- Some(res) => match res.def_id {
- Some(it) => it,
- None => return,
- },
- };
-
- if !is_last {
- curr = match def_id.loc(self.db) {
- DefLoc::Module { id, .. } => id,
- _ => return,
- }
- } else {
- self.update(module_id, |items| {
- let res = Resolution {
- def_id: Some(def_id),
- import: Some(ptr),
- };
- items.items.insert(name.clone(), res);
- })
- }
- }
- }
-
- fn update(&mut self, module_id: ModuleId, f: impl FnOnce(&mut ModuleScope)) {
- let module_items = self.result.per_module.get_mut(&module_id).unwrap();
- f(module_items)
- }
-}
-
-#[cfg(test)]
-mod tests {
- use ra_db::FilesDatabase;
- use crate::{
- AnalysisChange,
- mock_analysis::{MockAnalysis, analysis_and_position},
- hir::{self, HirDatabase},
-};
- use super::*;
-
- fn item_map(fixture: &str) -> (Arc<ItemMap>, ModuleId) {
- let (analysis, pos) = analysis_and_position(fixture);
- let db = analysis.imp.db;
- let source_root = db.file_source_root(pos.file_id);
- let descr = hir::Module::guess_from_position(&*db, pos)
- .unwrap()
- .unwrap();
- let module_id = descr.module_id;
- (db.item_map(source_root).unwrap(), module_id)
- }
-
- #[test]
- fn test_item_map() {
- let (item_map, module_id) = item_map(
- "
- //- /lib.rs
- mod foo;
-
- use crate::foo::bar::Baz;
- <|>
-
- //- /foo/mod.rs
- pub mod bar;
-
- //- /foo/bar.rs
- pub struct Baz;
- ",
- );
- let name = SmolStr::from("Baz");
- let resolution = &item_map.per_module[&module_id].items[&name];
- assert!(resolution.def_id.is_some());
- }
-
- #[test]
- fn typing_inside_a_function_should_not_invalidate_item_map() {
- let mock_analysis = MockAnalysis::with_files(
- "
- //- /lib.rs
- mod foo;
-
- use crate::foo::bar::Baz;
-
- fn foo() -> i32 {
- 1 + 1
- }
- //- /foo/mod.rs
- pub mod bar;
-
- //- /foo/bar.rs
- pub struct Baz;
- ",
- );
-
- let file_id = mock_analysis.id_of("/lib.rs");
- let mut host = mock_analysis.analysis_host();
-
- let source_root = host.analysis().imp.db.file_source_root(file_id);
-
- {
- let db = host.analysis().imp.db;
- let events = db.log_executed(|| {
- db.item_map(source_root).unwrap();
- });
- assert!(format!("{:?}", events).contains("item_map"))
- }
-
- let mut change = AnalysisChange::new();
-
- change.change_file(
- file_id,
- "
- mod foo;
-
- use crate::foo::bar::Baz;
-
- fn foo() -> i32 { 92 }
- "
- .to_string(),
- );
-
- host.apply_change(change);
-
- {
- let db = host.analysis().imp.db;
- let events = db.log_executed(|| {
- db.item_map(source_root).unwrap();
- });
- assert!(
- !format!("{:?}", events).contains("_item_map"),
- "{:#?}",
- events
- )
- }
- }
-}
+++ /dev/null
-use ra_syntax::{SmolStr, ast, AstNode, TextRange};
-
-#[derive(Debug, Clone, PartialEq, Eq)]
-pub(crate) struct Path {
- pub(crate) kind: PathKind,
- pub(crate) segments: Vec<SmolStr>,
-}
-
-#[derive(Debug, Clone, Copy, PartialEq, Eq)]
-pub(crate) enum PathKind {
- Plain,
- Self_,
- Super,
- Crate,
-}
-
-impl Path {
- /// Calls `cb` with all paths, represented by this use item.
- pub(crate) fn expand_use_item(item: ast::UseItem, mut cb: impl FnMut(Path, Option<TextRange>)) {
- if let Some(tree) = item.use_tree() {
- expand_use_tree(None, tree, &mut cb);
- }
- }
-
- /// Converts an `ast::Path` to `Path`. Works with use trees.
- pub(crate) fn from_ast(mut path: ast::Path) -> Option<Path> {
- let mut kind = PathKind::Plain;
- let mut segments = Vec::new();
- loop {
- let segment = path.segment()?;
- match segment.kind()? {
- ast::PathSegmentKind::Name(name) => segments.push(name.text()),
- ast::PathSegmentKind::CrateKw => {
- kind = PathKind::Crate;
- break;
- }
- ast::PathSegmentKind::SelfKw => {
- kind = PathKind::Self_;
- break;
- }
- ast::PathSegmentKind::SuperKw => {
- kind = PathKind::Super;
- break;
- }
- }
- path = match qualifier(path) {
- Some(it) => it,
- None => break,
- };
- }
- segments.reverse();
- return Some(Path { kind, segments });
-
- fn qualifier(path: ast::Path) -> Option<ast::Path> {
- if let Some(q) = path.qualifier() {
- return Some(q);
- }
- // TODO: this bottom up traversal is not too precise.
- // Should we handle do a top-down analysiss, recording results?
- let use_tree_list = path.syntax().ancestors().find_map(ast::UseTreeList::cast)?;
- let use_tree = use_tree_list.parent_use_tree();
- use_tree.path()
- }
- }
-
- /// `true` is this path is a single identifier, like `foo`
- pub(crate) fn is_ident(&self) -> bool {
- self.kind == PathKind::Plain && self.segments.len() == 1
- }
-}
-
-fn expand_use_tree(
- prefix: Option<Path>,
- tree: ast::UseTree,
- cb: &mut impl FnMut(Path, Option<TextRange>),
-) {
- if let Some(use_tree_list) = tree.use_tree_list() {
- let prefix = match tree.path() {
- None => prefix,
- Some(path) => match convert_path(prefix, path) {
- Some(it) => Some(it),
- None => return, // TODO: report errors somewhere
- },
- };
- for tree in use_tree_list.use_trees() {
- expand_use_tree(prefix.clone(), tree, cb);
- }
- } else {
- if let Some(ast_path) = tree.path() {
- if let Some(path) = convert_path(prefix, ast_path) {
- let range = if tree.has_star() {
- None
- } else {
- let range = ast_path.segment().unwrap().syntax().range();
- Some(range)
- };
- cb(path, range)
- }
- }
- }
-}
-
-fn convert_path(prefix: Option<Path>, path: ast::Path) -> Option<Path> {
- let prefix = if let Some(qual) = path.qualifier() {
- Some(convert_path(prefix, qual)?)
- } else {
- None
- };
- let segment = path.segment()?;
- let res = match segment.kind()? {
- ast::PathSegmentKind::Name(name) => {
- let mut res = prefix.unwrap_or_else(|| Path {
- kind: PathKind::Plain,
- segments: Vec::with_capacity(1),
- });
- res.segments.push(name.text());
- res
- }
- ast::PathSegmentKind::CrateKw => {
- if prefix.is_some() {
- return None;
- }
- Path {
- kind: PathKind::Crate,
- segments: Vec::new(),
- }
- }
- ast::PathSegmentKind::SelfKw => {
- if prefix.is_some() {
- return None;
- }
- Path {
- kind: PathKind::Self_,
- segments: Vec::new(),
- }
- }
- ast::PathSegmentKind::SuperKw => {
- if prefix.is_some() {
- return None;
- }
- Path {
- kind: PathKind::Super,
- segments: Vec::new(),
- }
- }
- };
- Some(res)
-}
+++ /dev/null
-use std::{
- sync::Arc,
- time::Instant,
-};
-
-use rustc_hash::FxHashMap;
-use ra_syntax::{
- AstNode, SyntaxNode, SmolStr,
- ast::{self, FnDef, FnDefNode, NameOwner, ModuleItemOwner}
-};
-use ra_db::SourceRootId;
-
-use crate::{
- FileId, Cancelable,
- hir::{
- FnId,
- SourceFileItems, SourceItemId,
- db::HirDatabase,
- function::FnScopes,
- module::{
- ModuleSource, ModuleSourceNode, ModuleId,
- imp::Submodule,
- nameres::{InputModuleItems, ItemMap, Resolver},
- },
- },
-};
-
-/// Resolve `FnId` to the corresponding `SyntaxNode`
-pub(super) fn fn_syntax(db: &impl HirDatabase, fn_id: FnId) -> FnDefNode {
- let item_id = fn_id.loc(db);
- let syntax = db.file_item(item_id);
- FnDef::cast(syntax.borrowed()).unwrap().owned()
-}
-
-pub(super) fn fn_scopes(db: &impl HirDatabase, fn_id: FnId) -> Arc<FnScopes> {
- let syntax = db.fn_syntax(fn_id);
- let res = FnScopes::new(syntax.borrowed());
- Arc::new(res)
-}
-
-pub(super) fn file_items(db: &impl HirDatabase, file_id: FileId) -> Arc<SourceFileItems> {
- let source_file = db.source_file(file_id);
- let source_file = source_file.borrowed();
- let mut res = SourceFileItems::default();
- source_file
- .syntax()
- .descendants()
- .filter_map(ast::ModuleItem::cast)
- .map(|it| it.syntax().owned())
- .for_each(|it| {
- res.alloc(it);
- });
- Arc::new(res)
-}
-
-pub(super) fn file_item(db: &impl HirDatabase, source_item_id: SourceItemId) -> SyntaxNode {
- db.file_items(source_item_id.file_id)[source_item_id.item_id].clone()
-}
-
-pub(crate) fn submodules(
- db: &impl HirDatabase,
- source: ModuleSource,
-) -> Cancelable<Arc<Vec<Submodule>>> {
- db.check_canceled()?;
- let file_id = source.file_id();
- let submodules = match source.resolve(db) {
- ModuleSourceNode::SourceFile(it) => collect_submodules(db, file_id, it.borrowed()),
- ModuleSourceNode::Module(it) => it
- .borrowed()
- .item_list()
- .map(|it| collect_submodules(db, file_id, it))
- .unwrap_or_else(Vec::new),
- };
- return Ok(Arc::new(submodules));
-
- fn collect_submodules<'a>(
- db: &impl HirDatabase,
- file_id: FileId,
- root: impl ast::ModuleItemOwner<'a>,
- ) -> Vec<Submodule> {
- modules(root)
- .map(|(name, m)| {
- if m.has_semi() {
- Submodule::Declaration(name)
- } else {
- let src = ModuleSource::new_inline(db, file_id, m);
- Submodule::Definition(name, src)
- }
- })
- .collect()
- }
-}
-
-pub(crate) fn modules<'a>(
- root: impl ast::ModuleItemOwner<'a>,
-) -> impl Iterator<Item = (SmolStr, ast::Module<'a>)> {
- root.items()
- .filter_map(|item| match item {
- ast::ModuleItem::Module(m) => Some(m),
- _ => None,
- })
- .filter_map(|module| {
- let name = module.name()?.text();
- Some((name, module))
- })
-}
-
-pub(super) fn input_module_items(
- db: &impl HirDatabase,
- source_root: SourceRootId,
- module_id: ModuleId,
-) -> Cancelable<Arc<InputModuleItems>> {
- let module_tree = db.module_tree(source_root)?;
- let source = module_id.source(&module_tree);
- let file_items = db.file_items(source.file_id());
- let res = match source.resolve(db) {
- ModuleSourceNode::SourceFile(it) => {
- let items = it.borrowed().items();
- InputModuleItems::new(&file_items, items)
- }
- ModuleSourceNode::Module(it) => {
- let items = it
- .borrowed()
- .item_list()
- .into_iter()
- .flat_map(|it| it.items());
- InputModuleItems::new(&file_items, items)
- }
- };
- Ok(Arc::new(res))
-}
-
-pub(super) fn item_map(
- db: &impl HirDatabase,
- source_root: SourceRootId,
-) -> Cancelable<Arc<ItemMap>> {
- let start = Instant::now();
- let module_tree = db.module_tree(source_root)?;
- let input = module_tree
- .modules()
- .map(|id| {
- let items = db.input_module_items(source_root, id)?;
- Ok((id, items))
- })
- .collect::<Cancelable<FxHashMap<_, _>>>()?;
- let resolver = Resolver {
- db: db,
- input: &input,
- source_root,
- module_tree,
- result: ItemMap::default(),
- };
- let res = resolver.resolve()?;
- let elapsed = start.elapsed();
- log::info!("item_map: {:?}", elapsed);
- Ok(Arc::new(res))
-}
use rayon::prelude::*;
use rustc_hash::FxHashSet;
use salsa::{Database, ParallelDatabase};
+use hir::{
+ self,
+ FnSignatureInfo,
+ Problem,
+};
use crate::{
completion::{completions, CompletionItem},
db,
- hir::{
- self,
- FnSignatureInfo,
- Problem,
- },
symbol_index::{SymbolIndex, SymbolsDatabase},
AnalysisChange, Cancelable, CrateId, Diagnostic, FileId,
FileSystemEdit, FilePosition, Query, SourceChange, SourceFileNodeEdit,
extern crate rustc_hash;
extern crate salsa;
-mod arena;
+macro_rules! ctry {
+ ($expr:expr) => {
+ match $expr {
+ None => return Ok(None),
+ Some(it) => it,
+ }
+ };
+}
+
mod db;
mod imp;
mod completion;
-mod hir;
mod symbol_index;
pub mod mock_analysis;
pub use crate::{
completion::CompletionItem,
- hir::FnSignatureInfo,
};
pub use ra_editor::{
FileSymbol, Fold, FoldKind, HighlightedRange, LineIndex, Runnable, RunnableKind, StructureNode,
};
+pub use hir::FnSignatureInfo;
pub use ra_db::{
Canceled, Cancelable, FilePosition,
fn is_send<T: Send>() {}
is_send::<Analysis>();
}
+
+//TODO: move to hir
+#[cfg(test)]
+mod hir_namres_tests {
+ use std::sync::Arc;
+ use ra_db::FilesDatabase;
+ use ra_syntax::SmolStr;
+ use hir::{self, db::HirDatabase};
+
+ use crate::{
+ AnalysisChange,
+ mock_analysis::{MockAnalysis, analysis_and_position},
+};
+
+ fn item_map(fixture: &str) -> (Arc<hir::ItemMap>, hir::ModuleId) {
+ let (analysis, pos) = analysis_and_position(fixture);
+ let db = analysis.imp.db;
+ let source_root = db.file_source_root(pos.file_id);
+ let descr = hir::Module::guess_from_position(&*db, pos)
+ .unwrap()
+ .unwrap();
+ let module_id = descr.module_id;
+ (db.item_map(source_root).unwrap(), module_id)
+ }
+
+ #[test]
+ fn test_item_map() {
+ let (item_map, module_id) = item_map(
+ "
+ //- /lib.rs
+ mod foo;
+
+ use crate::foo::bar::Baz;
+ <|>
+
+ //- /foo/mod.rs
+ pub mod bar;
+
+ //- /foo/bar.rs
+ pub struct Baz;
+ ",
+ );
+ let name = SmolStr::from("Baz");
+ let resolution = &item_map.per_module[&module_id].items[&name];
+ assert!(resolution.def_id.is_some());
+ }
+
+ #[test]
+ fn typing_inside_a_function_should_not_invalidate_item_map() {
+ let mock_analysis = MockAnalysis::with_files(
+ "
+ //- /lib.rs
+ mod foo;
+
+ use crate::foo::bar::Baz;
+
+ fn foo() -> i32 {
+ 1 + 1
+ }
+ //- /foo/mod.rs
+ pub mod bar;
+
+ //- /foo/bar.rs
+ pub struct Baz;
+ ",
+ );
+
+ let file_id = mock_analysis.id_of("/lib.rs");
+ let mut host = mock_analysis.analysis_host();
+
+ let source_root = host.analysis().imp.db.file_source_root(file_id);
+
+ {
+ let db = host.analysis().imp.db;
+ let events = db.log_executed(|| {
+ db.item_map(source_root).unwrap();
+ });
+ assert!(format!("{:?}", events).contains("item_map"))
+ }
+
+ let mut change = AnalysisChange::new();
+
+ change.change_file(
+ file_id,
+ "
+ mod foo;
+
+ use crate::foo::bar::Baz;
+
+ fn foo() -> i32 { 92 }
+ "
+ .to_string(),
+ );
+
+ host.apply_change(change);
+
+ {
+ let db = host.analysis().imp.db;
+ let events = db.log_executed(|| {
+ db.item_map(source_root).unwrap();
+ });
+ assert!(
+ !format!("{:?}", events).contains("_item_map"),
+ "{:#?}",
+ events
+ )
+ }
+ }
+}
marker::PhantomData,
};
-pub(crate) struct Id<T> {
+pub struct Id<T> {
idx: u32,
_ty: PhantomData<fn() -> T>,
}
DefLoc, DefId, FnId,
SourceFileItems, SourceItemId,
query_definitions,
- function::{FnScopes},
+ FnScopes,
module::{ModuleId, ModuleTree, ModuleSource,
nameres::{ItemMap, InputModuleItems}},
};
salsa::query_group! {
-pub(crate) trait HirDatabase: SyntaxDatabase
+pub trait HirDatabase: SyntaxDatabase
+ AsRef<LocationIntener<DefLoc, DefId>>
+ AsRef<LocationIntener<SourceItemId, FnId>>
{
FnId, HirDatabase, SourceItemId,
};
-pub(crate) use self::scope::FnScopes;
+pub use self::scope::FnScopes;
impl FnId {
- pub(crate) fn get(db: &impl HirDatabase, file_id: FileId, fn_def: ast::FnDef) -> FnId {
+ pub fn get(db: &impl HirDatabase, file_id: FileId, fn_def: ast::FnDef) -> FnId {
let file_items = db.file_items(file_id);
let item_id = file_items.id_of(fn_def.syntax());
let item_id = SourceItemId { file_id, item_id };
}
}
-pub(crate) struct Function {
+pub struct Function {
fn_id: FnId,
}
impl Function {
- pub(crate) fn guess_from_source(
+ pub fn guess_from_source(
db: &impl HirDatabase,
file_id: FileId,
fn_def: ast::FnDef,
Function { fn_id }
}
- pub(crate) fn guess_for_name_ref(
+ pub fn guess_for_name_ref(
db: &impl HirDatabase,
file_id: FileId,
name_ref: ast::NameRef,
Function::guess_for_node(db, file_id, name_ref.syntax())
}
- pub(crate) fn guess_for_bind_pat(
+ pub fn guess_for_bind_pat(
db: &impl HirDatabase,
file_id: FileId,
bind_pat: ast::BindPat,
Some(res)
}
- pub(crate) fn scope(&self, db: &impl HirDatabase) -> Arc<FnScopes> {
+ pub fn scope(&self, db: &impl HirDatabase) -> Arc<FnScopes> {
db.fn_scopes(self.fn_id)
}
- pub(crate) fn signature_info(&self, db: &impl HirDatabase) -> Option<FnSignatureInfo> {
+ pub fn signature_info(&self, db: &impl HirDatabase) -> Option<FnSignatureInfo> {
let syntax = db.fn_syntax(self.fn_id);
FnSignatureInfo::new(syntax.borrowed())
}
#[derive(Debug, PartialEq, Eq)]
pub struct FnScopes {
- pub(crate) self_param: Option<LocalSyntaxPtr>,
+ pub self_param: Option<LocalSyntaxPtr>,
scopes: Arena<ScopeData>,
scope_for: FxHashMap<LocalSyntaxPtr, ScopeId>,
}
}
#[derive(Debug, PartialEq, Eq)]
-pub(crate) struct ScopeData {
+pub struct ScopeData {
parent: Option<ScopeId>,
entries: Vec<ScopeEntry>,
}
impl FnScopes {
- pub(crate) fn new(fn_def: ast::FnDef) -> FnScopes {
+ pub fn new(fn_def: ast::FnDef) -> FnScopes {
let mut scopes = FnScopes {
self_param: fn_def
.param_list()
}
scopes
}
- pub(crate) fn entries(&self, scope: ScopeId) -> &[ScopeEntry] {
+ pub fn entries(&self, scope: ScopeId) -> &[ScopeEntry] {
&self.scopes[scope].entries
}
pub fn scope_chain<'a>(&'a self, node: SyntaxNodeRef) -> impl Iterator<Item = ScopeId> + 'a {
self.scopes[scope].parent
})
}
- pub(crate) fn resolve_local_name<'a>(
- &'a self,
- name_ref: ast::NameRef,
- ) -> Option<&'a ScopeEntry> {
+ pub fn resolve_local_name<'a>(&'a self, name_ref: ast::NameRef) -> Option<&'a ScopeEntry> {
let mut shadowed = FxHashSet::default();
let ret = self
.scope_chain(name_ref.syntax())
};
Some(res)
}
- pub(crate) fn name(&self) -> &SmolStr {
+ pub fn name(&self) -> &SmolStr {
&self.name
}
- pub(crate) fn ptr(&self) -> LocalSyntaxPtr {
+ pub fn ptr(&self) -> LocalSyntaxPtr {
self.ptr
}
}
};
}
-pub(crate) mod db;
+pub mod db;
mod query_definitions;
mod function;
mod module;
arena::{Arena, Id},
};
-pub(crate) use self::{
+pub use self::{
path::{Path, PathKind},
- module::{Module, ModuleId, Problem},
+ module::{Module, ModuleId, Problem, nameres::ItemMap},
function::{Function, FnScopes},
};
pub use self::function::FnSignatureInfo;
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
-pub(crate) struct FnId(u32);
+pub struct FnId(u32);
ra_db::impl_numeric_id!(FnId);
impl FnId {
- pub(crate) fn from_loc(
+ pub fn from_loc(
db: &impl AsRef<LocationIntener<SourceItemId, FnId>>,
loc: &SourceItemId,
) -> FnId {
db.as_ref().loc2id(loc)
}
- pub(crate) fn loc(self, db: &impl AsRef<LocationIntener<SourceItemId, FnId>>) -> SourceItemId {
+ pub fn loc(self, db: &impl AsRef<LocationIntener<SourceItemId, FnId>>) -> SourceItemId {
db.as_ref().id2loc(self)
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
-pub(crate) struct DefId(u32);
+pub struct DefId(u32);
ra_db::impl_numeric_id!(DefId);
#[derive(Clone, Debug, PartialEq, Eq, Hash)]
-pub(crate) enum DefLoc {
+pub enum DefLoc {
Module {
id: ModuleId,
source_root: SourceRootId,
}
impl DefId {
- pub(crate) fn loc(self, db: &impl AsRef<LocationIntener<DefLoc, DefId>>) -> DefLoc {
+ pub fn loc(self, db: &impl AsRef<LocationIntener<DefLoc, DefId>>) -> DefLoc {
db.as_ref().id2loc(self)
}
}
impl DefLoc {
- pub(crate) fn id(&self, db: &impl AsRef<LocationIntener<DefLoc, DefId>>) -> DefId {
+ pub fn id(&self, db: &impl AsRef<LocationIntener<DefLoc, DefId>>) -> DefId {
db.as_ref().loc2id(&self)
}
}
-pub(crate) enum Def {
+pub enum Def {
Module(Module),
Item,
}
impl DefId {
- pub(crate) fn resolve(self, db: &impl HirDatabase) -> Cancelable<Def> {
+ pub fn resolve(self, db: &impl HirDatabase) -> Cancelable<Def> {
let loc = self.loc(db);
let res = match loc {
DefLoc::Module { id, source_root } => {
pub(crate) type SourceFileItemId = Id<SyntaxNode>;
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
-pub(crate) struct SourceItemId {
+pub struct SourceItemId {
file_id: FileId,
item_id: SourceFileItemId,
}
/// Maps item's `SyntaxNode`s to `SourceFileItemId` and back.
#[derive(Debug, PartialEq, Eq, Default)]
-pub(crate) struct SourceFileItems {
+pub struct SourceFileItems {
arena: Arena<SyntaxNode>,
}
fn alloc(&mut self, item: SyntaxNode) -> SourceFileItemId {
self.arena.alloc(item)
}
- fn id_of(&self, item: SyntaxNodeRef) -> SourceFileItemId {
+ pub fn id_of(&self, item: SyntaxNodeRef) -> SourceFileItemId {
let (id, _item) = self
.arena
.iter()
};
#[derive(Clone, Hash, PartialEq, Eq, Debug)]
-pub(crate) enum Submodule {
+pub enum Submodule {
Declaration(SmolStr),
Definition(SmolStr, ModuleSource),
}
arena::{Arena, Id},
};
-pub(crate) use self::nameres::ModuleScope;
+pub use self::nameres::ModuleScope;
/// `Module` is API entry point to get all the information
/// about a particular module.
#[derive(Debug, Clone)]
-pub(crate) struct Module {
+pub struct Module {
tree: Arc<ModuleTree>,
source_root_id: SourceRootId,
- module_id: ModuleId,
+ //TODO: make private
+ pub module_id: ModuleId,
}
impl Module {
}
/// Returns a `ModuleScope`: a set of items, visible in this module.
- pub(crate) fn scope(&self, db: &impl HirDatabase) -> Cancelable<ModuleScope> {
+ pub fn scope(&self, db: &impl HirDatabase) -> Cancelable<ModuleScope> {
let item_map = db.item_map(self.source_root_id)?;
let res = item_map.per_module[&self.module_id].clone();
Ok(res)
}
- pub(crate) fn resolve_path(
- &self,
- db: &impl HirDatabase,
- path: Path,
- ) -> Cancelable<Option<DefId>> {
+ pub fn resolve_path(&self, db: &impl HirDatabase, path: Path) -> Cancelable<Option<DefId>> {
let mut curr = match path.kind {
PathKind::Crate => self.crate_root(),
PathKind::Self_ | PathKind::Plain => self.clone(),
/// (which can have multiple parents) to the precise world of modules (which
/// always have one parent).
#[derive(Default, Debug, PartialEq, Eq)]
-pub(crate) struct ModuleTree {
+pub struct ModuleTree {
mods: Arena<ModuleData>,
links: Arena<LinkData>,
}
/// `ModuleSource` is the syntax tree element that produced this module:
/// either a file, or an inlinde module.
#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
-pub(crate) enum ModuleSource {
+pub enum ModuleSource {
SourceFile(FileId),
Module(SourceItemId),
}
/// An owned syntax node for a module. Unlike `ModuleSource`,
/// this holds onto the AST for the whole file.
-pub(crate) enum ModuleSourceNode {
+pub enum ModuleSourceNode {
SourceFile(ast::SourceFileNode),
Module(ast::ModuleNode),
}
-pub(crate) type ModuleId = Id<ModuleData>;
+pub type ModuleId = Id<ModuleData>;
type LinkId = Id<LinkData>;
#[derive(Clone, Debug, Hash, PartialEq, Eq)]
}
#[derive(Debug, PartialEq, Eq, Hash)]
-pub(crate) struct ModuleData {
+pub struct ModuleData {
source: ModuleSource,
parent: Option<LinkId>,
children: Vec<LinkId>,
ModuleSource::Module(id)
}
- pub(crate) fn as_file(self) -> Option<FileId> {
+ pub fn as_file(self) -> Option<FileId> {
match self {
ModuleSource::SourceFile(f) => Some(f),
ModuleSource::Module(..) => None,
}
}
- pub(crate) fn file_id(self) -> FileId {
+ pub fn file_id(self) -> FileId {
match self {
ModuleSource::SourceFile(f) => f,
ModuleSource::Module(source_item_id) => source_item_id.file_id,
}
}
- pub(crate) fn resolve(self, db: &impl HirDatabase) -> ModuleSourceNode {
+ pub fn resolve(self, db: &impl HirDatabase) -> ModuleSourceNode {
match self {
ModuleSource::SourceFile(file_id) => {
let syntax = db.source_file(file_id);
/// Item map is the result of the name resolution. Item map contains, for each
/// module, the set of visible items.
#[derive(Default, Debug, PartialEq, Eq)]
-pub(crate) struct ItemMap {
- pub(crate) per_module: FxHashMap<ModuleId, ModuleScope>,
+pub struct ItemMap {
+ pub per_module: FxHashMap<ModuleId, ModuleScope>,
}
#[derive(Debug, Default, PartialEq, Eq, Clone)]
-pub(crate) struct ModuleScope {
- items: FxHashMap<SmolStr, Resolution>,
+pub struct ModuleScope {
+ pub items: FxHashMap<SmolStr, Resolution>,
}
impl ModuleScope {
- pub(crate) fn entries<'a>(&'a self) -> impl Iterator<Item = (&'a SmolStr, &Resolution)> + 'a {
+ pub fn entries<'a>(&'a self) -> impl Iterator<Item = (&'a SmolStr, &Resolution)> + 'a {
self.items.iter()
}
- pub(crate) fn get(&self, name: &SmolStr) -> Option<&Resolution> {
+ pub fn get(&self, name: &SmolStr) -> Option<&Resolution> {
self.items.get(name)
}
}
/// recomputing name res: if `InputModuleItems` are the same, we can avoid
/// running name resolution.
#[derive(Debug, Default, PartialEq, Eq)]
-pub(crate) struct InputModuleItems {
+pub struct InputModuleItems {
items: Vec<ModuleItem>,
imports: Vec<Import>,
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
-pub(crate) struct NamedImport {
- file_item_id: SourceFileItemId,
- relative_range: TextRange,
+pub struct NamedImport {
+ pub file_item_id: SourceFileItemId,
+ pub relative_range: TextRange,
}
impl NamedImport {
- pub(crate) fn range(&self, db: &impl HirDatabase, file_id: FileId) -> TextRange {
+ pub fn range(&self, db: &impl HirDatabase, file_id: FileId) -> TextRange {
let source_item_id = SourceItemId {
file_id,
item_id: self.file_item_id,
/// Resolution is basically `DefId` atm, but it should account for stuff like
/// multiple namespaces, ambiguity and errors.
#[derive(Debug, Clone, PartialEq, Eq)]
-pub(crate) struct Resolution {
+pub struct Resolution {
/// None for unresolved
- pub(crate) def_id: Option<DefId>,
+ pub def_id: Option<DefId>,
/// ident by whitch this is imported into local scope.
- pub(crate) import: Option<NamedImport>,
+ pub import: Option<NamedImport>,
}
// #[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
f(module_items)
}
}
-
-#[cfg(test)]
-mod tests {
- use ra_db::FilesDatabase;
- use crate::{
- AnalysisChange,
- mock_analysis::{MockAnalysis, analysis_and_position},
- hir::{self, HirDatabase},
-};
- use super::*;
-
- fn item_map(fixture: &str) -> (Arc<ItemMap>, ModuleId) {
- let (analysis, pos) = analysis_and_position(fixture);
- let db = analysis.imp.db;
- let source_root = db.file_source_root(pos.file_id);
- let descr = hir::Module::guess_from_position(&*db, pos)
- .unwrap()
- .unwrap();
- let module_id = descr.module_id;
- (db.item_map(source_root).unwrap(), module_id)
- }
-
- #[test]
- fn test_item_map() {
- let (item_map, module_id) = item_map(
- "
- //- /lib.rs
- mod foo;
-
- use crate::foo::bar::Baz;
- <|>
-
- //- /foo/mod.rs
- pub mod bar;
-
- //- /foo/bar.rs
- pub struct Baz;
- ",
- );
- let name = SmolStr::from("Baz");
- let resolution = &item_map.per_module[&module_id].items[&name];
- assert!(resolution.def_id.is_some());
- }
-
- #[test]
- fn typing_inside_a_function_should_not_invalidate_item_map() {
- let mock_analysis = MockAnalysis::with_files(
- "
- //- /lib.rs
- mod foo;
-
- use crate::foo::bar::Baz;
-
- fn foo() -> i32 {
- 1 + 1
- }
- //- /foo/mod.rs
- pub mod bar;
-
- //- /foo/bar.rs
- pub struct Baz;
- ",
- );
-
- let file_id = mock_analysis.id_of("/lib.rs");
- let mut host = mock_analysis.analysis_host();
-
- let source_root = host.analysis().imp.db.file_source_root(file_id);
-
- {
- let db = host.analysis().imp.db;
- let events = db.log_executed(|| {
- db.item_map(source_root).unwrap();
- });
- assert!(format!("{:?}", events).contains("item_map"))
- }
-
- let mut change = AnalysisChange::new();
-
- change.change_file(
- file_id,
- "
- mod foo;
-
- use crate::foo::bar::Baz;
-
- fn foo() -> i32 { 92 }
- "
- .to_string(),
- );
-
- host.apply_change(change);
-
- {
- let db = host.analysis().imp.db;
- let events = db.log_executed(|| {
- db.item_map(source_root).unwrap();
- });
- assert!(
- !format!("{:?}", events).contains("_item_map"),
- "{:#?}",
- events
- )
- }
- }
-}
use ra_syntax::{SmolStr, ast, AstNode, TextRange};
#[derive(Debug, Clone, PartialEq, Eq)]
-pub(crate) struct Path {
- pub(crate) kind: PathKind,
- pub(crate) segments: Vec<SmolStr>,
+pub struct Path {
+ pub kind: PathKind,
+ pub segments: Vec<SmolStr>,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
-pub(crate) enum PathKind {
+pub enum PathKind {
Plain,
Self_,
Super,
impl Path {
/// Calls `cb` with all paths, represented by this use item.
- pub(crate) fn expand_use_item(item: ast::UseItem, mut cb: impl FnMut(Path, Option<TextRange>)) {
+ pub fn expand_use_item(item: ast::UseItem, mut cb: impl FnMut(Path, Option<TextRange>)) {
if let Some(tree) = item.use_tree() {
expand_use_tree(None, tree, &mut cb);
}
}
/// Converts an `ast::Path` to `Path`. Works with use trees.
- pub(crate) fn from_ast(mut path: ast::Path) -> Option<Path> {
+ pub fn from_ast(mut path: ast::Path) -> Option<Path> {
let mut kind = PathKind::Plain;
let mut segments = Vec::new();
loop {
}
/// `true` is this path is a single identifier, like `foo`
- pub(crate) fn is_ident(&self) -> bool {
+ pub fn is_ident(&self) -> bool {
self.kind == PathKind::Plain && self.segments.len() == 1
}
}