jellyfin/MediaBrowser.Server.Implementations/UserViews/DynamicImageProvider.cs

177 lines
5.9 KiB
C#
Raw Normal View History

2015-03-14 17:02:51 +00:00
using MediaBrowser.Common.Configuration;
2015-04-08 14:38:02 +00:00
using MediaBrowser.Controller.Drawing;
2014-10-29 22:01:02 +00:00
using MediaBrowser.Controller.Entities;
2015-03-13 19:52:49 +00:00
using MediaBrowser.Controller.Entities.Audio;
2014-11-11 03:41:55 +00:00
using MediaBrowser.Controller.Entities.TV;
2014-10-29 22:01:02 +00:00
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Entities;
2015-03-14 15:38:16 +00:00
using MediaBrowser.Server.Implementations.Photos;
2014-11-11 04:26:53 +00:00
using MoreLinq;
2014-10-29 22:01:02 +00:00
using System;
using System.Collections.Generic;
2015-10-26 05:29:32 +00:00
using System.IO;
2014-10-29 22:01:02 +00:00
using System.Linq;
using System.Threading.Tasks;
2015-10-04 04:23:11 +00:00
using CommonIO;
2014-10-29 22:01:02 +00:00
2015-03-14 15:38:16 +00:00
namespace MediaBrowser.Server.Implementations.UserViews
2014-10-29 22:01:02 +00:00
{
2015-04-26 04:39:40 +00:00
public class DynamicImageProvider : BaseDynamicImageProvider<UserView>
2014-10-29 22:01:02 +00:00
{
private readonly IUserManager _userManager;
2015-05-05 15:25:00 +00:00
public DynamicImageProvider(IFileSystem fileSystem, IProviderManager providerManager, IApplicationPaths applicationPaths, IImageProcessor imageProcessor, IUserManager userManager)
2015-04-08 15:45:30 +00:00
: base(fileSystem, providerManager, applicationPaths, imageProcessor)
2014-10-29 22:01:02 +00:00
{
2015-03-07 03:53:31 +00:00
_userManager = userManager;
2015-03-14 04:50:23 +00:00
}
public override IEnumerable<ImageType> GetSupportedImages(IHasImages item)
{
var view = (UserView)item;
if (IsUsingCollectionStrip(view))
{
return new List<ImageType>
{
ImageType.Primary
};
}
return new List<ImageType>
{
2015-06-28 03:29:50 +00:00
ImageType.Primary
2015-03-14 04:50:23 +00:00
};
2014-10-29 22:01:02 +00:00
}
protected override async Task<List<BaseItem>> GetItemsWithImages(IHasImages item)
{
var view = (UserView)item;
2015-03-14 15:38:16 +00:00
if (string.Equals(view.ViewType, CollectionType.LiveTv, StringComparison.OrdinalIgnoreCase))
{
return new List<BaseItem>();
}
2015-11-09 18:18:37 +00:00
if (string.Equals(view.ViewType, SpecialFolder.MovieGenre, StringComparison.OrdinalIgnoreCase) ||
2014-11-11 04:26:53 +00:00
string.Equals(view.ViewType, SpecialFolder.TvGenre, StringComparison.OrdinalIgnoreCase))
{
2015-04-17 04:53:47 +00:00
var userItemsResult = await view.GetItems(new InternalItemsQuery
2014-11-11 04:26:53 +00:00
{
2015-04-17 04:53:47 +00:00
CollapseBoxSetItems = false
});
return userItemsResult.Items.ToList();
2014-11-11 04:26:53 +00:00
}
2015-03-14 04:50:23 +00:00
var isUsingCollectionStrip = IsUsingCollectionStrip(view);
2015-09-15 18:09:44 +00:00
var recursive = isUsingCollectionStrip && !new[] { CollectionType.Channels, CollectionType.BoxSets, CollectionType.Playlists }.Contains(view.ViewType ?? string.Empty, StringComparer.OrdinalIgnoreCase);
2015-03-14 04:50:23 +00:00
2014-10-29 22:01:02 +00:00
var result = await view.GetItems(new InternalItemsQuery
{
2016-03-27 21:11:27 +00:00
User = view.UserId.HasValue ? _userManager.GetUserById(view.UserId.Value) : null,
2015-03-14 04:50:23 +00:00
CollapseBoxSetItems = false,
Recursive = recursive,
2016-08-09 05:08:21 +00:00
ExcludeItemTypes = new[] { "UserView", "CollectionFolder", "Person" },
2014-10-29 22:01:02 +00:00
}).ConfigureAwait(false);
2014-11-11 03:41:55 +00:00
var items = result.Items.Select(i =>
{
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;
}
2015-03-13 19:52:49 +00:00
var audio = i as Audio;
if (audio != null)
{
2015-05-04 14:35:38 +00:00
var album = audio.AlbumEntity;
2015-03-14 04:50:23 +00:00
if (album != null && album.HasImage(ImageType.Primary))
2015-03-13 19:52:49 +00:00
{
2015-03-14 04:50:23 +00:00
return album;
2015-03-13 19:52:49 +00:00
}
}
2014-11-11 03:41:55 +00:00
return i;
}).DistinctBy(i => i.Id);
2015-03-14 04:50:23 +00:00
if (isUsingCollectionStrip)
2015-03-13 19:52:49 +00:00
{
return GetFinalItems(items.Where(i => i.HasImage(ImageType.Primary) || i.HasImage(ImageType.Thumb)).ToList(), 8);
}
2014-11-11 03:41:55 +00:00
return GetFinalItems(items.Where(i => i.HasImage(ImageType.Primary)).ToList());
2014-10-29 22:01:02 +00:00
}
2015-10-14 05:02:30 +00:00
protected override bool Supports(IHasImages item)
2014-10-29 22:01:02 +00:00
{
var view = item as UserView;
2015-07-27 05:03:34 +00:00
if (view != null)
2014-10-29 22:01:02 +00:00
{
2016-03-27 21:11:27 +00:00
return IsUsingCollectionStrip(view);
2014-10-29 22:01:02 +00:00
}
return false;
}
2015-03-13 19:52:49 +00:00
private bool IsUsingCollectionStrip(UserView view)
{
2015-03-14 04:50:23 +00:00
string[] collectionStripViewTypes =
{
CollectionType.Movies,
CollectionType.TvShows,
2015-09-15 18:09:44 +00:00
CollectionType.Music,
CollectionType.Games,
CollectionType.Books,
CollectionType.MusicVideos,
CollectionType.HomeVideos,
CollectionType.BoxSets,
CollectionType.Playlists,
CollectionType.Photos,
string.Empty
2015-03-14 04:50:23 +00:00
};
return collectionStripViewTypes.Contains(view.ViewType ?? string.Empty);
2015-03-13 19:52:49 +00:00
}
2015-03-13 20:09:07 +00:00
2015-10-26 05:29:32 +00:00
protected override async Task<string> CreateImage(IHasImages item, List<BaseItem> itemsWithImages, string outputPathWithoutExtension, ImageType imageType, int imageIndex)
2015-03-13 20:09:07 +00:00
{
2015-10-26 05:29:32 +00:00
var outputPath = Path.ChangeExtension(outputPathWithoutExtension, ".png");
2015-03-13 20:09:07 +00:00
var view = (UserView)item;
if (imageType == ImageType.Primary && IsUsingCollectionStrip(view))
{
2015-08-28 04:19:08 +00:00
if (itemsWithImages.Count == 0)
2015-03-16 03:41:12 +00:00
{
2015-10-26 05:29:32 +00:00
return null;
2015-03-16 03:41:12 +00:00
}
2015-08-14 17:24:07 +00:00
return await CreateThumbCollage(item, itemsWithImages, outputPath, 960, 540).ConfigureAwait(false);
2015-03-13 20:09:07 +00:00
}
return await base.CreateImage(item, itemsWithImages, outputPath, imageType, imageIndex).ConfigureAwait(false);
2015-03-13 20:09:07 +00:00
}
2014-10-29 22:01:02 +00:00
}
}