12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455 |
- using System;
- using System.ComponentModel.Composition;
- using System.IO;
- using System.Linq;
- using System.Threading.Tasks;
- using MediaBrowser.Controller.Events;
- using MediaBrowser.Controller.Providers;
- using MediaBrowser.Model.Entities;
- using MediaBrowser.TV.Entities;
- namespace MediaBrowser.TV.Providers
- {
- [Export(typeof(BaseMetadataProvider))]
- public class EpisodeImageFromMediaLocationProvider : BaseMetadataProvider
- {
- public override bool Supports(BaseEntity item)
- {
- return item is Episode;
- }
- public override Task Fetch(BaseEntity item, ItemResolveEventArgs args)
- {
- return Task.Run(() =>
- {
- Episode episode = item as Episode;
- string metadataFolder = Path.Combine(args.Parent.Path, "metadata");
- string episodeFileName = Path.GetFileName(episode.Path);
- Season season = args.Parent as Season;
- SetPrimaryImagePath(episode, season, metadataFolder, episodeFileName);
- });
- }
- private void SetPrimaryImagePath(Episode item, Season season, string metadataFolder, string episodeFileName)
- {
- string[] imageFiles = new string[] {
- Path.Combine(metadataFolder, Path.ChangeExtension(episodeFileName, ".jpg")),
- Path.Combine(metadataFolder, Path.ChangeExtension(episodeFileName, ".png"))
- };
- if (season == null)
- {
- // Gotta do this the slow way
- item.PrimaryImagePath = imageFiles.FirstOrDefault(f => File.Exists(f));
- }
- else
- {
- item.PrimaryImagePath = imageFiles.FirstOrDefault(f => season.MetadataFiles.Any(s => s.Equals(f, StringComparison.OrdinalIgnoreCase)));
- }
- }
- }
- }
|