From cd599ec202d49d044d6d71fd3471e3c5199476e1 Mon Sep 17 00:00:00 2001
From: longfangsong <longfangsong@icloud.com>
Date: Mon, 20 Sep 2021 20:43:13 +0800
Subject: [PATCH] Address comments

---
 .../src/handlers/promote_mod_file.rs          | 78 +++++++++++--------
 crates/ide_assists/src/tests/generated.rs     |  2 +-
 2 files changed, 45 insertions(+), 35 deletions(-)

diff --git a/crates/ide_assists/src/handlers/promote_mod_file.rs b/crates/ide_assists/src/handlers/promote_mod_file.rs
index 5991c0989fe..1a177893f0c 100644
--- a/crates/ide_assists/src/handlers/promote_mod_file.rs
+++ b/crates/ide_assists/src/handlers/promote_mod_file.rs
@@ -3,12 +3,38 @@
     base_db::AnchoredPathBuf,
 };
 use syntax::{
-    ast::{self},
-    AstNode, TextRange,
+    ast::{self, Whitespace},
+    AstNode, AstToken, SourceFile, TextRange, TextSize,
 };
 
 use crate::assist_context::{AssistContext, Assists};
 
+/// Trim(remove leading and trailing whitespace) `initial_range` in `source_file`, return the trimmed range.
+fn trimmed_text_range(source_file: &SourceFile, initial_range: TextRange) -> TextRange {
+    let mut trimmed_range = initial_range;
+    while source_file
+        .syntax()
+        .token_at_offset(trimmed_range.start())
+        .find_map(Whitespace::cast)
+        .is_some()
+        && trimmed_range.start() < trimmed_range.end()
+    {
+        let start = trimmed_range.start() + TextSize::from(1);
+        trimmed_range = TextRange::new(start, trimmed_range.end());
+    }
+    while source_file
+        .syntax()
+        .token_at_offset(trimmed_range.end())
+        .find_map(Whitespace::cast)
+        .is_some()
+        && trimmed_range.start() < trimmed_range.end()
+    {
+        let end = trimmed_range.end() - TextSize::from(1);
+        trimmed_range = TextRange::new(trimmed_range.start(), end);
+    }
+    trimmed_range
+}
+
 // Assist: promote_mod_file
 //
 // Moves inline module's contents to a separate file.
@@ -17,7 +43,7 @@
 // //- /main.rs
 // mod a;
 // //- /a.rs
-// $0fn t() {}
+// $0fn t() {}$0
 // ```
 // ->
 // ```
@@ -26,18 +52,23 @@
 pub(crate) fn promote_mod_file(acc: &mut Assists, ctx: &AssistContext) -> Option<()> {
     let source_file = ctx.find_node_at_offset::<ast::SourceFile>()?;
     let module = ctx.sema.to_module_def(ctx.frange.file_id)?;
-    if module.is_mod_rs(ctx.db()) {
+    // Enable this assist if the user select all "meaningful" content in the source file
+    let trimmed_selected_range = trimmed_text_range(&source_file, ctx.frange.range);
+    let trimmed_file_range = trimmed_text_range(&source_file, source_file.syntax().text_range());
+    if module.is_mod_rs(ctx.db()) || trimmed_selected_range != trimmed_file_range {
         return None;
     }
+
     let target = TextRange::new(
         source_file.syntax().text_range().start(),
         source_file.syntax().text_range().end(),
     );
-    let path = format!("./{}/mod.rs", module.name(ctx.db())?.to_string());
+    let module_name = module.name(ctx.db())?.to_string();
+    let path = format!("./{}/mod.rs", module_name);
     let dst = AnchoredPathBuf { anchor: ctx.frange.file_id, path };
     acc.add(
         AssistId("promote_mod_file", AssistKind::Refactor),
-        "Promote Module to directory",
+        format!("Turn {}.rs to {}/mod.rs", module_name, module_name),
         target,
         |builder| {
             builder.move_file(ctx.frange.file_id, dst);
@@ -60,7 +91,7 @@ fn trivial() {
 mod a;
 //- /a.rs
 $0fn t() {}
-"#,
+$0"#,
             r#"
 //- /a/mod.rs
 fn t() {}
@@ -69,44 +100,23 @@ fn t() {}
     }
 
     #[test]
-    fn cursor_can_be_putted_anywhere() {
-        check_assist(
+    fn must_select_all_file() {
+        check_assist_not_applicable(
             promote_mod_file,
             r#"
 //- /main.rs
 mod a;
 //- /a.rs
 fn t() {}$0
-"#,
-            r#"
-//- /a/mod.rs
-fn t() {}
 "#,
         );
-        check_assist(
-            promote_mod_file,
-            r#"
-//- /main.rs
-mod a;
-//- /a.rs
-fn t()$0 {}
-"#,
-            r#"
-//- /a/mod.rs
-fn t() {}
-"#,
-        );
-        check_assist(
+        check_assist_not_applicable(
             promote_mod_file,
             r#"
 //- /main.rs
 mod a;
 //- /a.rs
-fn t($0) {}
-"#,
-            r#"
-//- /a/mod.rs
-fn t() {}
+$0fn$0 t() {}
 "#,
         );
     }
@@ -147,8 +157,8 @@ fn works_in_mod() {
             r#"//- /main.rs
 mod a;
 //- /a.rs
-mod b;
-$0fn t() {}
+$0mod b;
+fn t() {}$0
 //- /a/b.rs
 fn t1() {}
 "#,
diff --git a/crates/ide_assists/src/tests/generated.rs b/crates/ide_assists/src/tests/generated.rs
index 98e0bd67489..4ec3e0bb1bd 100644
--- a/crates/ide_assists/src/tests/generated.rs
+++ b/crates/ide_assists/src/tests/generated.rs
@@ -1234,7 +1234,7 @@ fn doctest_promote_mod_file() {
 //- /main.rs
 mod a;
 //- /a.rs
-$0fn t() {}
+$0fn t() {}$0
 "#####,
         r#####"
 fn t() {}
-- 
2.44.0