tweak

hatoo · Sep 12, 2023 · b9ffa9d · b9ffa9d
1 parent 4e62f44
commit b9ffa9d
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -177,10 +177,12 @@ impl LanguageServer for Backend {
             .log_message(MessageType::INFO, "file saved!")
             .await;
     }
-    async fn did_close(&self, _: DidCloseTextDocumentParams) {
+    async fn did_close(&self, param: DidCloseTextDocumentParams) {
         self.client
             .log_message(MessageType::INFO, "file closed!")
             .await;
+
+        self.document_map.remove(&param.text_document.uri);
     }
 
     async fn semantic_tokens_full(
@@ -353,13 +355,13 @@ impl LanguageServer for Backend {
         }
 
         if markdown.is_empty() {
-            return Ok(None);
+            Ok(None)
+        } else {
+            Ok(Some(Hover {
+                contents: HoverContents::Scalar(MarkedString::String(markdown)),
+                range: None,
+            }))
         }
-
-        Ok(Some(Hover {
-            contents: HoverContents::Scalar(MarkedString::String(markdown)),
-            range: None,
-        }))
     }
 
     async fn completion(&self, params: CompletionParams) -> Result<Option<CompletionResponse>> {