@mseep/atlas-mcp-server/dist/mcp/server.js

A Model Context Protocol (MCP) server for ATLAS, a Neo4j-powered task management system for LLM Agents - implementing a three-tier architecture (Projects, Tasks, Knowledge) to manage complex workflows.

#!/usr/bin/env node
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import { config } from "../config/index.js";
import { BaseErrorCode, McpError } from "../types/errors.js";
import { logger } from "../utils/logger.js";
import { configureSecurity } from "../utils/security.js";
import { initializeNeo4jSchema } from "../services/neo4j/index.js"; // Changed import
// Import tool registrations
import { registerAtlasProjectCreateTool } from "./tools/atlas_project_create/index.js";
import { registerAtlasProjectDeleteTool } from "./tools/atlas_project_delete/index.js";
import { registerAtlasProjectListTool } from "./tools/atlas_project_list/index.js";
import { registerAtlasProjectUpdateTool } from "./tools/atlas_project_update/index.js";
import { registerAtlasTaskCreateTool } from "./tools/atlas_task_create/index.js";
import { registerAtlasTaskDeleteTool } from "./tools/atlas_task_delete/index.js";
import { registerAtlasTaskListTool } from "./tools/atlas_task_list/index.js";
import { registerAtlasTaskUpdateTool } from "./tools/atlas_task_update/index.js";
import { registerAtlasDatabaseCleanTool } from "./tools/atlas_database_clean/index.js";
import { registerAtlasKnowledgeAddTool } from "./tools/atlas_knowledge_add/index.js";
import { registerAtlasKnowledgeDeleteTool } from "./tools/atlas_knowledge_delete/index.js";
import { registerAtlasKnowledgeListTool } from "./tools/atlas_knowledge_list/index.js";
import { registerAtlasUnifiedSearchTool } from "./tools/atlas_unified_search/index.js";
import { registerAtlasDeepResearchTool } from "./tools/atlas_deep_research/index.js"; // Import the new tool registration
// Import resource registrations
import { registerMcpResources } from "./resources/index.js";
export const createMcpServer = async () => {
    try {
        // Configure security settings
        configureSecurity({
            authRequired: config.security.authRequired
        });
        // Initialize Neo4j database and services
        logger.info("Initializing Neo4j schema..."); // Updated log message
        await initializeNeo4jSchema(); // Changed function call
        logger.info("Neo4j schema initialized successfully"); // Updated log message
        const server = new McpServer({
            name: config.mcpServerName,
            version: config.mcpServerVersion,
            capabilities: {
                resources: {},
                tools: {
                    // Define global tool capabilities
                    requestContext: true, // Enable request context for all tools
                    rateLimit: {
                        windowMs: 60 * 1000, // 1 minute default window
                        maxRequests: 100 // 100 requests per minute default
                    },
                    permissions: {
                        required: config.security.authRequired // Make permissions optional based on security config
                    }
                }
            }
        });
        // Register tools
        registerAtlasProjectCreateTool(server); // atlas_project_create
        registerAtlasProjectListTool(server); // atlas_project_list
        registerAtlasProjectUpdateTool(server); // atlas_project_update
        registerAtlasProjectDeleteTool(server); // atlas_project_delete
        registerAtlasTaskCreateTool(server); // atlas_task_create
        registerAtlasTaskDeleteTool(server); // atlas_task_delete
        registerAtlasTaskListTool(server); // atlas_task_list
        registerAtlasTaskUpdateTool(server); // atlas_task_update
        registerAtlasDatabaseCleanTool(server); // atlas_database_clean
        registerAtlasKnowledgeAddTool(server); // atlas_knowledge_add
        registerAtlasKnowledgeDeleteTool(server); // atlas_knowledge_delete
        registerAtlasKnowledgeListTool(server); // atlas_knowledge_list
        registerAtlasUnifiedSearchTool(server); // atlas_unified_search
        registerAtlasDeepResearchTool(server); // atlas_deep_research (Register the new tool)
        // Register resources
        registerMcpResources(server);
        // Connect using stdio transport
        const transport = new StdioServerTransport();
        await server.connect(transport);
        logger.info("ATLAS MCP Server running on stdio", {
            name: config.mcpServerName,
            version: config.mcpServerVersion,
            authRequired: config.security.authRequired
        });
        return server;
    }
    catch (error) {
        // Handle initialization errors
        logger.error("Failed to initialize MCP server", {
            error: error instanceof Error ? error.message : 'Unknown error'
        });
        throw new McpError(BaseErrorCode.INTERNAL_ERROR, `Failed to initialize MCP server: ${error instanceof Error ? error.message : 'Unknown error'}`);
    }
};