cwtools-shared 0.1.0

package/dist/tools/toolHandlers.js

@@ -0,0 +1,291 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.defaultSharedToolDispatcher = void 0;
+const path = __importStar(require("path"));
+const diagnosticRouting_1 = require("../knowledge/diagnosticRouting");
+const rules_1 = require("../knowledge/rules");
+const profile_1 = require("../project/profile");
+const knowledge_1 = require("../project/knowledge");
+const localisation_1 = require("../safety/localisation");
+const pdxBlock_1 = require("./pdxBlock");
+const schema_1 = require("./schema");
+const symbols_1 = require("./symbols");
+function asRecord(value) {
+    return value && typeof value === 'object' && !Array.isArray(value)
+        ? value
+        : {};
+}
+async function executeLspTool(host, command, args, unavailableNextStep) {
+    const result = await host.lsp.executeCommand(command, args);
+    const record = asRecord(result);
+    const errorRecord = asRecord(record.error);
+    const unavailable = record.status === 'unavailable' || errorRecord.code === 'lsp_unavailable';
+    return {
+        ok: !unavailable,
+        status: unavailable ? 'unavailable' : 'ready',
+        source: 'cwtools-lsp',
+        data: result,
+        error: unavailable
+            ? {
+                code: String(errorRecord.code ?? 'lsp_unavailable'),
+                message: String(errorRecord.message ?? 'LSP command is unavailable.'),
+            }
+            : undefined,
+        nextSteps: unavailable ? [unavailableNextStep] : undefined,
+    };
+}
+const defaultSharedToolDispatcher = async (host, name, rawArgs) => {
+    const args = asRecord(rawArgs);
+    switch (name) {
+        case 'query_project_profile':
+            return (0, profile_1.queryProjectProfileWithHost)(host, args);
+        case 'query_project_knowledge':
+            return (0, knowledge_1.queryProjectKnowledgeWithHost)(host, args);
+        case 'query_workspace_index': {
+            if (!host.indexing) {
+                return (0, schema_1.toolUnavailable)(name, 'Workspace index is not available in this host.', [
+                    'Wire an LSP index command or a thin Node index before using this tool for semantic evidence.',
+                ]);
+            }
+            const data = await host.indexing.queryWorkspace(args);
+            const status = data.status === 'ready'
+                ? 'ready'
+                : data.status === 'partial'
+                    ? 'partial'
+                    : data.status === 'error'
+                        ? 'error'
+                        : data.status === 'unavailable'
+                            ? 'unavailable'
+                            : 'loading';
+            return {
+                ok: true,
+                status,
+                source: 'cwtools-index',
+                data,
+            };
+        }
+        case 'explore_pdx_project':
+            return executeLspTool(host, 'cwtools.ai.exploreProject', [
+                typeof args.query === 'string' ? args.query : '',
+                typeof args.file === 'string' && args.file ? toFileUri(args.file, host.workspaceRoot) : '',
+                typeof args.typeName === 'string' ? args.typeName : '',
+                args.exact === true,
+                args.depth ?? 1,
+                args.maxNodes ?? 30,
+                args.maxEdges ?? 80,
+                args.includeMetadata !== false,
+            ], 'Start the CWTools LSP process and wait for the game model to load before exploring the semantic graph.');
+        case 'query_localisation_index': {
+            if (!host.indexing) {
+                return (0, schema_1.toolUnavailable)(name, 'Localisation index is not available in this host.', [
+                    'Wire an LSP index command or a thin Node index before using this tool for localisation evidence.',
+                ]);
+            }
+            return {
+                ok: true,
+                status: 'ready',
+                source: 'cwtools-index',
+                data: await host.indexing.queryLocalisation(args),
+            };
+        }
+        case 'get_diagnostics':
+            {
+                const diagnostics = await host.diagnostics.getDiagnostics(args);
+                return {
+                    ok: diagnostics.ok,
+                    status: diagnostics.status === 'unavailable' ? 'unavailable' : diagnostics.status === 'stale' ? 'stale' : diagnostics.status === 'pending' ? 'loading' : 'ready',
+                    source: 'cwtools-diagnostics',
+                    data: diagnostics,
+                    error: diagnostics.error,
+                };
+            }
+        case 'analyze_diagnostic_error':
+            return {
+                ok: true,
+                status: 'ready',
+                source: 'cwtools-knowledge',
+                data: (0, diagnosticRouting_1.analyzeDiagnosticKnowledge)({
+                    code: typeof args.code === 'string' ? args.code : undefined,
+                    message: typeof args.message === 'string' ? args.message : undefined,
+                }),
+            };
+        case 'write_localisation':
+            return (0, localisation_1.writeLocalisationWithHost)(host, {
+                filePath: String(args.filePath ?? args.file ?? ''),
+                language: typeof args.language === 'string' ? args.language : undefined,
+                entries: Array.isArray(args.entries) ? args.entries : [],
+            });
+        case 'get_completion_at':
+            return (0, symbols_1.getCompletionAtWithHost)(host, {
+                file: String(args.file ?? ''),
+                line: Number(args.line ?? 0),
+                column: Number(args.column ?? 0),
+                limit: typeof args.limit === 'number' ? args.limit : undefined,
+            });
+        case 'document_symbols':
+            return (0, symbols_1.documentSymbolsWithHost)(host, {
+                file: String(args.file ?? ''),
+            });
+        case 'workspace_symbols':
+            return (0, symbols_1.workspaceSymbolsWithHost)(host, {
+                query: String(args.query ?? ''),
+                limit: typeof args.limit === 'number' ? args.limit : undefined,
+            });
+        case 'query_definition':
+            return (0, symbols_1.queryDefinitionWithHost)(host, {
+                file: String(args.file ?? ''),
+                line: Number(args.line ?? 0),
+                column: Number(args.column ?? 0),
+            });
+        case 'query_definition_by_name':
+            return (0, symbols_1.queryDefinitionByNameWithHost)(host, {
+                symbolName: String(args.symbolName ?? ''),
+            });
+        case 'query_references':
+            return (0, symbols_1.queryReferencesWithHost)(host, {
+                identifier: String(args.identifier ?? ''),
+                file: typeof args.file === 'string' ? args.file : undefined,
+                limit: typeof args.limit === 'number' ? args.limit : undefined,
+            });
+        case 'query_scope':
+            return executeLspTool(host, 'cwtools.ai.getScopeAtPosition', [toFileUri(String(args.file ?? ''), host.workspaceRoot), args.line, args.column], 'Start the CWTools LSP process or connect this host to an existing language server.');
+        case 'query_types':
+            return executeLspTool(host, 'cwtools.ai.queryTypes', [args.typeName, args.filter ?? '', args.limit ?? 30, args.vanilla === true], 'Start the CWTools LSP process or connect this host to an existing language server.');
+        case 'query_rules':
+            return (0, rules_1.queryRulesWithHost)(host, {
+                category: String(args.category ?? 'trigger'),
+                name: typeof args.name === 'string' ? args.name : undefined,
+                scope: typeof args.scope === 'string' ? args.scope : undefined,
+            });
+        case 'query_cwt_schema':
+            return (0, rules_1.queryCwtSchemaWithHost)(host, {
+                target: typeof args.target === 'string' ? args.target : undefined,
+                file: typeof args.file === 'string' ? args.file : undefined,
+                directory: typeof args.directory === 'string' ? args.directory : undefined,
+                name: typeof args.name === 'string' ? args.name : undefined,
+                includeContent: args.includeContent === true,
+                limit: typeof args.limit === 'number' ? args.limit : undefined,
+            });
+        case 'search_rule_capabilities':
+            return (0, rules_1.searchRuleCapabilitiesWithHost)(host, {
+                intent: typeof args.intent === 'string' ? args.intent : undefined,
+                category: typeof args.category === 'string' ? args.category : undefined,
+                currentScope: typeof args.currentScope === 'string' ? args.currentScope : undefined,
+                desiredPushScope: typeof args.desiredPushScope === 'string' ? args.desiredPushScope : undefined,
+                limit: typeof args.limit === 'number' ? args.limit : undefined,
+            });
+        case 'explain_scope':
+            return (0, rules_1.explainScopeWithHost)(host, {
+                scope: String(args.scope ?? ''),
+            });
+        case 'parse_pdx_fragment':
+            return executeLspTool(host, 'cwtools.ai.parseFragment', [String(args.code ?? '')], 'Start the CWTools LSP process or connect this host to an existing language server.');
+        case 'query_scripted_effects':
+            return executeLspTool(host, 'cwtools.ai.queryScriptedEffects', [args.filter ?? '', args.limit ?? 50], 'Start the CWTools LSP process or connect this host to an existing language server.');
+        case 'query_scripted_triggers':
+            return executeLspTool(host, 'cwtools.ai.queryScriptedTriggers', [args.filter ?? '', args.limit ?? 50], 'Start the CWTools LSP process or connect this host to an existing language server.');
+        case 'query_enums':
+            return executeLspTool(host, 'cwtools.ai.queryEnums', [args.enumName ?? '', args.limit ?? 500], 'Start the CWTools LSP process or connect this host to an existing language server.');
+        case 'query_static_modifiers':
+            return executeLspTool(host, 'cwtools.ai.queryStaticModifiers', [args.filter ?? '', args.limit ?? 300], 'Start the CWTools LSP process or connect this host to an existing language server.');
+        case 'query_variables':
+            return executeLspTool(host, 'cwtools.ai.queryVariables', [args.filter ?? ''], 'Start the CWTools LSP process or connect this host to an existing language server.');
+        case 'get_entity_info':
+            return executeLspTool(host, 'cwtools.ai.getEntityInfo', [toFileUri(String(args.file ?? ''), host.workspaceRoot)], 'Start the CWTools LSP process or connect this host to an existing language server.');
+        case 'get_pdx_block':
+            return (0, pdxBlock_1.getPdxBlockWithHost)(host, {
+                file: String(args.file ?? ''),
+                symbol: String(args.symbol ?? ''),
+            });
+        // Shader knowledge tools: each LSP command takes a single JSON record
+        // argument and returns { ok, ... } / { ok: false, status: 'error', ... }.
+        case 'query_shader_symbol':
+            return executeLspTool(host, 'cwtools.ai.shader.symbols', [shaderRecord({
+                    filter: typeof args.filter === 'string' ? args.filter : undefined,
+                    kind: typeof args.kind === 'string' ? args.kind : undefined,
+                    limit: typeof args.limit === 'number' ? args.limit : undefined,
+                    cursor: typeof args.cursor === 'number' ? args.cursor : undefined,
+                })], 'Start the CWTools LSP process and wait for the shader model to load before querying shader symbols.');
+        case 'query_shader_compile_unit':
+            return executeLspTool(host, 'cwtools.ai.shader.compileUnit', [{ uri: toFileUri(String(args.file ?? ''), host.workspaceRoot) }], 'Start the CWTools LSP process and wait for the shader model to load before resolving compile units.');
+        case 'query_shader_platform_variants':
+            return executeLspTool(host, 'cwtools.ai.shader.variants', [{ uri: toFileUri(String(args.file ?? ''), host.workspaceRoot) }], 'Start the CWTools LSP process and wait for the shader model to load before comparing shader platform variants.');
+        case 'query_shader_callers':
+            return executeLspTool(host, 'cwtools.ai.shader.callers', [shaderRecord({
+                    effectName: String(args.effectName ?? ''),
+                    limit: typeof args.limit === 'number' ? args.limit : undefined,
+                })], 'Start the CWTools LSP process and wait for the shader model to load before querying Effect callers.');
+        case 'explain_shader_reachability':
+            return executeLspTool(host, 'cwtools.ai.shader.reachability', [typeof args.effectName === 'string' && args.effectName
+                    ? { effectName: args.effectName }
+                    : shaderRecord({
+                        uri: toFileUri(String(args.file ?? ''), host.workspaceRoot),
+                        limit: typeof args.limit === 'number' ? args.limit : undefined,
+                        cursor: typeof args.cursor === 'number' ? args.cursor : undefined,
+                    })], 'Start the CWTools LSP process and wait for the shader model to load before explaining Effect reachability.');
+        case 'validate_shader':
+            return executeLspTool(host, 'cwtools.ai.shader.validate', [{ uri: toFileUri(String(args.file ?? ''), host.workspaceRoot) }], 'Start the CWTools LSP process and wait for the shader model to load before validating shader files.');
+        case 'compare_shader_with_vanilla':
+            return executeLspTool(host, 'cwtools.ai.shader.compareVanilla', [typeof args.effectName === 'string' && args.effectName
+                    ? { effectName: args.effectName }
+                    : { uri: toFileUri(String(args.file ?? ''), host.workspaceRoot) }], 'Start the CWTools LSP process and wait for the shader model to load before comparing with vanilla.');
+        default:
+            return {
+                ok: false,
+                status: 'error',
+                source: 'cwtools-shared',
+                error: {
+                    code: 'unknown_tool',
+                    message: `Unknown MCP tool: ${name}`,
+                },
+            };
+    }
+};
+exports.defaultSharedToolDispatcher = defaultSharedToolDispatcher;
+function toFileUri(filePath, workspaceRoot) {
+    const resolved = path.resolve(workspaceRoot, filePath).replace(/\\/g, '/');
+    const withLeadingSlash = resolved.startsWith('/') ? resolved : `/${resolved}`;
+    return `file://${encodeURI(withLeadingSlash).replace(/#/g, '%23')}`;
+}
+/** Drop undefined fields so the LSP command record only carries provided args. */
+function shaderRecord(fields) {
+    const record = {};
+    for (const [key, value] of Object.entries(fields)) {
+        if (value !== undefined)
+            record[key] = value;
+    }
+    return record;
+}

package/package.json

@@ -0,0 +1,24 @@
+{
+  "name": "cwtools-shared",
+  "version": "0.1.0",
+  "description": "Shared protocol and tool contracts for the CWTools MCP server",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/Aa728848/cwtools-mcp.git"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "package.json"
+  ],
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "prepack": "npm run build",
+    "test:contracts": "ts-mocha -p tsconfig.test.json \"src/test/**/*.test.ts\""
+  }
+}