mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-08 04:18:06 +02:00
214 lines
7.5 KiB
C#
214 lines
7.5 KiB
C#
#pragma warning disable CS1591
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Globalization;
|
|
using System.Linq;
|
|
using System.Net.Http;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using MediaBrowser.Common.Net;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Entities.TV;
|
|
using MediaBrowser.Controller.Providers;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.Providers;
|
|
|
|
namespace MediaBrowser.Providers.Plugins.Tmdb.TV
|
|
{
|
|
public class TmdbEpisodeProvider : IRemoteMetadataProvider<Episode, EpisodeInfo>, IHasOrder
|
|
{
|
|
private readonly IHttpClientFactory _httpClientFactory;
|
|
private readonly TmdbClientManager _tmdbClientManager;
|
|
|
|
public TmdbEpisodeProvider(IHttpClientFactory httpClientFactory, TmdbClientManager tmdbClientManager)
|
|
{
|
|
_httpClientFactory = httpClientFactory;
|
|
_tmdbClientManager = tmdbClientManager;
|
|
}
|
|
|
|
// After TheTvDb
|
|
public int Order => 1;
|
|
|
|
public string Name => TmdbUtils.ProviderName;
|
|
|
|
public async Task<IEnumerable<RemoteSearchResult>> GetSearchResults(EpisodeInfo searchInfo, CancellationToken cancellationToken)
|
|
{
|
|
// The search query must either provide an episode number or date
|
|
if (!searchInfo.IndexNumber.HasValue || !searchInfo.ParentIndexNumber.HasValue)
|
|
{
|
|
return Enumerable.Empty<RemoteSearchResult>();
|
|
}
|
|
|
|
var metadataResult = await GetMetadata(searchInfo, cancellationToken).ConfigureAwait(false);
|
|
|
|
if (!metadataResult.HasMetadata)
|
|
{
|
|
return Enumerable.Empty<RemoteSearchResult>();
|
|
}
|
|
|
|
var item = metadataResult.Item;
|
|
|
|
return new[]
|
|
{
|
|
new RemoteSearchResult
|
|
{
|
|
IndexNumber = item.IndexNumber,
|
|
Name = item.Name,
|
|
ParentIndexNumber = item.ParentIndexNumber,
|
|
PremiereDate = item.PremiereDate,
|
|
ProductionYear = item.ProductionYear,
|
|
ProviderIds = item.ProviderIds,
|
|
SearchProviderName = Name,
|
|
IndexNumberEnd = item.IndexNumberEnd
|
|
}
|
|
};
|
|
}
|
|
|
|
public async Task<MetadataResult<Episode>> GetMetadata(EpisodeInfo info, CancellationToken cancellationToken)
|
|
{
|
|
var metadataResult = new MetadataResult<Episode>();
|
|
|
|
// Allowing this will dramatically increase scan times
|
|
if (info.IsMissingEpisode)
|
|
{
|
|
return metadataResult;
|
|
}
|
|
|
|
info.SeriesProviderIds.TryGetValue(MetadataProvider.Tmdb.ToString(), out string tmdbId);
|
|
|
|
var seriesTmdbId = Convert.ToInt32(tmdbId, CultureInfo.InvariantCulture);
|
|
if (seriesTmdbId <= 0)
|
|
{
|
|
return metadataResult;
|
|
}
|
|
|
|
var seasonNumber = info.ParentIndexNumber;
|
|
var episodeNumber = info.IndexNumber;
|
|
|
|
if (!seasonNumber.HasValue || !episodeNumber.HasValue)
|
|
{
|
|
return metadataResult;
|
|
}
|
|
|
|
var episodeResult = await _tmdbClientManager
|
|
.GetEpisodeAsync(seriesTmdbId, seasonNumber.Value, episodeNumber.Value, info.SeriesDisplayOrder, info.MetadataLanguage, TmdbUtils.GetImageLanguagesParam(info.MetadataLanguage), cancellationToken)
|
|
.ConfigureAwait(false);
|
|
|
|
if (episodeResult == null)
|
|
{
|
|
return metadataResult;
|
|
}
|
|
|
|
metadataResult.HasMetadata = true;
|
|
metadataResult.QueriedById = true;
|
|
|
|
if (!string.IsNullOrEmpty(episodeResult.Overview))
|
|
{
|
|
// if overview is non-empty, we can assume that localized data was returned
|
|
metadataResult.ResultLanguage = info.MetadataLanguage;
|
|
}
|
|
|
|
var item = new Episode
|
|
{
|
|
IndexNumber = info.IndexNumber,
|
|
ParentIndexNumber = info.ParentIndexNumber,
|
|
IndexNumberEnd = info.IndexNumberEnd,
|
|
Name = episodeResult.Name,
|
|
PremiereDate = episodeResult.AirDate,
|
|
ProductionYear = episodeResult.AirDate?.Year,
|
|
Overview = episodeResult.Overview,
|
|
CommunityRating = Convert.ToSingle(episodeResult.VoteAverage)
|
|
};
|
|
|
|
var externalIds = episodeResult.ExternalIds;
|
|
if (!string.IsNullOrEmpty(externalIds?.TvdbId))
|
|
{
|
|
item.SetProviderId(MetadataProvider.Tvdb, externalIds.TvdbId);
|
|
}
|
|
|
|
if (!string.IsNullOrEmpty(externalIds?.ImdbId))
|
|
{
|
|
item.SetProviderId(MetadataProvider.Imdb, externalIds.ImdbId);
|
|
}
|
|
|
|
if (!string.IsNullOrEmpty(externalIds?.TvrageId))
|
|
{
|
|
item.SetProviderId(MetadataProvider.TvRage, externalIds.TvrageId);
|
|
}
|
|
|
|
if (episodeResult.Videos?.Results != null)
|
|
{
|
|
foreach (var video in episodeResult.Videos.Results)
|
|
{
|
|
if (TmdbUtils.IsTrailerType(video))
|
|
{
|
|
item.AddTrailerUrl("https://www.youtube.com/watch?v=" + video.Key);
|
|
}
|
|
}
|
|
}
|
|
|
|
var credits = episodeResult.Credits;
|
|
|
|
if (credits?.Cast != null)
|
|
{
|
|
foreach (var actor in credits.Cast.OrderBy(a => a.Order).Take(TmdbUtils.MaxCastMembers))
|
|
{
|
|
metadataResult.AddPerson(new PersonInfo
|
|
{
|
|
Name = actor.Name.Trim(),
|
|
Role = actor.Character,
|
|
Type = PersonType.Actor,
|
|
SortOrder = actor.Order
|
|
});
|
|
}
|
|
}
|
|
|
|
if (credits?.GuestStars != null)
|
|
{
|
|
foreach (var guest in credits.GuestStars.OrderBy(a => a.Order).Take(TmdbUtils.MaxCastMembers))
|
|
{
|
|
metadataResult.AddPerson(new PersonInfo
|
|
{
|
|
Name = guest.Name.Trim(),
|
|
Role = guest.Character,
|
|
Type = PersonType.GuestStar,
|
|
SortOrder = guest.Order
|
|
});
|
|
}
|
|
}
|
|
|
|
// and the rest from crew
|
|
if (credits?.Crew != null)
|
|
{
|
|
foreach (var person in credits.Crew)
|
|
{
|
|
// Normalize this
|
|
var type = TmdbUtils.MapCrewToPersonType(person);
|
|
|
|
if (!TmdbUtils.WantedCrewTypes.Contains(type, StringComparer.OrdinalIgnoreCase)
|
|
&& !TmdbUtils.WantedCrewTypes.Contains(person.Job ?? string.Empty, StringComparer.OrdinalIgnoreCase))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
metadataResult.AddPerson(new PersonInfo
|
|
{
|
|
Name = person.Name.Trim(),
|
|
Role = person.Job,
|
|
Type = type
|
|
});
|
|
}
|
|
}
|
|
|
|
metadataResult.Item = item;
|
|
|
|
return metadataResult;
|
|
}
|
|
|
|
public Task<HttpResponseMessage> GetImageResponse(string url, CancellationToken cancellationToken)
|
|
{
|
|
return _httpClientFactory.CreateClient(NamedClient.Default).GetAsync(url, cancellationToken);
|
|
}
|
|
}
|
|
}
|