jellyfin-server/MediaBrowser.Providers/Music/AlbumMetadataService.cs

127 lines
4.4 KiB
C#
Raw Normal View History

using System;
using System.Collections.Generic;
using System.Linq;
using MediaBrowser.Controller.Configuration;
using MediaBrowser.Controller.Entities;
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;
2016-10-25 19:02:04 +00:00
using MediaBrowser.Model.IO;
using MediaBrowser.Providers.Manager;
using Microsoft.Extensions.Logging;
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
{
2019-08-29 20:28:33 +00:00
public AlbumMetadataService(
IServerConfigurationManager serverConfigurationManager,
ILogger logger,
IProviderManager providerManager,
IFileSystem fileSystem,
ILibraryManager libraryManager)
2019-09-10 20:37:53 +00:00
: base(serverConfigurationManager, logger, providerManager, fileSystem, libraryManager)
2019-08-29 20:28:33 +00:00
{
}
/// <inheritdoc />
protected override bool EnableUpdatingPremiereDateFromChildren => true;
/// <inheritdoc />
protected override bool EnableUpdatingGenresFromChildren => true;
/// <inheritdoc />
protected override bool EnableUpdatingStudiosFromChildren => true;
/// <inheritdoc />
2018-09-12 17:26:21 +00:00
protected override IList<BaseItem> GetChildrenForMetadataUpdates(MusicAlbum item)
2019-09-10 20:37:53 +00:00
=> item.GetRecursiveChildren(i => i is Audio);
2018-09-12 17:26:21 +00:00
2019-08-29 20:28:33 +00:00
/// <inheritdoc />
2018-09-12 17:26:21 +00:00
protected override ItemUpdateType UpdateMetadataFromChildren(MusicAlbum item, IList<BaseItem> children, bool isFullRefresh, ItemUpdateType currentUpdateType)
{
var updateType = base.UpdateMetadataFromChildren(item, children, isFullRefresh, currentUpdateType);
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
{
2018-09-12 17:26:21 +00:00
if (!item.LockedFields.Contains(MetadataFields.Name))
2014-06-02 19:32:41 +00:00
{
2018-09-12 17:26:21 +00:00
var name = children.Select(i => i.Album).FirstOrDefault(i => !string.IsNullOrEmpty(i));
2014-01-31 19:55:21 +00:00
2019-09-10 20:37:53 +00:00
if (!string.IsNullOrEmpty(name)
&& !string.Equals(item.Name, name, StringComparison.Ordinal))
2014-06-02 19:32:41 +00:00
{
2019-09-10 20:37:53 +00:00
item.Name = name;
updateType |= ItemUpdateType.MetadataEdit;
2015-01-17 18:15:09 +00:00
}
2018-09-12 17:26:21 +00:00
}
2014-01-31 19:55:21 +00:00
2018-09-12 17:26:21 +00:00
var songs = children.Cast<Audio>().ToArray();
2014-06-02 19:32:41 +00:00
2019-09-10 20:37:53 +00:00
updateType |= SetAlbumArtistFromSongs(item, songs);
updateType |= SetArtistsFromSongs(item, songs);
2018-09-12 17:26:21 +00:00
}
2014-01-31 19:55:21 +00:00
2018-09-12 17:26:21 +00:00
return updateType;
}
2015-01-17 18:15:09 +00:00
2014-01-31 19:55:21 +00:00
private ItemUpdateType SetAlbumArtistFromSongs(MusicAlbum item, IEnumerable<Audio> songs)
{
2014-02-10 18:39:41 +00:00
var updateType = ItemUpdateType.None;
2019-01-07 23:27:46 +00:00
var artists = songs
2014-06-23 16:05:19 +00:00
.SelectMany(i => i.AlbumArtists)
.Distinct(StringComparer.OrdinalIgnoreCase)
.OrderBy(i => i)
2017-08-10 18:01:31 +00:00
.ToArray();
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)
2017-08-24 19:52:19 +00:00
.ToArray();
2014-01-31 19:55:21 +00:00
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;
}
2015-06-29 01:10:45 +00:00
2019-08-29 20:28:33 +00:00
/// <inheritdoc />
protected override void MergeData(
MetadataResult<MusicAlbum> source,
MetadataResult<MusicAlbum> target,
MetadataFields[] lockedFields,
bool replaceData,
bool mergeMetadataSettings)
2015-06-29 01:10:45 +00:00
{
ProviderUtils.MergeBaseItemData(source, target, lockedFields, replaceData, mergeMetadataSettings);
var sourceItem = source.Item;
var targetItem = target.Item;
2019-08-29 20:28:33 +00:00
if (replaceData || targetItem.Artists.Count == 0)
2015-06-29 01:10:45 +00:00
{
targetItem.Artists = sourceItem.Artists;
}
}
2014-01-31 19:55:21 +00:00
}
}