2019-01-13 14:54:44 -05:00
|
|
|
using System;
|
2013-02-20 20:33:05 -05:00
|
|
|
using System.Collections.Generic;
|
2013-05-25 01:17:32 -04:00
|
|
|
using System.IO;
|
2013-02-20 20:33:05 -05:00
|
|
|
using System.Linq;
|
|
|
|
using System.Threading;
|
|
|
|
using System.Threading.Tasks;
|
2021-12-20 07:31:07 -05:00
|
|
|
using Jellyfin.Extensions;
|
2019-01-13 14:22:24 -05:00
|
|
|
using MediaBrowser.Common.Configuration;
|
2017-05-21 03:25:49 -04:00
|
|
|
using MediaBrowser.Controller.Dto;
|
2019-01-13 14:22:24 -05:00
|
|
|
using MediaBrowser.Controller.Entities;
|
|
|
|
using MediaBrowser.Controller.Library;
|
|
|
|
using MediaBrowser.Controller.MediaEncoding;
|
|
|
|
using MediaBrowser.Controller.Persistence;
|
|
|
|
using MediaBrowser.Controller.Providers;
|
2016-05-09 00:56:41 -04:00
|
|
|
using MediaBrowser.Model.Entities;
|
2021-04-17 06:37:55 -04:00
|
|
|
using MediaBrowser.Model.Globalization;
|
2019-01-13 14:22:24 -05:00
|
|
|
using MediaBrowser.Model.IO;
|
2016-10-23 15:14:57 -04:00
|
|
|
using MediaBrowser.Model.Tasks;
|
2022-11-23 09:58:11 -05:00
|
|
|
using Microsoft.Extensions.Logging;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
2021-12-15 12:25:36 -05:00
|
|
|
namespace Emby.Server.Implementations.ScheduledTasks.Tasks
|
2013-02-20 20:33:05 -05:00
|
|
|
{
|
2013-02-23 02:57:11 -05:00
|
|
|
/// <summary>
|
2019-09-10 16:37:53 -04:00
|
|
|
/// Class ChapterImagesTask.
|
2013-02-23 02:57:11 -05:00
|
|
|
/// </summary>
|
2019-02-15 17:05:14 -05:00
|
|
|
public class ChapterImagesTask : IScheduledTask
|
2013-02-20 20:33:05 -05:00
|
|
|
{
|
2022-11-27 08:35:07 -05:00
|
|
|
private readonly ILogger<ChapterImagesTask> _logger;
|
2013-02-28 14:32:41 -05:00
|
|
|
private readonly ILibraryManager _libraryManager;
|
2013-06-18 15:16:27 -04:00
|
|
|
private readonly IItemRepository _itemRepo;
|
2014-02-20 11:37:41 -05:00
|
|
|
private readonly IApplicationPaths _appPaths;
|
|
|
|
private readonly IEncodingManager _encodingManager;
|
2015-09-13 19:07:54 -04:00
|
|
|
private readonly IFileSystem _fileSystem;
|
2020-03-26 15:28:30 -04:00
|
|
|
private readonly ILocalizationManager _localization;
|
2014-02-20 11:37:41 -05:00
|
|
|
|
2013-02-23 02:57:11 -05:00
|
|
|
/// <summary>
|
|
|
|
/// Initializes a new instance of the <see cref="ChapterImagesTask" /> class.
|
|
|
|
/// </summary>
|
2022-11-23 09:58:11 -05:00
|
|
|
/// <param name="logger">The logger.</param>.
|
2021-10-02 12:53:51 -04:00
|
|
|
/// <param name="libraryManager">The library manager.</param>.
|
|
|
|
/// <param name="itemRepo">The item repository.</param>
|
|
|
|
/// <param name="appPaths">The application paths.</param>
|
|
|
|
/// <param name="encodingManager">The encoding manager.</param>
|
|
|
|
/// <param name="fileSystem">The filesystem.</param>
|
|
|
|
/// <param name="localization">The localization manager.</param>
|
2020-03-26 17:26:25 -04:00
|
|
|
public ChapterImagesTask(
|
2022-11-23 09:58:11 -05:00
|
|
|
ILogger<ChapterImagesTask> logger,
|
2020-03-26 17:26:25 -04:00
|
|
|
ILibraryManager libraryManager,
|
|
|
|
IItemRepository itemRepo,
|
|
|
|
IApplicationPaths appPaths,
|
|
|
|
IEncodingManager encodingManager,
|
|
|
|
IFileSystem fileSystem,
|
|
|
|
ILocalizationManager localization)
|
2013-02-23 02:57:11 -05:00
|
|
|
{
|
2022-11-23 09:58:11 -05:00
|
|
|
_logger = logger;
|
2013-02-28 14:32:41 -05:00
|
|
|
_libraryManager = libraryManager;
|
2013-06-18 15:16:27 -04:00
|
|
|
_itemRepo = itemRepo;
|
2014-02-20 11:37:41 -05:00
|
|
|
_appPaths = appPaths;
|
|
|
|
_encodingManager = encodingManager;
|
2015-09-13 19:07:54 -04:00
|
|
|
_fileSystem = fileSystem;
|
2020-03-26 17:26:25 -04:00
|
|
|
_localization = localization;
|
2013-05-03 11:08:02 -04:00
|
|
|
}
|
2013-07-09 12:11:16 -04:00
|
|
|
|
2021-07-05 18:01:33 -04:00
|
|
|
/// <inheritdoc />
|
|
|
|
public string Name => _localization.GetLocalizedString("TaskRefreshChapterImages");
|
|
|
|
|
|
|
|
/// <inheritdoc />
|
|
|
|
public string Description => _localization.GetLocalizedString("TaskRefreshChapterImagesDescription");
|
|
|
|
|
|
|
|
/// <inheritdoc />
|
|
|
|
public string Category => _localization.GetLocalizedString("TasksLibraryCategory");
|
|
|
|
|
|
|
|
/// <inheritdoc />
|
|
|
|
public string Key => "RefreshChapterImages";
|
|
|
|
|
|
|
|
/// <inheritdoc />
|
2016-10-23 15:14:57 -04:00
|
|
|
public IEnumerable<TaskTriggerInfo> GetDefaultTriggers()
|
2013-02-20 20:33:05 -05:00
|
|
|
{
|
2019-09-10 16:37:53 -04:00
|
|
|
return new[]
|
|
|
|
{
|
2016-10-23 15:14:57 -04:00
|
|
|
new TaskTriggerInfo
|
2013-02-20 20:33:05 -05:00
|
|
|
{
|
2016-10-23 15:14:57 -04:00
|
|
|
Type = TaskTriggerInfo.TriggerDaily,
|
2016-12-23 03:50:32 -05:00
|
|
|
TimeOfDayTicks = TimeSpan.FromHours(2).Ticks,
|
2018-09-12 13:26:21 -04:00
|
|
|
MaxRuntimeTicks = TimeSpan.FromHours(4).Ticks
|
2016-10-23 15:14:57 -04:00
|
|
|
}
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
2022-02-15 12:59:46 -05:00
|
|
|
/// <inheritdoc />
|
|
|
|
public async Task ExecuteAsync(IProgress<double> progress, CancellationToken cancellationToken)
|
2013-02-20 20:33:05 -05:00
|
|
|
{
|
2016-05-09 00:56:41 -04:00
|
|
|
var videos = _libraryManager.GetItemList(new InternalItemsQuery
|
|
|
|
{
|
|
|
|
MediaTypes = new[] { MediaType.Video },
|
|
|
|
IsFolder = false,
|
2017-05-21 03:25:49 -04:00
|
|
|
Recursive = true,
|
|
|
|
DtoOptions = new DtoOptions(false)
|
2017-09-09 23:18:23 -04:00
|
|
|
{
|
|
|
|
EnableImages = false
|
|
|
|
},
|
|
|
|
SourceTypes = new SourceType[] { SourceType.Library },
|
|
|
|
HasChapterImages = false,
|
|
|
|
IsVirtualItem = false
|
2016-05-09 00:56:41 -04:00
|
|
|
})
|
|
|
|
.OfType<Video>()
|
2013-04-15 14:45:58 -04:00
|
|
|
.ToList();
|
2013-02-20 20:33:05 -05:00
|
|
|
|
|
|
|
var numComplete = 0;
|
|
|
|
|
2014-02-20 11:37:41 -05:00
|
|
|
var failHistoryPath = Path.Combine(_appPaths.CachePath, "chapter-failures.txt");
|
2013-05-25 01:17:32 -04:00
|
|
|
|
|
|
|
List<string> previouslyFailedImages;
|
2013-07-09 12:11:16 -04:00
|
|
|
|
2019-01-27 11:00:17 -05:00
|
|
|
if (File.Exists(failHistoryPath))
|
2013-05-25 01:17:32 -04:00
|
|
|
{
|
2019-01-27 11:00:17 -05:00
|
|
|
try
|
|
|
|
{
|
2019-01-26 17:09:07 -05:00
|
|
|
previouslyFailedImages = File.ReadAllText(failHistoryPath)
|
2020-11-14 10:28:49 -05:00
|
|
|
.Split('|', StringSplitOptions.RemoveEmptyEntries)
|
2019-01-27 11:00:17 -05:00
|
|
|
.ToList();
|
|
|
|
}
|
|
|
|
catch (IOException)
|
|
|
|
{
|
|
|
|
previouslyFailedImages = new List<string>();
|
|
|
|
}
|
2013-05-25 01:17:32 -04:00
|
|
|
}
|
2019-01-27 11:00:17 -05:00
|
|
|
else
|
2013-09-25 11:11:23 -04:00
|
|
|
{
|
|
|
|
previouslyFailedImages = new List<string>();
|
|
|
|
}
|
2013-05-25 01:17:32 -04:00
|
|
|
|
2019-09-10 16:37:53 -04:00
|
|
|
var directoryService = new DirectoryService(_fileSystem);
|
2017-11-05 16:51:23 -05:00
|
|
|
|
2013-04-15 14:45:58 -04:00
|
|
|
foreach (var video in videos)
|
2013-02-20 20:33:05 -05:00
|
|
|
{
|
2013-04-15 14:45:58 -04:00
|
|
|
cancellationToken.ThrowIfCancellationRequested();
|
|
|
|
|
2013-05-25 01:17:32 -04:00
|
|
|
var key = video.Path + video.DateModified.Ticks;
|
|
|
|
|
2021-12-20 07:31:07 -05:00
|
|
|
var extract = !previouslyFailedImages.Contains(key, StringComparison.OrdinalIgnoreCase);
|
2013-05-25 01:17:32 -04:00
|
|
|
|
2016-03-17 12:39:39 -04:00
|
|
|
try
|
2014-02-20 11:37:41 -05:00
|
|
|
{
|
2018-09-12 13:26:21 -04:00
|
|
|
var chapters = _itemRepo.GetChapters(video);
|
2014-02-20 11:37:41 -05:00
|
|
|
|
2018-09-12 13:26:21 -04:00
|
|
|
var success = await _encodingManager.RefreshChapterImages(video, directoryService, chapters, extract, true, cancellationToken).ConfigureAwait(false);
|
2013-06-03 22:02:49 -04:00
|
|
|
|
2016-03-17 12:39:39 -04:00
|
|
|
if (!success)
|
|
|
|
{
|
|
|
|
previouslyFailedImages.Add(key);
|
2013-06-03 22:02:49 -04:00
|
|
|
|
2019-01-26 15:47:11 -05:00
|
|
|
var parentPath = Path.GetDirectoryName(failHistoryPath);
|
2022-12-05 09:01:13 -05:00
|
|
|
if (parentPath is not null)
|
2021-05-20 15:28:18 -04:00
|
|
|
{
|
|
|
|
Directory.CreateDirectory(parentPath);
|
|
|
|
}
|
2013-02-20 20:33:05 -05:00
|
|
|
|
2021-02-12 18:39:18 -05:00
|
|
|
string text = string.Join('|', previouslyFailedImages);
|
2019-01-26 17:09:07 -05:00
|
|
|
File.WriteAllText(failHistoryPath, text);
|
2016-03-17 12:39:39 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
numComplete++;
|
|
|
|
double percent = numComplete;
|
|
|
|
percent /= videos.Count;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
2016-03-17 12:39:39 -04:00
|
|
|
progress.Report(100 * percent);
|
|
|
|
}
|
2022-11-23 09:58:11 -05:00
|
|
|
catch (ObjectDisposedException ex)
|
2016-03-17 12:39:39 -04:00
|
|
|
{
|
2020-06-14 05:11:11 -04:00
|
|
|
// TODO Investigate and properly fix.
|
2022-11-23 09:58:11 -05:00
|
|
|
_logger.LogError(ex, "Object Disposed");
|
2016-03-17 12:39:39 -04:00
|
|
|
break;
|
|
|
|
}
|
2013-04-15 14:45:58 -04:00
|
|
|
}
|
2013-02-20 20:33:05 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|