RecordingNotifier.cs 2.9 KB

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