symbols-mcp

/** * Store implementations for caching LSP data */ import { Diagnostic } from 'vscode-languageserver-protocol'; import { DiagnosticsStore, DiagnosticProviderStore, DiagnosticProvider, WindowLogStore, LogMessage, WorkspaceLoaderStore, } from '../types.js'; import { WorkspaceLoaderState, WorkspaceLoader } from '../workspace/types.js'; import logger from '../utils/logger.js'; export function createDiagnosticsStore(): DiagnosticsStore { const diagnostics = new Map<string, Diagnostic[]>(); return { diagnostics, addDiagnostics(uri: string, diagnostics: Diagnostic[]) { this.diagnostics.set(uri, diagnostics); }, getDiagnostics(uri: string): Diagnostic[] { return this.diagnostics.get(uri) || []; }, clear() { this.diagnostics.clear(); }, }; } export function createDiagnosticProviderStore(): DiagnosticProviderStore { const providers: DiagnosticProvider[] = []; return { providers, addProvider(provider: DiagnosticProvider) { // Avoid duplicates based on id const existingIndex = this.providers.findIndex( (p) => p.id === provider.id ); if (existingIndex >= 0) { this.providers[existingIndex] = provider; } else { this.providers.push(provider); } }, getProviders(): DiagnosticProvider[] { return [...this.providers]; }, getProvidersForDocument( uri: string, languageId?: string ): DiagnosticProvider[] { const filteredProviders = this.providers.filter((provider) => { if (!provider.documentSelector) return true; return provider.documentSelector.some((selector) => { // Check language match if ( selector.language && languageId && selector.language !== languageId ) { return false; } // Check scheme match (assuming file:// scheme) if (selector.scheme && !uri.startsWith(`${selector.scheme}:`)) { return false; } // Check pattern match (simplified glob matching) if (selector.pattern) { const regex = new RegExp(selector.pattern.replace(/\*/g, '.*')); if (!regex.test(uri)) { return false; } } return true; }); }); logger.debug('Retrieving providers for document', { uri, languageId, providerCount: filteredProviders.length, }); return filteredProviders; }, clear() { this.providers.length = 0; }, }; } export function createWindowLogStore(): WindowLogStore { const messages: LogMessage[] = []; return { messages, addMessage(message: LogMessage) { logger.info('[LSP-LOG]', message); this.messages.push(message); }, getMessages(): LogMessage[] { return [...this.messages]; }, clear() { this.messages.length = 0; }, }; } export function createWorkspaceLoaderStore(): WorkspaceLoaderStore { let state: WorkspaceLoaderState | null = null; let loader: WorkspaceLoader | null = null; return { state, loader, setState(newState: WorkspaceLoaderState) { state = newState; this.state = newState; }, setLoader(newLoader: WorkspaceLoader) { loader = newLoader; this.loader = newLoader; }, getState(): WorkspaceLoaderState | null { return state; }, getLoader(): WorkspaceLoader | null { return loader; }, updateState(method: string) { if (state && loader?.handleNotification) { const newState = loader.handleNotification(state, method); this.setState(newState); } }, isReady(): boolean { if (!state || !loader) { return false; } return loader.isReady(state); }, }; }