Skip to content

Commit fce1020

Browse files
committed
Better naming and fix some review comments
1 parent ef1aaeb commit fce1020

File tree

2 files changed

+14
-15
lines changed

2 files changed

+14
-15
lines changed

crates/ra_assists/src/assist_context.rs

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -292,24 +292,23 @@ impl AssistBuilder {
292292
}
293293

294294
pub(crate) struct AssistDirector {
295-
source_changes: Vec<SourceChange>,
296295
builders: FxHashMap<FileId, AssistBuilder>,
297296
}
298297

299298
impl AssistDirector {
300299
fn new() -> AssistDirector {
301-
AssistDirector { source_changes: vec![], builders: FxHashMap::default() }
300+
AssistDirector { builders: FxHashMap::default() }
302301
}
303302

304303
pub(crate) fn perform(&mut self, file_id: FileId, f: impl FnOnce(&mut AssistBuilder)) {
305304
let mut builder = self.builders.entry(file_id).or_insert(AssistBuilder::new(file_id));
306305
f(&mut builder);
307306
}
308307

309-
fn finish(mut self) -> Vec<SourceChange> {
310-
for (_, builder) in self.builders.into_iter().collect::<Vec<(FileId, AssistBuilder)>>() {
311-
self.source_changes.push(builder.finish());
312-
}
313-
self.source_changes
308+
fn finish(self) -> Vec<SourceChange> {
309+
self.builders
310+
.into_iter()
311+
.map(|(_, builder)| builder.finish())
312+
.collect::<Vec<SourceChange>>()
314313
}
315314
}

crates/ra_assists/src/handlers/extract_struct_from_enum_variant.rs

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -59,8 +59,8 @@ pub(crate) fn extract_struct_from_enum(acc: &mut Assists, ctx: &AssistContext) -
5959
let res = definition.find_usages(&ctx.db, None);
6060
let module_def = mod_def_for_target_module(ctx, &enum_name);
6161
let start_offset = variant.parent_enum().syntax().text_range().start();
62-
let mut seen_files_map: FxHashSet<Module> = FxHashSet::default();
63-
seen_files_map.insert(module_def.module(ctx.db).unwrap());
62+
let mut visited_modules_set: FxHashSet<Module> = FxHashSet::default();
63+
visited_modules_set.insert(module_def.module(ctx.db).unwrap());
6464
for reference in res {
6565
let source_file = ctx.sema.parse(reference.file_range.file_id);
6666
update_reference(
@@ -69,7 +69,7 @@ pub(crate) fn extract_struct_from_enum(acc: &mut Assists, ctx: &AssistContext) -
6969
reference,
7070
&source_file,
7171
&module_def,
72-
&mut seen_files_map,
72+
&mut visited_modules_set,
7373
);
7474
}
7575
extract_struct_def(
@@ -101,7 +101,7 @@ fn mod_def_for_target_module(ctx: &AssistContext, enum_name: &str) -> ModuleDef
101101
ImportsLocator::new(ctx.db).find_imports(enum_name).first().unwrap().left().unwrap()
102102
}
103103

104-
fn insert_use_import(
104+
fn insert_import(
105105
ctx: &AssistContext,
106106
builder: &mut AssistBuilder,
107107
path: &ast::PathExpr,
@@ -179,7 +179,7 @@ fn update_reference(
179179
reference: Reference,
180180
source_file: &SourceFile,
181181
module_def: &ModuleDef,
182-
seen_files_map: &mut FxHashSet<Module>,
182+
visited_modules_set: &mut FxHashSet<Module>,
183183
) -> Option<()> {
184184
let path_expr: ast::PathExpr = find_node_at_offset::<ast::PathExpr>(
185185
source_file.syntax(),
@@ -195,8 +195,8 @@ fn update_reference(
195195
);
196196
edit.perform(reference.file_range.file_id, |builder| {
197197
let module = ctx.sema.scope(&path_expr.syntax()).module().unwrap();
198-
if !seen_files_map.contains(&module) {
199-
if insert_use_import(
198+
if !visited_modules_set.contains(&module) {
199+
if insert_import(
200200
ctx,
201201
builder,
202202
&path_expr,
@@ -206,7 +206,7 @@ fn update_reference(
206206
)
207207
.is_some()
208208
{
209-
seen_files_map.insert(module);
209+
visited_modules_set.insert(module);
210210
}
211211
}
212212
builder.replace(inside_list_range, format!("{}{}", segment, list));

0 commit comments

Comments
 (0)