|
@@ -8,7 +8,6 @@ using MediaBrowser.Controller.Entities.TV;
|
|
|
using MediaBrowser.Controller.Library;
|
|
|
using MediaBrowser.Controller.Persistence;
|
|
|
using MediaBrowser.Controller.Providers;
|
|
|
-using MediaBrowser.Model.Chapters;
|
|
|
using MediaBrowser.Model.Configuration;
|
|
|
using MediaBrowser.Model.Entities;
|
|
|
using MediaBrowser.Model.Logging;
|
|
@@ -22,7 +21,6 @@ namespace MediaBrowser.Providers.Chapters
|
|
|
{
|
|
|
public class ChapterManager : IChapterManager
|
|
|
{
|
|
|
- private IChapterProvider[] _providers;
|
|
|
private readonly ILibraryManager _libraryManager;
|
|
|
private readonly ILogger _logger;
|
|
|
private readonly IServerConfigurationManager _config;
|
|
@@ -36,224 +34,6 @@ namespace MediaBrowser.Providers.Chapters
|
|
|
_itemRepo = itemRepo;
|
|
|
}
|
|
|
|
|
|
- public void AddParts(IEnumerable<IChapterProvider> chapterProviders)
|
|
|
- {
|
|
|
- _providers = chapterProviders.ToArray();
|
|
|
- }
|
|
|
-
|
|
|
- public Task<IEnumerable<RemoteChapterResult>> Search(Video video, CancellationToken cancellationToken)
|
|
|
- {
|
|
|
- VideoContentType mediaType;
|
|
|
-
|
|
|
- if (video is Episode)
|
|
|
- {
|
|
|
- mediaType = VideoContentType.Episode;
|
|
|
- }
|
|
|
- else if (video is Movie)
|
|
|
- {
|
|
|
- mediaType = VideoContentType.Movie;
|
|
|
- }
|
|
|
- else
|
|
|
- {
|
|
|
- // These are the only supported types
|
|
|
- return Task.FromResult<IEnumerable<RemoteChapterResult>>(new List<RemoteChapterResult>());
|
|
|
- }
|
|
|
-
|
|
|
- var request = new ChapterSearchRequest
|
|
|
- {
|
|
|
- ContentType = mediaType,
|
|
|
- IndexNumber = video.IndexNumber,
|
|
|
- Language = video.GetPreferredMetadataLanguage(),
|
|
|
- MediaPath = video.Path,
|
|
|
- Name = video.Name,
|
|
|
- ParentIndexNumber = video.ParentIndexNumber,
|
|
|
- ProductionYear = video.ProductionYear,
|
|
|
- ProviderIds = video.ProviderIds,
|
|
|
- RuntimeTicks = video.RunTimeTicks,
|
|
|
- SearchAllProviders = false
|
|
|
- };
|
|
|
-
|
|
|
- var episode = video as Episode;
|
|
|
-
|
|
|
- if (episode != null)
|
|
|
- {
|
|
|
- request.IndexNumberEnd = episode.IndexNumberEnd;
|
|
|
- request.SeriesName = episode.SeriesName;
|
|
|
- }
|
|
|
-
|
|
|
- return Search(request, cancellationToken);
|
|
|
- }
|
|
|
-
|
|
|
- public async Task<IEnumerable<RemoteChapterResult>> Search(ChapterSearchRequest request, CancellationToken cancellationToken)
|
|
|
- {
|
|
|
- var contentType = request.ContentType;
|
|
|
- var providers = GetInternalProviders(false)
|
|
|
- .Where(i => i.SupportedMediaTypes.Contains(contentType))
|
|
|
- .ToList();
|
|
|
-
|
|
|
- // If not searching all, search one at a time until something is found
|
|
|
- if (!request.SearchAllProviders)
|
|
|
- {
|
|
|
- foreach (var provider in providers)
|
|
|
- {
|
|
|
- try
|
|
|
- {
|
|
|
- var currentResults = await Search(request, provider, cancellationToken).ConfigureAwait(false);
|
|
|
-
|
|
|
- if (currentResults.Count > 0)
|
|
|
- {
|
|
|
- return currentResults;
|
|
|
- }
|
|
|
- }
|
|
|
- catch (Exception ex)
|
|
|
- {
|
|
|
- _logger.ErrorException("Error downloading subtitles from {0}", ex, provider.Name);
|
|
|
- }
|
|
|
- }
|
|
|
- return new List<RemoteChapterResult>();
|
|
|
- }
|
|
|
-
|
|
|
- var tasks = providers.Select(async i =>
|
|
|
- {
|
|
|
- try
|
|
|
- {
|
|
|
- return await Search(request, i, cancellationToken).ConfigureAwait(false);
|
|
|
- }
|
|
|
- catch (Exception ex)
|
|
|
- {
|
|
|
- _logger.ErrorException("Error downloading subtitles from {0}", ex, i.Name);
|
|
|
- return new List<RemoteChapterResult>();
|
|
|
- }
|
|
|
- });
|
|
|
-
|
|
|
- var results = await Task.WhenAll(tasks).ConfigureAwait(false);
|
|
|
-
|
|
|
- return results.SelectMany(i => i);
|
|
|
- }
|
|
|
-
|
|
|
- private async Task<List<RemoteChapterResult>> Search(ChapterSearchRequest request,
|
|
|
- IChapterProvider provider,
|
|
|
- CancellationToken cancellationToken)
|
|
|
- {
|
|
|
- var searchResults = await provider.Search(request, cancellationToken).ConfigureAwait(false);
|
|
|
-
|
|
|
- var list = searchResults.ToList();
|
|
|
-
|
|
|
- foreach (var result in list)
|
|
|
- {
|
|
|
- result.Id = GetProviderId(provider.Name) + "_" + result.Id;
|
|
|
- result.ProviderName = provider.Name;
|
|
|
- }
|
|
|
-
|
|
|
- return list;
|
|
|
- }
|
|
|
-
|
|
|
- public Task<ChapterResponse> GetChapters(string id, CancellationToken cancellationToken)
|
|
|
- {
|
|
|
- var parts = id.Split(new[] { '_' }, 2);
|
|
|
-
|
|
|
- var provider = GetProvider(parts.First());
|
|
|
- id = parts.Last();
|
|
|
-
|
|
|
- return provider.GetChapters(id, cancellationToken);
|
|
|
- }
|
|
|
-
|
|
|
- public IEnumerable<ChapterProviderInfo> GetProviders(string itemId)
|
|
|
- {
|
|
|
- var video = _libraryManager.GetItemById(itemId) as Video;
|
|
|
- VideoContentType mediaType;
|
|
|
-
|
|
|
- if (video is Episode)
|
|
|
- {
|
|
|
- mediaType = VideoContentType.Episode;
|
|
|
- }
|
|
|
- else if (video is Movie)
|
|
|
- {
|
|
|
- mediaType = VideoContentType.Movie;
|
|
|
- }
|
|
|
- else
|
|
|
- {
|
|
|
- // These are the only supported types
|
|
|
- return new List<ChapterProviderInfo>();
|
|
|
- }
|
|
|
-
|
|
|
- var providers = GetInternalProviders(false)
|
|
|
- .Where(i => i.SupportedMediaTypes.Contains(mediaType));
|
|
|
-
|
|
|
- return GetInfos(providers);
|
|
|
- }
|
|
|
-
|
|
|
- public IEnumerable<ChapterProviderInfo> GetProviders()
|
|
|
- {
|
|
|
- return GetInfos(GetInternalProviders(true));
|
|
|
- }
|
|
|
-
|
|
|
- private IEnumerable<IChapterProvider> GetInternalProviders(bool includeDisabledProviders)
|
|
|
- {
|
|
|
- var providers = _providers;
|
|
|
-
|
|
|
- if (!includeDisabledProviders)
|
|
|
- {
|
|
|
- var options = GetConfiguration();
|
|
|
-
|
|
|
- providers = providers
|
|
|
- .Where(i => !options.DisabledFetchers.Contains(i.Name))
|
|
|
- .ToArray();
|
|
|
- }
|
|
|
-
|
|
|
- return providers
|
|
|
- .OrderBy(GetConfiguredOrder)
|
|
|
- .ThenBy(GetDefaultOrder)
|
|
|
- .ToArray();
|
|
|
- }
|
|
|
-
|
|
|
- private IEnumerable<ChapterProviderInfo> GetInfos(IEnumerable<IChapterProvider> providers)
|
|
|
- {
|
|
|
- return providers.Select(i => new ChapterProviderInfo
|
|
|
- {
|
|
|
- Name = i.Name,
|
|
|
- Id = GetProviderId(i.Name)
|
|
|
- });
|
|
|
- }
|
|
|
-
|
|
|
- private string GetProviderId(string name)
|
|
|
- {
|
|
|
- return name.ToLower().GetMD5().ToString("N");
|
|
|
- }
|
|
|
-
|
|
|
- private IChapterProvider GetProvider(string id)
|
|
|
- {
|
|
|
- return _providers.First(i => string.Equals(id, GetProviderId(i.Name)));
|
|
|
- }
|
|
|
-
|
|
|
- private int GetConfiguredOrder(IChapterProvider provider)
|
|
|
- {
|
|
|
- var options = GetConfiguration();
|
|
|
-
|
|
|
- // See if there's a user-defined order
|
|
|
- var index = Array.IndexOf(options.FetcherOrder, provider.Name);
|
|
|
-
|
|
|
- if (index != -1)
|
|
|
- {
|
|
|
- return index;
|
|
|
- }
|
|
|
-
|
|
|
- // Not configured. Just return some high number to put it at the end.
|
|
|
- return 100;
|
|
|
- }
|
|
|
-
|
|
|
- private int GetDefaultOrder(IChapterProvider provider)
|
|
|
- {
|
|
|
- var hasOrder = provider as IHasOrder;
|
|
|
-
|
|
|
- if (hasOrder != null)
|
|
|
- {
|
|
|
- return hasOrder.Order;
|
|
|
- }
|
|
|
-
|
|
|
- return 0;
|
|
|
- }
|
|
|
-
|
|
|
public IEnumerable<ChapterInfo> GetChapters(string itemId)
|
|
|
{
|
|
|
return _itemRepo.GetChapters(new Guid(itemId));
|