From e4ff75b4d4eb793004e91279c82fde7766fc5a00 Mon Sep 17 00:00:00 2001
From: Blaž Hrastnik
Date: Wed, 12 May 2021 17:24:55 +0900
Subject: Add :fmt (formats the whole file).

---
 helix-term/src/commands.rs | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

(limited to 'helix-term/src')

diff --git a/helix-term/src/commands.rs b/helix-term/src/commands.rs
index e26dbbaa..4a42bbba 100644
--- a/helix-term/src/commands.rs
+++ b/helix-term/src/commands.rs
@@ -888,10 +888,34 @@ mod cmd {
         }
         tokio::spawn(doc.save());
     }
+
     fn new_file(editor: &mut Editor, args: &[&str], event: PromptEvent) {
         editor.new_file(Action::Replace);
     }
 
+    fn format(editor: &mut Editor, args: &[&str], event: PromptEvent) {
+        let (view, doc) = editor.current();
+
+        if let Some(language_server) = doc.language_server() {
+            // TODO: await, no blocking
+            let transaction = helix_lsp::block_on(
+                language_server
+                    .text_document_formatting(doc.identifier(), lsp::FormattingOptions::default()),
+            )
+            .map(|edits| {
+                helix_lsp::util::generate_transaction_from_edits(
+                    doc.text(),
+                    edits,
+                    language_server.offset_encoding(),
+                )
+            });
+
+            if let Ok(transaction) = transaction {
+                doc.apply(&transaction, view.id);
+            }
+        }
+    }
+
     pub const COMMAND_LIST: &[Command] = &[
         Command {
             name: "quit",
@@ -928,6 +952,13 @@ mod cmd {
             fun: new_file,
             completer: Some(completers::filename),
         },
+        Command {
+            name: "format",
+            alias: Some("fmt"),
+            doc: "Format the file using a formatter.",
+            fun: format,
+            completer: None,
+        },
     ];
 
     pub static COMMANDS: Lazy<HashMap<&'static str, &'static Command>> = Lazy::new(|| {
-- 
cgit v1.2.3-70-g09d2