buf fix for format on lines less than preious

movebit · Jan 30, 2024 · 138e828 · 138e828
1 parent 11d0f6e
commit 138e828
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/language/move-analyzer/src/movefmt.rs b/language/move-analyzer/src/movefmt.rs
@@ -48,10 +48,18 @@ pub fn on_movefmt_request(context: &Context, request: &Request, fmt_cfg: &FmtCon
     movefmt_cfg.set().max_width(fmt_cfg.max_width as usize);
     movefmt_cfg.set().indent_size(fmt_cfg.indent_size as usize);
     let content_format = movefmt::core::fmt::format_entry(content_origin.clone(), movefmt_cfg).unwrap();
+
+    let result_line = 
+        if content_format.clone().lines().count() >= content_origin.clone().lines().count() {
+            content_format.clone().lines().count()
+        } else {
+            content_origin.clone().lines().count()
+        };
+
     let result = Some(vec![TextEdit {
         range: lsp_types::Range {
             start: Position { line: 0, character: 0 },
-            end: Position { line: content_format.clone().lines().count() as u32, character: 0 },
+            end: Position { line: result_line as u32, character: 0 },
         },
         new_text: content_format,
     }]);