diff options
Diffstat (limited to 'crates/ra_ide/src/completion/complete_scope.rs')
-rw-r--r-- | crates/ra_ide/src/completion/complete_scope.rs | 136 |
1 files changed, 3 insertions, 133 deletions
diff --git a/crates/ra_ide/src/completion/complete_scope.rs b/crates/ra_ide/src/completion/complete_scope.rs index 64b04ec2b..e2ee86dd1 100644 --- a/crates/ra_ide/src/completion/complete_scope.rs +++ b/crates/ra_ide/src/completion/complete_scope.rs | |||
@@ -1,12 +1,6 @@ | |||
1 | //! FIXME: write short doc here | 1 | //! FIXME: write short doc here |
2 | 2 | ||
3 | use ra_assists::auto_import_text_edit; | 3 | use crate::completion::{CompletionContext, Completions}; |
4 | use ra_syntax::{ast, AstNode, SmolStr}; | ||
5 | use ra_text_edit::TextEditBuilder; | ||
6 | use rustc_hash::FxHashMap; | ||
7 | |||
8 | use crate::completion::{CompletionContext, CompletionItem, CompletionKind, Completions}; | ||
9 | use hir::{ModPath, PathKind}; | ||
10 | 4 | ||
11 | pub(super) fn complete_scope(acc: &mut Completions, ctx: &CompletionContext) { | 5 | pub(super) fn complete_scope(acc: &mut Completions, ctx: &CompletionContext) { |
12 | if !ctx.is_trivial_path { | 6 | if !ctx.is_trivial_path { |
@@ -16,138 +10,14 @@ pub(super) fn complete_scope(acc: &mut Completions, ctx: &CompletionContext) { | |||
16 | ctx.analyzer.process_all_names(ctx.db, &mut |name, res| { | 10 | ctx.analyzer.process_all_names(ctx.db, &mut |name, res| { |
17 | acc.add_resolution(ctx, name.to_string(), &res) | 11 | acc.add_resolution(ctx, name.to_string(), &res) |
18 | }); | 12 | }); |
19 | |||
20 | // auto-import | ||
21 | // We fetch ident from the original file, because we need to pre-filter auto-imports | ||
22 | if ast::NameRef::cast(ctx.token.parent()).is_some() { | ||
23 | let import_resolver = ImportResolver::new(); | ||
24 | let import_names = import_resolver.all_names(ctx.token.text()); | ||
25 | import_names.into_iter().for_each(|(name, path)| { | ||
26 | let edit = { | ||
27 | let mut builder = TextEditBuilder::default(); | ||
28 | builder.replace(ctx.source_range(), name.to_string()); | ||
29 | auto_import_text_edit( | ||
30 | &ctx.token.parent(), | ||
31 | &ctx.token.parent(), | ||
32 | &path, | ||
33 | &mut builder, | ||
34 | ); | ||
35 | builder.finish() | ||
36 | }; | ||
37 | |||
38 | // Hack: copied this check form conv.rs beacause auto import can produce edits | ||
39 | // that invalidate assert in conv_with. | ||
40 | if edit | ||
41 | .as_atoms() | ||
42 | .iter() | ||
43 | .filter(|atom| !ctx.source_range().is_subrange(&atom.delete)) | ||
44 | .all(|atom| ctx.source_range().intersection(&atom.delete).is_none()) | ||
45 | { | ||
46 | CompletionItem::new( | ||
47 | CompletionKind::Reference, | ||
48 | ctx.source_range(), | ||
49 | build_import_label(&name, &path), | ||
50 | ) | ||
51 | .text_edit(edit) | ||
52 | .add_to(acc); | ||
53 | } | ||
54 | }); | ||
55 | } | ||
56 | } | ||
57 | |||
58 | fn build_import_label(name: &str, path: &ModPath) -> String { | ||
59 | let mut buf = String::with_capacity(64); | ||
60 | buf.push_str(name); | ||
61 | buf.push_str(" ("); | ||
62 | buf.push_str(&path.to_string()); | ||
63 | buf.push_str(")"); | ||
64 | buf | ||
65 | } | ||
66 | |||
67 | #[derive(Debug, Clone, Default)] | ||
68 | pub(crate) struct ImportResolver { | ||
69 | // todo: use fst crate or something like that | ||
70 | dummy_names: Vec<(SmolStr, ModPath)>, | ||
71 | } | ||
72 | |||
73 | impl ImportResolver { | ||
74 | pub(crate) fn new() -> Self { | ||
75 | use hir::name; | ||
76 | |||
77 | let dummy_names = vec![ | ||
78 | ( | ||
79 | SmolStr::new("fmt"), | ||
80 | ModPath { kind: PathKind::Plain, segments: vec![name![std], name![fmt]] }, | ||
81 | ), | ||
82 | ( | ||
83 | SmolStr::new("io"), | ||
84 | ModPath { kind: PathKind::Plain, segments: vec![name![std], name![io]] }, | ||
85 | ), | ||
86 | ( | ||
87 | SmolStr::new("iter"), | ||
88 | ModPath { kind: PathKind::Plain, segments: vec![name![std], name![iter]] }, | ||
89 | ), | ||
90 | ( | ||
91 | SmolStr::new("hash"), | ||
92 | ModPath { kind: PathKind::Plain, segments: vec![name![std], name![hash]] }, | ||
93 | ), | ||
94 | ( | ||
95 | SmolStr::new("Debug"), | ||
96 | ModPath { | ||
97 | kind: PathKind::Plain, | ||
98 | segments: vec![name![std], name![fmt], name![Debug]], | ||
99 | }, | ||
100 | ), | ||
101 | ( | ||
102 | SmolStr::new("Display"), | ||
103 | ModPath { | ||
104 | kind: PathKind::Plain, | ||
105 | segments: vec![name![std], name![fmt], name![Display]], | ||
106 | }, | ||
107 | ), | ||
108 | ( | ||
109 | SmolStr::new("Hash"), | ||
110 | ModPath { | ||
111 | kind: PathKind::Plain, | ||
112 | segments: vec![name![std], name![hash], name![Hash]], | ||
113 | }, | ||
114 | ), | ||
115 | ( | ||
116 | SmolStr::new("Hasher"), | ||
117 | ModPath { | ||
118 | kind: PathKind::Plain, | ||
119 | segments: vec![name![std], name![hash], name![Hasher]], | ||
120 | }, | ||
121 | ), | ||
122 | ( | ||
123 | SmolStr::new("Iterator"), | ||
124 | ModPath { | ||
125 | kind: PathKind::Plain, | ||
126 | segments: vec![name![std], name![iter], name![Iterator]], | ||
127 | }, | ||
128 | ), | ||
129 | ]; | ||
130 | |||
131 | ImportResolver { dummy_names } | ||
132 | } | ||
133 | |||
134 | // Returns a map of importable items filtered by name. | ||
135 | // The map associates item name with its full path. | ||
136 | // todo: should return Resolutions | ||
137 | pub(crate) fn all_names(&self, name: &str) -> FxHashMap<SmolStr, ModPath> { | ||
138 | if name.len() > 1 { | ||
139 | self.dummy_names.iter().filter(|(n, _)| n.contains(name)).cloned().collect() | ||
140 | } else { | ||
141 | FxHashMap::default() | ||
142 | } | ||
143 | } | ||
144 | } | 13 | } |
145 | 14 | ||
146 | #[cfg(test)] | 15 | #[cfg(test)] |
147 | mod tests { | 16 | mod tests { |
148 | use crate::completion::{do_completion, CompletionItem, CompletionKind}; | ||
149 | use insta::assert_debug_snapshot; | 17 | use insta::assert_debug_snapshot; |
150 | 18 | ||
19 | use crate::completion::{do_completion, CompletionItem, CompletionKind}; | ||
20 | |||
151 | fn do_reference_completion(code: &str) -> Vec<CompletionItem> { | 21 | fn do_reference_completion(code: &str) -> Vec<CompletionItem> { |
152 | do_completion(code, CompletionKind::Reference) | 22 | do_completion(code, CompletionKind::Reference) |
153 | } | 23 | } |