blob: 3dca748d06f14f4ed010a7758e01d9bf213dcfba (
plain)
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
|
using System;
using System.Collections.Generic;
using System.Net.Http;
using System.Runtime.CompilerServices;
using System.Threading;
using MediaBrowser.Controller.Entities.Audio;
using MediaBrowser.Controller.Library;
using MediaBrowser.Model.Configuration;
using MediaBrowser.Model.Entities;
using MediaBrowser.Providers.Plugins.ListenBrainz.Api;
using Microsoft.Extensions.Logging;
namespace MediaBrowser.Providers.Plugins.ListenBrainz;
/// <summary>
/// ListenBrainz-based similar items provider for music artists.
/// </summary>
public class ListenBrainzSimilarArtistProvider : IRemoteSimilarItemsProvider<MusicArtist>
{
private readonly ListenBrainzLabsClient _labsClient;
private readonly ILogger<ListenBrainzSimilarArtistProvider> _logger;
/// <summary>
/// Initializes a new instance of the <see cref="ListenBrainzSimilarArtistProvider"/> class.
/// </summary>
/// <param name="labsClient">The ListenBrainz Labs API client.</param>
/// <param name="logger">The logger.</param>
public ListenBrainzSimilarArtistProvider(
ListenBrainzLabsClient labsClient,
ILogger<ListenBrainzSimilarArtistProvider> logger)
{
_labsClient = labsClient;
_logger = logger;
}
/// <inheritdoc/>
public string Name => "ListenBrainz";
/// <inheritdoc/>
public MetadataPluginType Type => MetadataPluginType.SimilarityProvider;
/// <inheritdoc/>
public TimeSpan? CacheDuration
{
get
{
var days = ListenBrainzPlugin.Instance?.Configuration.SimilarItemsCacheDays ?? 0;
return days > 0 ? TimeSpan.FromDays(days) : null;
}
}
/// <inheritdoc/>
public async IAsyncEnumerable<SimilarItemReference> GetSimilarItemsAsync(
MusicArtist item,
SimilarItemsQuery query,
[EnumeratorCancellation] CancellationToken cancellationToken)
{
ArgumentNullException.ThrowIfNull(item);
ArgumentNullException.ThrowIfNull(query);
if (!item.TryGetProviderId(MetadataProvider.MusicBrainzArtist, out var mbidStr) || !Guid.TryParse(mbidStr, out var mbid))
{
_logger.LogDebug("No MusicBrainz Artist ID found for {ArtistName}", item.Name);
yield break;
}
IReadOnlyList<Guid> similarMbids;
try
{
similarMbids = await _labsClient.GetSimilarArtistsAsync(mbid, cancellationToken).ConfigureAwait(false);
}
catch (HttpRequestException ex)
{
_logger.LogWarning(ex, "Failed to fetch similar artists from ListenBrainz for {ArtistMbid}", mbid);
yield break;
}
var providerName = MetadataProvider.MusicBrainzArtist.ToString();
foreach (var similarMbid in similarMbids)
{
yield return new SimilarItemReference
{
ProviderName = providerName,
ProviderId = similarMbid.ToString()
};
}
}
}
|
