DtoBuilder.cs 35 KB

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