Skip to content

Commit 2bf5f14

Browse files
committed
Use trimmed selection range
1 parent 2efcff7 commit 2bf5f14

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

crates/ide_assists/src/handlers/extract_module.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ use super::remove_unused_param::range_to_remove;
5151
// }
5252
// ```
5353
pub(crate) fn extract_module(acc: &mut Assists, ctx: &AssistContext) -> Option<()> {
54-
if ctx.frange.range.is_empty() {
54+
if ctx.has_empty_selection() {
5555
return None;
5656
}
5757

@@ -66,7 +66,7 @@ pub(crate) fn extract_module(acc: &mut Assists, ctx: &AssistContext) -> Option<(
6666
curr_parent_module = ast::Module::cast(mod_syn_opt);
6767
}
6868

69-
let mut module = extract_target(&node, ctx.frange.range)?;
69+
let mut module = extract_target(&node, ctx.selection_trimmed())?;
7070
if module.body_items.len() == 0 {
7171
return None;
7272
}
@@ -119,7 +119,7 @@ pub(crate) fn extract_module(acc: &mut Assists, ctx: &AssistContext) -> Option<(
119119

120120
let mut usages_to_be_updated_for_curr_file = vec![];
121121
for usages_to_be_updated_for_file in usages_to_be_processed {
122-
if usages_to_be_updated_for_file.0 == ctx.frange.file_id {
122+
if usages_to_be_updated_for_file.0 == ctx.file_id() {
123123
usages_to_be_updated_for_curr_file = usages_to_be_updated_for_file.1;
124124
continue;
125125
}
@@ -129,7 +129,7 @@ pub(crate) fn extract_module(acc: &mut Assists, ctx: &AssistContext) -> Option<(
129129
}
130130
}
131131

132-
builder.edit_file(ctx.frange.file_id);
132+
builder.edit_file(ctx.file_id());
133133
for usage_to_be_processed in usages_to_be_updated_for_curr_file {
134134
builder.replace(usage_to_be_processed.0, usage_to_be_processed.1)
135135
}
@@ -426,11 +426,11 @@ impl Module {
426426
ctx: &AssistContext,
427427
) -> Option<TextRange> {
428428
//We only need to find in the current file
429-
let selection_range = ctx.frange.range;
430-
let search_scope = SearchScope::single_file(ctx.frange.file_id);
429+
let selection_range = ctx.selection_trimmed();
430+
let curr_file_id = ctx.file_id();
431+
let search_scope = SearchScope::single_file(curr_file_id);
431432
let usage_res = def.usages(&ctx.sema).in_scope(search_scope).all();
432-
let curr_file_id = ctx.frange.file_id;
433-
let file = ctx.sema.parse(ctx.frange.file_id);
433+
let file = ctx.sema.parse(curr_file_id);
434434

435435
let mut exists_inside_sel = false;
436436
let mut exists_outside_sel = false;
@@ -470,7 +470,7 @@ impl Module {
470470
let use_stmt_opt: Option<ast::Use> = usage_res.into_iter().find_map(|x| {
471471
let file_id = x.0;
472472
let mut use_opt: Option<ast::Use> = None;
473-
if file_id == ctx.frange.file_id {
473+
if file_id == curr_file_id {
474474
(&x.1).into_iter().for_each(|x| {
475475
let node_opt: Option<ast::Use> = find_node_at_range(file.syntax(), x.range);
476476
if let Some(node) = node_opt {

0 commit comments

Comments
 (0)