61 lines
2.0 KiB
C#
61 lines
2.0 KiB
C#
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 MetadataProviderPriority Priority
|
|
{
|
|
get { return MetadataProviderPriority.First; }
|
|
}
|
|
|
|
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)));
|
|
}
|
|
}
|
|
}
|
|
}
|