Add display::Description

This commit is contained in:
Edwin Cheng 2019-06-09 23:59:59 +08:00
parent 2acf1e16fc
commit aacc894134
4 changed files with 163 additions and 67 deletions

View File

@ -4,6 +4,7 @@
mod function_signature; mod function_signature;
mod navigation_target; mod navigation_target;
mod structure; mod structure;
mod description;
use ra_syntax::{ast::{self, AstNode, TypeParamsOwner}, SyntaxKind::{ATTR, COMMENT}}; use ra_syntax::{ast::{self, AstNode, TypeParamsOwner}, SyntaxKind::{ATTR, COMMENT}};
@ -11,6 +12,8 @@ pub use navigation_target::NavigationTarget;
pub use structure::{StructureNode, file_structure}; pub use structure::{StructureNode, file_structure};
pub use function_signature::FunctionSignature; pub use function_signature::FunctionSignature;
pub(crate) use description::Description;
pub(crate) fn function_label(node: &ast::FnDef) -> String { pub(crate) fn function_label(node: &ast::FnDef) -> String {
FunctionSignature::from(node).to_string() FunctionSignature::from(node).to_string()
} }

View File

@ -0,0 +1,92 @@
use ra_syntax::{
ast::{self, NameOwner, VisibilityOwner, TypeAscriptionOwner, AstNode},
};
pub(crate) trait Description {
fn description(&self) -> Option<String>;
}
impl Description for ast::FnDef {
fn description(&self) -> Option<String> {
Some(crate::display::function_label(self))
}
}
impl Description for ast::StructDef {
fn description(&self) -> Option<String> {
visit_node(self, "struct ")
}
}
impl Description for ast::EnumDef {
fn description(&self) -> Option<String> {
visit_node(self, "enum ")
}
}
impl Description for ast::TraitDef {
fn description(&self) -> Option<String> {
visit_node(self, "trait ")
}
}
impl Description for ast::Module {
fn description(&self) -> Option<String> {
visit_node(self, "mod ")
}
}
impl Description for ast::TypeAliasDef {
fn description(&self) -> Option<String> {
visit_node(self, "type ")
}
}
impl Description for ast::ConstDef {
fn description(&self) -> Option<String> {
visit_ascribed_node(self, "const ")
}
}
impl Description for ast::StaticDef {
fn description(&self) -> Option<String> {
visit_ascribed_node(self, "static ")
}
}
impl Description for ast::NamedFieldDef {
fn description(&self) -> Option<String> {
visit_ascribed_node(self, "")
}
}
impl Description for ast::EnumVariant {
fn description(&self) -> Option<String> {
Some(self.name()?.text().to_string())
}
}
fn visit_ascribed_node<T>(node: &T, prefix: &str) -> Option<String>
where
T: NameOwner + VisibilityOwner + TypeAscriptionOwner,
{
let mut string = visit_node(node, prefix)?;
if let Some(type_ref) = node.ascribed_type() {
string.push_str(": ");
type_ref.syntax().text().push_to(&mut string);
}
Some(string)
}
fn visit_node<T>(node: &T, label: &str) -> Option<String>
where
T: NameOwner + VisibilityOwner,
{
let mut string =
node.visibility().map(|v| format!("{} ", v.syntax().text())).unwrap_or_default();
string.push_str(label);
string.push_str(node.name()?.text().as_str());
Some(string)
}

View File

@ -2,12 +2,13 @@ use ra_db::{FileId, SourceDatabase};
use ra_syntax::{ use ra_syntax::{
SyntaxNode, AstNode, SmolStr, TextRange, AstPtr, SyntaxNode, AstNode, SmolStr, TextRange, AstPtr,
SyntaxKind::{self, NAME}, SyntaxKind::{self, NAME},
ast::{self, NameOwner, VisibilityOwner, TypeAscriptionOwner, DocCommentsOwner}, ast::{self, DocCommentsOwner},
algo::visit::{visitor, Visitor}, algo::visit::{visitor, Visitor},
}; };
use hir::{ModuleSource, FieldSource, ImplItem}; use hir::{ModuleSource, FieldSource, ImplItem};
use crate::{FileSymbol, db::RootDatabase}; use crate::{FileSymbol, db::RootDatabase};
use super::description::Description;
/// `NavigationTarget` represents and element in the editor's UI which you can /// `NavigationTarget` represents and element in the editor's UI which you can
/// click on to navigate to a particular piece of code. /// click on to navigate to a particular piece of code.
@ -63,7 +64,7 @@ impl NavigationTarget {
} }
pub(crate) fn from_bind_pat(file_id: FileId, pat: &ast::BindPat) -> NavigationTarget { pub(crate) fn from_bind_pat(file_id: FileId, pat: &ast::BindPat) -> NavigationTarget {
NavigationTarget::from_named(file_id, pat, None) NavigationTarget::from_named(file_id, pat, None, None)
} }
pub(crate) fn from_symbol(db: &RootDatabase, symbol: FileSymbol) -> NavigationTarget { pub(crate) fn from_symbol(db: &RootDatabase, symbol: FileSymbol) -> NavigationTarget {
@ -77,7 +78,7 @@ impl NavigationTarget {
full_range: symbol.ptr.range(), full_range: symbol.ptr.range(),
focus_range: symbol.name_range, focus_range: symbol.name_range,
container_name: symbol.container_name.clone(), container_name: symbol.container_name.clone(),
description: description_inner(&node), description: description_from_symbol(&node),
docs: docs_from_symbol(db, &symbol), docs: docs_from_symbol(db, &symbol),
} }
} }
@ -92,7 +93,6 @@ impl NavigationTarget {
ast::PatKind::BindPat(pat) => return NavigationTarget::from_bind_pat(file_id, &pat), ast::PatKind::BindPat(pat) => return NavigationTarget::from_bind_pat(file_id, &pat),
_ => ("_".into(), pat.syntax_node_ptr().range()), _ => ("_".into(), pat.syntax_node_ptr().range()),
}; };
let node = pat.to_node(file.syntax()).syntax().to_owned();
NavigationTarget { NavigationTarget {
file_id, file_id,
@ -101,20 +101,16 @@ impl NavigationTarget {
focus_range: None, focus_range: None,
kind: NAME, kind: NAME,
container_name: None, container_name: None,
description: description_inner(&node), description: None, //< No documentation for Description
docs: None, //< No documentation for Pattern docs: None, //< No documentation for Pattern
} }
} }
pub(crate) fn from_self_param( pub(crate) fn from_self_param(
db: &RootDatabase,
file_id: FileId, file_id: FileId,
par: AstPtr<ast::SelfParam>, par: AstPtr<ast::SelfParam>,
) -> NavigationTarget { ) -> NavigationTarget {
let (name, full_range) = ("self".into(), par.syntax_node_ptr().range()); let (name, full_range) = ("self".into(), par.syntax_node_ptr().range());
let file = db.parse(file_id).tree;
let ast = par.to_node(file.syntax());
let node = ast.syntax().to_owned();
NavigationTarget { NavigationTarget {
file_id, file_id,
@ -123,8 +119,8 @@ impl NavigationTarget {
focus_range: None, focus_range: None,
kind: NAME, kind: NAME,
container_name: None, container_name: None,
description: description_inner(&node), description: None, //< No document node for SelfParam
docs: None, //< No document node for SelfParam docs: None, //< No document node for SelfParam
} }
} }
@ -134,7 +130,7 @@ impl NavigationTarget {
let name = module.name(db).map(|it| it.to_string().into()).unwrap_or_default(); let name = module.name(db).map(|it| it.to_string().into()).unwrap_or_default();
match source { match source {
ModuleSource::SourceFile(node) => { ModuleSource::SourceFile(node) => {
NavigationTarget::from_syntax(file_id, name, None, node.syntax(), None) NavigationTarget::from_syntax(file_id, name, None, node.syntax(), None, None)
} }
ModuleSource::Module(node) => NavigationTarget::from_syntax( ModuleSource::Module(node) => NavigationTarget::from_syntax(
file_id, file_id,
@ -142,6 +138,7 @@ impl NavigationTarget {
None, None,
node.syntax(), node.syntax(),
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
), ),
} }
} }
@ -156,6 +153,7 @@ impl NavigationTarget {
None, None,
source.syntax(), source.syntax(),
source.doc_comment_text(), source.doc_comment_text(),
source.description(),
); );
} }
NavigationTarget::from_module(db, module) NavigationTarget::from_module(db, module)
@ -163,7 +161,12 @@ impl NavigationTarget {
pub(crate) fn from_function(db: &RootDatabase, func: hir::Function) -> NavigationTarget { pub(crate) fn from_function(db: &RootDatabase, func: hir::Function) -> NavigationTarget {
let (file_id, fn_def) = func.source(db); let (file_id, fn_def) = func.source(db);
NavigationTarget::from_named(file_id.original_file(db), &*fn_def, fn_def.doc_comment_text()) NavigationTarget::from_named(
file_id.original_file(db),
&*fn_def,
fn_def.doc_comment_text(),
fn_def.description(),
)
} }
pub(crate) fn from_field(db: &RootDatabase, field: hir::StructField) -> NavigationTarget { pub(crate) fn from_field(db: &RootDatabase, field: hir::StructField) -> NavigationTarget {
@ -171,10 +174,10 @@ impl NavigationTarget {
let file_id = file_id.original_file(db); let file_id = file_id.original_file(db);
match field { match field {
FieldSource::Named(it) => { FieldSource::Named(it) => {
NavigationTarget::from_named(file_id, &*it, it.doc_comment_text()) NavigationTarget::from_named(file_id, &*it, it.doc_comment_text(), it.description())
} }
FieldSource::Pos(it) => { FieldSource::Pos(it) => {
NavigationTarget::from_syntax(file_id, "".into(), None, it.syntax(), None) NavigationTarget::from_syntax(file_id, "".into(), None, it.syntax(), None, None)
} }
} }
} }
@ -187,6 +190,7 @@ impl NavigationTarget {
file_id.original_file(db), file_id.original_file(db),
&*node, &*node,
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
) )
} }
hir::AdtDef::Union(s) => { hir::AdtDef::Union(s) => {
@ -195,6 +199,7 @@ impl NavigationTarget {
file_id.original_file(db), file_id.original_file(db),
&*node, &*node,
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
) )
} }
hir::AdtDef::Enum(s) => { hir::AdtDef::Enum(s) => {
@ -203,6 +208,7 @@ impl NavigationTarget {
file_id.original_file(db), file_id.original_file(db),
&*node, &*node,
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
) )
} }
} }
@ -221,6 +227,7 @@ impl NavigationTarget {
file_id.original_file(db), file_id.original_file(db),
&*node, &*node,
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
) )
} }
hir::ModuleDef::Union(s) => { hir::ModuleDef::Union(s) => {
@ -229,6 +236,7 @@ impl NavigationTarget {
file_id.original_file(db), file_id.original_file(db),
&*node, &*node,
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
) )
} }
hir::ModuleDef::Const(s) => { hir::ModuleDef::Const(s) => {
@ -237,6 +245,7 @@ impl NavigationTarget {
file_id.original_file(db), file_id.original_file(db),
&*node, &*node,
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
) )
} }
hir::ModuleDef::Static(s) => { hir::ModuleDef::Static(s) => {
@ -245,6 +254,7 @@ impl NavigationTarget {
file_id.original_file(db), file_id.original_file(db),
&*node, &*node,
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
) )
} }
hir::ModuleDef::Enum(e) => { hir::ModuleDef::Enum(e) => {
@ -253,6 +263,7 @@ impl NavigationTarget {
file_id.original_file(db), file_id.original_file(db),
&*node, &*node,
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
) )
} }
hir::ModuleDef::EnumVariant(var) => { hir::ModuleDef::EnumVariant(var) => {
@ -261,6 +272,7 @@ impl NavigationTarget {
file_id.original_file(db), file_id.original_file(db),
&*node, &*node,
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
) )
} }
hir::ModuleDef::Trait(e) => { hir::ModuleDef::Trait(e) => {
@ -269,6 +281,7 @@ impl NavigationTarget {
file_id.original_file(db), file_id.original_file(db),
&*node, &*node,
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
) )
} }
hir::ModuleDef::TypeAlias(e) => { hir::ModuleDef::TypeAlias(e) => {
@ -277,6 +290,7 @@ impl NavigationTarget {
file_id.original_file(db), file_id.original_file(db),
&*node, &*node,
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
) )
} }
hir::ModuleDef::BuiltinType(..) => { hir::ModuleDef::BuiltinType(..) => {
@ -297,6 +311,7 @@ impl NavigationTarget {
None, None,
node.syntax(), node.syntax(),
None, None,
None,
) )
} }
@ -309,6 +324,7 @@ impl NavigationTarget {
file_id.original_file(db), file_id.original_file(db),
&*node, &*node,
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
) )
} }
ImplItem::TypeAlias(a) => { ImplItem::TypeAlias(a) => {
@ -317,6 +333,7 @@ impl NavigationTarget {
file_id.original_file(db), file_id.original_file(db),
&*node, &*node,
node.doc_comment_text(), node.doc_comment_text(),
node.description(),
) )
} }
} }
@ -325,7 +342,12 @@ impl NavigationTarget {
pub(crate) fn from_macro_def(db: &RootDatabase, macro_call: hir::MacroDef) -> NavigationTarget { pub(crate) fn from_macro_def(db: &RootDatabase, macro_call: hir::MacroDef) -> NavigationTarget {
let (file_id, node) = macro_call.source(db); let (file_id, node) = macro_call.source(db);
log::debug!("nav target {}", node.syntax().debug_dump()); log::debug!("nav target {}", node.syntax().debug_dump());
NavigationTarget::from_named(file_id.original_file(db), &*node, node.doc_comment_text()) NavigationTarget::from_named(
file_id.original_file(db),
&*node,
node.doc_comment_text(),
None,
)
} }
#[cfg(test)] #[cfg(test)]
@ -357,11 +379,12 @@ impl NavigationTarget {
file_id: FileId, file_id: FileId,
node: &impl ast::NameOwner, node: &impl ast::NameOwner,
docs: Option<String>, docs: Option<String>,
description: Option<String>,
) -> NavigationTarget { ) -> NavigationTarget {
//FIXME: use `_` instead of empty string //FIXME: use `_` instead of empty string
let name = node.name().map(|it| it.text().clone()).unwrap_or_default(); let name = node.name().map(|it| it.text().clone()).unwrap_or_default();
let focus_range = node.name().map(|it| it.syntax().range()); let focus_range = node.name().map(|it| it.syntax().range());
NavigationTarget::from_syntax(file_id, name, focus_range, node.syntax(), docs) NavigationTarget::from_syntax(file_id, name, focus_range, node.syntax(), docs, description)
} }
fn from_syntax( fn from_syntax(
@ -370,6 +393,7 @@ impl NavigationTarget {
focus_range: Option<TextRange>, focus_range: Option<TextRange>,
node: &SyntaxNode, node: &SyntaxNode,
docs: Option<String>, docs: Option<String>,
description: Option<String>,
) -> NavigationTarget { ) -> NavigationTarget {
NavigationTarget { NavigationTarget {
file_id, file_id,
@ -379,7 +403,7 @@ impl NavigationTarget {
focus_range, focus_range,
// ptr: Some(LocalSyntaxPtr::new(node)), // ptr: Some(LocalSyntaxPtr::new(node)),
container_name: None, container_name: None,
description: description_inner(node), description,
docs, docs,
} }
} }
@ -408,46 +432,22 @@ fn docs_from_symbol(db: &RootDatabase, symbol: &FileSymbol) -> Option<String> {
.accept(&node)? .accept(&node)?
} }
/// Get a description of a node. /// Get a description of a symbol.
/// ///
/// e.g. `struct Name`, `enum Name`, `fn Name` /// e.g. `struct Name`, `enum Name`, `fn Name`
fn description_inner(node: &SyntaxNode) -> Option<String> { fn description_from_symbol(node: &SyntaxNode) -> Option<String> {
// FIXME: After type inference is done, add type information to improve the output // FIXME: After type inference is done, add type information to improve the output
fn visit_ascribed_node<T>(node: &T, prefix: &str) -> Option<String>
where
T: NameOwner + VisibilityOwner + TypeAscriptionOwner,
{
let mut string = visit_node(node, prefix)?;
if let Some(type_ref) = node.ascribed_type() {
string.push_str(": ");
type_ref.syntax().text().push_to(&mut string);
}
Some(string)
}
fn visit_node<T>(node: &T, label: &str) -> Option<String>
where
T: NameOwner + VisibilityOwner,
{
let mut string =
node.visibility().map(|v| format!("{} ", v.syntax().text())).unwrap_or_default();
string.push_str(label);
string.push_str(node.name()?.text().as_str());
Some(string)
}
visitor() visitor()
.visit(|node: &ast::FnDef| Some(crate::display::function_label(node))) .visit(|node: &ast::FnDef| node.description())
.visit(|node: &ast::StructDef| visit_node(node, "struct ")) .visit(|node: &ast::StructDef| node.description())
.visit(|node: &ast::EnumDef| visit_node(node, "enum ")) .visit(|node: &ast::EnumDef| node.description())
.visit(|node: &ast::TraitDef| visit_node(node, "trait ")) .visit(|node: &ast::TraitDef| node.description())
.visit(|node: &ast::Module| visit_node(node, "mod ")) .visit(|node: &ast::Module| node.description())
.visit(|node: &ast::TypeAliasDef| visit_node(node, "type ")) .visit(|node: &ast::TypeAliasDef| node.description())
.visit(|node: &ast::ConstDef| visit_ascribed_node(node, "const ")) .visit(|node: &ast::ConstDef| node.description())
.visit(|node: &ast::StaticDef| visit_ascribed_node(node, "static ")) .visit(|node: &ast::StaticDef| node.description())
.visit(|node: &ast::NamedFieldDef| visit_ascribed_node(node, "")) .visit(|node: &ast::NamedFieldDef| node.description())
.visit(|node: &ast::EnumVariant| Some(node.name()?.text().to_string())) .visit(|node: &ast::EnumVariant| node.description())
.accept(&node)? .accept(&node)?
} }

View File

@ -13,6 +13,7 @@ use crate::{
db::RootDatabase, db::RootDatabase,
RangeInfo, RangeInfo,
name_ref_kind::{NameRefKind::*, classify_name_ref}, name_ref_kind::{NameRefKind::*, classify_name_ref},
display::Description,
}; };
pub(crate) fn goto_definition( pub(crate) fn goto_definition(
@ -72,7 +73,7 @@ pub(crate) fn reference_definition(
} }
} }
Some(Pat(pat)) => return Exact(NavigationTarget::from_pat(db, file_id, pat)), Some(Pat(pat)) => return Exact(NavigationTarget::from_pat(db, file_id, pat)),
Some(SelfParam(par)) => return Exact(NavigationTarget::from_self_param(db, file_id, par)), Some(SelfParam(par)) => return Exact(NavigationTarget::from_self_param(file_id, par)),
Some(GenericParam(_)) => { Some(GenericParam(_)) => {
// FIXME: go to the generic param def // FIXME: go to the generic param def
} }
@ -115,37 +116,37 @@ pub(crate) fn name_definition(
fn named_target(file_id: FileId, node: &SyntaxNode) -> Option<NavigationTarget> { fn named_target(file_id: FileId, node: &SyntaxNode) -> Option<NavigationTarget> {
visitor() visitor()
.visit(|node: &ast::StructDef| { .visit(|node: &ast::StructDef| {
NavigationTarget::from_named(file_id, node, node.doc_comment_text()) NavigationTarget::from_named(file_id, node, node.doc_comment_text(), node.description())
}) })
.visit(|node: &ast::EnumDef| { .visit(|node: &ast::EnumDef| {
NavigationTarget::from_named(file_id, node, node.doc_comment_text()) NavigationTarget::from_named(file_id, node, node.doc_comment_text(), node.description())
}) })
.visit(|node: &ast::EnumVariant| { .visit(|node: &ast::EnumVariant| {
NavigationTarget::from_named(file_id, node, node.doc_comment_text()) NavigationTarget::from_named(file_id, node, node.doc_comment_text(), node.description())
}) })
.visit(|node: &ast::FnDef| { .visit(|node: &ast::FnDef| {
NavigationTarget::from_named(file_id, node, node.doc_comment_text()) NavigationTarget::from_named(file_id, node, node.doc_comment_text(), node.description())
}) })
.visit(|node: &ast::TypeAliasDef| { .visit(|node: &ast::TypeAliasDef| {
NavigationTarget::from_named(file_id, node, node.doc_comment_text()) NavigationTarget::from_named(file_id, node, node.doc_comment_text(), node.description())
}) })
.visit(|node: &ast::ConstDef| { .visit(|node: &ast::ConstDef| {
NavigationTarget::from_named(file_id, node, node.doc_comment_text()) NavigationTarget::from_named(file_id, node, node.doc_comment_text(), node.description())
}) })
.visit(|node: &ast::StaticDef| { .visit(|node: &ast::StaticDef| {
NavigationTarget::from_named(file_id, node, node.doc_comment_text()) NavigationTarget::from_named(file_id, node, node.doc_comment_text(), node.description())
}) })
.visit(|node: &ast::TraitDef| { .visit(|node: &ast::TraitDef| {
NavigationTarget::from_named(file_id, node, node.doc_comment_text()) NavigationTarget::from_named(file_id, node, node.doc_comment_text(), node.description())
}) })
.visit(|node: &ast::NamedFieldDef| { .visit(|node: &ast::NamedFieldDef| {
NavigationTarget::from_named(file_id, node, node.doc_comment_text()) NavigationTarget::from_named(file_id, node, node.doc_comment_text(), node.description())
}) })
.visit(|node: &ast::Module| { .visit(|node: &ast::Module| {
NavigationTarget::from_named(file_id, node, node.doc_comment_text()) NavigationTarget::from_named(file_id, node, node.doc_comment_text(), node.description())
}) })
.visit(|node: &ast::MacroCall| { .visit(|node: &ast::MacroCall| {
NavigationTarget::from_named(file_id, node, node.doc_comment_text()) NavigationTarget::from_named(file_id, node, node.doc_comment_text(), None)
}) })
.accept(node) .accept(node)
} }