ActivityManager.cs 3.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. using System;
  2. using System.Linq;
  3. using System.Threading.Tasks;
  4. using Jellyfin.Data.Entities;
  5. using Jellyfin.Data.Events;
  6. using Jellyfin.Data.Queries;
  7. using MediaBrowser.Model.Activity;
  8. using MediaBrowser.Model.Querying;
  9. using Microsoft.EntityFrameworkCore;
  10. namespace Jellyfin.Server.Implementations.Activity
  11. {
  12. /// <summary>
  13. /// Manages the storage and retrieval of <see cref="ActivityLog"/> instances.
  14. /// </summary>
  15. public class ActivityManager : IActivityManager
  16. {
  17. private readonly JellyfinDbProvider _provider;
  18. /// <summary>
  19. /// Initializes a new instance of the <see cref="ActivityManager"/> class.
  20. /// </summary>
  21. /// <param name="provider">The Jellyfin database provider.</param>
  22. public ActivityManager(JellyfinDbProvider provider)
  23. {
  24. _provider = provider;
  25. }
  26. /// <inheritdoc/>
  27. public event EventHandler<GenericEventArgs<ActivityLogEntry>> EntryCreated;
  28. /// <inheritdoc/>
  29. public async Task CreateAsync(ActivityLog entry)
  30. {
  31. await using var dbContext = _provider.CreateContext();
  32. dbContext.ActivityLogs.Add(entry);
  33. await dbContext.SaveChangesAsync().ConfigureAwait(false);
  34. EntryCreated?.Invoke(this, new GenericEventArgs<ActivityLogEntry>(ConvertToOldModel(entry)));
  35. }
  36. /// <inheritdoc/>
  37. public async Task<QueryResult<ActivityLogEntry>> GetPagedResultAsync(ActivityLogQuery query)
  38. {
  39. await using var dbContext = _provider.CreateContext();
  40. IQueryable<ActivityLog> entries = dbContext.ActivityLogs
  41. .AsQueryable()
  42. .OrderByDescending(entry => entry.DateCreated);
  43. if (query.MinDate.HasValue)
  44. {
  45. entries = entries.Where(entry => entry.DateCreated >= query.MinDate);
  46. }
  47. if (query.HasUserId.HasValue)
  48. {
  49. entries = entries.Where(entry => entry.UserId != Guid.Empty == query.HasUserId.Value );
  50. }
  51. return new QueryResult<ActivityLogEntry>
  52. {
  53. Items = await entries
  54. .Skip(query.StartIndex ?? 0)
  55. .Take(query.Limit ?? 100)
  56. .AsAsyncEnumerable()
  57. .Select(ConvertToOldModel)
  58. .ToListAsync()
  59. .ConfigureAwait(false),
  60. TotalRecordCount = await entries.CountAsync().ConfigureAwait(false)
  61. };
  62. }
  63. private static ActivityLogEntry ConvertToOldModel(ActivityLog entry)
  64. {
  65. return new ActivityLogEntry
  66. {
  67. Id = entry.Id,
  68. Name = entry.Name,
  69. Overview = entry.Overview,
  70. ShortOverview = entry.ShortOverview,
  71. Type = entry.Type,
  72. ItemId = entry.ItemId,
  73. UserId = entry.UserId,
  74. Date = entry.DateCreated,
  75. Severity = entry.LogSeverity
  76. };
  77. }
  78. }
  79. }