mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 03:26:07 +02:00
142 lines
4.3 KiB
C#
142 lines
4.3 KiB
C#
using MediaBrowser.Common.Extensions;
|
|
using MediaBrowser.Controller.Configuration;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Entities.Audio;
|
|
using MediaBrowser.Controller.Providers;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.Logging;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace MediaBrowser.Providers.Music
|
|
{
|
|
public class AlbumInfoFromSongProvider : BaseMetadataProvider
|
|
{
|
|
public AlbumInfoFromSongProvider(ILogManager logManager, IServerConfigurationManager configurationManager)
|
|
: base(logManager, configurationManager)
|
|
{
|
|
}
|
|
|
|
public override bool Supports(BaseItem item)
|
|
{
|
|
return item is MusicAlbum;
|
|
}
|
|
|
|
protected override bool RefreshOnVersionChange
|
|
{
|
|
get
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
protected override string ProviderVersion
|
|
{
|
|
get
|
|
{
|
|
return "1";
|
|
}
|
|
}
|
|
|
|
protected override bool NeedsRefreshInternal(BaseItem item, BaseProviderInfo providerInfo)
|
|
{
|
|
// If song metadata has changed
|
|
if (GetComparisonData((MusicAlbum)item) != providerInfo.FileStamp)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
return base.NeedsRefreshInternal(item, providerInfo);
|
|
}
|
|
/// <summary>
|
|
/// Gets the data.
|
|
/// </summary>
|
|
/// <param name="album">The album.</param>
|
|
/// <returns>Guid.</returns>
|
|
private Guid GetComparisonData(MusicAlbum album)
|
|
{
|
|
var songs = album.RecursiveChildren.OfType<Audio>().ToList();
|
|
|
|
return GetComparisonData(songs);
|
|
}
|
|
|
|
private Guid GetComparisonData(List<Audio> songs)
|
|
{
|
|
var albumNames = songs.Select(i => i.AlbumArtist)
|
|
.Where(i => !string.IsNullOrEmpty(i))
|
|
.Distinct(StringComparer.OrdinalIgnoreCase)
|
|
.ToList();
|
|
|
|
var studios = songs.SelectMany(i => i.Studios)
|
|
.Distinct(StringComparer.OrdinalIgnoreCase)
|
|
.ToList();
|
|
|
|
var genres = songs.SelectMany(i => i.Genres)
|
|
.Distinct(StringComparer.OrdinalIgnoreCase)
|
|
.ToList();
|
|
|
|
albumNames.AddRange(studios);
|
|
albumNames.AddRange(genres);
|
|
|
|
return string.Join(string.Empty, albumNames.OrderBy(i => i).ToArray()).GetMD5();
|
|
}
|
|
|
|
public override Task<bool> FetchAsync(BaseItem item, bool force, CancellationToken cancellationToken)
|
|
{
|
|
var album = (MusicAlbum)item;
|
|
|
|
BaseProviderInfo data;
|
|
if (!item.ProviderData.TryGetValue(Id, out data))
|
|
{
|
|
data = new BaseProviderInfo();
|
|
item.ProviderData[Id] = data;
|
|
}
|
|
|
|
var songs = album.RecursiveChildren.OfType<Audio>().ToList();
|
|
|
|
if (!item.LockedFields.Contains(MetadataFields.Name))
|
|
{
|
|
var name = songs.Select(i => i.Album).FirstOrDefault(i => !string.IsNullOrEmpty(i));
|
|
|
|
if (!string.IsNullOrEmpty(name))
|
|
{
|
|
album.Name = name;
|
|
}
|
|
}
|
|
|
|
var year = songs.Select(i => i.ProductionYear ?? 1800).FirstOrDefault(i => i != 1800);
|
|
|
|
if (year != 1800)
|
|
{
|
|
album.ProductionYear = year;
|
|
}
|
|
|
|
if (!item.LockedFields.Contains(MetadataFields.Studios))
|
|
{
|
|
album.Studios = songs.SelectMany(i => i.Studios)
|
|
.Distinct(StringComparer.OrdinalIgnoreCase)
|
|
.ToList();
|
|
}
|
|
|
|
if (!item.LockedFields.Contains(MetadataFields.Genres))
|
|
{
|
|
album.Genres = songs.SelectMany(i => i.Genres)
|
|
.Distinct(StringComparer.OrdinalIgnoreCase)
|
|
.ToList();
|
|
}
|
|
|
|
data.FileStamp = GetComparisonData(songs);
|
|
|
|
SetLastRefreshed(item, DateTime.UtcNow);
|
|
return TrueTaskResult;
|
|
}
|
|
|
|
public override MetadataProviderPriority Priority
|
|
{
|
|
get { return MetadataProviderPriority.Second; }
|
|
}
|
|
}
|
|
}
|