1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
using System;
using System.Collections.Generic;
using System.Globalization;
using System.Runtime.CompilerServices;
using System.Threading;
using MediaBrowser.Controller.Entities.TV;
using MediaBrowser.Controller.Library;
using MediaBrowser.Model.Configuration;
using MediaBrowser.Model.Entities;
using Microsoft.Extensions.Logging;
namespace MediaBrowser.Providers.Plugins.Tmdb.TV;
/// <summary>
/// TMDb-based similar items provider for TV series.
/// </summary>
public class TmdbSeriesSimilarProvider : IRemoteSimilarItemsProvider<Series>
{
private readonly TmdbClientManager _tmdbClientManager;
private readonly ILogger<TmdbSeriesSimilarProvider> _logger;
/// <summary>
/// Initializes a new instance of the <see cref="TmdbSeriesSimilarProvider"/> class.
/// </summary>
/// <param name="tmdbClientManager">The TMDb client manager.</param>
/// <param name="logger">The logger.</param>
public TmdbSeriesSimilarProvider(TmdbClientManager tmdbClientManager, ILogger<TmdbSeriesSimilarProvider> logger)
{
_tmdbClientManager = tmdbClientManager;
_logger = logger;
}
/// <inheritdoc/>
public string Name => TmdbUtils.ProviderName;
/// <inheritdoc/>
public MetadataPluginType Type => MetadataPluginType.SimilarityProvider;
/// <inheritdoc/>
public TimeSpan? CacheDuration
{
get
{
var days = Plugin.Instance?.Configuration.SimilarItemsCacheDays ?? 0;
return days > 0 ? TimeSpan.FromDays(days) : null;
}
}
/// <inheritdoc/>
public async IAsyncEnumerable<SimilarItemReference> GetSimilarItemsAsync(
Series item,
SimilarItemsQuery query,
[EnumeratorCancellation] CancellationToken cancellationToken)
{
if (!item.TryGetProviderId(MetadataProvider.Tmdb, out var tmdbIdStr) || !int.TryParse(tmdbIdStr, CultureInfo.InvariantCulture, out var tmdbId))
{
yield break;
}
var providerName = MetadataProvider.Tmdb.ToString();
var page = 1;
var totalPages = 1;
while (page <= totalPages && !cancellationToken.IsCancellationRequested)
{
IReadOnlyList<TMDbLib.Objects.Search.SearchTv> pageResults;
try
{
(pageResults, totalPages) = await _tmdbClientManager
.GetSeriesSimilarPageAsync(tmdbId, page, TmdbUtils.GetImageLanguagesParam(string.Empty), cancellationToken)
.ConfigureAwait(false);
}
catch (Exception ex)
{
_logger.LogWarning(ex, "Failed to get similar TV shows from TMDb for {TmdbId} page {Page}", tmdbId, page);
yield break;
}
if (pageResults.Count == 0)
{
yield break;
}
foreach (var similar in pageResults)
{
yield return new SimilarItemReference
{
ProviderName = providerName,
ProviderId = similar.Id.ToString(CultureInfo.InvariantCulture)
};
}
page++;
}
}
}
|
