DtoBuilder.cs 34 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042
  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. }
  446. private void SetGameProperties(BaseItemDto dto, Game item)
  447. {
  448. dto.Players = item.PlayersSupported;
  449. dto.GameSystem = item.GameSystem;
  450. }
  451. /// <summary>
  452. /// Since it can be slow to make all of these calculations independently, this method will provide a way to do them all at once
  453. /// </summary>
  454. /// <param name="folder">The folder.</param>
  455. /// <param name="user">The user.</param>
  456. /// <param name="dto">The dto.</param>
  457. /// <param name="userDataRepository">The user data repository.</param>
  458. /// <returns>Task.</returns>
  459. private static void SetSpecialCounts(Folder folder, User user, BaseItemDto dto, IUserDataRepository userDataRepository)
  460. {
  461. var rcentlyAddedItemCount = 0;
  462. var recursiveItemCount = 0;
  463. var unplayed = 0;
  464. double totalPercentPlayed = 0;
  465. // Loop through each recursive child
  466. foreach (var child in folder.GetRecursiveChildren(user, true).Where(i => !i.IsFolder).ToList())
  467. {
  468. var userdata = userDataRepository.GetUserData(user.Id, child.GetUserDataKey());
  469. recursiveItemCount++;
  470. // Check is recently added
  471. if (child.IsRecentlyAdded())
  472. {
  473. rcentlyAddedItemCount++;
  474. }
  475. var isUnplayed = true;
  476. // Incrememt totalPercentPlayed
  477. if (userdata != null)
  478. {
  479. if (userdata.Played)
  480. {
  481. totalPercentPlayed += 100;
  482. isUnplayed = false;
  483. }
  484. else if (userdata.PlaybackPositionTicks > 0 && child.RunTimeTicks.HasValue && child.RunTimeTicks.Value > 0)
  485. {
  486. double itemPercent = userdata.PlaybackPositionTicks;
  487. itemPercent /= child.RunTimeTicks.Value;
  488. totalPercentPlayed += itemPercent;
  489. }
  490. }
  491. if (isUnplayed)
  492. {
  493. unplayed++;
  494. }
  495. }
  496. dto.RecursiveItemCount = recursiveItemCount;
  497. dto.RecentlyAddedItemCount = rcentlyAddedItemCount;
  498. dto.RecursiveUnplayedItemCount = unplayed;
  499. if (recursiveItemCount > 0)
  500. {
  501. dto.PlayedPercentage = totalPercentPlayed / recursiveItemCount;
  502. }
  503. }
  504. /// <summary>
  505. /// Attaches People DTO's to a DTOBaseItem
  506. /// </summary>
  507. /// <param name="dto">The dto.</param>
  508. /// <param name="item">The item.</param>
  509. /// <returns>Task.</returns>
  510. private async Task AttachPeople(BaseItemDto dto, BaseItem item)
  511. {
  512. // Ordering by person type to ensure actors and artists are at the front.
  513. // This is taking advantage of the fact that they both begin with A
  514. // This should be improved in the future
  515. var people = item.People.OrderBy(i => i.Type).ToList();
  516. // Attach People by transforming them into BaseItemPerson (DTO)
  517. dto.People = new BaseItemPerson[people.Count];
  518. var entities = await Task.WhenAll(people.Select(p => p.Name).Distinct(StringComparer.OrdinalIgnoreCase).Select(c =>
  519. Task.Run(async () =>
  520. {
  521. try
  522. {
  523. return await _libraryManager.GetPerson(c).ConfigureAwait(false);
  524. }
  525. catch (IOException ex)
  526. {
  527. _logger.ErrorException("Error getting person {0}", ex, c);
  528. return null;
  529. }
  530. })
  531. )).ConfigureAwait(false);
  532. var dictionary = entities.Where(i => i != null).ToDictionary(i => i.Name, StringComparer.OrdinalIgnoreCase);
  533. for (var i = 0; i < people.Count; i++)
  534. {
  535. var person = people[i];
  536. var baseItemPerson = new BaseItemPerson
  537. {
  538. Name = person.Name,
  539. Role = person.Role,
  540. Type = person.Type
  541. };
  542. Person entity;
  543. if (dictionary.TryGetValue(person.Name, out entity))
  544. {
  545. var primaryImagePath = entity.PrimaryImagePath;
  546. if (!string.IsNullOrEmpty(primaryImagePath))
  547. {
  548. baseItemPerson.PrimaryImageTag = GetImageCacheTag(entity, ImageType.Primary, primaryImagePath);
  549. }
  550. }
  551. dto.People[i] = baseItemPerson;
  552. }
  553. }
  554. /// <summary>
  555. /// Attaches the studios.
  556. /// </summary>
  557. /// <param name="dto">The dto.</param>
  558. /// <param name="item">The item.</param>
  559. /// <returns>Task.</returns>
  560. private async Task AttachStudios(BaseItemDto dto, BaseItem item)
  561. {
  562. var studios = item.Studios.ToList();
  563. dto.Studios = new StudioDto[studios.Count];
  564. var entities = await Task.WhenAll(studios.Distinct(StringComparer.OrdinalIgnoreCase).Select(c =>
  565. Task.Run(async () =>
  566. {
  567. try
  568. {
  569. return await _libraryManager.GetStudio(c).ConfigureAwait(false);
  570. }
  571. catch (IOException ex)
  572. {
  573. _logger.ErrorException("Error getting studio {0}", ex, c);
  574. return null;
  575. }
  576. })
  577. )).ConfigureAwait(false);
  578. var dictionary = entities.Where(i => i != null).ToDictionary(i => i.Name, StringComparer.OrdinalIgnoreCase);
  579. for (var i = 0; i < studios.Count; i++)
  580. {
  581. var studio = studios[i];
  582. var studioDto = new StudioDto
  583. {
  584. Name = studio
  585. };
  586. Studio entity;
  587. if (dictionary.TryGetValue(studio, out entity))
  588. {
  589. var primaryImagePath = entity.PrimaryImagePath;
  590. if (!string.IsNullOrEmpty(primaryImagePath))
  591. {
  592. studioDto.PrimaryImageTag = GetImageCacheTag(entity, ImageType.Primary, primaryImagePath);
  593. }
  594. }
  595. dto.Studios[i] = studioDto;
  596. }
  597. }
  598. /// <summary>
  599. /// If an item does not any backdrops, this can be used to find the first parent that does have one
  600. /// </summary>
  601. /// <param name="item">The item.</param>
  602. /// <returns>BaseItem.</returns>
  603. private BaseItem GetParentBackdropItem(BaseItem item)
  604. {
  605. var parent = item.Parent;
  606. while (parent != null)
  607. {
  608. if (parent.BackdropImagePaths != null && parent.BackdropImagePaths.Count > 0)
  609. {
  610. return parent;
  611. }
  612. parent = parent.Parent;
  613. }
  614. return null;
  615. }
  616. /// <summary>
  617. /// If an item does not have a logo, this can be used to find the first parent that does have one
  618. /// </summary>
  619. /// <param name="item">The item.</param>
  620. /// <param name="type">The type.</param>
  621. /// <returns>BaseItem.</returns>
  622. private BaseItem GetParentImageItem(BaseItem item, ImageType type)
  623. {
  624. var parent = item.Parent;
  625. while (parent != null)
  626. {
  627. if (parent.HasImage(type))
  628. {
  629. return parent;
  630. }
  631. parent = parent.Parent;
  632. }
  633. return null;
  634. }
  635. /// <summary>
  636. /// Converts a UserItemData to a DTOUserItemData
  637. /// </summary>
  638. /// <param name="data">The data.</param>
  639. /// <returns>DtoUserItemData.</returns>
  640. /// <exception cref="System.ArgumentNullException"></exception>
  641. public static UserItemDataDto GetUserItemDataDto(UserItemData data)
  642. {
  643. if (data == null)
  644. {
  645. throw new ArgumentNullException("data");
  646. }
  647. return new UserItemDataDto
  648. {
  649. IsFavorite = data.IsFavorite,
  650. Likes = data.Likes,
  651. PlaybackPositionTicks = data.PlaybackPositionTicks,
  652. PlayCount = data.PlayCount,
  653. Rating = data.Rating,
  654. Played = data.Played,
  655. LastPlayedDate = data.LastPlayedDate
  656. };
  657. }
  658. /// <summary>
  659. /// Gets the chapter info dto.
  660. /// </summary>
  661. /// <param name="chapterInfo">The chapter info.</param>
  662. /// <param name="item">The item.</param>
  663. /// <returns>ChapterInfoDto.</returns>
  664. private ChapterInfoDto GetChapterInfoDto(ChapterInfo chapterInfo, BaseItem item)
  665. {
  666. var dto = new ChapterInfoDto
  667. {
  668. Name = chapterInfo.Name,
  669. StartPositionTicks = chapterInfo.StartPositionTicks
  670. };
  671. if (!string.IsNullOrEmpty(chapterInfo.ImagePath))
  672. {
  673. dto.ImageTag = GetImageCacheTag(item, ImageType.Chapter, chapterInfo.ImagePath);
  674. }
  675. return dto;
  676. }
  677. /// <summary>
  678. /// Converts a BaseItem to a BaseItemInfo
  679. /// </summary>
  680. /// <param name="item">The item.</param>
  681. /// <returns>BaseItemInfo.</returns>
  682. /// <exception cref="System.ArgumentNullException">item</exception>
  683. public static BaseItemInfo GetBaseItemInfo(BaseItem item)
  684. {
  685. if (item == null)
  686. {
  687. throw new ArgumentNullException("item");
  688. }
  689. var info = new BaseItemInfo
  690. {
  691. Id = GetClientItemId(item),
  692. Name = item.Name,
  693. MediaType = item.MediaType,
  694. Type = item.GetType().Name,
  695. IsFolder = item.IsFolder,
  696. RunTimeTicks = item.RunTimeTicks
  697. };
  698. var imagePath = item.PrimaryImagePath;
  699. if (!string.IsNullOrEmpty(imagePath))
  700. {
  701. try
  702. {
  703. info.PrimaryImageTag = Kernel.Instance.ImageManager.GetImageCacheTag(item, ImageType.Primary, imagePath);
  704. }
  705. catch (IOException)
  706. {
  707. }
  708. }
  709. return info;
  710. }
  711. /// <summary>
  712. /// Gets client-side Id of a server-side BaseItem
  713. /// </summary>
  714. /// <param name="item">The item.</param>
  715. /// <returns>System.String.</returns>
  716. /// <exception cref="System.ArgumentNullException">item</exception>
  717. public static string GetClientItemId(BaseItem item)
  718. {
  719. if (item == null)
  720. {
  721. throw new ArgumentNullException("item");
  722. }
  723. var indexFolder = item as IndexFolder;
  724. if (indexFolder != null)
  725. {
  726. return GetClientItemId(indexFolder.Parent) + IndexFolderDelimeter + (indexFolder.IndexName ?? string.Empty) + IndexFolderDelimeter + indexFolder.Id;
  727. }
  728. return item.Id.ToString("N");
  729. }
  730. /// <summary>
  731. /// Gets a BaseItem based upon it's client-side item id
  732. /// </summary>
  733. /// <param name="id">The id.</param>
  734. /// <param name="userManager">The user manager.</param>
  735. /// <param name="libraryManager">The library manager.</param>
  736. /// <param name="userId">The user id.</param>
  737. /// <returns>BaseItem.</returns>
  738. public static BaseItem GetItemByClientId(string id, IUserManager userManager, ILibraryManager libraryManager, Guid? userId = null)
  739. {
  740. if (string.IsNullOrEmpty(id))
  741. {
  742. throw new ArgumentNullException("id");
  743. }
  744. // If the item is an indexed folder we have to do a special routine to get it
  745. var isIndexFolder = id.IndexOf(IndexFolderDelimeter, StringComparison.OrdinalIgnoreCase) != -1;
  746. if (isIndexFolder)
  747. {
  748. if (userId.HasValue)
  749. {
  750. return GetIndexFolder(id, userId.Value, userManager, libraryManager);
  751. }
  752. }
  753. BaseItem item = null;
  754. if (userId.HasValue || !isIndexFolder)
  755. {
  756. item = libraryManager.GetItemById(new Guid(id));
  757. }
  758. // If we still don't find it, look within individual user views
  759. if (item == null && !userId.HasValue && isIndexFolder)
  760. {
  761. foreach (var user in userManager.Users)
  762. {
  763. item = GetItemByClientId(id, userManager, libraryManager, user.Id);
  764. if (item != null)
  765. {
  766. break;
  767. }
  768. }
  769. }
  770. return item;
  771. }
  772. /// <summary>
  773. /// Finds an index folder based on an Id and userId
  774. /// </summary>
  775. /// <param name="id">The id.</param>
  776. /// <param name="userId">The user id.</param>
  777. /// <param name="userManager">The user manager.</param>
  778. /// <param name="libraryManager">The library manager.</param>
  779. /// <returns>BaseItem.</returns>
  780. private static BaseItem GetIndexFolder(string id, Guid userId, IUserManager userManager, ILibraryManager libraryManager)
  781. {
  782. var user = userManager.GetUserById(userId);
  783. var stringSeparators = new[] { IndexFolderDelimeter };
  784. // Split using the delimeter
  785. var values = id.Split(stringSeparators, StringSplitOptions.None).ToList();
  786. // Get the top folder normally using the first id
  787. var folder = GetItemByClientId(values[0], userManager, libraryManager, userId) as Folder;
  788. values.RemoveAt(0);
  789. // Get indexed folders using the remaining values in the id string
  790. return GetIndexFolder(values, folder, user);
  791. }
  792. /// <summary>
  793. /// Gets indexed folders based on a list of index names and folder id's
  794. /// </summary>
  795. /// <param name="values">The values.</param>
  796. /// <param name="parentFolder">The parent folder.</param>
  797. /// <param name="user">The user.</param>
  798. /// <returns>BaseItem.</returns>
  799. private static BaseItem GetIndexFolder(List<string> values, Folder parentFolder, User user)
  800. {
  801. // The index name is first
  802. var indexBy = values[0];
  803. // The index folder id is next
  804. var indexFolderId = new Guid(values[1]);
  805. // Remove them from the lst
  806. values.RemoveRange(0, 2);
  807. // Get the IndexFolder
  808. var indexFolder = parentFolder.GetChildren(user, false, indexBy).FirstOrDefault(i => i.Id == indexFolderId) as Folder;
  809. // Nested index folder
  810. if (values.Count > 0)
  811. {
  812. return GetIndexFolder(values, indexFolder, user);
  813. }
  814. return indexFolder;
  815. }
  816. /// <summary>
  817. /// Gets the backdrop image tags.
  818. /// </summary>
  819. /// <param name="item">The item.</param>
  820. /// <returns>List{System.String}.</returns>
  821. private List<Guid> GetBackdropImageTags(BaseItem item)
  822. {
  823. return item.BackdropImagePaths
  824. .Select(p => GetImageCacheTag(item, ImageType.Backdrop, p))
  825. .Where(i => i.HasValue)
  826. .Select(i => i.Value)
  827. .ToList();
  828. }
  829. /// <summary>
  830. /// Gets the screenshot image tags.
  831. /// </summary>
  832. /// <param name="item">The item.</param>
  833. /// <returns>List{Guid}.</returns>
  834. private List<Guid> GetScreenshotImageTags(BaseItem item)
  835. {
  836. return item.ScreenshotImagePaths
  837. .Select(p => GetImageCacheTag(item, ImageType.Screenshot, p))
  838. .Where(i => i.HasValue)
  839. .Select(i => i.Value)
  840. .ToList();
  841. }
  842. private Guid? GetImageCacheTag(BaseItem item, ImageType type, string path)
  843. {
  844. try
  845. {
  846. return Kernel.Instance.ImageManager.GetImageCacheTag(item, ImageType.Screenshot, path);
  847. }
  848. catch (IOException ex)
  849. {
  850. _logger.ErrorException("Error getting {0} image info for {1}", ex, type, path);
  851. return null;
  852. }
  853. }
  854. }
  855. }