jellyfin/MediaBrowser.LocalMetadata/Images/LocalImageProvider.cs

423 lines
15 KiB
C#
Raw Normal View History

2015-10-30 16:45:22 +00:00
using MediaBrowser.Controller.Entities;
2014-01-31 19:55:21 +00:00
using MediaBrowser.Controller.Entities.Audio;
using MediaBrowser.Controller.Entities.TV;
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Entities;
2014-07-26 17:30:15 +00:00
using System;
using System.Collections.Generic;
using System.Globalization;
using System.Linq;
2016-10-25 19:02:04 +00:00
using MediaBrowser.Controller.IO;
using MediaBrowser.Model.IO;
namespace MediaBrowser.LocalMetadata.Images
{
2015-04-11 01:42:37 +00:00
public class LocalImageProvider : ILocalImageFileProvider, IHasOrder
{
2014-07-26 17:30:15 +00:00
private readonly IFileSystem _fileSystem;
public LocalImageProvider(IFileSystem fileSystem)
{
_fileSystem = fileSystem;
}
2014-10-20 03:04:45 +00:00
public string Name
{
get { return "Local Images"; }
}
public int Order
{
get { return 0; }
}
2017-08-07 21:06:13 +00:00
public bool Supports(IHasMetadata item)
{
2014-02-10 20:11:46 +00:00
if (item.SupportsLocalMetadata)
{
// Episode has it's own provider
2014-07-22 01:29:06 +00:00
if (item.IsOwnedItem || item is Episode || item is Audio || item is Photo)
{
return false;
}
return true;
}
2014-01-31 19:55:21 +00:00
2014-02-10 20:11:46 +00:00
if (item.LocationType == LocationType.Virtual)
{
var season = item as Season;
if (season != null)
{
var series = season.Series;
if (series != null && series.LocationType == LocationType.FileSystem)
{
return true;
}
}
}
return false;
}
2017-08-07 21:06:13 +00:00
private IEnumerable<FileSystemMetadata> GetFiles(IHasMetadata item, bool includeDirectories, IDirectoryService directoryService)
{
if (item.LocationType != LocationType.FileSystem)
{
2015-10-04 03:38:46 +00:00
return new List<FileSystemMetadata>();
}
2014-02-06 05:17:00 +00:00
var path = item.ContainingFolderPath;
if (includeDirectories)
{
2014-02-08 22:38:02 +00:00
return directoryService.GetFileSystemEntries(path)
2015-11-12 20:51:39 +00:00
.Where(i => BaseItem.SupportedImageExtensions.Contains(i.Extension, StringComparer.OrdinalIgnoreCase) || i.IsDirectory)
2014-12-19 04:20:07 +00:00
.OrderBy(i => BaseItem.SupportedImageExtensionsList.IndexOf(i.Extension ?? string.Empty));
}
2014-02-08 22:38:02 +00:00
return directoryService.GetFiles(path)
2014-12-19 04:20:07 +00:00
.Where(i => BaseItem.SupportedImageExtensions.Contains(i.Extension, StringComparer.OrdinalIgnoreCase))
.OrderBy(i => BaseItem.SupportedImageExtensionsList.IndexOf(i.Extension ?? string.Empty));
}
2017-08-07 21:06:13 +00:00
public List<LocalImageInfo> GetImages(IHasMetadata item, IDirectoryService directoryService)
{
2014-02-08 22:38:02 +00:00
var files = GetFiles(item, true, directoryService).ToList();
var list = new List<LocalImageInfo>();
2014-02-08 22:38:02 +00:00
PopulateImages(item, list, files, true, directoryService);
2014-02-08 20:02:35 +00:00
return list;
}
2017-08-07 21:06:13 +00:00
public List<LocalImageInfo> GetImages(IHasMetadata item, string path, bool isPathInMediaFolder, IDirectoryService directoryService)
2014-02-08 20:02:35 +00:00
{
2017-03-29 06:26:48 +00:00
return GetImages(item, new[] { path }, isPathInMediaFolder, directoryService);
2014-02-08 20:02:35 +00:00
}
2017-08-07 21:06:13 +00:00
public List<LocalImageInfo> GetImages(IHasMetadata item, IEnumerable<string> paths, bool arePathsInMediaFolders, IDirectoryService directoryService)
2014-02-08 20:02:35 +00:00
{
IEnumerable<FileSystemMetadata> files = paths.SelectMany(i => _fileSystem.GetFiles(i, BaseItem.SupportedImageExtensions, true, false));
2017-03-29 06:26:48 +00:00
files = files
.OrderBy(i => BaseItem.SupportedImageExtensionsList.IndexOf(i.Extension ?? string.Empty));
2014-02-08 20:02:35 +00:00
var list = new List<LocalImageInfo>();
2017-03-29 06:26:48 +00:00
PopulateImages(item, list, files.ToList(), false, directoryService);
return list;
}
2017-08-07 21:06:13 +00:00
private void PopulateImages(IHasMetadata item, List<LocalImageInfo> images, List<FileSystemMetadata> files, bool supportParentSeriesFiles, IDirectoryService directoryService)
{
2016-01-22 02:59:07 +00:00
if (supportParentSeriesFiles)
{
var season = item as Season;
if (season != null)
{
PopulateSeasonImagesFromSeriesFolder(season, images, directoryService);
}
}
2017-03-29 06:26:48 +00:00
2014-12-09 04:57:18 +00:00
var imagePrefix = item.FileNameWithoutExtension + "-";
2017-07-31 05:16:22 +00:00
var isInMixedFolder = item.IsInMixedFolder;
2014-12-09 04:57:18 +00:00
PopulatePrimaryImages(item, images, files, imagePrefix, isInMixedFolder);
2014-12-09 04:57:18 +00:00
AddImage(files, images, "logo", imagePrefix, isInMixedFolder, ImageType.Logo);
AddImage(files, images, "clearart", imagePrefix, isInMixedFolder, ImageType.Art);
2016-01-22 02:29:00 +00:00
// For music albums, prefer cdart before disc
if (item is MusicAlbum)
{
AddImage(files, images, "cdart", imagePrefix, isInMixedFolder, ImageType.Disc);
AddImage(files, images, "disc", imagePrefix, isInMixedFolder, ImageType.Disc);
}
else
{
AddImage(files, images, "disc", imagePrefix, isInMixedFolder, ImageType.Disc);
AddImage(files, images, "cdart", imagePrefix, isInMixedFolder, ImageType.Disc);
}
2014-12-09 04:57:18 +00:00
AddImage(files, images, "box", imagePrefix, isInMixedFolder, ImageType.Box);
AddImage(files, images, "back", imagePrefix, isInMixedFolder, ImageType.BoxRear);
AddImage(files, images, "boxrear", imagePrefix, isInMixedFolder, ImageType.BoxRear);
AddImage(files, images, "menu", imagePrefix, isInMixedFolder, ImageType.Menu);
// Banner
2014-12-09 04:57:18 +00:00
AddImage(files, images, "banner", imagePrefix, isInMixedFolder, ImageType.Banner);
// Thumb
2014-12-09 04:57:18 +00:00
AddImage(files, images, "landscape", imagePrefix, isInMixedFolder, ImageType.Thumb);
2016-01-22 02:59:07 +00:00
AddImage(files, images, "thumb", imagePrefix, isInMixedFolder, ImageType.Thumb);
2014-12-09 04:57:18 +00:00
PopulateBackdrops(item, images, files, imagePrefix, isInMixedFolder, directoryService);
PopulateScreenshots(images, files, imagePrefix, isInMixedFolder);
}
2017-08-19 19:43:35 +00:00
private static readonly string[] CommonImageFileNames = new[]
{
"poster",
"folder",
"cover",
"default"
};
private static readonly string[] MusicImageFileNames = new[]
{
"folder",
"poster",
"cover",
"default"
};
private static readonly string[] PersonImageFileNames = new[]
{
"folder",
"poster"
};
private static readonly string[] SeriesImageFileNames = new[]
{
"poster",
"folder",
"cover",
"default",
"show"
};
private static readonly string[] VideoImageFileNames = new[]
{
"poster",
"folder",
"cover",
"default",
"movie"
};
2017-08-07 21:06:13 +00:00
private void PopulatePrimaryImages(IHasMetadata item, List<LocalImageInfo> images, List<FileSystemMetadata> files, string imagePrefix, bool isInMixedFolder)
{
2017-08-19 19:43:35 +00:00
string[] imageFileNames;
2017-08-19 19:43:35 +00:00
if (item is MusicAlbum || item is MusicArtist || item is PhotoAlbum)
2016-01-22 02:59:07 +00:00
{
// these prefer folder
2017-08-19 19:43:35 +00:00
imageFileNames = MusicImageFileNames;
2016-01-22 02:59:07 +00:00
}
2017-08-19 19:43:35 +00:00
else if (item is Person)
2016-01-22 02:59:07 +00:00
{
2017-08-19 19:43:35 +00:00
// these prefer folder
imageFileNames = PersonImageFileNames;
2016-01-22 02:59:07 +00:00
}
2017-08-19 19:43:35 +00:00
else if (item is Series)
{
2017-08-19 19:43:35 +00:00
imageFileNames = SeriesImageFileNames;
}
2017-08-19 19:43:35 +00:00
else if (item is Video && !(item is Episode))
{
imageFileNames = VideoImageFileNames;
}
else
{
2017-08-19 19:43:35 +00:00
imageFileNames = CommonImageFileNames;
}
2014-12-09 04:57:18 +00:00
var fileNameWithoutExtension = item.FileNameWithoutExtension;
if (!string.IsNullOrEmpty(fileNameWithoutExtension))
{
AddImage(files, images, fileNameWithoutExtension, ImageType.Primary);
}
2017-08-19 19:43:35 +00:00
foreach (var name in imageFileNames)
2015-03-27 04:17:04 +00:00
{
AddImage(files, images, imagePrefix + name, ImageType.Primary);
}
2014-12-09 04:57:18 +00:00
if (!isInMixedFolder)
{
2017-08-19 19:43:35 +00:00
foreach (var name in imageFileNames)
{
AddImage(files, images, name, ImageType.Primary);
}
}
}
2017-08-07 21:06:13 +00:00
private void PopulateBackdrops(IHasMetadata item, List<LocalImageInfo> images, List<FileSystemMetadata> files, string imagePrefix, bool isInMixedFolder, IDirectoryService directoryService)
{
2014-02-27 16:25:04 +00:00
if (!string.IsNullOrEmpty(item.Path))
{
2014-12-09 04:57:18 +00:00
var name = item.FileNameWithoutExtension;
if (!string.IsNullOrEmpty(name))
{
AddImage(files, images, imagePrefix + name + "-fanart", ImageType.Backdrop);
2014-12-09 04:57:18 +00:00
// Support without the prefix if it's in it's own folder
if (!isInMixedFolder)
{
AddImage(files, images, name + "-fanart", ImageType.Backdrop);
}
}
}
2014-12-09 04:57:18 +00:00
PopulateBackdrops(images, files, imagePrefix, "fanart", "fanart-", isInMixedFolder, ImageType.Backdrop);
PopulateBackdrops(images, files, imagePrefix, "background", "background-", isInMixedFolder, ImageType.Backdrop);
PopulateBackdrops(images, files, imagePrefix, "art", "art-", isInMixedFolder, ImageType.Backdrop);
var extraFanartFolder = files
.FirstOrDefault(i => string.Equals(i.Name, "extrafanart", StringComparison.OrdinalIgnoreCase));
if (extraFanartFolder != null)
{
2014-02-08 22:38:02 +00:00
PopulateBackdropsFromExtraFanart(extraFanartFolder.FullName, images, directoryService);
}
2016-01-22 02:59:07 +00:00
PopulateBackdrops(images, files, imagePrefix, "backdrop", "backdrop", isInMixedFolder, ImageType.Backdrop);
}
2014-02-10 18:39:41 +00:00
private void PopulateBackdropsFromExtraFanart(string path, List<LocalImageInfo> images, IDirectoryService directoryService)
{
2017-03-29 06:26:48 +00:00
var imageFiles = _fileSystem.GetFiles(path, BaseItem.SupportedImageExtensions, false, false);
2017-03-29 19:16:18 +00:00
images.AddRange(imageFiles.Where(i => i.Length > 0).Select(i => new LocalImageInfo
{
FileInfo = i,
Type = ImageType.Backdrop
}));
}
2015-10-04 03:38:46 +00:00
private void PopulateScreenshots(List<LocalImageInfo> images, List<FileSystemMetadata> files, string imagePrefix, bool isInMixedFolder)
{
2014-12-09 04:57:18 +00:00
PopulateBackdrops(images, files, imagePrefix, "screenshot", "screenshot", isInMixedFolder, ImageType.Screenshot);
}
2015-10-04 03:38:46 +00:00
private void PopulateBackdrops(List<LocalImageInfo> images, List<FileSystemMetadata> files, string imagePrefix, string firstFileName, string subsequentFileNamePrefix, bool isInMixedFolder, ImageType type)
{
AddImage(files, images, imagePrefix + firstFileName, type);
var unfound = 0;
for (var i = 1; i <= 20; i++)
{
// Screenshot Image
var found = AddImage(files, images, imagePrefix + subsequentFileNamePrefix + i, type);
if (!found)
{
unfound++;
if (unfound >= 3)
{
break;
}
}
}
2014-12-09 04:57:18 +00:00
// Support without the prefix
if (!isInMixedFolder)
{
AddImage(files, images, firstFileName, type);
unfound = 0;
for (var i = 1; i <= 20; i++)
{
// Screenshot Image
var found = AddImage(files, images, subsequentFileNamePrefix + i, type);
if (!found)
{
unfound++;
if (unfound >= 3)
{
break;
}
}
}
}
}
private readonly CultureInfo _usCulture = new CultureInfo("en-US");
2014-07-26 17:30:15 +00:00
2014-02-10 18:39:41 +00:00
private void PopulateSeasonImagesFromSeriesFolder(Season season, List<LocalImageInfo> images, IDirectoryService directoryService)
{
var seasonNumber = season.IndexNumber;
var series = season.Series;
if (!seasonNumber.HasValue || series.LocationType != LocationType.FileSystem)
{
return;
}
2014-02-08 22:38:02 +00:00
var seriesFiles = GetFiles(series, false, directoryService).ToList();
// Try using the season name
var prefix = season.Name.ToLower().Replace(" ", string.Empty);
var filenamePrefixes = new List<string> { prefix };
var seasonMarker = seasonNumber.Value == 0
? "-specials"
: seasonNumber.Value.ToString("00", _usCulture);
// Get this one directly from the file system since we have to go up a level
if (!string.Equals(prefix, seasonMarker, StringComparison.OrdinalIgnoreCase))
{
filenamePrefixes.Add("season" + seasonMarker);
}
foreach (var filename in filenamePrefixes)
{
AddImage(seriesFiles, images, filename + "-poster", ImageType.Primary);
AddImage(seriesFiles, images, filename + "-fanart", ImageType.Backdrop);
AddImage(seriesFiles, images, filename + "-banner", ImageType.Banner);
AddImage(seriesFiles, images, filename + "-landscape", ImageType.Thumb);
}
}
2015-10-04 03:38:46 +00:00
private bool AddImage(List<FileSystemMetadata> files, List<LocalImageInfo> images, string name, string imagePrefix, bool isInMixedFolder, ImageType type)
2014-12-09 04:57:18 +00:00
{
var added = AddImage(files, images, imagePrefix + name, type);
if (!isInMixedFolder)
{
if (AddImage(files, images, name, type))
{
added = true;
}
}
return added;
}
2015-10-04 03:38:46 +00:00
private bool AddImage(IEnumerable<FileSystemMetadata> files, List<LocalImageInfo> images, string name, ImageType type)
{
2015-10-04 03:38:46 +00:00
var image = GetImage(files, name);
if (image != null)
{
images.Add(new LocalImageInfo
{
FileInfo = image,
Type = type
});
return true;
}
return false;
}
2015-10-04 03:38:46 +00:00
private FileSystemMetadata GetImage(IEnumerable<FileSystemMetadata> files, string name)
{
2017-03-29 19:16:18 +00:00
return files.FirstOrDefault(i => !i.IsDirectory && string.Equals(name, _fileSystem.GetFileNameWithoutExtension(i), StringComparison.OrdinalIgnoreCase) && i.Length > 0);
}
}
}