mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 03:26:07 +02:00
102 lines
3.5 KiB
C#
102 lines
3.5 KiB
C#
using MediaBrowser.Controller.Channels;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Model.Entities;
|
|
using System;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using MediaBrowser.Controller.Entities.Movies;
|
|
using MediaBrowser.Controller.Entities.TV;
|
|
|
|
namespace MediaBrowser.Server.Implementations.Library
|
|
{
|
|
public class LocalTrailerPostScanTask : ILibraryPostScanTask
|
|
{
|
|
private readonly ILibraryManager _libraryManager;
|
|
private readonly IChannelManager _channelManager;
|
|
|
|
public LocalTrailerPostScanTask(ILibraryManager libraryManager, IChannelManager channelManager)
|
|
{
|
|
_libraryManager = libraryManager;
|
|
_channelManager = channelManager;
|
|
}
|
|
|
|
public async Task Run(IProgress<double> progress, CancellationToken cancellationToken)
|
|
{
|
|
var items = _libraryManager.GetItemList(new InternalItemsQuery
|
|
{
|
|
IncludeItemTypes = new[] { typeof(BoxSet).Name, typeof(Game).Name, typeof(Movie).Name, typeof(Series).Name },
|
|
Recursive = true
|
|
|
|
}).OfType<IHasTrailers>().ToList();
|
|
|
|
var trailerTypes = Enum.GetNames(typeof(TrailerType))
|
|
.Select(i => (TrailerType)Enum.Parse(typeof(TrailerType), i, true))
|
|
.Except(new[] { TrailerType.LocalTrailer })
|
|
.ToArray();
|
|
|
|
var trailers = _libraryManager.GetItemList(new InternalItemsQuery
|
|
{
|
|
IncludeItemTypes = new[] { typeof(Trailer).Name },
|
|
TrailerTypes = trailerTypes,
|
|
Recursive = true
|
|
|
|
}).ToArray();
|
|
|
|
var numComplete = 0;
|
|
|
|
foreach (var item in items)
|
|
{
|
|
cancellationToken.ThrowIfCancellationRequested();
|
|
|
|
await AssignTrailers(item, trailers).ConfigureAwait(false);
|
|
|
|
numComplete++;
|
|
double percent = numComplete;
|
|
percent /= items.Count;
|
|
progress.Report(percent * 100);
|
|
}
|
|
|
|
progress.Report(100);
|
|
}
|
|
|
|
private async Task AssignTrailers(IHasTrailers item, BaseItem[] channelTrailers)
|
|
{
|
|
if (item is Game)
|
|
{
|
|
return;
|
|
}
|
|
|
|
var imdbId = item.GetProviderId(MetadataProviders.Imdb);
|
|
var tmdbId = item.GetProviderId(MetadataProviders.Tmdb);
|
|
|
|
var trailers = channelTrailers.Where(i =>
|
|
{
|
|
if (!string.IsNullOrWhiteSpace(imdbId) &&
|
|
string.Equals(imdbId, i.GetProviderId(MetadataProviders.Imdb), StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return true;
|
|
}
|
|
if (!string.IsNullOrWhiteSpace(tmdbId) &&
|
|
string.Equals(tmdbId, i.GetProviderId(MetadataProviders.Tmdb), StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return true;
|
|
}
|
|
return false;
|
|
});
|
|
|
|
var trailerIds = trailers.Select(i => i.Id)
|
|
.ToList();
|
|
|
|
if (!trailerIds.SequenceEqual(item.RemoteTrailerIds))
|
|
{
|
|
item.RemoteTrailerIds = trailerIds;
|
|
|
|
var baseItem = (BaseItem)item;
|
|
await baseItem.UpdateToRepository(ItemUpdateType.MetadataImport, CancellationToken.None)
|
|
.ConfigureAwait(false);
|
|
}
|
|
}
|
|
}
|
|
}
|