mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-07 03:48:06 +02:00
126 lines
3.8 KiB
C#
126 lines
3.8 KiB
C#
using MediaBrowser.Common.ScheduledTasks;
|
|
using MediaBrowser.Controller;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Model.Logging;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace MediaBrowser.Server.Implementations.ScheduledTasks
|
|
{
|
|
/// <summary>
|
|
/// Class ChapterImagesTask
|
|
/// </summary>
|
|
class ChapterImagesTask : IScheduledTask
|
|
{
|
|
/// <summary>
|
|
/// The _kernel
|
|
/// </summary>
|
|
private readonly Kernel _kernel;
|
|
/// <summary>
|
|
/// The _logger
|
|
/// </summary>
|
|
private readonly ILogger _logger;
|
|
private readonly ILibraryManager _libraryManager;
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="ChapterImagesTask" /> class.
|
|
/// </summary>
|
|
/// <param name="kernel">The kernel.</param>
|
|
/// <param name="logger">The logger.</param>
|
|
/// <param name="libraryManager">The library manager.</param>
|
|
public ChapterImagesTask(Kernel kernel, ILogger logger, ILibraryManager libraryManager)
|
|
{
|
|
_kernel = kernel;
|
|
_logger = logger;
|
|
_libraryManager = libraryManager;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Creates the triggers that define when the task will run
|
|
/// </summary>
|
|
/// <returns>IEnumerable{BaseTaskTrigger}.</returns>
|
|
public IEnumerable<ITaskTrigger> GetDefaultTriggers()
|
|
{
|
|
return new ITaskTrigger[]
|
|
{
|
|
new DailyTrigger { TimeOfDay = TimeSpan.FromHours(4) }
|
|
};
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns the task to be executed
|
|
/// </summary>
|
|
/// <param name="cancellationToken">The cancellation token.</param>
|
|
/// <param name="progress">The progress.</param>
|
|
/// <returns>Task.</returns>
|
|
public Task Execute(CancellationToken cancellationToken, IProgress<double> progress)
|
|
{
|
|
var videos = _libraryManager.RootFolder.RecursiveChildren.OfType<Video>().Where(v => v.Chapters != null).ToList();
|
|
|
|
var numComplete = 0;
|
|
|
|
var tasks = videos.Select(v => Task.Run(async () =>
|
|
{
|
|
try
|
|
{
|
|
await _kernel.FFMpegManager.PopulateChapterImages(v, cancellationToken, true, true);
|
|
}
|
|
catch (OperationCanceledException)
|
|
{
|
|
throw;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.ErrorException("Error creating chapter images for {0}", ex, v.Name);
|
|
}
|
|
finally
|
|
{
|
|
lock (progress)
|
|
{
|
|
numComplete++;
|
|
double percent = numComplete;
|
|
percent /= videos.Count;
|
|
|
|
progress.Report(100 * percent);
|
|
}
|
|
}
|
|
}));
|
|
|
|
return Task.WhenAll(tasks);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the name of the task
|
|
/// </summary>
|
|
/// <value>The name.</value>
|
|
public string Name
|
|
{
|
|
get { return "Create video chapter thumbnails"; }
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the description.
|
|
/// </summary>
|
|
/// <value>The description.</value>
|
|
public string Description
|
|
{
|
|
get { return "Creates thumbnails for videos that have chapters."; }
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the category.
|
|
/// </summary>
|
|
/// <value>The category.</value>
|
|
public string Category
|
|
{
|
|
get
|
|
{
|
|
return "Library";
|
|
}
|
|
}
|
|
}
|
|
}
|