Remove symbols when sentence deleted.

coq · Jul 12, 2024 · a3e8b67 · a3e8b67
1 parent 29f1f95
commit a3e8b67
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/language-server/dm/document.ml b/language-server/dm/document.ml
@@ -193,8 +193,9 @@ let remove_sentence parsed id =
   | Some sentence ->
     let sentences_by_id = SM.remove id parsed.sentences_by_id in
     let sentences_by_end = LM.remove sentence.stop parsed.sentences_by_end in
+    let outline = List.filter (fun (e : outline_element) -> e.id != id) parsed.outline in
     (* TODO clean up the schedule and free cached states *)
-    { parsed with sentences_by_id; sentences_by_end; }
+    { parsed with sentences_by_id; sentences_by_end; outline }
 
 let sentences parsed =
   List.map snd @@ SM.bindings parsed.sentences_by_id

diff --git a/language-server/vscoqtop/lspManager.ml b/language-server/vscoqtop/lspManager.ml
@@ -444,8 +444,8 @@ let documentSymbol id params =
   let Lsp.Types.DocumentSymbolParams.{ textDocument = {uri}} = params in
   match Hashtbl.find_opt states (DocumentUri.to_path uri) with
   | None -> log @@ "[documentSymbol] ignoring event on non existant document"; Error("Document does not exist"), []
-  | Some st -> log @@ "[documentSymbol] getting symbols";
-    let symbols = Dm.DocumentManager.get_document_symbols st in
+  | Some tab -> log @@ "[documentSymbol] getting symbols";
+    let symbols = Dm.DocumentManager.get_document_symbols tab.st in
     Ok(Some (`DocumentSymbol symbols)), []
 
 let coqtopResetCoq id params =