imcp 0.1.5 → 0.1.6

package/src/core/loaders/ConfigurationProvider.ts

@@ -0,0 +1,462 @@
+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();