jellyfin/Emby.Server.Implementations/UserViews/CollectionFolderImageProvider.cs

179 lines
6.2 KiB
C#
Raw Normal View History

2015-07-27 05:03:34 +00:00
using MediaBrowser.Common.Configuration;
using MediaBrowser.Controller.Drawing;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Entities.Audio;
using MediaBrowser.Controller.Entities.TV;
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Entities;
using System;
using System.Collections.Generic;
2015-10-26 05:29:32 +00:00
using System.IO;
2015-07-27 05:03:34 +00:00
using System.Linq;
using System.Threading.Tasks;
2016-11-03 07:14:14 +00:00
using Emby.Server.Implementations.Images;
2016-10-25 19:02:04 +00:00
using MediaBrowser.Model.IO;
2016-09-05 20:07:36 +00:00
using MediaBrowser.Controller.Collections;
2017-05-21 07:25:49 +00:00
using MediaBrowser.Controller.Dto;
2016-09-05 20:07:36 +00:00
using MediaBrowser.Controller.Entities.Movies;
2016-10-25 19:02:04 +00:00
using MediaBrowser.Controller.IO;
2016-09-05 20:07:36 +00:00
using MediaBrowser.Controller.Library;
2016-10-22 02:08:34 +00:00
using MediaBrowser.Model.Extensions;
2016-09-05 20:07:36 +00:00
using MediaBrowser.Model.Querying;
2015-07-27 05:03:34 +00:00
2016-11-03 07:14:14 +00:00
namespace Emby.Server.Implementations.UserViews
2015-07-27 05:03:34 +00:00
{
public class CollectionFolderImageProvider : BaseDynamicImageProvider<CollectionFolder>
{
public CollectionFolderImageProvider(IFileSystem fileSystem, IProviderManager providerManager, IApplicationPaths applicationPaths, IImageProcessor imageProcessor) : base(fileSystem, providerManager, applicationPaths, imageProcessor)
{
}
2017-08-07 21:06:13 +00:00
public override IEnumerable<ImageType> GetSupportedImages(IHasMetadata item)
2015-07-27 05:03:34 +00:00
{
return new List<ImageType>
{
ImageType.Primary
};
}
2017-08-07 21:06:13 +00:00
protected override List<BaseItem> GetItemsWithImages(IHasMetadata item)
2015-07-27 05:03:34 +00:00
{
var view = (CollectionFolder)item;
var recursive = !new[] { CollectionType.Playlists, CollectionType.Channels }.Contains(view.CollectionType ?? string.Empty, StringComparer.OrdinalIgnoreCase);
2017-08-01 16:45:57 +00:00
var result = view.GetItemList(new InternalItemsQuery
2015-07-27 05:03:34 +00:00
{
CollapseBoxSetItems = false,
Recursive = recursive,
2017-05-21 07:25:49 +00:00
ExcludeItemTypes = new[] { "UserView", "CollectionFolder", "Playlist" },
DtoOptions = new DtoOptions(false)
2015-07-27 05:03:34 +00:00
2017-05-26 06:48:54 +00:00
});
2015-07-27 05:03:34 +00:00
2017-08-01 16:45:57 +00:00
var items = result.Select(i =>
2015-07-27 05:03:34 +00:00
{
var episode = i as Episode;
if (episode != null)
{
var series = episode.Series;
if (series != null)
{
return series;
}
return episode;
}
var season = i as Season;
if (season != null)
{
var series = season.Series;
if (series != null)
{
return series;
}
return season;
}
var audio = i as Audio;
if (audio != null)
{
var album = audio.AlbumEntity;
if (album != null && album.HasImage(ImageType.Primary))
{
return album;
}
}
return i;
}).DistinctBy(i => i.Id);
return GetFinalItems(items.Where(i => i.HasImage(ImageType.Primary) || i.HasImage(ImageType.Thumb)), 8);
2015-07-27 05:03:34 +00:00
}
2017-08-07 21:06:13 +00:00
protected override bool Supports(IHasMetadata item)
2015-07-27 05:03:34 +00:00
{
return item is CollectionFolder;
}
2017-08-07 21:06:13 +00:00
protected override string CreateImage(IHasMetadata item, List<BaseItem> itemsWithImages, string outputPathWithoutExtension, ImageType imageType, int imageIndex)
2015-07-27 05:03:34 +00:00
{
2015-10-26 05:29:32 +00:00
var outputPath = Path.ChangeExtension(outputPathWithoutExtension, ".png");
2015-07-27 05:03:34 +00:00
if (imageType == ImageType.Primary)
{
if (itemsWithImages.Count == 0)
{
2015-10-26 05:29:32 +00:00
return null;
2015-07-27 05:03:34 +00:00
}
2017-05-19 17:09:37 +00:00
return CreateThumbCollage(item, itemsWithImages, outputPath, 960, 540);
2015-07-27 05:03:34 +00:00
}
2017-05-19 17:09:37 +00:00
return base.CreateImage(item, itemsWithImages, outputPath, imageType, imageIndex);
2015-07-27 05:03:34 +00:00
}
}
2016-09-05 20:07:36 +00:00
public class ManualCollectionFolderImageProvider : BaseDynamicImageProvider<ManualCollectionsFolder>
{
private readonly ILibraryManager _libraryManager;
public ManualCollectionFolderImageProvider(IFileSystem fileSystem, IProviderManager providerManager, IApplicationPaths applicationPaths, IImageProcessor imageProcessor, ILibraryManager libraryManager) : base(fileSystem, providerManager, applicationPaths, imageProcessor)
{
_libraryManager = libraryManager;
}
2017-08-07 21:06:13 +00:00
public override IEnumerable<ImageType> GetSupportedImages(IHasMetadata item)
2016-09-05 20:07:36 +00:00
{
return new List<ImageType>
{
ImageType.Primary
};
}
2017-08-07 21:06:13 +00:00
protected override List<BaseItem> GetItemsWithImages(IHasMetadata item)
2016-09-05 20:07:36 +00:00
{
var view = (ManualCollectionsFolder)item;
var recursive = !new[] { CollectionType.Playlists, CollectionType.Channels }.Contains(view.CollectionType ?? string.Empty, StringComparer.OrdinalIgnoreCase);
var items = _libraryManager.GetItemList(new InternalItemsQuery
{
Recursive = recursive,
IncludeItemTypes = new[] { typeof(BoxSet).Name },
Limit = 20,
2017-09-04 19:28:22 +00:00
OrderBy = new [] { new Tuple<string, SortOrder>(ItemSortBy.Random, SortOrder.Ascending) },
2017-05-21 07:25:49 +00:00
DtoOptions = new DtoOptions(false)
2016-09-05 20:07:36 +00:00
});
return GetFinalItems(items.Where(i => i.HasImage(ImageType.Primary) || i.HasImage(ImageType.Thumb)), 8);
2016-09-05 20:07:36 +00:00
}
2017-08-07 21:06:13 +00:00
protected override bool Supports(IHasMetadata item)
2016-09-05 20:07:36 +00:00
{
return item is ManualCollectionsFolder;
}
2017-08-07 21:06:13 +00:00
protected override string CreateImage(IHasMetadata item, List<BaseItem> itemsWithImages, string outputPathWithoutExtension, ImageType imageType, int imageIndex)
2016-09-05 20:07:36 +00:00
{
var outputPath = Path.ChangeExtension(outputPathWithoutExtension, ".png");
if (imageType == ImageType.Primary)
{
if (itemsWithImages.Count == 0)
{
return null;
}
2017-05-19 17:09:37 +00:00
return CreateThumbCollage(item, itemsWithImages, outputPath, 960, 540);
2016-09-05 20:07:36 +00:00
}
2017-05-19 17:09:37 +00:00
return base.CreateImage(item, itemsWithImages, outputPath, imageType, imageIndex);
2016-09-05 20:07:36 +00:00
}
}
2015-07-27 05:03:34 +00:00
}