/**
 * Copyright (c) 2024-2025 Тарабанов Александр Викторович
 * All rights reserved.
 * 
 * This software is proprietary and confidential.
 * Unauthorized copying, modification, or distribution is prohibited.
 * 
 * For licensing inquiries: info@hb3-accelerator.com
 * Website: https://hb3-accelerator.com
 * GitHub: https://github.com/HB3-ACCELERATOR
 */

/**
 * Конфигурационный сервис для Ollama
 * Централизует все настройки и URL для Ollama API
 * 
 * ВАЖНО: Настройки берутся из таблицы ai_providers_settings (через aiProviderSettingsService)
 */

const logger = require('../utils/logger');

// Кэш для настроек из БД
let settingsCache = null;

/**
 * Загружает настройки Ollama из базы данных
 * @returns {Promise<Object>} Настройки Ollama провайдера
 */
async function loadSettingsFromDb() {
  try {
    const aiProviderSettingsService = require('./aiProviderSettingsService');
    const settings = await aiProviderSettingsService.getProviderSettings('ollama');
    
    if (settings) {
      settingsCache = settings;
      logger.info(`[ollamaConfig] Loaded settings from DB: model=${settings.selected_model}, base_url=${settings.base_url}`);
    }
    
    return settings;
  } catch (error) {
    logger.error('[ollamaConfig] Ошибка загрузки настроек Ollama из БД:', error.message);
    return null;
  }
}

/**
 * Получает базовый URL для Ollama (синхронная версия)
 * @returns {string} Базовый URL Ollama
 */
function getBaseUrl() {
  // Приоритет: кэш из БД > Docker дефолт
  if (settingsCache && settingsCache.base_url) {
    return settingsCache.base_url;
  }
  // URL по умолчанию для Docker
  return 'http://ollama:11434';
}

/**
 * Получает базовый URL для Ollama (асинхронная версия)
 * @returns {Promise<string>} Базовый URL Ollama
 */
async function getBaseUrlAsync() {
  try {
    if (!settingsCache) {
      await loadSettingsFromDb();
    }
    
    if (settingsCache && settingsCache.base_url) {
      return settingsCache.base_url;
    }
  } catch (error) {
    logger.warn('[ollamaConfig] Failed to load base_url from DB, using default');
  }
  
  return 'http://ollama:11434';
}

/**
 * Получает URL для конкретного API endpoint Ollama
 * @param {string} endpoint - Endpoint API (например: 'tags', 'generate')
 * @returns {string} Полный URL для API endpoint
 */
function getApiUrl(endpoint) {
  const baseUrl = getBaseUrl();
  return `${baseUrl}/api/${endpoint}`;
}

/**
 * Получает модель по умолчанию для Ollama (синхронная версия)
 * @returns {string} Название модели
 */
function getDefaultModel() {
  // Приоритет: кэш из БД > дефолт
  if (settingsCache && settingsCache.selected_model) {
    return settingsCache.selected_model;
  }
  // Дефолтное значение если БД недоступна
  return 'qwen2.5:7b';
}

/**
 * Получает модель асинхронно из БД
 * @returns {Promise<string>} Название модели из БД
 */
async function getDefaultModelAsync() {
  try {
    if (!settingsCache) {
      await loadSettingsFromDb();
    }
    
    if (settingsCache && settingsCache.selected_model) {
      logger.info(`[ollamaConfig] Using model from DB: ${settingsCache.selected_model}`);
      return settingsCache.selected_model;
    }
  } catch (error) {
    logger.warn('[ollamaConfig] Failed to load model from DB, using default');
  }
  return 'qwen2.5:7b';
}

/**
 * Получает embedding модель асинхронно из БД
 * @returns {Promise<string>} Название embedding модели из БД
 */
async function getEmbeddingModel() {
  try {
    if (!settingsCache) {
      await loadSettingsFromDb();
    }
    
    if (settingsCache && settingsCache.embedding_model) {
      logger.info(`[ollamaConfig] Using embedding model from DB: ${settingsCache.embedding_model}`);
      return settingsCache.embedding_model;
    }
  } catch (error) {
    logger.warn('[ollamaConfig] Failed to load embedding model from DB, using default');
  }
  return 'mxbai-embed-large:latest';
}

/**
 * Получает timeout для запросов к Ollama
 * @returns {number} Timeout в миллисекундах
 */
function getTimeout() {
  return 30000; // 30 секунд
}

/**
 * Получает все конфигурационные параметры Ollama (синхронная версия)
 * @returns {Object} Объект с конфигурацией
 */
function getConfig() {
  return {
    baseUrl: getBaseUrl(),
    defaultModel: getDefaultModel(),
    timeout: getTimeout(),
    apiUrl: {
      tags: getApiUrl('tags'),
      generate: getApiUrl('generate'),
      chat: getApiUrl('chat'),
      models: getApiUrl('models'),
      show: getApiUrl('show'),
      pull: getApiUrl('pull'),
      push: getApiUrl('push')
    }
  };
}

/**
 * Получает все конфигурационные параметры Ollama (асинхронная версия)
 * @returns {Promise<Object>} Объект с конфигурацией
 */
async function getConfigAsync() {
  const baseUrl = await getBaseUrlAsync();
  const defaultModel = await getDefaultModelAsync();
  const embeddingModel = await getEmbeddingModel();
  
  return {
    baseUrl,
    defaultModel,
    embeddingModel,
    timeout: getTimeout(),
    apiUrl: {
      tags: `${baseUrl}/api/tags`,
      generate: `${baseUrl}/api/generate`,
      chat: `${baseUrl}/api/chat`,
      models: `${baseUrl}/api/models`,
      show: `${baseUrl}/api/show`,
      pull: `${baseUrl}/api/pull`,
      push: `${baseUrl}/api/push`
    }
  };
}

/**
 * Очищает кэш настроек (для перезагрузки)
 */
function clearCache() {
  settingsCache = null;
  logger.info('[ollamaConfig] Settings cache cleared');
}

/**
 * Проверяет доступность Ollama сервиса
 * @returns {Promise<Object>} Статус здоровья сервиса
 */
async function checkHealth() {
  try {
    const baseUrl = getBaseUrl();
    const response = await fetch(`${baseUrl}/api/tags`);
    
    if (!response.ok) {
      return { 
        status: 'error', 
        error: `Ollama вернул код ${response.status}`,
        baseUrl 
      };
    }

    const data = await response.json();
    return { 
      status: 'ok', 
      baseUrl,
      model: getDefaultModel(),
      availableModels: data.models?.length || 0
    };
  } catch (error) {
    return { 
      status: 'error', 
      error: error.message,
      baseUrl: getBaseUrl()
    };
  }
}

module.exports = {
  getBaseUrl,
  getBaseUrlAsync,
  getApiUrl,
  getDefaultModel,
  getDefaultModelAsync,
  getEmbeddingModel,
  getTimeout,
  getConfig,
  getConfigAsync,
  loadSettingsFromDb,
  clearCache,
  checkHealth
};