2014-03-12 20:23:16 +00:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Text;
|
|
|
|
|
using System.Text.RegularExpressions;
|
|
|
|
|
using System.Xml;
|
|
|
|
|
using System.Xml.Linq;
|
|
|
|
|
using NLog;
|
|
|
|
|
using NzbDrone.Common.Disk;
|
2014-12-02 06:26:25 +00:00
|
|
|
|
using NzbDrone.Common.Extensions;
|
2014-03-12 20:23:16 +00:00
|
|
|
|
using NzbDrone.Core.MediaCover;
|
|
|
|
|
using NzbDrone.Core.MediaFiles;
|
|
|
|
|
using NzbDrone.Core.Metadata.Files;
|
|
|
|
|
using NzbDrone.Core.Tv;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Metadata.Consumers.Wdtv
|
|
|
|
|
{
|
|
|
|
|
public class WdtvMetadata : MetadataBase<WdtvMetadataSettings>
|
|
|
|
|
{
|
|
|
|
|
private readonly IMapCoversToLocal _mediaCoverService;
|
|
|
|
|
private readonly IDiskProvider _diskProvider;
|
|
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
2014-04-30 23:39:54 +00:00
|
|
|
|
public WdtvMetadata(IMapCoversToLocal mediaCoverService,
|
2014-03-12 20:23:16 +00:00
|
|
|
|
IDiskProvider diskProvider,
|
|
|
|
|
Logger logger)
|
|
|
|
|
{
|
|
|
|
|
_mediaCoverService = mediaCoverService;
|
|
|
|
|
_diskProvider = diskProvider;
|
|
|
|
|
_logger = logger;
|
|
|
|
|
}
|
|
|
|
|
|
2014-06-19 14:18:50 +00:00
|
|
|
|
private static readonly Regex SeasonImagesRegex = new Regex(@"^(season (?<season>\d+))|(?<specials>specials)", RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
2014-03-12 20:23:16 +00:00
|
|
|
|
|
2015-04-25 16:10:43 +00:00
|
|
|
|
public override string Name
|
|
|
|
|
{
|
|
|
|
|
get
|
|
|
|
|
{
|
|
|
|
|
return "WDTV";
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2014-04-30 23:39:54 +00:00
|
|
|
|
public override List<MetadataFile> AfterRename(Series series, List<MetadataFile> existingMetadataFiles, List<EpisodeFile> episodeFiles)
|
2014-03-12 20:23:16 +00:00
|
|
|
|
{
|
|
|
|
|
var episodeFilesMetadata = existingMetadataFiles.Where(c => c.EpisodeFileId > 0).ToList();
|
|
|
|
|
var updatedMetadataFiles = new List<MetadataFile>();
|
|
|
|
|
|
|
|
|
|
foreach (var episodeFile in episodeFiles)
|
|
|
|
|
{
|
|
|
|
|
var metadataFiles = episodeFilesMetadata.Where(m => m.EpisodeFileId == episodeFile.Id).ToList();
|
|
|
|
|
|
|
|
|
|
foreach (var metadataFile in metadataFiles)
|
|
|
|
|
{
|
|
|
|
|
string newFilename;
|
|
|
|
|
|
|
|
|
|
if (metadataFile.Type == MetadataType.EpisodeImage)
|
|
|
|
|
{
|
2014-07-23 23:43:54 +00:00
|
|
|
|
newFilename = GetEpisodeImageFilename(episodeFile.RelativePath);
|
2014-03-12 20:23:16 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
else if (metadataFile.Type == MetadataType.EpisodeMetadata)
|
|
|
|
|
{
|
2014-07-23 23:43:54 +00:00
|
|
|
|
newFilename = GetEpisodeMetadataFilename(episodeFile.RelativePath);
|
2014-03-12 20:23:16 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
_logger.Trace("Unknown episode file metadata: {0}", metadataFile.RelativePath);
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var existingFilename = Path.Combine(series.Path, metadataFile.RelativePath);
|
2014-07-23 23:43:54 +00:00
|
|
|
|
newFilename = Path.Combine(series.Path, newFilename);
|
2014-03-12 20:23:16 +00:00
|
|
|
|
|
|
|
|
|
if (!newFilename.PathEquals(existingFilename))
|
|
|
|
|
{
|
|
|
|
|
_diskProvider.MoveFile(existingFilename, newFilename);
|
2014-04-19 15:09:22 +00:00
|
|
|
|
metadataFile.RelativePath = series.Path.GetRelativePath(newFilename);
|
2014-03-12 20:23:16 +00:00
|
|
|
|
|
|
|
|
|
updatedMetadataFiles.Add(metadataFile);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
2014-04-30 23:39:54 +00:00
|
|
|
|
return updatedMetadataFiles;
|
2014-03-12 20:23:16 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public override MetadataFile FindMetadataFile(Series series, string path)
|
|
|
|
|
{
|
|
|
|
|
var filename = Path.GetFileName(path);
|
|
|
|
|
|
|
|
|
|
if (filename == null) return null;
|
|
|
|
|
|
|
|
|
|
var metadata = new MetadataFile
|
|
|
|
|
{
|
|
|
|
|
SeriesId = series.Id,
|
|
|
|
|
Consumer = GetType().Name,
|
2014-04-19 15:09:22 +00:00
|
|
|
|
RelativePath = series.Path.GetRelativePath(path)
|
2014-03-12 20:23:16 +00:00
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
//Series and season images are both named folder.jpg, only season ones sit in season folders
|
2014-06-19 14:18:50 +00:00
|
|
|
|
if (Path.GetFileName(filename).Equals("folder.jpg", StringComparison.InvariantCultureIgnoreCase))
|
2014-03-12 20:23:16 +00:00
|
|
|
|
{
|
|
|
|
|
var parentdir = Directory.GetParent(path);
|
|
|
|
|
var seasonMatch = SeasonImagesRegex.Match(parentdir.Name);
|
|
|
|
|
if (seasonMatch.Success)
|
|
|
|
|
{
|
|
|
|
|
metadata.Type = MetadataType.SeasonImage;
|
|
|
|
|
|
2014-06-19 14:18:50 +00:00
|
|
|
|
if (seasonMatch.Groups["specials"].Success)
|
2014-03-12 20:23:16 +00:00
|
|
|
|
{
|
|
|
|
|
metadata.SeasonNumber = 0;
|
|
|
|
|
}
|
2014-06-19 14:18:50 +00:00
|
|
|
|
|
2014-03-12 20:23:16 +00:00
|
|
|
|
else
|
|
|
|
|
{
|
2014-06-19 14:18:50 +00:00
|
|
|
|
metadata.SeasonNumber = Convert.ToInt32(seasonMatch.Groups["season"].Value);
|
2014-03-12 20:23:16 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return metadata;
|
|
|
|
|
}
|
2014-06-19 14:18:50 +00:00
|
|
|
|
|
2014-12-07 07:23:11 +00:00
|
|
|
|
metadata.Type = MetadataType.SeriesImage;
|
|
|
|
|
return metadata;
|
2014-03-12 20:23:16 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var parseResult = Parser.Parser.ParseTitle(filename);
|
|
|
|
|
|
|
|
|
|
if (parseResult != null &&
|
|
|
|
|
!parseResult.FullSeason)
|
|
|
|
|
{
|
|
|
|
|
switch (Path.GetExtension(filename).ToLowerInvariant())
|
|
|
|
|
{
|
|
|
|
|
case ".xml":
|
|
|
|
|
metadata.Type = MetadataType.EpisodeMetadata;
|
|
|
|
|
return metadata;
|
|
|
|
|
case ".metathumb":
|
|
|
|
|
metadata.Type = MetadataType.EpisodeImage;
|
|
|
|
|
return metadata;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
|
2014-04-30 23:39:54 +00:00
|
|
|
|
public override MetadataFileResult SeriesMetadata(Series series)
|
2014-03-12 20:23:16 +00:00
|
|
|
|
{
|
2014-04-30 23:39:54 +00:00
|
|
|
|
//Series metadata is not supported
|
|
|
|
|
return null;
|
2014-03-12 20:23:16 +00:00
|
|
|
|
}
|
|
|
|
|
|
2014-04-30 23:39:54 +00:00
|
|
|
|
public override MetadataFileResult EpisodeMetadata(Series series, EpisodeFile episodeFile)
|
2014-03-12 20:23:16 +00:00
|
|
|
|
{
|
2014-04-30 23:39:54 +00:00
|
|
|
|
if (!Settings.EpisodeMetadata)
|
2014-03-12 20:23:16 +00:00
|
|
|
|
{
|
2014-04-30 23:39:54 +00:00
|
|
|
|
return null;
|
2014-03-12 20:23:16 +00:00
|
|
|
|
}
|
|
|
|
|
|
2014-07-23 23:43:54 +00:00
|
|
|
|
_logger.Debug("Generating Episode Metadata for: {0}", Path.Combine(series.Path, episodeFile.RelativePath));
|
2014-03-12 20:23:16 +00:00
|
|
|
|
|
2015-10-03 17:45:26 +00:00
|
|
|
|
var xmlResult = string.Empty;
|
2014-03-12 20:23:16 +00:00
|
|
|
|
foreach (var episode in episodeFile.Episodes.Value)
|
|
|
|
|
{
|
|
|
|
|
var sb = new StringBuilder();
|
|
|
|
|
var xws = new XmlWriterSettings();
|
|
|
|
|
xws.OmitXmlDeclaration = true;
|
|
|
|
|
xws.Indent = false;
|
|
|
|
|
|
|
|
|
|
using (var xw = XmlWriter.Create(sb, xws))
|
|
|
|
|
{
|
|
|
|
|
var doc = new XDocument();
|
|
|
|
|
|
|
|
|
|
var details = new XElement("details");
|
|
|
|
|
details.Add(new XElement("id", series.Id));
|
2015-10-03 17:45:26 +00:00
|
|
|
|
details.Add(new XElement("title", string.Format("{0} - {1}x{2:00} - {3}", series.Title, episode.SeasonNumber, episode.EpisodeNumber, episode.Title)));
|
2014-03-12 20:23:16 +00:00
|
|
|
|
details.Add(new XElement("series_name", series.Title));
|
|
|
|
|
details.Add(new XElement("episode_name", episode.Title));
|
2014-12-15 14:34:00 +00:00
|
|
|
|
details.Add(new XElement("season_number", episode.SeasonNumber.ToString("00")));
|
|
|
|
|
details.Add(new XElement("episode_number", episode.EpisodeNumber.ToString("00")));
|
2014-03-12 20:23:16 +00:00
|
|
|
|
details.Add(new XElement("firstaired", episode.AirDate));
|
2015-10-03 17:45:26 +00:00
|
|
|
|
details.Add(new XElement("genre", string.Join(" / ", series.Genres)));
|
|
|
|
|
details.Add(new XElement("actor", string.Join(" / ", series.Actors.ConvertAll(c => c.Name + " - " + c.Character))));
|
2014-03-12 20:23:16 +00:00
|
|
|
|
details.Add(new XElement("overview", episode.Overview));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
//Todo: get guest stars, writer and director
|
|
|
|
|
//details.Add(new XElement("credits", tvdbEpisode.Writer.FirstOrDefault()));
|
|
|
|
|
//details.Add(new XElement("director", tvdbEpisode.Directors.FirstOrDefault()));
|
|
|
|
|
|
|
|
|
|
doc.Add(details);
|
|
|
|
|
doc.Save(xw);
|
|
|
|
|
|
|
|
|
|
xmlResult += doc.ToString();
|
|
|
|
|
xmlResult += Environment.NewLine;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2014-07-23 23:43:54 +00:00
|
|
|
|
var filename = GetEpisodeMetadataFilename(episodeFile.RelativePath);
|
2014-03-12 20:23:16 +00:00
|
|
|
|
|
2014-04-30 23:39:54 +00:00
|
|
|
|
return new MetadataFileResult(filename, xmlResult.Trim(Environment.NewLine.ToCharArray()));
|
2014-03-12 20:23:16 +00:00
|
|
|
|
}
|
|
|
|
|
|
2014-04-30 23:39:54 +00:00
|
|
|
|
public override List<ImageFileResult> SeriesImages(Series series)
|
2014-03-12 20:23:16 +00:00
|
|
|
|
{
|
2014-04-30 23:39:54 +00:00
|
|
|
|
if (!Settings.SeriesImages)
|
|
|
|
|
{
|
|
|
|
|
return new List<ImageFileResult>();
|
|
|
|
|
}
|
2014-03-12 20:23:16 +00:00
|
|
|
|
|
2014-04-30 23:39:54 +00:00
|
|
|
|
//Because we only support one image, attempt to get the Poster type, then if that fails grab the first
|
|
|
|
|
var image = series.Images.SingleOrDefault(c => c.CoverType == MediaCoverTypes.Poster) ?? series.Images.FirstOrDefault();
|
|
|
|
|
if (image == null)
|
2014-03-12 20:23:16 +00:00
|
|
|
|
{
|
2014-04-30 23:39:54 +00:00
|
|
|
|
_logger.Trace("Failed to find suitable Series image for series {0}.", series.Title);
|
|
|
|
|
return new List<ImageFileResult>();
|
2014-03-12 20:23:16 +00:00
|
|
|
|
}
|
|
|
|
|
|
2014-04-30 23:39:54 +00:00
|
|
|
|
var source = _mediaCoverService.GetCoverPath(series.Id, image.CoverType);
|
|
|
|
|
var destination = Path.Combine(series.Path, "folder" + Path.GetExtension(source));
|
2014-03-12 20:23:16 +00:00
|
|
|
|
|
2014-04-30 23:39:54 +00:00
|
|
|
|
return new List<ImageFileResult>
|
|
|
|
|
{
|
|
|
|
|
new ImageFileResult(destination, source)
|
|
|
|
|
};
|
|
|
|
|
}
|
2014-03-12 20:23:16 +00:00
|
|
|
|
|
2014-04-30 23:39:54 +00:00
|
|
|
|
public override List<ImageFileResult> SeasonImages(Series series, Season season)
|
|
|
|
|
{
|
|
|
|
|
if (!Settings.SeasonImages)
|
2014-03-12 20:23:16 +00:00
|
|
|
|
{
|
2014-04-30 23:39:54 +00:00
|
|
|
|
return new List<ImageFileResult>();
|
2014-03-12 20:23:16 +00:00
|
|
|
|
}
|
2014-04-30 23:39:54 +00:00
|
|
|
|
|
|
|
|
|
var seasonFolders = GetSeasonFolders(series);
|
2014-03-12 20:23:16 +00:00
|
|
|
|
|
2014-04-30 23:39:54 +00:00
|
|
|
|
//Work out the path to this season - if we don't have a matching path then skip this season.
|
|
|
|
|
string seasonFolder;
|
|
|
|
|
if (!seasonFolders.TryGetValue(season.SeasonNumber, out seasonFolder))
|
|
|
|
|
{
|
|
|
|
|
_logger.Trace("Failed to find season folder for series {0}, season {1}.", series.Title, season.SeasonNumber);
|
|
|
|
|
return new List<ImageFileResult>();
|
|
|
|
|
}
|
2014-03-12 20:23:16 +00:00
|
|
|
|
|
2014-04-30 23:39:54 +00:00
|
|
|
|
//WDTV only supports one season image, so first of all try for poster otherwise just use whatever is first in the collection
|
|
|
|
|
var image = season.Images.SingleOrDefault(c => c.CoverType == MediaCoverTypes.Poster) ?? season.Images.FirstOrDefault();
|
|
|
|
|
if (image == null)
|
|
|
|
|
{
|
|
|
|
|
_logger.Trace("Failed to find suitable season image for series {0}, season {1}.", series.Title, season.SeasonNumber);
|
|
|
|
|
return new List<ImageFileResult>();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var path = Path.Combine(series.Path, seasonFolder, "folder.jpg");
|
2014-03-12 20:23:16 +00:00
|
|
|
|
|
2014-04-30 23:39:54 +00:00
|
|
|
|
return new List<ImageFileResult>{ new ImageFileResult(path, image.Url) };
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public override List<ImageFileResult> EpisodeImages(Series series, EpisodeFile episodeFile)
|
|
|
|
|
{
|
|
|
|
|
if (!Settings.EpisodeImages)
|
|
|
|
|
{
|
|
|
|
|
return new List<ImageFileResult>();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var screenshot = episodeFile.Episodes.Value.First().Images.SingleOrDefault(i => i.CoverType == MediaCoverTypes.Screenshot);
|
|
|
|
|
|
|
|
|
|
if (screenshot == null)
|
|
|
|
|
{
|
|
|
|
|
_logger.Trace("Episode screenshot not available");
|
|
|
|
|
return new List<ImageFileResult>();
|
|
|
|
|
}
|
|
|
|
|
|
2014-07-23 23:43:54 +00:00
|
|
|
|
return new List<ImageFileResult>{ new ImageFileResult(GetEpisodeImageFilename(episodeFile.RelativePath), screenshot.Url) };
|
2014-03-12 20:23:16 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private string GetEpisodeMetadataFilename(string episodeFilePath)
|
|
|
|
|
{
|
|
|
|
|
return Path.ChangeExtension(episodeFilePath, "xml");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private string GetEpisodeImageFilename(string episodeFilePath)
|
|
|
|
|
{
|
|
|
|
|
return Path.ChangeExtension(episodeFilePath, "metathumb");
|
|
|
|
|
}
|
2014-04-30 23:39:54 +00:00
|
|
|
|
|
2015-10-03 17:45:26 +00:00
|
|
|
|
private Dictionary<int, string> GetSeasonFolders(Series series)
|
2014-04-30 23:39:54 +00:00
|
|
|
|
{
|
2015-10-03 17:45:26 +00:00
|
|
|
|
var seasonFolderMap = new Dictionary<int, string>();
|
2014-04-30 23:39:54 +00:00
|
|
|
|
|
|
|
|
|
foreach (var folder in _diskProvider.GetDirectories(series.Path))
|
|
|
|
|
{
|
|
|
|
|
var directoryinfo = new DirectoryInfo(folder);
|
|
|
|
|
var seasonMatch = SeasonImagesRegex.Match(directoryinfo.Name);
|
|
|
|
|
|
|
|
|
|
if (seasonMatch.Success)
|
|
|
|
|
{
|
|
|
|
|
var seasonNumber = seasonMatch.Groups["season"].Value;
|
|
|
|
|
|
|
|
|
|
if (seasonNumber.Contains("specials"))
|
|
|
|
|
{
|
|
|
|
|
seasonFolderMap[0] = folder;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
int matchedSeason;
|
2015-10-03 17:45:26 +00:00
|
|
|
|
if (int.TryParse(seasonNumber, out matchedSeason))
|
2014-04-30 23:39:54 +00:00
|
|
|
|
{
|
|
|
|
|
seasonFolderMap[matchedSeason] = folder;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
_logger.Debug("Failed to parse season number from {0} for series {1}.", folder, series.Title);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
_logger.Debug("Rejecting folder {0} for series {1}.", Path.GetDirectoryName(folder), series.Title);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return seasonFolderMap;
|
|
|
|
|
}
|
2014-03-12 20:23:16 +00:00
|
|
|
|
}
|
|
|
|
|
}
|