jellyfin/MediaBrowser.Server.Implementations/ScheduledTasks/ChapterImagesTask.cs

198 lines
6.3 KiB
C#
Raw Normal View History

2014-02-20 16:37:41 +00:00
using MediaBrowser.Common.Configuration;
using MediaBrowser.Common.ScheduledTasks;
2013-02-21 01:33:05 +00:00
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Library;
2014-02-20 16:37:41 +00:00
using MediaBrowser.Controller.MediaEncoding;
2013-06-18 19:16:27 +00:00
using MediaBrowser.Controller.Persistence;
2013-02-23 07:57:11 +00:00
using MediaBrowser.Model.Logging;
2013-02-21 01:33:05 +00:00
using System;
using System.Collections.Generic;
using System.IO;
2013-02-21 01:33:05 +00:00
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
2015-10-04 04:23:11 +00:00
using CommonIO;
2016-05-09 04:56:41 +00:00
using MediaBrowser.Model.Entities;
2013-02-21 01:33:05 +00:00
2013-03-05 03:34:02 +00:00
namespace MediaBrowser.Server.Implementations.ScheduledTasks
2013-02-21 01:33:05 +00:00
{
2013-02-23 07:57:11 +00:00
/// <summary>
/// Class ChapterImagesTask
/// </summary>
class ChapterImagesTask : IScheduledTask
2013-02-21 01:33:05 +00:00
{
/// <summary>
/// The _logger
/// </summary>
private readonly ILogger _logger;
/// <summary>
/// The _library manager
/// </summary>
private readonly ILibraryManager _libraryManager;
/// <summary>
/// The current new item timer
/// </summary>
/// <value>The new item timer.</value>
private Timer NewItemTimer { get; set; }
2013-06-18 19:16:27 +00:00
private readonly IItemRepository _itemRepo;
2014-02-20 16:37:41 +00:00
private readonly IApplicationPaths _appPaths;
private readonly IEncodingManager _encodingManager;
2015-09-13 23:07:54 +00:00
private readonly IFileSystem _fileSystem;
2014-02-20 16:37:41 +00:00
2013-02-23 07:57:11 +00:00
/// <summary>
/// Initializes a new instance of the <see cref="ChapterImagesTask" /> class.
/// </summary>
/// <param name="logManager">The log manager.</param>
/// <param name="libraryManager">The library manager.</param>
2013-06-18 19:16:27 +00:00
/// <param name="itemRepo">The item repo.</param>
2015-09-13 23:07:54 +00:00
public ChapterImagesTask(ILogManager logManager, ILibraryManager libraryManager, IItemRepository itemRepo, IApplicationPaths appPaths, IEncodingManager encodingManager, IFileSystem fileSystem)
2013-02-23 07:57:11 +00:00
{
_logger = logManager.GetLogger(GetType().Name);
_libraryManager = libraryManager;
2013-06-18 19:16:27 +00:00
_itemRepo = itemRepo;
2014-02-20 16:37:41 +00:00
_appPaths = appPaths;
_encodingManager = encodingManager;
2015-09-13 23:07:54 +00:00
_fileSystem = fileSystem;
}
2013-02-21 01:33:05 +00:00
/// <summary>
/// Creates the triggers that define when the task will run
/// </summary>
/// <returns>IEnumerable{BaseTaskTrigger}.</returns>
public IEnumerable<ITaskTrigger> GetDefaultTriggers()
2013-02-21 01:33:05 +00:00
{
2013-02-24 21:53:54 +00:00
return new ITaskTrigger[]
2013-02-21 01:33:05 +00:00
{
new DailyTrigger
{
2015-03-07 06:51:09 +00:00
TimeOfDay = TimeSpan.FromHours(1),
TaskOptions = new TaskExecutionOptions
{
2015-03-07 06:51:09 +00:00
MaxRuntimeMs = Convert.ToInt32(TimeSpan.FromHours(4).TotalMilliseconds)
}
}
2013-02-21 01:33:05 +00:00
};
}
/// <summary>
/// Returns the task to be executed
/// </summary>
/// <param name="cancellationToken">The cancellation token.</param>
/// <param name="progress">The progress.</param>
/// <returns>Task.</returns>
2013-04-15 18:45:58 +00:00
public async Task Execute(CancellationToken cancellationToken, IProgress<double> progress)
2013-02-21 01:33:05 +00:00
{
2016-05-09 04:56:41 +00:00
var videos = _libraryManager.GetItemList(new InternalItemsQuery
{
MediaTypes = new[] { MediaType.Video },
IsFolder = false,
Recursive = true
})
.OfType<Video>()
2013-04-15 18:45:58 +00:00
.ToList();
2013-02-21 01:33:05 +00:00
var numComplete = 0;
2014-02-20 16:37:41 +00:00
var failHistoryPath = Path.Combine(_appPaths.CachePath, "chapter-failures.txt");
List<string> previouslyFailedImages;
try
{
2016-05-09 04:56:41 +00:00
previouslyFailedImages = _fileSystem.ReadAllText(failHistoryPath)
.Split(new[] { '|' }, StringSplitOptions.RemoveEmptyEntries)
.ToList();
}
catch (FileNotFoundException)
{
previouslyFailedImages = new List<string>();
}
2013-09-25 15:11:23 +00:00
catch (DirectoryNotFoundException)
{
previouslyFailedImages = new List<string>();
}
2013-04-15 18:45:58 +00:00
foreach (var video in videos)
2013-02-21 01:33:05 +00:00
{
2013-04-15 18:45:58 +00:00
cancellationToken.ThrowIfCancellationRequested();
var key = video.Path + video.DateModified.Ticks;
var extract = !previouslyFailedImages.Contains(key, StringComparer.OrdinalIgnoreCase);
2016-03-17 16:39:39 +00:00
try
2014-02-20 16:37:41 +00:00
{
2016-03-17 16:39:39 +00:00
var chapters = _itemRepo.GetChapters(video.Id).ToList();
2014-02-20 16:37:41 +00:00
2016-03-17 16:39:39 +00:00
var success = await _encodingManager.RefreshChapterImages(new ChapterImageRefreshOptions
{
SaveChapters = true,
ExtractImages = extract,
Video = video,
Chapters = chapters
2016-03-17 16:39:39 +00:00
}, CancellationToken.None);
2013-06-04 02:02:49 +00:00
2016-03-17 16:39:39 +00:00
if (!success)
{
previouslyFailedImages.Add(key);
2013-06-04 02:02:49 +00:00
2016-03-17 16:39:39 +00:00
var parentPath = Path.GetDirectoryName(failHistoryPath);
2013-06-04 02:02:49 +00:00
2016-03-17 16:39:39 +00:00
_fileSystem.CreateDirectory(parentPath);
2013-02-21 01:33:05 +00:00
2016-03-17 16:39:39 +00:00
_fileSystem.WriteAllText(failHistoryPath, string.Join("|", previouslyFailedImages.ToArray()));
}
numComplete++;
double percent = numComplete;
percent /= videos.Count;
2013-02-21 01:33:05 +00:00
2016-03-17 16:39:39 +00:00
progress.Report(100 * percent);
}
catch (ObjectDisposedException)
{
break;
}
2013-04-15 18:45:58 +00:00
}
2013-02-21 01:33:05 +00:00
}
/// <summary>
/// Gets the name of the task
/// </summary>
/// <value>The name.</value>
public string Name
2013-02-21 01:33:05 +00:00
{
2013-09-05 17:05:39 +00:00
get
{
return "Chapter image extraction";
}
2013-02-21 01:33:05 +00:00
}
/// <summary>
/// Gets the description.
/// </summary>
/// <value>The description.</value>
public string Description
2013-02-21 01:33:05 +00:00
{
get { return "Creates thumbnails for videos that have chapters."; }
}
/// <summary>
/// Gets the category.
/// </summary>
/// <value>The category.</value>
public string Category
2013-02-21 01:33:05 +00:00
{
get
{
return "Library";
}
}
}
}