SubtitleScheduledTask.cs 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166
  1. using MediaBrowser.Common.Configuration;
  2. using MediaBrowser.Controller.Configuration;
  3. using MediaBrowser.Controller.Entities;
  4. using MediaBrowser.Controller.Entities.Movies;
  5. using MediaBrowser.Controller.Entities.TV;
  6. using MediaBrowser.Controller.Library;
  7. using MediaBrowser.Controller.Subtitles;
  8. using MediaBrowser.Model.Entities;
  9. using MediaBrowser.Model.Logging;
  10. using MediaBrowser.Model.Providers;
  11. using System;
  12. using System.Collections.Generic;
  13. using System.Linq;
  14. using System.Threading;
  15. using System.Threading.Tasks;
  16. using System.IO;
  17. using MediaBrowser.Controller.Dto;
  18. using MediaBrowser.Model.Serialization;
  19. using MediaBrowser.Model.Tasks;
  20. namespace MediaBrowser.Providers.MediaInfo
  21. {
  22. public class SubtitleScheduledTask : IScheduledTask
  23. {
  24. private readonly ILibraryManager _libraryManager;
  25. private readonly IServerConfigurationManager _config;
  26. private readonly ISubtitleManager _subtitleManager;
  27. private readonly IMediaSourceManager _mediaSourceManager;
  28. private readonly ILogger _logger;
  29. private readonly IJsonSerializer _json;
  30. public SubtitleScheduledTask(ILibraryManager libraryManager, IJsonSerializer json, IServerConfigurationManager config, ISubtitleManager subtitleManager, ILogger logger, IMediaSourceManager mediaSourceManager)
  31. {
  32. _libraryManager = libraryManager;
  33. _config = config;
  34. _subtitleManager = subtitleManager;
  35. _logger = logger;
  36. _mediaSourceManager = mediaSourceManager;
  37. _json = json;
  38. }
  39. public string Name
  40. {
  41. get { return "Download missing subtitles"; }
  42. }
  43. public string Description
  44. {
  45. get { return "Searches the internet for missing subtitles based on metadata configuration."; }
  46. }
  47. public string Category
  48. {
  49. get { return "Library"; }
  50. }
  51. private SubtitleOptions GetOptions()
  52. {
  53. return _config.GetConfiguration<SubtitleOptions>("subtitles");
  54. }
  55. public async Task Execute(CancellationToken cancellationToken, IProgress<double> progress)
  56. {
  57. var options = GetOptions();
  58. var types = new List<string>();
  59. if (options.DownloadEpisodeSubtitles)
  60. {
  61. types.Add("Episode");
  62. }
  63. if (options.DownloadMovieSubtitles)
  64. {
  65. types.Add("Movie");
  66. }
  67. if (types.Count == 0)
  68. {
  69. return;
  70. }
  71. var videos = _libraryManager.GetItemList(new InternalItemsQuery
  72. {
  73. MediaTypes = new string[] { MediaType.Video },
  74. IsVirtualItem = false,
  75. IncludeItemTypes = types.ToArray(),
  76. DtoOptions = new DtoOptions(true)
  77. }).OfType<Video>()
  78. .Where(i => i.LocationType != LocationType.Remote)
  79. .ToList();
  80. if (videos.Count == 0)
  81. {
  82. return;
  83. }
  84. var numComplete = 0;
  85. foreach (var video in videos)
  86. {
  87. try
  88. {
  89. await DownloadSubtitles(video, options, cancellationToken).ConfigureAwait(false);
  90. }
  91. catch (Exception ex)
  92. {
  93. _logger.ErrorException("Error downloading subtitles for {0}", ex, video.Path);
  94. }
  95. // Update progress
  96. numComplete++;
  97. double percent = numComplete;
  98. percent /= videos.Count;
  99. progress.Report(100 * percent);
  100. }
  101. }
  102. private async Task<bool> DownloadSubtitles(Video video, SubtitleOptions options, CancellationToken cancellationToken)
  103. {
  104. if ((options.DownloadEpisodeSubtitles &&
  105. video is Episode) ||
  106. (options.DownloadMovieSubtitles &&
  107. video is Movie))
  108. {
  109. var mediaStreams = _mediaSourceManager.GetStaticMediaSources(video, false).First().MediaStreams;
  110. var downloadedLanguages = await new SubtitleDownloader(_logger,
  111. _subtitleManager)
  112. .DownloadSubtitles(video,
  113. mediaStreams,
  114. options.SkipIfEmbeddedSubtitlesPresent,
  115. options.SkipIfAudioTrackMatches,
  116. options.RequirePerfectMatch,
  117. options.DownloadLanguages,
  118. cancellationToken).ConfigureAwait(false);
  119. // Rescan
  120. if (downloadedLanguages.Count > 0)
  121. {
  122. await video.RefreshMetadata(cancellationToken).ConfigureAwait(false);
  123. return false;
  124. }
  125. return true;
  126. }
  127. return false;
  128. }
  129. public IEnumerable<TaskTriggerInfo> GetDefaultTriggers()
  130. {
  131. return new[] {
  132. // Every so often
  133. new TaskTriggerInfo { Type = TaskTriggerInfo.TriggerInterval, IntervalTicks = TimeSpan.FromHours(24).Ticks}
  134. };
  135. }
  136. public string Key
  137. {
  138. get { return "DownloadSubtitles"; }
  139. }
  140. }
  141. }