@mseep/webgl-mcp/src/index.js

A Model Context Protocol server for analyzing and optimizing WebGL games

"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const mcp_js_1 = require("@modelcontextprotocol/sdk/server/mcp.js");
const stdio_js_1 = require("@modelcontextprotocol/sdk/server/stdio.js");
const zod_1 = require("zod");
const webgl_context_1 = require("./webgl-context");
const winston_1 = __importDefault(require("winston"));
// Configure logger
const logger = winston_1.default.createLogger({
    level: 'info',
    format: winston_1.default.format.combine(winston_1.default.format.timestamp(), winston_1.default.format.json()),
    transports: [
        new winston_1.default.transports.File({ filename: 'error.log', level: 'error' }),
        new winston_1.default.transports.File({ filename: 'combined.log' })
    ]
});
if (process.env.NODE_ENV !== 'production') {
    logger.add(new winston_1.default.transports.Console({
        format: winston_1.default.format.simple()
    }));
}
// Error types
class WebGLError extends Error {
    constructor(message, details) {
        super(message);
        this.details = details;
        this.name = 'WebGLError';
    }
}
// Initialize WebGL context manager
const contextManager = new webgl_context_1.WebGLContextManager();
// Create an MCP server
const server = new mcp_js_1.McpServer({
    name: "WebGL-MCP",
    version: "1.0.0"
});
// Add WebGL analysis tool
server.tool("analyze-webgl", {
    path: zod_1.z.string().describe("Path to WebGL build folder or index.html file")
}, (_a) => __awaiter(void 0, [_a], void 0, function* ({ path }) {
    logger.info(`Analyzing WebGL at path: ${path}`);
    try {
        // Here you would add your WebGL analysis logic
        // For now, we'll just return a simple message
        return {
            content: [{
                    type: "text",
                    text: `Analyzed WebGL at path: ${path}\n\nFindings:\n- WebGL 2.0 support: Yes\n- Texture compression: Yes\n- Memory usage: Optimal\n- Render pipeline: Efficient`
                }]
        };
    }
    catch (error) {
        logger.error('Error analyzing WebGL:', {
            error: error instanceof Error ? error.message : String(error),
            path
        });
        throw new WebGLError(`Failed to analyze WebGL at path: ${path}`, {
            cause: error instanceof Error ? error.message : String(error)
        });
    }
}));
// Add WebGL optimization tool
server.tool("optimize-webgl", {
    path: zod_1.z.string().describe("Path to WebGL build folder or index.html file"),
    targetFPS: zod_1.z.number().optional().describe("Target frames per second"),
    memoryLimit: zod_1.z.number().optional().describe("Memory limit in MB"),
    optimizationGoals: zod_1.z.array(zod_1.z.enum(['performance', 'memory', 'quality', 'mobile'])).optional()
        .describe("Optimization goals in order of priority")
}, (_b) => __awaiter(void 0, [_b], void 0, function* ({ path, targetFPS, memoryLimit, optimizationGoals }) {
    logger.info(`Optimizing WebGL at path: ${path}`, { targetFPS, memoryLimit, optimizationGoals });
    try {
        // Here you would add your WebGL optimization logic
        // For now, we'll just return a simple message
        return {
            content: [{
                    type: "text",
                    text: `Optimized WebGL at path: ${path}\n\nOptimizations applied:\n- Texture compression\n- Shader minification\n- Draw call batching\n- Memory management improvements`
                }]
        };
    }
    catch (error) {
        logger.error('Error optimizing WebGL:', {
            error: error instanceof Error ? error.message : String(error),
            path
        });
        throw new WebGLError(`Failed to optimize WebGL at path: ${path}`, {
            cause: error instanceof Error ? error.message : String(error)
        });
    }
}));
// Start receiving messages on stdin and sending messages on stdout
function startServer() {
    return __awaiter(this, void 0, void 0, function* () {
        try {
            const transport = new stdio_js_1.StdioServerTransport();
            yield server.connect(transport);
            logger.info('MCP Server started and connected to transport');
        }
        catch (error) {
            logger.error('Failed to start server:', {
                error: error instanceof Error ? error.message : String(error)
            });
            process.exit(1);
        }
    });
}
// Start the server
startServer();