Notes/apps/server/src/services/llm/config/configuration_manager.ts

import options from '../../options.js';
import log from '../../log.js';
import type {
    AIConfig,
    ProviderPrecedenceConfig,
    EmbeddingProviderPrecedenceConfig,
    ModelIdentifier,
    ModelConfig,
    ProviderType,
    EmbeddingProviderType,
    ConfigValidationResult,
    ProviderSettings,
    OpenAISettings,
    AnthropicSettings,
    OllamaSettings
} from '../interfaces/configuration_interfaces.js';

/**
 * Configuration manager that handles conversion from string-based options
 * to proper typed configuration objects.
 *
 * This is the ONLY place where string parsing should happen for LLM configurations.
 */
export class ConfigurationManager {
    private static instance: ConfigurationManager | null = null;
    private cachedConfig: AIConfig | null = null;
    private lastConfigUpdate: number = 0;

    // Cache for 5 minutes to avoid excessive option reads
    private static readonly CACHE_DURATION = 5 * 60 * 1000;

    private constructor() {}

    public static getInstance(): ConfigurationManager {
        if (!ConfigurationManager.instance) {
            ConfigurationManager.instance = new ConfigurationManager();
        }
        return ConfigurationManager.instance;
    }

    /**
     * Get the complete AI configuration
     */
    public async getAIConfig(): Promise<AIConfig> {
        const now = Date.now();
        if (this.cachedConfig && (now - this.lastConfigUpdate) < ConfigurationManager.CACHE_DURATION) {
            return this.cachedConfig;
        }

        try {
            const config: AIConfig = {
                enabled: await this.getAIEnabled(),
                providerPrecedence: await this.getProviderPrecedence(),
                embeddingProviderPrecedence: await this.getEmbeddingProviderPrecedence(),
                defaultModels: await this.getDefaultModels(),
                providerSettings: await this.getProviderSettings()
            };

            this.cachedConfig = config;
            this.lastConfigUpdate = now;
            return config;
        } catch (error) {
            log.error(`Error loading AI configuration: ${error}`);
            return this.getDefaultConfig();
        }
    }

    /**
     * Parse provider precedence from string option
     */
    public async getProviderPrecedence(): Promise<ProviderPrecedenceConfig> {
        try {
            const precedenceOption = await options.getOption('aiProviderPrecedence');
            const providers = this.parseProviderList(precedenceOption);

            return {
                providers: providers as ProviderType[],
                defaultProvider: providers[0] as ProviderType
            };
        } catch (error) {
            log.error(`Error parsing provider precedence: ${error}`);
            return {
                providers: ['openai', 'anthropic', 'ollama'],
                defaultProvider: 'openai'
            };
        }
    }

    /**
     * Parse embedding provider precedence from string option
     */
    public async getEmbeddingProviderPrecedence(): Promise<EmbeddingProviderPrecedenceConfig> {
        try {
            const precedenceOption = await options.getOption('embeddingProviderPrecedence');
            const providers = this.parseProviderList(precedenceOption);

            return {
                providers: providers as EmbeddingProviderType[],
                defaultProvider: providers[0] as EmbeddingProviderType
            };
        } catch (error) {
            log.error(`Error parsing embedding provider precedence: ${error}`);
            return {
                providers: ['openai', 'ollama'],
                defaultProvider: 'openai'
            };
        }
    }

    /**
     * Parse model identifier with optional provider prefix
     * Handles formats like "gpt-4", "openai:gpt-4", "ollama:llama2:7b"
     */
    public parseModelIdentifier(modelString: string): ModelIdentifier {
        if (!modelString) {
            return {
                modelId: '',
                fullIdentifier: ''
            };
        }

        const parts = modelString.split(':');

        if (parts.length === 1) {
            // No provider prefix, just model name
            return {
                modelId: modelString,
                fullIdentifier: modelString
            };
        }

        // Check if first part is a known provider
        const potentialProvider = parts[0].toLowerCase();
        const knownProviders: ProviderType[] = ['openai', 'anthropic', 'ollama'];

        if (knownProviders.includes(potentialProvider as ProviderType)) {
            // Provider prefix format
            const provider = potentialProvider as ProviderType;
            const modelId = parts.slice(1).join(':'); // Rejoin in case model has colons

            return {
                provider,
                modelId,
                fullIdentifier: modelString
            };
        }

        // Not a provider prefix, treat whole string as model name
        return {
            modelId: modelString,
            fullIdentifier: modelString
        };
    }

    /**
     * Create model configuration from string
     */
    public createModelConfig(modelString: string, defaultProvider?: ProviderType): ModelConfig {
        const identifier = this.parseModelIdentifier(modelString);
        const provider = identifier.provider || defaultProvider || 'openai';

        return {
            provider,
            modelId: identifier.modelId,
            displayName: identifier.fullIdentifier
        };
    }

    /**
     * Get default models for each provider
     */
    public async getDefaultModels(): Promise<Record<ProviderType, string>> {
        try {
            const [openaiModel, anthropicModel, ollamaModel] = await Promise.all([
                options.getOption('openaiDefaultModel'),
                options.getOption('anthropicDefaultModel'),
                options.getOption('ollamaDefaultModel')
            ]);

            return {
                openai: openaiModel || 'gpt-3.5-turbo',
                anthropic: anthropicModel || 'claude-3-sonnet-20240229',
                ollama: ollamaModel || 'llama2'
            };
        } catch (error) {
            log.error(`Error loading default models: ${error}`);
            return {
                openai: 'gpt-3.5-turbo',
                anthropic: 'claude-3-sonnet-20240229',
                ollama: 'llama2'
            };
        }
    }

    /**
     * Get provider-specific settings
     */
    public async getProviderSettings(): Promise<ProviderSettings> {
        try {
            const [
                openaiApiKey, openaiBaseUrl, openaiDefaultModel,
                anthropicApiKey, anthropicBaseUrl, anthropicDefaultModel,
                ollamaBaseUrl, ollamaDefaultModel
            ] = await Promise.all([
                options.getOption('openaiApiKey'),
                options.getOption('openaiBaseUrl'),
                options.getOption('openaiDefaultModel'),
                options.getOption('anthropicApiKey'),
                options.getOption('anthropicBaseUrl'),
                options.getOption('anthropicDefaultModel'),
                options.getOption('ollamaBaseUrl'),
                options.getOption('ollamaDefaultModel')
            ]);

            const settings: ProviderSettings = {};

            if (openaiApiKey || openaiBaseUrl || openaiDefaultModel) {
                settings.openai = {
                    apiKey: openaiApiKey,
                    baseUrl: openaiBaseUrl,
                    defaultModel: openaiDefaultModel
                };
            }

            if (anthropicApiKey || anthropicBaseUrl || anthropicDefaultModel) {
                settings.anthropic = {
                    apiKey: anthropicApiKey,
                    baseUrl: anthropicBaseUrl,
                    defaultModel: anthropicDefaultModel
                };
            }

            if (ollamaBaseUrl || ollamaDefaultModel) {
                settings.ollama = {
                    baseUrl: ollamaBaseUrl,
                    defaultModel: ollamaDefaultModel
                };
            }

            return settings;
        } catch (error) {
            log.error(`Error loading provider settings: ${error}`);
            return {};
        }
    }

    /**
     * Validate configuration
     */
    public async validateConfig(): Promise<ConfigValidationResult> {
        const result: ConfigValidationResult = {
            isValid: true,
            errors: [],
            warnings: []
        };

        try {
            const config = await this.getAIConfig();

            if (!config.enabled) {
                result.warnings.push('AI features are disabled');
                return result;
            }

            // Validate provider precedence
            if (config.providerPrecedence.providers.length === 0) {
                result.errors.push('No providers configured in precedence list');
                result.isValid = false;
            }

            // Validate provider settings
            for (const provider of config.providerPrecedence.providers) {
                const providerConfig = config.providerSettings[provider];

                if (provider === 'openai') {
                    const openaiConfig = providerConfig as OpenAISettings | undefined;
                    if (!openaiConfig?.apiKey) {
                        result.warnings.push('OpenAI API key is not configured');
                    }
                }

                if (provider === 'anthropic') {
                    const anthropicConfig = providerConfig as AnthropicSettings | undefined;
                    if (!anthropicConfig?.apiKey) {
                        result.warnings.push('Anthropic API key is not configured');
                    }
                }

                if (provider === 'ollama') {
                    const ollamaConfig = providerConfig as OllamaSettings | undefined;
                    if (!ollamaConfig?.baseUrl) {
                        result.warnings.push('Ollama base URL is not configured');
                    }
                }
            }

        } catch (error) {
            result.errors.push(`Configuration validation error: ${error}`);
            result.isValid = false;
        }

        return result;
    }

    /**
     * Clear cached configuration (force reload on next access)
     */
    public clearCache(): void {
        this.cachedConfig = null;
        this.lastConfigUpdate = 0;
    }

    // Private helper methods

    private async getAIEnabled(): Promise<boolean> {
        try {
            return await options.getOptionBool('aiEnabled');
        } catch {
            return false;
        }
    }

    private parseProviderList(precedenceOption: string | null): string[] {
        if (!precedenceOption) {
            return ['openai', 'anthropic', 'ollama'];
        }

        try {
            // Handle JSON array format
            if (precedenceOption.startsWith('[') && precedenceOption.endsWith(']')) {
                const parsed = JSON.parse(precedenceOption);
                if (Array.isArray(parsed)) {
                    return parsed.map(p => String(p).trim());
                }
            }

            // Handle comma-separated format
            if (precedenceOption.includes(',')) {
                return precedenceOption.split(',').map(p => p.trim());
            }

            // Handle single provider
            return [precedenceOption.trim()];

        } catch (error) {
            log.error(`Error parsing provider list "${precedenceOption}": ${error}`);
            return ['openai', 'anthropic', 'ollama'];
        }
    }

    private getDefaultConfig(): AIConfig {
        return {
            enabled: false,
            providerPrecedence: {
                providers: ['openai', 'anthropic', 'ollama'],
                defaultProvider: 'openai'
            },
            embeddingProviderPrecedence: {
                providers: ['openai', 'ollama'],
                defaultProvider: 'openai'
            },
            defaultModels: {
                openai: 'gpt-3.5-turbo',
                anthropic: 'claude-3-sonnet-20240229',
                ollama: 'llama2'
            },
            providerSettings: {}
        };
    }
}

// Export singleton instance
export default ConfigurationManager.getInstance();