ls: Refactor textDocument/didSave in a similar vein

helix-lsp/src/client.rs

@@ -490,11 +490,11 @@ impl Client {
 
     // will_save / will_save_wait_until
 
-    pub async fn text_document_did_save(
+    pub fn text_document_did_save(
         &self,
         text_document: lsp::TextDocumentIdentifier,
         text: &Rope,
-    ) -> Result<()> {
+    ) -> Option<impl Future<Output = Result<()>>> {
         let capabilities = self.capabilities.get().unwrap();
 
         let include_text = match &capabilities.text_document_sync {
@@ -507,17 +507,18 @@ impl Client {
                     include_text,
                 }) => include_text.unwrap_or(false),
                 // Supported(false)
-                _ => return Ok(()),
+                _ => return None,
             },
             // unsupported
-            _ => return Ok(()),
+            _ => return None,
         };
 
-        self.notify::<lsp::notification::DidSaveTextDocument>(lsp::DidSaveTextDocumentParams {
+        Some(self.notify::<lsp::notification::DidSaveTextDocument>(
-            text_document,
+            lsp::DidSaveTextDocumentParams {
-            text: include_text.then(|| text.into()),
+                text_document,
-        })
+                text: include_text.then(|| text.into()),
-        .await
+            },
+        ))
     }
 
     pub fn completion(

helix-view/src/document.rs

@@ -471,10 +471,10 @@ impl Document {
             let mut file = File::create(path).await?;
             to_writer(&mut file, encoding, &text).await?;
 
-            if let Some(language_server) = language_server {
+            if let Some(notification) = language_server.and_then(|language_server| {
-                language_server
+                language_server.text_document_did_save(identifier, &text)
-                    .text_document_did_save(identifier, &text)
+            }) {
-                    .await?;
+                notification.await?;
             }
 
             Ok(())