diff options
Diffstat (limited to 'crates/ide_completion')
22 files changed, 1714 insertions, 873 deletions
diff --git a/crates/ide_completion/Cargo.toml b/crates/ide_completion/Cargo.toml index 6bd8a5500..ba81c9e04 100644 --- a/crates/ide_completion/Cargo.toml +++ b/crates/ide_completion/Cargo.toml | |||
@@ -15,6 +15,7 @@ itertools = "0.10.0" | |||
15 | log = "0.4.8" | 15 | log = "0.4.8" |
16 | rustc-hash = "1.1.0" | 16 | rustc-hash = "1.1.0" |
17 | either = "1.6.1" | 17 | either = "1.6.1" |
18 | once_cell = "1.7" | ||
18 | 19 | ||
19 | stdx = { path = "../stdx", version = "0.0.0" } | 20 | stdx = { path = "../stdx", version = "0.0.0" } |
20 | syntax = { path = "../syntax", version = "0.0.0" } | 21 | syntax = { path = "../syntax", version = "0.0.0" } |
diff --git a/crates/ide_completion/src/completions.rs b/crates/ide_completion/src/completions.rs index 78154bf3e..151bf3783 100644 --- a/crates/ide_completion/src/completions.rs +++ b/crates/ide_completion/src/completions.rs | |||
@@ -18,7 +18,7 @@ pub(crate) mod unqualified_path; | |||
18 | 18 | ||
19 | use std::iter; | 19 | use std::iter; |
20 | 20 | ||
21 | use hir::{known, ModPath, ScopeDef, Type}; | 21 | use hir::known; |
22 | use ide_db::SymbolKind; | 22 | use ide_db::SymbolKind; |
23 | 23 | ||
24 | use crate::{ | 24 | use crate::{ |
@@ -69,12 +69,17 @@ impl Completions { | |||
69 | items.into_iter().for_each(|item| self.add(item.into())) | 69 | items.into_iter().for_each(|item| self.add(item.into())) |
70 | } | 70 | } |
71 | 71 | ||
72 | pub(crate) fn add_field(&mut self, ctx: &CompletionContext, field: hir::Field, ty: &Type) { | 72 | pub(crate) fn add_field(&mut self, ctx: &CompletionContext, field: hir::Field, ty: &hir::Type) { |
73 | let item = render_field(RenderContext::new(ctx), field, ty); | 73 | let item = render_field(RenderContext::new(ctx), field, ty); |
74 | self.add(item); | 74 | self.add(item); |
75 | } | 75 | } |
76 | 76 | ||
77 | pub(crate) fn add_tuple_field(&mut self, ctx: &CompletionContext, field: usize, ty: &Type) { | 77 | pub(crate) fn add_tuple_field( |
78 | &mut self, | ||
79 | ctx: &CompletionContext, | ||
80 | field: usize, | ||
81 | ty: &hir::Type, | ||
82 | ) { | ||
78 | let item = render_tuple_field(RenderContext::new(ctx), field, ty); | 83 | let item = render_tuple_field(RenderContext::new(ctx), field, ty); |
79 | self.add(item); | 84 | self.add(item); |
80 | } | 85 | } |
@@ -89,8 +94,8 @@ impl Completions { | |||
89 | pub(crate) fn add_resolution( | 94 | pub(crate) fn add_resolution( |
90 | &mut self, | 95 | &mut self, |
91 | ctx: &CompletionContext, | 96 | ctx: &CompletionContext, |
92 | local_name: String, | 97 | local_name: hir::Name, |
93 | resolution: &ScopeDef, | 98 | resolution: &hir::ScopeDef, |
94 | ) { | 99 | ) { |
95 | if let Some(item) = render_resolution(RenderContext::new(ctx), local_name, resolution) { | 100 | if let Some(item) = render_resolution(RenderContext::new(ctx), local_name, resolution) { |
96 | self.add(item); | 101 | self.add(item); |
@@ -100,7 +105,7 @@ impl Completions { | |||
100 | pub(crate) fn add_macro( | 105 | pub(crate) fn add_macro( |
101 | &mut self, | 106 | &mut self, |
102 | ctx: &CompletionContext, | 107 | ctx: &CompletionContext, |
103 | name: Option<String>, | 108 | name: Option<hir::Name>, |
104 | macro_: hir::MacroDef, | 109 | macro_: hir::MacroDef, |
105 | ) { | 110 | ) { |
106 | let name = match name { | 111 | let name = match name { |
@@ -116,7 +121,7 @@ impl Completions { | |||
116 | &mut self, | 121 | &mut self, |
117 | ctx: &CompletionContext, | 122 | ctx: &CompletionContext, |
118 | func: hir::Function, | 123 | func: hir::Function, |
119 | local_name: Option<String>, | 124 | local_name: Option<hir::Name>, |
120 | ) { | 125 | ) { |
121 | if let Some(item) = render_fn(RenderContext::new(ctx), None, local_name, func) { | 126 | if let Some(item) = render_fn(RenderContext::new(ctx), None, local_name, func) { |
122 | self.add(item) | 127 | self.add(item) |
@@ -127,7 +132,7 @@ impl Completions { | |||
127 | &mut self, | 132 | &mut self, |
128 | ctx: &CompletionContext, | 133 | ctx: &CompletionContext, |
129 | func: hir::Function, | 134 | func: hir::Function, |
130 | local_name: Option<String>, | 135 | local_name: Option<hir::Name>, |
131 | ) { | 136 | ) { |
132 | if let Some(item) = render_method(RenderContext::new(ctx), None, local_name, func) { | 137 | if let Some(item) = render_method(RenderContext::new(ctx), None, local_name, func) { |
133 | self.add(item) | 138 | self.add(item) |
@@ -149,7 +154,7 @@ impl Completions { | |||
149 | &mut self, | 154 | &mut self, |
150 | ctx: &CompletionContext, | 155 | ctx: &CompletionContext, |
151 | variant: hir::Variant, | 156 | variant: hir::Variant, |
152 | path: ModPath, | 157 | path: hir::ModPath, |
153 | ) { | 158 | ) { |
154 | if let Some(item) = render_variant_pat(RenderContext::new(ctx), variant, None, Some(path)) { | 159 | if let Some(item) = render_variant_pat(RenderContext::new(ctx), variant, None, Some(path)) { |
155 | self.add(item); | 160 | self.add(item); |
@@ -183,7 +188,7 @@ impl Completions { | |||
183 | &mut self, | 188 | &mut self, |
184 | ctx: &CompletionContext, | 189 | ctx: &CompletionContext, |
185 | variant: hir::Variant, | 190 | variant: hir::Variant, |
186 | path: ModPath, | 191 | path: hir::ModPath, |
187 | ) { | 192 | ) { |
188 | let item = render_variant(RenderContext::new(ctx), None, None, variant, Some(path)); | 193 | let item = render_variant(RenderContext::new(ctx), None, None, variant, Some(path)); |
189 | self.add(item); | 194 | self.add(item); |
@@ -193,7 +198,7 @@ impl Completions { | |||
193 | &mut self, | 198 | &mut self, |
194 | ctx: &CompletionContext, | 199 | ctx: &CompletionContext, |
195 | variant: hir::Variant, | 200 | variant: hir::Variant, |
196 | local_name: Option<String>, | 201 | local_name: Option<hir::Name>, |
197 | ) { | 202 | ) { |
198 | let item = render_variant(RenderContext::new(ctx), None, local_name, variant, None); | 203 | let item = render_variant(RenderContext::new(ctx), None, local_name, variant, None); |
199 | self.add(item); | 204 | self.add(item); |
diff --git a/crates/ide_completion/src/completions/attribute.rs b/crates/ide_completion/src/completions/attribute.rs index b1505c74b..76d926157 100644 --- a/crates/ide_completion/src/completions/attribute.rs +++ b/crates/ide_completion/src/completions/attribute.rs | |||
@@ -3,9 +3,9 @@ | |||
3 | //! This module uses a bit of static metadata to provide completions | 3 | //! This module uses a bit of static metadata to provide completions |
4 | //! for built-in attributes. | 4 | //! for built-in attributes. |
5 | 5 | ||
6 | use itertools::Itertools; | 6 | use once_cell::sync::Lazy; |
7 | use rustc_hash::FxHashSet; | 7 | use rustc_hash::{FxHashMap, FxHashSet}; |
8 | use syntax::{ast, AstNode, T}; | 8 | use syntax::{ast, AstNode, NodeOrToken, SyntaxKind, T}; |
9 | 9 | ||
10 | use crate::{ | 10 | use crate::{ |
11 | context::CompletionContext, | 11 | context::CompletionContext, |
@@ -14,33 +14,40 @@ use crate::{ | |||
14 | Completions, | 14 | Completions, |
15 | }; | 15 | }; |
16 | 16 | ||
17 | pub(crate) fn complete_attribute(acc: &mut Completions, ctx: &CompletionContext) -> Option<()> { | 17 | mod derive; |
18 | if ctx.mod_declaration_under_caret.is_some() { | 18 | mod lint; |
19 | return None; | 19 | pub(crate) use self::lint::LintCompletion; |
20 | } | ||
21 | 20 | ||
21 | pub(crate) fn complete_attribute(acc: &mut Completions, ctx: &CompletionContext) -> Option<()> { | ||
22 | let attribute = ctx.attribute_under_caret.as_ref()?; | 22 | let attribute = ctx.attribute_under_caret.as_ref()?; |
23 | match (attribute.path(), attribute.token_tree()) { | 23 | match (attribute.path().and_then(|p| p.as_single_name_ref()), attribute.token_tree()) { |
24 | (Some(path), Some(token_tree)) => { | 24 | (Some(path), Some(token_tree)) => match path.text().as_str() { |
25 | let path = path.syntax().text(); | 25 | "derive" => derive::complete_derive(acc, ctx, token_tree), |
26 | if path == "derive" { | 26 | "feature" => lint::complete_lint(acc, ctx, token_tree, FEATURES), |
27 | complete_derive(acc, ctx, token_tree) | 27 | "allow" | "warn" | "deny" | "forbid" => { |
28 | } else if path == "feature" { | 28 | lint::complete_lint(acc, ctx, token_tree.clone(), lint::DEFAULT_LINT_COMPLETIONS); |
29 | complete_lint(acc, ctx, token_tree, FEATURES) | 29 | lint::complete_lint(acc, ctx, token_tree, CLIPPY_LINTS); |
30 | } else if path == "allow" || path == "warn" || path == "deny" || path == "forbid" { | ||
31 | complete_lint(acc, ctx, token_tree.clone(), DEFAULT_LINT_COMPLETIONS); | ||
32 | complete_lint(acc, ctx, token_tree, CLIPPY_LINTS); | ||
33 | } | 30 | } |
34 | } | 31 | _ => (), |
35 | (_, Some(_token_tree)) => {} | 32 | }, |
36 | _ => complete_attribute_start(acc, ctx, attribute), | 33 | (None, Some(_)) => (), |
34 | _ => complete_new_attribute(acc, ctx, attribute), | ||
37 | } | 35 | } |
38 | Some(()) | 36 | Some(()) |
39 | } | 37 | } |
40 | 38 | ||
41 | fn complete_attribute_start(acc: &mut Completions, ctx: &CompletionContext, attribute: &ast::Attr) { | 39 | fn complete_new_attribute(acc: &mut Completions, ctx: &CompletionContext, attribute: &ast::Attr) { |
40 | let attribute_annotated_item_kind = attribute.syntax().parent().map(|it| it.kind()); | ||
41 | let attributes = attribute_annotated_item_kind.and_then(|kind| { | ||
42 | if ast::Expr::can_cast(kind) { | ||
43 | Some(EXPR_ATTRIBUTES) | ||
44 | } else { | ||
45 | KIND_TO_ATTRIBUTES.get(&kind).copied() | ||
46 | } | ||
47 | }); | ||
42 | let is_inner = attribute.kind() == ast::AttrKind::Inner; | 48 | let is_inner = attribute.kind() == ast::AttrKind::Inner; |
43 | for attr_completion in ATTRIBUTES.iter().filter(|compl| is_inner || !compl.prefer_inner) { | 49 | |
50 | let add_completion = |attr_completion: &AttrCompletion| { | ||
44 | let mut item = CompletionItem::new( | 51 | let mut item = CompletionItem::new( |
45 | CompletionKind::Attribute, | 52 | CompletionKind::Attribute, |
46 | ctx.source_range(), | 53 | ctx.source_range(), |
@@ -56,9 +63,19 @@ fn complete_attribute_start(acc: &mut Completions, ctx: &CompletionContext, attr | |||
56 | item.insert_snippet(cap, snippet); | 63 | item.insert_snippet(cap, snippet); |
57 | } | 64 | } |
58 | 65 | ||
59 | if attribute.kind() == ast::AttrKind::Inner || !attr_completion.prefer_inner { | 66 | if is_inner || !attr_completion.prefer_inner { |
60 | acc.add(item.build()); | 67 | acc.add(item.build()); |
61 | } | 68 | } |
69 | }; | ||
70 | |||
71 | match attributes { | ||
72 | Some(applicable) => applicable | ||
73 | .iter() | ||
74 | .flat_map(|name| ATTRIBUTES.binary_search_by(|attr| attr.key().cmp(name)).ok()) | ||
75 | .flat_map(|idx| ATTRIBUTES.get(idx)) | ||
76 | .for_each(add_completion), | ||
77 | None if is_inner => ATTRIBUTES.iter().for_each(add_completion), | ||
78 | None => ATTRIBUTES.iter().filter(|compl| !compl.prefer_inner).for_each(add_completion), | ||
62 | } | 79 | } |
63 | } | 80 | } |
64 | 81 | ||
@@ -70,6 +87,10 @@ struct AttrCompletion { | |||
70 | } | 87 | } |
71 | 88 | ||
72 | impl AttrCompletion { | 89 | impl AttrCompletion { |
90 | fn key(&self) -> &'static str { | ||
91 | self.lookup.unwrap_or(self.label) | ||
92 | } | ||
93 | |||
73 | const fn prefer_inner(self) -> AttrCompletion { | 94 | const fn prefer_inner(self) -> AttrCompletion { |
74 | AttrCompletion { prefer_inner: true, ..self } | 95 | AttrCompletion { prefer_inner: true, ..self } |
75 | } | 96 | } |
@@ -83,30 +104,122 @@ const fn attr( | |||
83 | AttrCompletion { label, lookup, snippet, prefer_inner: false } | 104 | AttrCompletion { label, lookup, snippet, prefer_inner: false } |
84 | } | 105 | } |
85 | 106 | ||
107 | macro_rules! attrs { | ||
108 | // attributes applicable to all items | ||
109 | [@ { item $($tt:tt)* } {$($acc:tt)*}] => { | ||
110 | attrs!(@ { $($tt)* } { $($acc)*, "deprecated", "doc", "dochidden", "docalias", "must_use", "no_mangle" }) | ||
111 | }; | ||
112 | // attributes applicable to all adts | ||
113 | [@ { adt $($tt:tt)* } {$($acc:tt)*}] => { | ||
114 | attrs!(@ { $($tt)* } { $($acc)*, "derive", "repr" }) | ||
115 | }; | ||
116 | // attributes applicable to all linkable things aka functions/statics | ||
117 | [@ { linkable $($tt:tt)* } {$($acc:tt)*}] => { | ||
118 | attrs!(@ { $($tt)* } { $($acc)*, "export_name", "link_name", "link_section" }) | ||
119 | }; | ||
120 | // error fallback for nicer error message | ||
121 | [@ { $ty:ident $($tt:tt)* } {$($acc:tt)*}] => { | ||
122 | compile_error!(concat!("unknown attr subtype ", stringify!($ty))) | ||
123 | }; | ||
124 | // general push down accumulation | ||
125 | [@ { $lit:literal $($tt:tt)*} {$($acc:tt)*}] => { | ||
126 | attrs!(@ { $($tt)* } { $($acc)*, $lit }) | ||
127 | }; | ||
128 | [@ {$($tt:tt)+} {$($tt2:tt)*}] => { | ||
129 | compile_error!(concat!("Unexpected input ", stringify!($($tt)+))) | ||
130 | }; | ||
131 | // final output construction | ||
132 | [@ {} {$($tt:tt)*}] => { &[$($tt)*] as _ }; | ||
133 | // starting matcher | ||
134 | [$($tt:tt),*] => { | ||
135 | attrs!(@ { $($tt)* } { "allow", "cfg", "cfg_attr", "deny", "forbid", "warn" }) | ||
136 | }; | ||
137 | } | ||
138 | |||
139 | #[rustfmt::skip] | ||
140 | static KIND_TO_ATTRIBUTES: Lazy<FxHashMap<SyntaxKind, &[&str]>> = Lazy::new(|| { | ||
141 | use SyntaxKind::*; | ||
142 | std::array::IntoIter::new([ | ||
143 | ( | ||
144 | SOURCE_FILE, | ||
145 | attrs!( | ||
146 | item, | ||
147 | "crate_name", "feature", "no_implicit_prelude", "no_main", "no_std", | ||
148 | "recursion_limit", "type_length_limit", "windows_subsystem" | ||
149 | ), | ||
150 | ), | ||
151 | (MODULE, attrs!(item, "no_implicit_prelude", "path")), | ||
152 | (ITEM_LIST, attrs!(item, "no_implicit_prelude")), | ||
153 | (MACRO_RULES, attrs!(item, "macro_export", "macro_use")), | ||
154 | (MACRO_DEF, attrs!(item)), | ||
155 | (EXTERN_CRATE, attrs!(item, "macro_use", "no_link")), | ||
156 | (USE, attrs!(item)), | ||
157 | (TYPE_ALIAS, attrs!(item)), | ||
158 | (STRUCT, attrs!(item, adt, "non_exhaustive")), | ||
159 | (ENUM, attrs!(item, adt, "non_exhaustive")), | ||
160 | (UNION, attrs!(item, adt)), | ||
161 | (CONST, attrs!(item)), | ||
162 | ( | ||
163 | FN, | ||
164 | attrs!( | ||
165 | item, linkable, | ||
166 | "cold", "ignore", "inline", "must_use", "panic_handler", "proc_macro", | ||
167 | "proc_macro_derive", "proc_macro_attribute", "should_panic", "target_feature", | ||
168 | "test", "track_caller" | ||
169 | ), | ||
170 | ), | ||
171 | (STATIC, attrs!(item, linkable, "global_allocator", "used")), | ||
172 | (TRAIT, attrs!(item, "must_use")), | ||
173 | (IMPL, attrs!(item, "automatically_derived")), | ||
174 | (ASSOC_ITEM_LIST, attrs!(item)), | ||
175 | (EXTERN_BLOCK, attrs!(item, "link")), | ||
176 | (EXTERN_ITEM_LIST, attrs!(item, "link")), | ||
177 | (MACRO_CALL, attrs!()), | ||
178 | (SELF_PARAM, attrs!()), | ||
179 | (PARAM, attrs!()), | ||
180 | (RECORD_FIELD, attrs!()), | ||
181 | (VARIANT, attrs!("non_exhaustive")), | ||
182 | (TYPE_PARAM, attrs!()), | ||
183 | (CONST_PARAM, attrs!()), | ||
184 | (LIFETIME_PARAM, attrs!()), | ||
185 | (LET_STMT, attrs!()), | ||
186 | (EXPR_STMT, attrs!()), | ||
187 | (LITERAL, attrs!()), | ||
188 | (RECORD_EXPR_FIELD_LIST, attrs!()), | ||
189 | (RECORD_EXPR_FIELD, attrs!()), | ||
190 | (MATCH_ARM_LIST, attrs!()), | ||
191 | (MATCH_ARM, attrs!()), | ||
192 | (IDENT_PAT, attrs!()), | ||
193 | (RECORD_PAT_FIELD, attrs!()), | ||
194 | ]) | ||
195 | .collect() | ||
196 | }); | ||
197 | const EXPR_ATTRIBUTES: &[&str] = attrs!(); | ||
198 | |||
86 | /// https://doc.rust-lang.org/reference/attributes.html#built-in-attributes-index | 199 | /// https://doc.rust-lang.org/reference/attributes.html#built-in-attributes-index |
200 | // Keep these sorted for the binary search! | ||
87 | const ATTRIBUTES: &[AttrCompletion] = &[ | 201 | const ATTRIBUTES: &[AttrCompletion] = &[ |
88 | attr("allow(…)", Some("allow"), Some("allow(${0:lint})")), | 202 | attr("allow(…)", Some("allow"), Some("allow(${0:lint})")), |
89 | attr("automatically_derived", None, None), | 203 | attr("automatically_derived", None, None), |
90 | attr("cfg_attr(…)", Some("cfg_attr"), Some("cfg_attr(${1:predicate}, ${0:attr})")), | ||
91 | attr("cfg(…)", Some("cfg"), Some("cfg(${0:predicate})")), | 204 | attr("cfg(…)", Some("cfg"), Some("cfg(${0:predicate})")), |
205 | attr("cfg_attr(…)", Some("cfg_attr"), Some("cfg_attr(${1:predicate}, ${0:attr})")), | ||
92 | attr("cold", None, None), | 206 | attr("cold", None, None), |
93 | attr(r#"crate_name = """#, Some("crate_name"), Some(r#"crate_name = "${0:crate_name}""#)) | 207 | attr(r#"crate_name = """#, Some("crate_name"), Some(r#"crate_name = "${0:crate_name}""#)) |
94 | .prefer_inner(), | 208 | .prefer_inner(), |
95 | attr("deny(…)", Some("deny"), Some("deny(${0:lint})")), | 209 | attr("deny(…)", Some("deny"), Some("deny(${0:lint})")), |
96 | attr(r#"deprecated"#, Some("deprecated"), Some(r#"deprecated"#)), | 210 | attr(r#"deprecated"#, Some("deprecated"), Some(r#"deprecated"#)), |
97 | attr("derive(…)", Some("derive"), Some(r#"derive(${0:Debug})"#)), | 211 | attr("derive(…)", Some("derive"), Some(r#"derive(${0:Debug})"#)), |
212 | attr(r#"doc = "…""#, Some("doc"), Some(r#"doc = "${0:docs}""#)), | ||
213 | attr(r#"doc(alias = "…")"#, Some("docalias"), Some(r#"doc(alias = "${0:docs}")"#)), | ||
214 | attr(r#"doc(hidden)"#, Some("dochidden"), Some(r#"doc(hidden)"#)), | ||
98 | attr( | 215 | attr( |
99 | r#"export_name = "…""#, | 216 | r#"export_name = "…""#, |
100 | Some("export_name"), | 217 | Some("export_name"), |
101 | Some(r#"export_name = "${0:exported_symbol_name}""#), | 218 | Some(r#"export_name = "${0:exported_symbol_name}""#), |
102 | ), | 219 | ), |
103 | attr(r#"doc(alias = "…")"#, Some("docalias"), Some(r#"doc(alias = "${0:docs}")"#)), | ||
104 | attr(r#"doc = "…""#, Some("doc"), Some(r#"doc = "${0:docs}""#)), | ||
105 | attr(r#"doc(hidden)"#, Some("dochidden"), Some(r#"doc(hidden)"#)), | ||
106 | attr("feature(…)", Some("feature"), Some("feature(${0:flag})")).prefer_inner(), | 220 | attr("feature(…)", Some("feature"), Some("feature(${0:flag})")).prefer_inner(), |
107 | attr("forbid(…)", Some("forbid"), Some("forbid(${0:lint})")), | 221 | attr("forbid(…)", Some("forbid"), Some("forbid(${0:lint})")), |
108 | // FIXME: resolve through macro resolution? | 222 | attr("global_allocator", None, None), |
109 | attr("global_allocator", None, None).prefer_inner(), | ||
110 | attr(r#"ignore = "…""#, Some("ignore"), Some(r#"ignore = "${0:reason}""#)), | 223 | attr(r#"ignore = "…""#, Some("ignore"), Some(r#"ignore = "${0:reason}""#)), |
111 | attr("inline", Some("inline"), Some("inline")), | 224 | attr("inline", Some("inline"), Some("inline")), |
112 | attr("link", None, None), | 225 | attr("link", None, None), |
@@ -119,13 +232,13 @@ const ATTRIBUTES: &[AttrCompletion] = &[ | |||
119 | attr("macro_export", None, None), | 232 | attr("macro_export", None, None), |
120 | attr("macro_use", None, None), | 233 | attr("macro_use", None, None), |
121 | attr(r#"must_use"#, Some("must_use"), Some(r#"must_use"#)), | 234 | attr(r#"must_use"#, Some("must_use"), Some(r#"must_use"#)), |
122 | attr("no_link", None, None).prefer_inner(), | ||
123 | attr("no_implicit_prelude", None, None).prefer_inner(), | 235 | attr("no_implicit_prelude", None, None).prefer_inner(), |
236 | attr("no_link", None, None).prefer_inner(), | ||
124 | attr("no_main", None, None).prefer_inner(), | 237 | attr("no_main", None, None).prefer_inner(), |
125 | attr("no_mangle", None, None), | 238 | attr("no_mangle", None, None), |
126 | attr("no_std", None, None).prefer_inner(), | 239 | attr("no_std", None, None).prefer_inner(), |
127 | attr("non_exhaustive", None, None), | 240 | attr("non_exhaustive", None, None), |
128 | attr("panic_handler", None, None).prefer_inner(), | 241 | attr("panic_handler", None, None), |
129 | attr(r#"path = "…""#, Some("path"), Some(r#"path ="${0:path}""#)), | 242 | attr(r#"path = "…""#, Some("path"), Some(r#"path ="${0:path}""#)), |
130 | attr("proc_macro", None, None), | 243 | attr("proc_macro", None, None), |
131 | attr("proc_macro_attribute", None, None), | 244 | attr("proc_macro_attribute", None, None), |
@@ -153,412 +266,518 @@ const ATTRIBUTES: &[AttrCompletion] = &[ | |||
153 | .prefer_inner(), | 266 | .prefer_inner(), |
154 | ]; | 267 | ]; |
155 | 268 | ||
156 | fn complete_derive(acc: &mut Completions, ctx: &CompletionContext, derive_input: ast::TokenTree) { | 269 | fn parse_comma_sep_input(derive_input: ast::TokenTree) -> Option<FxHashSet<String>> { |
157 | if let Ok(existing_derives) = parse_comma_sep_input(derive_input) { | 270 | let (l_paren, r_paren) = derive_input.l_paren_token().zip(derive_input.r_paren_token())?; |
158 | for derive_completion in DEFAULT_DERIVE_COMPLETIONS | 271 | let mut input_derives = FxHashSet::default(); |
159 | .iter() | 272 | let mut tokens = derive_input |
160 | .filter(|completion| !existing_derives.contains(completion.label)) | 273 | .syntax() |
161 | { | 274 | .children_with_tokens() |
162 | let mut components = vec![derive_completion.label]; | 275 | .filter_map(NodeOrToken::into_token) |
163 | components.extend( | 276 | .skip_while(|token| token != &l_paren) |
164 | derive_completion | 277 | .skip(1) |
165 | .dependencies | 278 | .take_while(|token| token != &r_paren) |
166 | .iter() | 279 | .peekable(); |
167 | .filter(|&&dependency| !existing_derives.contains(dependency)), | 280 | let mut input = String::new(); |
168 | ); | 281 | while tokens.peek().is_some() { |
169 | let lookup = components.join(", "); | 282 | for token in tokens.by_ref().take_while(|t| t.kind() != T![,]) { |
170 | let label = components.iter().rev().join(", "); | 283 | input.push_str(token.text()); |
171 | let mut item = | ||
172 | CompletionItem::new(CompletionKind::Attribute, ctx.source_range(), label); | ||
173 | item.lookup_by(lookup).kind(CompletionItemKind::Attribute); | ||
174 | item.add_to(acc); | ||
175 | } | ||
176 | |||
177 | for custom_derive_name in get_derive_names_in_scope(ctx).difference(&existing_derives) { | ||
178 | let mut item = CompletionItem::new( | ||
179 | CompletionKind::Attribute, | ||
180 | ctx.source_range(), | ||
181 | custom_derive_name, | ||
182 | ); | ||
183 | item.kind(CompletionItemKind::Attribute); | ||
184 | item.add_to(acc); | ||
185 | } | 284 | } |
186 | } | ||
187 | } | ||
188 | 285 | ||
189 | fn complete_lint( | 286 | if !input.is_empty() { |
190 | acc: &mut Completions, | 287 | input_derives.insert(input.trim().to_owned()); |
191 | ctx: &CompletionContext, | ||
192 | derive_input: ast::TokenTree, | ||
193 | lints_completions: &[LintCompletion], | ||
194 | ) { | ||
195 | if let Ok(existing_lints) = parse_comma_sep_input(derive_input) { | ||
196 | for lint_completion in lints_completions | ||
197 | .into_iter() | ||
198 | .filter(|completion| !existing_lints.contains(completion.label)) | ||
199 | { | ||
200 | let mut item = CompletionItem::new( | ||
201 | CompletionKind::Attribute, | ||
202 | ctx.source_range(), | ||
203 | lint_completion.label, | ||
204 | ); | ||
205 | item.kind(CompletionItemKind::Attribute).detail(lint_completion.description); | ||
206 | item.add_to(acc) | ||
207 | } | 288 | } |
208 | } | ||
209 | } | ||
210 | 289 | ||
211 | fn parse_comma_sep_input(derive_input: ast::TokenTree) -> Result<FxHashSet<String>, ()> { | 290 | input.clear(); |
212 | match (derive_input.left_delimiter_token(), derive_input.right_delimiter_token()) { | ||
213 | (Some(left_paren), Some(right_paren)) | ||
214 | if left_paren.kind() == T!['('] && right_paren.kind() == T![')'] => | ||
215 | { | ||
216 | let mut input_derives = FxHashSet::default(); | ||
217 | let mut current_derive = String::new(); | ||
218 | for token in derive_input | ||
219 | .syntax() | ||
220 | .children_with_tokens() | ||
221 | .filter_map(|token| token.into_token()) | ||
222 | .skip_while(|token| token != &left_paren) | ||
223 | .skip(1) | ||
224 | .take_while(|token| token != &right_paren) | ||
225 | { | ||
226 | if T![,] == token.kind() { | ||
227 | if !current_derive.is_empty() { | ||
228 | input_derives.insert(current_derive); | ||
229 | current_derive = String::new(); | ||
230 | } | ||
231 | } else { | ||
232 | current_derive.push_str(token.text().trim()); | ||
233 | } | ||
234 | } | ||
235 | |||
236 | if !current_derive.is_empty() { | ||
237 | input_derives.insert(current_derive); | ||
238 | } | ||
239 | Ok(input_derives) | ||
240 | } | ||
241 | _ => Err(()), | ||
242 | } | 291 | } |
243 | } | ||
244 | |||
245 | fn get_derive_names_in_scope(ctx: &CompletionContext) -> FxHashSet<String> { | ||
246 | let mut result = FxHashSet::default(); | ||
247 | ctx.scope.process_all_names(&mut |name, scope_def| { | ||
248 | if let hir::ScopeDef::MacroDef(mac) = scope_def { | ||
249 | // FIXME kind() doesn't check whether proc-macro is a derive | ||
250 | if mac.kind() == hir::MacroKind::Derive || mac.kind() == hir::MacroKind::ProcMacro { | ||
251 | result.insert(name.to_string()); | ||
252 | } | ||
253 | } | ||
254 | }); | ||
255 | result | ||
256 | } | ||
257 | |||
258 | struct DeriveCompletion { | ||
259 | label: &'static str, | ||
260 | dependencies: &'static [&'static str], | ||
261 | } | ||
262 | |||
263 | /// Standard Rust derives and the information about their dependencies | ||
264 | /// (the dependencies are needed so that the main derive don't break the compilation when added) | ||
265 | const DEFAULT_DERIVE_COMPLETIONS: &[DeriveCompletion] = &[ | ||
266 | DeriveCompletion { label: "Clone", dependencies: &[] }, | ||
267 | DeriveCompletion { label: "Copy", dependencies: &["Clone"] }, | ||
268 | DeriveCompletion { label: "Debug", dependencies: &[] }, | ||
269 | DeriveCompletion { label: "Default", dependencies: &[] }, | ||
270 | DeriveCompletion { label: "Hash", dependencies: &[] }, | ||
271 | DeriveCompletion { label: "PartialEq", dependencies: &[] }, | ||
272 | DeriveCompletion { label: "Eq", dependencies: &["PartialEq"] }, | ||
273 | DeriveCompletion { label: "PartialOrd", dependencies: &["PartialEq"] }, | ||
274 | DeriveCompletion { label: "Ord", dependencies: &["PartialOrd", "Eq", "PartialEq"] }, | ||
275 | ]; | ||
276 | 292 | ||
277 | pub(crate) struct LintCompletion { | 293 | Some(input_derives) |
278 | pub(crate) label: &'static str, | ||
279 | pub(crate) description: &'static str, | ||
280 | } | 294 | } |
281 | 295 | ||
282 | #[rustfmt::skip] | ||
283 | const DEFAULT_LINT_COMPLETIONS: &[LintCompletion] = &[ | ||
284 | LintCompletion { label: "absolute_paths_not_starting_with_crate", description: r#"fully qualified paths that start with a module name instead of `crate`, `self`, or an extern crate name"# }, | ||
285 | LintCompletion { label: "anonymous_parameters", description: r#"detects anonymous parameters"# }, | ||
286 | LintCompletion { label: "box_pointers", description: r#"use of owned (Box type) heap memory"# }, | ||
287 | LintCompletion { label: "deprecated_in_future", description: r#"detects use of items that will be deprecated in a future version"# }, | ||
288 | LintCompletion { label: "elided_lifetimes_in_paths", description: r#"hidden lifetime parameters in types are deprecated"# }, | ||
289 | LintCompletion { label: "explicit_outlives_requirements", description: r#"outlives requirements can be inferred"# }, | ||
290 | LintCompletion { label: "indirect_structural_match", description: r#"pattern with const indirectly referencing non-structural-match type"# }, | ||
291 | LintCompletion { label: "keyword_idents", description: r#"detects edition keywords being used as an identifier"# }, | ||
292 | LintCompletion { label: "macro_use_extern_crate", description: r#"the `#[macro_use]` attribute is now deprecated in favor of using macros via the module system"# }, | ||
293 | LintCompletion { label: "meta_variable_misuse", description: r#"possible meta-variable misuse at macro definition"# }, | ||
294 | LintCompletion { label: "missing_copy_implementations", description: r#"detects potentially-forgotten implementations of `Copy`"# }, | ||
295 | LintCompletion { label: "missing_crate_level_docs", description: r#"detects crates with no crate-level documentation"# }, | ||
296 | LintCompletion { label: "missing_debug_implementations", description: r#"detects missing implementations of Debug"# }, | ||
297 | LintCompletion { label: "missing_docs", description: r#"detects missing documentation for public members"# }, | ||
298 | LintCompletion { label: "missing_doc_code_examples", description: r#"detects publicly-exported items without code samples in their documentation"# }, | ||
299 | LintCompletion { label: "non_ascii_idents", description: r#"detects non-ASCII identifiers"# }, | ||
300 | LintCompletion { label: "private_doc_tests", description: r#"detects code samples in docs of private items not documented by rustdoc"# }, | ||
301 | LintCompletion { label: "single_use_lifetimes", description: r#"detects lifetime parameters that are only used once"# }, | ||
302 | LintCompletion { label: "trivial_casts", description: r#"detects trivial casts which could be removed"# }, | ||
303 | LintCompletion { label: "trivial_numeric_casts", description: r#"detects trivial casts of numeric types which could be removed"# }, | ||
304 | LintCompletion { label: "unaligned_references", description: r#"detects unaligned references to fields of packed structs"# }, | ||
305 | LintCompletion { label: "unreachable_pub", description: r#"`pub` items not reachable from crate root"# }, | ||
306 | LintCompletion { label: "unsafe_code", description: r#"usage of `unsafe` code"# }, | ||
307 | LintCompletion { label: "unsafe_op_in_unsafe_fn", description: r#"unsafe operations in unsafe functions without an explicit unsafe block are deprecated"# }, | ||
308 | LintCompletion { label: "unstable_features", description: r#"enabling unstable features (deprecated. do not use)"# }, | ||
309 | LintCompletion { label: "unused_crate_dependencies", description: r#"crate dependencies that are never used"# }, | ||
310 | LintCompletion { label: "unused_extern_crates", description: r#"extern crates that are never used"# }, | ||
311 | LintCompletion { label: "unused_import_braces", description: r#"unnecessary braces around an imported item"# }, | ||
312 | LintCompletion { label: "unused_lifetimes", description: r#"detects lifetime parameters that are never used"# }, | ||
313 | LintCompletion { label: "unused_qualifications", description: r#"detects unnecessarily qualified names"# }, | ||
314 | LintCompletion { label: "unused_results", description: r#"unused result of an expression in a statement"# }, | ||
315 | LintCompletion { label: "variant_size_differences", description: r#"detects enums with widely varying variant sizes"# }, | ||
316 | LintCompletion { label: "array_into_iter", description: r#"detects calling `into_iter` on arrays"# }, | ||
317 | LintCompletion { label: "asm_sub_register", description: r#"using only a subset of a register for inline asm inputs"# }, | ||
318 | LintCompletion { label: "bare_trait_objects", description: r#"suggest using `dyn Trait` for trait objects"# }, | ||
319 | LintCompletion { label: "bindings_with_variant_name", description: r#"detects pattern bindings with the same name as one of the matched variants"# }, | ||
320 | LintCompletion { label: "cenum_impl_drop_cast", description: r#"a C-like enum implementing Drop is cast"# }, | ||
321 | LintCompletion { label: "clashing_extern_declarations", description: r#"detects when an extern fn has been declared with the same name but different types"# }, | ||
322 | LintCompletion { label: "coherence_leak_check", description: r#"distinct impls distinguished only by the leak-check code"# }, | ||
323 | LintCompletion { label: "confusable_idents", description: r#"detects visually confusable pairs between identifiers"# }, | ||
324 | LintCompletion { label: "dead_code", description: r#"detect unused, unexported items"# }, | ||
325 | LintCompletion { label: "deprecated", description: r#"detects use of deprecated items"# }, | ||
326 | LintCompletion { label: "ellipsis_inclusive_range_patterns", description: r#"`...` range patterns are deprecated"# }, | ||
327 | LintCompletion { label: "exported_private_dependencies", description: r#"public interface leaks type from a private dependency"# }, | ||
328 | LintCompletion { label: "illegal_floating_point_literal_pattern", description: r#"floating-point literals cannot be used in patterns"# }, | ||
329 | LintCompletion { label: "improper_ctypes", description: r#"proper use of libc types in foreign modules"# }, | ||
330 | LintCompletion { label: "improper_ctypes_definitions", description: r#"proper use of libc types in foreign item definitions"# }, | ||
331 | LintCompletion { label: "incomplete_features", description: r#"incomplete features that may function improperly in some or all cases"# }, | ||
332 | LintCompletion { label: "inline_no_sanitize", description: r#"detects incompatible use of `#[inline(always)]` and `#[no_sanitize(...)]`"# }, | ||
333 | LintCompletion { label: "intra_doc_link_resolution_failure", description: r#"failures in resolving intra-doc link targets"# }, | ||
334 | LintCompletion { label: "invalid_codeblock_attributes", description: r#"codeblock attribute looks a lot like a known one"# }, | ||
335 | LintCompletion { label: "invalid_value", description: r#"an invalid value is being created (such as a NULL reference)"# }, | ||
336 | LintCompletion { label: "irrefutable_let_patterns", description: r#"detects irrefutable patterns in if-let and while-let statements"# }, | ||
337 | LintCompletion { label: "late_bound_lifetime_arguments", description: r#"detects generic lifetime arguments in path segments with late bound lifetime parameters"# }, | ||
338 | LintCompletion { label: "mixed_script_confusables", description: r#"detects Unicode scripts whose mixed script confusables codepoints are solely used"# }, | ||
339 | LintCompletion { label: "mutable_borrow_reservation_conflict", description: r#"reservation of a two-phased borrow conflicts with other shared borrows"# }, | ||
340 | LintCompletion { label: "non_camel_case_types", description: r#"types, variants, traits and type parameters should have camel case names"# }, | ||
341 | LintCompletion { label: "non_shorthand_field_patterns", description: r#"using `Struct { x: x }` instead of `Struct { x }` in a pattern"# }, | ||
342 | LintCompletion { label: "non_snake_case", description: r#"variables, methods, functions, lifetime parameters and modules should have snake case names"# }, | ||
343 | LintCompletion { label: "non_upper_case_globals", description: r#"static constants should have uppercase identifiers"# }, | ||
344 | LintCompletion { label: "no_mangle_generic_items", description: r#"generic items must be mangled"# }, | ||
345 | LintCompletion { label: "overlapping_patterns", description: r#"detects overlapping patterns"# }, | ||
346 | LintCompletion { label: "path_statements", description: r#"path statements with no effect"# }, | ||
347 | LintCompletion { label: "private_in_public", description: r#"detect private items in public interfaces not caught by the old implementation"# }, | ||
348 | LintCompletion { label: "proc_macro_derive_resolution_fallback", description: r#"detects proc macro derives using inaccessible names from parent modules"# }, | ||
349 | LintCompletion { label: "redundant_semicolons", description: r#"detects unnecessary trailing semicolons"# }, | ||
350 | LintCompletion { label: "renamed_and_removed_lints", description: r#"lints that have been renamed or removed"# }, | ||
351 | LintCompletion { label: "safe_packed_borrows", description: r#"safe borrows of fields of packed structs were erroneously allowed"# }, | ||
352 | LintCompletion { label: "stable_features", description: r#"stable features found in `#[feature]` directive"# }, | ||
353 | LintCompletion { label: "trivial_bounds", description: r#"these bounds don't depend on an type parameters"# }, | ||
354 | LintCompletion { label: "type_alias_bounds", description: r#"bounds in type aliases are not enforced"# }, | ||
355 | LintCompletion { label: "tyvar_behind_raw_pointer", description: r#"raw pointer to an inference variable"# }, | ||
356 | LintCompletion { label: "uncommon_codepoints", description: r#"detects uncommon Unicode codepoints in identifiers"# }, | ||
357 | LintCompletion { label: "unconditional_recursion", description: r#"functions that cannot return without calling themselves"# }, | ||
358 | LintCompletion { label: "unknown_lints", description: r#"unrecognized lint attribute"# }, | ||
359 | LintCompletion { label: "unnameable_test_items", description: r#"detects an item that cannot be named being marked as `#[test_case]`"# }, | ||
360 | LintCompletion { label: "unreachable_code", description: r#"detects unreachable code paths"# }, | ||
361 | LintCompletion { label: "unreachable_patterns", description: r#"detects unreachable patterns"# }, | ||
362 | LintCompletion { label: "unstable_name_collisions", description: r#"detects name collision with an existing but unstable method"# }, | ||
363 | LintCompletion { label: "unused_allocation", description: r#"detects unnecessary allocations that can be eliminated"# }, | ||
364 | LintCompletion { label: "unused_assignments", description: r#"detect assignments that will never be read"# }, | ||
365 | LintCompletion { label: "unused_attributes", description: r#"detects attributes that were not used by the compiler"# }, | ||
366 | LintCompletion { label: "unused_braces", description: r#"unnecessary braces around an expression"# }, | ||
367 | LintCompletion { label: "unused_comparisons", description: r#"comparisons made useless by limits of the types involved"# }, | ||
368 | LintCompletion { label: "unused_doc_comments", description: r#"detects doc comments that aren't used by rustdoc"# }, | ||
369 | LintCompletion { label: "unused_features", description: r#"unused features found in crate-level `#[feature]` directives"# }, | ||
370 | LintCompletion { label: "unused_imports", description: r#"imports that are never used"# }, | ||
371 | LintCompletion { label: "unused_labels", description: r#"detects labels that are never used"# }, | ||
372 | LintCompletion { label: "unused_macros", description: r#"detects macros that were not used"# }, | ||
373 | LintCompletion { label: "unused_must_use", description: r#"unused result of a type flagged as `#[must_use]`"# }, | ||
374 | LintCompletion { label: "unused_mut", description: r#"detect mut variables which don't need to be mutable"# }, | ||
375 | LintCompletion { label: "unused_parens", description: r#"`if`, `match`, `while` and `return` do not need parentheses"# }, | ||
376 | LintCompletion { label: "unused_unsafe", description: r#"unnecessary use of an `unsafe` block"# }, | ||
377 | LintCompletion { label: "unused_variables", description: r#"detect variables which are not used in any way"# }, | ||
378 | LintCompletion { label: "warnings", description: r#"mass-change the level for lints which produce warnings"# }, | ||
379 | LintCompletion { label: "where_clauses_object_safety", description: r#"checks the object safety of where clauses"# }, | ||
380 | LintCompletion { label: "while_true", description: r#"suggest using `loop { }` instead of `while true { }`"# }, | ||
381 | LintCompletion { label: "ambiguous_associated_items", description: r#"ambiguous associated items"# }, | ||
382 | LintCompletion { label: "arithmetic_overflow", description: r#"arithmetic operation overflows"# }, | ||
383 | LintCompletion { label: "conflicting_repr_hints", description: r#"conflicts between `#[repr(..)]` hints that were previously accepted and used in practice"# }, | ||
384 | LintCompletion { label: "const_err", description: r#"constant evaluation detected erroneous expression"# }, | ||
385 | LintCompletion { label: "ill_formed_attribute_input", description: r#"ill-formed attribute inputs that were previously accepted and used in practice"# }, | ||
386 | LintCompletion { label: "incomplete_include", description: r#"trailing content in included file"# }, | ||
387 | LintCompletion { label: "invalid_type_param_default", description: r#"type parameter default erroneously allowed in invalid location"# }, | ||
388 | LintCompletion { label: "macro_expanded_macro_exports_accessed_by_absolute_paths", description: r#"macro-expanded `macro_export` macros from the current crate cannot be referred to by absolute paths"# }, | ||
389 | LintCompletion { label: "missing_fragment_specifier", description: r#"detects missing fragment specifiers in unused `macro_rules!` patterns"# }, | ||
390 | LintCompletion { label: "mutable_transmutes", description: r#"mutating transmuted &mut T from &T may cause undefined behavior"# }, | ||
391 | LintCompletion { label: "no_mangle_const_items", description: r#"const items will not have their symbols exported"# }, | ||
392 | LintCompletion { label: "order_dependent_trait_objects", description: r#"trait-object types were treated as different depending on marker-trait order"# }, | ||
393 | LintCompletion { label: "overflowing_literals", description: r#"literal out of range for its type"# }, | ||
394 | LintCompletion { label: "patterns_in_fns_without_body", description: r#"patterns in functions without body were erroneously allowed"# }, | ||
395 | LintCompletion { label: "pub_use_of_private_extern_crate", description: r#"detect public re-exports of private extern crates"# }, | ||
396 | LintCompletion { label: "soft_unstable", description: r#"a feature gate that doesn't break dependent crates"# }, | ||
397 | LintCompletion { label: "unconditional_panic", description: r#"operation will cause a panic at runtime"# }, | ||
398 | LintCompletion { label: "unknown_crate_types", description: r#"unknown crate type found in `#[crate_type]` directive"# }, | ||
399 | ]; | ||
400 | |||
401 | #[cfg(test)] | 296 | #[cfg(test)] |
402 | mod tests { | 297 | mod tests { |
298 | use super::*; | ||
299 | |||
403 | use expect_test::{expect, Expect}; | 300 | use expect_test::{expect, Expect}; |
404 | 301 | ||
405 | use crate::{test_utils::completion_list, CompletionKind}; | 302 | use crate::{test_utils::completion_list, CompletionKind}; |
406 | 303 | ||
304 | #[test] | ||
305 | fn attributes_are_sorted() { | ||
306 | let mut attrs = ATTRIBUTES.iter().map(|attr| attr.key()); | ||
307 | let mut prev = attrs.next().unwrap(); | ||
308 | |||
309 | attrs.for_each(|next| { | ||
310 | assert!( | ||
311 | prev < next, | ||
312 | r#"ATTRIBUTES array is not sorted, "{}" should come after "{}""#, | ||
313 | prev, | ||
314 | next | ||
315 | ); | ||
316 | prev = next; | ||
317 | }); | ||
318 | } | ||
319 | |||
407 | fn check(ra_fixture: &str, expect: Expect) { | 320 | fn check(ra_fixture: &str, expect: Expect) { |
408 | let actual = completion_list(ra_fixture, CompletionKind::Attribute); | 321 | let actual = completion_list(ra_fixture, CompletionKind::Attribute); |
409 | expect.assert_eq(&actual); | 322 | expect.assert_eq(&actual); |
410 | } | 323 | } |
411 | 324 | ||
412 | #[test] | 325 | #[test] |
413 | fn empty_derive_completion() { | 326 | fn test_attribute_completion_inside_nested_attr() { |
327 | check(r#"#[cfg($0)]"#, expect![[]]) | ||
328 | } | ||
329 | |||
330 | #[test] | ||
331 | fn test_attribute_completion_with_existing_attr() { | ||
414 | check( | 332 | check( |
415 | r#" | 333 | r#"#[no_mangle] #[$0] mcall!();"#, |
416 | #[derive($0)] | ||
417 | struct Test {} | ||
418 | "#, | ||
419 | expect![[r#" | 334 | expect![[r#" |
420 | at Clone | 335 | at allow(…) |
421 | at Clone, Copy | 336 | at cfg(…) |
422 | at Debug | 337 | at cfg_attr(…) |
423 | at Default | 338 | at deny(…) |
424 | at Hash | 339 | at forbid(…) |
425 | at PartialEq | 340 | at warn(…) |
426 | at PartialEq, Eq | 341 | "#]], |
427 | at PartialEq, PartialOrd | 342 | ) |
428 | at PartialEq, Eq, PartialOrd, Ord | 343 | } |
344 | |||
345 | #[test] | ||
346 | fn complete_attribute_on_source_file() { | ||
347 | check( | ||
348 | r#"#![$0]"#, | ||
349 | expect![[r#" | ||
350 | at allow(…) | ||
351 | at cfg(…) | ||
352 | at cfg_attr(…) | ||
353 | at deny(…) | ||
354 | at forbid(…) | ||
355 | at warn(…) | ||
356 | at deprecated | ||
357 | at doc = "…" | ||
358 | at doc(hidden) | ||
359 | at doc(alias = "…") | ||
360 | at must_use | ||
361 | at no_mangle | ||
362 | at crate_name = "" | ||
363 | at feature(…) | ||
364 | at no_implicit_prelude | ||
365 | at no_main | ||
366 | at no_std | ||
367 | at recursion_limit = … | ||
368 | at type_length_limit = … | ||
369 | at windows_subsystem = "…" | ||
429 | "#]], | 370 | "#]], |
430 | ); | 371 | ); |
431 | } | 372 | } |
432 | 373 | ||
433 | #[test] | 374 | #[test] |
434 | fn no_completion_for_incorrect_derive() { | 375 | fn complete_attribute_on_module() { |
435 | check( | 376 | check( |
436 | r#" | 377 | r#"#[$0] mod foo;"#, |
437 | #[derive{$0)] | 378 | expect![[r#" |
438 | struct Test {} | 379 | at allow(…) |
439 | "#, | 380 | at cfg(…) |
440 | expect![[r#""#]], | 381 | at cfg_attr(…) |
441 | ) | 382 | at deny(…) |
383 | at forbid(…) | ||
384 | at warn(…) | ||
385 | at deprecated | ||
386 | at doc = "…" | ||
387 | at doc(hidden) | ||
388 | at doc(alias = "…") | ||
389 | at must_use | ||
390 | at no_mangle | ||
391 | at path = "…" | ||
392 | "#]], | ||
393 | ); | ||
394 | check( | ||
395 | r#"mod foo {#![$0]}"#, | ||
396 | expect![[r#" | ||
397 | at allow(…) | ||
398 | at cfg(…) | ||
399 | at cfg_attr(…) | ||
400 | at deny(…) | ||
401 | at forbid(…) | ||
402 | at warn(…) | ||
403 | at deprecated | ||
404 | at doc = "…" | ||
405 | at doc(hidden) | ||
406 | at doc(alias = "…") | ||
407 | at must_use | ||
408 | at no_mangle | ||
409 | at no_implicit_prelude | ||
410 | "#]], | ||
411 | ); | ||
442 | } | 412 | } |
443 | 413 | ||
444 | #[test] | 414 | #[test] |
445 | fn derive_with_input_completion() { | 415 | fn complete_attribute_on_macro_rules() { |
446 | check( | 416 | check( |
447 | r#" | 417 | r#"#[$0] macro_rules! foo {}"#, |
448 | #[derive(serde::Serialize, PartialEq, $0)] | ||
449 | struct Test {} | ||
450 | "#, | ||
451 | expect![[r#" | 418 | expect![[r#" |
452 | at Clone | 419 | at allow(…) |
453 | at Clone, Copy | 420 | at cfg(…) |
454 | at Debug | 421 | at cfg_attr(…) |
455 | at Default | 422 | at deny(…) |
456 | at Hash | 423 | at forbid(…) |
457 | at Eq | 424 | at warn(…) |
458 | at PartialOrd | 425 | at deprecated |
459 | at Eq, PartialOrd, Ord | 426 | at doc = "…" |
427 | at doc(hidden) | ||
428 | at doc(alias = "…") | ||
429 | at must_use | ||
430 | at no_mangle | ||
431 | at macro_export | ||
432 | at macro_use | ||
460 | "#]], | 433 | "#]], |
461 | ) | 434 | ); |
462 | } | 435 | } |
463 | 436 | ||
464 | #[test] | 437 | #[test] |
465 | fn test_attribute_completion() { | 438 | fn complete_attribute_on_macro_def() { |
466 | check( | 439 | check( |
467 | r#"#[$0]"#, | 440 | r#"#[$0] macro foo {}"#, |
468 | expect![[r#" | 441 | expect![[r#" |
469 | at allow(…) | 442 | at allow(…) |
470 | at automatically_derived | 443 | at cfg(…) |
471 | at cfg_attr(…) | 444 | at cfg_attr(…) |
445 | at deny(…) | ||
446 | at forbid(…) | ||
447 | at warn(…) | ||
448 | at deprecated | ||
449 | at doc = "…" | ||
450 | at doc(hidden) | ||
451 | at doc(alias = "…") | ||
452 | at must_use | ||
453 | at no_mangle | ||
454 | "#]], | ||
455 | ); | ||
456 | } | ||
457 | |||
458 | #[test] | ||
459 | fn complete_attribute_on_extern_crate() { | ||
460 | check( | ||
461 | r#"#[$0] extern crate foo;"#, | ||
462 | expect![[r#" | ||
463 | at allow(…) | ||
472 | at cfg(…) | 464 | at cfg(…) |
473 | at cold | 465 | at cfg_attr(…) |
474 | at deny(…) | 466 | at deny(…) |
467 | at forbid(…) | ||
468 | at warn(…) | ||
475 | at deprecated | 469 | at deprecated |
476 | at derive(…) | 470 | at doc = "…" |
477 | at export_name = "…" | 471 | at doc(hidden) |
478 | at doc(alias = "…") | 472 | at doc(alias = "…") |
473 | at must_use | ||
474 | at no_mangle | ||
475 | at macro_use | ||
476 | "#]], | ||
477 | ); | ||
478 | } | ||
479 | |||
480 | #[test] | ||
481 | fn complete_attribute_on_use() { | ||
482 | check( | ||
483 | r#"#[$0] use foo;"#, | ||
484 | expect![[r#" | ||
485 | at allow(…) | ||
486 | at cfg(…) | ||
487 | at cfg_attr(…) | ||
488 | at deny(…) | ||
489 | at forbid(…) | ||
490 | at warn(…) | ||
491 | at deprecated | ||
479 | at doc = "…" | 492 | at doc = "…" |
480 | at doc(hidden) | 493 | at doc(hidden) |
494 | at doc(alias = "…") | ||
495 | at must_use | ||
496 | at no_mangle | ||
497 | "#]], | ||
498 | ); | ||
499 | } | ||
500 | |||
501 | #[test] | ||
502 | fn complete_attribute_on_type_alias() { | ||
503 | check( | ||
504 | r#"#[$0] type foo = ();"#, | ||
505 | expect![[r#" | ||
506 | at allow(…) | ||
507 | at cfg(…) | ||
508 | at cfg_attr(…) | ||
509 | at deny(…) | ||
481 | at forbid(…) | 510 | at forbid(…) |
482 | at ignore = "…" | 511 | at warn(…) |
483 | at inline | 512 | at deprecated |
484 | at link | 513 | at doc = "…" |
485 | at link_name = "…" | 514 | at doc(hidden) |
486 | at link_section = "…" | 515 | at doc(alias = "…") |
487 | at macro_export | 516 | at must_use |
488 | at macro_use | 517 | at no_mangle |
518 | "#]], | ||
519 | ); | ||
520 | } | ||
521 | |||
522 | #[test] | ||
523 | fn complete_attribute_on_struct() { | ||
524 | check( | ||
525 | r#"#[$0] struct Foo;"#, | ||
526 | expect![[r#" | ||
527 | at allow(…) | ||
528 | at cfg(…) | ||
529 | at cfg_attr(…) | ||
530 | at deny(…) | ||
531 | at forbid(…) | ||
532 | at warn(…) | ||
533 | at deprecated | ||
534 | at doc = "…" | ||
535 | at doc(hidden) | ||
536 | at doc(alias = "…") | ||
489 | at must_use | 537 | at must_use |
490 | at no_mangle | 538 | at no_mangle |
539 | at derive(…) | ||
540 | at repr(…) | ||
491 | at non_exhaustive | 541 | at non_exhaustive |
492 | at path = "…" | 542 | "#]], |
493 | at proc_macro | 543 | ); |
494 | at proc_macro_attribute | 544 | } |
495 | at proc_macro_derive(…) | 545 | |
546 | #[test] | ||
547 | fn complete_attribute_on_enum() { | ||
548 | check( | ||
549 | r#"#[$0] enum Foo {}"#, | ||
550 | expect![[r#" | ||
551 | at allow(…) | ||
552 | at cfg(…) | ||
553 | at cfg_attr(…) | ||
554 | at deny(…) | ||
555 | at forbid(…) | ||
556 | at warn(…) | ||
557 | at deprecated | ||
558 | at doc = "…" | ||
559 | at doc(hidden) | ||
560 | at doc(alias = "…") | ||
561 | at must_use | ||
562 | at no_mangle | ||
563 | at derive(…) | ||
496 | at repr(…) | 564 | at repr(…) |
497 | at should_panic | 565 | at non_exhaustive |
498 | at target_feature = "…" | 566 | "#]], |
499 | at test | 567 | ); |
500 | at track_caller | 568 | } |
501 | at used | 569 | |
570 | #[test] | ||
571 | fn complete_attribute_on_const() { | ||
572 | check( | ||
573 | r#"#[$0] const FOO: () = ();"#, | ||
574 | expect![[r#" | ||
575 | at allow(…) | ||
576 | at cfg(…) | ||
577 | at cfg_attr(…) | ||
578 | at deny(…) | ||
579 | at forbid(…) | ||
502 | at warn(…) | 580 | at warn(…) |
581 | at deprecated | ||
582 | at doc = "…" | ||
583 | at doc(hidden) | ||
584 | at doc(alias = "…") | ||
585 | at must_use | ||
586 | at no_mangle | ||
503 | "#]], | 587 | "#]], |
504 | ) | 588 | ); |
505 | } | 589 | } |
506 | 590 | ||
507 | #[test] | 591 | #[test] |
508 | fn test_attribute_completion_inside_nested_attr() { | 592 | fn complete_attribute_on_static() { |
509 | check(r#"#[cfg($0)]"#, expect![[]]) | 593 | check( |
594 | r#"#[$0] static FOO: () = ()"#, | ||
595 | expect![[r#" | ||
596 | at allow(…) | ||
597 | at cfg(…) | ||
598 | at cfg_attr(…) | ||
599 | at deny(…) | ||
600 | at forbid(…) | ||
601 | at warn(…) | ||
602 | at deprecated | ||
603 | at doc = "…" | ||
604 | at doc(hidden) | ||
605 | at doc(alias = "…") | ||
606 | at must_use | ||
607 | at no_mangle | ||
608 | at export_name = "…" | ||
609 | at link_name = "…" | ||
610 | at link_section = "…" | ||
611 | at global_allocator | ||
612 | at used | ||
613 | "#]], | ||
614 | ); | ||
510 | } | 615 | } |
511 | 616 | ||
512 | #[test] | 617 | #[test] |
513 | fn test_inner_attribute_completion() { | 618 | fn complete_attribute_on_trait() { |
514 | check( | 619 | check( |
515 | r"#![$0]", | 620 | r#"#[$0] trait Foo {}"#, |
516 | expect![[r#" | 621 | expect![[r#" |
517 | at allow(…) | 622 | at allow(…) |
518 | at automatically_derived | 623 | at cfg(…) |
519 | at cfg_attr(…) | 624 | at cfg_attr(…) |
625 | at deny(…) | ||
626 | at forbid(…) | ||
627 | at warn(…) | ||
628 | at deprecated | ||
629 | at doc = "…" | ||
630 | at doc(hidden) | ||
631 | at doc(alias = "…") | ||
632 | at must_use | ||
633 | at no_mangle | ||
634 | at must_use | ||
635 | "#]], | ||
636 | ); | ||
637 | } | ||
638 | |||
639 | #[test] | ||
640 | fn complete_attribute_on_impl() { | ||
641 | check( | ||
642 | r#"#[$0] impl () {}"#, | ||
643 | expect![[r#" | ||
644 | at allow(…) | ||
520 | at cfg(…) | 645 | at cfg(…) |
521 | at cold | 646 | at cfg_attr(…) |
522 | at crate_name = "" | ||
523 | at deny(…) | 647 | at deny(…) |
648 | at forbid(…) | ||
649 | at warn(…) | ||
524 | at deprecated | 650 | at deprecated |
525 | at derive(…) | 651 | at doc = "…" |
526 | at export_name = "…" | 652 | at doc(hidden) |
653 | at doc(alias = "…") | ||
654 | at must_use | ||
655 | at no_mangle | ||
656 | at automatically_derived | ||
657 | "#]], | ||
658 | ); | ||
659 | check( | ||
660 | r#"impl () {#![$0]}"#, | ||
661 | expect![[r#" | ||
662 | at allow(…) | ||
663 | at cfg(…) | ||
664 | at cfg_attr(…) | ||
665 | at deny(…) | ||
666 | at forbid(…) | ||
667 | at warn(…) | ||
668 | at deprecated | ||
669 | at doc = "…" | ||
670 | at doc(hidden) | ||
527 | at doc(alias = "…") | 671 | at doc(alias = "…") |
672 | at must_use | ||
673 | at no_mangle | ||
674 | "#]], | ||
675 | ); | ||
676 | } | ||
677 | |||
678 | #[test] | ||
679 | fn complete_attribute_on_extern_block() { | ||
680 | check( | ||
681 | r#"#[$0] extern {}"#, | ||
682 | expect![[r#" | ||
683 | at allow(…) | ||
684 | at cfg(…) | ||
685 | at cfg_attr(…) | ||
686 | at deny(…) | ||
687 | at forbid(…) | ||
688 | at warn(…) | ||
689 | at deprecated | ||
528 | at doc = "…" | 690 | at doc = "…" |
529 | at doc(hidden) | 691 | at doc(hidden) |
530 | at feature(…) | 692 | at doc(alias = "…") |
693 | at must_use | ||
694 | at no_mangle | ||
695 | at link | ||
696 | "#]], | ||
697 | ); | ||
698 | check( | ||
699 | r#"extern {#![$0]}"#, | ||
700 | expect![[r#" | ||
701 | at allow(…) | ||
702 | at cfg(…) | ||
703 | at cfg_attr(…) | ||
704 | at deny(…) | ||
531 | at forbid(…) | 705 | at forbid(…) |
532 | at global_allocator | 706 | at warn(…) |
533 | at ignore = "…" | 707 | at deprecated |
534 | at inline | 708 | at doc = "…" |
709 | at doc(hidden) | ||
710 | at doc(alias = "…") | ||
711 | at must_use | ||
712 | at no_mangle | ||
535 | at link | 713 | at link |
714 | "#]], | ||
715 | ); | ||
716 | } | ||
717 | |||
718 | #[test] | ||
719 | fn complete_attribute_on_variant() { | ||
720 | check( | ||
721 | r#"enum Foo { #[$0] Bar }"#, | ||
722 | expect![[r#" | ||
723 | at allow(…) | ||
724 | at cfg(…) | ||
725 | at cfg_attr(…) | ||
726 | at deny(…) | ||
727 | at forbid(…) | ||
728 | at warn(…) | ||
729 | at non_exhaustive | ||
730 | "#]], | ||
731 | ); | ||
732 | } | ||
733 | |||
734 | #[test] | ||
735 | fn complete_attribute_on_fn() { | ||
736 | check( | ||
737 | r#"#[$0] fn main() {}"#, | ||
738 | expect![[r#" | ||
739 | at allow(…) | ||
740 | at cfg(…) | ||
741 | at cfg_attr(…) | ||
742 | at deny(…) | ||
743 | at forbid(…) | ||
744 | at warn(…) | ||
745 | at deprecated | ||
746 | at doc = "…" | ||
747 | at doc(hidden) | ||
748 | at doc(alias = "…") | ||
749 | at must_use | ||
750 | at no_mangle | ||
751 | at export_name = "…" | ||
536 | at link_name = "…" | 752 | at link_name = "…" |
537 | at link_section = "…" | 753 | at link_section = "…" |
538 | at macro_export | 754 | at cold |
539 | at macro_use | 755 | at ignore = "…" |
756 | at inline | ||
540 | at must_use | 757 | at must_use |
541 | at no_link | ||
542 | at no_implicit_prelude | ||
543 | at no_main | ||
544 | at no_mangle | ||
545 | at no_std | ||
546 | at non_exhaustive | ||
547 | at panic_handler | 758 | at panic_handler |
548 | at path = "…" | ||
549 | at proc_macro | 759 | at proc_macro |
550 | at proc_macro_attribute | ||
551 | at proc_macro_derive(…) | 760 | at proc_macro_derive(…) |
552 | at recursion_limit = … | 761 | at proc_macro_attribute |
553 | at repr(…) | ||
554 | at should_panic | 762 | at should_panic |
555 | at target_feature = "…" | 763 | at target_feature = "…" |
556 | at test | 764 | at test |
557 | at track_caller | 765 | at track_caller |
558 | at type_length_limit = … | 766 | "#]], |
559 | at used | 767 | ); |
768 | } | ||
769 | |||
770 | #[test] | ||
771 | fn complete_attribute_on_expr() { | ||
772 | check( | ||
773 | r#"fn main() { #[$0] foo() }"#, | ||
774 | expect![[r#" | ||
775 | at allow(…) | ||
776 | at cfg(…) | ||
777 | at cfg_attr(…) | ||
778 | at deny(…) | ||
779 | at forbid(…) | ||
560 | at warn(…) | 780 | at warn(…) |
561 | at windows_subsystem = "…" | ||
562 | "#]], | 781 | "#]], |
563 | ); | 782 | ); |
564 | } | 783 | } |
diff --git a/crates/ide_completion/src/completions/attribute/derive.rs b/crates/ide_completion/src/completions/attribute/derive.rs new file mode 100644 index 000000000..0bc3eab98 --- /dev/null +++ b/crates/ide_completion/src/completions/attribute/derive.rs | |||
@@ -0,0 +1,147 @@ | |||
1 | //! Completion for derives | ||
2 | use itertools::Itertools; | ||
3 | use rustc_hash::FxHashSet; | ||
4 | use syntax::ast; | ||
5 | |||
6 | use crate::{ | ||
7 | context::CompletionContext, | ||
8 | item::{CompletionItem, CompletionItemKind, CompletionKind}, | ||
9 | Completions, | ||
10 | }; | ||
11 | |||
12 | pub(super) fn complete_derive( | ||
13 | acc: &mut Completions, | ||
14 | ctx: &CompletionContext, | ||
15 | derive_input: ast::TokenTree, | ||
16 | ) { | ||
17 | if let Some(existing_derives) = super::parse_comma_sep_input(derive_input) { | ||
18 | for derive_completion in DEFAULT_DERIVE_COMPLETIONS | ||
19 | .iter() | ||
20 | .filter(|completion| !existing_derives.contains(completion.label)) | ||
21 | { | ||
22 | let mut components = vec![derive_completion.label]; | ||
23 | components.extend( | ||
24 | derive_completion | ||
25 | .dependencies | ||
26 | .iter() | ||
27 | .filter(|&&dependency| !existing_derives.contains(dependency)), | ||
28 | ); | ||
29 | let lookup = components.join(", "); | ||
30 | let label = components.iter().rev().join(", "); | ||
31 | let mut item = | ||
32 | CompletionItem::new(CompletionKind::Attribute, ctx.source_range(), label); | ||
33 | item.lookup_by(lookup).kind(CompletionItemKind::Attribute); | ||
34 | item.add_to(acc); | ||
35 | } | ||
36 | |||
37 | for custom_derive_name in get_derive_names_in_scope(ctx).difference(&existing_derives) { | ||
38 | let mut item = CompletionItem::new( | ||
39 | CompletionKind::Attribute, | ||
40 | ctx.source_range(), | ||
41 | custom_derive_name, | ||
42 | ); | ||
43 | item.kind(CompletionItemKind::Attribute); | ||
44 | item.add_to(acc); | ||
45 | } | ||
46 | } | ||
47 | } | ||
48 | |||
49 | fn get_derive_names_in_scope(ctx: &CompletionContext) -> FxHashSet<String> { | ||
50 | let mut result = FxHashSet::default(); | ||
51 | ctx.scope.process_all_names(&mut |name, scope_def| { | ||
52 | if let hir::ScopeDef::MacroDef(mac) = scope_def { | ||
53 | if mac.kind() == hir::MacroKind::Derive { | ||
54 | result.insert(name.to_string()); | ||
55 | } | ||
56 | } | ||
57 | }); | ||
58 | result | ||
59 | } | ||
60 | |||
61 | struct DeriveCompletion { | ||
62 | label: &'static str, | ||
63 | dependencies: &'static [&'static str], | ||
64 | } | ||
65 | |||
66 | /// Standard Rust derives and the information about their dependencies | ||
67 | /// (the dependencies are needed so that the main derive don't break the compilation when added) | ||
68 | const DEFAULT_DERIVE_COMPLETIONS: &[DeriveCompletion] = &[ | ||
69 | DeriveCompletion { label: "Clone", dependencies: &[] }, | ||
70 | DeriveCompletion { label: "Copy", dependencies: &["Clone"] }, | ||
71 | DeriveCompletion { label: "Debug", dependencies: &[] }, | ||
72 | DeriveCompletion { label: "Default", dependencies: &[] }, | ||
73 | DeriveCompletion { label: "Hash", dependencies: &[] }, | ||
74 | DeriveCompletion { label: "PartialEq", dependencies: &[] }, | ||
75 | DeriveCompletion { label: "Eq", dependencies: &["PartialEq"] }, | ||
76 | DeriveCompletion { label: "PartialOrd", dependencies: &["PartialEq"] }, | ||
77 | DeriveCompletion { label: "Ord", dependencies: &["PartialOrd", "Eq", "PartialEq"] }, | ||
78 | ]; | ||
79 | |||
80 | #[cfg(test)] | ||
81 | mod tests { | ||
82 | use expect_test::{expect, Expect}; | ||
83 | |||
84 | use crate::{test_utils::completion_list, CompletionKind}; | ||
85 | |||
86 | fn check(ra_fixture: &str, expect: Expect) { | ||
87 | let actual = completion_list(ra_fixture, CompletionKind::Attribute); | ||
88 | expect.assert_eq(&actual); | ||
89 | } | ||
90 | |||
91 | #[test] | ||
92 | fn no_completion_for_incorrect_derive() { | ||
93 | check(r#"#[derive{$0)] struct Test;"#, expect![[]]) | ||
94 | } | ||
95 | |||
96 | #[test] | ||
97 | fn empty_derive() { | ||
98 | check( | ||
99 | r#"#[derive($0)] struct Test;"#, | ||
100 | expect![[r#" | ||
101 | at Clone | ||
102 | at Clone, Copy | ||
103 | at Debug | ||
104 | at Default | ||
105 | at Hash | ||
106 | at PartialEq | ||
107 | at PartialEq, Eq | ||
108 | at PartialEq, PartialOrd | ||
109 | at PartialEq, Eq, PartialOrd, Ord | ||
110 | "#]], | ||
111 | ); | ||
112 | } | ||
113 | |||
114 | #[test] | ||
115 | fn derive_with_input() { | ||
116 | check( | ||
117 | r#"#[derive(serde::Serialize, PartialEq, $0)] struct Test;"#, | ||
118 | expect![[r#" | ||
119 | at Clone | ||
120 | at Clone, Copy | ||
121 | at Debug | ||
122 | at Default | ||
123 | at Hash | ||
124 | at Eq | ||
125 | at PartialOrd | ||
126 | at Eq, PartialOrd, Ord | ||
127 | "#]], | ||
128 | ) | ||
129 | } | ||
130 | |||
131 | #[test] | ||
132 | fn derive_with_input2() { | ||
133 | check( | ||
134 | r#"#[derive($0 serde::Serialize, PartialEq)] struct Test;"#, | ||
135 | expect![[r#" | ||
136 | at Clone | ||
137 | at Clone, Copy | ||
138 | at Debug | ||
139 | at Default | ||
140 | at Hash | ||
141 | at Eq | ||
142 | at PartialOrd | ||
143 | at Eq, PartialOrd, Ord | ||
144 | "#]], | ||
145 | ) | ||
146 | } | ||
147 | } | ||
diff --git a/crates/ide_completion/src/completions/attribute/lint.rs b/crates/ide_completion/src/completions/attribute/lint.rs new file mode 100644 index 000000000..403630dce --- /dev/null +++ b/crates/ide_completion/src/completions/attribute/lint.rs | |||
@@ -0,0 +1,187 @@ | |||
1 | //! Completion for lints | ||
2 | use syntax::ast; | ||
3 | |||
4 | use crate::{ | ||
5 | context::CompletionContext, | ||
6 | item::{CompletionItem, CompletionItemKind, CompletionKind}, | ||
7 | Completions, | ||
8 | }; | ||
9 | |||
10 | pub(super) fn complete_lint( | ||
11 | acc: &mut Completions, | ||
12 | ctx: &CompletionContext, | ||
13 | derive_input: ast::TokenTree, | ||
14 | lints_completions: &[LintCompletion], | ||
15 | ) { | ||
16 | if let Some(existing_lints) = super::parse_comma_sep_input(derive_input) { | ||
17 | for lint_completion in lints_completions | ||
18 | .into_iter() | ||
19 | .filter(|completion| !existing_lints.contains(completion.label)) | ||
20 | { | ||
21 | let mut item = CompletionItem::new( | ||
22 | CompletionKind::Attribute, | ||
23 | ctx.source_range(), | ||
24 | lint_completion.label, | ||
25 | ); | ||
26 | item.kind(CompletionItemKind::Attribute).detail(lint_completion.description); | ||
27 | item.add_to(acc) | ||
28 | } | ||
29 | } | ||
30 | } | ||
31 | |||
32 | pub(crate) struct LintCompletion { | ||
33 | pub(crate) label: &'static str, | ||
34 | pub(crate) description: &'static str, | ||
35 | } | ||
36 | |||
37 | #[rustfmt::skip] | ||
38 | pub(super) const DEFAULT_LINT_COMPLETIONS: &[LintCompletion] = &[ | ||
39 | LintCompletion { label: "absolute_paths_not_starting_with_crate", description: r#"fully qualified paths that start with a module name instead of `crate`, `self`, or an extern crate name"# }, | ||
40 | LintCompletion { label: "anonymous_parameters", description: r#"detects anonymous parameters"# }, | ||
41 | LintCompletion { label: "box_pointers", description: r#"use of owned (Box type) heap memory"# }, | ||
42 | LintCompletion { label: "deprecated_in_future", description: r#"detects use of items that will be deprecated in a future version"# }, | ||
43 | LintCompletion { label: "elided_lifetimes_in_paths", description: r#"hidden lifetime parameters in types are deprecated"# }, | ||
44 | LintCompletion { label: "explicit_outlives_requirements", description: r#"outlives requirements can be inferred"# }, | ||
45 | LintCompletion { label: "indirect_structural_match", description: r#"pattern with const indirectly referencing non-structural-match type"# }, | ||
46 | LintCompletion { label: "keyword_idents", description: r#"detects edition keywords being used as an identifier"# }, | ||
47 | LintCompletion { label: "macro_use_extern_crate", description: r#"the `#[macro_use]` attribute is now deprecated in favor of using macros via the module system"# }, | ||
48 | LintCompletion { label: "meta_variable_misuse", description: r#"possible meta-variable misuse at macro definition"# }, | ||
49 | LintCompletion { label: "missing_copy_implementations", description: r#"detects potentially-forgotten implementations of `Copy`"# }, | ||
50 | LintCompletion { label: "missing_crate_level_docs", description: r#"detects crates with no crate-level documentation"# }, | ||
51 | LintCompletion { label: "missing_debug_implementations", description: r#"detects missing implementations of Debug"# }, | ||
52 | LintCompletion { label: "missing_docs", description: r#"detects missing documentation for public members"# }, | ||
53 | LintCompletion { label: "missing_doc_code_examples", description: r#"detects publicly-exported items without code samples in their documentation"# }, | ||
54 | LintCompletion { label: "non_ascii_idents", description: r#"detects non-ASCII identifiers"# }, | ||
55 | LintCompletion { label: "private_doc_tests", description: r#"detects code samples in docs of private items not documented by rustdoc"# }, | ||
56 | LintCompletion { label: "single_use_lifetimes", description: r#"detects lifetime parameters that are only used once"# }, | ||
57 | LintCompletion { label: "trivial_casts", description: r#"detects trivial casts which could be removed"# }, | ||
58 | LintCompletion { label: "trivial_numeric_casts", description: r#"detects trivial casts of numeric types which could be removed"# }, | ||
59 | LintCompletion { label: "unaligned_references", description: r#"detects unaligned references to fields of packed structs"# }, | ||
60 | LintCompletion { label: "unreachable_pub", description: r#"`pub` items not reachable from crate root"# }, | ||
61 | LintCompletion { label: "unsafe_code", description: r#"usage of `unsafe` code"# }, | ||
62 | LintCompletion { label: "unsafe_op_in_unsafe_fn", description: r#"unsafe operations in unsafe functions without an explicit unsafe block are deprecated"# }, | ||
63 | LintCompletion { label: "unstable_features", description: r#"enabling unstable features (deprecated. do not use)"# }, | ||
64 | LintCompletion { label: "unused_crate_dependencies", description: r#"crate dependencies that are never used"# }, | ||
65 | LintCompletion { label: "unused_extern_crates", description: r#"extern crates that are never used"# }, | ||
66 | LintCompletion { label: "unused_import_braces", description: r#"unnecessary braces around an imported item"# }, | ||
67 | LintCompletion { label: "unused_lifetimes", description: r#"detects lifetime parameters that are never used"# }, | ||
68 | LintCompletion { label: "unused_qualifications", description: r#"detects unnecessarily qualified names"# }, | ||
69 | LintCompletion { label: "unused_results", description: r#"unused result of an expression in a statement"# }, | ||
70 | LintCompletion { label: "variant_size_differences", description: r#"detects enums with widely varying variant sizes"# }, | ||
71 | LintCompletion { label: "array_into_iter", description: r#"detects calling `into_iter` on arrays"# }, | ||
72 | LintCompletion { label: "asm_sub_register", description: r#"using only a subset of a register for inline asm inputs"# }, | ||
73 | LintCompletion { label: "bare_trait_objects", description: r#"suggest using `dyn Trait` for trait objects"# }, | ||
74 | LintCompletion { label: "bindings_with_variant_name", description: r#"detects pattern bindings with the same name as one of the matched variants"# }, | ||
75 | LintCompletion { label: "cenum_impl_drop_cast", description: r#"a C-like enum implementing Drop is cast"# }, | ||
76 | LintCompletion { label: "clashing_extern_declarations", description: r#"detects when an extern fn has been declared with the same name but different types"# }, | ||
77 | LintCompletion { label: "coherence_leak_check", description: r#"distinct impls distinguished only by the leak-check code"# }, | ||
78 | LintCompletion { label: "confusable_idents", description: r#"detects visually confusable pairs between identifiers"# }, | ||
79 | LintCompletion { label: "dead_code", description: r#"detect unused, unexported items"# }, | ||
80 | LintCompletion { label: "deprecated", description: r#"detects use of deprecated items"# }, | ||
81 | LintCompletion { label: "ellipsis_inclusive_range_patterns", description: r#"`...` range patterns are deprecated"# }, | ||
82 | LintCompletion { label: "exported_private_dependencies", description: r#"public interface leaks type from a private dependency"# }, | ||
83 | LintCompletion { label: "illegal_floating_point_literal_pattern", description: r#"floating-point literals cannot be used in patterns"# }, | ||
84 | LintCompletion { label: "improper_ctypes", description: r#"proper use of libc types in foreign modules"# }, | ||
85 | LintCompletion { label: "improper_ctypes_definitions", description: r#"proper use of libc types in foreign item definitions"# }, | ||
86 | LintCompletion { label: "incomplete_features", description: r#"incomplete features that may function improperly in some or all cases"# }, | ||
87 | LintCompletion { label: "inline_no_sanitize", description: r#"detects incompatible use of `#[inline(always)]` and `#[no_sanitize(...)]`"# }, | ||
88 | LintCompletion { label: "intra_doc_link_resolution_failure", description: r#"failures in resolving intra-doc link targets"# }, | ||
89 | LintCompletion { label: "invalid_codeblock_attributes", description: r#"codeblock attribute looks a lot like a known one"# }, | ||
90 | LintCompletion { label: "invalid_value", description: r#"an invalid value is being created (such as a NULL reference)"# }, | ||
91 | LintCompletion { label: "irrefutable_let_patterns", description: r#"detects irrefutable patterns in if-let and while-let statements"# }, | ||
92 | LintCompletion { label: "late_bound_lifetime_arguments", description: r#"detects generic lifetime arguments in path segments with late bound lifetime parameters"# }, | ||
93 | LintCompletion { label: "mixed_script_confusables", description: r#"detects Unicode scripts whose mixed script confusables codepoints are solely used"# }, | ||
94 | LintCompletion { label: "mutable_borrow_reservation_conflict", description: r#"reservation of a two-phased borrow conflicts with other shared borrows"# }, | ||
95 | LintCompletion { label: "non_camel_case_types", description: r#"types, variants, traits and type parameters should have camel case names"# }, | ||
96 | LintCompletion { label: "non_shorthand_field_patterns", description: r#"using `Struct { x: x }` instead of `Struct { x }` in a pattern"# }, | ||
97 | LintCompletion { label: "non_snake_case", description: r#"variables, methods, functions, lifetime parameters and modules should have snake case names"# }, | ||
98 | LintCompletion { label: "non_upper_case_globals", description: r#"static constants should have uppercase identifiers"# }, | ||
99 | LintCompletion { label: "no_mangle_generic_items", description: r#"generic items must be mangled"# }, | ||
100 | LintCompletion { label: "overlapping_patterns", description: r#"detects overlapping patterns"# }, | ||
101 | LintCompletion { label: "path_statements", description: r#"path statements with no effect"# }, | ||
102 | LintCompletion { label: "private_in_public", description: r#"detect private items in public interfaces not caught by the old implementation"# }, | ||
103 | LintCompletion { label: "proc_macro_derive_resolution_fallback", description: r#"detects proc macro derives using inaccessible names from parent modules"# }, | ||
104 | LintCompletion { label: "redundant_semicolons", description: r#"detects unnecessary trailing semicolons"# }, | ||
105 | LintCompletion { label: "renamed_and_removed_lints", description: r#"lints that have been renamed or removed"# }, | ||
106 | LintCompletion { label: "safe_packed_borrows", description: r#"safe borrows of fields of packed structs were erroneously allowed"# }, | ||
107 | LintCompletion { label: "stable_features", description: r#"stable features found in `#[feature]` directive"# }, | ||
108 | LintCompletion { label: "trivial_bounds", description: r#"these bounds don't depend on an type parameters"# }, | ||
109 | LintCompletion { label: "type_alias_bounds", description: r#"bounds in type aliases are not enforced"# }, | ||
110 | LintCompletion { label: "tyvar_behind_raw_pointer", description: r#"raw pointer to an inference variable"# }, | ||
111 | LintCompletion { label: "uncommon_codepoints", description: r#"detects uncommon Unicode codepoints in identifiers"# }, | ||
112 | LintCompletion { label: "unconditional_recursion", description: r#"functions that cannot return without calling themselves"# }, | ||
113 | LintCompletion { label: "unknown_lints", description: r#"unrecognized lint attribute"# }, | ||
114 | LintCompletion { label: "unnameable_test_items", description: r#"detects an item that cannot be named being marked as `#[test_case]`"# }, | ||
115 | LintCompletion { label: "unreachable_code", description: r#"detects unreachable code paths"# }, | ||
116 | LintCompletion { label: "unreachable_patterns", description: r#"detects unreachable patterns"# }, | ||
117 | LintCompletion { label: "unstable_name_collisions", description: r#"detects name collision with an existing but unstable method"# }, | ||
118 | LintCompletion { label: "unused_allocation", description: r#"detects unnecessary allocations that can be eliminated"# }, | ||
119 | LintCompletion { label: "unused_assignments", description: r#"detect assignments that will never be read"# }, | ||
120 | LintCompletion { label: "unused_attributes", description: r#"detects attributes that were not used by the compiler"# }, | ||
121 | LintCompletion { label: "unused_braces", description: r#"unnecessary braces around an expression"# }, | ||
122 | LintCompletion { label: "unused_comparisons", description: r#"comparisons made useless by limits of the types involved"# }, | ||
123 | LintCompletion { label: "unused_doc_comments", description: r#"detects doc comments that aren't used by rustdoc"# }, | ||
124 | LintCompletion { label: "unused_features", description: r#"unused features found in crate-level `#[feature]` directives"# }, | ||
125 | LintCompletion { label: "unused_imports", description: r#"imports that are never used"# }, | ||
126 | LintCompletion { label: "unused_labels", description: r#"detects labels that are never used"# }, | ||
127 | LintCompletion { label: "unused_macros", description: r#"detects macros that were not used"# }, | ||
128 | LintCompletion { label: "unused_must_use", description: r#"unused result of a type flagged as `#[must_use]`"# }, | ||
129 | LintCompletion { label: "unused_mut", description: r#"detect mut variables which don't need to be mutable"# }, | ||
130 | LintCompletion { label: "unused_parens", description: r#"`if`, `match`, `while` and `return` do not need parentheses"# }, | ||
131 | LintCompletion { label: "unused_unsafe", description: r#"unnecessary use of an `unsafe` block"# }, | ||
132 | LintCompletion { label: "unused_variables", description: r#"detect variables which are not used in any way"# }, | ||
133 | LintCompletion { label: "warnings", description: r#"mass-change the level for lints which produce warnings"# }, | ||
134 | LintCompletion { label: "where_clauses_object_safety", description: r#"checks the object safety of where clauses"# }, | ||
135 | LintCompletion { label: "while_true", description: r#"suggest using `loop { }` instead of `while true { }`"# }, | ||
136 | LintCompletion { label: "ambiguous_associated_items", description: r#"ambiguous associated items"# }, | ||
137 | LintCompletion { label: "arithmetic_overflow", description: r#"arithmetic operation overflows"# }, | ||
138 | LintCompletion { label: "conflicting_repr_hints", description: r#"conflicts between `#[repr(..)]` hints that were previously accepted and used in practice"# }, | ||
139 | LintCompletion { label: "const_err", description: r#"constant evaluation detected erroneous expression"# }, | ||
140 | LintCompletion { label: "ill_formed_attribute_input", description: r#"ill-formed attribute inputs that were previously accepted and used in practice"# }, | ||
141 | LintCompletion { label: "incomplete_include", description: r#"trailing content in included file"# }, | ||
142 | LintCompletion { label: "invalid_type_param_default", description: r#"type parameter default erroneously allowed in invalid location"# }, | ||
143 | LintCompletion { label: "macro_expanded_macro_exports_accessed_by_absolute_paths", description: r#"macro-expanded `macro_export` macros from the current crate cannot be referred to by absolute paths"# }, | ||
144 | LintCompletion { label: "missing_fragment_specifier", description: r#"detects missing fragment specifiers in unused `macro_rules!` patterns"# }, | ||
145 | LintCompletion { label: "mutable_transmutes", description: r#"mutating transmuted &mut T from &T may cause undefined behavior"# }, | ||
146 | LintCompletion { label: "no_mangle_const_items", description: r#"const items will not have their symbols exported"# }, | ||
147 | LintCompletion { label: "order_dependent_trait_objects", description: r#"trait-object types were treated as different depending on marker-trait order"# }, | ||
148 | LintCompletion { label: "overflowing_literals", description: r#"literal out of range for its type"# }, | ||
149 | LintCompletion { label: "patterns_in_fns_without_body", description: r#"patterns in functions without body were erroneously allowed"# }, | ||
150 | LintCompletion { label: "pub_use_of_private_extern_crate", description: r#"detect public re-exports of private extern crates"# }, | ||
151 | LintCompletion { label: "soft_unstable", description: r#"a feature gate that doesn't break dependent crates"# }, | ||
152 | LintCompletion { label: "unconditional_panic", description: r#"operation will cause a panic at runtime"# }, | ||
153 | LintCompletion { label: "unknown_crate_types", description: r#"unknown crate type found in `#[crate_type]` directive"# }, | ||
154 | ]; | ||
155 | |||
156 | #[cfg(test)] | ||
157 | mod tests { | ||
158 | |||
159 | use crate::test_utils::check_edit; | ||
160 | |||
161 | #[test] | ||
162 | fn check_empty() { | ||
163 | check_edit( | ||
164 | "deprecated", | ||
165 | r#"#[allow($0)] struct Test;"#, | ||
166 | r#"#[allow(deprecated)] struct Test;"#, | ||
167 | ) | ||
168 | } | ||
169 | |||
170 | #[test] | ||
171 | fn check_with_existing() { | ||
172 | check_edit( | ||
173 | "deprecated", | ||
174 | r#"#[allow(keyword_idents, $0)] struct Test;"#, | ||
175 | r#"#[allow(keyword_idents, deprecated)] struct Test;"#, | ||
176 | ) | ||
177 | } | ||
178 | |||
179 | #[test] | ||
180 | fn check_qualified() { | ||
181 | check_edit( | ||
182 | "deprecated", | ||
183 | r#"#[allow(keyword_idents, $0)] struct Test;"#, | ||
184 | r#"#[allow(keyword_idents, deprecated)] struct Test;"#, | ||
185 | ) | ||
186 | } | ||
187 | } | ||
diff --git a/crates/ide_completion/src/completions/flyimport.rs b/crates/ide_completion/src/completions/flyimport.rs index be9cfbded..df27e7a84 100644 --- a/crates/ide_completion/src/completions/flyimport.rs +++ b/crates/ide_completion/src/completions/flyimport.rs | |||
@@ -110,7 +110,11 @@ pub(crate) fn import_on_the_fly(acc: &mut Completions, ctx: &CompletionContext) | |||
110 | if !ctx.config.enable_imports_on_the_fly { | 110 | if !ctx.config.enable_imports_on_the_fly { |
111 | return None; | 111 | return None; |
112 | } | 112 | } |
113 | if ctx.use_item_syntax.is_some() || ctx.is_path_disallowed() { | 113 | if ctx.use_item_syntax.is_some() |
114 | || ctx.is_path_disallowed() | ||
115 | || ctx.expects_item() | ||
116 | || ctx.expects_assoc_item() | ||
117 | { | ||
114 | return None; | 118 | return None; |
115 | } | 119 | } |
116 | let potential_import_name = { | 120 | let potential_import_name = { |
diff --git a/crates/ide_completion/src/completions/fn_param.rs b/crates/ide_completion/src/completions/fn_param.rs index 0ea558489..cb90e8a3e 100644 --- a/crates/ide_completion/src/completions/fn_param.rs +++ b/crates/ide_completion/src/completions/fn_param.rs | |||
@@ -128,4 +128,19 @@ fn outer(text: String) { | |||
128 | "#]], | 128 | "#]], |
129 | ) | 129 | ) |
130 | } | 130 | } |
131 | |||
132 | #[test] | ||
133 | fn completes_non_ident_pat_param() { | ||
134 | check( | ||
135 | r#" | ||
136 | struct Bar { bar: u32 } | ||
137 | |||
138 | fn foo(Bar { bar }: Bar) {} | ||
139 | fn foo2($0) {} | ||
140 | "#, | ||
141 | expect![[r#" | ||
142 | bn Bar { bar }: Bar | ||
143 | "#]], | ||
144 | ) | ||
145 | } | ||
131 | } | 146 | } |
diff --git a/crates/ide_completion/src/completions/keyword.rs b/crates/ide_completion/src/completions/keyword.rs index 58e35bad9..0d035c611 100644 --- a/crates/ide_completion/src/completions/keyword.rs +++ b/crates/ide_completion/src/completions/keyword.rs | |||
@@ -4,7 +4,10 @@ use std::iter; | |||
4 | 4 | ||
5 | use syntax::{SyntaxKind, T}; | 5 | use syntax::{SyntaxKind, T}; |
6 | 6 | ||
7 | use crate::{CompletionContext, CompletionItem, CompletionItemKind, CompletionKind, Completions}; | 7 | use crate::{ |
8 | patterns::ImmediateLocation, CompletionContext, CompletionItem, CompletionItemKind, | ||
9 | CompletionKind, Completions, | ||
10 | }; | ||
8 | 11 | ||
9 | pub(crate) fn complete_use_tree_keyword(acc: &mut Completions, ctx: &CompletionContext) { | 12 | pub(crate) fn complete_use_tree_keyword(acc: &mut Completions, ctx: &CompletionContext) { |
10 | // complete keyword "crate" in use stmt | 13 | // complete keyword "crate" in use stmt |
@@ -44,96 +47,95 @@ pub(crate) fn complete_expr_keyword(acc: &mut Completions, ctx: &CompletionConte | |||
44 | cov_mark::hit!(no_keyword_completion_in_comments); | 47 | cov_mark::hit!(no_keyword_completion_in_comments); |
45 | return; | 48 | return; |
46 | } | 49 | } |
47 | if ctx.record_lit_syntax.is_some() { | 50 | if matches!(ctx.completion_location, Some(ImmediateLocation::RecordExpr(_))) { |
48 | cov_mark::hit!(no_keyword_completion_in_record_lit); | 51 | cov_mark::hit!(no_keyword_completion_in_record_lit); |
49 | return; | 52 | return; |
50 | } | 53 | } |
54 | let mut add_keyword = |kw, snippet| add_keyword(ctx, acc, kw, snippet); | ||
51 | 55 | ||
52 | let has_trait_or_impl_parent = ctx.has_impl_or_trait_parent(); | 56 | let expects_assoc_item = ctx.expects_assoc_item(); |
53 | let has_block_expr_parent = ctx.has_block_expr_parent(); | 57 | let has_block_expr_parent = ctx.has_block_expr_parent(); |
54 | let has_item_list_parent = ctx.has_item_list_parent(); | 58 | let expects_item = ctx.expects_item(); |
59 | |||
55 | if ctx.has_impl_or_trait_prev_sibling() { | 60 | if ctx.has_impl_or_trait_prev_sibling() { |
56 | add_keyword(ctx, acc, "where", "where "); | 61 | add_keyword("where", "where "); |
57 | return; | 62 | return; |
58 | } | 63 | } |
59 | if ctx.previous_token_is(T![unsafe]) { | 64 | if ctx.previous_token_is(T![unsafe]) { |
60 | if has_item_list_parent || has_block_expr_parent { | 65 | if expects_item || expects_assoc_item || has_block_expr_parent { |
61 | add_keyword(ctx, acc, "fn", "fn $1($2) {\n $0\n}") | 66 | add_keyword("fn", "fn $1($2) {\n $0\n}") |
62 | } | 67 | } |
63 | 68 | ||
64 | if has_item_list_parent || has_block_expr_parent { | 69 | if expects_item || has_block_expr_parent { |
65 | add_keyword(ctx, acc, "trait", "trait $1 {\n $0\n}"); | 70 | add_keyword("trait", "trait $1 {\n $0\n}"); |
66 | add_keyword(ctx, acc, "impl", "impl $1 {\n $0\n}"); | 71 | add_keyword("impl", "impl $1 {\n $0\n}"); |
67 | } | 72 | } |
68 | 73 | ||
69 | return; | 74 | return; |
70 | } | 75 | } |
71 | if has_item_list_parent || has_trait_or_impl_parent || has_block_expr_parent { | ||
72 | add_keyword(ctx, acc, "fn", "fn $1($2) {\n $0\n}"); | ||
73 | } | ||
74 | if has_item_list_parent || has_block_expr_parent { | ||
75 | add_keyword(ctx, acc, "use", "use "); | ||
76 | add_keyword(ctx, acc, "impl", "impl $1 {\n $0\n}"); | ||
77 | add_keyword(ctx, acc, "trait", "trait $1 {\n $0\n}"); | ||
78 | } | ||
79 | 76 | ||
80 | if has_item_list_parent { | 77 | if expects_item || ctx.expects_non_trait_assoc_item() || ctx.expect_record_field() { |
81 | add_keyword(ctx, acc, "enum", "enum $1 {\n $0\n}"); | 78 | add_keyword("pub(crate)", "pub(crate) "); |
82 | add_keyword(ctx, acc, "struct", "struct $0"); | 79 | add_keyword("pub", "pub "); |
83 | add_keyword(ctx, acc, "union", "union $1 {\n $0\n}"); | ||
84 | } | 80 | } |
85 | 81 | ||
86 | if ctx.is_expr { | 82 | if expects_item || expects_assoc_item || has_block_expr_parent { |
87 | add_keyword(ctx, acc, "match", "match $1 {\n $0\n}"); | 83 | add_keyword("unsafe", "unsafe "); |
88 | add_keyword(ctx, acc, "while", "while $1 {\n $0\n}"); | 84 | add_keyword("fn", "fn $1($2) {\n $0\n}"); |
89 | add_keyword(ctx, acc, "while let", "while let $1 = $2 {\n $0\n}"); | 85 | add_keyword("const", "const $0"); |
90 | add_keyword(ctx, acc, "loop", "loop {\n $0\n}"); | 86 | add_keyword("type", "type $0"); |
91 | add_keyword(ctx, acc, "if", "if $1 {\n $0\n}"); | ||
92 | add_keyword(ctx, acc, "if let", "if let $1 = $2 {\n $0\n}"); | ||
93 | add_keyword(ctx, acc, "for", "for $1 in $2 {\n $0\n}"); | ||
94 | } | 87 | } |
95 | 88 | ||
96 | if ctx.previous_token_is(T![if]) || ctx.previous_token_is(T![while]) || has_block_expr_parent { | 89 | if expects_item || has_block_expr_parent { |
97 | add_keyword(ctx, acc, "let", "let "); | 90 | add_keyword("use", "use $0"); |
91 | add_keyword("impl", "impl $1 {\n $0\n}"); | ||
92 | add_keyword("trait", "trait $1 {\n $0\n}"); | ||
93 | add_keyword("static", "static $0"); | ||
94 | add_keyword("extern", "extern $0"); | ||
95 | add_keyword("mod", "mod $0"); | ||
98 | } | 96 | } |
99 | 97 | ||
100 | if ctx.after_if { | 98 | if expects_item { |
101 | add_keyword(ctx, acc, "else", "else {\n $0\n}"); | 99 | add_keyword("enum", "enum $1 {\n $0\n}"); |
102 | add_keyword(ctx, acc, "else if", "else if $1 {\n $0\n}"); | 100 | add_keyword("struct", "struct $0"); |
103 | } | 101 | add_keyword("union", "union $1 {\n $0\n}"); |
104 | if has_item_list_parent || has_block_expr_parent { | ||
105 | add_keyword(ctx, acc, "mod", "mod $0"); | ||
106 | } | 102 | } |
107 | if ctx.has_ident_or_ref_pat_parent() { | 103 | |
108 | add_keyword(ctx, acc, "mut", "mut "); | 104 | if ctx.expects_expression() { |
105 | if !has_block_expr_parent { | ||
106 | add_keyword("unsafe", "unsafe {\n $0\n}"); | ||
107 | } | ||
108 | add_keyword("match", "match $1 {\n $0\n}"); | ||
109 | add_keyword("while", "while $1 {\n $0\n}"); | ||
110 | add_keyword("while let", "while let $1 = $2 {\n $0\n}"); | ||
111 | add_keyword("loop", "loop {\n $0\n}"); | ||
112 | add_keyword("if", "if $1 {\n $0\n}"); | ||
113 | add_keyword("if let", "if let $1 = $2 {\n $0\n}"); | ||
114 | add_keyword("for", "for $1 in $2 {\n $0\n}"); | ||
109 | } | 115 | } |
110 | if has_item_list_parent || has_trait_or_impl_parent || has_block_expr_parent { | 116 | |
111 | add_keyword(ctx, acc, "const", "const "); | 117 | if ctx.previous_token_is(T![if]) || ctx.previous_token_is(T![while]) || has_block_expr_parent { |
112 | add_keyword(ctx, acc, "type", "type "); | 118 | add_keyword("let", "let "); |
113 | } | 119 | } |
114 | if has_item_list_parent || has_block_expr_parent { | 120 | |
115 | add_keyword(ctx, acc, "static", "static "); | 121 | if ctx.after_if() { |
116 | }; | 122 | add_keyword("else", "else {\n $0\n}"); |
117 | if has_item_list_parent || has_block_expr_parent { | 123 | add_keyword("else if", "else if $1 {\n $0\n}"); |
118 | add_keyword(ctx, acc, "extern", "extern "); | ||
119 | } | 124 | } |
120 | if has_item_list_parent || has_trait_or_impl_parent || has_block_expr_parent || ctx.is_match_arm | 125 | |
121 | { | 126 | if ctx.expects_ident_pat_or_ref_expr() { |
122 | add_keyword(ctx, acc, "unsafe", "unsafe "); | 127 | add_keyword("mut", "mut "); |
123 | } | 128 | } |
129 | |||
124 | if ctx.in_loop_body { | 130 | if ctx.in_loop_body { |
125 | if ctx.can_be_stmt { | 131 | if ctx.can_be_stmt { |
126 | add_keyword(ctx, acc, "continue", "continue;"); | 132 | add_keyword("continue", "continue;"); |
127 | add_keyword(ctx, acc, "break", "break;"); | 133 | add_keyword("break", "break;"); |
128 | } else { | 134 | } else { |
129 | add_keyword(ctx, acc, "continue", "continue"); | 135 | add_keyword("continue", "continue"); |
130 | add_keyword(ctx, acc, "break", "break"); | 136 | add_keyword("break", "break"); |
131 | } | 137 | } |
132 | } | 138 | } |
133 | if has_item_list_parent || ctx.has_impl_parent() || ctx.has_field_list_parent() { | ||
134 | add_keyword(ctx, acc, "pub(crate)", "pub(crate) "); | ||
135 | add_keyword(ctx, acc, "pub", "pub "); | ||
136 | } | ||
137 | 139 | ||
138 | if !ctx.is_trivial_path { | 140 | if !ctx.is_trivial_path { |
139 | return; | 141 | return; |
@@ -144,8 +146,6 @@ pub(crate) fn complete_expr_keyword(acc: &mut Completions, ctx: &CompletionConte | |||
144 | }; | 146 | }; |
145 | 147 | ||
146 | add_keyword( | 148 | add_keyword( |
147 | ctx, | ||
148 | acc, | ||
149 | "return", | 149 | "return", |
150 | match (ctx.can_be_stmt, fn_def.ret_type().is_some()) { | 150 | match (ctx.can_be_stmt, fn_def.ret_type().is_some()) { |
151 | (true, true) => "return $0;", | 151 | (true, true) => "return $0;", |
@@ -162,15 +162,12 @@ fn add_keyword(ctx: &CompletionContext, acc: &mut Completions, kw: &str, snippet | |||
162 | 162 | ||
163 | match ctx.config.snippet_cap { | 163 | match ctx.config.snippet_cap { |
164 | Some(cap) => { | 164 | Some(cap) => { |
165 | let tmp; | 165 | if snippet.ends_with('}') && ctx.incomplete_let { |
166 | let snippet = if snippet.ends_with('}') && ctx.incomplete_let { | ||
167 | cov_mark::hit!(let_semi); | 166 | cov_mark::hit!(let_semi); |
168 | tmp = format!("{};", snippet); | 167 | item.insert_snippet(cap, format!("{};", snippet)); |
169 | &tmp | ||
170 | } else { | 168 | } else { |
171 | snippet | 169 | item.insert_snippet(cap, snippet); |
172 | }; | 170 | } |
173 | item.insert_snippet(cap, snippet); | ||
174 | } | 171 | } |
175 | None => { | 172 | None => { |
176 | item.insert_text(if snippet.contains('$') { kw } else { snippet }); | 173 | item.insert_text(if snippet.contains('$') { kw } else { snippet }); |
@@ -233,21 +230,21 @@ mod tests { | |||
233 | check( | 230 | check( |
234 | r"m$0", | 231 | r"m$0", |
235 | expect![[r#" | 232 | expect![[r#" |
233 | kw pub(crate) | ||
234 | kw pub | ||
235 | kw unsafe | ||
236 | kw fn | 236 | kw fn |
237 | kw const | ||
238 | kw type | ||
237 | kw use | 239 | kw use |
238 | kw impl | 240 | kw impl |
239 | kw trait | 241 | kw trait |
242 | kw static | ||
243 | kw extern | ||
244 | kw mod | ||
240 | kw enum | 245 | kw enum |
241 | kw struct | 246 | kw struct |
242 | kw union | 247 | kw union |
243 | kw mod | ||
244 | kw const | ||
245 | kw type | ||
246 | kw static | ||
247 | kw extern | ||
248 | kw unsafe | ||
249 | kw pub(crate) | ||
250 | kw pub | ||
251 | "#]], | 248 | "#]], |
252 | ); | 249 | ); |
253 | } | 250 | } |
@@ -257,10 +254,16 @@ mod tests { | |||
257 | check( | 254 | check( |
258 | r"fn quux() { $0 }", | 255 | r"fn quux() { $0 }", |
259 | expect![[r#" | 256 | expect![[r#" |
257 | kw unsafe | ||
260 | kw fn | 258 | kw fn |
259 | kw const | ||
260 | kw type | ||
261 | kw use | 261 | kw use |
262 | kw impl | 262 | kw impl |
263 | kw trait | 263 | kw trait |
264 | kw static | ||
265 | kw extern | ||
266 | kw mod | ||
264 | kw match | 267 | kw match |
265 | kw while | 268 | kw while |
266 | kw while let | 269 | kw while let |
@@ -269,12 +272,6 @@ mod tests { | |||
269 | kw if let | 272 | kw if let |
270 | kw for | 273 | kw for |
271 | kw let | 274 | kw let |
272 | kw mod | ||
273 | kw const | ||
274 | kw type | ||
275 | kw static | ||
276 | kw extern | ||
277 | kw unsafe | ||
278 | kw return | 275 | kw return |
279 | "#]], | 276 | "#]], |
280 | ); | 277 | ); |
@@ -285,10 +282,16 @@ mod tests { | |||
285 | check( | 282 | check( |
286 | r"fn quux() { if true { $0 } }", | 283 | r"fn quux() { if true { $0 } }", |
287 | expect![[r#" | 284 | expect![[r#" |
285 | kw unsafe | ||
288 | kw fn | 286 | kw fn |
287 | kw const | ||
288 | kw type | ||
289 | kw use | 289 | kw use |
290 | kw impl | 290 | kw impl |
291 | kw trait | 291 | kw trait |
292 | kw static | ||
293 | kw extern | ||
294 | kw mod | ||
292 | kw match | 295 | kw match |
293 | kw while | 296 | kw while |
294 | kw while let | 297 | kw while let |
@@ -297,12 +300,6 @@ mod tests { | |||
297 | kw if let | 300 | kw if let |
298 | kw for | 301 | kw for |
299 | kw let | 302 | kw let |
300 | kw mod | ||
301 | kw const | ||
302 | kw type | ||
303 | kw static | ||
304 | kw extern | ||
305 | kw unsafe | ||
306 | kw return | 303 | kw return |
307 | "#]], | 304 | "#]], |
308 | ); | 305 | ); |
@@ -313,10 +310,16 @@ mod tests { | |||
313 | check( | 310 | check( |
314 | r#"fn quux() { if true { () } $0 }"#, | 311 | r#"fn quux() { if true { () } $0 }"#, |
315 | expect![[r#" | 312 | expect![[r#" |
313 | kw unsafe | ||
316 | kw fn | 314 | kw fn |
315 | kw const | ||
316 | kw type | ||
317 | kw use | 317 | kw use |
318 | kw impl | 318 | kw impl |
319 | kw trait | 319 | kw trait |
320 | kw static | ||
321 | kw extern | ||
322 | kw mod | ||
320 | kw match | 323 | kw match |
321 | kw while | 324 | kw while |
322 | kw while let | 325 | kw while let |
@@ -327,12 +330,6 @@ mod tests { | |||
327 | kw let | 330 | kw let |
328 | kw else | 331 | kw else |
329 | kw else if | 332 | kw else if |
330 | kw mod | ||
331 | kw const | ||
332 | kw type | ||
333 | kw static | ||
334 | kw extern | ||
335 | kw unsafe | ||
336 | kw return | 333 | kw return |
337 | "#]], | 334 | "#]], |
338 | ); | 335 | ); |
@@ -354,6 +351,7 @@ fn quux() -> i32 { | |||
354 | } | 351 | } |
355 | "#, | 352 | "#, |
356 | expect![[r#" | 353 | expect![[r#" |
354 | kw unsafe | ||
357 | kw match | 355 | kw match |
358 | kw while | 356 | kw while |
359 | kw while let | 357 | kw while let |
@@ -361,7 +359,6 @@ fn quux() -> i32 { | |||
361 | kw if | 359 | kw if |
362 | kw if let | 360 | kw if let |
363 | kw for | 361 | kw for |
364 | kw unsafe | ||
365 | kw return | 362 | kw return |
366 | "#]], | 363 | "#]], |
367 | ); | 364 | ); |
@@ -372,10 +369,10 @@ fn quux() -> i32 { | |||
372 | check( | 369 | check( |
373 | r"trait My { $0 }", | 370 | r"trait My { $0 }", |
374 | expect![[r#" | 371 | expect![[r#" |
372 | kw unsafe | ||
375 | kw fn | 373 | kw fn |
376 | kw const | 374 | kw const |
377 | kw type | 375 | kw type |
378 | kw unsafe | ||
379 | "#]], | 376 | "#]], |
380 | ); | 377 | ); |
381 | } | 378 | } |
@@ -385,12 +382,27 @@ fn quux() -> i32 { | |||
385 | check( | 382 | check( |
386 | r"impl My { $0 }", | 383 | r"impl My { $0 }", |
387 | expect![[r#" | 384 | expect![[r#" |
385 | kw pub(crate) | ||
386 | kw pub | ||
387 | kw unsafe | ||
388 | kw fn | 388 | kw fn |
389 | kw const | 389 | kw const |
390 | kw type | 390 | kw type |
391 | kw unsafe | 391 | "#]], |
392 | ); | ||
393 | } | ||
394 | |||
395 | #[test] | ||
396 | fn test_keywords_in_impl_def_with_attr() { | ||
397 | check( | ||
398 | r"impl My { #[foo] $0 }", | ||
399 | expect![[r#" | ||
392 | kw pub(crate) | 400 | kw pub(crate) |
393 | kw pub | 401 | kw pub |
402 | kw unsafe | ||
403 | kw fn | ||
404 | kw const | ||
405 | kw type | ||
394 | "#]], | 406 | "#]], |
395 | ); | 407 | ); |
396 | } | 408 | } |
@@ -400,10 +412,16 @@ fn quux() -> i32 { | |||
400 | check( | 412 | check( |
401 | r"fn my() { loop { $0 } }", | 413 | r"fn my() { loop { $0 } }", |
402 | expect![[r#" | 414 | expect![[r#" |
415 | kw unsafe | ||
403 | kw fn | 416 | kw fn |
417 | kw const | ||
418 | kw type | ||
404 | kw use | 419 | kw use |
405 | kw impl | 420 | kw impl |
406 | kw trait | 421 | kw trait |
422 | kw static | ||
423 | kw extern | ||
424 | kw mod | ||
407 | kw match | 425 | kw match |
408 | kw while | 426 | kw while |
409 | kw while let | 427 | kw while let |
@@ -412,12 +430,6 @@ fn quux() -> i32 { | |||
412 | kw if let | 430 | kw if let |
413 | kw for | 431 | kw for |
414 | kw let | 432 | kw let |
415 | kw mod | ||
416 | kw const | ||
417 | kw type | ||
418 | kw static | ||
419 | kw extern | ||
420 | kw unsafe | ||
421 | kw continue | 433 | kw continue |
422 | kw break | 434 | kw break |
423 | kw return | 435 | kw return |
@@ -564,6 +576,7 @@ pub mod future { | |||
564 | check( | 576 | check( |
565 | r#"fn main() { let _ = $0 }"#, | 577 | r#"fn main() { let _ = $0 }"#, |
566 | expect![[r#" | 578 | expect![[r#" |
579 | kw unsafe | ||
567 | kw match | 580 | kw match |
568 | kw while | 581 | kw while |
569 | kw while let | 582 | kw while let |
@@ -624,6 +637,7 @@ fn foo() { | |||
624 | } | 637 | } |
625 | "#, | 638 | "#, |
626 | expect![[r#" | 639 | expect![[r#" |
640 | kw unsafe | ||
627 | kw match | 641 | kw match |
628 | kw while | 642 | kw while |
629 | kw while let | 643 | kw while let |
diff --git a/crates/ide_completion/src/completions/lifetime.rs b/crates/ide_completion/src/completions/lifetime.rs index 5f6285b84..8ccccb646 100644 --- a/crates/ide_completion/src/completions/lifetime.rs +++ b/crates/ide_completion/src/completions/lifetime.rs | |||
@@ -16,15 +16,14 @@ pub(crate) fn complete_lifetime(acc: &mut Completions, ctx: &CompletionContext) | |||
16 | (Some(lt), Some(lp)) if lp == lt.clone() => return, | 16 | (Some(lt), Some(lp)) if lp == lt.clone() => return, |
17 | (Some(_), Some(lp)) => { | 17 | (Some(_), Some(lp)) => { |
18 | lp_string = lp.to_string(); | 18 | lp_string = lp.to_string(); |
19 | Some(&lp_string) | 19 | Some(&*lp_string) |
20 | } | 20 | } |
21 | _ => None, | 21 | _ => None, |
22 | }; | 22 | }; |
23 | 23 | ||
24 | ctx.scope.process_all_names(&mut |name, res| { | 24 | ctx.scope.process_all_names(&mut |name, res| { |
25 | if let ScopeDef::GenericParam(hir::GenericParam::LifetimeParam(_)) = res { | 25 | if let ScopeDef::GenericParam(hir::GenericParam::LifetimeParam(_)) = res { |
26 | let name = name.to_string(); | 26 | if param_lifetime != Some(&*name.to_string()) { |
27 | if param_lifetime != Some(&name) { | ||
28 | acc.add_resolution(ctx, name, &res); | 27 | acc.add_resolution(ctx, name, &res); |
29 | } | 28 | } |
30 | } | 29 | } |
@@ -41,7 +40,7 @@ pub(crate) fn complete_label(acc: &mut Completions, ctx: &CompletionContext) { | |||
41 | } | 40 | } |
42 | ctx.scope.process_all_names(&mut |name, res| { | 41 | ctx.scope.process_all_names(&mut |name, res| { |
43 | if let ScopeDef::Label(_) = res { | 42 | if let ScopeDef::Label(_) = res { |
44 | acc.add_resolution(ctx, name.to_string(), &res); | 43 | acc.add_resolution(ctx, name, &res); |
45 | } | 44 | } |
46 | }); | 45 | }); |
47 | } | 46 | } |
diff --git a/crates/ide_completion/src/completions/macro_in_item_position.rs b/crates/ide_completion/src/completions/macro_in_item_position.rs index c5e377500..202e71215 100644 --- a/crates/ide_completion/src/completions/macro_in_item_position.rs +++ b/crates/ide_completion/src/completions/macro_in_item_position.rs | |||
@@ -2,6 +2,7 @@ | |||
2 | 2 | ||
3 | use crate::{CompletionContext, Completions}; | 3 | use crate::{CompletionContext, Completions}; |
4 | 4 | ||
5 | // Ideally this should be removed and moved into `(un)qualified_path` respectively | ||
5 | pub(crate) fn complete_macro_in_item_position(acc: &mut Completions, ctx: &CompletionContext) { | 6 | pub(crate) fn complete_macro_in_item_position(acc: &mut Completions, ctx: &CompletionContext) { |
6 | // Show only macros in top level. | 7 | // Show only macros in top level. |
7 | if !ctx.is_new_item { | 8 | if !ctx.is_new_item { |
@@ -10,7 +11,11 @@ pub(crate) fn complete_macro_in_item_position(acc: &mut Completions, ctx: &Compl | |||
10 | 11 | ||
11 | ctx.scope.process_all_names(&mut |name, res| { | 12 | ctx.scope.process_all_names(&mut |name, res| { |
12 | if let hir::ScopeDef::MacroDef(mac) = res { | 13 | if let hir::ScopeDef::MacroDef(mac) = res { |
13 | acc.add_macro(ctx, Some(name.to_string()), mac); | 14 | acc.add_macro(ctx, Some(name.clone()), mac); |
15 | } | ||
16 | // FIXME: This should be done in qualified_path/unqualified_path instead? | ||
17 | if let hir::ScopeDef::ModuleDef(hir::ModuleDef::Module(_)) = res { | ||
18 | acc.add_resolution(ctx, name, &res); | ||
14 | } | 19 | } |
15 | }) | 20 | }) |
16 | } | 21 | } |
diff --git a/crates/ide_completion/src/completions/mod_.rs b/crates/ide_completion/src/completions/mod_.rs index 4f9415736..6a5746fb9 100644 --- a/crates/ide_completion/src/completions/mod_.rs +++ b/crates/ide_completion/src/completions/mod_.rs | |||
@@ -9,14 +9,14 @@ use ide_db::{ | |||
9 | }; | 9 | }; |
10 | use rustc_hash::FxHashSet; | 10 | use rustc_hash::FxHashSet; |
11 | 11 | ||
12 | use crate::CompletionItem; | 12 | use crate::{patterns::ImmediateLocation, CompletionItem}; |
13 | 13 | ||
14 | use crate::{context::CompletionContext, item::CompletionKind, Completions}; | 14 | use crate::{context::CompletionContext, item::CompletionKind, Completions}; |
15 | 15 | ||
16 | /// Complete mod declaration, i.e. `mod $0 ;` | 16 | /// Complete mod declaration, i.e. `mod $0 ;` |
17 | pub(crate) fn complete_mod(acc: &mut Completions, ctx: &CompletionContext) -> Option<()> { | 17 | pub(crate) fn complete_mod(acc: &mut Completions, ctx: &CompletionContext) -> Option<()> { |
18 | let mod_under_caret = match &ctx.mod_declaration_under_caret { | 18 | let mod_under_caret = match &ctx.completion_location { |
19 | Some(mod_under_caret) if mod_under_caret.item_list().is_none() => mod_under_caret, | 19 | Some(ImmediateLocation::ModDeclaration(mod_under_caret)) => mod_under_caret, |
20 | _ => return None, | 20 | _ => return None, |
21 | }; | 21 | }; |
22 | 22 | ||
diff --git a/crates/ide_completion/src/completions/pattern.rs b/crates/ide_completion/src/completions/pattern.rs index b84e9a967..8a728c67e 100644 --- a/crates/ide_completion/src/completions/pattern.rs +++ b/crates/ide_completion/src/completions/pattern.rs | |||
@@ -51,7 +51,7 @@ pub(crate) fn complete_pattern(acc: &mut Completions, ctx: &CompletionContext) { | |||
51 | _ => false, | 51 | _ => false, |
52 | }; | 52 | }; |
53 | if add_resolution { | 53 | if add_resolution { |
54 | acc.add_resolution(ctx, name.to_string(), &res); | 54 | acc.add_resolution(ctx, name, &res); |
55 | } | 55 | } |
56 | }); | 56 | }); |
57 | } | 57 | } |
diff --git a/crates/ide_completion/src/completions/qualified_path.rs b/crates/ide_completion/src/completions/qualified_path.rs index ed48f61af..de58ce1cd 100644 --- a/crates/ide_completion/src/completions/qualified_path.rs +++ b/crates/ide_completion/src/completions/qualified_path.rs | |||
@@ -1,13 +1,13 @@ | |||
1 | //! Completion of paths, i.e. `some::prefix::$0`. | 1 | //! Completion of paths, i.e. `some::prefix::$0`. |
2 | 2 | ||
3 | use hir::{Adt, HasVisibility, PathResolution, ScopeDef}; | 3 | use hir::HasVisibility; |
4 | use rustc_hash::FxHashSet; | 4 | use rustc_hash::FxHashSet; |
5 | use syntax::AstNode; | 5 | use syntax::AstNode; |
6 | 6 | ||
7 | use crate::{CompletionContext, Completions}; | 7 | use crate::{CompletionContext, Completions}; |
8 | 8 | ||
9 | pub(crate) fn complete_qualified_path(acc: &mut Completions, ctx: &CompletionContext) { | 9 | pub(crate) fn complete_qualified_path(acc: &mut Completions, ctx: &CompletionContext) { |
10 | if ctx.is_path_disallowed() { | 10 | if ctx.is_path_disallowed() || ctx.expects_item() { |
11 | return; | 11 | return; |
12 | } | 12 | } |
13 | let path = match &ctx.path_qual { | 13 | let path = match &ctx.path_qual { |
@@ -20,6 +20,20 @@ pub(crate) fn complete_qualified_path(acc: &mut Completions, ctx: &CompletionCon | |||
20 | None => return, | 20 | None => return, |
21 | }; | 21 | }; |
22 | let context_module = ctx.scope.module(); | 22 | let context_module = ctx.scope.module(); |
23 | if ctx.expects_assoc_item() { | ||
24 | if let hir::PathResolution::Def(hir::ModuleDef::Module(module)) = resolution { | ||
25 | let module_scope = module.scope(ctx.db, context_module); | ||
26 | for (name, def) in module_scope { | ||
27 | if let hir::ScopeDef::MacroDef(macro_def) = def { | ||
28 | acc.add_macro(ctx, Some(name.clone()), macro_def); | ||
29 | } | ||
30 | if let hir::ScopeDef::ModuleDef(hir::ModuleDef::Module(_)) = def { | ||
31 | acc.add_resolution(ctx, name, &def); | ||
32 | } | ||
33 | } | ||
34 | } | ||
35 | return; | ||
36 | } | ||
23 | 37 | ||
24 | // Add associated types on type parameters and `Self`. | 38 | // Add associated types on type parameters and `Self`. |
25 | resolution.assoc_type_shorthand_candidates(ctx.db, |_, alias| { | 39 | resolution.assoc_type_shorthand_candidates(ctx.db, |_, alias| { |
@@ -28,11 +42,11 @@ pub(crate) fn complete_qualified_path(acc: &mut Completions, ctx: &CompletionCon | |||
28 | }); | 42 | }); |
29 | 43 | ||
30 | match resolution { | 44 | match resolution { |
31 | PathResolution::Def(hir::ModuleDef::Module(module)) => { | 45 | hir::PathResolution::Def(hir::ModuleDef::Module(module)) => { |
32 | let module_scope = module.scope(ctx.db, context_module); | 46 | let module_scope = module.scope(ctx.db, context_module); |
33 | for (name, def) in module_scope { | 47 | for (name, def) in module_scope { |
34 | if ctx.use_item_syntax.is_some() { | 48 | if ctx.use_item_syntax.is_some() { |
35 | if let ScopeDef::Unknown = def { | 49 | if let hir::ScopeDef::Unknown = def { |
36 | if let Some(name_ref) = ctx.name_ref_syntax.as_ref() { | 50 | if let Some(name_ref) = ctx.name_ref_syntax.as_ref() { |
37 | if name_ref.syntax().text() == name.to_string().as_str() { | 51 | if name_ref.syntax().text() == name.to_string().as_str() { |
38 | // for `use self::foo$0`, don't suggest `foo` as a completion | 52 | // for `use self::foo$0`, don't suggest `foo` as a completion |
@@ -43,20 +57,20 @@ pub(crate) fn complete_qualified_path(acc: &mut Completions, ctx: &CompletionCon | |||
43 | } | 57 | } |
44 | } | 58 | } |
45 | 59 | ||
46 | acc.add_resolution(ctx, name.to_string(), &def); | 60 | acc.add_resolution(ctx, name, &def); |
47 | } | 61 | } |
48 | } | 62 | } |
49 | PathResolution::Def(def @ hir::ModuleDef::Adt(_)) | 63 | hir::PathResolution::Def(def @ hir::ModuleDef::Adt(_)) |
50 | | PathResolution::Def(def @ hir::ModuleDef::TypeAlias(_)) | 64 | | hir::PathResolution::Def(def @ hir::ModuleDef::TypeAlias(_)) |
51 | | PathResolution::Def(def @ hir::ModuleDef::BuiltinType(_)) => { | 65 | | hir::PathResolution::Def(def @ hir::ModuleDef::BuiltinType(_)) => { |
52 | if let hir::ModuleDef::Adt(Adt::Enum(e)) = def { | 66 | if let hir::ModuleDef::Adt(hir::Adt::Enum(e)) = def { |
53 | add_enum_variants(ctx, acc, e); | 67 | add_enum_variants(ctx, acc, e); |
54 | } | 68 | } |
55 | let ty = match def { | 69 | let ty = match def { |
56 | hir::ModuleDef::Adt(adt) => adt.ty(ctx.db), | 70 | hir::ModuleDef::Adt(adt) => adt.ty(ctx.db), |
57 | hir::ModuleDef::TypeAlias(a) => { | 71 | hir::ModuleDef::TypeAlias(a) => { |
58 | let ty = a.ty(ctx.db); | 72 | let ty = a.ty(ctx.db); |
59 | if let Some(Adt::Enum(e)) = ty.as_adt() { | 73 | if let Some(hir::Adt::Enum(e)) = ty.as_adt() { |
60 | cov_mark::hit!(completes_variant_through_alias); | 74 | cov_mark::hit!(completes_variant_through_alias); |
61 | add_enum_variants(ctx, acc, e); | 75 | add_enum_variants(ctx, acc, e); |
62 | } | 76 | } |
@@ -103,7 +117,7 @@ pub(crate) fn complete_qualified_path(acc: &mut Completions, ctx: &CompletionCon | |||
103 | }); | 117 | }); |
104 | } | 118 | } |
105 | } | 119 | } |
106 | PathResolution::Def(hir::ModuleDef::Trait(t)) => { | 120 | hir::PathResolution::Def(hir::ModuleDef::Trait(t)) => { |
107 | // Handles `Trait::assoc` as well as `<Ty as Trait>::assoc`. | 121 | // Handles `Trait::assoc` as well as `<Ty as Trait>::assoc`. |
108 | for item in t.items(ctx.db) { | 122 | for item in t.items(ctx.db) { |
109 | if context_module.map_or(false, |m| !item.is_visible_from(ctx.db, m)) { | 123 | if context_module.map_or(false, |m| !item.is_visible_from(ctx.db, m)) { |
@@ -116,15 +130,15 @@ pub(crate) fn complete_qualified_path(acc: &mut Completions, ctx: &CompletionCon | |||
116 | } | 130 | } |
117 | } | 131 | } |
118 | } | 132 | } |
119 | PathResolution::TypeParam(_) | PathResolution::SelfType(_) => { | 133 | hir::PathResolution::TypeParam(_) | hir::PathResolution::SelfType(_) => { |
120 | if let Some(krate) = ctx.krate { | 134 | if let Some(krate) = ctx.krate { |
121 | let ty = match resolution { | 135 | let ty = match resolution { |
122 | PathResolution::TypeParam(param) => param.ty(ctx.db), | 136 | hir::PathResolution::TypeParam(param) => param.ty(ctx.db), |
123 | PathResolution::SelfType(impl_def) => impl_def.self_ty(ctx.db), | 137 | hir::PathResolution::SelfType(impl_def) => impl_def.self_ty(ctx.db), |
124 | _ => return, | 138 | _ => return, |
125 | }; | 139 | }; |
126 | 140 | ||
127 | if let Some(Adt::Enum(e)) = ty.as_adt() { | 141 | if let Some(hir::Adt::Enum(e)) = ty.as_adt() { |
128 | add_enum_variants(ctx, acc, e); | 142 | add_enum_variants(ctx, acc, e); |
129 | } | 143 | } |
130 | 144 | ||
@@ -594,7 +608,7 @@ fn main() { T::$0; } | |||
594 | macro_rules! foo { () => {} } | 608 | macro_rules! foo { () => {} } |
595 | 609 | ||
596 | fn main() { let _ = crate::$0 } | 610 | fn main() { let _ = crate::$0 } |
597 | "#, | 611 | "#, |
598 | expect![[r##" | 612 | expect![[r##" |
599 | fn main() fn() | 613 | fn main() fn() |
600 | ma foo!(…) #[macro_export] macro_rules! foo | 614 | ma foo!(…) #[macro_export] macro_rules! foo |
@@ -603,6 +617,44 @@ fn main() { let _ = crate::$0 } | |||
603 | } | 617 | } |
604 | 618 | ||
605 | #[test] | 619 | #[test] |
620 | fn completes_in_assoc_item_list() { | ||
621 | check( | ||
622 | r#" | ||
623 | #[macro_export] | ||
624 | macro_rules! foo { () => {} } | ||
625 | mod bar {} | ||
626 | |||
627 | struct MyStruct {} | ||
628 | impl MyStruct { | ||
629 | crate::$0 | ||
630 | } | ||
631 | "#, | ||
632 | expect![[r##" | ||
633 | md bar | ||
634 | ma foo! #[macro_export] macro_rules! foo | ||
635 | "##]], | ||
636 | ); | ||
637 | } | ||
638 | |||
639 | #[test] | ||
640 | #[ignore] // FIXME doesn't complete anything atm | ||
641 | fn completes_in_item_list() { | ||
642 | check( | ||
643 | r#" | ||
644 | struct MyStruct {} | ||
645 | macro_rules! foo {} | ||
646 | mod bar {} | ||
647 | |||
648 | crate::$0 | ||
649 | "#, | ||
650 | expect![[r#" | ||
651 | md bar | ||
652 | ma foo! macro_rules! foo | ||
653 | "#]], | ||
654 | ) | ||
655 | } | ||
656 | |||
657 | #[test] | ||
606 | fn test_super_super_completion() { | 658 | fn test_super_super_completion() { |
607 | check( | 659 | check( |
608 | r#" | 660 | r#" |
diff --git a/crates/ide_completion/src/completions/record.rs b/crates/ide_completion/src/completions/record.rs index e1526b70b..227c08d01 100644 --- a/crates/ide_completion/src/completions/record.rs +++ b/crates/ide_completion/src/completions/record.rs | |||
@@ -2,21 +2,21 @@ | |||
2 | use ide_db::{helpers::FamousDefs, SymbolKind}; | 2 | use ide_db::{helpers::FamousDefs, SymbolKind}; |
3 | use syntax::ast::Expr; | 3 | use syntax::ast::Expr; |
4 | 4 | ||
5 | use crate::{item::CompletionKind, CompletionContext, CompletionItem, Completions}; | 5 | use crate::{ |
6 | item::CompletionKind, patterns::ImmediateLocation, CompletionContext, CompletionItem, | ||
7 | Completions, | ||
8 | }; | ||
6 | 9 | ||
7 | pub(crate) fn complete_record(acc: &mut Completions, ctx: &CompletionContext) -> Option<()> { | 10 | pub(crate) fn complete_record(acc: &mut Completions, ctx: &CompletionContext) -> Option<()> { |
8 | let missing_fields = match (ctx.record_pat_syntax.as_ref(), ctx.record_lit_syntax.as_ref()) { | 11 | let missing_fields = match &ctx.completion_location { |
9 | (None, None) => return None, | 12 | Some(ImmediateLocation::RecordExpr(record_expr)) => { |
10 | (Some(_), Some(_)) => unreachable!("A record cannot be both a literal and a pattern"), | 13 | let ty = ctx.sema.type_of_expr(&Expr::RecordExpr(record_expr.clone())); |
11 | (Some(record_pat), _) => ctx.sema.record_pattern_missing_fields(record_pat), | ||
12 | (_, Some(record_lit)) => { | ||
13 | let ty = ctx.sema.type_of_expr(&Expr::RecordExpr(record_lit.clone())); | ||
14 | let default_trait = FamousDefs(&ctx.sema, ctx.krate).core_default_Default(); | 14 | let default_trait = FamousDefs(&ctx.sema, ctx.krate).core_default_Default(); |
15 | let impl_default_trait = default_trait | 15 | let impl_default_trait = default_trait |
16 | .zip(ty) | 16 | .zip(ty) |
17 | .map_or(false, |(default_trait, ty)| ty.impls_trait(ctx.db, default_trait, &[])); | 17 | .map_or(false, |(default_trait, ty)| ty.impls_trait(ctx.db, default_trait, &[])); |
18 | 18 | ||
19 | let missing_fields = ctx.sema.record_literal_missing_fields(record_lit); | 19 | let missing_fields = ctx.sema.record_literal_missing_fields(record_expr); |
20 | if impl_default_trait && !missing_fields.is_empty() { | 20 | if impl_default_trait && !missing_fields.is_empty() { |
21 | let completion_text = "..Default::default()"; | 21 | let completion_text = "..Default::default()"; |
22 | let mut item = CompletionItem::new( | 22 | let mut item = CompletionItem::new( |
@@ -32,6 +32,10 @@ pub(crate) fn complete_record(acc: &mut Completions, ctx: &CompletionContext) -> | |||
32 | 32 | ||
33 | missing_fields | 33 | missing_fields |
34 | } | 34 | } |
35 | Some(ImmediateLocation::RecordPat(record_pat)) => { | ||
36 | ctx.sema.record_pattern_missing_fields(record_pat) | ||
37 | } | ||
38 | _ => return None, | ||
35 | }; | 39 | }; |
36 | 40 | ||
37 | for (field, ty) in missing_fields { | 41 | for (field, ty) in missing_fields { |
diff --git a/crates/ide_completion/src/completions/unqualified_path.rs b/crates/ide_completion/src/completions/unqualified_path.rs index 046a393ae..9db8516d0 100644 --- a/crates/ide_completion/src/completions/unqualified_path.rs +++ b/crates/ide_completion/src/completions/unqualified_path.rs | |||
@@ -1,7 +1,6 @@ | |||
1 | //! Completion of names from the current scope, e.g. locals and imported items. | 1 | //! Completion of names from the current scope, e.g. locals and imported items. |
2 | 2 | ||
3 | use hir::ScopeDef; | 3 | use hir::ScopeDef; |
4 | use syntax::AstNode; | ||
5 | 4 | ||
6 | use crate::{CompletionContext, Completions}; | 5 | use crate::{CompletionContext, Completions}; |
7 | 6 | ||
@@ -9,10 +8,30 @@ pub(crate) fn complete_unqualified_path(acc: &mut Completions, ctx: &CompletionC | |||
9 | if !ctx.is_trivial_path { | 8 | if !ctx.is_trivial_path { |
10 | return; | 9 | return; |
11 | } | 10 | } |
12 | if ctx.is_path_disallowed() { | 11 | if ctx.is_path_disallowed() || ctx.expects_item() { |
12 | return; | ||
13 | } | ||
14 | if ctx.expects_assoc_item() { | ||
15 | ctx.scope.process_all_names(&mut |name, def| { | ||
16 | if let ScopeDef::MacroDef(macro_def) = def { | ||
17 | acc.add_macro(ctx, Some(name.clone()), macro_def); | ||
18 | } | ||
19 | if let ScopeDef::ModuleDef(hir::ModuleDef::Module(_)) = def { | ||
20 | acc.add_resolution(ctx, name, &def); | ||
21 | } | ||
22 | }); | ||
13 | return; | 23 | return; |
14 | } | 24 | } |
15 | 25 | ||
26 | if ctx.expects_use_tree() { | ||
27 | cov_mark::hit!(only_completes_modules_in_import); | ||
28 | ctx.scope.process_all_names(&mut |name, res| { | ||
29 | if let ScopeDef::ModuleDef(hir::ModuleDef::Module(_)) = res { | ||
30 | acc.add_resolution(ctx, name, &res); | ||
31 | } | ||
32 | }); | ||
33 | return; | ||
34 | } | ||
16 | if let Some(hir::Adt::Enum(e)) = | 35 | if let Some(hir::Adt::Enum(e)) = |
17 | ctx.expected_type.as_ref().and_then(|ty| ty.strip_references().as_adt()) | 36 | ctx.expected_type.as_ref().and_then(|ty| ty.strip_references().as_adt()) |
18 | { | 37 | { |
@@ -26,15 +45,7 @@ pub(crate) fn complete_unqualified_path(acc: &mut Completions, ctx: &CompletionC | |||
26 | cov_mark::hit!(skip_lifetime_completion); | 45 | cov_mark::hit!(skip_lifetime_completion); |
27 | return; | 46 | return; |
28 | } | 47 | } |
29 | if ctx.use_item_syntax.is_some() { | 48 | acc.add_resolution(ctx, name, &res); |
30 | if let (ScopeDef::Unknown, Some(name_ref)) = (&res, &ctx.name_ref_syntax) { | ||
31 | if name_ref.syntax().text() == name.to_string().as_str() { | ||
32 | cov_mark::hit!(self_fulfilling_completion); | ||
33 | return; | ||
34 | } | ||
35 | } | ||
36 | } | ||
37 | acc.add_resolution(ctx, name.to_string(), &res); | ||
38 | }); | 49 | }); |
39 | } | 50 | } |
40 | 51 | ||
@@ -57,15 +68,17 @@ mod tests { | |||
57 | } | 68 | } |
58 | 69 | ||
59 | #[test] | 70 | #[test] |
60 | fn self_fulfilling_completion() { | 71 | fn only_completes_modules_in_import() { |
61 | cov_mark::check!(self_fulfilling_completion); | 72 | cov_mark::check!(only_completes_modules_in_import); |
62 | check( | 73 | check( |
63 | r#" | 74 | r#" |
64 | use foo$0 | 75 | use f$0 |
65 | use std::collections; | 76 | |
77 | struct Foo; | ||
78 | mod foo {} | ||
66 | "#, | 79 | "#, |
67 | expect![[r#" | 80 | expect![[r#" |
68 | ?? collections | 81 | md foo |
69 | "#]], | 82 | "#]], |
70 | ); | 83 | ); |
71 | } | 84 | } |
@@ -647,7 +660,7 @@ fn f() {} | |||
647 | } | 660 | } |
648 | 661 | ||
649 | #[test] | 662 | #[test] |
650 | fn completes_type_or_trait_in_impl_block() { | 663 | fn completes_target_type_or_trait_in_impl_block() { |
651 | check( | 664 | check( |
652 | r#" | 665 | r#" |
653 | trait MyTrait {} | 666 | trait MyTrait {} |
@@ -662,4 +675,41 @@ impl My$0 | |||
662 | "#]], | 675 | "#]], |
663 | ) | 676 | ) |
664 | } | 677 | } |
678 | |||
679 | #[test] | ||
680 | fn completes_in_assoc_item_list() { | ||
681 | check( | ||
682 | r#" | ||
683 | macro_rules! foo {} | ||
684 | mod bar {} | ||
685 | |||
686 | struct MyStruct {} | ||
687 | impl MyStruct { | ||
688 | $0 | ||
689 | } | ||
690 | "#, | ||
691 | expect![[r#" | ||
692 | md bar | ||
693 | ma foo! macro_rules! foo | ||
694 | "#]], | ||
695 | ) | ||
696 | } | ||
697 | |||
698 | // FIXME: The completions here currently come from `macro_in_item_position`, but they shouldn't | ||
699 | #[test] | ||
700 | fn completes_in_item_list() { | ||
701 | check( | ||
702 | r#" | ||
703 | struct MyStruct {} | ||
704 | macro_rules! foo {} | ||
705 | mod bar {} | ||
706 | |||
707 | $0 | ||
708 | "#, | ||
709 | expect![[r#" | ||
710 | md bar | ||
711 | ma foo!(…) macro_rules! foo | ||
712 | "#]], | ||
713 | ) | ||
714 | } | ||
665 | } | 715 | } |
diff --git a/crates/ide_completion/src/context.rs b/crates/ide_completion/src/context.rs index 66577df94..7c46c815d 100644 --- a/crates/ide_completion/src/context.rs +++ b/crates/ide_completion/src/context.rs | |||
@@ -17,9 +17,8 @@ use text_edit::Indel; | |||
17 | 17 | ||
18 | use crate::{ | 18 | use crate::{ |
19 | patterns::{ | 19 | patterns::{ |
20 | for_is_prev2, has_bind_pat_parent, has_block_expr_parent, has_field_list_parent, | 20 | determine_location, determine_prev_sibling, for_is_prev2, inside_impl_trait_block, |
21 | has_impl_parent, has_item_list_or_source_file_parent, has_prev_sibling, has_ref_parent, | 21 | is_in_loop_body, previous_token, ImmediateLocation, ImmediatePrevSibling, |
22 | has_trait_parent, inside_impl_trait_block, is_in_loop_body, is_match_arm, previous_token, | ||
23 | }, | 22 | }, |
24 | CompletionConfig, | 23 | CompletionConfig, |
25 | }; | 24 | }; |
@@ -30,24 +29,6 @@ pub(crate) enum PatternRefutability { | |||
30 | Irrefutable, | 29 | Irrefutable, |
31 | } | 30 | } |
32 | 31 | ||
33 | /// Direct parent container of the cursor position | ||
34 | #[derive(Copy, Clone, Debug, PartialEq, Eq)] | ||
35 | pub(crate) enum ImmediateLocation { | ||
36 | Impl, | ||
37 | Trait, | ||
38 | RecordFieldList, | ||
39 | RefPatOrExpr, | ||
40 | IdentPat, | ||
41 | BlockExpr, | ||
42 | ItemList, | ||
43 | } | ||
44 | |||
45 | #[derive(Copy, Clone, Debug, PartialEq, Eq)] | ||
46 | pub(crate) enum PrevSibling { | ||
47 | Trait, | ||
48 | Impl, | ||
49 | } | ||
50 | |||
51 | /// `CompletionContext` is created early during completion to figure out, where | 32 | /// `CompletionContext` is created early during completion to figure out, where |
52 | /// exactly is the cursor, syntax-wise. | 33 | /// exactly is the cursor, syntax-wise. |
53 | #[derive(Debug)] | 34 | #[derive(Debug)] |
@@ -73,11 +54,6 @@ pub(crate) struct CompletionContext<'a> { | |||
73 | /// The parent impl of the cursor position if it exists. | 54 | /// The parent impl of the cursor position if it exists. |
74 | pub(super) impl_def: Option<ast::Impl>, | 55 | pub(super) impl_def: Option<ast::Impl>, |
75 | 56 | ||
76 | /// RecordExpr the token is a field of | ||
77 | pub(super) record_lit_syntax: Option<ast::RecordExpr>, | ||
78 | /// RecordPat the token is a field of | ||
79 | pub(super) record_pat_syntax: Option<ast::RecordPat>, | ||
80 | |||
81 | // potentially set if we are completing a lifetime | 57 | // potentially set if we are completing a lifetime |
82 | pub(super) lifetime_syntax: Option<ast::Lifetime>, | 58 | pub(super) lifetime_syntax: Option<ast::Lifetime>, |
83 | pub(super) lifetime_param_syntax: Option<ast::LifetimeParam>, | 59 | pub(super) lifetime_param_syntax: Option<ast::LifetimeParam>, |
@@ -89,6 +65,8 @@ pub(crate) struct CompletionContext<'a> { | |||
89 | pub(super) is_param: bool, | 65 | pub(super) is_param: bool, |
90 | 66 | ||
91 | pub(super) completion_location: Option<ImmediateLocation>, | 67 | pub(super) completion_location: Option<ImmediateLocation>, |
68 | pub(super) prev_sibling: Option<ImmediatePrevSibling>, | ||
69 | pub(super) attribute_under_caret: Option<ast::Attr>, | ||
92 | 70 | ||
93 | /// FIXME: `ActiveParameter` is string-based, which is very very wrong | 71 | /// FIXME: `ActiveParameter` is string-based, which is very very wrong |
94 | pub(super) active_parameter: Option<ActiveParameter>, | 72 | pub(super) active_parameter: Option<ActiveParameter>, |
@@ -96,7 +74,6 @@ pub(crate) struct CompletionContext<'a> { | |||
96 | pub(super) is_trivial_path: bool, | 74 | pub(super) is_trivial_path: bool, |
97 | /// If not a trivial path, the prefix (qualifier). | 75 | /// If not a trivial path, the prefix (qualifier). |
98 | pub(super) path_qual: Option<ast::Path>, | 76 | pub(super) path_qual: Option<ast::Path>, |
99 | pub(super) after_if: bool, | ||
100 | /// `true` if we are a statement or a last expr in the block. | 77 | /// `true` if we are a statement or a last expr in the block. |
101 | pub(super) can_be_stmt: bool, | 78 | pub(super) can_be_stmt: bool, |
102 | /// `true` if we expect an expression at the cursor position. | 79 | /// `true` if we expect an expression at the cursor position. |
@@ -114,19 +91,15 @@ pub(crate) struct CompletionContext<'a> { | |||
114 | pub(super) is_macro_call: bool, | 91 | pub(super) is_macro_call: bool, |
115 | pub(super) is_path_type: bool, | 92 | pub(super) is_path_type: bool, |
116 | pub(super) has_type_args: bool, | 93 | pub(super) has_type_args: bool, |
117 | pub(super) attribute_under_caret: Option<ast::Attr>, | ||
118 | pub(super) mod_declaration_under_caret: Option<ast::Module>, | ||
119 | pub(super) locals: Vec<(String, Local)>, | 94 | pub(super) locals: Vec<(String, Local)>, |
120 | 95 | ||
121 | // keyword patterns | ||
122 | pub(super) previous_token: Option<SyntaxToken>, | 96 | pub(super) previous_token: Option<SyntaxToken>, |
123 | pub(super) prev_sibling: Option<PrevSibling>, | ||
124 | pub(super) in_loop_body: bool, | 97 | pub(super) in_loop_body: bool, |
125 | pub(super) is_match_arm: bool, | ||
126 | pub(super) incomplete_let: bool, | 98 | pub(super) incomplete_let: bool, |
127 | 99 | ||
128 | no_completion_required: bool, | 100 | no_completion_required: bool, |
129 | } | 101 | } |
102 | |||
130 | impl<'a> CompletionContext<'a> { | 103 | impl<'a> CompletionContext<'a> { |
131 | pub(super) fn new( | 104 | pub(super) fn new( |
132 | db: &'a RootDatabase, | 105 | db: &'a RootDatabase, |
@@ -176,8 +149,6 @@ impl<'a> CompletionContext<'a> { | |||
176 | lifetime_param_syntax: None, | 149 | lifetime_param_syntax: None, |
177 | function_def: None, | 150 | function_def: None, |
178 | use_item_syntax: None, | 151 | use_item_syntax: None, |
179 | record_lit_syntax: None, | ||
180 | record_pat_syntax: None, | ||
181 | impl_def: None, | 152 | impl_def: None, |
182 | active_parameter: ActiveParameter::at(db, position), | 153 | active_parameter: ActiveParameter::at(db, position), |
183 | is_label_ref: false, | 154 | is_label_ref: false, |
@@ -185,7 +156,6 @@ impl<'a> CompletionContext<'a> { | |||
185 | is_pat_or_const: None, | 156 | is_pat_or_const: None, |
186 | is_trivial_path: false, | 157 | is_trivial_path: false, |
187 | path_qual: None, | 158 | path_qual: None, |
188 | after_if: false, | ||
189 | can_be_stmt: false, | 159 | can_be_stmt: false, |
190 | is_expr: false, | 160 | is_expr: false, |
191 | is_new_item: false, | 161 | is_new_item: false, |
@@ -196,15 +166,13 @@ impl<'a> CompletionContext<'a> { | |||
196 | is_macro_call: false, | 166 | is_macro_call: false, |
197 | is_path_type: false, | 167 | is_path_type: false, |
198 | has_type_args: false, | 168 | has_type_args: false, |
199 | attribute_under_caret: None, | ||
200 | mod_declaration_under_caret: None, | ||
201 | previous_token: None, | 169 | previous_token: None, |
202 | in_loop_body: false, | 170 | in_loop_body: false, |
203 | completion_location: None, | 171 | completion_location: None, |
204 | prev_sibling: None, | 172 | prev_sibling: None, |
205 | is_match_arm: false, | ||
206 | no_completion_required: false, | 173 | no_completion_required: false, |
207 | incomplete_let: false, | 174 | incomplete_let: false, |
175 | attribute_under_caret: None, | ||
208 | locals, | 176 | locals, |
209 | }; | 177 | }; |
210 | 178 | ||
@@ -247,7 +215,6 @@ impl<'a> CompletionContext<'a> { | |||
247 | break; | 215 | break; |
248 | } | 216 | } |
249 | } | 217 | } |
250 | ctx.fill_keyword_patterns(&speculative_file, offset); | ||
251 | ctx.fill(&original_file, speculative_file, offset); | 218 | ctx.fill(&original_file, speculative_file, offset); |
252 | Some(ctx) | 219 | Some(ctx) |
253 | } | 220 | } |
@@ -281,88 +248,63 @@ impl<'a> CompletionContext<'a> { | |||
281 | self.previous_token.as_ref().map_or(false, |tok| tok.kind() == kind) | 248 | self.previous_token.as_ref().map_or(false, |tok| tok.kind() == kind) |
282 | } | 249 | } |
283 | 250 | ||
284 | pub(crate) fn has_impl_or_trait_parent(&self) -> bool { | 251 | pub(crate) fn expects_assoc_item(&self) -> bool { |
285 | matches!( | 252 | matches!( |
286 | self.completion_location, | 253 | self.completion_location, |
287 | Some(ImmediateLocation::Trait) | Some(ImmediateLocation::Impl) | 254 | Some(ImmediateLocation::Trait) | Some(ImmediateLocation::Impl) |
288 | ) | 255 | ) |
289 | } | 256 | } |
290 | 257 | ||
291 | pub(crate) fn has_block_expr_parent(&self) -> bool { | 258 | pub(crate) fn expects_use_tree(&self) -> bool { |
292 | matches!(self.completion_location, Some(ImmediateLocation::BlockExpr)) | 259 | matches!(self.completion_location, Some(ImmediateLocation::Use)) |
293 | } | 260 | } |
294 | 261 | ||
295 | pub(crate) fn has_item_list_parent(&self) -> bool { | 262 | pub(crate) fn expects_non_trait_assoc_item(&self) -> bool { |
263 | matches!(self.completion_location, Some(ImmediateLocation::Impl)) | ||
264 | } | ||
265 | |||
266 | pub(crate) fn expects_item(&self) -> bool { | ||
296 | matches!(self.completion_location, Some(ImmediateLocation::ItemList)) | 267 | matches!(self.completion_location, Some(ImmediateLocation::ItemList)) |
297 | } | 268 | } |
298 | 269 | ||
299 | pub(crate) fn has_ident_or_ref_pat_parent(&self) -> bool { | 270 | pub(crate) fn expects_expression(&self) -> bool { |
271 | self.is_expr | ||
272 | } | ||
273 | |||
274 | pub(crate) fn has_block_expr_parent(&self) -> bool { | ||
275 | matches!(self.completion_location, Some(ImmediateLocation::BlockExpr)) | ||
276 | } | ||
277 | |||
278 | pub(crate) fn expects_ident_pat_or_ref_expr(&self) -> bool { | ||
300 | matches!( | 279 | matches!( |
301 | self.completion_location, | 280 | self.completion_location, |
302 | Some(ImmediateLocation::IdentPat) | Some(ImmediateLocation::RefPatOrExpr) | 281 | Some(ImmediateLocation::IdentPat) | Some(ImmediateLocation::RefExpr) |
303 | ) | 282 | ) |
304 | } | 283 | } |
305 | 284 | ||
306 | pub(crate) fn has_impl_parent(&self) -> bool { | 285 | pub(crate) fn expect_record_field(&self) -> bool { |
307 | matches!(self.completion_location, Some(ImmediateLocation::Impl)) | 286 | matches!(self.completion_location, Some(ImmediateLocation::RecordField)) |
308 | } | ||
309 | |||
310 | pub(crate) fn has_field_list_parent(&self) -> bool { | ||
311 | matches!(self.completion_location, Some(ImmediateLocation::RecordFieldList)) | ||
312 | } | 287 | } |
313 | 288 | ||
314 | pub(crate) fn has_impl_or_trait_prev_sibling(&self) -> bool { | 289 | pub(crate) fn has_impl_or_trait_prev_sibling(&self) -> bool { |
315 | self.prev_sibling.is_some() | 290 | matches!( |
291 | self.prev_sibling, | ||
292 | Some(ImmediatePrevSibling::ImplDefType) | Some(ImmediatePrevSibling::TraitDefName) | ||
293 | ) | ||
316 | } | 294 | } |
317 | 295 | ||
318 | pub(crate) fn is_path_disallowed(&self) -> bool { | 296 | pub(crate) fn after_if(&self) -> bool { |
319 | self.record_lit_syntax.is_some() | 297 | matches!(self.prev_sibling, Some(ImmediatePrevSibling::IfExpr)) |
320 | || self.record_pat_syntax.is_some() | ||
321 | || self.attribute_under_caret.is_some() | ||
322 | || self.mod_declaration_under_caret.is_some() | ||
323 | || self.has_impl_or_trait_parent() | ||
324 | } | 298 | } |
325 | 299 | ||
326 | fn fill_keyword_patterns(&mut self, file_with_fake_ident: &SyntaxNode, offset: TextSize) { | 300 | pub(crate) fn is_path_disallowed(&self) -> bool { |
327 | let fake_ident_token = file_with_fake_ident.token_at_offset(offset).right_biased().unwrap(); | 301 | matches!( |
328 | let syntax_element = NodeOrToken::Token(fake_ident_token); | 302 | self.completion_location, |
329 | self.previous_token = previous_token(syntax_element.clone()); | 303 | Some(ImmediateLocation::Attribute(_)) |
330 | self.in_loop_body = is_in_loop_body(syntax_element.clone()); | 304 | | Some(ImmediateLocation::ModDeclaration(_)) |
331 | self.is_match_arm = is_match_arm(syntax_element.clone()); | 305 | | Some(ImmediateLocation::RecordPat(_)) |
332 | if has_prev_sibling(syntax_element.clone(), IMPL) { | 306 | | Some(ImmediateLocation::RecordExpr(_)) |
333 | self.prev_sibling = Some(PrevSibling::Impl) | 307 | ) || self.attribute_under_caret.is_some() |
334 | } else if has_prev_sibling(syntax_element.clone(), TRAIT) { | ||
335 | self.prev_sibling = Some(PrevSibling::Trait) | ||
336 | } | ||
337 | |||
338 | if has_block_expr_parent(syntax_element.clone()) { | ||
339 | self.completion_location = Some(ImmediateLocation::BlockExpr); | ||
340 | } else if has_bind_pat_parent(syntax_element.clone()) { | ||
341 | self.completion_location = Some(ImmediateLocation::IdentPat); | ||
342 | } else if has_ref_parent(syntax_element.clone()) { | ||
343 | self.completion_location = Some(ImmediateLocation::RefPatOrExpr); | ||
344 | } else if has_impl_parent(syntax_element.clone()) { | ||
345 | self.completion_location = Some(ImmediateLocation::Impl); | ||
346 | } else if has_field_list_parent(syntax_element.clone()) { | ||
347 | self.completion_location = Some(ImmediateLocation::RecordFieldList); | ||
348 | } else if has_trait_parent(syntax_element.clone()) { | ||
349 | self.completion_location = Some(ImmediateLocation::Trait); | ||
350 | } else if has_item_list_or_source_file_parent(syntax_element.clone()) { | ||
351 | self.completion_location = Some(ImmediateLocation::ItemList); | ||
352 | } | ||
353 | |||
354 | self.mod_declaration_under_caret = | ||
355 | find_node_at_offset::<ast::Module>(&file_with_fake_ident, offset) | ||
356 | .filter(|module| module.item_list().is_none()); | ||
357 | self.incomplete_let = | ||
358 | syntax_element.ancestors().take(6).find_map(ast::LetStmt::cast).map_or(false, |it| { | ||
359 | it.syntax().text_range().end() == syntax_element.text_range().end() | ||
360 | }); | ||
361 | |||
362 | let inside_impl_trait_block = inside_impl_trait_block(syntax_element.clone()); | ||
363 | let fn_is_prev = self.previous_token_is(T![fn]); | ||
364 | let for_is_prev2 = for_is_prev2(syntax_element.clone()); | ||
365 | self.no_completion_required = (fn_is_prev && !inside_impl_trait_block) || for_is_prev2; | ||
366 | } | 308 | } |
367 | 309 | ||
368 | fn fill_impl_def(&mut self) { | 310 | fn fill_impl_def(&mut self) { |
@@ -480,23 +422,43 @@ impl<'a> CompletionContext<'a> { | |||
480 | file_with_fake_ident: SyntaxNode, | 422 | file_with_fake_ident: SyntaxNode, |
481 | offset: TextSize, | 423 | offset: TextSize, |
482 | ) { | 424 | ) { |
425 | let fake_ident_token = file_with_fake_ident.token_at_offset(offset).right_biased().unwrap(); | ||
426 | let syntax_element = NodeOrToken::Token(fake_ident_token); | ||
427 | self.previous_token = previous_token(syntax_element.clone()); | ||
428 | self.attribute_under_caret = syntax_element.ancestors().find_map(ast::Attr::cast); | ||
429 | self.no_completion_required = { | ||
430 | let inside_impl_trait_block = inside_impl_trait_block(syntax_element.clone()); | ||
431 | let fn_is_prev = self.previous_token_is(T![fn]); | ||
432 | let for_is_prev2 = for_is_prev2(syntax_element.clone()); | ||
433 | (fn_is_prev && !inside_impl_trait_block) || for_is_prev2 | ||
434 | }; | ||
435 | self.in_loop_body = is_in_loop_body(syntax_element.clone()); | ||
436 | |||
437 | self.incomplete_let = | ||
438 | syntax_element.ancestors().take(6).find_map(ast::LetStmt::cast).map_or(false, |it| { | ||
439 | it.syntax().text_range().end() == syntax_element.text_range().end() | ||
440 | }); | ||
441 | |||
483 | let (expected_type, expected_name) = self.expected_type_and_name(); | 442 | let (expected_type, expected_name) = self.expected_type_and_name(); |
484 | self.expected_type = expected_type; | 443 | self.expected_type = expected_type; |
485 | self.expected_name = expected_name; | 444 | self.expected_name = expected_name; |
486 | self.attribute_under_caret = find_node_at_offset(&file_with_fake_ident, offset); | 445 | |
487 | let name_like = match find_node_at_offset(&&file_with_fake_ident, offset) { | 446 | let name_like = match find_node_at_offset(&&file_with_fake_ident, offset) { |
488 | Some(it) => it, | 447 | Some(it) => it, |
489 | None => return, | 448 | None => return, |
490 | }; | 449 | }; |
450 | self.completion_location = | ||
451 | determine_location(&self.sema, original_file, offset, &name_like); | ||
452 | self.prev_sibling = determine_prev_sibling(&name_like); | ||
491 | match name_like { | 453 | match name_like { |
492 | ast::NameLike::Lifetime(lifetime) => { | 454 | ast::NameLike::Lifetime(lifetime) => { |
493 | self.classify_lifetime(original_file, lifetime, offset); | 455 | self.classify_lifetime(original_file, lifetime, offset); |
494 | } | 456 | } |
495 | ast::NameLike::NameRef(name_ref) => { | 457 | ast::NameLike::NameRef(name_ref) => { |
496 | self.classify_name_ref(original_file, name_ref, offset); | 458 | self.classify_name_ref(original_file, name_ref); |
497 | } | 459 | } |
498 | ast::NameLike::Name(name) => { | 460 | ast::NameLike::Name(name) => { |
499 | self.classify_name(original_file, name, offset); | 461 | self.classify_name(name); |
500 | } | 462 | } |
501 | } | 463 | } |
502 | } | 464 | } |
@@ -530,7 +492,7 @@ impl<'a> CompletionContext<'a> { | |||
530 | } | 492 | } |
531 | } | 493 | } |
532 | 494 | ||
533 | fn classify_name(&mut self, original_file: &SyntaxNode, name: ast::Name, offset: TextSize) { | 495 | fn classify_name(&mut self, name: ast::Name) { |
534 | if let Some(bind_pat) = name.syntax().parent().and_then(ast::IdentPat::cast) { | 496 | if let Some(bind_pat) = name.syntax().parent().and_then(ast::IdentPat::cast) { |
535 | self.is_pat_or_const = Some(PatternRefutability::Refutable); | 497 | self.is_pat_or_const = Some(PatternRefutability::Refutable); |
536 | // if any of these is here our bind pat can't be a const pat anymore | 498 | // if any of these is here our bind pat can't be a const pat anymore |
@@ -568,28 +530,12 @@ impl<'a> CompletionContext<'a> { | |||
568 | 530 | ||
569 | self.fill_impl_def(); | 531 | self.fill_impl_def(); |
570 | } | 532 | } |
533 | |||
571 | self.is_param |= is_node::<ast::Param>(name.syntax()); | 534 | self.is_param |= is_node::<ast::Param>(name.syntax()); |
572 | if ast::RecordPatField::for_field_name(&name).is_some() { | ||
573 | self.record_pat_syntax = | ||
574 | self.sema.find_node_at_offset_with_macros(&original_file, offset); | ||
575 | } | ||
576 | } | 535 | } |
577 | 536 | ||
578 | fn classify_name_ref( | 537 | fn classify_name_ref(&mut self, original_file: &SyntaxNode, name_ref: ast::NameRef) { |
579 | &mut self, | ||
580 | original_file: &SyntaxNode, | ||
581 | name_ref: ast::NameRef, | ||
582 | offset: TextSize, | ||
583 | ) { | ||
584 | self.fill_impl_def(); | 538 | self.fill_impl_def(); |
585 | if ast::RecordExprField::for_field_name(&name_ref).is_some() { | ||
586 | self.record_lit_syntax = | ||
587 | self.sema.find_node_at_offset_with_macros(original_file, offset); | ||
588 | } | ||
589 | if ast::RecordPatField::for_field_name_ref(&name_ref).is_some() { | ||
590 | self.record_pat_syntax = | ||
591 | self.sema.find_node_at_offset_with_macros(&original_file, offset); | ||
592 | } | ||
593 | 539 | ||
594 | self.name_ref_syntax = | 540 | self.name_ref_syntax = |
595 | find_node_at_offset(original_file, name_ref.syntax().text_range().start()); | 541 | find_node_at_offset(original_file, name_ref.syntax().text_range().start()); |
@@ -676,17 +622,6 @@ impl<'a> CompletionContext<'a> { | |||
676 | }) | 622 | }) |
677 | .unwrap_or(false); | 623 | .unwrap_or(false); |
678 | self.is_expr = path.syntax().parent().and_then(ast::PathExpr::cast).is_some(); | 624 | self.is_expr = path.syntax().parent().and_then(ast::PathExpr::cast).is_some(); |
679 | |||
680 | if let Some(off) = name_ref.syntax().text_range().start().checked_sub(2.into()) { | ||
681 | if let Some(if_expr) = | ||
682 | self.sema.find_node_at_offset_with_macros::<ast::IfExpr>(original_file, off) | ||
683 | { | ||
684 | if if_expr.syntax().text_range().end() < name_ref.syntax().text_range().start() | ||
685 | { | ||
686 | self.after_if = true; | ||
687 | } | ||
688 | } | ||
689 | } | ||
690 | } | 625 | } |
691 | 626 | ||
692 | if let Some(field_expr) = ast::FieldExpr::cast(parent.clone()) { | 627 | if let Some(field_expr) = ast::FieldExpr::cast(parent.clone()) { |
diff --git a/crates/ide_completion/src/patterns.rs b/crates/ide_completion/src/patterns.rs index 04f2c532b..26516046b 100644 --- a/crates/ide_completion/src/patterns.rs +++ b/crates/ide_completion/src/patterns.rs | |||
@@ -1,38 +1,197 @@ | |||
1 | //! Patterns telling us certain facts about current syntax element, they are used in completion context | 1 | //! Patterns telling us certain facts about current syntax element, they are used in completion context |
2 | 2 | ||
3 | use hir::Semantics; | ||
4 | use ide_db::RootDatabase; | ||
3 | use syntax::{ | 5 | use syntax::{ |
4 | algo::non_trivia_sibling, | 6 | algo::non_trivia_sibling, |
5 | ast::{self, LoopBodyOwner}, | 7 | ast::{self, LoopBodyOwner}, |
6 | match_ast, AstNode, Direction, NodeOrToken, SyntaxElement, | 8 | match_ast, AstNode, Direction, SyntaxElement, |
7 | SyntaxKind::{self, *}, | 9 | SyntaxKind::*, |
8 | SyntaxNode, SyntaxToken, T, | 10 | SyntaxNode, SyntaxToken, TextSize, T, |
9 | }; | 11 | }; |
10 | 12 | ||
11 | #[cfg(test)] | 13 | #[cfg(test)] |
12 | use crate::test_utils::{check_pattern_is_applicable, check_pattern_is_not_applicable}; | 14 | use crate::test_utils::{check_pattern_is_applicable, check_pattern_is_not_applicable}; |
13 | 15 | ||
14 | pub(crate) fn has_trait_parent(element: SyntaxElement) -> bool { | 16 | /// Direct parent container of the cursor position |
15 | not_same_range_ancestor(element) | 17 | #[derive(Copy, Clone, Debug, PartialEq, Eq)] |
16 | .filter(|it| it.kind() == ASSOC_ITEM_LIST) | 18 | pub(crate) enum ImmediatePrevSibling { |
17 | .and_then(|it| it.parent()) | 19 | IfExpr, |
18 | .filter(|it| it.kind() == TRAIT) | 20 | TraitDefName, |
19 | .is_some() | 21 | ImplDefType, |
20 | } | 22 | } |
21 | #[test] | 23 | |
22 | fn test_has_trait_parent() { | 24 | /// Direct parent container of the cursor position |
23 | check_pattern_is_applicable(r"trait A { f$0 }", has_trait_parent); | 25 | #[derive(Clone, Debug, PartialEq, Eq)] |
26 | pub(crate) enum ImmediateLocation { | ||
27 | Use, | ||
28 | Impl, | ||
29 | Trait, | ||
30 | RecordField, | ||
31 | RefExpr, | ||
32 | IdentPat, | ||
33 | BlockExpr, | ||
34 | ItemList, | ||
35 | // Fake file ast node | ||
36 | Attribute(ast::Attr), | ||
37 | // Fake file ast node | ||
38 | ModDeclaration(ast::Module), | ||
39 | // Original file ast node | ||
40 | /// The record expr of the field name we are completing | ||
41 | RecordExpr(ast::RecordExpr), | ||
42 | // Original file ast node | ||
43 | /// The record pat of the field name we are completing | ||
44 | RecordPat(ast::RecordPat), | ||
24 | } | 45 | } |
25 | 46 | ||
26 | pub(crate) fn has_impl_parent(element: SyntaxElement) -> bool { | 47 | pub(crate) fn determine_prev_sibling(name_like: &ast::NameLike) -> Option<ImmediatePrevSibling> { |
27 | not_same_range_ancestor(element) | 48 | let node = match name_like { |
28 | .filter(|it| it.kind() == ASSOC_ITEM_LIST) | 49 | ast::NameLike::NameRef(name_ref) => maximize_name_ref(name_ref), |
29 | .and_then(|it| it.parent()) | 50 | ast::NameLike::Name(n) => n.syntax().clone(), |
30 | .filter(|it| it.kind() == IMPL) | 51 | ast::NameLike::Lifetime(lt) => lt.syntax().clone(), |
31 | .is_some() | 52 | }; |
53 | let node = match node.parent().and_then(ast::MacroCall::cast) { | ||
54 | // When a path is being typed after the name of a trait/type of an impl it is being | ||
55 | // parsed as a macro, so when the trait/impl has a block following it an we are between the | ||
56 | // name and block the macro will attach the block to itself so maximizing fails to take | ||
57 | // that into account | ||
58 | // FIXME path expr and statement have a similar problem with attrs | ||
59 | Some(call) | ||
60 | if call.excl_token().is_none() | ||
61 | && call.token_tree().map_or(false, |t| t.l_curly_token().is_some()) | ||
62 | && call.semicolon_token().is_none() => | ||
63 | { | ||
64 | call.syntax().clone() | ||
65 | } | ||
66 | _ => node, | ||
67 | }; | ||
68 | let prev_sibling = non_trivia_sibling(node.into(), Direction::Prev)?.into_node()?; | ||
69 | let res = match_ast! { | ||
70 | match prev_sibling { | ||
71 | ast::ExprStmt(it) => { | ||
72 | let node = it.expr().filter(|_| it.semicolon_token().is_none())?.syntax().clone(); | ||
73 | match_ast! { | ||
74 | match node { | ||
75 | ast::IfExpr(_it) => ImmediatePrevSibling::IfExpr, | ||
76 | _ => return None, | ||
77 | } | ||
78 | } | ||
79 | }, | ||
80 | ast::Trait(it) => if it.assoc_item_list().is_none() { | ||
81 | ImmediatePrevSibling::TraitDefName | ||
82 | } else { | ||
83 | return None | ||
84 | }, | ||
85 | ast::Impl(it) => if it.assoc_item_list().is_none() | ||
86 | && (it.for_token().is_none() || it.self_ty().is_some()) { | ||
87 | ImmediatePrevSibling::ImplDefType | ||
88 | } else { | ||
89 | return None | ||
90 | }, | ||
91 | _ => return None, | ||
92 | } | ||
93 | }; | ||
94 | Some(res) | ||
32 | } | 95 | } |
33 | #[test] | 96 | |
34 | fn test_has_impl_parent() { | 97 | pub(crate) fn determine_location( |
35 | check_pattern_is_applicable(r"impl A { f$0 }", has_impl_parent); | 98 | sema: &Semantics<RootDatabase>, |
99 | original_file: &SyntaxNode, | ||
100 | offset: TextSize, | ||
101 | name_like: &ast::NameLike, | ||
102 | ) -> Option<ImmediateLocation> { | ||
103 | let node = match name_like { | ||
104 | ast::NameLike::NameRef(name_ref) => { | ||
105 | if ast::RecordExprField::for_field_name(&name_ref).is_some() { | ||
106 | return sema | ||
107 | .find_node_at_offset_with_macros(original_file, offset) | ||
108 | .map(ImmediateLocation::RecordExpr); | ||
109 | } | ||
110 | if ast::RecordPatField::for_field_name_ref(&name_ref).is_some() { | ||
111 | return sema | ||
112 | .find_node_at_offset_with_macros(original_file, offset) | ||
113 | .map(ImmediateLocation::RecordPat); | ||
114 | } | ||
115 | maximize_name_ref(name_ref) | ||
116 | } | ||
117 | ast::NameLike::Name(name) => { | ||
118 | if ast::RecordPatField::for_field_name(&name).is_some() { | ||
119 | return sema | ||
120 | .find_node_at_offset_with_macros(original_file, offset) | ||
121 | .map(ImmediateLocation::RecordPat); | ||
122 | } | ||
123 | name.syntax().clone() | ||
124 | } | ||
125 | ast::NameLike::Lifetime(lt) => lt.syntax().clone(), | ||
126 | }; | ||
127 | |||
128 | let parent = match node.parent() { | ||
129 | Some(parent) => match ast::MacroCall::cast(parent.clone()) { | ||
130 | // When a path is being typed in an (Assoc)ItemList the parser will always emit a macro_call. | ||
131 | // This is usually fine as the node expansion code above already accounts for that with | ||
132 | // the ancestors call, but there is one exception to this which is that when an attribute | ||
133 | // precedes it the code above will not walk the Path to the parent MacroCall as their ranges differ. | ||
134 | // FIXME path expr and statement have a similar problem | ||
135 | Some(call) | ||
136 | if call.excl_token().is_none() | ||
137 | && call.token_tree().is_none() | ||
138 | && call.semicolon_token().is_none() => | ||
139 | { | ||
140 | call.syntax().parent()? | ||
141 | } | ||
142 | _ => parent, | ||
143 | }, | ||
144 | // SourceFile | ||
145 | None => { | ||
146 | return match node.kind() { | ||
147 | MACRO_ITEMS | SOURCE_FILE => Some(ImmediateLocation::ItemList), | ||
148 | _ => None, | ||
149 | } | ||
150 | } | ||
151 | }; | ||
152 | |||
153 | let res = match_ast! { | ||
154 | match parent { | ||
155 | ast::IdentPat(_it) => ImmediateLocation::IdentPat, | ||
156 | ast::Use(_it) => ImmediateLocation::Use, | ||
157 | ast::BlockExpr(_it) => ImmediateLocation::BlockExpr, | ||
158 | ast::SourceFile(_it) => ImmediateLocation::ItemList, | ||
159 | ast::ItemList(_it) => ImmediateLocation::ItemList, | ||
160 | ast::RefExpr(_it) => ImmediateLocation::RefExpr, | ||
161 | ast::RecordField(_it) => ImmediateLocation::RecordField, | ||
162 | ast::AssocItemList(it) => match it.syntax().parent().map(|it| it.kind()) { | ||
163 | Some(IMPL) => ImmediateLocation::Impl, | ||
164 | Some(TRAIT) => ImmediateLocation::Trait, | ||
165 | _ => return None, | ||
166 | }, | ||
167 | ast::Module(it) => if it.item_list().is_none() { | ||
168 | ImmediateLocation::ModDeclaration(it) | ||
169 | } else { | ||
170 | return None | ||
171 | }, | ||
172 | ast::Attr(it) => ImmediateLocation::Attribute(it), | ||
173 | _ => return None, | ||
174 | } | ||
175 | }; | ||
176 | Some(res) | ||
177 | } | ||
178 | |||
179 | fn maximize_name_ref(name_ref: &ast::NameRef) -> SyntaxNode { | ||
180 | // Maximize a nameref to its enclosing path if its the last segment of said path | ||
181 | if let Some(segment) = name_ref.syntax().parent().and_then(ast::PathSegment::cast) { | ||
182 | let p = segment.parent_path(); | ||
183 | if p.parent_path().is_none() { | ||
184 | if let Some(it) = p | ||
185 | .syntax() | ||
186 | .ancestors() | ||
187 | .take_while(|it| it.text_range() == p.syntax().text_range()) | ||
188 | .last() | ||
189 | { | ||
190 | return it; | ||
191 | } | ||
192 | } | ||
193 | } | ||
194 | name_ref.syntax().clone() | ||
36 | } | 195 | } |
37 | 196 | ||
38 | pub(crate) fn inside_impl_trait_block(element: SyntaxElement) -> bool { | 197 | pub(crate) fn inside_impl_trait_block(element: SyntaxElement) -> bool { |
@@ -53,68 +212,6 @@ fn test_inside_impl_trait_block() { | |||
53 | check_pattern_is_not_applicable(r"impl A { fn f$0 }", inside_impl_trait_block); | 212 | check_pattern_is_not_applicable(r"impl A { fn f$0 }", inside_impl_trait_block); |
54 | } | 213 | } |
55 | 214 | ||
56 | pub(crate) fn has_field_list_parent(element: SyntaxElement) -> bool { | ||
57 | not_same_range_ancestor(element).filter(|it| it.kind() == RECORD_FIELD_LIST).is_some() | ||
58 | } | ||
59 | #[test] | ||
60 | fn test_has_field_list_parent() { | ||
61 | check_pattern_is_applicable(r"struct Foo { f$0 }", has_field_list_parent); | ||
62 | check_pattern_is_applicable(r"struct Foo { f$0 pub f: i32}", has_field_list_parent); | ||
63 | } | ||
64 | |||
65 | pub(crate) fn has_block_expr_parent(element: SyntaxElement) -> bool { | ||
66 | not_same_range_ancestor(element).filter(|it| it.kind() == BLOCK_EXPR).is_some() | ||
67 | } | ||
68 | #[test] | ||
69 | fn test_has_block_expr_parent() { | ||
70 | check_pattern_is_applicable(r"fn my_fn() { let a = 2; f$0 }", has_block_expr_parent); | ||
71 | } | ||
72 | |||
73 | pub(crate) fn has_bind_pat_parent(element: SyntaxElement) -> bool { | ||
74 | element.ancestors().any(|it| it.kind() == IDENT_PAT) | ||
75 | } | ||
76 | |||
77 | #[test] | ||
78 | fn test_has_bind_pat_parent() { | ||
79 | check_pattern_is_applicable(r"fn my_fn(m$0) {}", has_bind_pat_parent); | ||
80 | check_pattern_is_applicable(r"fn my_fn() { let m$0 }", has_bind_pat_parent); | ||
81 | } | ||
82 | |||
83 | pub(crate) fn has_ref_parent(element: SyntaxElement) -> bool { | ||
84 | not_same_range_ancestor(element) | ||
85 | .filter(|it| it.kind() == REF_PAT || it.kind() == REF_EXPR) | ||
86 | .is_some() | ||
87 | } | ||
88 | #[test] | ||
89 | fn test_has_ref_parent() { | ||
90 | check_pattern_is_applicable(r"fn my_fn(&m$0) {}", has_ref_parent); | ||
91 | check_pattern_is_applicable(r"fn my() { let &m$0 }", has_ref_parent); | ||
92 | } | ||
93 | |||
94 | pub(crate) fn has_item_list_or_source_file_parent(element: SyntaxElement) -> bool { | ||
95 | match not_same_range_ancestor(element) { | ||
96 | Some(it) => it.kind() == SOURCE_FILE || it.kind() == ITEM_LIST, | ||
97 | None => true, | ||
98 | } | ||
99 | } | ||
100 | #[test] | ||
101 | fn test_has_item_list_or_source_file_parent() { | ||
102 | check_pattern_is_applicable(r"i$0", has_item_list_or_source_file_parent); | ||
103 | check_pattern_is_applicable(r"mod foo { f$0 }", has_item_list_or_source_file_parent); | ||
104 | } | ||
105 | |||
106 | pub(crate) fn is_match_arm(element: SyntaxElement) -> bool { | ||
107 | not_same_range_ancestor(element.clone()).filter(|it| it.kind() == MATCH_ARM).is_some() | ||
108 | && previous_sibling_or_ancestor_sibling(element) | ||
109 | .and_then(|it| it.into_token()) | ||
110 | .filter(|it| it.kind() == FAT_ARROW) | ||
111 | .is_some() | ||
112 | } | ||
113 | #[test] | ||
114 | fn test_is_match_arm() { | ||
115 | check_pattern_is_applicable(r"fn my_fn() { match () { () => m$0 } }", is_match_arm); | ||
116 | } | ||
117 | |||
118 | pub(crate) fn previous_token(element: SyntaxElement) -> Option<SyntaxToken> { | 215 | pub(crate) fn previous_token(element: SyntaxElement) -> Option<SyntaxToken> { |
119 | element.into_token().and_then(|it| previous_non_trivia_token(it)) | 216 | element.into_token().and_then(|it| previous_non_trivia_token(it)) |
120 | } | 217 | } |
@@ -134,14 +231,6 @@ fn test_for_is_prev2() { | |||
134 | check_pattern_is_applicable(r"for i i$0", for_is_prev2); | 231 | check_pattern_is_applicable(r"for i i$0", for_is_prev2); |
135 | } | 232 | } |
136 | 233 | ||
137 | pub(crate) fn has_prev_sibling(element: SyntaxElement, kind: SyntaxKind) -> bool { | ||
138 | previous_sibling_or_ancestor_sibling(element).filter(|it| it.kind() == kind).is_some() | ||
139 | } | ||
140 | #[test] | ||
141 | fn test_has_impl_as_prev_sibling() { | ||
142 | check_pattern_is_applicable(r"impl A w$0 {}", |it| has_prev_sibling(it, IMPL)); | ||
143 | } | ||
144 | |||
145 | pub(crate) fn is_in_loop_body(element: SyntaxElement) -> bool { | 234 | pub(crate) fn is_in_loop_body(element: SyntaxElement) -> bool { |
146 | element | 235 | element |
147 | .ancestors() | 236 | .ancestors() |
@@ -160,14 +249,6 @@ pub(crate) fn is_in_loop_body(element: SyntaxElement) -> bool { | |||
160 | .is_some() | 249 | .is_some() |
161 | } | 250 | } |
162 | 251 | ||
163 | fn not_same_range_ancestor(element: SyntaxElement) -> Option<SyntaxNode> { | ||
164 | element | ||
165 | .ancestors() | ||
166 | .take_while(|it| it.text_range() == element.text_range()) | ||
167 | .last() | ||
168 | .and_then(|it| it.parent()) | ||
169 | } | ||
170 | |||
171 | fn previous_non_trivia_token(token: SyntaxToken) -> Option<SyntaxToken> { | 252 | fn previous_non_trivia_token(token: SyntaxToken) -> Option<SyntaxToken> { |
172 | let mut token = token.prev_token(); | 253 | let mut token = token.prev_token(); |
173 | while let Some(inner) = token.clone() { | 254 | while let Some(inner) = token.clone() { |
@@ -180,17 +261,119 @@ fn previous_non_trivia_token(token: SyntaxToken) -> Option<SyntaxToken> { | |||
180 | None | 261 | None |
181 | } | 262 | } |
182 | 263 | ||
183 | fn previous_sibling_or_ancestor_sibling(element: SyntaxElement) -> Option<SyntaxElement> { | 264 | #[cfg(test)] |
184 | let token_sibling = non_trivia_sibling(element.clone(), Direction::Prev); | 265 | mod tests { |
185 | if let Some(sibling) = token_sibling { | 266 | use syntax::algo::find_node_at_offset; |
186 | Some(sibling) | 267 | |
187 | } else { | 268 | use crate::test_utils::position; |
188 | // if not trying to find first ancestor which has such a sibling | 269 | |
189 | let range = element.text_range(); | 270 | use super::*; |
190 | let top_node = element.ancestors().take_while(|it| it.text_range() == range).last()?; | 271 | |
191 | let prev_sibling_node = top_node.ancestors().find(|it| { | 272 | fn check_location(code: &str, loc: impl Into<Option<ImmediateLocation>>) { |
192 | non_trivia_sibling(NodeOrToken::Node(it.to_owned()), Direction::Prev).is_some() | 273 | let (db, pos) = position(code); |
193 | })?; | 274 | |
194 | non_trivia_sibling(NodeOrToken::Node(prev_sibling_node), Direction::Prev) | 275 | let sema = Semantics::new(&db); |
276 | let original_file = sema.parse(pos.file_id); | ||
277 | |||
278 | let name_like = find_node_at_offset(original_file.syntax(), pos.offset).unwrap(); | ||
279 | assert_eq!( | ||
280 | determine_location(&sema, original_file.syntax(), pos.offset, &name_like), | ||
281 | loc.into() | ||
282 | ); | ||
283 | } | ||
284 | |||
285 | fn check_prev_sibling(code: &str, sibling: impl Into<Option<ImmediatePrevSibling>>) { | ||
286 | check_pattern_is_applicable(code, |e| { | ||
287 | let name = &e.parent().and_then(ast::NameLike::cast).expect("Expected a namelike"); | ||
288 | assert_eq!(determine_prev_sibling(name), sibling.into()); | ||
289 | true | ||
290 | }); | ||
291 | } | ||
292 | |||
293 | #[test] | ||
294 | fn test_trait_loc() { | ||
295 | check_location(r"trait A { f$0 }", ImmediateLocation::Trait); | ||
296 | check_location(r"trait A { #[attr] f$0 }", ImmediateLocation::Trait); | ||
297 | check_location(r"trait A { f$0 fn f() {} }", ImmediateLocation::Trait); | ||
298 | check_location(r"trait A { fn f() {} f$0 }", ImmediateLocation::Trait); | ||
299 | check_location(r"trait A$0 {}", None); | ||
300 | check_location(r"trait A { fn f$0 }", None); | ||
301 | } | ||
302 | |||
303 | #[test] | ||
304 | fn test_impl_loc() { | ||
305 | check_location(r"impl A { f$0 }", ImmediateLocation::Impl); | ||
306 | check_location(r"impl A { #[attr] f$0 }", ImmediateLocation::Impl); | ||
307 | check_location(r"impl A { f$0 fn f() {} }", ImmediateLocation::Impl); | ||
308 | check_location(r"impl A { fn f() {} f$0 }", ImmediateLocation::Impl); | ||
309 | check_location(r"impl A$0 {}", None); | ||
310 | check_location(r"impl A { fn f$0 }", None); | ||
311 | } | ||
312 | |||
313 | #[test] | ||
314 | fn test_use_loc() { | ||
315 | check_location(r"use f$0", ImmediateLocation::Use); | ||
316 | check_location(r"use f$0;", ImmediateLocation::Use); | ||
317 | check_location(r"use f::{f$0}", None); | ||
318 | check_location(r"use {f$0}", None); | ||
319 | } | ||
320 | |||
321 | #[test] | ||
322 | fn test_record_field_loc() { | ||
323 | check_location(r"struct Foo { f$0 }", ImmediateLocation::RecordField); | ||
324 | check_location(r"struct Foo { f$0 pub f: i32}", ImmediateLocation::RecordField); | ||
325 | check_location(r"struct Foo { pub f: i32, f$0 }", ImmediateLocation::RecordField); | ||
326 | } | ||
327 | |||
328 | #[test] | ||
329 | fn test_block_expr_loc() { | ||
330 | check_location(r"fn my_fn() { let a = 2; f$0 }", ImmediateLocation::BlockExpr); | ||
331 | check_location(r"fn my_fn() { f$0 f }", ImmediateLocation::BlockExpr); | ||
332 | } | ||
333 | |||
334 | #[test] | ||
335 | fn test_ident_pat_loc() { | ||
336 | check_location(r"fn my_fn(m$0) {}", ImmediateLocation::IdentPat); | ||
337 | check_location(r"fn my_fn() { let m$0 }", ImmediateLocation::IdentPat); | ||
338 | check_location(r"fn my_fn(&m$0) {}", ImmediateLocation::IdentPat); | ||
339 | check_location(r"fn my_fn() { let &m$0 }", ImmediateLocation::IdentPat); | ||
340 | } | ||
341 | |||
342 | #[test] | ||
343 | fn test_ref_expr_loc() { | ||
344 | check_location(r"fn my_fn() { let x = &m$0 foo; }", ImmediateLocation::RefExpr); | ||
345 | } | ||
346 | |||
347 | #[test] | ||
348 | fn test_item_list_loc() { | ||
349 | check_location(r"i$0", ImmediateLocation::ItemList); | ||
350 | check_location(r"#[attr] i$0", ImmediateLocation::ItemList); | ||
351 | check_location(r"fn f() {} i$0", ImmediateLocation::ItemList); | ||
352 | check_location(r"mod foo { f$0 }", ImmediateLocation::ItemList); | ||
353 | check_location(r"mod foo { #[attr] f$0 }", ImmediateLocation::ItemList); | ||
354 | check_location(r"mod foo { fn f() {} f$0 }", ImmediateLocation::ItemList); | ||
355 | check_location(r"mod foo$0 {}", None); | ||
356 | } | ||
357 | |||
358 | #[test] | ||
359 | fn test_impl_prev_sibling() { | ||
360 | check_prev_sibling(r"impl A w$0 ", ImmediatePrevSibling::ImplDefType); | ||
361 | check_prev_sibling(r"impl A w$0 {}", ImmediatePrevSibling::ImplDefType); | ||
362 | check_prev_sibling(r"impl A for A w$0 ", ImmediatePrevSibling::ImplDefType); | ||
363 | check_prev_sibling(r"impl A for A w$0 {}", ImmediatePrevSibling::ImplDefType); | ||
364 | check_prev_sibling(r"impl A for w$0 {}", None); | ||
365 | check_prev_sibling(r"impl A for w$0", None); | ||
366 | } | ||
367 | |||
368 | #[test] | ||
369 | fn test_trait_prev_sibling() { | ||
370 | check_prev_sibling(r"trait A w$0 ", ImmediatePrevSibling::TraitDefName); | ||
371 | check_prev_sibling(r"trait A w$0 {}", ImmediatePrevSibling::TraitDefName); | ||
372 | } | ||
373 | |||
374 | #[test] | ||
375 | fn test_if_expr_prev_sibling() { | ||
376 | check_prev_sibling(r"fn foo() { if true {} w$0", ImmediatePrevSibling::IfExpr); | ||
377 | check_prev_sibling(r"fn foo() { if true {}; w$0", None); | ||
195 | } | 378 | } |
196 | } | 379 | } |
diff --git a/crates/ide_completion/src/render.rs b/crates/ide_completion/src/render.rs index 91300c56e..425dd0247 100644 --- a/crates/ide_completion/src/render.rs +++ b/crates/ide_completion/src/render.rs | |||
@@ -10,9 +10,7 @@ pub(crate) mod type_alias; | |||
10 | 10 | ||
11 | mod builder_ext; | 11 | mod builder_ext; |
12 | 12 | ||
13 | use hir::{ | 13 | use hir::{AsAssocItem, HasAttrs, HirDisplay}; |
14 | AsAssocItem, Documentation, HasAttrs, HirDisplay, ModuleDef, Mutability, ScopeDef, Type, | ||
15 | }; | ||
16 | use ide_db::{ | 14 | use ide_db::{ |
17 | helpers::{item_name, SnippetCap}, | 15 | helpers::{item_name, SnippetCap}, |
18 | RootDatabase, SymbolKind, | 16 | RootDatabase, SymbolKind, |
@@ -21,31 +19,30 @@ use syntax::TextRange; | |||
21 | 19 | ||
22 | use crate::{ | 20 | use crate::{ |
23 | item::{CompletionRelevanceTypeMatch, ImportEdit}, | 21 | item::{CompletionRelevanceTypeMatch, ImportEdit}, |
22 | render::{enum_variant::render_variant, function::render_fn, macro_::render_macro}, | ||
24 | CompletionContext, CompletionItem, CompletionItemKind, CompletionKind, CompletionRelevance, | 23 | CompletionContext, CompletionItem, CompletionItemKind, CompletionKind, CompletionRelevance, |
25 | }; | 24 | }; |
26 | 25 | ||
27 | use crate::render::{enum_variant::render_variant, function::render_fn, macro_::render_macro}; | ||
28 | |||
29 | pub(crate) fn render_field<'a>( | 26 | pub(crate) fn render_field<'a>( |
30 | ctx: RenderContext<'a>, | 27 | ctx: RenderContext<'a>, |
31 | field: hir::Field, | 28 | field: hir::Field, |
32 | ty: &Type, | 29 | ty: &hir::Type, |
33 | ) -> CompletionItem { | 30 | ) -> CompletionItem { |
34 | Render::new(ctx).add_field(field, ty) | 31 | Render::new(ctx).render_field(field, ty) |
35 | } | 32 | } |
36 | 33 | ||
37 | pub(crate) fn render_tuple_field<'a>( | 34 | pub(crate) fn render_tuple_field<'a>( |
38 | ctx: RenderContext<'a>, | 35 | ctx: RenderContext<'a>, |
39 | field: usize, | 36 | field: usize, |
40 | ty: &Type, | 37 | ty: &hir::Type, |
41 | ) -> CompletionItem { | 38 | ) -> CompletionItem { |
42 | Render::new(ctx).add_tuple_field(field, ty) | 39 | Render::new(ctx).render_tuple_field(field, ty) |
43 | } | 40 | } |
44 | 41 | ||
45 | pub(crate) fn render_resolution<'a>( | 42 | pub(crate) fn render_resolution<'a>( |
46 | ctx: RenderContext<'a>, | 43 | ctx: RenderContext<'a>, |
47 | local_name: String, | 44 | local_name: hir::Name, |
48 | resolution: &ScopeDef, | 45 | resolution: &hir::ScopeDef, |
49 | ) -> Option<CompletionItem> { | 46 | ) -> Option<CompletionItem> { |
50 | Render::new(ctx).render_resolution(local_name, None, resolution) | 47 | Render::new(ctx).render_resolution(local_name, None, resolution) |
51 | } | 48 | } |
@@ -54,12 +51,12 @@ pub(crate) fn render_resolution_with_import<'a>( | |||
54 | ctx: RenderContext<'a>, | 51 | ctx: RenderContext<'a>, |
55 | import_edit: ImportEdit, | 52 | import_edit: ImportEdit, |
56 | ) -> Option<CompletionItem> { | 53 | ) -> Option<CompletionItem> { |
57 | let resolution = ScopeDef::from(import_edit.import.original_item); | 54 | let resolution = hir::ScopeDef::from(import_edit.import.original_item); |
58 | let local_name = match resolution { | 55 | let local_name = match resolution { |
59 | ScopeDef::ModuleDef(ModuleDef::Function(f)) => f.name(ctx.completion.db).to_string(), | 56 | hir::ScopeDef::ModuleDef(hir::ModuleDef::Function(f)) => f.name(ctx.completion.db), |
60 | ScopeDef::ModuleDef(ModuleDef::Const(c)) => c.name(ctx.completion.db)?.to_string(), | 57 | hir::ScopeDef::ModuleDef(hir::ModuleDef::Const(c)) => c.name(ctx.completion.db)?, |
61 | ScopeDef::ModuleDef(ModuleDef::TypeAlias(t)) => t.name(ctx.completion.db).to_string(), | 58 | hir::ScopeDef::ModuleDef(hir::ModuleDef::TypeAlias(t)) => t.name(ctx.completion.db), |
62 | _ => item_name(ctx.db(), import_edit.import.original_item)?.to_string(), | 59 | _ => item_name(ctx.db(), import_edit.import.original_item)?, |
63 | }; | 60 | }; |
64 | Render::new(ctx).render_resolution(local_name, Some(import_edit), &resolution).map( | 61 | Render::new(ctx).render_resolution(local_name, Some(import_edit), &resolution).map( |
65 | |mut item| { | 62 | |mut item| { |
@@ -113,7 +110,7 @@ impl<'a> RenderContext<'a> { | |||
113 | || assoc.containing_trait(db).map(|trait_| self.is_deprecated(trait_)).unwrap_or(false) | 110 | || assoc.containing_trait(db).map(|trait_| self.is_deprecated(trait_)).unwrap_or(false) |
114 | } | 111 | } |
115 | 112 | ||
116 | fn docs(&self, node: impl HasAttrs) -> Option<Documentation> { | 113 | fn docs(&self, node: impl HasAttrs) -> Option<hir::Documentation> { |
117 | node.docs(self.db()) | 114 | node.docs(self.db()) |
118 | } | 115 | } |
119 | } | 116 | } |
@@ -129,14 +126,11 @@ impl<'a> Render<'a> { | |||
129 | Render { ctx } | 126 | Render { ctx } |
130 | } | 127 | } |
131 | 128 | ||
132 | fn add_field(&mut self, field: hir::Field, ty: &Type) -> CompletionItem { | 129 | fn render_field(&self, field: hir::Field, ty: &hir::Type) -> CompletionItem { |
133 | let is_deprecated = self.ctx.is_deprecated(field); | 130 | let is_deprecated = self.ctx.is_deprecated(field); |
134 | let name = field.name(self.ctx.db()); | 131 | let name = field.name(self.ctx.db()).to_string(); |
135 | let mut item = CompletionItem::new( | 132 | let mut item = |
136 | CompletionKind::Reference, | 133 | CompletionItem::new(CompletionKind::Reference, self.ctx.source_range(), name.clone()); |
137 | self.ctx.source_range(), | ||
138 | name.to_string(), | ||
139 | ); | ||
140 | item.kind(SymbolKind::Field) | 134 | item.kind(SymbolKind::Field) |
141 | .detail(ty.display(self.ctx.db()).to_string()) | 135 | .detail(ty.display(self.ctx.db()).to_string()) |
142 | .set_documentation(field.docs(self.ctx.db())) | 136 | .set_documentation(field.docs(self.ctx.db())) |
@@ -144,7 +138,7 @@ impl<'a> Render<'a> { | |||
144 | 138 | ||
145 | item.set_relevance(CompletionRelevance { | 139 | item.set_relevance(CompletionRelevance { |
146 | type_match: compute_type_match(self.ctx.completion, ty), | 140 | type_match: compute_type_match(self.ctx.completion, ty), |
147 | exact_name_match: compute_exact_name_match(self.ctx.completion, name.to_string()), | 141 | exact_name_match: compute_exact_name_match(self.ctx.completion, &name), |
148 | ..CompletionRelevance::default() | 142 | ..CompletionRelevance::default() |
149 | }); | 143 | }); |
150 | 144 | ||
@@ -157,7 +151,7 @@ impl<'a> Render<'a> { | |||
157 | item.build() | 151 | item.build() |
158 | } | 152 | } |
159 | 153 | ||
160 | fn add_tuple_field(&mut self, field: usize, ty: &Type) -> CompletionItem { | 154 | fn render_tuple_field(&self, field: usize, ty: &hir::Type) -> CompletionItem { |
161 | let mut item = CompletionItem::new( | 155 | let mut item = CompletionItem::new( |
162 | CompletionKind::Reference, | 156 | CompletionKind::Reference, |
163 | self.ctx.source_range(), | 157 | self.ctx.source_range(), |
@@ -171,71 +165,82 @@ impl<'a> Render<'a> { | |||
171 | 165 | ||
172 | fn render_resolution( | 166 | fn render_resolution( |
173 | self, | 167 | self, |
174 | local_name: String, | 168 | local_name: hir::Name, |
175 | import_to_add: Option<ImportEdit>, | 169 | import_to_add: Option<ImportEdit>, |
176 | resolution: &ScopeDef, | 170 | resolution: &hir::ScopeDef, |
177 | ) -> Option<CompletionItem> { | 171 | ) -> Option<CompletionItem> { |
178 | let _p = profile::span("render_resolution"); | 172 | let _p = profile::span("render_resolution"); |
179 | use hir::ModuleDef::*; | 173 | use hir::ModuleDef::*; |
180 | 174 | ||
181 | let completion_kind = match resolution { | 175 | let completion_kind = match resolution { |
182 | ScopeDef::ModuleDef(BuiltinType(..)) => CompletionKind::BuiltinType, | 176 | hir::ScopeDef::ModuleDef(BuiltinType(..)) => CompletionKind::BuiltinType, |
183 | _ => CompletionKind::Reference, | 177 | _ => CompletionKind::Reference, |
184 | }; | 178 | }; |
185 | 179 | ||
186 | let kind = match resolution { | 180 | let kind = match resolution { |
187 | ScopeDef::ModuleDef(Function(func)) => { | 181 | hir::ScopeDef::ModuleDef(Function(func)) => { |
188 | return render_fn(self.ctx, import_to_add, Some(local_name), *func); | 182 | return render_fn(self.ctx, import_to_add, Some(local_name), *func); |
189 | } | 183 | } |
190 | ScopeDef::ModuleDef(Variant(_)) if self.ctx.completion.is_pat_or_const.is_some() => { | 184 | hir::ScopeDef::ModuleDef(Variant(_)) |
185 | if self.ctx.completion.is_pat_or_const.is_some() => | ||
186 | { | ||
191 | CompletionItemKind::SymbolKind(SymbolKind::Variant) | 187 | CompletionItemKind::SymbolKind(SymbolKind::Variant) |
192 | } | 188 | } |
193 | ScopeDef::ModuleDef(Variant(var)) => { | 189 | hir::ScopeDef::ModuleDef(Variant(var)) => { |
194 | let item = render_variant(self.ctx, import_to_add, Some(local_name), *var, None); | 190 | let item = render_variant(self.ctx, import_to_add, Some(local_name), *var, None); |
195 | return Some(item); | 191 | return Some(item); |
196 | } | 192 | } |
197 | ScopeDef::MacroDef(mac) => { | 193 | hir::ScopeDef::MacroDef(mac) => { |
198 | let item = render_macro(self.ctx, import_to_add, local_name, *mac); | 194 | let item = render_macro(self.ctx, import_to_add, local_name, *mac); |
199 | return item; | 195 | return item; |
200 | } | 196 | } |
201 | 197 | ||
202 | ScopeDef::ModuleDef(Module(..)) => CompletionItemKind::SymbolKind(SymbolKind::Module), | 198 | hir::ScopeDef::ModuleDef(Module(..)) => { |
203 | ScopeDef::ModuleDef(Adt(adt)) => CompletionItemKind::SymbolKind(match adt { | 199 | CompletionItemKind::SymbolKind(SymbolKind::Module) |
200 | } | ||
201 | hir::ScopeDef::ModuleDef(Adt(adt)) => CompletionItemKind::SymbolKind(match adt { | ||
204 | hir::Adt::Struct(_) => SymbolKind::Struct, | 202 | hir::Adt::Struct(_) => SymbolKind::Struct, |
205 | hir::Adt::Union(_) => SymbolKind::Union, | 203 | hir::Adt::Union(_) => SymbolKind::Union, |
206 | hir::Adt::Enum(_) => SymbolKind::Enum, | 204 | hir::Adt::Enum(_) => SymbolKind::Enum, |
207 | }), | 205 | }), |
208 | ScopeDef::ModuleDef(Const(..)) => CompletionItemKind::SymbolKind(SymbolKind::Const), | 206 | hir::ScopeDef::ModuleDef(Const(..)) => { |
209 | ScopeDef::ModuleDef(Static(..)) => CompletionItemKind::SymbolKind(SymbolKind::Static), | 207 | CompletionItemKind::SymbolKind(SymbolKind::Const) |
210 | ScopeDef::ModuleDef(Trait(..)) => CompletionItemKind::SymbolKind(SymbolKind::Trait), | 208 | } |
211 | ScopeDef::ModuleDef(TypeAlias(..)) => { | 209 | hir::ScopeDef::ModuleDef(Static(..)) => { |
210 | CompletionItemKind::SymbolKind(SymbolKind::Static) | ||
211 | } | ||
212 | hir::ScopeDef::ModuleDef(Trait(..)) => { | ||
213 | CompletionItemKind::SymbolKind(SymbolKind::Trait) | ||
214 | } | ||
215 | hir::ScopeDef::ModuleDef(TypeAlias(..)) => { | ||
212 | CompletionItemKind::SymbolKind(SymbolKind::TypeAlias) | 216 | CompletionItemKind::SymbolKind(SymbolKind::TypeAlias) |
213 | } | 217 | } |
214 | ScopeDef::ModuleDef(BuiltinType(..)) => CompletionItemKind::BuiltinType, | 218 | hir::ScopeDef::ModuleDef(BuiltinType(..)) => CompletionItemKind::BuiltinType, |
215 | ScopeDef::GenericParam(param) => CompletionItemKind::SymbolKind(match param { | 219 | hir::ScopeDef::GenericParam(param) => CompletionItemKind::SymbolKind(match param { |
216 | hir::GenericParam::TypeParam(_) => SymbolKind::TypeParam, | 220 | hir::GenericParam::TypeParam(_) => SymbolKind::TypeParam, |
217 | hir::GenericParam::LifetimeParam(_) => SymbolKind::LifetimeParam, | 221 | hir::GenericParam::LifetimeParam(_) => SymbolKind::LifetimeParam, |
218 | hir::GenericParam::ConstParam(_) => SymbolKind::ConstParam, | 222 | hir::GenericParam::ConstParam(_) => SymbolKind::ConstParam, |
219 | }), | 223 | }), |
220 | ScopeDef::Local(..) => CompletionItemKind::SymbolKind(SymbolKind::Local), | 224 | hir::ScopeDef::Local(..) => CompletionItemKind::SymbolKind(SymbolKind::Local), |
221 | ScopeDef::Label(..) => CompletionItemKind::SymbolKind(SymbolKind::Label), | 225 | hir::ScopeDef::Label(..) => CompletionItemKind::SymbolKind(SymbolKind::Label), |
222 | ScopeDef::AdtSelfType(..) | ScopeDef::ImplSelfType(..) => { | 226 | hir::ScopeDef::AdtSelfType(..) | hir::ScopeDef::ImplSelfType(..) => { |
223 | CompletionItemKind::SymbolKind(SymbolKind::SelfParam) | 227 | CompletionItemKind::SymbolKind(SymbolKind::SelfParam) |
224 | } | 228 | } |
225 | ScopeDef::Unknown => { | 229 | hir::ScopeDef::Unknown => { |
226 | let mut item = CompletionItem::new( | 230 | let mut item = CompletionItem::new( |
227 | CompletionKind::Reference, | 231 | CompletionKind::Reference, |
228 | self.ctx.source_range(), | 232 | self.ctx.source_range(), |
229 | local_name, | 233 | local_name.to_string(), |
230 | ); | 234 | ); |
231 | item.kind(CompletionItemKind::UnresolvedReference).add_import(import_to_add); | 235 | item.kind(CompletionItemKind::UnresolvedReference).add_import(import_to_add); |
232 | return Some(item.build()); | 236 | return Some(item.build()); |
233 | } | 237 | } |
234 | }; | 238 | }; |
235 | 239 | ||
240 | let local_name = local_name.to_string(); | ||
236 | let mut item = | 241 | let mut item = |
237 | CompletionItem::new(completion_kind, self.ctx.source_range(), local_name.clone()); | 242 | CompletionItem::new(completion_kind, self.ctx.source_range(), local_name.clone()); |
238 | if let ScopeDef::Local(local) = resolution { | 243 | if let hir::ScopeDef::Local(local) = resolution { |
239 | let ty = local.ty(self.ctx.db()); | 244 | let ty = local.ty(self.ctx.db()); |
240 | if !ty.is_unknown() { | 245 | if !ty.is_unknown() { |
241 | item.detail(ty.display(self.ctx.db()).to_string()); | 246 | item.detail(ty.display(self.ctx.db()).to_string()); |
@@ -260,8 +265,10 @@ impl<'a> Render<'a> { | |||
260 | { | 265 | { |
261 | if let Some(cap) = self.ctx.snippet_cap() { | 266 | if let Some(cap) = self.ctx.snippet_cap() { |
262 | let has_non_default_type_params = match resolution { | 267 | let has_non_default_type_params = match resolution { |
263 | ScopeDef::ModuleDef(Adt(it)) => it.has_non_default_type_params(self.ctx.db()), | 268 | hir::ScopeDef::ModuleDef(Adt(it)) => { |
264 | ScopeDef::ModuleDef(TypeAlias(it)) => { | 269 | it.has_non_default_type_params(self.ctx.db()) |
270 | } | ||
271 | hir::ScopeDef::ModuleDef(TypeAlias(it)) => { | ||
265 | it.has_non_default_type_params(self.ctx.db()) | 272 | it.has_non_default_type_params(self.ctx.db()) |
266 | } | 273 | } |
267 | _ => false, | 274 | _ => false, |
@@ -281,26 +288,26 @@ impl<'a> Render<'a> { | |||
281 | Some(item.build()) | 288 | Some(item.build()) |
282 | } | 289 | } |
283 | 290 | ||
284 | fn docs(&self, resolution: &ScopeDef) -> Option<Documentation> { | 291 | fn docs(&self, resolution: &hir::ScopeDef) -> Option<hir::Documentation> { |
285 | use hir::ModuleDef::*; | 292 | use hir::ModuleDef::*; |
286 | match resolution { | 293 | match resolution { |
287 | ScopeDef::ModuleDef(Module(it)) => it.docs(self.ctx.db()), | 294 | hir::ScopeDef::ModuleDef(Module(it)) => it.docs(self.ctx.db()), |
288 | ScopeDef::ModuleDef(Adt(it)) => it.docs(self.ctx.db()), | 295 | hir::ScopeDef::ModuleDef(Adt(it)) => it.docs(self.ctx.db()), |
289 | ScopeDef::ModuleDef(Variant(it)) => it.docs(self.ctx.db()), | 296 | hir::ScopeDef::ModuleDef(Variant(it)) => it.docs(self.ctx.db()), |
290 | ScopeDef::ModuleDef(Const(it)) => it.docs(self.ctx.db()), | 297 | hir::ScopeDef::ModuleDef(Const(it)) => it.docs(self.ctx.db()), |
291 | ScopeDef::ModuleDef(Static(it)) => it.docs(self.ctx.db()), | 298 | hir::ScopeDef::ModuleDef(Static(it)) => it.docs(self.ctx.db()), |
292 | ScopeDef::ModuleDef(Trait(it)) => it.docs(self.ctx.db()), | 299 | hir::ScopeDef::ModuleDef(Trait(it)) => it.docs(self.ctx.db()), |
293 | ScopeDef::ModuleDef(TypeAlias(it)) => it.docs(self.ctx.db()), | 300 | hir::ScopeDef::ModuleDef(TypeAlias(it)) => it.docs(self.ctx.db()), |
294 | _ => None, | 301 | _ => None, |
295 | } | 302 | } |
296 | } | 303 | } |
297 | 304 | ||
298 | fn is_deprecated(&self, resolution: &ScopeDef) -> bool { | 305 | fn is_deprecated(&self, resolution: &hir::ScopeDef) -> bool { |
299 | match resolution { | 306 | match resolution { |
300 | ScopeDef::ModuleDef(it) => self.ctx.is_deprecated_assoc_item(*it), | 307 | hir::ScopeDef::ModuleDef(it) => self.ctx.is_deprecated_assoc_item(*it), |
301 | ScopeDef::MacroDef(it) => self.ctx.is_deprecated(*it), | 308 | hir::ScopeDef::MacroDef(it) => self.ctx.is_deprecated(*it), |
302 | ScopeDef::GenericParam(it) => self.ctx.is_deprecated(*it), | 309 | hir::ScopeDef::GenericParam(it) => self.ctx.is_deprecated(*it), |
303 | ScopeDef::AdtSelfType(it) => self.ctx.is_deprecated(*it), | 310 | hir::ScopeDef::AdtSelfType(it) => self.ctx.is_deprecated(*it), |
304 | _ => false, | 311 | _ => false, |
305 | } | 312 | } |
306 | } | 313 | } |
@@ -327,21 +334,23 @@ fn compute_type_match( | |||
327 | } | 334 | } |
328 | } | 335 | } |
329 | 336 | ||
330 | fn compute_exact_name_match(ctx: &CompletionContext, completion_name: impl Into<String>) -> bool { | 337 | fn compute_exact_name_match(ctx: &CompletionContext, completion_name: &str) -> bool { |
331 | let completion_name = completion_name.into(); | ||
332 | ctx.expected_name.as_ref().map_or(false, |name| name.text() == completion_name) | 338 | ctx.expected_name.as_ref().map_or(false, |name| name.text() == completion_name) |
333 | } | 339 | } |
334 | 340 | ||
335 | fn compute_ref_match(ctx: &CompletionContext, completion_ty: &hir::Type) -> Option<Mutability> { | 341 | fn compute_ref_match( |
342 | ctx: &CompletionContext, | ||
343 | completion_ty: &hir::Type, | ||
344 | ) -> Option<hir::Mutability> { | ||
336 | let expected_type = ctx.expected_type.as_ref()?; | 345 | let expected_type = ctx.expected_type.as_ref()?; |
337 | if completion_ty != expected_type { | 346 | if completion_ty != expected_type { |
338 | let expected_type_without_ref = expected_type.remove_ref()?; | 347 | let expected_type_without_ref = expected_type.remove_ref()?; |
339 | if completion_ty.autoderef(ctx.db).any(|deref_ty| deref_ty == expected_type_without_ref) { | 348 | if completion_ty.autoderef(ctx.db).any(|deref_ty| deref_ty == expected_type_without_ref) { |
340 | cov_mark::hit!(suggest_ref); | 349 | cov_mark::hit!(suggest_ref); |
341 | let mutability = if expected_type.is_mutable_reference() { | 350 | let mutability = if expected_type.is_mutable_reference() { |
342 | Mutability::Mut | 351 | hir::Mutability::Mut |
343 | } else { | 352 | } else { |
344 | Mutability::Shared | 353 | hir::Mutability::Shared |
345 | }; | 354 | }; |
346 | return Some(mutability); | 355 | return Some(mutability); |
347 | }; | 356 | }; |
diff --git a/crates/ide_completion/src/render/enum_variant.rs b/crates/ide_completion/src/render/enum_variant.rs index 0c0c71134..28f056e77 100644 --- a/crates/ide_completion/src/render/enum_variant.rs +++ b/crates/ide_completion/src/render/enum_variant.rs | |||
@@ -1,6 +1,8 @@ | |||
1 | //! Renderer for `enum` variants. | 1 | //! Renderer for `enum` variants. |
2 | 2 | ||
3 | use hir::{HasAttrs, HirDisplay, ModPath, StructKind}; | 3 | use std::iter; |
4 | |||
5 | use hir::{HasAttrs, HirDisplay}; | ||
4 | use ide_db::SymbolKind; | 6 | use ide_db::SymbolKind; |
5 | use itertools::Itertools; | 7 | use itertools::Itertools; |
6 | 8 | ||
@@ -13,9 +15,9 @@ use crate::{ | |||
13 | pub(crate) fn render_variant<'a>( | 15 | pub(crate) fn render_variant<'a>( |
14 | ctx: RenderContext<'a>, | 16 | ctx: RenderContext<'a>, |
15 | import_to_add: Option<ImportEdit>, | 17 | import_to_add: Option<ImportEdit>, |
16 | local_name: Option<String>, | 18 | local_name: Option<hir::Name>, |
17 | variant: hir::Variant, | 19 | variant: hir::Variant, |
18 | path: Option<ModPath>, | 20 | path: Option<hir::ModPath>, |
19 | ) -> CompletionItem { | 21 | ) -> CompletionItem { |
20 | let _p = profile::span("render_enum_variant"); | 22 | let _p = profile::span("render_enum_variant"); |
21 | EnumRender::new(ctx, local_name, variant, path).render(import_to_add) | 23 | EnumRender::new(ctx, local_name, variant, path).render(import_to_add) |
@@ -24,42 +26,45 @@ pub(crate) fn render_variant<'a>( | |||
24 | #[derive(Debug)] | 26 | #[derive(Debug)] |
25 | struct EnumRender<'a> { | 27 | struct EnumRender<'a> { |
26 | ctx: RenderContext<'a>, | 28 | ctx: RenderContext<'a>, |
27 | name: String, | 29 | name: hir::Name, |
28 | variant: hir::Variant, | 30 | variant: hir::Variant, |
29 | path: Option<ModPath>, | 31 | path: Option<hir::ModPath>, |
30 | qualified_name: String, | 32 | qualified_name: hir::ModPath, |
31 | short_qualified_name: String, | 33 | short_qualified_name: hir::ModPath, |
32 | variant_kind: StructKind, | 34 | variant_kind: hir::StructKind, |
33 | } | 35 | } |
34 | 36 | ||
35 | impl<'a> EnumRender<'a> { | 37 | impl<'a> EnumRender<'a> { |
36 | fn new( | 38 | fn new( |
37 | ctx: RenderContext<'a>, | 39 | ctx: RenderContext<'a>, |
38 | local_name: Option<String>, | 40 | local_name: Option<hir::Name>, |
39 | variant: hir::Variant, | 41 | variant: hir::Variant, |
40 | path: Option<ModPath>, | 42 | path: Option<hir::ModPath>, |
41 | ) -> EnumRender<'a> { | 43 | ) -> EnumRender<'a> { |
42 | let name = local_name.unwrap_or_else(|| variant.name(ctx.db()).to_string()); | 44 | let name = local_name.unwrap_or_else(|| variant.name(ctx.db())); |
43 | let variant_kind = variant.kind(ctx.db()); | 45 | let variant_kind = variant.kind(ctx.db()); |
44 | 46 | ||
45 | let (qualified_name, short_qualified_name) = match &path { | 47 | let (qualified_name, short_qualified_name) = match &path { |
46 | Some(path) => { | 48 | Some(path) => { |
47 | let full = path.to_string(); | 49 | let short = hir::ModPath::from_segments( |
48 | let segments = path.segments(); | 50 | hir::PathKind::Plain, |
49 | let short = segments[segments.len().saturating_sub(2)..].iter().join("::"); | 51 | path.segments().iter().skip(path.segments().len().saturating_sub(2)).cloned(), |
50 | (full, short) | 52 | ); |
53 | (path.clone(), short) | ||
51 | } | 54 | } |
52 | None => (name.to_string(), name.to_string()), | 55 | None => ( |
56 | hir::ModPath::from_segments(hir::PathKind::Plain, iter::once(name.clone())), | ||
57 | hir::ModPath::from_segments(hir::PathKind::Plain, iter::once(name.clone())), | ||
58 | ), | ||
53 | }; | 59 | }; |
54 | 60 | ||
55 | EnumRender { ctx, name, variant, path, qualified_name, short_qualified_name, variant_kind } | 61 | EnumRender { ctx, name, variant, path, qualified_name, short_qualified_name, variant_kind } |
56 | } | 62 | } |
57 | |||
58 | fn render(self, import_to_add: Option<ImportEdit>) -> CompletionItem { | 63 | fn render(self, import_to_add: Option<ImportEdit>) -> CompletionItem { |
59 | let mut item = CompletionItem::new( | 64 | let mut item = CompletionItem::new( |
60 | CompletionKind::Reference, | 65 | CompletionKind::Reference, |
61 | self.ctx.source_range(), | 66 | self.ctx.source_range(), |
62 | self.qualified_name.clone(), | 67 | self.qualified_name.to_string(), |
63 | ); | 68 | ); |
64 | item.kind(SymbolKind::Variant) | 69 | item.kind(SymbolKind::Variant) |
65 | .set_documentation(self.variant.docs(self.ctx.db())) | 70 | .set_documentation(self.variant.docs(self.ctx.db())) |
@@ -67,12 +72,16 @@ impl<'a> EnumRender<'a> { | |||
67 | .add_import(import_to_add) | 72 | .add_import(import_to_add) |
68 | .detail(self.detail()); | 73 | .detail(self.detail()); |
69 | 74 | ||
70 | if self.variant_kind == StructKind::Tuple { | 75 | if self.variant_kind == hir::StructKind::Tuple { |
71 | cov_mark::hit!(inserts_parens_for_tuple_enums); | 76 | cov_mark::hit!(inserts_parens_for_tuple_enums); |
72 | let params = Params::Anonymous(self.variant.fields(self.ctx.db()).len()); | 77 | let params = Params::Anonymous(self.variant.fields(self.ctx.db()).len()); |
73 | item.add_call_parens(self.ctx.completion, self.short_qualified_name, params); | 78 | item.add_call_parens( |
79 | self.ctx.completion, | ||
80 | self.short_qualified_name.to_string(), | ||
81 | params, | ||
82 | ); | ||
74 | } else if self.path.is_some() { | 83 | } else if self.path.is_some() { |
75 | item.lookup_by(self.short_qualified_name); | 84 | item.lookup_by(self.short_qualified_name.to_string()); |
76 | } | 85 | } |
77 | 86 | ||
78 | let ty = self.variant.parent_enum(self.ctx.completion.db).ty(self.ctx.completion.db); | 87 | let ty = self.variant.parent_enum(self.ctx.completion.db).ty(self.ctx.completion.db); |
@@ -96,11 +105,11 @@ impl<'a> EnumRender<'a> { | |||
96 | .map(|field| (field.name(self.ctx.db()), field.ty(self.ctx.db()))); | 105 | .map(|field| (field.name(self.ctx.db()), field.ty(self.ctx.db()))); |
97 | 106 | ||
98 | match self.variant_kind { | 107 | match self.variant_kind { |
99 | StructKind::Tuple | StructKind::Unit => format!( | 108 | hir::StructKind::Tuple | hir::StructKind::Unit => format!( |
100 | "({})", | 109 | "({})", |
101 | detail_types.map(|(_, t)| t.display(self.ctx.db()).to_string()).format(", ") | 110 | detail_types.map(|(_, t)| t.display(self.ctx.db()).to_string()).format(", ") |
102 | ), | 111 | ), |
103 | StructKind::Record => format!( | 112 | hir::StructKind::Record => format!( |
104 | "{{ {} }}", | 113 | "{{ {} }}", |
105 | detail_types | 114 | detail_types |
106 | .map(|(n, t)| format!("{}: {}", n, t.display(self.ctx.db()).to_string())) | 115 | .map(|(n, t)| format!("{}: {}", n, t.display(self.ctx.db()).to_string())) |
diff --git a/crates/ide_completion/src/render/function.rs b/crates/ide_completion/src/render/function.rs index d681e2c91..63bd66926 100644 --- a/crates/ide_completion/src/render/function.rs +++ b/crates/ide_completion/src/render/function.rs | |||
@@ -1,6 +1,6 @@ | |||
1 | //! Renderer for function calls. | 1 | //! Renderer for function calls. |
2 | 2 | ||
3 | use hir::{HasSource, HirDisplay, Type}; | 3 | use hir::{HasSource, HirDisplay}; |
4 | use ide_db::SymbolKind; | 4 | use ide_db::SymbolKind; |
5 | use itertools::Itertools; | 5 | use itertools::Itertools; |
6 | use syntax::ast::Fn; | 6 | use syntax::ast::Fn; |
@@ -16,7 +16,7 @@ use crate::{ | |||
16 | pub(crate) fn render_fn<'a>( | 16 | pub(crate) fn render_fn<'a>( |
17 | ctx: RenderContext<'a>, | 17 | ctx: RenderContext<'a>, |
18 | import_to_add: Option<ImportEdit>, | 18 | import_to_add: Option<ImportEdit>, |
19 | local_name: Option<String>, | 19 | local_name: Option<hir::Name>, |
20 | fn_: hir::Function, | 20 | fn_: hir::Function, |
21 | ) -> Option<CompletionItem> { | 21 | ) -> Option<CompletionItem> { |
22 | let _p = profile::span("render_fn"); | 22 | let _p = profile::span("render_fn"); |
@@ -26,7 +26,7 @@ pub(crate) fn render_fn<'a>( | |||
26 | pub(crate) fn render_method<'a>( | 26 | pub(crate) fn render_method<'a>( |
27 | ctx: RenderContext<'a>, | 27 | ctx: RenderContext<'a>, |
28 | import_to_add: Option<ImportEdit>, | 28 | import_to_add: Option<ImportEdit>, |
29 | local_name: Option<String>, | 29 | local_name: Option<hir::Name>, |
30 | fn_: hir::Function, | 30 | fn_: hir::Function, |
31 | ) -> Option<CompletionItem> { | 31 | ) -> Option<CompletionItem> { |
32 | let _p = profile::span("render_method"); | 32 | let _p = profile::span("render_method"); |
@@ -45,11 +45,11 @@ struct FunctionRender<'a> { | |||
45 | impl<'a> FunctionRender<'a> { | 45 | impl<'a> FunctionRender<'a> { |
46 | fn new( | 46 | fn new( |
47 | ctx: RenderContext<'a>, | 47 | ctx: RenderContext<'a>, |
48 | local_name: Option<String>, | 48 | local_name: Option<hir::Name>, |
49 | fn_: hir::Function, | 49 | fn_: hir::Function, |
50 | is_method: bool, | 50 | is_method: bool, |
51 | ) -> Option<FunctionRender<'a>> { | 51 | ) -> Option<FunctionRender<'a>> { |
52 | let name = local_name.unwrap_or_else(|| fn_.name(ctx.db()).to_string()); | 52 | let name = local_name.unwrap_or_else(|| fn_.name(ctx.db())).to_string(); |
53 | let ast_node = fn_.source(ctx.db())?.value; | 53 | let ast_node = fn_.source(ctx.db())?.value; |
54 | 54 | ||
55 | Some(FunctionRender { ctx, name, func: fn_, ast_node, is_method }) | 55 | Some(FunctionRender { ctx, name, func: fn_, ast_node, is_method }) |
@@ -74,7 +74,7 @@ impl<'a> FunctionRender<'a> { | |||
74 | let ret_type = self.func.ret_type(self.ctx.db()); | 74 | let ret_type = self.func.ret_type(self.ctx.db()); |
75 | item.set_relevance(CompletionRelevance { | 75 | item.set_relevance(CompletionRelevance { |
76 | type_match: compute_type_match(self.ctx.completion, &ret_type), | 76 | type_match: compute_type_match(self.ctx.completion, &ret_type), |
77 | exact_name_match: compute_exact_name_match(self.ctx.completion, self.name.clone()), | 77 | exact_name_match: compute_exact_name_match(self.ctx.completion, &self.name), |
78 | ..CompletionRelevance::default() | 78 | ..CompletionRelevance::default() |
79 | }); | 79 | }); |
80 | 80 | ||
@@ -129,7 +129,7 @@ impl<'a> FunctionRender<'a> { | |||
129 | format!("-> {}", ret_ty.display(self.ctx.db())) | 129 | format!("-> {}", ret_ty.display(self.ctx.db())) |
130 | } | 130 | } |
131 | 131 | ||
132 | fn add_arg(&self, arg: &str, ty: &Type) -> String { | 132 | fn add_arg(&self, arg: &str, ty: &hir::Type) -> String { |
133 | if let Some(derefed_ty) = ty.remove_ref() { | 133 | if let Some(derefed_ty) = ty.remove_ref() { |
134 | for (name, local) in self.ctx.completion.locals.iter() { | 134 | for (name, local) in self.ctx.completion.locals.iter() { |
135 | if name == arg && local.ty(self.ctx.db()) == derefed_ty { | 135 | if name == arg && local.ty(self.ctx.db()) == derefed_ty { |
diff --git a/crates/ide_completion/src/render/macro_.rs b/crates/ide_completion/src/render/macro_.rs index 7578ad50b..0dfba8acc 100644 --- a/crates/ide_completion/src/render/macro_.rs +++ b/crates/ide_completion/src/render/macro_.rs | |||
@@ -1,6 +1,6 @@ | |||
1 | //! Renderer for macro invocations. | 1 | //! Renderer for macro invocations. |
2 | 2 | ||
3 | use hir::{Documentation, HasSource}; | 3 | use hir::HasSource; |
4 | use ide_db::SymbolKind; | 4 | use ide_db::SymbolKind; |
5 | use syntax::display::macro_label; | 5 | use syntax::display::macro_label; |
6 | 6 | ||
@@ -12,7 +12,7 @@ use crate::{ | |||
12 | pub(crate) fn render_macro<'a>( | 12 | pub(crate) fn render_macro<'a>( |
13 | ctx: RenderContext<'a>, | 13 | ctx: RenderContext<'a>, |
14 | import_to_add: Option<ImportEdit>, | 14 | import_to_add: Option<ImportEdit>, |
15 | name: String, | 15 | name: hir::Name, |
16 | macro_: hir::MacroDef, | 16 | macro_: hir::MacroDef, |
17 | ) -> Option<CompletionItem> { | 17 | ) -> Option<CompletionItem> { |
18 | let _p = profile::span("render_macro"); | 18 | let _p = profile::span("render_macro"); |
@@ -24,13 +24,14 @@ struct MacroRender<'a> { | |||
24 | ctx: RenderContext<'a>, | 24 | ctx: RenderContext<'a>, |
25 | name: String, | 25 | name: String, |
26 | macro_: hir::MacroDef, | 26 | macro_: hir::MacroDef, |
27 | docs: Option<Documentation>, | 27 | docs: Option<hir::Documentation>, |
28 | bra: &'static str, | 28 | bra: &'static str, |
29 | ket: &'static str, | 29 | ket: &'static str, |
30 | } | 30 | } |
31 | 31 | ||
32 | impl<'a> MacroRender<'a> { | 32 | impl<'a> MacroRender<'a> { |
33 | fn new(ctx: RenderContext<'a>, name: String, macro_: hir::MacroDef) -> MacroRender<'a> { | 33 | fn new(ctx: RenderContext<'a>, name: hir::Name, macro_: hir::MacroDef) -> MacroRender<'a> { |
34 | let name = name.to_string(); | ||
34 | let docs = ctx.docs(macro_); | 35 | let docs = ctx.docs(macro_); |
35 | let docs_str = docs.as_ref().map_or("", |s| s.as_str()); | 36 | let docs_str = docs.as_ref().map_or("", |s| s.as_str()); |
36 | let (bra, ket) = guess_macro_braces(&name, docs_str); | 37 | let (bra, ket) = guess_macro_braces(&name, docs_str); |
@@ -74,7 +75,11 @@ impl<'a> MacroRender<'a> { | |||
74 | if self.needs_bang() && self.ctx.snippet_cap().is_some() { | 75 | if self.needs_bang() && self.ctx.snippet_cap().is_some() { |
75 | format!("{}!{}…{}", self.name, self.bra, self.ket) | 76 | format!("{}!{}…{}", self.name, self.bra, self.ket) |
76 | } else { | 77 | } else { |
77 | self.banged_name() | 78 | if self.macro_.kind() == hir::MacroKind::Derive { |
79 | self.name.to_string() | ||
80 | } else { | ||
81 | self.banged_name() | ||
82 | } | ||
78 | } | 83 | } |
79 | } | 84 | } |
80 | 85 | ||
diff --git a/crates/ide_completion/src/test_utils.rs b/crates/ide_completion/src/test_utils.rs index 37be575e5..93c7c872c 100644 --- a/crates/ide_completion/src/test_utils.rs +++ b/crates/ide_completion/src/test_utils.rs | |||
@@ -12,7 +12,7 @@ use ide_db::{ | |||
12 | use itertools::Itertools; | 12 | use itertools::Itertools; |
13 | use stdx::{format_to, trim_indent}; | 13 | use stdx::{format_to, trim_indent}; |
14 | use syntax::{AstNode, NodeOrToken, SyntaxElement}; | 14 | use syntax::{AstNode, NodeOrToken, SyntaxElement}; |
15 | use test_utils::{assert_eq_text, RangeOrOffset}; | 15 | use test_utils::assert_eq_text; |
16 | 16 | ||
17 | use crate::{item::CompletionKind, CompletionConfig, CompletionItem}; | 17 | use crate::{item::CompletionKind, CompletionConfig, CompletionItem}; |
18 | 18 | ||
@@ -36,10 +36,7 @@ pub(crate) fn position(ra_fixture: &str) -> (RootDatabase, FilePosition) { | |||
36 | let mut database = RootDatabase::default(); | 36 | let mut database = RootDatabase::default(); |
37 | database.apply_change(change_fixture.change); | 37 | database.apply_change(change_fixture.change); |
38 | let (file_id, range_or_offset) = change_fixture.file_position.expect("expected a marker ($0)"); | 38 | let (file_id, range_or_offset) = change_fixture.file_position.expect("expected a marker ($0)"); |
39 | let offset = match range_or_offset { | 39 | let offset = range_or_offset.expect_offset(); |
40 | RangeOrOffset::Range(_) => panic!(), | ||
41 | RangeOrOffset::Offset(it) => it, | ||
42 | }; | ||
43 | (database, FilePosition { file_id, offset }) | 40 | (database, FilePosition { file_id, offset }) |
44 | } | 41 | } |
45 | 42 | ||
@@ -52,10 +49,11 @@ pub(crate) fn do_completion_with_config( | |||
52 | code: &str, | 49 | code: &str, |
53 | kind: CompletionKind, | 50 | kind: CompletionKind, |
54 | ) -> Vec<CompletionItem> { | 51 | ) -> Vec<CompletionItem> { |
55 | let mut kind_completions: Vec<CompletionItem> = | 52 | get_all_items(config, code) |
56 | get_all_items(config, code).into_iter().filter(|c| c.completion_kind == kind).collect(); | 53 | .into_iter() |
57 | kind_completions.sort_by(|l, r| l.label().cmp(r.label())); | 54 | .filter(|c| c.completion_kind == kind) |
58 | kind_completions | 55 | .sorted_by(|l, r| l.label().cmp(r.label())) |
56 | .collect() | ||
59 | } | 57 | } |
60 | 58 | ||
61 | pub(crate) fn completion_list(code: &str, kind: CompletionKind) -> String { | 59 | pub(crate) fn completion_list(code: &str, kind: CompletionKind) -> String { |
@@ -132,7 +130,7 @@ pub(crate) fn check_edit_with_config( | |||
132 | assert_eq_text!(&ra_fixture_after, &actual) | 130 | assert_eq_text!(&ra_fixture_after, &actual) |
133 | } | 131 | } |
134 | 132 | ||
135 | pub(crate) fn check_pattern_is_applicable(code: &str, check: fn(SyntaxElement) -> bool) { | 133 | pub(crate) fn check_pattern_is_applicable(code: &str, check: impl FnOnce(SyntaxElement) -> bool) { |
136 | let (db, pos) = position(code); | 134 | let (db, pos) = position(code); |
137 | 135 | ||
138 | let sema = Semantics::new(&db); | 136 | let sema = Semantics::new(&db); |