Nx MCP Server

import { isExecutorStringNode, lspLogger, } from '@nx-console/language-server-utils'; import { gte } from '@nx-console/nx-version'; import { importNxPackagePath } from '@nx-console/shared-npm'; import { getExecutors, nxWorkspace, } from '@nx-console/shared-nx-workspace-info'; import { getNxVersion } from '@nx-console/shared-nx-workspace-info'; import type { ProjectConfiguration } from 'nx/src/devkit-exports'; import { dirname } from 'path'; import { JSONDocument } from 'vscode-json-languageservice'; import { DefinitionParams, LocationLink } from 'vscode-languageserver'; import { TextDocument } from 'vscode-languageserver-textdocument'; import { URI } from 'vscode-uri'; export async function getDefinition( workingPath: string, definitionParams: DefinitionParams, jsonAst: JSONDocument, document: TextDocument, ): Promise<LocationLink[] | undefined> { const offset = document.offsetAt(definitionParams.position); const node = jsonAst.getNodeFromOffset(offset); if (!node || !isExecutorStringNode(node)) { return undefined; } const executors = await getExecutors(workingPath); const executor = executors.find((e) => e.name === node.value); if (!executor) { return undefined; } const nxVersion = await getNxVersion(workingPath); const { resolveImplementation } = await importNxPackagePath< typeof import('nx/src/config/schema-utils') >(workingPath, 'src/config/schema-utils', lspLogger); let executorFile: string; if (gte(nxVersion, '20.3.0')) { const projectGraph = (await nxWorkspace(workingPath, lspLogger)) .projectGraph; const projects = Object.entries(projectGraph.nodes).reduce( (acc, [projectName, project]) => { acc[projectName] = project.data; return acc; }, {} as Record<string, ProjectConfiguration>, ); executorFile = resolveImplementation( executor.implementationPath, dirname(executor.configPath), executor.collectionName, projects, ); } else { executorFile = (resolveImplementation as any)( executor.implementationPath, dirname(executor.configPath), ) as string; } return [ LocationLink.create( URI.file(executorFile).toString(), // Link to start of file because we cannot find the exact location without parsing the file { start: { line: 0, character: 0, }, end: { line: 0, character: 0, }, }, { start: { line: 0, character: 0, }, end: { line: 0, character: 0, }, }, { // So that the underline is // "executor": "nx:build" // ^^^^^^^^ start: document.positionAt(node.offset + 1), end: document.positionAt(node.offset + node.length - 1), }, ), ]; }