imcp 0.1.7 → 0.1.8-dev

package/src/core/loaders/ConfigurationProvider.ts

@@ -1,462 +0,0 @@
-import fs from 'fs/promises';
-import path from 'path';
-import { exec } from 'child_process';
-import { promisify } from 'util';
-import { fileURLToPath } from 'url';
-import { GITHUB_REPO, LOCAL_FEEDS_DIR, SETTINGS_DIR, SUPPORTED_CLIENTS } from '../metadatas/constants.js';
-import { Logger } from '../../utils/logger.js';
-import { checkGithubAuth } from '../../utils/githubAuth.js';
-import {
-    MCPConfiguration,
-    MCPServerCategory,
-    FeedConfiguration,
-    InstallationStatus,
-    RequirementStatus,
-    MCPServerStatus,
-    OperationStatus,
-    ClientSettings,
-    McpConfig
-} from '../metadatas/types.js';
-import { ConfigurationLoader } from './ConfigurationLoader.js';
-
-const execAsync = promisify(exec);
-const __dirname = path.dirname(fileURLToPath(import.meta.url));
-
-export class ConfigurationProvider {
-    private static instance: ConfigurationProvider;
-    private configPath: string;
-    private configuration: MCPConfiguration;
-    private configLock: Promise<void> = Promise.resolve();
-    private tempDir: string;
-
-    private constructor() {
-        // Initialize configuration in user's appdata/imcp directory
-        this.configPath = path.join(SETTINGS_DIR, 'configurations.json');
-        this.configuration = {
-            localServerCategories: [],
-            feeds: {},
-            clientMCPSettings: {}
-        };
-        this.tempDir = path.join(LOCAL_FEEDS_DIR, '../temp');
-    }
-
-    public static getInstance(): ConfigurationProvider {
-        if (!ConfigurationProvider.instance) {
-            ConfigurationProvider.instance = new ConfigurationProvider();
-        }
-        return ConfigurationProvider.instance;
-    }
-
-    private async withLock<T>(operation: () => Promise<T>): Promise<T> {
-        const current = this.configLock;
-        let resolve: () => void;
-        this.configLock = new Promise<void>(r => resolve = r);
-        try {
-            await current;
-            return await operation();
-        } finally {
-            resolve!();
-        }
-    }
-
-    async initialize(feedFile?: string, settings?: { prLink?:string, adhocServers?: string[] }): Promise<void> {
-        await this.withLock(async () => {
-            const configDir = path.dirname(this.configPath);
-            await fs.mkdir(configDir, { recursive: true });
-            // remove the old configuration file if it exists
-            try {
-                await fs.rm(this.configPath, { recursive: true, force: true });
-            } catch (error) {
-                if ((error as NodeJS.ErrnoException).code !== 'ENOENT') {
-                    throw error;
-                }
-            }
-
-            try {
-                try {
-                    const config = JSON.parse(await fs.readFile(this.configPath, 'utf8'));
-                    this.configuration = config;
-                } catch (error) {
-                    if ((error as NodeJS.ErrnoException).code !== 'ENOENT') {
-                        throw error;
-                    }
-                    // File doesn't exist, use default empty configuration
-                    await this.saveConfiguration();
-                }
-
-                // Always load feeds and client settings, whether file existed or not
-                await this.loadFeedsIntoConfiguration(feedFile, settings);
-                await this.loadClientMCPSettings();
-            } catch (error) {
-                Logger.error('Error during initialization', error);
-                throw error;
-            }
-        });
-    }
-
-    private async saveConfiguration(): Promise<void> {
-        const configDir = path.dirname(this.configPath);
-        await fs.mkdir(configDir, { recursive: true });
-        await fs.writeFile(this.configPath, JSON.stringify(this.configuration, null, 2));
-    }
-
-    async getServerCategories(): Promise<MCPServerCategory[]> {
-        return await this.withLock(async () => {
-            return this.configuration.localServerCategories;
-        });
-    }
-
-    async getServerCategory(categoryName: string): Promise<MCPServerCategory | undefined> {
-        return await this.withLock(async () => {
-            return this.configuration.localServerCategories.find(s => s.name === categoryName);
-        });
-    }
-
-    async getClientMcpSettings(): Promise<Record<string, Record<string, any>> | undefined> {
-        return await this.withLock(async () => {
-            return this.configuration.clientMCPSettings;
-        });
-    }
-
-    async getFeedConfiguration(categoryName: string): Promise<FeedConfiguration | undefined> {
-        return await this.withLock(async () => {
-            return this.configuration.feeds[categoryName];
-        });
-    }
-
-    async getServerMcpConfig(categoryName: string, serverName: string): Promise<McpConfig | undefined> {
-        return await this.withLock(async () => {
-            return this.configuration.feeds[categoryName]?.mcpServers.find(s => s.name === serverName);
-        });
-    }
-
-    async getInstallationStatus(categoryName: string): Promise<InstallationStatus | undefined> {
-        return await this.withLock(async () => {
-            // Inline getServerCategory logic
-            const category = this.configuration.localServerCategories.find(s => s.name === categoryName);
-            return category?.installationStatus;
-        });
-    }
-
-    async getServerStatus(categoryName: string, serverName: string): Promise<MCPServerStatus | undefined> {
-        return await this.withLock(async () => {
-            // Inline getInstallationStatus logic
-            const category = this.configuration.localServerCategories.find(s => s.name === categoryName);
-            const status = category?.installationStatus;
-            return status?.serversStatus[serverName];
-        });
-    }
-
-    async getRequirementStatus(categoryName: string, requirementName: string): Promise<RequirementStatus | undefined> {
-        return await this.withLock(async () => {
-            // Inline getInstallationStatus logic
-            const category = this.configuration.localServerCategories.find(s => s.name === categoryName);
-            const status = category?.installationStatus;
-            return status?.requirementsStatus[requirementName];
-        });
-    }
-
-    async updateInstallationStatus(
-        categoryName: string,
-        requirementStatus: Record<string, RequirementStatus>,
-        serverStatus: Record<string, MCPServerStatus>
-    ): Promise<boolean> {
-        return await this.withLock(async () => {
-            // Inline getServerCategory logic
-            const category = this.configuration.localServerCategories.find(s => s.name === categoryName);
-            if (!category) return false;
-
-            if (!category.installationStatus) {
-                category.installationStatus = {
-                    requirementsStatus: {},
-                    serversStatus: {},
-                    lastUpdated: new Date().toISOString()
-                };
-            }
-
-            category.installationStatus.requirementsStatus = {
-                ...category.installationStatus.requirementsStatus,
-                ...requirementStatus
-            };
-
-            category.installationStatus.serversStatus = {
-                ...category.installationStatus.serversStatus,
-                ...serverStatus
-            };
-
-            category.installationStatus.lastUpdated = new Date().toISOString();
-            await this.saveConfiguration();
-            return true;
-        });
-    }
-
-    async updateRequirementStatus(
-        categoryName: string,
-        requirementName: string,
-        status: RequirementStatus
-    ): Promise<boolean> {
-        return await this.withLock(async () => {
-            // Inline getServerCategory logic
-            const category = this.configuration.localServerCategories.find(s => s.name === categoryName);
-            if (!category?.installationStatus) return false;
-
-            category.installationStatus.requirementsStatus[requirementName] = status;
-            category.installationStatus.lastUpdated = new Date().toISOString();
-            await this.saveConfiguration();
-            return true;
-        });
-    }
-
-    async updateRequirementOperationStatus(
-        categoryName: string,
-        requirementName: string,
-        operationStatus: OperationStatus
-    ): Promise<boolean> {
-        return await this.withLock(async () => {
-            // Inline getServerCategory logic
-            const category = this.configuration.localServerCategories.find(s => s.name === categoryName);
-            if (!category?.installationStatus?.requirementsStatus[requirementName]) return false;
-
-            category.installationStatus.requirementsStatus[requirementName].operationStatus = operationStatus;
-            category.installationStatus.lastUpdated = new Date().toISOString();
-            await this.saveConfiguration();
-            return true;
-        });
-    }
-
-    async updateServerStatus(
-        categoryName: string,
-        serverName: string,
-        status: MCPServerStatus
-    ): Promise<boolean> {
-        return await this.withLock(async () => {
-            // Inline getServerCategory logic
-            const category = this.configuration.localServerCategories.find(s => s.name === categoryName);
-            if (!category?.installationStatus) return false;
-
-            category.installationStatus.serversStatus[serverName] = status;
-            category.installationStatus.lastUpdated = new Date().toISOString();
-            await this.saveConfiguration();
-            return true;
-        });
-    }
-
-    async updateServerOperationStatus(
-        categoryName: string,
-        serverName: string,
-        clientName: string,
-        operationStatus: OperationStatus
-    ): Promise<boolean> {
-        return await this.withLock(async () => {
-            // Inline getServerCategory logic
-            const category = this.configuration.localServerCategories.find(s => s.name === categoryName);
-            if (!category?.installationStatus?.serversStatus[serverName]) return false;
-
-            category.installationStatus.serversStatus[serverName].installedStatus[clientName] = operationStatus;
-            category.installationStatus.lastUpdated = new Date().toISOString();
-            await this.saveConfiguration();
-            return true;
-        });
-    }
-
-    async isRequirementsReady(categoryName: string, serverName: string): Promise<boolean> {
-        return await this.withLock(async () => {
-            // Inline getServerCategory logic
-            const category = this.configuration.localServerCategories.find(s => s.name === categoryName);
-            if (!category?.feedConfiguration) return false;
-
-            const serverConfig = category.feedConfiguration.mcpServers.find(s => s.name === serverName);
-            if (!serverConfig?.dependencies?.requirements) return true; // No requirements means ready
-
-            const requirementNames = serverConfig.dependencies.requirements.map(r => r.name);
-            // Inline getInstallationStatus logic (using the already fetched category)
-            const status = category?.installationStatus;
-
-            if (!status?.requirementsStatus) return false;
-
-            return requirementNames.every(name => {
-                const reqStatus = status.requirementsStatus[name];
-                return reqStatus?.installed && !reqStatus?.error;
-            });
-        });
-    }
-
-    async GetServerRequirementStatus(categoryName: string, serverName: string): Promise<RequirementStatus[]> {
-        return await this.withLock(async () => {
-            // Inline getServerCategory logic
-            const category = this.configuration.localServerCategories.find(s => s.name === categoryName);
-            if (!category?.feedConfiguration) return [];
-            const serverConfig = category.feedConfiguration.mcpServers.find(s => s.name === serverName);
-            if (!serverConfig?.dependencies?.requirements) return [];
-            const requirementNames = serverConfig.dependencies.requirements.map(r => r.name);
-
-            return requirementNames
-                .map(name => category?.installationStatus?.requirementsStatus[name])
-                .filter(x => x !== undefined) as RequirementStatus[];
-        });
-    }
-
-    async isServerReady(categoryName: string, serverName: string, clients: string[]): Promise<boolean> {
-        return await this.withLock(async () => {
-            // Inline the logic from getServerStatus and getInstallationStatus to avoid nested lock
-            const category = this.configuration.localServerCategories.find(s => s.name === categoryName);
-            const installationStatus = category?.installationStatus;
-            const serverStatus = installationStatus?.serversStatus[serverName];
-
-            if (!serverStatus) return false;
-
-            return clients.every(clientName => {
-                // Add optional chaining for safety in case installedStatus is missing
-                const clientStatus = serverStatus.installedStatus?.[clientName];
-                return clientStatus?.status === 'completed' && !clientStatus?.error;
-            });
-        });
-    }
-    async syncFeeds(): Promise<void> {
-        return await this.withLock(async () => {
-            Logger.log('Starting feed synchronization...');
-            try {
-                // Check GitHub authentication first
-                await checkGithubAuth();
-
-                Logger.debug({
-                    action: 'create_directories',
-                    paths: {
-                        localFeeds: LOCAL_FEEDS_DIR,
-                        tempDir: this.tempDir
-                    }
-                });
-
-                await fs.mkdir(LOCAL_FEEDS_DIR, { recursive: true });
-                await fs.mkdir(this.tempDir, { recursive: true });
-
-                // Clean up temp directory
-                await fs.rm(this.tempDir, { recursive: true, force: true });
-
-                // Download latest release
-                Logger.debug('Downloading latest release...');
-                const { downloadGithubRelease } = await import('../../utils/githubUtils.js');
-                const { version, downloadPath } = await downloadGithubRelease(
-                    GITHUB_REPO.repoName,
-                    'latest',
-                    GITHUB_REPO.feedAssetsName,
-                    undefined,
-                    true,
-                    this.tempDir
-                );
-
-                Logger.debug({
-                    action: 'download_release',
-                    downloadPath,
-                    version,
-                    repoName: GITHUB_REPO.repoName,
-                });
-
-                Logger.debug('Updating local feeds...');
-                await fs.rm(LOCAL_FEEDS_DIR, { recursive: true, force: true });
-                const sourceFeedsDir = downloadPath;
-
-                try {
-                    await fs.access(downloadPath);
-                } catch (err) {
-                    throw new Error(`Could not find feeds directory in downloaded path: ${sourceFeedsDir}`);
-                }
-
-                await fs.cp(sourceFeedsDir, LOCAL_FEEDS_DIR, { recursive: true, force: true });
-                Logger.log('Successfully updated local feeds');
-
-            } catch (error) {
-                Logger.error('Error during feed synchronization', error);
-                throw new Error('Failed to sync feeds. Use --verbose for detailed error information.');
-            }
-        });
-    }
-
-    private async loadFeedsIntoConfiguration(feedFile?: string, settings?: { prLink?:string, adhocServers?: string[] }): Promise<void> {
-        this.configuration = await ConfigurationLoader.loadFeedsIntoConfiguration(this.configuration, feedFile, settings);
-        await this.saveConfiguration();
-    }
-
-    private async loadClientMCPSettings(): Promise<void> {
-        this.configuration = await ConfigurationLoader.loadClientMCPSettings(this.configuration);
-        await this.saveConfiguration();
-    }
-
-    // Public method to reload client MCP settings
-    async reloadClientMCPSettings(): Promise<void> {
-        return await this.withLock(async () => {
-            await this.loadClientMCPSettings();
-        });
-    }
-
-    async removeServerFromClientMCPSettings(serverName: string, target?: string): Promise<void> {
-        return await this.withLock(async () => {
-            // Load utils in async context to avoid circular dependencies
-            const { readJsonFile, writeJsonFile } = await import('../../utils/clientUtils.js');
-
-            // Filter clients if target is specified
-            const clientEntries = Object.entries(SUPPORTED_CLIENTS as Record<string, ClientSettings>);
-            const targetClients = target
-                ? clientEntries.filter(([clientName]) => clientName === target)
-                : clientEntries;
-
-            for (const [clientName, clientSettings] of targetClients) {
-                const settingPath = process.env.CODE_INSIDERS
-                    ? clientSettings.codeInsiderSettingPath
-                    : clientSettings.codeSettingPath;
-
-                try {
-                    const content = await readJsonFile(settingPath, true);
-                    let modified = false;
-
-                    // Handle GitHub Copilot's different structure
-                    if (clientName === 'GithubCopilot' && content.mcp?.servers?.[serverName]) {
-                        delete content.mcp.servers[serverName];
-                        modified = true;
-                    } else if (content.mcpServers?.[serverName]) {
-                        delete content.mcpServers[serverName];
-                        modified = true;
-                    }
-
-                    // Only write if we actually modified the content
-                    if (modified) {
-                        await writeJsonFile(settingPath, content);
-                        Logger.debug(`Removed server ${serverName} from client ${clientName} settings`);
-                    }
-                } catch (error) {
-                    Logger.error(`Failed to remove server ${serverName} from client ${clientName} settings:`, error);
-                }
-            }
-
-            // Also update our in-memory configuration
-            if (this.configuration.clientMCPSettings) {
-                if (target) {
-                    // Only update settings for the target client
-                    const clientSettings = this.configuration.clientMCPSettings[target];
-                    if (clientSettings) {
-                        if (clientSettings.mcpServers?.[serverName]) {
-                            delete clientSettings.mcpServers[serverName];
-                        }
-                        if (clientSettings.servers?.[serverName]) {  // For GitHub Copilot
-                            delete clientSettings.servers[serverName];
-                        }
-                    }
-                } else {
-                    // Update all clients if no target specified
-                    for (const clientSettings of Object.values(this.configuration.clientMCPSettings)) {
-                        if (clientSettings.mcpServers?.[serverName]) {
-                            delete clientSettings.mcpServers[serverName];
-                        }
-                        if (clientSettings.servers?.[serverName]) {  // For GitHub Copilot
-                            delete clientSettings.servers[serverName];
-                        }
-                    }
-                }
-                await this.saveConfiguration();
-            }
-        });
-    }
-}
-
-// Export a singleton instance
-export const configProvider = ConfigurationProvider.getInstance();