CollectionFolderImageProvider.cs 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118
  1. using MediaBrowser.Common.Configuration;
  2. using MediaBrowser.Common.IO;
  3. using MediaBrowser.Controller.Drawing;
  4. using MediaBrowser.Controller.Entities;
  5. using MediaBrowser.Controller.Entities.Audio;
  6. using MediaBrowser.Controller.Entities.TV;
  7. using MediaBrowser.Controller.Providers;
  8. using MediaBrowser.Model.Entities;
  9. using MediaBrowser.Server.Implementations.Photos;
  10. using MoreLinq;
  11. using System;
  12. using System.Collections.Generic;
  13. using System.IO;
  14. using System.Linq;
  15. using System.Threading.Tasks;
  16. using CommonIO;
  17. namespace MediaBrowser.Server.Implementations.UserViews
  18. {
  19. public class CollectionFolderImageProvider : BaseDynamicImageProvider<CollectionFolder>
  20. {
  21. public CollectionFolderImageProvider(IFileSystem fileSystem, IProviderManager providerManager, IApplicationPaths applicationPaths, IImageProcessor imageProcessor) : base(fileSystem, providerManager, applicationPaths, imageProcessor)
  22. {
  23. }
  24. public override IEnumerable<ImageType> GetSupportedImages(IHasImages item)
  25. {
  26. return new List<ImageType>
  27. {
  28. ImageType.Primary
  29. };
  30. }
  31. protected override async Task<List<BaseItem>> GetItemsWithImages(IHasImages item)
  32. {
  33. var view = (CollectionFolder)item;
  34. var recursive = !new[] { CollectionType.Playlists, CollectionType.Channels }.Contains(view.CollectionType ?? string.Empty, StringComparer.OrdinalIgnoreCase);
  35. var result = await view.GetItems(new InternalItemsQuery
  36. {
  37. CollapseBoxSetItems = false,
  38. Recursive = recursive,
  39. ExcludeItemTypes = new[] { "UserView", "CollectionFolder", "Playlist" }
  40. }).ConfigureAwait(false);
  41. var items = result.Items.Select(i =>
  42. {
  43. var episode = i as Episode;
  44. if (episode != null)
  45. {
  46. var series = episode.Series;
  47. if (series != null)
  48. {
  49. return series;
  50. }
  51. var episodeSeason = episode.Season;
  52. if (episodeSeason != null)
  53. {
  54. return episodeSeason;
  55. }
  56. return episode;
  57. }
  58. var season = i as Season;
  59. if (season != null)
  60. {
  61. var series = season.Series;
  62. if (series != null)
  63. {
  64. return series;
  65. }
  66. return season;
  67. }
  68. var audio = i as Audio;
  69. if (audio != null)
  70. {
  71. var album = audio.AlbumEntity;
  72. if (album != null && album.HasImage(ImageType.Primary))
  73. {
  74. return album;
  75. }
  76. }
  77. return i;
  78. }).DistinctBy(i => i.Id);
  79. return GetFinalItems(items.Where(i => i.HasImage(ImageType.Primary) || i.HasImage(ImageType.Thumb)).ToList(), 8);
  80. }
  81. protected override bool Supports(IHasImages item)
  82. {
  83. return item is CollectionFolder;
  84. }
  85. protected override async Task<string> CreateImage(IHasImages item, List<BaseItem> itemsWithImages, string outputPathWithoutExtension, ImageType imageType, int imageIndex)
  86. {
  87. var outputPath = Path.ChangeExtension(outputPathWithoutExtension, ".png");
  88. if (imageType == ImageType.Primary)
  89. {
  90. if (itemsWithImages.Count == 0)
  91. {
  92. return null;
  93. }
  94. return await CreateThumbCollage(item, itemsWithImages, outputPath, 960, 540).ConfigureAwait(false);
  95. }
  96. return await base.CreateImage(item, itemsWithImages, outputPath, imageType, imageIndex).ConfigureAwait(false);
  97. }
  98. }
  99. }