using Microsoft.EntityFrameworkCore;
using TakeoutSaaS.Domain.Identity.Entities;
using TakeoutSaaS.Domain.Identity.Repositories;

namespace TakeoutSaaS.Infrastructure.Identity.Persistence;

/// <summary>
/// EF 角色-权限仓储。
/// </summary>
public sealed class EfRolePermissionRepository(IdentityDbContext dbContext) : IRolePermissionRepository
{
    public Task<IReadOnlyList<RolePermission>> GetByRoleIdsAsync(long tenantId, IEnumerable<long> roleIds, CancellationToken cancellationToken = default)
        => dbContext.RolePermissions.AsNoTracking()
            .Where(x => x.TenantId == tenantId && roleIds.Contains(x.RoleId))
            .ToListAsync(cancellationToken)
            .ContinueWith(t => (IReadOnlyList<RolePermission>)t.Result, cancellationToken);

    public async Task AddRangeAsync(IEnumerable<RolePermission> rolePermissions, CancellationToken cancellationToken = default)
    {
        var toAdd = rolePermissions as RolePermission[] ?? rolePermissions.ToArray();
        if (toAdd.Length == 0)
        {
            return;
        }

        await dbContext.RolePermissions.AddRangeAsync(toAdd, cancellationToken);
    }

    public async Task ReplaceRolePermissionsAsync(long tenantId, long roleId, IEnumerable<long> permissionIds, CancellationToken cancellationToken = default)
    {
        var strategy = dbContext.Database.CreateExecutionStrategy();
        await strategy.ExecuteAsync(async () =>
        {
            await using var trx = await dbContext.Database.BeginTransactionAsync(cancellationToken);

            // 1. 删除旧记录（原生 SQL，避免跟踪干扰）
            await dbContext.Database.ExecuteSqlRawAsync(
                "DELETE FROM \"role_permissions\" WHERE \"TenantId\" = {0} AND \"RoleId\" = {1};",
                parameters: new object[] { tenantId, roleId },
                cancellationToken: cancellationToken);

            // 2. 插入新记录（防重复）
            foreach (var permissionId in permissionIds.Distinct())
            {
                await dbContext.Database.ExecuteSqlRawAsync(
                    "INSERT INTO \"role_permissions\" (\"TenantId\",\"RoleId\",\"PermissionId\",\"CreatedAt\",\"CreatedBy\",\"UpdatedAt\",\"UpdatedBy\",\"DeletedAt\",\"DeletedBy\") VALUES ({0},{1},{2},NOW(),NULL,NULL,NULL,NULL,NULL) ON CONFLICT DO NOTHING;",
                    parameters: new object[] { tenantId, roleId, permissionId },
                    cancellationToken: cancellationToken);
            }

            await trx.CommitAsync(cancellationToken);
        });
    }

    public Task SaveChangesAsync(CancellationToken cancellationToken = default)
        => dbContext.SaveChangesAsync(cancellationToken);
}