jellyfin-server/MediaBrowser.Api/ItemRefreshService.cs

166 lines
6.5 KiB
C#
Raw Normal View History

2014-04-27 03:42:05 +00:00
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Entities.Audio;
2013-06-27 19:29:58 +00:00
using MediaBrowser.Controller.Library;
2014-07-02 18:34:08 +00:00
using MediaBrowser.Controller.Net;
using MediaBrowser.Controller.Providers;
2013-12-07 15:52:38 +00:00
using ServiceStack;
2013-06-27 19:29:58 +00:00
using System;
2013-09-04 17:02:19 +00:00
using System.Linq;
2013-06-27 19:29:58 +00:00
using System.Threading;
using System.Threading.Tasks;
namespace MediaBrowser.Api
{
public class BaseRefreshRequest : IReturnVoid
2013-06-27 19:29:58 +00:00
{
[ApiMember(Name = "MetadataRefreshMode", Description = "Specifies the metadata refresh mode", IsRequired = false, DataType = "boolean", ParameterType = "query", Verb = "POST")]
public MetadataRefreshMode MetadataRefreshMode { get; set; }
[ApiMember(Name = "ImageRefreshMode", Description = "Specifies the image refresh mode", IsRequired = false, DataType = "boolean", ParameterType = "query", Verb = "POST")]
public ImageRefreshMode ImageRefreshMode { get; set; }
[ApiMember(Name = "ReplaceAllMetadata", Description = "Determines if metadata should be replaced. Only applicable if mode is FullRefresh", IsRequired = false, DataType = "boolean", ParameterType = "query", Verb = "POST")]
public bool ReplaceAllMetadata { get; set; }
[ApiMember(Name = "ReplaceAllImages", Description = "Determines if images should be replaced. Only applicable if mode is FullRefresh", IsRequired = false, DataType = "boolean", ParameterType = "query", Verb = "POST")]
public bool ReplaceAllImages { get; set; }
}
2013-06-27 19:29:58 +00:00
[Route("/Items/{Id}/Refresh", "POST")]
[Api(Description = "Refreshes metadata for an item")]
public class RefreshItem : BaseRefreshRequest
{
2013-06-27 19:29:58 +00:00
[ApiMember(Name = "Recursive", Description = "Indicates if the refresh should occur recursively.", IsRequired = false, DataType = "bool", ParameterType = "query", Verb = "POST")]
public bool Recursive { get; set; }
[ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "POST")]
public string Id { get; set; }
}
2014-07-02 18:34:08 +00:00
[Authenticated]
2013-06-27 19:29:58 +00:00
public class ItemRefreshService : BaseApiService
{
private readonly ILibraryManager _libraryManager;
2014-04-27 03:42:05 +00:00
public ItemRefreshService(ILibraryManager libraryManager)
2013-06-27 19:29:58 +00:00
{
_libraryManager = libraryManager;
}
2014-04-27 03:42:05 +00:00
private async Task RefreshArtist(RefreshItem request, MusicArtist item)
{
var cancellationToken = CancellationToken.None;
2013-09-09 18:24:18 +00:00
var albums = _libraryManager.RootFolder
.RecursiveChildren
.OfType<MusicAlbum>()
.Where(i => i.HasArtist(item.Name))
.ToList();
var musicArtists = albums
.Select(i => i.Parent)
.OfType<MusicArtist>()
.ToList();
2014-02-09 07:27:44 +00:00
var options = GetRefreshOptions(request);
var musicArtistRefreshTasks = musicArtists.Select(i => i.ValidateChildren(new Progress<double>(), cancellationToken, options, true));
2013-09-09 18:24:18 +00:00
await Task.WhenAll(musicArtistRefreshTasks).ConfigureAwait(false);
try
{
2014-02-09 07:27:44 +00:00
await item.RefreshMetadata(options, CancellationToken.None).ConfigureAwait(false);
}
catch (Exception ex)
{
Logger.ErrorException("Error refreshing library", ex);
}
}
2013-06-27 19:29:58 +00:00
/// <summary>
/// Posts the specified request.
/// </summary>
/// <param name="request">The request.</param>
public void Post(RefreshItem request)
{
2014-04-27 03:42:05 +00:00
var item = _libraryManager.GetItemById(request.Id);
var task = item is MusicArtist ? RefreshArtist(request, (MusicArtist)item) : RefreshItem(request, item);
2013-06-27 19:29:58 +00:00
Task.WaitAll(task);
}
2013-06-27 19:29:58 +00:00
/// <summary>
/// Refreshes the item.
/// </summary>
/// <param name="request">The request.</param>
/// <returns>Task.</returns>
2014-04-27 03:42:05 +00:00
private async Task RefreshItem(RefreshItem request, BaseItem item)
2013-06-27 19:29:58 +00:00
{
2014-02-09 07:27:44 +00:00
var options = GetRefreshOptions(request);
2013-06-27 19:29:58 +00:00
try
{
2014-02-09 07:27:44 +00:00
await item.RefreshMetadata(options, CancellationToken.None).ConfigureAwait(false);
2013-06-27 19:29:58 +00:00
2013-11-21 20:48:26 +00:00
if (item.IsFolder)
2013-06-27 19:29:58 +00:00
{
// Collection folders don't validate their children so we'll have to simulate that here
2013-11-21 20:48:26 +00:00
var collectionFolder = item as CollectionFolder;
if (collectionFolder != null)
{
await RefreshCollectionFolderChildren(request, collectionFolder).ConfigureAwait(false);
}
else
{
2013-11-21 20:48:26 +00:00
var folder = (Folder)item;
2014-02-09 07:27:44 +00:00
await folder.ValidateChildren(new Progress<double>(), CancellationToken.None, options, request.Recursive).ConfigureAwait(false);
}
2013-06-27 19:29:58 +00:00
}
}
catch (Exception ex)
{
Logger.ErrorException("Error refreshing library", ex);
}
}
/// <summary>
/// Refreshes the collection folder children.
/// </summary>
/// <param name="request">The request.</param>
/// <param name="collectionFolder">The collection folder.</param>
/// <returns>Task.</returns>
private async Task RefreshCollectionFolderChildren(RefreshItem request, CollectionFolder collectionFolder)
{
var options = GetRefreshOptions(request);
foreach (var child in collectionFolder.Children.ToList())
{
await child.RefreshMetadata(options, CancellationToken.None).ConfigureAwait(false);
2013-11-21 20:48:26 +00:00
if (child.IsFolder)
{
2013-11-21 20:48:26 +00:00
var folder = (Folder)child;
2014-02-09 07:27:44 +00:00
await folder.ValidateChildren(new Progress<double>(), CancellationToken.None, options, request.Recursive).ConfigureAwait(false);
}
}
}
private MetadataRefreshOptions GetRefreshOptions(BaseRefreshRequest request)
{
return new MetadataRefreshOptions
{
MetadataRefreshMode = request.MetadataRefreshMode,
ImageRefreshMode = request.ImageRefreshMode,
ReplaceAllImages = request.ReplaceAllImages,
2014-07-20 04:46:29 +00:00
ReplaceAllMetadata = request.ReplaceAllMetadata,
ForceSave = true
};
}
2013-06-27 19:29:58 +00:00
}
}