imcp 0.1.5 → 0.1.6

package/src/core/loaders/ConfigurationLoader.ts

@@ -0,0 +1,298 @@
+import fs from 'fs/promises';
+import path from 'path';
+import { LOCAL_FEEDS_DIR, SUPPORTED_CLIENTS } from '../metadatas/constants.js';
+import { Logger } from '../../utils/logger.js';
+import { readJsonFile } from '../../utils/clientUtils.js';
+import {
+    MCPConfiguration,
+    MCPServerCategory,
+    MCPServerStatus,
+    OperationStatus,
+    FeedConfiguration,
+    RequirementStatus,
+    InstallationStatus
+} from '../metadatas/types.js';
+
+export class ConfigurationLoader {
+    /**
+     * Updates the installed status for a server and client combination
+     */
+    private static updateServerInstalledStatus(
+        serverStatus: MCPServerStatus,
+        clientName: string,
+        operationStatus: OperationStatus
+    ): void {
+        if (!serverStatus.installedStatus) {
+            serverStatus.installedStatus = {};
+        }
+        serverStatus.installedStatus[clientName] = operationStatus;
+    }
+
+    /**
+     * Removes a client's status from a server
+     */
+    private static removeClientStatus(
+        serverStatus: MCPServerStatus,
+        clientName: string
+    ): void {
+        if (serverStatus.installedStatus && serverStatus.installedStatus[clientName]) {
+            delete serverStatus.installedStatus[clientName];
+        }
+    }
+
+    /**
+     * Synchronizes server categories with client MCP settings.
+     * Uses clientMCPSettings as source of truth for installation status.
+     */
+    static syncServerCategoriesWithClientSettings(configuration: MCPConfiguration): MCPConfiguration {
+        if (!configuration.clientMCPSettings) {
+            return configuration;
+        }
+
+        configuration.localServerCategories = configuration.localServerCategories.map(category => {
+            if (!category.installationStatus?.serversStatus) {
+                return category;
+            }
+
+            const updatedServerStatus = { ...category.installationStatus.serversStatus };
+            const clientSettings = configuration.clientMCPSettings as Record<string, Record<string, any>>;
+
+            for (const [clientName, settings] of Object.entries(clientSettings)) {
+                const clientServers = clientName === 'GithubCopilot'
+                    ? settings.servers || {}
+                    : settings.mcpServers || {};
+
+                Object.keys(updatedServerStatus).forEach(serverName => {
+                    if (clientServers[serverName]) {
+                        if (!updatedServerStatus[serverName].installedStatus[clientName]) {
+                            const operationStatus: OperationStatus = {
+                                status: 'completed',
+                                type: 'install',
+                                target: 'server',
+                                message: `Server ${serverName} is configured for client ${clientName}`
+                            };
+                            ConfigurationLoader.updateServerInstalledStatus(updatedServerStatus[serverName], clientName, operationStatus);
+                        }
+                    } else {
+                        ConfigurationLoader.removeClientStatus(updatedServerStatus[serverName], clientName);
+                    }
+                });
+            }
+
+            return {
+                ...category,
+                installationStatus: {
+                    ...category.installationStatus,
+                    serversStatus: updatedServerStatus,
+                    lastUpdated: new Date().toISOString()
+                }
+            };
+        });
+
+        return configuration;
+    }
+
+    /**
+     * Initializes installation status for a feed configuration
+     */
+    private static initializeInstallationStatus(feedConfig?: FeedConfiguration): InstallationStatus {
+        const requirementsStatus: Record<string, RequirementStatus> = {};
+        const serversStatus: Record<string, MCPServerStatus> = {};
+
+        if (feedConfig) {
+            if (feedConfig.requirements) {
+                for (const req of feedConfig.requirements) {
+                    requirementsStatus[req.name] = {
+                        name: req.name,
+                        type: req.type,
+                        installed: false,
+                        version: req.version,
+                        error: undefined
+                    };
+                }
+            }
+            if (feedConfig.mcpServers) {
+                for (const mcp of feedConfig.mcpServers) {
+                    serversStatus[mcp.name] = {
+                        name: mcp.name,
+                        error: undefined,
+                        installedStatus: {}
+                    };
+                }
+            }
+        }
+
+        return {
+            requirementsStatus,
+            serversStatus,
+            lastUpdated: new Date().toISOString()
+        };
+    }
+
+    /**
+     * Synchronizes server categories with feeds
+     */
+    private static async syncServerCategoriesWithFeeds(configuration: MCPConfiguration): Promise<MCPConfiguration> {
+        // Filter out categories that don't have corresponding feeds and update existing ones
+        configuration.localServerCategories = configuration.localServerCategories
+            .filter(server => configuration.feeds[server.name])
+            .map(server => {
+                server.feedConfiguration = configuration.feeds[server.name];
+
+                if (
+                    !server.installationStatus ||
+                    !server.installationStatus.requirementsStatus ||
+                    Object.keys(server.installationStatus.requirementsStatus).length === 0 ||
+                    !server.installationStatus.serversStatus ||
+                    Object.keys(server.installationStatus.serversStatus).length < Object.keys(server.feedConfiguration?.mcpServers || []).length
+                ) {
+                    server.installationStatus = ConfigurationLoader.initializeInstallationStatus(server.feedConfiguration);
+                }
+
+                return server;
+            });
+
+        // Add new categories for feeds that don't have a corresponding category
+        const existingServerNames = new Set(configuration.localServerCategories.map(category => category.name));
+
+        for (const [feedName, feedConfig] of Object.entries(configuration.feeds)) {
+            if (!existingServerNames.has(feedName)) {
+                const newServerCategory: MCPServerCategory = {
+                    name: feedName,
+                    displayName: feedConfig.displayName || feedName,
+                    type: 'local',
+                    description: feedConfig.description || `Local MCP server category: ${feedName}`,
+                    installationStatus: ConfigurationLoader.initializeInstallationStatus(feedConfig),
+                    feedConfiguration: feedConfig
+                };
+
+                configuration.localServerCategories.push(newServerCategory);
+                console.log(`Created new local server entry for feed: ${feedName}`);
+            }
+        }
+
+        return configuration;
+    }
+
+    /**
+     * Loads feed configurations into the MCP configuration
+     */
+    static async loadFeedsIntoConfiguration(configuration: MCPConfiguration, feedFile?: string, settings?: { prLink?:string, adhocServers?: string[] }): Promise<MCPConfiguration> {
+        try {
+            await fs.mkdir(LOCAL_FEEDS_DIR, { recursive: true });
+            const feeds: Record<string, FeedConfiguration> = {};
+
+            // Load provided feed file if specified
+            if (feedFile) {
+                try {
+                    const content = await fs.readFile(feedFile, 'utf8');
+                    let config = JSON.parse(content) as FeedConfiguration;
+                    if (config && config.name) {
+                        // Update systemTags if feedFile is provided
+                        if (!config.systemTags) {
+                            config.systemTags = {};
+                        }
+                        config.systemTags['adhoc'] = 'true';
+
+                        // Update PullRequest if prLink is provided in settings
+                        if (settings?.prLink) {
+                            config.PullRequest = settings.prLink;
+                        }
+
+                        // Update mcpServers systemTags if adhocServers are provided in settings
+                        if (settings?.adhocServers && settings.adhocServers.length > 0 && config.mcpServers) {
+                            config.mcpServers = config.mcpServers.map(server => {
+                                if (settings.adhocServers!.includes(server.name)) {
+                                    if (!server.systemTags) {
+                                        server.systemTags = {};
+                                    }
+                                    server.systemTags['adhoc'] = 'true';
+                                }
+                                return server;
+                            });
+                        }
+
+                        feeds[config.name] = config;
+                        console.log(`Loaded and processed feed configuration from provided file: ${feedFile}`);
+                    }
+                } catch (error) {
+                    console.log(`Error loading feed configuration from provided file ${feedFile}:`, error);
+                }
+            }
+
+            // Load feeds from LOCAL_FEEDS_DIR
+            const files = await fs.readdir(LOCAL_FEEDS_DIR);
+            const jsonFiles = files.filter(file => file.endsWith('.json'));
+
+            if (jsonFiles.length === 0 && !feedFile) {
+                console.log(`No feed configuration files found in ${LOCAL_FEEDS_DIR}`);
+                return configuration;
+            }
+
+            for (const file of jsonFiles) {
+                try {
+                    const filePath = path.join(LOCAL_FEEDS_DIR, file);
+                    const content = await fs.readFile(filePath, 'utf8');
+                    const config = JSON.parse(content) as FeedConfiguration;
+                    if (config && config.name) {
+                        // If feed exists from provided file, skip the local one
+                        if (!feeds[config.name]) {
+                            feeds[config.name] = config;
+                        } else {
+                            console.log(`Skipping local feed ${config.name} as it was provided via --feed-file`);
+                        }
+                    }
+                } catch (error) {
+                    console.warn(`Error loading feed configuration from ${file}:`, error);
+                }
+            }
+
+            configuration.feeds = feeds;
+            return await ConfigurationLoader.syncServerCategoriesWithFeeds(configuration);
+        } catch (error) {
+            console.error("Error loading feed configurations:", error);
+            throw error;
+        }
+    }
+
+    /**
+     * Loads MCP client settings into the configuration
+     */
+    static async loadClientMCPSettings(configuration: MCPConfiguration): Promise<MCPConfiguration> {
+        try {
+            Logger.debug('Starting to load MCP client settings...');
+            const settings: Record<string, Record<string, any>> = {};
+
+            for (const [clientName, clientSettings] of Object.entries(SUPPORTED_CLIENTS)) {
+                const settingPath = process.env.CODE_INSIDERS
+                    ? clientSettings.codeInsiderSettingPath
+                    : clientSettings.codeSettingPath;
+                try {
+                    let content = await readJsonFile(settingPath, true);
+
+                    if (clientName === 'GithubCopilot') {
+                        if (!content.mcp) {
+                            content = {
+                                servers: {},
+                                inputs: []
+                            };
+                        } else {
+                            content = content.mcp;
+                        }
+                    }
+                    settings[clientName] = content;
+                    Logger.debug(`Successfully loaded MCP settings for ${clientName}`);
+                } catch (error) {
+                    Logger.debug(`Warning: Could not load MCP settings for client ${clientName}: ${error instanceof Error ? error.message : String(error)}`);
+                    settings[clientName] = {};
+                }
+            }
+
+            configuration.clientMCPSettings = settings;
+            return ConfigurationLoader.syncServerCategoriesWithClientSettings(configuration);
+        } catch (error) {
+            Logger.error('Error loading client MCP settings:', error);
+            throw error;
+        }
+    }
+}