feat: Add commands to restart and stop LSP servers, enhance LSP client disposal by clearing diagnostics

Author: bajrangCoder

src/cm/commandRegistry.js

@@ -56,6 +56,7 @@ import {
 import { indentUnit as indentUnitFacet } from "@codemirror/language";
 import {
 	closeLintPanel,
+	forceLinting,
 	nextDiagnostic,
 	openLintPanel,
 	previousDiagnostic,
@@ -76,6 +77,8 @@ import {
 } from "@codemirror/lsp-client";
 import { Compartment, EditorSelection } from "@codemirror/state";
 import { keymap } from "@codemirror/view";
+import { clearDiagnosticsEffect, clientManager } from "cm/lsp";
+import toast from "components/toast";
 import prompt from "dialogs/prompt";
 import actions from "handlers/quickTools";
 import keyBindings from "lib/keyBindings";
@@ -938,6 +941,47 @@ function registerLspCommands() {
 			return lspCloseReferencePanel(resolvedView);
 		},
 	});
+	addCommand({
+		name: "restartAllLspServers",
+		description: "Restart all running LSP servers",
+		readOnly: true,
+		requiresView: false,
+		async run() {
+			const activeClients = clientManager.getActiveClients();
+			if (!activeClients.length) {
+				toast("No LSP servers are currently running");
+				return true;
+			}
+			const count = activeClients.length;
+			toast(`Restarting ${count} LSP server${count > 1 ? "s" : ""}...`);
+
+			// Dispose all clients (also clears diagnostics)
+			await clientManager.dispose();
+
+			// Trigger reconnect for active file
+			editorManager?.restartLsp?.();
+			return true;
+		},
+	});
+	addCommand({
+		name: "stopAllLspServers",
+		description: "Stop all running LSP servers",
+		readOnly: true,
+		requiresView: false,
+		async run() {
+			const activeClients = clientManager.getActiveClients();
+			if (!activeClients.length) {
+				toast("No LSP servers are currently running");
+				return true;
+			}
+			const count = activeClients.length;
+
+			// Dispose all clients
+			await clientManager.dispose();
+			toast(`Stopped ${count} LSP server${count > 1 ? "s" : ""}`);
+			return true;
+		},
+	});
 }
 
 function registerLintCommands() {

src/cm/lsp/clientManager.ts

@@ -12,11 +12,12 @@ import {
 	serverDiagnostics,
 	signatureHelp,
 } from "@codemirror/lsp-client";
-import { Extension, MapMode } from "@codemirror/state";
+import { EditorState, Extension, MapMode } from "@codemirror/state";
 import { EditorView, keymap } from "@codemirror/view";
 import lspStatusBar from "components/lspStatusBar";
 import NotificationManager from "lib/notificationManager";
 import Uri from "utils/Uri";
+import { clearDiagnosticsEffect } from "./diagnostics";
 import { inlayHintsExtension } from "./inlayHints";
 import { ensureServerRunning } from "./serverLauncher";
 import serverRegistry from "./serverRegistry";
@@ -250,6 +251,54 @@ export class LspClientManager {
 	}
 
 	async dispose(): Promise<void> {
+		try {
+			interface FileWithSession {
+				id?: string;
+				type?: string;
+				session?: EditorState;
+			}
+
+			interface EditorManagerLike {
+				files?: FileWithSession[];
+				editor?: EditorView;
+				activeFile?: FileWithSession;
+			}
+
+			const em = (globalThis as Record<string, unknown>).editorManager as
+				| EditorManagerLike
+				| undefined;
+
+			if (em?.editor) {
+				try {
+					em.editor.dispatch({ effects: clearDiagnosticsEffect() });
+					if (em.activeFile?.type === "editor") {
+						em.activeFile.session = em.editor.state;
+					}
+				} catch {
+					/* View may be disposed */
+				}
+			}
+
+			if (em?.files) {
+				for (const file of em.files) {
+					if (file?.type !== "editor" || file.id === em.activeFile?.id)
+						continue;
+					const session = file.session;
+					if (session && typeof session.update === "function") {
+						try {
+							file.session = session.update({
+								effects: clearDiagnosticsEffect(),
+							}).state;
+						} catch {
+							/* State update failed */
+						}
+					}
+				}
+			}
+		} catch {
+			/* Ignore errors */
+		}
+
 		const disposeOps: Promise<void>[] = [];
 		for (const [key, state] of this.#clients.entries()) {
 			disposeOps.push(state.dispose());

src/lib/editorManager.js

@@ -1123,6 +1123,16 @@ async function EditorManager($header, $body) {
 				events.emit(detailedEvent, ...detailedEventArgs);
 			}
 		},
+		/**
+		 * Restart LSP for the active file
+		 * Useful after stopping/restarting language servers
+		 */
+		restartLsp() {
+			const activeFile = manager.activeFile;
+			if (activeFile?.type === "editor") {
+				void configureLspForFile(activeFile);
+			}
+		},
 	};
 
 	if (typeof document !== "undefined") {