RecordingNotifier.cs 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  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 Microsoft.Extensions.Logging;
  9. namespace Emby.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, MediaBrowser.Model.Events.GenericEventArgs<TimerEventInfo> e)
  32. {
  33. SendMessage("SeriesTimerCreated", e.Argument);
  34. }
  35. private void _liveTvManager_TimerCreated(object sender, MediaBrowser.Model.Events.GenericEventArgs<TimerEventInfo> e)
  36. {
  37. SendMessage("TimerCreated", e.Argument);
  38. }
  39. private void _liveTvManager_SeriesTimerCancelled(object sender, MediaBrowser.Model.Events.GenericEventArgs<TimerEventInfo> e)
  40. {
  41. SendMessage("SeriesTimerCancelled", e.Argument);
  42. }
  43. private void _liveTvManager_TimerCancelled(object sender, MediaBrowser.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).ToList();
  50. try
  51. {
  52. await _sessionManager.SendMessageToUserSessions(users, name, info, CancellationToken.None);
  53. }
  54. catch (ObjectDisposedException)
  55. {
  56. // TODO Log exception or Investigate and properly fix.
  57. }
  58. catch (Exception ex)
  59. {
  60. _logger.LogError(ex, "Error sending message");
  61. }
  62. }
  63. public void Dispose()
  64. {
  65. _liveTvManager.TimerCancelled -= _liveTvManager_TimerCancelled;
  66. _liveTvManager.SeriesTimerCancelled -= _liveTvManager_SeriesTimerCancelled;
  67. _liveTvManager.TimerCreated -= _liveTvManager_TimerCreated;
  68. _liveTvManager.SeriesTimerCreated -= _liveTvManager_SeriesTimerCreated;
  69. }
  70. }
  71. }