RecordingNotifier.cs 3.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. #pragma warning disable CS1591
  2. using System;
  3. using System.Linq;
  4. using System.Threading;
  5. using System.Threading.Tasks;
  6. using Jellyfin.Data.Enums;
  7. using Jellyfin.Data.Events;
  8. using MediaBrowser.Controller.Library;
  9. using MediaBrowser.Controller.LiveTv;
  10. using MediaBrowser.Controller.Plugins;
  11. using MediaBrowser.Controller.Session;
  12. using MediaBrowser.Model.Session;
  13. using Microsoft.Extensions.Logging;
  14. namespace Emby.Server.Implementations.EntryPoints
  15. {
  16. public sealed class RecordingNotifier : IServerEntryPoint
  17. {
  18. private readonly ILiveTvManager _liveTvManager;
  19. private readonly ISessionManager _sessionManager;
  20. private readonly IUserManager _userManager;
  21. private readonly ILogger<RecordingNotifier> _logger;
  22. public RecordingNotifier(
  23. ISessionManager sessionManager,
  24. IUserManager userManager,
  25. ILogger<RecordingNotifier> logger,
  26. ILiveTvManager liveTvManager)
  27. {
  28. _sessionManager = sessionManager;
  29. _userManager = userManager;
  30. _logger = logger;
  31. _liveTvManager = liveTvManager;
  32. }
  33. /// <inheritdoc />
  34. public Task RunAsync()
  35. {
  36. _liveTvManager.TimerCancelled += OnLiveTvManagerTimerCancelled;
  37. _liveTvManager.SeriesTimerCancelled += OnLiveTvManagerSeriesTimerCancelled;
  38. _liveTvManager.TimerCreated += OnLiveTvManagerTimerCreated;
  39. _liveTvManager.SeriesTimerCreated += OnLiveTvManagerSeriesTimerCreated;
  40. return Task.CompletedTask;
  41. }
  42. private async void OnLiveTvManagerSeriesTimerCreated(object sender, GenericEventArgs<TimerEventInfo> e)
  43. {
  44. await SendMessage(SessionMessageType.SeriesTimerCreated, e.Argument).ConfigureAwait(false);
  45. }
  46. private async void OnLiveTvManagerTimerCreated(object sender, GenericEventArgs<TimerEventInfo> e)
  47. {
  48. await SendMessage(SessionMessageType.TimerCreated, e.Argument).ConfigureAwait(false);
  49. }
  50. private async void OnLiveTvManagerSeriesTimerCancelled(object sender, GenericEventArgs<TimerEventInfo> e)
  51. {
  52. await SendMessage(SessionMessageType.SeriesTimerCancelled, e.Argument).ConfigureAwait(false);
  53. }
  54. private async void OnLiveTvManagerTimerCancelled(object sender, GenericEventArgs<TimerEventInfo> e)
  55. {
  56. await SendMessage(SessionMessageType.TimerCancelled, e.Argument).ConfigureAwait(false);
  57. }
  58. private async Task SendMessage(SessionMessageType name, TimerEventInfo info)
  59. {
  60. var users = _userManager.Users.Where(i => i.HasPermission(PermissionKind.EnableLiveTvAccess)).Select(i => i.Id).ToList();
  61. try
  62. {
  63. await _sessionManager.SendMessageToUserSessions(users, name, info, CancellationToken.None).ConfigureAwait(false);
  64. }
  65. catch (Exception ex)
  66. {
  67. _logger.LogError(ex, "Error sending message");
  68. }
  69. }
  70. /// <inheritdoc />
  71. public void Dispose()
  72. {
  73. _liveTvManager.TimerCancelled -= OnLiveTvManagerTimerCancelled;
  74. _liveTvManager.SeriesTimerCancelled -= OnLiveTvManagerSeriesTimerCancelled;
  75. _liveTvManager.TimerCreated -= OnLiveTvManagerTimerCreated;
  76. _liveTvManager.SeriesTimerCreated -= OnLiveTvManagerSeriesTimerCreated;
  77. }
  78. }
  79. }