jellyfin/MediaBrowser.XbmcMetadata/Parsers/EpisodeNfoParser.cs

298 lines
10 KiB
C#
Raw Normal View History

using MediaBrowser.Common.Configuration;
using MediaBrowser.Controller.Entities.TV;
2013-06-09 16:47:28 +00:00
using MediaBrowser.Controller.Providers;
2013-02-21 20:26:35 +00:00
using MediaBrowser.Model.Logging;
using System;
2014-02-09 04:52:52 +00:00
using System.Collections.Generic;
using System.Globalization;
2013-08-12 19:18:31 +00:00
using System.Threading;
2013-02-21 01:33:05 +00:00
using System.Xml;
2016-10-26 02:53:47 +00:00
using MediaBrowser.Model.IO;
using MediaBrowser.Model.Xml;
using System.IO;
using System.Text;
2013-02-21 01:33:05 +00:00
namespace MediaBrowser.XbmcMetadata.Parsers
2013-02-21 01:33:05 +00:00
{
public class EpisodeNfoParser : BaseNfoParser<Episode>
2013-02-21 01:33:05 +00:00
{
2015-08-02 17:31:08 +00:00
public void Fetch(MetadataResult<Episode> item,
List<LocalImageInfo> images,
string metadataFile,
CancellationToken cancellationToken)
2013-08-12 19:18:31 +00:00
{
2015-06-29 01:10:45 +00:00
Fetch(item, metadataFile, cancellationToken);
2013-02-21 20:26:35 +00:00
}
private static readonly CultureInfo UsCulture = new CultureInfo("en-US");
protected override void Fetch(MetadataResult<Episode> item, string metadataFile, XmlReaderSettings settings, CancellationToken cancellationToken)
{
using (var fileStream = FileSystem.OpenRead(metadataFile))
{
using (var streamReader = new StreamReader(fileStream, Encoding.UTF8))
{
item.ResetPeople();
var xml = streamReader.ReadToEnd();
var srch = "</episodedetails>";
var index = xml.IndexOf(srch, StringComparison.OrdinalIgnoreCase);
if (index != -1)
{
xml = xml.Substring(0, index + srch.Length);
}
using (var ms = new MemoryStream())
{
var bytes = Encoding.UTF8.GetBytes(xml);
ms.Write(bytes, 0, bytes.Length);
ms.Position = 0;
// These are not going to be valid xml so no sense in causing the provider to fail and spamming the log with exceptions
try
{
// Use XmlReader for best performance
using (var reader = XmlReader.Create(ms, settings))
{
reader.MoveToContent();
reader.Read();
// Loop through each element
while (!reader.EOF && reader.ReadState == ReadState.Interactive)
{
cancellationToken.ThrowIfCancellationRequested();
if (reader.NodeType == XmlNodeType.Element)
{
FetchDataFromXmlNode(reader, item);
}
else
{
reader.Read();
}
}
}
}
catch (XmlException)
{
}
}
}
}
}
2013-02-21 01:33:05 +00:00
/// <summary>
/// Fetches the data from XML node.
/// </summary>
/// <param name="reader">The reader.</param>
2015-06-29 01:10:45 +00:00
/// <param name="itemResult">The item result.</param>
2015-08-02 17:31:08 +00:00
protected override void FetchDataFromXmlNode(XmlReader reader, MetadataResult<Episode> itemResult)
2013-02-21 01:33:05 +00:00
{
2015-06-29 01:10:45 +00:00
var item = itemResult.Item;
2013-02-21 01:33:05 +00:00
switch (reader.Name)
{
case "season":
2013-02-21 01:33:05 +00:00
{
var number = reader.ReadElementContentAsString();
if (!string.IsNullOrWhiteSpace(number))
{
int num;
if (int.TryParse(number, out num))
{
item.ParentIndexNumber = num;
}
}
break;
}
case "episode":
2013-02-21 01:33:05 +00:00
{
var number = reader.ReadElementContentAsString();
if (!string.IsNullOrWhiteSpace(number))
{
int num;
if (int.TryParse(number, out num))
{
item.IndexNumber = num;
}
}
break;
}
case "episodenumberend":
2013-10-25 14:26:22 +00:00
{
var number = reader.ReadElementContentAsString();
if (!string.IsNullOrWhiteSpace(number))
{
int num;
if (int.TryParse(number, out num))
{
item.IndexNumberEnd = num;
}
}
break;
}
case "absolute_number":
{
var val = reader.ReadElementContentAsString();
if (!string.IsNullOrWhiteSpace(val))
{
int rval;
// int.TryParse is local aware, so it can be probamatic, force us culture
if (int.TryParse(val, NumberStyles.Integer, UsCulture, out rval))
{
item.AbsoluteEpisodeNumber = rval;
}
}
break;
}
2013-12-08 18:07:45 +00:00
case "DVD_episodenumber":
{
var number = reader.ReadElementContentAsString();
if (!string.IsNullOrWhiteSpace(number))
{
float num;
2013-12-08 18:07:45 +00:00
if (float.TryParse(number, NumberStyles.Any, UsCulture, out num))
2013-12-08 18:07:45 +00:00
{
item.DvdEpisodeNumber = num;
2013-12-08 18:07:45 +00:00
}
}
break;
}
case "DVD_season":
{
var number = reader.ReadElementContentAsString();
if (!string.IsNullOrWhiteSpace(number))
{
float num;
2013-12-08 18:07:45 +00:00
if (float.TryParse(number, NumberStyles.Any, UsCulture, out num))
2013-12-08 18:07:45 +00:00
{
item.DvdSeasonNumber = Convert.ToInt32(num);
2013-12-08 18:07:45 +00:00
}
}
break;
}
case "airsbefore_episode":
2013-11-15 21:31:33 +00:00
{
var val = reader.ReadElementContentAsString();
2013-11-15 21:31:33 +00:00
if (!string.IsNullOrWhiteSpace(val))
2013-11-15 21:31:33 +00:00
{
int rval;
2013-11-15 21:31:33 +00:00
// int.TryParse is local aware, so it can be probamatic, force us culture
if (int.TryParse(val, NumberStyles.Integer, UsCulture, out rval))
2013-11-15 21:31:33 +00:00
{
item.AirsBeforeEpisodeNumber = rval;
2013-11-15 21:31:33 +00:00
}
}
break;
}
case "airsafter_season":
{
var val = reader.ReadElementContentAsString();
if (!string.IsNullOrWhiteSpace(val))
{
int rval;
// int.TryParse is local aware, so it can be probamatic, force us culture
if (int.TryParse(val, NumberStyles.Integer, UsCulture, out rval))
{
item.AirsAfterSeasonNumber = rval;
}
}
break;
}
case "airsbefore_season":
{
var val = reader.ReadElementContentAsString();
if (!string.IsNullOrWhiteSpace(val))
{
int rval;
// int.TryParse is local aware, so it can be probamatic, force us culture
if (int.TryParse(val, NumberStyles.Integer, UsCulture, out rval))
{
item.AirsBeforeSeasonNumber = rval;
}
}
2013-11-15 21:31:33 +00:00
break;
}
2016-06-24 20:07:35 +00:00
case "displayseason":
{
var val = reader.ReadElementContentAsString();
if (!string.IsNullOrWhiteSpace(val))
{
int rval;
// int.TryParse is local aware, so it can be probamatic, force us culture
if (int.TryParse(val, NumberStyles.Integer, UsCulture, out rval))
{
2016-08-01 18:16:07 +00:00
item.AirsBeforeSeasonNumber = rval;
2016-06-24 20:07:35 +00:00
}
}
break;
}
case "displayepisode":
{
var val = reader.ReadElementContentAsString();
if (!string.IsNullOrWhiteSpace(val))
{
int rval;
// int.TryParse is local aware, so it can be probamatic, force us culture
if (int.TryParse(val, NumberStyles.Integer, UsCulture, out rval))
{
2016-08-01 18:16:07 +00:00
item.AirsBeforeEpisodeNumber = rval;
2016-06-24 20:07:35 +00:00
}
}
break;
}
2013-02-21 01:33:05 +00:00
default:
2015-06-29 01:10:45 +00:00
base.FetchDataFromXmlNode(reader, itemResult);
2013-02-21 01:33:05 +00:00
break;
}
}
2016-10-26 02:53:47 +00:00
public EpisodeNfoParser(ILogger logger, IConfigurationManager config, IProviderManager providerManager, IFileSystem fileSystem, IXmlReaderSettingsFactory xmlReaderSettingsFactory) : base(logger, config, providerManager, fileSystem, xmlReaderSettingsFactory)
{
}
2013-02-21 01:33:05 +00:00
}
}