DtoBuilder.cs 36 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109
  1. using MediaBrowser.Common.Extensions;
  2. using MediaBrowser.Controller.Entities;
  3. using MediaBrowser.Controller.Entities.Audio;
  4. using MediaBrowser.Controller.Entities.Movies;
  5. using MediaBrowser.Controller.Entities.TV;
  6. using MediaBrowser.Controller.Library;
  7. using MediaBrowser.Controller.Persistence;
  8. using MediaBrowser.Model.Drawing;
  9. using MediaBrowser.Model.Dto;
  10. using MediaBrowser.Model.Entities;
  11. using MediaBrowser.Model.Logging;
  12. using MediaBrowser.Model.Querying;
  13. using System;
  14. using System.Collections.Generic;
  15. using System.IO;
  16. using System.Linq;
  17. using System.Threading.Tasks;
  18. using MoreLinq;
  19. namespace MediaBrowser.Controller.Dto
  20. {
  21. /// <summary>
  22. /// Generates DTO's from domain entities
  23. /// </summary>
  24. public class DtoBuilder
  25. {
  26. /// <summary>
  27. /// The index folder delimeter
  28. /// </summary>
  29. const string IndexFolderDelimeter = "-index-";
  30. private readonly ILogger _logger;
  31. private readonly ILibraryManager _libraryManager;
  32. private readonly IUserDataRepository _userDataRepository;
  33. private readonly IItemRepository _itemRepo;
  34. public DtoBuilder(ILogger logger, ILibraryManager libraryManager, IUserDataRepository userDataRepository, IItemRepository itemRepo)
  35. {
  36. _logger = logger;
  37. _libraryManager = libraryManager;
  38. _userDataRepository = userDataRepository;
  39. _itemRepo = itemRepo;
  40. }
  41. /// <summary>
  42. /// Converts a BaseItem to a DTOBaseItem
  43. /// </summary>
  44. /// <param name="item">The item.</param>
  45. /// <param name="fields">The fields.</param>
  46. /// <param name="user">The user.</param>
  47. /// <param name="owner">The owner.</param>
  48. /// <returns>Task{DtoBaseItem}.</returns>
  49. /// <exception cref="System.ArgumentNullException">item</exception>
  50. public async Task<BaseItemDto> GetBaseItemDto(BaseItem item, List<ItemFields> fields, User user = null, BaseItem owner = null)
  51. {
  52. if (item == null)
  53. {
  54. throw new ArgumentNullException("item");
  55. }
  56. if (fields == null)
  57. {
  58. throw new ArgumentNullException("fields");
  59. }
  60. var dto = new BaseItemDto();
  61. var tasks = new List<Task>();
  62. if (fields.Contains(ItemFields.Studios))
  63. {
  64. tasks.Add(AttachStudios(dto, item));
  65. }
  66. if (fields.Contains(ItemFields.People))
  67. {
  68. tasks.Add(AttachPeople(dto, item));
  69. }
  70. if (fields.Contains(ItemFields.PrimaryImageAspectRatio))
  71. {
  72. try
  73. {
  74. await AttachPrimaryImageAspectRatio(dto, item, _logger).ConfigureAwait(false);
  75. }
  76. catch (Exception ex)
  77. {
  78. // Have to use a catch-all unfortunately because some .net image methods throw plain Exceptions
  79. _logger.ErrorException("Error generating PrimaryImageAspectRatio for {0}", ex, item.Name);
  80. }
  81. }
  82. if (fields.Contains(ItemFields.DisplayPreferencesId))
  83. {
  84. dto.DisplayPreferencesId = item.DisplayPreferencesId.ToString("N");
  85. }
  86. if (user != null)
  87. {
  88. AttachUserSpecificInfo(dto, item, user, fields);
  89. }
  90. AttachBasicFields(dto, item, owner, fields);
  91. if (fields.Contains(ItemFields.SoundtrackIds))
  92. {
  93. dto.SoundtrackIds = item.SoundtrackIds
  94. .Select(i => i.ToString("N"))
  95. .ToArray();
  96. }
  97. // Make sure all the tasks we kicked off have completed.
  98. if (tasks.Count > 0)
  99. {
  100. await Task.WhenAll(tasks).ConfigureAwait(false);
  101. }
  102. return dto;
  103. }
  104. /// <summary>
  105. /// Attaches the user specific info.
  106. /// </summary>
  107. /// <param name="dto">The dto.</param>
  108. /// <param name="item">The item.</param>
  109. /// <param name="user">The user.</param>
  110. /// <param name="fields">The fields.</param>
  111. private void AttachUserSpecificInfo(BaseItemDto dto, BaseItem item, User user, List<ItemFields> fields)
  112. {
  113. if (item.IsFolder)
  114. {
  115. var hasItemCounts = fields.Contains(ItemFields.ItemCounts);
  116. if (hasItemCounts || fields.Contains(ItemFields.CumulativeRunTimeTicks))
  117. {
  118. var folder = (Folder)item;
  119. if (hasItemCounts)
  120. {
  121. dto.ChildCount = folder.GetChildren(user, true).Count();
  122. }
  123. SetSpecialCounts(folder, user, dto, _userDataRepository);
  124. }
  125. }
  126. var userData = _userDataRepository.GetUserData(user.Id, item.GetUserDataKey());
  127. dto.UserData = GetUserItemDataDto(userData);
  128. if (item.IsFolder)
  129. {
  130. dto.UserData.Played = dto.PlayedPercentage.HasValue && dto.PlayedPercentage.Value >= 100;
  131. }
  132. }
  133. /// <summary>
  134. /// Attaches the primary image aspect ratio.
  135. /// </summary>
  136. /// <param name="dto">The dto.</param>
  137. /// <param name="item">The item.</param>
  138. /// <param name="logger">The _logger.</param>
  139. /// <returns>Task.</returns>
  140. internal static async Task AttachPrimaryImageAspectRatio(IItemDto dto, BaseItem item, ILogger logger)
  141. {
  142. var path = item.PrimaryImagePath;
  143. if (string.IsNullOrEmpty(path))
  144. {
  145. return;
  146. }
  147. var metaFileEntry = item.ResolveArgs.GetMetaFileByPath(path);
  148. // See if we can avoid a file system lookup by looking for the file in ResolveArgs
  149. var dateModified = metaFileEntry == null ? File.GetLastWriteTimeUtc(path) : metaFileEntry.LastWriteTimeUtc;
  150. ImageSize size;
  151. try
  152. {
  153. size = await Kernel.Instance.ImageManager.GetImageSize(path, dateModified).ConfigureAwait(false);
  154. }
  155. catch (FileNotFoundException)
  156. {
  157. logger.Error("Image file does not exist: {0}", path);
  158. return;
  159. }
  160. catch (Exception ex)
  161. {
  162. logger.ErrorException("Failed to determine primary image aspect ratio for {0}", ex, path);
  163. return;
  164. }
  165. dto.OriginalPrimaryImageAspectRatio = size.Width / size.Height;
  166. var supportedEnhancers = Kernel.Instance.ImageManager.ImageEnhancers.Where(i =>
  167. {
  168. try
  169. {
  170. return i.Supports(item, ImageType.Primary);
  171. }
  172. catch (Exception ex)
  173. {
  174. logger.ErrorException("Error in image enhancer: {0}", ex, i.GetType().Name);
  175. return false;
  176. }
  177. }).ToList();
  178. foreach (var enhancer in supportedEnhancers)
  179. {
  180. try
  181. {
  182. size = enhancer.GetEnhancedImageSize(item, ImageType.Primary, 0, size);
  183. }
  184. catch (Exception ex)
  185. {
  186. logger.ErrorException("Error in image enhancer: {0}", ex, enhancer.GetType().Name);
  187. }
  188. }
  189. dto.PrimaryImageAspectRatio = size.Width / size.Height;
  190. }
  191. /// <summary>
  192. /// Sets simple property values on a DTOBaseItem
  193. /// </summary>
  194. /// <param name="dto">The dto.</param>
  195. /// <param name="item">The item.</param>
  196. /// <param name="owner">The owner.</param>
  197. /// <param name="fields">The fields.</param>
  198. private void AttachBasicFields(BaseItemDto dto, BaseItem item, BaseItem owner, List<ItemFields> fields)
  199. {
  200. if (fields.Contains(ItemFields.DateCreated))
  201. {
  202. dto.DateCreated = item.DateCreated;
  203. }
  204. if (fields.Contains(ItemFields.OriginalRunTimeTicks))
  205. {
  206. dto.OriginalRunTimeTicks = item.OriginalRunTimeTicks;
  207. }
  208. dto.DisplayMediaType = item.DisplayMediaType;
  209. if (fields.Contains(ItemFields.MetadataSettings))
  210. {
  211. dto.LockedFields = item.LockedFields;
  212. dto.EnableInternetProviders = !item.DontFetchMeta;
  213. }
  214. if (fields.Contains(ItemFields.Budget))
  215. {
  216. dto.Budget = item.Budget;
  217. }
  218. if (fields.Contains(ItemFields.Revenue))
  219. {
  220. dto.Revenue = item.Revenue;
  221. }
  222. dto.EndDate = item.EndDate;
  223. if (fields.Contains(ItemFields.HomePageUrl))
  224. {
  225. dto.HomePageUrl = item.HomePageUrl;
  226. }
  227. if (fields.Contains(ItemFields.Tags))
  228. {
  229. dto.Tags = item.Tags;
  230. }
  231. if (fields.Contains(ItemFields.ProductionLocations))
  232. {
  233. dto.ProductionLocations = item.ProductionLocations;
  234. }
  235. dto.AspectRatio = item.AspectRatio;
  236. dto.BackdropImageTags = GetBackdropImageTags(item);
  237. dto.ScreenshotImageTags = GetScreenshotImageTags(item);
  238. if (fields.Contains(ItemFields.Genres))
  239. {
  240. dto.Genres = item.Genres;
  241. }
  242. dto.ImageTags = new Dictionary<ImageType, Guid>();
  243. foreach (var image in item.Images)
  244. {
  245. var type = image.Key;
  246. var tag = GetImageCacheTag(item, type, image.Value);
  247. if (tag.HasValue)
  248. {
  249. dto.ImageTags[type] = tag.Value;
  250. }
  251. }
  252. dto.Id = GetClientItemId(item);
  253. dto.IndexNumber = item.IndexNumber;
  254. dto.IsFolder = item.IsFolder;
  255. dto.Language = item.Language;
  256. dto.MediaType = item.MediaType;
  257. dto.LocationType = item.LocationType;
  258. dto.CriticRating = item.CriticRating;
  259. if (fields.Contains(ItemFields.CriticRatingSummary))
  260. {
  261. dto.CriticRatingSummary = item.CriticRatingSummary;
  262. }
  263. var localTrailerCount = item.LocalTrailerIds.Count;
  264. if (localTrailerCount > 0)
  265. {
  266. dto.LocalTrailerCount = localTrailerCount;
  267. }
  268. dto.Name = item.Name;
  269. dto.OfficialRating = item.OfficialRating;
  270. var hasOverview = fields.Contains(ItemFields.Overview);
  271. var hasHtmlOverview = fields.Contains(ItemFields.OverviewHtml);
  272. if (hasOverview || hasHtmlOverview)
  273. {
  274. var strippedOverview = string.IsNullOrEmpty(item.Overview) ? item.Overview : item.Overview.StripHtml();
  275. if (hasOverview)
  276. {
  277. dto.Overview = strippedOverview;
  278. }
  279. // Only supply the html version if there was actually html content
  280. if (hasHtmlOverview)
  281. {
  282. dto.OverviewHtml = item.Overview;
  283. }
  284. }
  285. // If there are no backdrops, indicate what parent has them in case the Ui wants to allow inheritance
  286. if (dto.BackdropImageTags.Count == 0)
  287. {
  288. var parentWithBackdrop = GetParentBackdropItem(item, owner);
  289. if (parentWithBackdrop != null)
  290. {
  291. dto.ParentBackdropItemId = GetClientItemId(parentWithBackdrop);
  292. dto.ParentBackdropImageTags = GetBackdropImageTags(parentWithBackdrop);
  293. }
  294. }
  295. if (item.Parent != null && fields.Contains(ItemFields.ParentId))
  296. {
  297. dto.ParentId = GetClientItemId(item.Parent);
  298. }
  299. dto.ParentIndexNumber = item.ParentIndexNumber;
  300. // If there is no logo, indicate what parent has one in case the Ui wants to allow inheritance
  301. if (!dto.HasLogo)
  302. {
  303. var parentWithLogo = GetParentImageItem(item, ImageType.Logo, owner);
  304. if (parentWithLogo != null)
  305. {
  306. dto.ParentLogoItemId = GetClientItemId(parentWithLogo);
  307. dto.ParentLogoImageTag = GetImageCacheTag(parentWithLogo, ImageType.Logo, parentWithLogo.GetImage(ImageType.Logo));
  308. }
  309. }
  310. // If there is no art, indicate what parent has one in case the Ui wants to allow inheritance
  311. if (!dto.HasArtImage)
  312. {
  313. var parentWithImage = GetParentImageItem(item, ImageType.Art, owner);
  314. if (parentWithImage != null)
  315. {
  316. dto.ParentArtItemId = GetClientItemId(parentWithImage);
  317. dto.ParentArtImageTag = GetImageCacheTag(parentWithImage, ImageType.Art, parentWithImage.GetImage(ImageType.Art));
  318. }
  319. }
  320. if (fields.Contains(ItemFields.Path))
  321. {
  322. dto.Path = item.Path;
  323. }
  324. dto.PremiereDate = item.PremiereDate;
  325. dto.ProductionYear = item.ProductionYear;
  326. if (fields.Contains(ItemFields.ProviderIds))
  327. {
  328. dto.ProviderIds = item.ProviderIds;
  329. }
  330. dto.RunTimeTicks = item.RunTimeTicks;
  331. if (fields.Contains(ItemFields.SortName))
  332. {
  333. dto.SortName = item.SortName;
  334. }
  335. if (fields.Contains(ItemFields.CustomRating))
  336. {
  337. dto.CustomRating = item.CustomRating;
  338. }
  339. if (fields.Contains(ItemFields.Taglines))
  340. {
  341. dto.Taglines = item.Taglines;
  342. }
  343. if (fields.Contains(ItemFields.RemoteTrailers))
  344. {
  345. dto.RemoteTrailers = item.RemoteTrailers;
  346. }
  347. dto.Type = item.GetType().Name;
  348. dto.CommunityRating = item.CommunityRating;
  349. if (item.IsFolder)
  350. {
  351. var folder = (Folder)item;
  352. if (fields.Contains(ItemFields.IndexOptions))
  353. {
  354. dto.IndexOptions = folder.IndexByOptionStrings.ToArray();
  355. }
  356. }
  357. // Add audio info
  358. var audio = item as Audio;
  359. if (audio != null)
  360. {
  361. dto.Album = audio.Album;
  362. dto.AlbumArtist = audio.AlbumArtist;
  363. dto.Artists = new[] { audio.Artist };
  364. var albumParent = audio.FindParent<MusicAlbum>();
  365. if (albumParent != null)
  366. {
  367. dto.AlbumId = GetClientItemId(albumParent);
  368. var imagePath = albumParent.PrimaryImagePath;
  369. if (!string.IsNullOrEmpty(imagePath))
  370. {
  371. dto.AlbumPrimaryImageTag = GetImageCacheTag(albumParent, ImageType.Primary, imagePath);
  372. }
  373. }
  374. }
  375. var album = item as MusicAlbum;
  376. if (album != null)
  377. {
  378. var songs = album.RecursiveChildren.OfType<Audio>().ToList();
  379. dto.AlbumArtist = songs.Select(i => i.AlbumArtist).FirstOrDefault(i => !string.IsNullOrEmpty(i));
  380. dto.Artists =
  381. songs.Select(i => i.Artist ?? string.Empty)
  382. .Where(i => !string.IsNullOrEmpty(i))
  383. .Distinct(StringComparer.OrdinalIgnoreCase)
  384. .ToArray();
  385. }
  386. // Add video info
  387. var video = item as Video;
  388. if (video != null)
  389. {
  390. dto.VideoType = video.VideoType;
  391. dto.Video3DFormat = video.Video3DFormat;
  392. dto.IsoType = video.IsoType;
  393. dto.PartCount = video.AdditionalPartIds.Count + 1;
  394. if (fields.Contains(ItemFields.Chapters))
  395. {
  396. dto.Chapters = _itemRepo.GetChapters(video.Id).Select(c => GetChapterInfoDto(c, item)).ToList();
  397. }
  398. }
  399. if (fields.Contains(ItemFields.MediaStreams))
  400. {
  401. // Add VideoInfo
  402. var iHasMediaStreams = item as IHasMediaStreams;
  403. if (iHasMediaStreams != null)
  404. {
  405. dto.MediaStreams = iHasMediaStreams.MediaStreams;
  406. }
  407. }
  408. // Add MovieInfo
  409. var movie = item as Movie;
  410. if (movie != null)
  411. {
  412. var specialFeatureCount = movie.SpecialFeatureIds.Count;
  413. if (specialFeatureCount > 0)
  414. {
  415. dto.SpecialFeatureCount = specialFeatureCount;
  416. }
  417. }
  418. // Add EpisodeInfo
  419. var episode = item as Episode;
  420. if (episode != null)
  421. {
  422. dto.IndexNumberEnd = episode.IndexNumberEnd;
  423. }
  424. // Add SeriesInfo
  425. var series = item as Series;
  426. if (series != null)
  427. {
  428. dto.AirDays = series.AirDays;
  429. dto.AirTime = series.AirTime;
  430. dto.Status = series.Status;
  431. dto.SpecialFeatureCount = series.SpecialFeatureIds.Count;
  432. dto.SeasonCount = series.SeasonCount;
  433. }
  434. if (episode != null)
  435. {
  436. series = item.FindParent<Series>();
  437. dto.SeriesId = GetClientItemId(series);
  438. dto.SeriesName = series.Name;
  439. }
  440. // Add SeasonInfo
  441. var season = item as Season;
  442. if (season != null)
  443. {
  444. series = item.FindParent<Series>();
  445. dto.SeriesId = GetClientItemId(series);
  446. dto.SeriesName = series.Name;
  447. }
  448. var game = item as Game;
  449. if (game != null)
  450. {
  451. SetGameProperties(dto, game);
  452. }
  453. var musicVideo = item as MusicVideo;
  454. if (musicVideo != null)
  455. {
  456. SetMusicVideoProperties(dto, musicVideo);
  457. }
  458. var book = item as Book;
  459. if (book != null)
  460. {
  461. SetBookProperties(dto, book);
  462. }
  463. }
  464. private void SetBookProperties(BaseItemDto dto, Book item)
  465. {
  466. dto.SeriesName = item.SeriesName;
  467. }
  468. private void SetMusicVideoProperties(BaseItemDto dto, MusicVideo item)
  469. {
  470. if (!string.IsNullOrEmpty(item.Album))
  471. {
  472. var parentAlbum = _libraryManager.RootFolder
  473. .RecursiveChildren
  474. .OfType<MusicAlbum>()
  475. .FirstOrDefault(i => string.Equals(i.Name, item.Album, StringComparison.OrdinalIgnoreCase));
  476. if (parentAlbum != null)
  477. {
  478. dto.AlbumId = GetClientItemId(parentAlbum);
  479. }
  480. }
  481. dto.Album = item.Album;
  482. dto.Artists = string.IsNullOrEmpty(item.Artist) ? new string[] { } : new[] { item.Artist };
  483. }
  484. private void SetGameProperties(BaseItemDto dto, Game item)
  485. {
  486. dto.Players = item.PlayersSupported;
  487. dto.GameSystem = item.GameSystem;
  488. }
  489. /// <summary>
  490. /// Since it can be slow to make all of these calculations independently, this method will provide a way to do them all at once
  491. /// </summary>
  492. /// <param name="folder">The folder.</param>
  493. /// <param name="user">The user.</param>
  494. /// <param name="dto">The dto.</param>
  495. /// <param name="userDataRepository">The user data repository.</param>
  496. /// <returns>Task.</returns>
  497. private static void SetSpecialCounts(Folder folder, User user, BaseItemDto dto, IUserDataRepository userDataRepository)
  498. {
  499. var rcentlyAddedItemCount = 0;
  500. var recursiveItemCount = 0;
  501. var unplayed = 0;
  502. long runtime = 0;
  503. double totalPercentPlayed = 0;
  504. // Loop through each recursive child
  505. foreach (var child in folder.GetRecursiveChildren(user, true).Where(i => !i.IsFolder).ToList())
  506. {
  507. var userdata = userDataRepository.GetUserData(user.Id, child.GetUserDataKey());
  508. recursiveItemCount++;
  509. // Check is recently added
  510. if (child.IsRecentlyAdded())
  511. {
  512. rcentlyAddedItemCount++;
  513. }
  514. var isUnplayed = true;
  515. // Incrememt totalPercentPlayed
  516. if (userdata != null)
  517. {
  518. if (userdata.Played)
  519. {
  520. totalPercentPlayed += 100;
  521. isUnplayed = false;
  522. }
  523. else if (userdata.PlaybackPositionTicks > 0 && child.RunTimeTicks.HasValue && child.RunTimeTicks.Value > 0)
  524. {
  525. double itemPercent = userdata.PlaybackPositionTicks;
  526. itemPercent /= child.RunTimeTicks.Value;
  527. totalPercentPlayed += itemPercent;
  528. }
  529. }
  530. if (isUnplayed)
  531. {
  532. unplayed++;
  533. }
  534. runtime += child.RunTimeTicks ?? 0;
  535. }
  536. dto.RecursiveItemCount = recursiveItemCount;
  537. dto.RecentlyAddedItemCount = rcentlyAddedItemCount;
  538. dto.RecursiveUnplayedItemCount = unplayed;
  539. if (recursiveItemCount > 0)
  540. {
  541. dto.PlayedPercentage = totalPercentPlayed / recursiveItemCount;
  542. }
  543. if (runtime > 0)
  544. {
  545. dto.CumulativeRunTimeTicks = runtime;
  546. }
  547. }
  548. /// <summary>
  549. /// Attaches People DTO's to a DTOBaseItem
  550. /// </summary>
  551. /// <param name="dto">The dto.</param>
  552. /// <param name="item">The item.</param>
  553. /// <returns>Task.</returns>
  554. private async Task AttachPeople(BaseItemDto dto, BaseItem item)
  555. {
  556. // Ordering by person type to ensure actors and artists are at the front.
  557. // This is taking advantage of the fact that they both begin with A
  558. // This should be improved in the future
  559. var people = item.People.OrderBy(i => i.Type).ToList();
  560. // Attach People by transforming them into BaseItemPerson (DTO)
  561. dto.People = new BaseItemPerson[people.Count];
  562. var entities = await Task.WhenAll(people.Select(p => p.Name)
  563. .Distinct(StringComparer.OrdinalIgnoreCase).Select(c =>
  564. Task.Run(async () =>
  565. {
  566. try
  567. {
  568. return await _libraryManager.GetPerson(c).ConfigureAwait(false);
  569. }
  570. catch (IOException ex)
  571. {
  572. _logger.ErrorException("Error getting person {0}", ex, c);
  573. return null;
  574. }
  575. })
  576. )).ConfigureAwait(false);
  577. var dictionary = entities.Where(i => i != null)
  578. .DistinctBy(i => i.Name)
  579. .ToDictionary(i => i.Name, StringComparer.OrdinalIgnoreCase);
  580. for (var i = 0; i < people.Count; i++)
  581. {
  582. var person = people[i];
  583. var baseItemPerson = new BaseItemPerson
  584. {
  585. Name = person.Name,
  586. Role = person.Role,
  587. Type = person.Type
  588. };
  589. Person entity;
  590. if (dictionary.TryGetValue(person.Name, out entity))
  591. {
  592. var primaryImagePath = entity.PrimaryImagePath;
  593. if (!string.IsNullOrEmpty(primaryImagePath))
  594. {
  595. baseItemPerson.PrimaryImageTag = GetImageCacheTag(entity, ImageType.Primary, primaryImagePath);
  596. }
  597. }
  598. dto.People[i] = baseItemPerson;
  599. }
  600. }
  601. /// <summary>
  602. /// Attaches the studios.
  603. /// </summary>
  604. /// <param name="dto">The dto.</param>
  605. /// <param name="item">The item.</param>
  606. /// <returns>Task.</returns>
  607. private async Task AttachStudios(BaseItemDto dto, BaseItem item)
  608. {
  609. var studios = item.Studios.ToList();
  610. dto.Studios = new StudioDto[studios.Count];
  611. var entities = await Task.WhenAll(studios.Distinct(StringComparer.OrdinalIgnoreCase).Select(c =>
  612. Task.Run(async () =>
  613. {
  614. try
  615. {
  616. return await _libraryManager.GetStudio(c).ConfigureAwait(false);
  617. }
  618. catch (IOException ex)
  619. {
  620. _logger.ErrorException("Error getting studio {0}", ex, c);
  621. return null;
  622. }
  623. })
  624. )).ConfigureAwait(false);
  625. var dictionary = entities
  626. .Where(i => i != null)
  627. .ToDictionary(i => i.Name, StringComparer.OrdinalIgnoreCase);
  628. for (var i = 0; i < studios.Count; i++)
  629. {
  630. var studio = studios[i];
  631. var studioDto = new StudioDto
  632. {
  633. Name = studio
  634. };
  635. Studio entity;
  636. if (dictionary.TryGetValue(studio, out entity))
  637. {
  638. var primaryImagePath = entity.PrimaryImagePath;
  639. if (!string.IsNullOrEmpty(primaryImagePath))
  640. {
  641. studioDto.PrimaryImageTag = GetImageCacheTag(entity, ImageType.Primary, primaryImagePath);
  642. }
  643. }
  644. dto.Studios[i] = studioDto;
  645. }
  646. }
  647. /// <summary>
  648. /// If an item does not any backdrops, this can be used to find the first parent that does have one
  649. /// </summary>
  650. /// <param name="item">The item.</param>
  651. /// <param name="owner">The owner.</param>
  652. /// <returns>BaseItem.</returns>
  653. private BaseItem GetParentBackdropItem(BaseItem item, BaseItem owner)
  654. {
  655. var parent = item.Parent ?? owner;
  656. while (parent != null)
  657. {
  658. if (parent.BackdropImagePaths != null && parent.BackdropImagePaths.Count > 0)
  659. {
  660. return parent;
  661. }
  662. parent = parent.Parent;
  663. }
  664. return null;
  665. }
  666. /// <summary>
  667. /// If an item does not have a logo, this can be used to find the first parent that does have one
  668. /// </summary>
  669. /// <param name="item">The item.</param>
  670. /// <param name="type">The type.</param>
  671. /// <param name="owner">The owner.</param>
  672. /// <returns>BaseItem.</returns>
  673. private BaseItem GetParentImageItem(BaseItem item, ImageType type, BaseItem owner)
  674. {
  675. var parent = item.Parent ?? owner;
  676. while (parent != null)
  677. {
  678. if (parent.HasImage(type))
  679. {
  680. return parent;
  681. }
  682. parent = parent.Parent;
  683. }
  684. return null;
  685. }
  686. /// <summary>
  687. /// Converts a UserItemData to a DTOUserItemData
  688. /// </summary>
  689. /// <param name="data">The data.</param>
  690. /// <returns>DtoUserItemData.</returns>
  691. /// <exception cref="System.ArgumentNullException"></exception>
  692. public static UserItemDataDto GetUserItemDataDto(UserItemData data)
  693. {
  694. if (data == null)
  695. {
  696. throw new ArgumentNullException("data");
  697. }
  698. return new UserItemDataDto
  699. {
  700. IsFavorite = data.IsFavorite,
  701. Likes = data.Likes,
  702. PlaybackPositionTicks = data.PlaybackPositionTicks,
  703. PlayCount = data.PlayCount,
  704. Rating = data.Rating,
  705. Played = data.Played,
  706. LastPlayedDate = data.LastPlayedDate
  707. };
  708. }
  709. /// <summary>
  710. /// Gets the chapter info dto.
  711. /// </summary>
  712. /// <param name="chapterInfo">The chapter info.</param>
  713. /// <param name="item">The item.</param>
  714. /// <returns>ChapterInfoDto.</returns>
  715. private ChapterInfoDto GetChapterInfoDto(ChapterInfo chapterInfo, BaseItem item)
  716. {
  717. var dto = new ChapterInfoDto
  718. {
  719. Name = chapterInfo.Name,
  720. StartPositionTicks = chapterInfo.StartPositionTicks
  721. };
  722. if (!string.IsNullOrEmpty(chapterInfo.ImagePath))
  723. {
  724. dto.ImageTag = GetImageCacheTag(item, ImageType.Chapter, chapterInfo.ImagePath);
  725. }
  726. return dto;
  727. }
  728. /// <summary>
  729. /// Converts a BaseItem to a BaseItemInfo
  730. /// </summary>
  731. /// <param name="item">The item.</param>
  732. /// <returns>BaseItemInfo.</returns>
  733. /// <exception cref="System.ArgumentNullException">item</exception>
  734. public static BaseItemInfo GetBaseItemInfo(BaseItem item)
  735. {
  736. if (item == null)
  737. {
  738. throw new ArgumentNullException("item");
  739. }
  740. var info = new BaseItemInfo
  741. {
  742. Id = GetClientItemId(item),
  743. Name = item.Name,
  744. MediaType = item.MediaType,
  745. Type = item.GetType().Name,
  746. IsFolder = item.IsFolder,
  747. RunTimeTicks = item.RunTimeTicks
  748. };
  749. var imagePath = item.PrimaryImagePath;
  750. if (!string.IsNullOrEmpty(imagePath))
  751. {
  752. try
  753. {
  754. info.PrimaryImageTag = Kernel.Instance.ImageManager.GetImageCacheTag(item, ImageType.Primary, imagePath);
  755. }
  756. catch (IOException)
  757. {
  758. }
  759. }
  760. return info;
  761. }
  762. /// <summary>
  763. /// Gets client-side Id of a server-side BaseItem
  764. /// </summary>
  765. /// <param name="item">The item.</param>
  766. /// <returns>System.String.</returns>
  767. /// <exception cref="System.ArgumentNullException">item</exception>
  768. public static string GetClientItemId(BaseItem item)
  769. {
  770. if (item == null)
  771. {
  772. throw new ArgumentNullException("item");
  773. }
  774. var indexFolder = item as IndexFolder;
  775. if (indexFolder != null)
  776. {
  777. return GetClientItemId(indexFolder.Parent) + IndexFolderDelimeter + (indexFolder.IndexName ?? string.Empty) + IndexFolderDelimeter + indexFolder.Id;
  778. }
  779. return item.Id.ToString("N");
  780. }
  781. /// <summary>
  782. /// Gets a BaseItem based upon it's client-side item id
  783. /// </summary>
  784. /// <param name="id">The id.</param>
  785. /// <param name="userManager">The user manager.</param>
  786. /// <param name="libraryManager">The library manager.</param>
  787. /// <param name="userId">The user id.</param>
  788. /// <returns>BaseItem.</returns>
  789. public static BaseItem GetItemByClientId(string id, IUserManager userManager, ILibraryManager libraryManager, Guid? userId = null)
  790. {
  791. if (string.IsNullOrEmpty(id))
  792. {
  793. throw new ArgumentNullException("id");
  794. }
  795. // If the item is an indexed folder we have to do a special routine to get it
  796. var isIndexFolder = id.IndexOf(IndexFolderDelimeter, StringComparison.OrdinalIgnoreCase) != -1;
  797. if (isIndexFolder)
  798. {
  799. if (userId.HasValue)
  800. {
  801. return GetIndexFolder(id, userId.Value, userManager, libraryManager);
  802. }
  803. }
  804. BaseItem item = null;
  805. if (userId.HasValue || !isIndexFolder)
  806. {
  807. item = libraryManager.GetItemById(new Guid(id));
  808. }
  809. // If we still don't find it, look within individual user views
  810. if (item == null && !userId.HasValue && isIndexFolder)
  811. {
  812. foreach (var user in userManager.Users)
  813. {
  814. item = GetItemByClientId(id, userManager, libraryManager, user.Id);
  815. if (item != null)
  816. {
  817. break;
  818. }
  819. }
  820. }
  821. return item;
  822. }
  823. /// <summary>
  824. /// Finds an index folder based on an Id and userId
  825. /// </summary>
  826. /// <param name="id">The id.</param>
  827. /// <param name="userId">The user id.</param>
  828. /// <param name="userManager">The user manager.</param>
  829. /// <param name="libraryManager">The library manager.</param>
  830. /// <returns>BaseItem.</returns>
  831. private static BaseItem GetIndexFolder(string id, Guid userId, IUserManager userManager, ILibraryManager libraryManager)
  832. {
  833. var user = userManager.GetUserById(userId);
  834. var stringSeparators = new[] { IndexFolderDelimeter };
  835. // Split using the delimeter
  836. var values = id.Split(stringSeparators, StringSplitOptions.None).ToList();
  837. // Get the top folder normally using the first id
  838. var folder = GetItemByClientId(values[0], userManager, libraryManager, userId) as Folder;
  839. values.RemoveAt(0);
  840. // Get indexed folders using the remaining values in the id string
  841. return GetIndexFolder(values, folder, user);
  842. }
  843. /// <summary>
  844. /// Gets indexed folders based on a list of index names and folder id's
  845. /// </summary>
  846. /// <param name="values">The values.</param>
  847. /// <param name="parentFolder">The parent folder.</param>
  848. /// <param name="user">The user.</param>
  849. /// <returns>BaseItem.</returns>
  850. private static BaseItem GetIndexFolder(List<string> values, Folder parentFolder, User user)
  851. {
  852. // The index name is first
  853. var indexBy = values[0];
  854. // The index folder id is next
  855. var indexFolderId = new Guid(values[1]);
  856. // Remove them from the lst
  857. values.RemoveRange(0, 2);
  858. // Get the IndexFolder
  859. var indexFolder = parentFolder.GetChildren(user, false, indexBy).FirstOrDefault(i => i.Id == indexFolderId) as Folder;
  860. // Nested index folder
  861. if (values.Count > 0)
  862. {
  863. return GetIndexFolder(values, indexFolder, user);
  864. }
  865. return indexFolder;
  866. }
  867. /// <summary>
  868. /// Gets the backdrop image tags.
  869. /// </summary>
  870. /// <param name="item">The item.</param>
  871. /// <returns>List{System.String}.</returns>
  872. private List<Guid> GetBackdropImageTags(BaseItem item)
  873. {
  874. return item.BackdropImagePaths
  875. .Select(p => GetImageCacheTag(item, ImageType.Backdrop, p))
  876. .Where(i => i.HasValue)
  877. .Select(i => i.Value)
  878. .ToList();
  879. }
  880. /// <summary>
  881. /// Gets the screenshot image tags.
  882. /// </summary>
  883. /// <param name="item">The item.</param>
  884. /// <returns>List{Guid}.</returns>
  885. private List<Guid> GetScreenshotImageTags(BaseItem item)
  886. {
  887. return item.ScreenshotImagePaths
  888. .Select(p => GetImageCacheTag(item, ImageType.Screenshot, p))
  889. .Where(i => i.HasValue)
  890. .Select(i => i.Value)
  891. .ToList();
  892. }
  893. private Guid? GetImageCacheTag(BaseItem item, ImageType type, string path)
  894. {
  895. try
  896. {
  897. return Kernel.Instance.ImageManager.GetImageCacheTag(item, type, path);
  898. }
  899. catch (IOException ex)
  900. {
  901. _logger.ErrorException("Error getting {0} image info for {1}", ex, type, path);
  902. return null;
  903. }
  904. }
  905. }
  906. }