mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-07 03:48:06 +02:00
172 lines
5.9 KiB
C#
172 lines
5.9 KiB
C#
using MediaBrowser.Common.Configuration;
|
|
using MediaBrowser.Common.IO;
|
|
using MediaBrowser.Controller.Drawing;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Entities.Audio;
|
|
using MediaBrowser.Controller.Entities.TV;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Controller.Providers;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Server.Implementations.Photos;
|
|
using MoreLinq;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace MediaBrowser.Server.Implementations.UserViews
|
|
{
|
|
public class DynamicImageProvider : BaseDynamicImageProvider<UserView>
|
|
{
|
|
private readonly IUserManager _userManager;
|
|
|
|
public DynamicImageProvider(IFileSystem fileSystem, IProviderManager providerManager, IApplicationPaths applicationPaths, IImageProcessor imageProcessor, IUserManager userManager)
|
|
: base(fileSystem, providerManager, applicationPaths, imageProcessor)
|
|
{
|
|
_userManager = userManager;
|
|
}
|
|
|
|
public override IEnumerable<ImageType> GetSupportedImages(IHasImages item)
|
|
{
|
|
var view = (UserView)item;
|
|
if (IsUsingCollectionStrip(view))
|
|
{
|
|
return new List<ImageType>
|
|
{
|
|
ImageType.Primary
|
|
};
|
|
}
|
|
|
|
return new List<ImageType>
|
|
{
|
|
ImageType.Primary
|
|
};
|
|
}
|
|
|
|
protected override async Task<List<BaseItem>> GetItemsWithImages(IHasImages item)
|
|
{
|
|
var view = (UserView)item;
|
|
|
|
if (string.Equals(view.ViewType, CollectionType.LiveTv, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return new List<BaseItem>();
|
|
}
|
|
|
|
if (string.Equals(view.ViewType, SpecialFolder.GameGenre, StringComparison.OrdinalIgnoreCase) ||
|
|
string.Equals(view.ViewType, SpecialFolder.MusicGenre, StringComparison.OrdinalIgnoreCase) ||
|
|
string.Equals(view.ViewType, SpecialFolder.MovieGenre, StringComparison.OrdinalIgnoreCase) ||
|
|
string.Equals(view.ViewType, SpecialFolder.TvGenre, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
var userItemsResult = await view.GetItems(new InternalItemsQuery
|
|
{
|
|
CollapseBoxSetItems = false
|
|
});
|
|
|
|
return userItemsResult.Items.ToList();
|
|
}
|
|
|
|
var isUsingCollectionStrip = IsUsingCollectionStrip(view);
|
|
var recursive = isUsingCollectionStrip && !new[] { CollectionType.Playlists, CollectionType.Channels }.Contains(view.ViewType ?? string.Empty, StringComparer.OrdinalIgnoreCase);
|
|
|
|
var result = await view.GetItems(new InternalItemsQuery
|
|
{
|
|
User = (view.UserId.HasValue ? _userManager.GetUserById(view.UserId.Value) : null),
|
|
CollapseBoxSetItems = false,
|
|
Recursive = recursive,
|
|
ExcludeItemTypes = new[] { "UserView", "CollectionFolder", "Playlist" }
|
|
|
|
}).ConfigureAwait(false);
|
|
|
|
var items = result.Items.Select(i =>
|
|
{
|
|
var episode = i as Episode;
|
|
if (episode != null)
|
|
{
|
|
var series = episode.Series;
|
|
if (series != null)
|
|
{
|
|
return series;
|
|
}
|
|
var episodeSeason = episode.Season;
|
|
if (episodeSeason != null)
|
|
{
|
|
return episodeSeason;
|
|
}
|
|
|
|
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);
|
|
|
|
if (isUsingCollectionStrip)
|
|
{
|
|
return GetFinalItems(items.Where(i => i.HasImage(ImageType.Primary) || i.HasImage(ImageType.Thumb)).ToList(), 8);
|
|
}
|
|
|
|
return GetFinalItems(items.Where(i => i.HasImage(ImageType.Primary)).ToList());
|
|
}
|
|
|
|
public override bool Supports(IHasImages item)
|
|
{
|
|
var view = item as UserView;
|
|
if (view != null)
|
|
{
|
|
return (IsUsingCollectionStrip(view));
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
private bool IsUsingCollectionStrip(UserView view)
|
|
{
|
|
string[] collectionStripViewTypes =
|
|
{
|
|
CollectionType.Movies,
|
|
CollectionType.TvShows,
|
|
CollectionType.Music
|
|
};
|
|
|
|
return collectionStripViewTypes.Contains(view.ViewType ?? string.Empty);
|
|
}
|
|
|
|
protected override async Task<bool> CreateImage(IHasImages item, List<BaseItem> itemsWithImages, string outputPath, ImageType imageType, int imageIndex)
|
|
{
|
|
var view = (UserView)item;
|
|
if (imageType == ImageType.Primary && IsUsingCollectionStrip(view))
|
|
{
|
|
if (itemsWithImages.Count == 0)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
return await CreateThumbCollage(item, itemsWithImages, outputPath, 960, 540).ConfigureAwait(false);
|
|
}
|
|
|
|
return await base.CreateImage(item, itemsWithImages, outputPath, imageType, imageIndex).ConfigureAwait(false);
|
|
}
|
|
}
|
|
}
|