ChapterImagesTask.cs 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122
  1. using MediaBrowser.Common.ScheduledTasks;
  2. using MediaBrowser.Controller;
  3. using MediaBrowser.Controller.Entities;
  4. using MediaBrowser.Model.Logging;
  5. using System;
  6. using System.Collections.Generic;
  7. using System.Linq;
  8. using System.Threading;
  9. using System.Threading.Tasks;
  10. namespace MediaBrowser.Server.Implementations.ScheduledTasks.Tasks
  11. {
  12. /// <summary>
  13. /// Class ChapterImagesTask
  14. /// </summary>
  15. class ChapterImagesTask : IScheduledTask
  16. {
  17. /// <summary>
  18. /// The _kernel
  19. /// </summary>
  20. private readonly Kernel _kernel;
  21. /// <summary>
  22. /// The _logger
  23. /// </summary>
  24. private readonly ILogger _logger;
  25. /// <summary>
  26. /// Initializes a new instance of the <see cref="ChapterImagesTask" /> class.
  27. /// </summary>
  28. /// <param name="kernel">The kernel.</param>
  29. /// <param name="logger">The logger.</param>
  30. public ChapterImagesTask(Kernel kernel, ILogger logger)
  31. {
  32. _kernel = kernel;
  33. _logger = logger;
  34. }
  35. /// <summary>
  36. /// Creates the triggers that define when the task will run
  37. /// </summary>
  38. /// <returns>IEnumerable{BaseTaskTrigger}.</returns>
  39. public IEnumerable<ITaskTrigger> GetDefaultTriggers()
  40. {
  41. return new ITaskTrigger[]
  42. {
  43. new DailyTrigger { TimeOfDay = TimeSpan.FromHours(4) }
  44. };
  45. }
  46. /// <summary>
  47. /// Returns the task to be executed
  48. /// </summary>
  49. /// <param name="cancellationToken">The cancellation token.</param>
  50. /// <param name="progress">The progress.</param>
  51. /// <returns>Task.</returns>
  52. public Task Execute(CancellationToken cancellationToken, IProgress<double> progress)
  53. {
  54. var videos = _kernel.RootFolder.RecursiveChildren.OfType<Video>().Where(v => v.Chapters != null).ToList();
  55. var numComplete = 0;
  56. var tasks = videos.Select(v => Task.Run(async () =>
  57. {
  58. try
  59. {
  60. await _kernel.FFMpegManager.PopulateChapterImages(v, cancellationToken, true, true);
  61. }
  62. catch (OperationCanceledException)
  63. {
  64. throw;
  65. }
  66. catch (Exception ex)
  67. {
  68. _logger.ErrorException("Error creating chapter images for {0}", ex, v.Name);
  69. }
  70. finally
  71. {
  72. lock (progress)
  73. {
  74. numComplete++;
  75. double percent = numComplete;
  76. percent /= videos.Count;
  77. progress.Report(100 * percent);
  78. }
  79. }
  80. }));
  81. return Task.WhenAll(tasks);
  82. }
  83. /// <summary>
  84. /// Gets the name of the task
  85. /// </summary>
  86. /// <value>The name.</value>
  87. public string Name
  88. {
  89. get { return "Create video chapter thumbnails"; }
  90. }
  91. /// <summary>
  92. /// Gets the description.
  93. /// </summary>
  94. /// <value>The description.</value>
  95. public string Description
  96. {
  97. get { return "Creates thumbnails for videos that have chapters."; }
  98. }
  99. /// <summary>
  100. /// Gets the category.
  101. /// </summary>
  102. /// <value>The category.</value>
  103. public string Category
  104. {
  105. get
  106. {
  107. return "Library";
  108. }
  109. }
  110. }
  111. }