VberAdmin_V4/SourceCode/Zero/VberYue.Zero/Notifications/NotificationStore.cs

using System.Linq.Expressions;

using Abp;
using Abp.Dependency;
using Abp.Domain.Repositories;
using Abp.Domain.Uow;
using Abp.Linq.Expressions;
using Abp.Linq.Extensions;
using Abp.Notifications;

namespace VberZero.Notifications;

/// <summary>
/// Implements <see cref="INotificationStore"/> using repositories.
/// </summary>
public class NotificationStore : INotificationStore, ITransientDependency
{
    private readonly IRepository<NotificationInfo, Guid> _notificationRepository;
    private readonly IRepository<TenantNotificationInfo, Guid> _tenantNotificationRepository;
    private readonly IRepository<UserNotificationInfo, Guid> _userNotificationRepository;
    private readonly IRepository<NotificationSubscriptionInfo, Guid> _notificationSubscriptionRepository;
    private readonly IUnitOfWorkManager _unitOfWorkManager;

    /// <summary>
    /// Initializes a new instance of the <see cref="NotificationStore"/> class.
    /// </summary>
    public NotificationStore(
        IRepository<NotificationInfo, Guid> notificationRepository,
        IRepository<TenantNotificationInfo, Guid> tenantNotificationRepository,
        IRepository<UserNotificationInfo, Guid> userNotificationRepository,
        IRepository<NotificationSubscriptionInfo, Guid> notificationSubscriptionRepository,
        IUnitOfWorkManager unitOfWorkManager)
    {
        _notificationRepository = notificationRepository;
        _tenantNotificationRepository = tenantNotificationRepository;
        _userNotificationRepository = userNotificationRepository;
        _notificationSubscriptionRepository = notificationSubscriptionRepository;
        _unitOfWorkManager = unitOfWorkManager;
    }

    public virtual async Task InsertSubscriptionAsync(NotificationSubscriptionInfo subscription)
    {
        await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(subscription.TenantId))
            {
                await _notificationSubscriptionRepository.InsertAsync(subscription);
                await _unitOfWorkManager.Current.SaveChangesAsync();
            }
        });
    }

    public virtual void InsertSubscription(NotificationSubscriptionInfo subscription)
    {
        _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(subscription.TenantId))
            {
                _notificationSubscriptionRepository.Insert(subscription);
                _unitOfWorkManager.Current.SaveChanges();
            }
        });
    }

    public virtual async Task DeleteSubscriptionAsync(
        UserIdentifier user,
        string notificationName,
        string entityTypeName,
        string entityId)
    {
        await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                await _notificationSubscriptionRepository.DeleteAsync(s =>
                    s.UserId == user.UserId &&
                    s.NotificationName == notificationName &&
                    s.EntityTypeName == entityTypeName &&
                    s.EntityId == entityId
                );

                await _unitOfWorkManager.Current.SaveChangesAsync();
            }
        });
    }

    public virtual void DeleteSubscription(
        UserIdentifier user,
        string notificationName,
        string entityTypeName,
        string entityId)
    {
        _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                _notificationSubscriptionRepository.Delete(s =>
                    s.UserId == user.UserId &&
                    s.NotificationName == notificationName &&
                    s.EntityTypeName == entityTypeName &&
                    s.EntityId == entityId
                );

                _unitOfWorkManager.Current.SaveChanges();
            }
        });
    }

    public virtual async Task InsertNotificationAsync(NotificationInfo notification)
    {
        await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(null))
            {
                await _notificationRepository.InsertAsync(notification);
                await _unitOfWorkManager.Current.SaveChangesAsync();
            }
        });
    }

    public virtual void InsertNotification(NotificationInfo notification)
    {
        _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(null))
            {
                _notificationRepository.Insert(notification);
                _unitOfWorkManager.Current.SaveChanges();
            }
        });
    }

    public virtual async Task<NotificationInfo> GetNotificationOrNullAsync(Guid notificationId)
    {
        return await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(null))
            {
                return await _notificationRepository.FirstOrDefaultAsync(notificationId);
            }
        });
    }

    public virtual NotificationInfo GetNotificationOrNull(Guid notificationId)
    {
        return _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(null))
            {
                return _notificationRepository.FirstOrDefault(notificationId);
            }
        });
    }

    public virtual async Task InsertUserNotificationAsync(UserNotificationInfo userNotification)
    {
        await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(userNotification.TenantId))
            {
                await _userNotificationRepository.InsertAsync(userNotification);
                await _unitOfWorkManager.Current.SaveChangesAsync();
            }
        });
    }

    public virtual void InsertUserNotification(UserNotificationInfo userNotification)
    {
        _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(userNotification.TenantId))
            {
                _userNotificationRepository.Insert(userNotification);
                _unitOfWorkManager.Current.SaveChanges();
            }
        });
    }

    public virtual async Task<List<NotificationSubscriptionInfo>> GetSubscriptionsAsync(
        string notificationName,
        string entityTypeName,
        string entityId)
    {
        return await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.DisableFilter(AbpDataFilters.MayHaveTenant))
            {
                return await _notificationSubscriptionRepository.GetAllListAsync(s =>
                    s.NotificationName == notificationName &&
                    s.EntityTypeName == entityTypeName &&
                    s.EntityId == entityId
                );
            }
        });
    }

    public virtual List<NotificationSubscriptionInfo> GetSubscriptions(
        string notificationName,
        string entityTypeName,
        string entityId)
    {
        return _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.DisableFilter(AbpDataFilters.MayHaveTenant))
            {
                return _notificationSubscriptionRepository.GetAllList(s =>
                    s.NotificationName == notificationName &&
                    s.EntityTypeName == entityTypeName &&
                    s.EntityId == entityId
                );
            }
        });
    }

    public virtual async Task<List<NotificationSubscriptionInfo>> GetSubscriptionsAsync(
        int?[] tenantIds,
        string notificationName,
        string entityTypeName,
        string entityId)
    {
        return await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            var subscriptions = new List<NotificationSubscriptionInfo>();

            foreach (var tenantId in tenantIds)
            {
                subscriptions.AddRange(await GetSubscriptionsAsync(tenantId, notificationName, entityTypeName,
                    entityId));
            }

            return subscriptions;
        });
    }

    public virtual List<NotificationSubscriptionInfo> GetSubscriptions(
        int?[] tenantIds,
        string notificationName,
        string entityTypeName,
        string entityId)
    {
        return _unitOfWorkManager.WithUnitOfWork(() =>
        {
            var subscriptions = new List<NotificationSubscriptionInfo>();

            foreach (var tenantId in tenantIds)
            {
                subscriptions.AddRange(GetSubscriptions(tenantId, notificationName, entityTypeName, entityId));
            }

            return subscriptions;
        });
    }

    public virtual async Task<List<NotificationSubscriptionInfo>> GetSubscriptionsAsync(UserIdentifier user)
    {
        return await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                return await _notificationSubscriptionRepository.GetAllListAsync(s => s.UserId == user.UserId);
            }
        });
    }

    public virtual List<NotificationSubscriptionInfo> GetSubscriptions(UserIdentifier user)
    {
        return _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                return _notificationSubscriptionRepository.GetAllList(s => s.UserId == user.UserId);
            }
        });
    }

    protected virtual async Task<List<NotificationSubscriptionInfo>> GetSubscriptionsAsync(
        int? tenantId,
        string notificationName,
        string entityTypeName,
        string entityId)
    {
        return await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(tenantId))
            {
                return await _notificationSubscriptionRepository.GetAllListAsync(s =>
                    s.NotificationName == notificationName &&
                    s.EntityTypeName == entityTypeName &&
                    s.EntityId == entityId
                );
            }
        });
    }

    protected virtual List<NotificationSubscriptionInfo> GetSubscriptions(
        int? tenantId,
        string notificationName,
        string entityTypeName,
        string entityId)
    {
        return _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(tenantId))
            {
                return _notificationSubscriptionRepository.GetAllList(s =>
                    s.NotificationName == notificationName &&
                    s.EntityTypeName == entityTypeName &&
                    s.EntityId == entityId
                );
            }
        });
    }

    public virtual async Task<bool> IsSubscribedAsync(
        UserIdentifier user,
        string notificationName,
        string entityTypeName,
        string entityId)
    {
        return await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                return await _notificationSubscriptionRepository.CountAsync(s =>
                    s.UserId == user.UserId &&
                    s.NotificationName == notificationName &&
                    s.EntityTypeName == entityTypeName &&
                    s.EntityId == entityId
                ) > 0;
            }
        });
    }

    public virtual bool IsSubscribed(
        UserIdentifier user,
        string notificationName,
        string entityTypeName,
        string entityId)
    {
        return _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                return _notificationSubscriptionRepository.Count(s =>
                    s.UserId == user.UserId &&
                    s.NotificationName == notificationName &&
                    s.EntityTypeName == entityTypeName &&
                    s.EntityId == entityId
                ) > 0;
            }
        });
    }

    public virtual async Task UpdateUserNotificationStateAsync(
        int? tenantId,
        Guid userNotificationId,
        UserNotificationState state)
    {
        await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(tenantId))
            {
                var userNotification = await _userNotificationRepository.FirstOrDefaultAsync(userNotificationId);
                if (userNotification == null)
                {
                    return;
                }

                userNotification.State = state;
                await _unitOfWorkManager.Current.SaveChangesAsync();
            }
        });
    }

    public virtual void UpdateUserNotificationState(
        int? tenantId,
        Guid userNotificationId,
        UserNotificationState state)
    {
        _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(tenantId))
            {
                var userNotification = _userNotificationRepository.FirstOrDefault(userNotificationId);
                if (userNotification == null)
                {
                    return;
                }

                userNotification.State = state;
                _unitOfWorkManager.Current.SaveChanges();
            }
        });
    }

    public virtual async Task UpdateAllUserNotificationStatesAsync(UserIdentifier user, UserNotificationState state)
    {
        await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                var userNotifications = await _userNotificationRepository.GetAllListAsync(
                    un => un.UserId == user.UserId
                );

                foreach (var userNotification in userNotifications)
                {
                    userNotification.State = state;
                }

                await _unitOfWorkManager.Current.SaveChangesAsync();
            }
        });
    }

    public virtual void UpdateAllUserNotificationStates(UserIdentifier user, UserNotificationState state)
    {
        _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                var userNotifications = _userNotificationRepository.GetAllList(
                    un => un.UserId == user.UserId
                );

                foreach (var userNotification in userNotifications)
                {
                    userNotification.State = state;
                }

                _unitOfWorkManager.Current.SaveChanges();
            }
        });
    }

    public virtual async Task DeleteUserNotificationAsync(int? tenantId, Guid userNotificationId)
    {
        await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(tenantId))
            {
                await _userNotificationRepository.DeleteAsync(userNotificationId);
                await _unitOfWorkManager.Current.SaveChangesAsync();
            }
        });
    }

    public virtual void DeleteUserNotification(int? tenantId, Guid userNotificationId)
    {
        _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(tenantId))
            {
                _userNotificationRepository.Delete(userNotificationId);
                _unitOfWorkManager.Current.SaveChanges();
            }
        });
    }

    public virtual async Task DeleteAllUserNotificationsAsync(
        UserIdentifier user,
        UserNotificationState? state = null,
        DateTime? startDate = null,
        DateTime? endDate = null)
    {
        await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                var predicate = CreateNotificationFilterPredicate(user, state, startDate, endDate);

                await _userNotificationRepository.DeleteAsync(predicate);
                await _unitOfWorkManager.Current.SaveChangesAsync();
            }
        });
    }

    public virtual void DeleteAllUserNotifications(
        UserIdentifier user,
        UserNotificationState? state = null,
        DateTime? startDate = null,
        DateTime? endDate = null)
    {
        _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                var predicate = CreateNotificationFilterPredicate(user, state, startDate, endDate);

                _userNotificationRepository.Delete(predicate);
                _unitOfWorkManager.Current.SaveChanges();
            }
        });
    }

    private Expression<Func<UserNotificationInfo, bool>> CreateNotificationFilterPredicate(
        UserIdentifier user,
        UserNotificationState? state = null,
        DateTime? startDate = null,
        DateTime? endDate = null)
    {
        var predicate = PredicateBuilder.New<UserNotificationInfo>();
        predicate = predicate.And(p => p.UserId == user.UserId);

        if (startDate.HasValue)
        {
            predicate = predicate.And(p => p.CreationTime >= startDate);
        }

        if (endDate.HasValue)
        {
            predicate = predicate.And(p => p.CreationTime <= endDate);
        }

        if (state.HasValue)
        {
            predicate = predicate.And(p => p.State == state);
        }

        return predicate;
    }

    public virtual async Task<List<UserNotificationInfoWithNotificationInfo>>
        GetUserNotificationsWithNotificationsAsync(
            UserIdentifier user,
            UserNotificationState? state = null,
            int skipCount = 0,
            int maxResultCount = int.MaxValue,
            DateTime? startDate = null,
            DateTime? endDate = null)
    {
        var result = _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                var query = from userNotificationInfo in _userNotificationRepository.GetAll()
                            join tenantNotificationInfo in _tenantNotificationRepository.GetAll() on userNotificationInfo
                                .TenantNotificationId equals tenantNotificationInfo.Id
                            where userNotificationInfo.UserId == user.UserId
                            orderby tenantNotificationInfo.CreationTime descending
                            select new
                            {
                                userNotificationInfo,
                                tenantNotificationInfo
                            };

                if (state.HasValue)
                {
                    query = query.Where(x => x.userNotificationInfo.State == state.Value);
                }

                if (startDate.HasValue)
                {
                    query = query.Where(x => x.tenantNotificationInfo.CreationTime >= startDate);
                }

                if (endDate.HasValue)
                {
                    query = query.Where(x => x.tenantNotificationInfo.CreationTime <= endDate);
                }

                query = query.PageBy(skipCount, maxResultCount);

                var list = query.ToList();

                return list.Select(
                    a => new UserNotificationInfoWithNotificationInfo(
                        a.userNotificationInfo,
                        a.tenantNotificationInfo
                    )
                ).ToList();
            }
        });

        return await Task.FromResult(result);
    }

    public virtual List<UserNotificationInfoWithNotificationInfo> GetUserNotificationsWithNotifications(
        UserIdentifier user,
        UserNotificationState? state = null,
        int skipCount = 0,
        int maxResultCount = int.MaxValue,
        DateTime? startDate = null,
        DateTime? endDate = null)
    {
        return _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                var query = from userNotificationInfo in _userNotificationRepository.GetAll()
                            join tenantNotificationInfo in _tenantNotificationRepository.GetAll() on userNotificationInfo
                                .TenantNotificationId equals tenantNotificationInfo.Id
                            where userNotificationInfo.UserId == user.UserId
                            orderby tenantNotificationInfo.CreationTime descending
                            select new
                            {
                                userNotificationInfo,
                                tenantNotificationInfo
                            };

                if (state.HasValue)
                {
                    query = query.Where(x => x.userNotificationInfo.State == state.Value);
                }

                if (startDate.HasValue)
                {
                    query = query.Where(x => x.tenantNotificationInfo.CreationTime >= startDate);
                }

                if (endDate.HasValue)
                {
                    query = query.Where(x => x.tenantNotificationInfo.CreationTime <= endDate);
                }

                query = query.PageBy(skipCount, maxResultCount);

                var list = query.ToList();

                return list.Select(
                    a => new UserNotificationInfoWithNotificationInfo(
                        a.userNotificationInfo,
                        a.tenantNotificationInfo
                    )
                ).ToList();
            }
        });
    }

    public virtual async Task<int> GetUserNotificationCountAsync(
        UserIdentifier user,
        UserNotificationState? state = null,
        DateTime? startDate = null,
        DateTime? endDate = null)
    {
        return await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                var predicate = CreateNotificationFilterPredicate(user, state, startDate, endDate);
                return await _userNotificationRepository.CountAsync(predicate);
            }
        });
    }

    public virtual int GetUserNotificationCount(
        UserIdentifier user,
        UserNotificationState? state = null,
        DateTime? startDate = null,
        DateTime? endDate = null)
    {
        return _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(user.TenantId))
            {
                var predicate = CreateNotificationFilterPredicate(user, state, startDate, endDate);
                return _userNotificationRepository.Count(predicate);
            }
        });
    }

    public virtual async Task<UserNotificationInfoWithNotificationInfo>
        GetUserNotificationWithNotificationOrNullAsync(
            int? tenantId,
            Guid userNotificationId)
    {
        var result = _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(tenantId))
            {
                var query = from userNotificationInfo in _userNotificationRepository.GetAll()
                            join tenantNotificationInfo in _tenantNotificationRepository.GetAll() on userNotificationInfo
                                .TenantNotificationId equals tenantNotificationInfo.Id
                            where userNotificationInfo.Id == userNotificationId
                            select new
                            {
                                userNotificationInfo,
                                tenantNotificationInfo
                            };

                var item = query.FirstOrDefault();
                if (item == null)
                {
                    return null;
                }

                return new UserNotificationInfoWithNotificationInfo(
                    item.userNotificationInfo,
                    item.tenantNotificationInfo
                );
            }
        });

        return await Task.FromResult(result);
    }

    public virtual UserNotificationInfoWithNotificationInfo GetUserNotificationWithNotificationOrNull(
        int? tenantId,
        Guid userNotificationId)
    {
        return _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(tenantId))
            {
                var query = from userNotificationInfo in _userNotificationRepository.GetAll()
                            join tenantNotificationInfo in _tenantNotificationRepository.GetAll() on userNotificationInfo
                                .TenantNotificationId equals tenantNotificationInfo.Id
                            where userNotificationInfo.Id == userNotificationId
                            select new
                            {
                                userNotificationInfo,
                                tenantNotificationInfo
                            };

                var item = query.FirstOrDefault();
                if (item == null)
                {
                    return null;
                }

                return new UserNotificationInfoWithNotificationInfo(
                    item.userNotificationInfo,
                    item.tenantNotificationInfo
                );
            }
        });
    }

    public virtual async Task InsertTenantNotificationAsync(TenantNotificationInfo tenantNotificationInfo)
    {
        await _unitOfWorkManager.WithUnitOfWorkAsync(async () =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(tenantNotificationInfo.TenantId))
            {
                await _tenantNotificationRepository.InsertAsync(tenantNotificationInfo);
            }
        });
    }

    public virtual void InsertTenantNotification(TenantNotificationInfo tenantNotificationInfo)
    {
        _unitOfWorkManager.WithUnitOfWork(() =>
        {
            using (_unitOfWorkManager.Current.SetTenantId(tenantNotificationInfo.TenantId))
            {
                _tenantNotificationRepository.Insert(tenantNotificationInfo);
            }
        });
    }

    public virtual async Task DeleteNotificationAsync(NotificationInfo notification)
    {
        await _unitOfWorkManager.WithUnitOfWorkAsync(async () => await _notificationRepository.DeleteAsync(notification));
    }

    public virtual void DeleteNotification(NotificationInfo notification)
    {
        _unitOfWorkManager.WithUnitOfWork(() =>
        {
            _notificationRepository.Delete(notification);
        });
    }
}