using MiaoYu.Core.CodeGenerator.Abstractions;

namespace MiaoYu.Core.CodeGenerator.Core;

/// <summary>
/// 数据源扩展方法
/// </summary>
public static class DataSourceExtensions
{
    /// <summary>
    /// 从 Schema 中提取数据库标识
    /// </summary>
    /// <param name="schema">Schema字符串</param>
    /// <returns>数据库标识</returns>
    public static string ExtractDatabaseKey(this string schema)
    {
        if (string.IsNullOrWhiteSpace(schema))
            return DataSourceConstants.Admin;

        if (schema.Contains("."))
        {
            var parts = schema.Split('.');
            return parts.Length > 1 ? parts[1] : DataSourceConstants.Admin;
        }

        return DataSourceConstants.Admin;
    }

    /// <summary>
    /// 清理 Schema（移除数据库标识）
    /// </summary>
    /// <param name="schema">Schema字符串</param>
    /// <returns>清理后的Schema</returns>
    public static string CleanSchema(this string schema)
    {
        if (string.IsNullOrWhiteSpace(schema))
            return schema;

        return schema.Contains(".")
            ? schema.Split('.')[0]
            : schema;
    }
}