jellyfin/MediaBrowser.Api/ItemLookupService.cs

322 lines
12 KiB
C#
Raw Normal View History

2014-03-01 22:34:27 +00:00
using MediaBrowser.Common.Extensions;
using MediaBrowser.Controller;
using MediaBrowser.Controller.Entities;
2014-03-14 03:23:58 +00:00
using MediaBrowser.Controller.Entities.Audio;
2014-03-01 22:34:27 +00:00
using MediaBrowser.Controller.Entities.Movies;
using MediaBrowser.Controller.Entities.TV;
2014-03-02 15:42:21 +00:00
using MediaBrowser.Controller.Library;
2014-07-02 18:34:08 +00:00
using MediaBrowser.Controller.Net;
2014-03-01 22:34:27 +00:00
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Providers;
using ServiceStack;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
2015-10-04 04:23:11 +00:00
using CommonIO;
2016-01-15 19:15:51 +00:00
using MediaBrowser.Model.Serialization;
2014-03-01 22:34:27 +00:00
namespace MediaBrowser.Api
{
2014-11-15 02:31:03 +00:00
[Route("/Items/{Id}/ExternalIdInfos", "GET", Summary = "Gets external id infos for an item")]
[Authenticated(Roles = "Admin")]
2014-03-01 22:34:27 +00:00
public class GetExternalIdInfos : IReturn<List<ExternalIdInfo>>
{
/// <summary>
/// Gets or sets the id.
/// </summary>
/// <value>The id.</value>
[ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")]
public string Id { get; set; }
}
[Route("/Items/RemoteSearch/Movie", "POST")]
[Authenticated]
2014-03-01 22:34:27 +00:00
public class GetMovieRemoteSearchResults : RemoteSearchQuery<MovieInfo>, IReturn<List<RemoteSearchResult>>
{
}
2016-06-23 17:04:18 +00:00
[Route("/Items/RemoteSearch/Trailer", "POST")]
[Authenticated]
public class GetTrailerRemoteSearchResults : RemoteSearchQuery<TrailerInfo>, IReturn<List<RemoteSearchResult>>
{
}
2014-03-01 22:34:27 +00:00
[Route("/Items/RemoteSearch/AdultVideo", "POST")]
[Authenticated]
2014-03-01 22:34:27 +00:00
public class GetAdultVideoRemoteSearchResults : RemoteSearchQuery<ItemLookupInfo>, IReturn<List<RemoteSearchResult>>
{
}
[Route("/Items/RemoteSearch/Series", "POST")]
[Authenticated]
2014-03-01 22:34:27 +00:00
public class GetSeriesRemoteSearchResults : RemoteSearchQuery<SeriesInfo>, IReturn<List<RemoteSearchResult>>
{
}
[Route("/Items/RemoteSearch/Game", "POST")]
[Authenticated]
2014-03-01 22:34:27 +00:00
public class GetGameRemoteSearchResults : RemoteSearchQuery<GameInfo>, IReturn<List<RemoteSearchResult>>
{
}
[Route("/Items/RemoteSearch/BoxSet", "POST")]
[Authenticated]
2014-03-01 22:34:27 +00:00
public class GetBoxSetRemoteSearchResults : RemoteSearchQuery<BoxSetInfo>, IReturn<List<RemoteSearchResult>>
{
}
2014-03-14 03:23:58 +00:00
[Route("/Items/RemoteSearch/MusicArtist", "POST")]
[Authenticated]
2014-03-14 03:23:58 +00:00
public class GetMusicArtistRemoteSearchResults : RemoteSearchQuery<ArtistInfo>, IReturn<List<RemoteSearchResult>>
{
}
[Route("/Items/RemoteSearch/MusicAlbum", "POST")]
[Authenticated]
2014-03-14 03:23:58 +00:00
public class GetMusicAlbumRemoteSearchResults : RemoteSearchQuery<AlbumInfo>, IReturn<List<RemoteSearchResult>>
{
}
2014-03-01 22:34:27 +00:00
[Route("/Items/RemoteSearch/Person", "POST")]
2014-11-15 02:31:03 +00:00
[Authenticated(Roles = "Admin")]
2014-03-01 22:34:27 +00:00
public class GetPersonRemoteSearchResults : RemoteSearchQuery<PersonLookupInfo>, IReturn<List<RemoteSearchResult>>
{
}
2014-11-15 02:31:03 +00:00
[Route("/Items/RemoteSearch/Image", "GET", Summary = "Gets a remote image")]
2014-03-01 22:34:27 +00:00
public class GetRemoteSearchImage
{
[ApiMember(Name = "ImageUrl", Description = "The image url", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "GET")]
public string ImageUrl { get; set; }
[ApiMember(Name = "ProviderName", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "GET")]
public string ProviderName { get; set; }
}
2014-03-02 15:42:21 +00:00
2014-11-15 02:31:03 +00:00
[Route("/Items/RemoteSearch/Apply/{Id}", "POST", Summary = "Applies search criteria to an item and refreshes metadata")]
[Authenticated(Roles = "Admin")]
2014-03-02 15:42:21 +00:00
public class ApplySearchCriteria : RemoteSearchResult, IReturnVoid
{
2015-01-11 20:31:09 +00:00
[ApiMember(Name = "Id", Description = "The item id", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "POST")]
2014-03-02 15:42:21 +00:00
public string Id { get; set; }
2015-01-11 20:31:09 +00:00
[ApiMember(Name = "ReplaceAllImages", Description = "Whether or not to replace all images", IsRequired = false, DataType = "boolean", ParameterType = "query", Verb = "POST")]
public bool ReplaceAllImages { get; set; }
public ApplySearchCriteria()
{
ReplaceAllImages = true;
}
2014-03-02 15:42:21 +00:00
}
2014-03-01 22:34:27 +00:00
public class ItemLookupService : BaseApiService
{
private readonly IProviderManager _providerManager;
private readonly IServerApplicationPaths _appPaths;
private readonly IFileSystem _fileSystem;
2014-03-02 15:42:21 +00:00
private readonly ILibraryManager _libraryManager;
2016-01-15 19:15:51 +00:00
private readonly IJsonSerializer _json;
2014-03-01 22:34:27 +00:00
2016-01-15 19:15:51 +00:00
public ItemLookupService(IProviderManager providerManager, IServerApplicationPaths appPaths, IFileSystem fileSystem, ILibraryManager libraryManager, IJsonSerializer json)
2014-03-01 22:34:27 +00:00
{
_providerManager = providerManager;
_appPaths = appPaths;
_fileSystem = fileSystem;
2014-03-02 15:42:21 +00:00
_libraryManager = libraryManager;
2016-01-15 19:15:51 +00:00
_json = json;
2014-03-01 22:34:27 +00:00
}
public object Get(GetExternalIdInfos request)
{
2014-04-25 20:15:50 +00:00
var item = _libraryManager.GetItemById(request.Id);
2014-03-01 22:34:27 +00:00
var infos = _providerManager.GetExternalIdInfos(item).ToList();
return ToOptimizedResult(infos);
}
2016-06-23 17:04:18 +00:00
public async Task<object> Post(GetTrailerRemoteSearchResults request)
{
var result = await _providerManager.GetRemoteSearchResults<Trailer, TrailerInfo>(request, CancellationToken.None).ConfigureAwait(false);
return ToOptimizedResult(result);
}
2016-06-19 06:18:29 +00:00
public async Task<object> Post(GetMovieRemoteSearchResults request)
2014-03-01 22:34:27 +00:00
{
2016-06-19 06:18:29 +00:00
var result = await _providerManager.GetRemoteSearchResults<Movie, MovieInfo>(request, CancellationToken.None).ConfigureAwait(false);
2014-03-01 22:34:27 +00:00
return ToOptimizedResult(result);
}
2016-06-19 06:18:29 +00:00
public async Task<object> Post(GetSeriesRemoteSearchResults request)
2014-03-01 22:34:27 +00:00
{
2016-06-19 06:18:29 +00:00
var result = await _providerManager.GetRemoteSearchResults<Series, SeriesInfo>(request, CancellationToken.None).ConfigureAwait(false);
2014-03-01 22:34:27 +00:00
return ToOptimizedResult(result);
}
2016-06-19 06:18:29 +00:00
public async Task<object> Post(GetGameRemoteSearchResults request)
2014-03-01 22:34:27 +00:00
{
2016-06-19 06:18:29 +00:00
var result = await _providerManager.GetRemoteSearchResults<Game, GameInfo>(request, CancellationToken.None).ConfigureAwait(false);
2014-03-01 22:34:27 +00:00
return ToOptimizedResult(result);
}
2016-06-19 06:18:29 +00:00
public async Task<object> Post(GetBoxSetRemoteSearchResults request)
2014-03-01 22:34:27 +00:00
{
2016-06-19 06:18:29 +00:00
var result = await _providerManager.GetRemoteSearchResults<BoxSet, BoxSetInfo>(request, CancellationToken.None).ConfigureAwait(false);
2014-03-01 22:34:27 +00:00
return ToOptimizedResult(result);
}
2016-06-19 06:18:29 +00:00
public async Task<object> Post(GetPersonRemoteSearchResults request)
2014-03-01 22:34:27 +00:00
{
2016-06-19 06:18:29 +00:00
var result = await _providerManager.GetRemoteSearchResults<Person, PersonLookupInfo>(request, CancellationToken.None).ConfigureAwait(false);
2014-03-01 22:34:27 +00:00
return ToOptimizedResult(result);
}
2016-06-19 06:18:29 +00:00
public async Task<object> Post(GetMusicAlbumRemoteSearchResults request)
2014-03-14 03:23:58 +00:00
{
2016-06-19 06:18:29 +00:00
var result = await _providerManager.GetRemoteSearchResults<MusicAlbum, AlbumInfo>(request, CancellationToken.None).ConfigureAwait(false);
2014-03-14 03:23:58 +00:00
return ToOptimizedResult(result);
}
2016-06-19 06:18:29 +00:00
public async Task<object> Post(GetMusicArtistRemoteSearchResults request)
2014-03-14 03:23:58 +00:00
{
2016-06-19 06:18:29 +00:00
var result = await _providerManager.GetRemoteSearchResults<MusicArtist, ArtistInfo>(request, CancellationToken.None).ConfigureAwait(false);
2014-03-14 03:23:58 +00:00
return ToOptimizedResult(result);
}
2016-06-23 05:26:49 +00:00
public Task<object> Get(GetRemoteSearchImage request)
2014-03-01 22:34:27 +00:00
{
2016-06-23 05:26:49 +00:00
return GetRemoteImage(request);
2014-03-01 22:34:27 +00:00
}
2014-03-02 15:42:21 +00:00
public void Post(ApplySearchCriteria request)
{
var item = _libraryManager.GetItemById(new Guid(request.Id));
2014-10-14 04:59:34 +00:00
//foreach (var key in request.ProviderIds)
//{
// var value = key.Value;
// if (!string.IsNullOrWhiteSpace(value))
// {
// item.SetProviderId(key.Key, value);
// }
//}
2016-01-15 19:15:51 +00:00
Logger.Info("Setting provider id's to item {0}-{1}: {2}", item.Id, item.Name, _json.SerializeToString(request.ProviderIds));
2016-04-06 02:18:56 +00:00
// Since the refresh process won't erase provider Ids, we need to set this explicitly now.
2014-10-14 04:59:34 +00:00
item.ProviderIds = request.ProviderIds;
2016-04-06 02:18:56 +00:00
//item.ProductionYear = request.ProductionYear;
//item.Name = request.Name;
2014-03-02 15:42:21 +00:00
2016-04-06 02:18:56 +00:00
var task = _providerManager.RefreshFullItem(item, new MetadataRefreshOptions(_fileSystem)
2014-03-02 15:42:21 +00:00
{
MetadataRefreshMode = MetadataRefreshMode.FullRefresh,
ImageRefreshMode = ImageRefreshMode.FullRefresh,
2014-06-13 14:24:14 +00:00
ReplaceAllMetadata = true,
2016-04-06 02:18:56 +00:00
ReplaceAllImages = request.ReplaceAllImages,
SearchResult = request
2015-03-14 15:38:16 +00:00
}, CancellationToken.None);
Task.WaitAll(task);
2014-03-02 15:42:21 +00:00
}
2014-03-01 22:34:27 +00:00
/// <summary>
/// Gets the remote image.
/// </summary>
/// <param name="request">The request.</param>
/// <returns>Task{System.Object}.</returns>
private async Task<object> GetRemoteImage(GetRemoteSearchImage request)
{
var urlHash = request.ImageUrl.GetMD5();
var pointerCachePath = GetFullCachePath(urlHash.ToString());
string contentPath;
try
{
using (var reader = new StreamReader(pointerCachePath))
{
contentPath = await reader.ReadToEndAsync().ConfigureAwait(false);
}
2016-04-06 02:18:56 +00:00
if (_fileSystem.FileExists(contentPath))
2014-03-01 22:34:27 +00:00
{
2016-06-19 06:18:29 +00:00
return await ResultFactory.GetStaticFileResult(Request, contentPath).ConfigureAwait(false);
2014-03-01 22:34:27 +00:00
}
}
catch (DirectoryNotFoundException)
{
// Means the file isn't cached yet
}
catch (FileNotFoundException)
{
// Means the file isn't cached yet
}
await DownloadImage(request.ProviderName, request.ImageUrl, urlHash, pointerCachePath).ConfigureAwait(false);
// Read the pointer file again
using (var reader = new StreamReader(pointerCachePath))
{
contentPath = await reader.ReadToEndAsync().ConfigureAwait(false);
}
2016-06-19 06:18:29 +00:00
return await ResultFactory.GetStaticFileResult(Request, contentPath).ConfigureAwait(false);
2014-03-01 22:34:27 +00:00
}
/// <summary>
/// Downloads the image.
/// </summary>
/// <param name="providerName">Name of the provider.</param>
/// <param name="url">The URL.</param>
/// <param name="urlHash">The URL hash.</param>
/// <param name="pointerCachePath">The pointer cache path.</param>
/// <returns>Task.</returns>
private async Task DownloadImage(string providerName, string url, Guid urlHash, string pointerCachePath)
{
var result = await _providerManager.GetSearchImage(providerName, url, CancellationToken.None).ConfigureAwait(false);
var ext = result.ContentType.Split('/').Last();
var fullCachePath = GetFullCachePath(urlHash + "." + ext);
2016-04-06 02:18:56 +00:00
_fileSystem.CreateDirectory(Path.GetDirectoryName(fullCachePath));
2014-03-01 22:34:27 +00:00
using (var stream = result.Content)
{
using (var filestream = _fileSystem.GetFileStream(fullCachePath, FileMode.Create, FileAccess.Write, FileShare.Read, true))
{
await stream.CopyToAsync(filestream).ConfigureAwait(false);
}
}
2016-04-06 02:18:56 +00:00
_fileSystem.CreateDirectory(Path.GetDirectoryName(pointerCachePath));
2014-03-01 22:34:27 +00:00
using (var writer = new StreamWriter(pointerCachePath))
{
await writer.WriteAsync(fullCachePath).ConfigureAwait(false);
}
}
/// <summary>
/// Gets the full cache path.
/// </summary>
/// <param name="filename">The filename.</param>
/// <returns>System.String.</returns>
private string GetFullCachePath(string filename)
{
return Path.Combine(_appPaths.CachePath, "remote-images", filename.Substring(0, 1), filename);
}
}
}