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

251 lines
9.2 KiB
C#
Raw Normal View History

2015-03-14 10:02:51 -07:00
using MediaBrowser.Common.Configuration;
using MediaBrowser.Common.IO;
2015-04-08 07:38:02 -07:00
using MediaBrowser.Controller.Drawing;
2014-10-29 15:01:02 -07:00
using MediaBrowser.Controller.Entities;
2015-03-13 12:52:49 -07:00
using MediaBrowser.Controller.Entities.Audio;
2014-11-10 20:41:55 -07:00
using MediaBrowser.Controller.Entities.TV;
2014-10-29 15:01:02 -07:00
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Entities;
2015-03-14 08:38:16 -07:00
using MediaBrowser.Server.Implementations.Photos;
2014-11-10 21:26:53 -07:00
using MoreLinq;
2014-10-29 15:01:02 -07:00
using System;
using System.Collections.Generic;
2015-03-14 10:02:51 -07:00
using System.Globalization;
2015-03-13 13:09:07 -07:00
using System.IO;
2014-10-29 15:01:02 -07:00
using System.Linq;
using System.Threading.Tasks;
2015-03-14 08:38:16 -07:00
namespace MediaBrowser.Server.Implementations.UserViews
2014-10-29 15:01:02 -07:00
{
2015-04-25 21:39:40 -07:00
public class DynamicImageProvider : BaseDynamicImageProvider<UserView>
2014-10-29 15:01:02 -07:00
{
private readonly IUserManager _userManager;
2015-05-05 08:25:00 -07:00
public DynamicImageProvider(IFileSystem fileSystem, IProviderManager providerManager, IApplicationPaths applicationPaths, IImageProcessor imageProcessor, IUserManager userManager)
2015-04-08 08:45:30 -07:00
: base(fileSystem, providerManager, applicationPaths, imageProcessor)
2014-10-29 15:01:02 -07:00
{
2015-03-06 20:53:31 -07:00
_userManager = userManager;
2015-03-13 21:50:23 -07: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-27 20:29:50 -07:00
ImageType.Primary
2015-03-13 21:50:23 -07:00
};
2014-10-29 15:01:02 -07:00
}
protected override async Task<List<BaseItem>> GetItemsWithImages(IHasImages item)
{
var view = (UserView)item;
2015-03-14 08:38:16 -07:00
if (string.Equals(view.ViewType, CollectionType.LiveTv, StringComparison.OrdinalIgnoreCase))
{
return new List<BaseItem>();
}
2015-04-16 21:53:47 -07:00
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) ||
2014-11-10 21:26:53 -07:00
string.Equals(view.ViewType, SpecialFolder.TvGenre, StringComparison.OrdinalIgnoreCase))
{
2015-04-16 21:53:47 -07:00
var userItemsResult = await view.GetItems(new InternalItemsQuery
2014-11-10 21:26:53 -07:00
{
2015-07-26 22:03:34 -07:00
User = view.UserId.HasValue ? _userManager.GetUserById(view.UserId.Value) : null,
2015-04-16 21:53:47 -07:00
CollapseBoxSetItems = false
});
return userItemsResult.Items.ToList();
2014-11-10 21:26:53 -07:00
}
2015-03-13 21:50:23 -07:00
var isUsingCollectionStrip = IsUsingCollectionStrip(view);
2015-03-14 08:38:16 -07:00
var recursive = isUsingCollectionStrip && !new[] { CollectionType.Playlists, CollectionType.Channels }.Contains(view.ViewType ?? string.Empty, StringComparer.OrdinalIgnoreCase);
2015-03-13 21:50:23 -07:00
2014-10-29 15:01:02 -07:00
var result = await view.GetItems(new InternalItemsQuery
{
2015-07-26 22:03:34 -07:00
User = view.UserId.HasValue ? _userManager.GetUserById(view.UserId.Value) : null,
2015-03-13 21:50:23 -07:00
CollapseBoxSetItems = false,
Recursive = recursive,
2015-04-03 09:58:12 -07:00
ExcludeItemTypes = new[] { "UserView", "CollectionFolder", "Playlist" }
2014-10-29 15:01:02 -07:00
}).ConfigureAwait(false);
2014-11-10 20:41:55 -07:00
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;
}
2015-03-13 12:52:49 -07:00
var audio = i as Audio;
if (audio != null)
{
2015-05-04 07:35:38 -07:00
var album = audio.AlbumEntity;
2015-03-13 21:50:23 -07:00
if (album != null && album.HasImage(ImageType.Primary))
2015-03-13 12:52:49 -07:00
{
2015-03-13 21:50:23 -07:00
return album;
2015-03-13 12:52:49 -07:00
}
}
2014-11-10 20:41:55 -07:00
return i;
}).DistinctBy(i => i.Id);
2015-03-13 21:50:23 -07:00
if (isUsingCollectionStrip)
2015-03-13 12:52:49 -07:00
{
return GetFinalItems(items.Where(i => i.HasImage(ImageType.Primary) || i.HasImage(ImageType.Thumb)).ToList(), 8);
}
2014-11-10 20:41:55 -07:00
return GetFinalItems(items.Where(i => i.HasImage(ImageType.Primary)).ToList());
2014-10-29 15:01:02 -07:00
}
2015-03-13 12:37:19 -07:00
public override bool Supports(IHasImages item)
2014-10-29 15:01:02 -07:00
{
var view = item as UserView;
2015-07-26 22:03:34 -07:00
if (view != null)
2014-10-29 15:01:02 -07:00
{
var supported = new[]
{
SpecialFolder.TvFavoriteEpisodes,
SpecialFolder.TvFavoriteSeries,
SpecialFolder.TvGenres,
SpecialFolder.TvLatest,
SpecialFolder.TvNextUp,
SpecialFolder.TvResume,
SpecialFolder.TvShowSeries,
SpecialFolder.MovieCollections,
SpecialFolder.MovieFavorites,
SpecialFolder.MovieGenres,
SpecialFolder.MovieLatest,
SpecialFolder.MovieMovies,
SpecialFolder.MovieResume,
SpecialFolder.MusicArtists,
SpecialFolder.MusicAlbumArtists,
SpecialFolder.MusicAlbums,
SpecialFolder.MusicGenres,
SpecialFolder.MusicLatest,
2015-04-12 22:12:02 -07:00
SpecialFolder.MusicPlaylists,
2014-10-29 15:01:02 -07:00
SpecialFolder.MusicSongs,
SpecialFolder.MusicFavorites,
SpecialFolder.MusicFavoriteArtists,
SpecialFolder.MusicFavoriteAlbums,
SpecialFolder.MusicFavoriteSongs
2015-07-26 22:03:34 -07:00
2014-10-29 15:01:02 -07:00
};
2015-07-26 22:03:34 -07:00
return (IsUsingCollectionStrip(view) || supported.Contains(view.ViewType, StringComparer.OrdinalIgnoreCase));
2014-10-29 15:01:02 -07:00
}
return false;
}
2015-03-13 12:52:49 -07:00
private bool IsUsingCollectionStrip(UserView view)
{
2015-03-13 21:50:23 -07:00
string[] collectionStripViewTypes =
{
CollectionType.Movies,
CollectionType.TvShows,
CollectionType.Games,
CollectionType.Music,
CollectionType.BoxSets,
CollectionType.Playlists,
2015-03-14 08:38:16 -07:00
CollectionType.Channels,
2015-03-15 18:48:25 -07:00
CollectionType.LiveTv,
CollectionType.Books,
2015-03-17 19:39:55 -07:00
CollectionType.Photos,
CollectionType.HomeVideos,
2015-03-18 09:40:16 -07:00
CollectionType.MusicVideos,
2015-03-17 19:39:55 -07:00
string.Empty
2015-03-13 21:50:23 -07:00
};
return collectionStripViewTypes.Contains(view.ViewType ?? string.Empty);
2015-03-13 12:52:49 -07:00
}
2015-03-13 13:09:07 -07:00
protected override async Task<bool> CreateImage(IHasImages item, List<BaseItem> itemsWithImages, string outputPath, ImageType imageType, int imageIndex)
2015-03-13 13:09:07 -07:00
{
var view = (UserView)item;
if (imageType == ImageType.Primary && IsUsingCollectionStrip(view))
{
2015-03-15 20:41:12 -07:00
if (itemsWithImages.Count == 0 && !string.Equals(view.ViewType, CollectionType.LiveTv, StringComparison.OrdinalIgnoreCase))
{
2015-04-08 08:45:30 -07:00
return false;
2015-03-15 20:41:12 -07:00
}
2015-05-11 09:32:15 -07:00
return await CreateThumbCollage(item, itemsWithImages, outputPath, 960, 540, false, item.Name).ConfigureAwait(false);
2015-03-13 13:09:07 -07:00
}
return await base.CreateImage(item, itemsWithImages, outputPath, imageType, imageIndex).ConfigureAwait(false);
2015-03-13 13:09:07 -07:00
}
2015-04-08 07:38:02 -07:00
protected override IEnumerable<String> GetStripCollageImagePaths(IHasImages primaryItem, IEnumerable<BaseItem> items)
2015-03-13 13:09:07 -07:00
{
2015-04-08 07:38:02 -07:00
var userView = primaryItem as UserView;
if (userView != null && string.Equals(userView.ViewType, CollectionType.LiveTv, StringComparison.OrdinalIgnoreCase))
2015-03-14 08:38:16 -07:00
{
var list = new List<string>();
for (int i = 1; i <= 8; i++)
{
list.Add(ExtractLiveTvResource(i.ToString(CultureInfo.InvariantCulture), ApplicationPaths));
}
return list;
}
2015-04-08 07:38:02 -07:00
return base.GetStripCollageImagePaths(primaryItem, items);
2015-03-13 13:09:07 -07:00
}
2015-03-14 08:38:16 -07:00
private string ExtractLiveTvResource(string name, IApplicationPaths paths)
{
var namespacePath = GetType().Namespace + ".livetv." + name + ".jpg";
var tempPath = Path.Combine(paths.TempDirectory, Guid.NewGuid().ToString("N") + ".jpg");
Directory.CreateDirectory(Path.GetDirectoryName(tempPath));
using (var stream = GetType().Assembly.GetManifestResourceStream(namespacePath))
{
using (var fileStream = new FileStream(tempPath, FileMode.Create, FileAccess.Write, FileShare.Read))
{
stream.CopyTo(fileStream);
}
}
return tempPath;
}
2014-10-29 15:01:02 -07:00
}
}