diff options
Diffstat (limited to 'crates/ra_hir_def/src/docs.rs')
-rw-r--r-- | crates/ra_hir_def/src/docs.rs | 43 |
1 files changed, 41 insertions, 2 deletions
diff --git a/crates/ra_hir_def/src/docs.rs b/crates/ra_hir_def/src/docs.rs index b221ae1ce..74b9f8199 100644 --- a/crates/ra_hir_def/src/docs.rs +++ b/crates/ra_hir_def/src/docs.rs | |||
@@ -70,6 +70,45 @@ impl Documentation { | |||
70 | } | 70 | } |
71 | } | 71 | } |
72 | 72 | ||
73 | pub(crate) fn docs_from_ast(node: &impl ast::DocCommentsOwner) -> Option<Documentation> { | 73 | pub(crate) fn docs_from_ast<N>(node: &N) -> Option<Documentation> |
74 | node.doc_comment_text().map(|it| Documentation::new(&it)) | 74 | where |
75 | N: ast::DocCommentsOwner + ast::AttrsOwner, | ||
76 | { | ||
77 | let doc_comment_text = node.doc_comment_text(); | ||
78 | let doc_attr_text = expand_doc_attrs(node); | ||
79 | let docs = merge_doc_comments_and_attrs(doc_comment_text, doc_attr_text); | ||
80 | docs.map(|it| Documentation::new(&it)) | ||
81 | } | ||
82 | |||
83 | fn merge_doc_comments_and_attrs( | ||
84 | doc_comment_text: Option<String>, | ||
85 | doc_attr_text: Option<String>, | ||
86 | ) -> Option<String> { | ||
87 | match (doc_comment_text, doc_attr_text) { | ||
88 | (Some(mut comment_text), Some(attr_text)) => { | ||
89 | comment_text.push_str("\n\n"); | ||
90 | comment_text.push_str(&attr_text); | ||
91 | Some(comment_text) | ||
92 | } | ||
93 | (Some(comment_text), None) => Some(comment_text), | ||
94 | (None, Some(attr_text)) => Some(attr_text), | ||
95 | (None, None) => None, | ||
96 | } | ||
97 | } | ||
98 | |||
99 | fn expand_doc_attrs(owner: &dyn ast::AttrsOwner) -> Option<String> { | ||
100 | let mut docs = String::new(); | ||
101 | for attr in owner.attrs() { | ||
102 | if let Some(("doc", value)) = | ||
103 | attr.as_simple_key_value().as_ref().map(|(k, v)| (k.as_str(), v.as_str())) | ||
104 | { | ||
105 | docs.push_str(value); | ||
106 | docs.push_str("\n\n"); | ||
107 | } | ||
108 | } | ||
109 | if docs.is_empty() { | ||
110 | None | ||
111 | } else { | ||
112 | Some(docs) | ||
113 | } | ||
75 | } | 114 | } |