jellyfin/MediaBrowser.Providers/Music/AlbumMetadataService.cs

171 lines
6.6 KiB
C#
Raw Normal View History

2014-02-02 13:36:31 +00:00
using MediaBrowser.Common.IO;
using MediaBrowser.Controller.Configuration;
2014-01-31 19:55:21 +00:00
using MediaBrowser.Controller.Entities.Audio;
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.Logging;
using MediaBrowser.Providers.Manager;
using System;
using System.Collections.Generic;
using System.Linq;
2015-01-17 18:15:09 +00:00
using System.Threading.Tasks;
2014-01-31 19:55:21 +00:00
namespace MediaBrowser.Providers.Music
{
2014-02-07 03:10:13 +00:00
public class AlbumMetadataService : MetadataService<MusicAlbum, AlbumInfo>
2014-01-31 19:55:21 +00:00
{
2015-06-29 01:10:45 +00:00
public AlbumMetadataService(IServerConfigurationManager serverConfigurationManager, ILogger logger, IProviderManager providerManager, IProviderRepository providerRepo, IFileSystem fileSystem, IUserDataManager userDataManager, ILibraryManager libraryManager) : base(serverConfigurationManager, logger, providerManager, providerRepo, fileSystem, userDataManager, libraryManager)
2014-01-31 19:55:21 +00:00
{
}
2015-01-17 18:15:09 +00:00
protected override async Task<ItemUpdateType> BeforeSave(MusicAlbum item, bool isFullRefresh, ItemUpdateType currentUpdateType)
2014-01-31 19:55:21 +00:00
{
2015-01-17 18:15:09 +00:00
var updateType = await base.BeforeSave(item, isFullRefresh, currentUpdateType).ConfigureAwait(false);
2014-01-31 19:55:21 +00:00
2015-01-17 18:15:09 +00:00
if (isFullRefresh || currentUpdateType > ItemUpdateType.None)
2014-01-31 19:55:21 +00:00
{
2015-01-17 18:15:09 +00:00
if (!item.IsLocked)
2014-06-02 19:32:41 +00:00
{
2015-01-25 06:34:50 +00:00
var songs = item.GetRecursiveChildren(i => i is Audio)
.Cast<Audio>()
.ToList();
2014-01-31 19:55:21 +00:00
2015-01-17 18:15:09 +00:00
if (!item.LockedFields.Contains(MetadataFields.Genres))
2014-06-02 19:32:41 +00:00
{
2015-01-17 18:15:09 +00:00
var currentList = item.Genres.ToList();
2014-01-31 19:55:21 +00:00
2015-01-17 18:15:09 +00:00
item.Genres = songs.SelectMany(i => i.Genres)
.Distinct(StringComparer.OrdinalIgnoreCase)
.ToList();
2014-01-31 19:55:21 +00:00
2015-01-17 18:15:09 +00:00
if (currentList.Count != item.Genres.Count || !currentList.OrderBy(i => i).SequenceEqual(item.Genres.OrderBy(i => i), StringComparer.OrdinalIgnoreCase))
{
updateType = updateType | ItemUpdateType.MetadataEdit;
}
}
2014-01-31 19:55:21 +00:00
2015-01-17 18:15:09 +00:00
if (!item.LockedFields.Contains(MetadataFields.Studios))
2014-06-02 19:32:41 +00:00
{
2015-01-17 18:15:09 +00:00
var currentList = item.Studios.ToList();
2014-01-31 19:55:21 +00:00
2015-01-17 18:15:09 +00:00
item.Studios = songs.SelectMany(i => i.Studios)
.Distinct(StringComparer.OrdinalIgnoreCase)
.ToList();
2014-06-02 19:32:41 +00:00
2015-01-17 18:15:09 +00:00
if (currentList.Count != item.Studios.Count || !currentList.OrderBy(i => i).SequenceEqual(item.Studios.OrderBy(i => i), StringComparer.OrdinalIgnoreCase))
2014-06-02 19:32:41 +00:00
{
2014-12-28 06:21:39 +00:00
updateType = updateType | ItemUpdateType.MetadataEdit;
2014-06-02 19:32:41 +00:00
}
2014-01-31 19:55:21 +00:00
}
2015-01-17 18:15:09 +00:00
if (!item.LockedFields.Contains(MetadataFields.Name))
{
var name = songs.Select(i => i.Album).FirstOrDefault(i => !string.IsNullOrEmpty(i));
if (!string.IsNullOrEmpty(name))
{
if (!string.Equals(item.Name, name, StringComparison.Ordinal))
{
item.Name = name;
updateType = updateType | ItemUpdateType.MetadataEdit;
}
}
}
updateType = updateType | SetAlbumArtistFromSongs(item, songs);
updateType = updateType | SetArtistsFromSongs(item, songs);
updateType = updateType | SetDateFromSongs(item, songs);
}
2014-12-28 06:21:39 +00:00
}
2014-01-31 19:55:21 +00:00
return updateType;
}
private ItemUpdateType SetAlbumArtistFromSongs(MusicAlbum item, IEnumerable<Audio> songs)
{
2014-02-10 18:39:41 +00:00
var updateType = ItemUpdateType.None;
2014-01-31 19:55:21 +00:00
var artists = songs
2014-06-23 16:05:19 +00:00
.SelectMany(i => i.AlbumArtists)
.Distinct(StringComparer.OrdinalIgnoreCase)
.OrderBy(i => i)
.ToList();
2014-01-31 19:55:21 +00:00
if (!item.AlbumArtists.SequenceEqual(artists, StringComparer.OrdinalIgnoreCase))
2014-01-31 19:55:21 +00:00
{
item.AlbumArtists = artists;
2014-12-28 06:21:39 +00:00
updateType = updateType | ItemUpdateType.MetadataEdit;
2014-01-31 19:55:21 +00:00
}
return updateType;
}
private ItemUpdateType SetArtistsFromSongs(MusicAlbum item, IEnumerable<Audio> songs)
{
2014-02-10 18:39:41 +00:00
var updateType = ItemUpdateType.None;
2014-01-31 19:55:21 +00:00
var artists = songs
.SelectMany(i => i.Artists)
2014-01-31 19:55:21 +00:00
.Distinct(StringComparer.OrdinalIgnoreCase)
.OrderBy(i => i)
2014-01-31 19:55:21 +00:00
.ToList();
if (!item.Artists.SequenceEqual(artists, StringComparer.OrdinalIgnoreCase))
2014-01-31 19:55:21 +00:00
{
item.Artists = artists;
2014-12-28 06:21:39 +00:00
updateType = updateType | ItemUpdateType.MetadataEdit;
2014-01-31 19:55:21 +00:00
}
return updateType;
}
private ItemUpdateType SetDateFromSongs(MusicAlbum item, List<Audio> songs)
{
2014-02-10 18:39:41 +00:00
var updateType = ItemUpdateType.None;
2014-01-31 19:55:21 +00:00
var date = songs.Select(i => i.PremiereDate)
.FirstOrDefault(i => i.HasValue);
var originalPremiereDate = item.PremiereDate;
var originalProductionYear = item.ProductionYear;
if (date.HasValue)
{
item.PremiereDate = date.Value;
item.ProductionYear = date.Value.Year;
}
else
{
var year = songs.Select(i => i.ProductionYear ?? 1800).FirstOrDefault(i => i != 1800);
if (year != 1800)
{
item.ProductionYear = year;
}
}
if ((originalPremiereDate ?? DateTime.MinValue) != (item.PremiereDate ?? DateTime.MinValue) ||
(originalProductionYear ?? -1) != (item.ProductionYear ?? -1))
{
2014-12-28 06:21:39 +00:00
updateType = updateType | ItemUpdateType.MetadataEdit;
2014-01-31 19:55:21 +00:00
}
return updateType;
}
2015-06-29 01:10:45 +00:00
protected override void MergeData(MetadataResult<MusicAlbum> source, MetadataResult<MusicAlbum> target, List<MetadataFields> lockedFields, bool replaceData, bool mergeMetadataSettings)
{
ProviderUtils.MergeBaseItemData(source, target, lockedFields, replaceData, mergeMetadataSettings);
var sourceItem = source.Item;
var targetItem = target.Item;
if (replaceData || targetItem.Artists.Count == 0)
{
targetItem.Artists = sourceItem.Artists;
}
}
2014-01-31 19:55:21 +00:00
}
}