implement textDocument/didSave (#196)

Author: joshuadavidthomas

crates/djls-server/src/server.rs

@@ -208,6 +208,21 @@ impl LanguageServer for DjangoLanguageServer {
         .await;
     }
 
+    async fn did_save(&self, params: lsp_types::DidSaveTextDocumentParams) {
+        tracing::info!("Saved document: {:?}", params.text_document.uri);
+
+        self.with_session_mut(|session| {
+            let Some(url) =
+                paths::parse_lsp_uri(&params.text_document.uri, paths::LspContext::DidSave)
+            else {
+                return;
+            };
+
+            session.save_document(&url);
+        })
+        .await;
+    }
+
     async fn did_change(&self, params: lsp_types::DidChangeTextDocumentParams) {
         tracing::info!("Changed document: {:?}", params.text_document.uri);
 

crates/djls-server/src/session.rs

@@ -172,6 +172,17 @@ impl Session {
         }
     }
 
+    pub fn save_document(&mut self, url: &Url) {
+        // Touch file in database to trigger re-analysis
+        if let Some(path) = paths::url_to_path(url) {
+            self.with_db_mut(|db| {
+                if db.has_file(&path) {
+                    db.touch_file(&path);
+                }
+            });
+        }
+    }
+
     /// Close a document.
     ///
     /// Removes from workspace buffers and triggers database invalidation to fall back to disk.

crates/djls-workspace/src/paths.rs

@@ -42,6 +42,8 @@ pub enum LspContext {
     DidChange,
     /// textDocument/didClose notification
     DidClose,
+    /// textDocument/didSave notification
+    DidSave,
     /// textDocument/completion request
     Completion,
 }
@@ -52,6 +54,7 @@ impl std::fmt::Display for LspContext {
             Self::DidOpen => write!(f, "didOpen"),
             Self::DidChange => write!(f, "didChange"),
             Self::DidClose => write!(f, "didClose"),
+            Self::DidSave => write!(f, "didSave"),
             Self::Completion => write!(f, "completion"),
         }
     }