use ra_db::{FileId, SourceDatabase}; use ra_syntax::{ SyntaxNode, AstNode, SmolStr, TextRange, AstPtr, SyntaxKind::{self, NAME}, ast::{self, DocCommentsOwner}, algo::visit::{visitor, Visitor}, }; use hir::{ModuleSource, FieldSource, ImplItem}; use crate::{FileSymbol, db::RootDatabase}; use super::short_label::ShortLabel; /// `NavigationTarget` represents and element in the editor's UI which you can /// click on to navigate to a particular piece of code. /// /// Typically, a `NavigationTarget` corresponds to some element in the source /// code, like a function or a struct, but this is not strictly required. #[derive(Debug, Clone)] pub struct NavigationTarget { file_id: FileId, name: SmolStr, kind: SyntaxKind, full_range: TextRange, focus_range: Option, container_name: Option, description: Option, docs: Option, } impl NavigationTarget { /// When `focus_range` is specified, returns it. otherwise /// returns `full_range` pub fn range(&self) -> TextRange { self.focus_range.unwrap_or(self.full_range) } pub fn name(&self) -> &SmolStr { &self.name } pub fn container_name(&self) -> Option<&SmolStr> { self.container_name.as_ref() } pub fn kind(&self) -> SyntaxKind { self.kind } pub fn file_id(&self) -> FileId { self.file_id } pub fn full_range(&self) -> TextRange { self.full_range } pub fn docs(&self) -> Option<&str> { self.docs.as_ref().map(String::as_str) } pub fn description(&self) -> Option<&str> { self.description.as_ref().map(String::as_str) } /// A "most interesting" range withing the `full_range`. /// /// Typically, `full_range` is the whole syntax node, /// including doc comments, and `focus_range` is the range of the identifier. pub fn focus_range(&self) -> Option { self.focus_range } pub(crate) fn from_bind_pat(file_id: FileId, pat: &ast::BindPat) -> NavigationTarget { NavigationTarget::from_named(file_id, pat, None, None) } pub(crate) fn from_symbol(db: &RootDatabase, symbol: FileSymbol) -> NavigationTarget { NavigationTarget { file_id: symbol.file_id, name: symbol.name.clone(), kind: symbol.ptr.kind(), full_range: symbol.ptr.range(), focus_range: symbol.name_range, container_name: symbol.container_name.clone(), description: description_from_symbol(db, &symbol), docs: docs_from_symbol(db, &symbol), } } pub(crate) fn from_pat( db: &RootDatabase, file_id: FileId, pat: AstPtr, ) -> NavigationTarget { let file = db.parse(file_id).tree; let (name, full_range) = match pat.to_node(file.syntax()).kind() { ast::PatKind::BindPat(pat) => return NavigationTarget::from_bind_pat(file_id, &pat), _ => ("_".into(), pat.syntax_node_ptr().range()), }; NavigationTarget { file_id, name, full_range, focus_range: None, kind: NAME, container_name: None, description: None, //< No documentation for Description docs: None, //< No documentation for Pattern } } pub(crate) fn from_self_param( file_id: FileId, par: AstPtr, ) -> NavigationTarget { let (name, full_range) = ("self".into(), par.syntax_node_ptr().range()); NavigationTarget { file_id, name, full_range, focus_range: None, kind: NAME, container_name: None, description: None, //< No document node for SelfParam docs: None, //< No document node for SelfParam } } pub(crate) fn from_module(db: &RootDatabase, module: hir::Module) -> NavigationTarget { let (file_id, source) = module.definition_source(db); let file_id = file_id.as_original_file(); let name = module.name(db).map(|it| it.to_string().into()).unwrap_or_default(); match source { ModuleSource::SourceFile(node) => { NavigationTarget::from_syntax(file_id, name, None, node.syntax(), None, None) } ModuleSource::Module(node) => NavigationTarget::from_syntax( file_id, name, None, node.syntax(), node.doc_comment_text(), node.short_label(), ), } } pub(crate) fn from_module_to_decl(db: &RootDatabase, module: hir::Module) -> NavigationTarget { let name = module.name(db).map(|it| it.to_string().into()).unwrap_or_default(); if let Some((file_id, source)) = module.declaration_source(db) { let file_id = file_id.as_original_file(); return NavigationTarget::from_syntax( file_id, name, None, source.syntax(), source.doc_comment_text(), source.short_label(), ); } NavigationTarget::from_module(db, module) } pub(crate) fn from_function(db: &RootDatabase, func: hir::Function) -> NavigationTarget { let (file_id, fn_def) = func.source(db); NavigationTarget::from_named( file_id.original_file(db), &*fn_def, fn_def.doc_comment_text(), fn_def.short_label(), ) } pub(crate) fn from_field(db: &RootDatabase, field: hir::StructField) -> NavigationTarget { let (file_id, field) = field.source(db); let file_id = file_id.original_file(db); match field { FieldSource::Named(it) => { NavigationTarget::from_named(file_id, &*it, it.doc_comment_text(), it.short_label()) } FieldSource::Pos(it) => { NavigationTarget::from_syntax(file_id, "".into(), None, it.syntax(), None, None) } } } pub(crate) fn from_adt_def(db: &RootDatabase, adt_def: hir::AdtDef) -> NavigationTarget { match adt_def { hir::AdtDef::Struct(s) => { let (file_id, node) = s.source(db); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), node.short_label(), ) } hir::AdtDef::Union(s) => { let (file_id, node) = s.source(db); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), node.short_label(), ) } hir::AdtDef::Enum(s) => { let (file_id, node) = s.source(db); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), node.short_label(), ) } } } pub(crate) fn from_def( db: &RootDatabase, module_def: hir::ModuleDef, ) -> Option { let nav = match module_def { hir::ModuleDef::Module(module) => NavigationTarget::from_module(db, module), hir::ModuleDef::Function(func) => NavigationTarget::from_function(db, func), hir::ModuleDef::Struct(s) => { let (file_id, node) = s.source(db); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), node.short_label(), ) } hir::ModuleDef::Union(s) => { let (file_id, node) = s.source(db); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), node.short_label(), ) } hir::ModuleDef::Const(s) => { let (file_id, node) = s.source(db); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), node.short_label(), ) } hir::ModuleDef::Static(s) => { let (file_id, node) = s.source(db); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), node.short_label(), ) } hir::ModuleDef::Enum(e) => { let (file_id, node) = e.source(db); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), node.short_label(), ) } hir::ModuleDef::EnumVariant(var) => { let (file_id, node) = var.source(db); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), node.short_label(), ) } hir::ModuleDef::Trait(e) => { let (file_id, node) = e.source(db); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), node.short_label(), ) } hir::ModuleDef::TypeAlias(e) => { let (file_id, node) = e.source(db); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), node.short_label(), ) } hir::ModuleDef::BuiltinType(..) => { return None; } }; Some(nav) } pub(crate) fn from_impl_block( db: &RootDatabase, impl_block: hir::ImplBlock, ) -> NavigationTarget { let (file_id, node) = impl_block.source(db); NavigationTarget::from_syntax( file_id.as_original_file(), "impl".into(), None, node.syntax(), None, None, ) } pub(crate) fn from_impl_item(db: &RootDatabase, impl_item: hir::ImplItem) -> NavigationTarget { match impl_item { ImplItem::Method(f) => NavigationTarget::from_function(db, f), ImplItem::Const(c) => { let (file_id, node) = c.source(db); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), node.short_label(), ) } ImplItem::TypeAlias(a) => { let (file_id, node) = a.source(db); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), node.short_label(), ) } } } pub(crate) fn from_macro_def(db: &RootDatabase, macro_call: hir::MacroDef) -> NavigationTarget { let (file_id, node) = macro_call.source(db); log::debug!("nav target {}", node.syntax().debug_dump()); NavigationTarget::from_named( file_id.original_file(db), &*node, node.doc_comment_text(), None, ) } #[cfg(test)] pub(crate) fn assert_match(&self, expected: &str) { let actual = self.debug_render(); test_utils::assert_eq_text!(expected.trim(), actual.trim(),); } #[cfg(test)] pub(crate) fn debug_render(&self) -> String { let mut buf = format!( "{} {:?} {:?} {:?}", self.name(), self.kind(), self.file_id(), self.full_range() ); if let Some(focus_range) = self.focus_range() { buf.push_str(&format!(" {:?}", focus_range)) } if let Some(container_name) = self.container_name() { buf.push_str(&format!(" {}", container_name)) } buf } /// Allows `NavigationTarget` to be created from a `NameOwner` pub(crate) fn from_named( file_id: FileId, node: &impl ast::NameOwner, docs: Option, description: Option, ) -> NavigationTarget { //FIXME: use `_` instead of empty string let name = node.name().map(|it| it.text().clone()).unwrap_or_default(); let focus_range = node.name().map(|it| it.syntax().range()); NavigationTarget::from_syntax(file_id, name, focus_range, node.syntax(), docs, description) } fn from_syntax( file_id: FileId, name: SmolStr, focus_range: Option, node: &SyntaxNode, docs: Option, description: Option, ) -> NavigationTarget { NavigationTarget { file_id, name, kind: node.kind(), full_range: node.range(), focus_range, // ptr: Some(LocalSyntaxPtr::new(node)), container_name: None, description, docs, } } } fn docs_from_symbol(db: &RootDatabase, symbol: &FileSymbol) -> Option { let file = db.parse(symbol.file_id).tree; let node = symbol.ptr.to_node(file.syntax()).to_owned(); fn doc_comments(node: &N) -> Option { node.doc_comment_text() } visitor() .visit(doc_comments::) .visit(doc_comments::) .visit(doc_comments::) .visit(doc_comments::) .visit(doc_comments::) .visit(doc_comments::) .visit(doc_comments::) .visit(doc_comments::) .visit(doc_comments::) .visit(doc_comments::) .visit(doc_comments::) .accept(&node)? } /// Get a description of a symbol. /// /// e.g. `struct Name`, `enum Name`, `fn Name` fn description_from_symbol(db: &RootDatabase, symbol: &FileSymbol) -> Option { let file = db.parse(symbol.file_id).tree; let node = symbol.ptr.to_node(file.syntax()).to_owned(); // FIXME: After type inference is done, add type information to improve the output visitor() .visit(|node: &ast::FnDef| node.short_label()) .visit(|node: &ast::StructDef| node.short_label()) .visit(|node: &ast::EnumDef| node.short_label()) .visit(|node: &ast::TraitDef| node.short_label()) .visit(|node: &ast::Module| node.short_label()) .visit(|node: &ast::TypeAliasDef| node.short_label()) .visit(|node: &ast::ConstDef| node.short_label()) .visit(|node: &ast::StaticDef| node.short_label()) .visit(|node: &ast::NamedFieldDef| node.short_label()) .visit(|node: &ast::EnumVariant| node.short_label()) .accept(&node)? }