jellyfin/Jellyfin.Server.Implementations/Activity/ActivityManager.cs

using System;
using System.Linq;
using System.Threading.Tasks;
using Jellyfin.Data.Entities;
using Jellyfin.Data.Events;
using Jellyfin.Data.Queries;
using MediaBrowser.Model.Activity;
using MediaBrowser.Model.Querying;
using Microsoft.EntityFrameworkCore;

namespace Jellyfin.Server.Implementations.Activity
{
    /// <summary>
    /// Manages the storage and retrieval of <see cref="ActivityLog"/> instances.
    /// </summary>
    public class ActivityManager : IActivityManager
    {
        private readonly IDbContextFactory<JellyfinDb> _provider;

        /// <summary>
        /// Initializes a new instance of the <see cref="ActivityManager"/> class.
        /// </summary>
        /// <param name="provider">The Jellyfin database provider.</param>
        public ActivityManager(IDbContextFactory<JellyfinDb> provider)
        {
            _provider = provider;
        }

        /// <inheritdoc/>
        public event EventHandler<GenericEventArgs<ActivityLogEntry>>? EntryCreated;

        /// <inheritdoc/>
        public async Task CreateAsync(ActivityLog entry)
        {
            var dbContext = await _provider.CreateDbContextAsync().ConfigureAwait(false);
            await using (dbContext.ConfigureAwait(false))
            {
                dbContext.ActivityLogs.Add(entry);
                await dbContext.SaveChangesAsync().ConfigureAwait(false);
            }

            EntryCreated?.Invoke(this, new GenericEventArgs<ActivityLogEntry>(ConvertToOldModel(entry)));
        }

        /// <inheritdoc/>
        public async Task<QueryResult<ActivityLogEntry>> GetPagedResultAsync(ActivityLogQuery query)
        {
            var dbContext = await _provider.CreateDbContextAsync().ConfigureAwait(false);
            await using (dbContext.ConfigureAwait(false))
            {
                var entries = dbContext.ActivityLogs
                    .OrderByDescending(entry => entry.DateCreated)
                    .Where(entry => query.MinDate == null || entry.DateCreated >= query.MinDate)
                    .Where(entry => !query.HasUserId.HasValue || entry.UserId.Equals(default) != query.HasUserId.Value);

                return new QueryResult<ActivityLogEntry>(
                    query.Skip,
                    await entries.CountAsync().ConfigureAwait(false),
                    await entries
                        .Skip(query.Skip ?? 0)
                        .Take(query.Limit ?? 100)
                        .Select(entity => new ActivityLogEntry(entity.Name, entity.Type, entity.UserId)
                        {
                            Id = entity.Id,
                            Overview = entity.Overview,
                            ShortOverview = entity.ShortOverview,
                            ItemId = entity.ItemId,
                            Date = entity.DateCreated,
                            Severity = entity.LogSeverity
                        })
                        .AsQueryable()
                        .ToListAsync()
                        .ConfigureAwait(false));
            }
        }

        /// <inheritdoc />
        public async Task CleanAsync(DateTime startDate)
        {
            var dbContext = await _provider.CreateDbContextAsync().ConfigureAwait(false);
            await using (dbContext.ConfigureAwait(false))
            {
                var entries = dbContext.ActivityLogs
                    .Where(entry => entry.DateCreated <= startDate);

                dbContext.RemoveRange(entries);
                await dbContext.SaveChangesAsync().ConfigureAwait(false);
            }
        }

        private static ActivityLogEntry ConvertToOldModel(ActivityLog entry)
        {
            return new ActivityLogEntry(entry.Name, entry.Type, entry.UserId)
            {
                Id = entry.Id,
                Overview = entry.Overview,
                ShortOverview = entry.ShortOverview,
                ItemId = entry.ItemId,
                Date = entry.DateCreated,
                Severity = entry.LogSeverity
            };
        }
    }
}