RecordingNotifier.cs 3.1 KB

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