ItemLookupService.cs 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313
  1. using MediaBrowser.Common.Extensions;
  2. using MediaBrowser.Common.IO;
  3. using MediaBrowser.Controller;
  4. using MediaBrowser.Controller.Entities;
  5. using MediaBrowser.Controller.Entities.Audio;
  6. using MediaBrowser.Controller.Entities.Movies;
  7. using MediaBrowser.Controller.Entities.TV;
  8. using MediaBrowser.Controller.Library;
  9. using MediaBrowser.Controller.Providers;
  10. using MediaBrowser.Model.Entities;
  11. using MediaBrowser.Model.Providers;
  12. using ServiceStack;
  13. using System;
  14. using System.Collections.Generic;
  15. using System.IO;
  16. using System.Linq;
  17. using System.Threading;
  18. using System.Threading.Tasks;
  19. namespace MediaBrowser.Api
  20. {
  21. [Route("/Items/{Id}/ExternalIdInfos", "GET")]
  22. [Api(Description = "Gets external id infos for an item")]
  23. public class GetExternalIdInfos : IReturn<List<ExternalIdInfo>>
  24. {
  25. /// <summary>
  26. /// Gets or sets the id.
  27. /// </summary>
  28. /// <value>The id.</value>
  29. [ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")]
  30. public string Id { get; set; }
  31. }
  32. [Route("/Items/RemoteSearch/Movie", "POST")]
  33. [Api(Description = "Gets external id infos for an item")]
  34. public class GetMovieRemoteSearchResults : RemoteSearchQuery<MovieInfo>, IReturn<List<RemoteSearchResult>>
  35. {
  36. }
  37. [Route("/Items/RemoteSearch/Trailer", "POST")]
  38. [Api(Description = "Gets external id infos for an item")]
  39. public class GetTrailerRemoteSearchResults : RemoteSearchQuery<TrailerInfo>, IReturn<List<RemoteSearchResult>>
  40. {
  41. }
  42. [Route("/Items/RemoteSearch/AdultVideo", "POST")]
  43. [Api(Description = "Gets external id infos for an item")]
  44. public class GetAdultVideoRemoteSearchResults : RemoteSearchQuery<ItemLookupInfo>, IReturn<List<RemoteSearchResult>>
  45. {
  46. }
  47. [Route("/Items/RemoteSearch/Series", "POST")]
  48. [Api(Description = "Gets external id infos for an item")]
  49. public class GetSeriesRemoteSearchResults : RemoteSearchQuery<SeriesInfo>, IReturn<List<RemoteSearchResult>>
  50. {
  51. }
  52. [Route("/Items/RemoteSearch/Game", "POST")]
  53. [Api(Description = "Gets external id infos for an item")]
  54. public class GetGameRemoteSearchResults : RemoteSearchQuery<GameInfo>, IReturn<List<RemoteSearchResult>>
  55. {
  56. }
  57. [Route("/Items/RemoteSearch/BoxSet", "POST")]
  58. [Api(Description = "Gets external id infos for an item")]
  59. public class GetBoxSetRemoteSearchResults : RemoteSearchQuery<BoxSetInfo>, IReturn<List<RemoteSearchResult>>
  60. {
  61. }
  62. [Route("/Items/RemoteSearch/MusicArtist", "POST")]
  63. [Api(Description = "Gets external id infos for an item")]
  64. public class GetMusicArtistRemoteSearchResults : RemoteSearchQuery<ArtistInfo>, IReturn<List<RemoteSearchResult>>
  65. {
  66. }
  67. [Route("/Items/RemoteSearch/MusicAlbum", "POST")]
  68. [Api(Description = "Gets external id infos for an item")]
  69. public class GetMusicAlbumRemoteSearchResults : RemoteSearchQuery<AlbumInfo>, IReturn<List<RemoteSearchResult>>
  70. {
  71. }
  72. [Route("/Items/RemoteSearch/Person", "POST")]
  73. [Api(Description = "Gets external id infos for an item")]
  74. public class GetPersonRemoteSearchResults : RemoteSearchQuery<PersonLookupInfo>, IReturn<List<RemoteSearchResult>>
  75. {
  76. }
  77. [Route("/Items/RemoteSearch/Image", "GET")]
  78. [Api(Description = "Gets a remote image")]
  79. public class GetRemoteSearchImage
  80. {
  81. [ApiMember(Name = "ImageUrl", Description = "The image url", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "GET")]
  82. public string ImageUrl { get; set; }
  83. [ApiMember(Name = "ProviderName", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "GET")]
  84. public string ProviderName { get; set; }
  85. }
  86. [Route("/Items/RemoteSearch/Apply/{Id}", "POST")]
  87. [Api(Description = "Applies search criteria to an item and refreshes metadata")]
  88. public class ApplySearchCriteria : RemoteSearchResult, IReturnVoid
  89. {
  90. [ApiMember(Name = "Id", Description = "The item id", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "GET")]
  91. public string Id { get; set; }
  92. }
  93. public class ItemLookupService : BaseApiService
  94. {
  95. private readonly IProviderManager _providerManager;
  96. private readonly IServerApplicationPaths _appPaths;
  97. private readonly IFileSystem _fileSystem;
  98. private readonly ILibraryManager _libraryManager;
  99. public ItemLookupService(IProviderManager providerManager, IServerApplicationPaths appPaths, IFileSystem fileSystem, ILibraryManager libraryManager)
  100. {
  101. _providerManager = providerManager;
  102. _appPaths = appPaths;
  103. _fileSystem = fileSystem;
  104. _libraryManager = libraryManager;
  105. }
  106. public object Get(GetExternalIdInfos request)
  107. {
  108. var item = _libraryManager.GetItemById(request.Id);
  109. var infos = _providerManager.GetExternalIdInfos(item).ToList();
  110. return ToOptimizedResult(infos);
  111. }
  112. public object Post(GetMovieRemoteSearchResults request)
  113. {
  114. var result = _providerManager.GetRemoteSearchResults<Movie, MovieInfo>(request, CancellationToken.None).Result;
  115. return ToOptimizedResult(result);
  116. }
  117. public object Post(GetAdultVideoRemoteSearchResults request)
  118. {
  119. var result = _providerManager.GetRemoteSearchResults<AdultVideo, ItemLookupInfo>(request, CancellationToken.None).Result;
  120. return ToOptimizedResult(result);
  121. }
  122. public object Post(GetSeriesRemoteSearchResults request)
  123. {
  124. var result = _providerManager.GetRemoteSearchResults<Series, SeriesInfo>(request, CancellationToken.None).Result;
  125. return ToOptimizedResult(result);
  126. }
  127. public object Post(GetGameRemoteSearchResults request)
  128. {
  129. var result = _providerManager.GetRemoteSearchResults<Game, GameInfo>(request, CancellationToken.None).Result;
  130. return ToOptimizedResult(result);
  131. }
  132. public object Post(GetBoxSetRemoteSearchResults request)
  133. {
  134. var result = _providerManager.GetRemoteSearchResults<BoxSet, BoxSetInfo>(request, CancellationToken.None).Result;
  135. return ToOptimizedResult(result);
  136. }
  137. public object Post(GetPersonRemoteSearchResults request)
  138. {
  139. var result = _providerManager.GetRemoteSearchResults<Person, PersonLookupInfo>(request, CancellationToken.None).Result;
  140. return ToOptimizedResult(result);
  141. }
  142. public object Post(GetTrailerRemoteSearchResults request)
  143. {
  144. var result = _providerManager.GetRemoteSearchResults<Trailer, TrailerInfo>(request, CancellationToken.None).Result;
  145. return ToOptimizedResult(result);
  146. }
  147. public object Post(GetMusicAlbumRemoteSearchResults request)
  148. {
  149. var result = _providerManager.GetRemoteSearchResults<MusicAlbum, AlbumInfo>(request, CancellationToken.None).Result;
  150. return ToOptimizedResult(result);
  151. }
  152. public object Post(GetMusicArtistRemoteSearchResults request)
  153. {
  154. var result = _providerManager.GetRemoteSearchResults<MusicArtist, ArtistInfo>(request, CancellationToken.None).Result;
  155. return ToOptimizedResult(result);
  156. }
  157. public object Get(GetRemoteSearchImage request)
  158. {
  159. var result = GetRemoteImage(request).Result;
  160. return result;
  161. }
  162. public void Post(ApplySearchCriteria request)
  163. {
  164. var item = _libraryManager.GetItemById(new Guid(request.Id));
  165. foreach (var key in request.ProviderIds)
  166. {
  167. var value = key.Value;
  168. if (!string.IsNullOrWhiteSpace(value))
  169. {
  170. item.SetProviderId(key.Key, value);
  171. }
  172. }
  173. var task = item.RefreshMetadata(new MetadataRefreshOptions
  174. {
  175. MetadataRefreshMode = MetadataRefreshMode.FullRefresh,
  176. ImageRefreshMode = ImageRefreshMode.FullRefresh,
  177. ReplaceAllMetadata = true
  178. }, CancellationToken.None);
  179. Task.WaitAll(task);
  180. }
  181. /// <summary>
  182. /// Gets the remote image.
  183. /// </summary>
  184. /// <param name="request">The request.</param>
  185. /// <returns>Task{System.Object}.</returns>
  186. private async Task<object> GetRemoteImage(GetRemoteSearchImage request)
  187. {
  188. var urlHash = request.ImageUrl.GetMD5();
  189. var pointerCachePath = GetFullCachePath(urlHash.ToString());
  190. string contentPath;
  191. try
  192. {
  193. using (var reader = new StreamReader(pointerCachePath))
  194. {
  195. contentPath = await reader.ReadToEndAsync().ConfigureAwait(false);
  196. }
  197. if (File.Exists(contentPath))
  198. {
  199. return ToStaticFileResult(contentPath);
  200. }
  201. }
  202. catch (DirectoryNotFoundException)
  203. {
  204. // Means the file isn't cached yet
  205. }
  206. catch (FileNotFoundException)
  207. {
  208. // Means the file isn't cached yet
  209. }
  210. await DownloadImage(request.ProviderName, request.ImageUrl, urlHash, pointerCachePath).ConfigureAwait(false);
  211. // Read the pointer file again
  212. using (var reader = new StreamReader(pointerCachePath))
  213. {
  214. contentPath = await reader.ReadToEndAsync().ConfigureAwait(false);
  215. }
  216. return ToStaticFileResult(contentPath);
  217. }
  218. /// <summary>
  219. /// Downloads the image.
  220. /// </summary>
  221. /// <param name="providerName">Name of the provider.</param>
  222. /// <param name="url">The URL.</param>
  223. /// <param name="urlHash">The URL hash.</param>
  224. /// <param name="pointerCachePath">The pointer cache path.</param>
  225. /// <returns>Task.</returns>
  226. private async Task DownloadImage(string providerName, string url, Guid urlHash, string pointerCachePath)
  227. {
  228. var result = await _providerManager.GetSearchImage(providerName, url, CancellationToken.None).ConfigureAwait(false);
  229. var ext = result.ContentType.Split('/').Last();
  230. var fullCachePath = GetFullCachePath(urlHash + "." + ext);
  231. Directory.CreateDirectory(Path.GetDirectoryName(fullCachePath));
  232. using (var stream = result.Content)
  233. {
  234. using (var filestream = _fileSystem.GetFileStream(fullCachePath, FileMode.Create, FileAccess.Write, FileShare.Read, true))
  235. {
  236. await stream.CopyToAsync(filestream).ConfigureAwait(false);
  237. }
  238. }
  239. Directory.CreateDirectory(Path.GetDirectoryName(pointerCachePath));
  240. using (var writer = new StreamWriter(pointerCachePath))
  241. {
  242. await writer.WriteAsync(fullCachePath).ConfigureAwait(false);
  243. }
  244. }
  245. /// <summary>
  246. /// Gets the full cache path.
  247. /// </summary>
  248. /// <param name="filename">The filename.</param>
  249. /// <returns>System.String.</returns>
  250. private string GetFullCachePath(string filename)
  251. {
  252. return Path.Combine(_appPaths.CachePath, "remote-images", filename.Substring(0, 1), filename);
  253. }
  254. }
  255. }