using MiaoYu.Core.CodeGenerator.Abstractions;

namespace MiaoYu.Core.CodeGenerator.Core;

/// <summary>
/// 数据源管理器
/// </summary>
[Component]
public class DataSourceManager : IScopedDependency
{
    private readonly IEnumerable<IDataSourceProvider> _providers;

    /// <summary>
    /// 构造函数（通过依赖注入自动收集所有IDataSourceProvider实现）
    /// </summary>
    /// <param name="providers">所有数据源提供者</param>
    public DataSourceManager(IEnumerable<IDataSourceProvider> providers)
    {
        _providers = providers.OrderBy(p => p.Config.Order);
    }

    /// <summary>
    /// 获取所有数据源提供者
    /// </summary>
    /// <returns>数据源提供者集合</returns>
    public IEnumerable<IDataSourceProvider> GetAllProviders() => _providers;

    /// <summary>
    /// 根据数据库标识获取数据源提供者
    /// </summary>
    /// <param name="databaseKey">数据库标识（如：Admin, MiaoYuChat）</param>
    /// <returns>数据源提供者，如果未找到返回null</returns>
    public IDataSourceProvider? GetProvider(string databaseKey)
    {
        if (string.IsNullOrWhiteSpace(databaseKey))
            return null;

        return _providers.FirstOrDefault(p =>
            p.Config.DatabaseKey.Equals(databaseKey, StringComparison.OrdinalIgnoreCase));
    }

    /// <summary>
    /// 获取所有数据源的表信息
    /// </summary>
    /// <returns>所有表信息列表</returns>
    public List<DbTableInfo> GetAllTables()
    {
        var allTables = new List<DbTableInfo>();

        foreach (var provider in _providers)
        {
            try
            {
                var tables = provider.GetTables();
                if (tables != null && tables.Count > 0)
                {
                    allTables.AddRange(tables);
                }
            }
            catch (Exception ex)
            {
                LogUtil.Log.Warning($"获取数据源 {provider.Config.DatabaseKey} 的表信息失败: {ex.Message}");
            }
        }

        return allTables;
    }
}