123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362 |
- using MediaBrowser.Common.IO;
- using MediaBrowser.Controller.Entities;
- using MediaBrowser.Controller.Entities.Audio;
- using MediaBrowser.Controller.Entities.Movies;
- using MediaBrowser.Controller.Entities.TV;
- using MediaBrowser.Controller.Providers;
- using MediaBrowser.Model.Entities;
- using System;
- using System.Collections.Generic;
- using System.Globalization;
- using System.IO;
- using System.Linq;
- namespace MediaBrowser.LocalMetadata.Images
- {
- public class LocalImageProvider : ILocalImageFileProvider
- {
- private readonly IFileSystem _fileSystem;
- public LocalImageProvider(IFileSystem fileSystem)
- {
- _fileSystem = fileSystem;
- }
- public string Name
- {
- get { return "Local Images"; }
- }
- public int Order
- {
- get { return 0; }
- }
- public bool Supports(IHasImages item)
- {
- if (item.SupportsLocalMetadata)
- {
- // Episode has it's own provider
- if (item.IsOwnedItem || item is Episode || item is Audio || item is Photo)
- {
- return false;
- }
- return true;
- }
- 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;
- }
- private IEnumerable<FileSystemInfo> GetFiles(IHasImages item, bool includeDirectories, IDirectoryService directoryService)
- {
- if (item.LocationType != LocationType.FileSystem)
- {
- return new List<FileSystemInfo>();
- }
- var path = item.ContainingFolderPath;
- if (includeDirectories)
- {
- return directoryService.GetFileSystemEntries(path)
- .Where(i => BaseItem.SupportedImageExtensions.Contains(i.Extension, StringComparer.OrdinalIgnoreCase) ||
- (i.Attributes & FileAttributes.Directory) == FileAttributes.Directory);
- }
- return directoryService.GetFiles(path)
- .Where(i => BaseItem.SupportedImageExtensions.Contains(i.Extension, StringComparer.OrdinalIgnoreCase));
- }
- public List<LocalImageInfo> GetImages(IHasImages item, IDirectoryService directoryService)
- {
- var files = GetFiles(item, true, directoryService).ToList();
- var list = new List<LocalImageInfo>();
- PopulateImages(item, list, files, true, directoryService);
- return list;
- }
- public List<LocalImageInfo> GetImages(IHasImages item, string path, bool checkForCacheKeyFiles, IDirectoryService directoryService)
- {
- return GetImages(item, new[] { path }, checkForCacheKeyFiles, directoryService);
- }
- public List<LocalImageInfo> GetImages(IHasImages item, IEnumerable<string> paths, bool checkForCacheKeyFiles, IDirectoryService directoryService)
- {
- var files = paths.SelectMany(directoryService.GetFiles)
- .Where(i =>
- {
- var ext = i.Extension;
- return !string.IsNullOrEmpty(ext) &&
- BaseItem.SupportedImageExtensions.Contains(ext, StringComparer.OrdinalIgnoreCase);
- })
- .ToList();
- var list = new List<LocalImageInfo>();
- PopulateImages(item, list, files, false, directoryService);
- if (checkForCacheKeyFiles)
- {
- AddCacheKeyImage(files, list, ImageType.Primary);
- AddCacheKeyImage(files, list, ImageType.Thumb);
- }
- return list;
- }
- private void PopulateImages(IHasImages item, List<LocalImageInfo> images, List<FileSystemInfo> files, bool supportParentSeriesFiles, IDirectoryService directoryService)
- {
- var imagePrefix = string.Empty;
- var baseItem = item as BaseItem;
- if (baseItem != null && baseItem.IsInMixedFolder)
- {
- imagePrefix = _fileSystem.GetFileNameWithoutExtension(item.Path) + "-";
- }
- PopulatePrimaryImages(item, images, files, imagePrefix);
- PopulateBackdrops(item, images, files, imagePrefix, directoryService);
- PopulateScreenshots(images, files, imagePrefix);
- AddImage(files, images, imagePrefix + "logo", ImageType.Logo);
- AddImage(files, images, imagePrefix + "clearart", ImageType.Art);
- AddImage(files, images, imagePrefix + "disc", ImageType.Disc);
- AddImage(files, images, imagePrefix + "cdart", ImageType.Disc);
- AddImage(files, images, imagePrefix + "box", ImageType.Box);
- AddImage(files, images, imagePrefix + "back", ImageType.BoxRear);
- AddImage(files, images, imagePrefix + "boxrear", ImageType.BoxRear);
- AddImage(files, images, imagePrefix + "menu", ImageType.Menu);
- // Banner
- AddImage(files, images, imagePrefix + "banner", ImageType.Banner);
- // Thumb
- AddImage(files, images, imagePrefix + "thumb", ImageType.Thumb);
- AddImage(files, images, imagePrefix + "landscape", ImageType.Thumb);
- if (supportParentSeriesFiles)
- {
- var season = item as Season;
- if (season != null)
- {
- PopulateSeasonImagesFromSeriesFolder(season, images, directoryService);
- }
- }
- }
- private void PopulatePrimaryImages(IHasImages item, List<LocalImageInfo> images, List<FileSystemInfo> files, string imagePrefix)
- {
- AddImage(files, images, imagePrefix + "folder", ImageType.Primary);
- AddImage(files, images, imagePrefix + "cover", ImageType.Primary);
- AddImage(files, images, imagePrefix + "poster", ImageType.Primary);
- AddImage(files, images, imagePrefix + "default", ImageType.Primary);
- // Support plex/xbmc convention
- if (item is Series)
- {
- AddImage(files, images, imagePrefix + "show", ImageType.Primary);
- }
- // Support plex/xbmc convention
- if (item is Video && !(item is Episode))
- {
- AddImage(files, images, imagePrefix + "movie", ImageType.Primary);
- }
- if (!string.IsNullOrEmpty(item.Path))
- {
- var name = _fileSystem.GetFileNameWithoutExtension(item.Path);
- if (!string.IsNullOrEmpty(name))
- {
- AddImage(files, images, name, ImageType.Primary);
- AddImage(files, images, name + "-poster", ImageType.Primary);
- }
- }
- }
- private void PopulateBackdrops(IHasImages item, List<LocalImageInfo> images, List<FileSystemInfo> files, string imagePrefix, IDirectoryService directoryService)
- {
- PopulateBackdrops(images, files, imagePrefix, "backdrop", "backdrop", ImageType.Backdrop);
- if (!string.IsNullOrEmpty(item.Path))
- {
- var name = _fileSystem.GetFileNameWithoutExtension(item.Path);
- if (!string.IsNullOrEmpty(name))
- {
- AddImage(files, images, imagePrefix + name + "-fanart", ImageType.Backdrop);
- }
- }
- PopulateBackdrops(images, files, imagePrefix, "fanart", "fanart-", ImageType.Backdrop);
- PopulateBackdrops(images, files, imagePrefix, "background", "background-", ImageType.Backdrop);
- PopulateBackdrops(images, files, imagePrefix, "art", "art-", ImageType.Backdrop);
- var extraFanartFolder = files
- .FirstOrDefault(i => string.Equals(i.Name, "extrafanart", StringComparison.OrdinalIgnoreCase));
- if (extraFanartFolder != null)
- {
- PopulateBackdropsFromExtraFanart(extraFanartFolder.FullName, images, directoryService);
- }
- }
- private void PopulateBackdropsFromExtraFanart(string path, List<LocalImageInfo> images, IDirectoryService directoryService)
- {
- var imageFiles = directoryService.GetFiles(path)
- .Where(i =>
- {
- var extension = i.Extension;
- if (string.IsNullOrEmpty(extension))
- {
- return false;
- }
- return BaseItem.SupportedImageExtensions.Contains(extension, StringComparer.OrdinalIgnoreCase);
- });
- images.AddRange(imageFiles.Select(i => new LocalImageInfo
- {
- FileInfo = i,
- Type = ImageType.Backdrop
- }));
- }
- private void PopulateScreenshots(List<LocalImageInfo> images, List<FileSystemInfo> files, string imagePrefix)
- {
- PopulateBackdrops(images, files, imagePrefix, "screenshot", "screenshot", ImageType.Screenshot);
- }
- private void PopulateBackdrops(List<LocalImageInfo> images, List<FileSystemInfo> files, string imagePrefix, string firstFileName, string subsequentFileNamePrefix, 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;
- }
- }
- }
- }
- private readonly CultureInfo _usCulture = new CultureInfo("en-US");
- 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;
- }
- 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);
- }
- }
- private bool AddImage(IEnumerable<FileSystemInfo> files, List<LocalImageInfo> images, string name, ImageType type)
- {
- var image = GetImage(files, name) as FileInfo;
- if (image != null)
- {
- images.Add(new LocalImageInfo
- {
- FileInfo = image,
- Type = type
- });
- return true;
- }
- return false;
- }
- private void AddCacheKeyImage(IEnumerable<FileSystemInfo> files, List<LocalImageInfo> images, ImageType type)
- {
- var candidates = files
- .Where(i => _fileSystem.GetFileNameWithoutExtension(i).StartsWith(type.ToString() + "_key_", StringComparison.OrdinalIgnoreCase))
- .ToList();
- var image = BaseItem.SupportedImageExtensions
- .Select(i => candidates.FirstOrDefault(c => string.Equals(c.Extension, i, StringComparison.OrdinalIgnoreCase)))
- .FirstOrDefault(i => i != null) as FileInfo;
- if (image != null)
- {
- images.Add(new LocalImageInfo
- {
- FileInfo = image,
- Type = type
- });
- }
- }
- private FileSystemInfo GetImage(IEnumerable<FileSystemInfo> files, string name)
- {
- var candidates = files
- .Where(i => string.Equals(name, _fileSystem.GetFileNameWithoutExtension(i), StringComparison.OrdinalIgnoreCase))
- .ToList();
- return BaseItem.SupportedImageExtensions
- .Select(i => candidates.FirstOrDefault(c => string.Equals(c.Extension, i, StringComparison.OrdinalIgnoreCase)))
- .FirstOrDefault(i => i != null);
- }
- }
- }
|