From 2fcbf58fd12596168fc661d6fb1db9150a1c969f Mon Sep 17 00:00:00 2001
From: Ary Borenszweig <asterite@gmail.com>
Date: Wed, 4 Dec 2024 15:02:08 -0300
Subject: [PATCH] Fix LSP auto-import text indent

---
 tooling/lsp/src/requests/completion/tests.rs | 48 ++++++++++++++++++++
 tooling/lsp/src/use_segment_positions.rs     |  4 +-
 2 files changed, 50 insertions(+), 2 deletions(-)

diff --git a/tooling/lsp/src/requests/completion/tests.rs b/tooling/lsp/src/requests/completion/tests.rs
index 9306e38a48a..97c7ad86d5a 100644
--- a/tooling/lsp/src/requests/completion/tests.rs
+++ b/tooling/lsp/src/requests/completion/tests.rs
@@ -1586,6 +1586,54 @@ fn main() {
         assert_eq!(changed, expected);
     }
 
+    #[test]
+    async fn test_auto_import_inserts_after_last_use_in_nested_module() {
+        let src = r#"mod foo {
+    pub mod bar {
+        pub fn hello_world() {}
+    }
+}
+
+mod baz {
+    fn qux() {}
+}
+
+mod other {
+    use baz::qux;
+
+    fn main() {
+        hel>|<
+    }
+}"#;
+
+        let expected = r#"mod foo {
+    pub mod bar {
+        pub fn hello_world() {}
+    }
+}
+
+mod baz {
+    fn qux() {}
+}
+
+mod other {
+    use baz::qux;
+    use super::foo::bar::hello_world;
+
+    fn main() {
+        hel
+    }
+}"#;
+        let mut items = get_completions(src).await;
+        assert_eq!(items.len(), 1);
+
+        let item = items.remove(0);
+
+        let changed =
+            apply_text_edits(&src.replace(">|<", ""), &item.additional_text_edits.unwrap());
+        assert_eq!(changed, expected);
+    }
+
     #[test]
     async fn test_does_not_auto_import_test_functions() {
         let src = r#"
diff --git a/tooling/lsp/src/use_segment_positions.rs b/tooling/lsp/src/use_segment_positions.rs
index f9a3f429029..246ff653245 100644
--- a/tooling/lsp/src/use_segment_positions.rs
+++ b/tooling/lsp/src/use_segment_positions.rs
@@ -318,7 +318,7 @@ fn new_use_completion_item_additional_text_edits(
     request: UseCompletionItemAdditionTextEditsRequest,
 ) -> Vec<TextEdit> {
     let line = request.auto_import_line as u32;
-    let character = (request.nesting * 4) as u32;
+    let character = 0;
     let indent = " ".repeat(request.nesting * 4);
     let mut newlines = "\n";
 
@@ -331,6 +331,6 @@ fn new_use_completion_item_additional_text_edits(
 
     vec![TextEdit {
         range: Range { start: Position { line, character }, end: Position { line, character } },
-        new_text: format!("use {};{}{}", request.full_path, newlines, indent),
+        new_text: format!("{}use {};{}", indent, request.full_path, newlines),
     }]
 }