TvdbSeriesProvider.cs 50 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293
  1. using MediaBrowser.Common.Configuration;
  2. using MediaBrowser.Common.IO;
  3. using MediaBrowser.Common.Net;
  4. using MediaBrowser.Controller.Configuration;
  5. using MediaBrowser.Controller.Entities;
  6. using MediaBrowser.Controller.Entities.TV;
  7. using MediaBrowser.Controller.Library;
  8. using MediaBrowser.Controller.Providers;
  9. using MediaBrowser.Model.Configuration;
  10. using MediaBrowser.Model.Entities;
  11. using MediaBrowser.Model.IO;
  12. using MediaBrowser.Model.Logging;
  13. using MediaBrowser.Model.Net;
  14. using MediaBrowser.Model.Providers;
  15. using System;
  16. using System.Collections.Generic;
  17. using System.Globalization;
  18. using System.IO;
  19. using System.Linq;
  20. using System.Net;
  21. using System.Text;
  22. using System.Threading;
  23. using System.Threading.Tasks;
  24. using System.Xml;
  25. using CommonIO;
  26. namespace MediaBrowser.Providers.TV
  27. {
  28. public class TvdbSeriesProvider : IRemoteMetadataProvider<Series, SeriesInfo>, IItemIdentityProvider<SeriesInfo>, IHasOrder
  29. {
  30. private const string TvdbSeriesOffset = "TvdbSeriesOffset";
  31. private const string TvdbSeriesOffsetFormat = "{0}-{1}";
  32. internal readonly SemaphoreSlim TvDbResourcePool = new SemaphoreSlim(2, 2);
  33. internal static TvdbSeriesProvider Current { get; private set; }
  34. private readonly IZipClient _zipClient;
  35. private readonly IHttpClient _httpClient;
  36. private readonly IFileSystem _fileSystem;
  37. private readonly IServerConfigurationManager _config;
  38. private readonly CultureInfo _usCulture = new CultureInfo("en-US");
  39. private readonly ILogger _logger;
  40. private readonly ISeriesOrderManager _seriesOrder;
  41. private readonly ILibraryManager _libraryManager;
  42. public TvdbSeriesProvider(IZipClient zipClient, IHttpClient httpClient, IFileSystem fileSystem, IServerConfigurationManager config, ILogger logger, ISeriesOrderManager seriesOrder, ILibraryManager libraryManager)
  43. {
  44. _zipClient = zipClient;
  45. _httpClient = httpClient;
  46. _fileSystem = fileSystem;
  47. _config = config;
  48. _logger = logger;
  49. _seriesOrder = seriesOrder;
  50. _libraryManager = libraryManager;
  51. Current = this;
  52. }
  53. private const string SeriesSearchUrl = "http://www.thetvdb.com/api/GetSeries.php?seriesname={0}&language={1}";
  54. private const string SeriesGetZip = "http://www.thetvdb.com/api/{0}/series/{1}/all/{2}.zip";
  55. public async Task<IEnumerable<RemoteSearchResult>> GetSearchResults(SeriesInfo searchInfo, CancellationToken cancellationToken)
  56. {
  57. var seriesId = searchInfo.GetProviderId(MetadataProviders.Tvdb);
  58. if (string.IsNullOrWhiteSpace(seriesId))
  59. {
  60. return await FindSeries(searchInfo.Name, searchInfo.MetadataLanguage, cancellationToken).ConfigureAwait(false);
  61. }
  62. var metadata = await GetMetadata(searchInfo, cancellationToken).ConfigureAwait(false);
  63. var list = new List<RemoteSearchResult>();
  64. if (metadata.HasMetadata)
  65. {
  66. var res = new RemoteSearchResult
  67. {
  68. Name = metadata.Item.Name,
  69. PremiereDate = metadata.Item.PremiereDate,
  70. ProductionYear = metadata.Item.ProductionYear,
  71. ProviderIds = metadata.Item.ProviderIds,
  72. SearchProviderName = Name
  73. };
  74. list.Add(res);
  75. }
  76. return list;
  77. }
  78. public async Task<MetadataResult<Series>> GetMetadata(SeriesInfo itemId, CancellationToken cancellationToken)
  79. {
  80. var result = new MetadataResult<Series>();
  81. var seriesId = itemId.GetProviderId(MetadataProviders.Tvdb);
  82. if (string.IsNullOrWhiteSpace(seriesId))
  83. {
  84. await Identify(itemId).ConfigureAwait(false);
  85. seriesId = itemId.GetProviderId(MetadataProviders.Tvdb);
  86. }
  87. cancellationToken.ThrowIfCancellationRequested();
  88. if (!string.IsNullOrWhiteSpace(seriesId))
  89. {
  90. await EnsureSeriesInfo(seriesId, itemId.MetadataLanguage, cancellationToken).ConfigureAwait(false);
  91. result.Item = new Series();
  92. result.HasMetadata = true;
  93. FetchSeriesData(result, seriesId, cancellationToken);
  94. await FindAnimeSeriesIndex(result.Item, itemId).ConfigureAwait(false);
  95. }
  96. return result;
  97. }
  98. private async Task FindAnimeSeriesIndex(Series series, SeriesInfo info)
  99. {
  100. var index = await _seriesOrder.FindSeriesIndex(SeriesOrderTypes.Anime, series.Name);
  101. if (index == null)
  102. return;
  103. var offset = info.AnimeSeriesIndex - index;
  104. var id = string.Format(TvdbSeriesOffsetFormat, series.GetProviderId(MetadataProviders.Tvdb), offset);
  105. series.SetProviderId(TvdbSeriesOffset, id);
  106. }
  107. internal static int? GetSeriesOffset(Dictionary<string, string> seriesProviderIds)
  108. {
  109. string idString;
  110. if (!seriesProviderIds.TryGetValue(TvdbSeriesOffset, out idString))
  111. return null;
  112. var parts = idString.Split('-');
  113. if (parts.Length < 2)
  114. return null;
  115. int offset;
  116. if (int.TryParse(parts[1], out offset))
  117. return offset;
  118. return null;
  119. }
  120. /// <summary>
  121. /// Fetches the series data.
  122. /// </summary>
  123. /// <param name="result">The result.</param>
  124. /// <param name="seriesId">The series id.</param>
  125. /// <param name="cancellationToken">The cancellation token.</param>
  126. /// <returns>Task{System.Boolean}.</returns>
  127. private void FetchSeriesData(MetadataResult<Series> result, string seriesId, CancellationToken cancellationToken)
  128. {
  129. var series = result.Item;
  130. series.SetProviderId(MetadataProviders.Tvdb, seriesId);
  131. var seriesDataPath = GetSeriesDataPath(_config.ApplicationPaths, seriesId);
  132. var seriesXmlFilename = series.GetPreferredMetadataLanguage().ToLower() + ".xml";
  133. var seriesXmlPath = Path.Combine(seriesDataPath, seriesXmlFilename);
  134. var actorsXmlPath = Path.Combine(seriesDataPath, "actors.xml");
  135. FetchSeriesInfo(series, seriesXmlPath, cancellationToken);
  136. cancellationToken.ThrowIfCancellationRequested();
  137. result.ResetPeople();
  138. FetchActors(result, actorsXmlPath);
  139. }
  140. /// <summary>
  141. /// Downloads the series zip.
  142. /// </summary>
  143. /// <param name="seriesId">The series id.</param>
  144. /// <param name="seriesDataPath">The series data path.</param>
  145. /// <param name="lastTvDbUpdateTime">The last tv database update time.</param>
  146. /// <param name="preferredMetadataLanguage">The preferred metadata language.</param>
  147. /// <param name="cancellationToken">The cancellation token.</param>
  148. /// <returns>Task.</returns>
  149. internal async Task DownloadSeriesZip(string seriesId, string seriesDataPath, long? lastTvDbUpdateTime, string preferredMetadataLanguage, CancellationToken cancellationToken)
  150. {
  151. try
  152. {
  153. await DownloadSeriesZip(seriesId, seriesDataPath, lastTvDbUpdateTime, preferredMetadataLanguage, preferredMetadataLanguage, cancellationToken).ConfigureAwait(false);
  154. return;
  155. }
  156. catch (HttpException ex)
  157. {
  158. if (!ex.StatusCode.HasValue || ex.StatusCode.Value != HttpStatusCode.NotFound)
  159. {
  160. throw;
  161. }
  162. }
  163. if (!string.Equals(preferredMetadataLanguage, "en", StringComparison.OrdinalIgnoreCase))
  164. {
  165. await DownloadSeriesZip(seriesId, seriesDataPath, lastTvDbUpdateTime, "en", preferredMetadataLanguage, cancellationToken).ConfigureAwait(false);
  166. }
  167. }
  168. private async Task DownloadSeriesZip(string seriesId, string seriesDataPath, long? lastTvDbUpdateTime, string preferredMetadataLanguage, string saveAsMetadataLanguage, CancellationToken cancellationToken)
  169. {
  170. var url = string.Format(SeriesGetZip, TVUtils.TvdbApiKey, seriesId, preferredMetadataLanguage);
  171. using (var zipStream = await _httpClient.Get(new HttpRequestOptions
  172. {
  173. Url = url,
  174. ResourcePool = TvDbResourcePool,
  175. CancellationToken = cancellationToken
  176. }).ConfigureAwait(false))
  177. {
  178. // Delete existing files
  179. DeleteXmlFiles(seriesDataPath);
  180. // Copy to memory stream because we need a seekable stream
  181. using (var ms = new MemoryStream())
  182. {
  183. await zipStream.CopyToAsync(ms).ConfigureAwait(false);
  184. ms.Position = 0;
  185. _zipClient.ExtractAllFromZip(ms, seriesDataPath, true);
  186. }
  187. }
  188. // Sanitize all files, except for extracted episode files
  189. foreach (var file in Directory.EnumerateFiles(seriesDataPath, "*.xml", SearchOption.AllDirectories).ToList()
  190. .Where(i => !Path.GetFileName(i).StartsWith("episode-", StringComparison.OrdinalIgnoreCase)))
  191. {
  192. await SanitizeXmlFile(file).ConfigureAwait(false);
  193. }
  194. var downloadLangaugeXmlFile = Path.Combine(seriesDataPath, preferredMetadataLanguage + ".xml");
  195. var saveAsLanguageXmlFile = Path.Combine(seriesDataPath, saveAsMetadataLanguage + ".xml");
  196. if (!string.Equals(downloadLangaugeXmlFile, saveAsLanguageXmlFile, StringComparison.OrdinalIgnoreCase))
  197. {
  198. _fileSystem.CopyFile(downloadLangaugeXmlFile, saveAsLanguageXmlFile, true);
  199. }
  200. await ExtractEpisodes(seriesDataPath, downloadLangaugeXmlFile, lastTvDbUpdateTime).ConfigureAwait(false);
  201. }
  202. public TvdbOptions GetTvDbOptions()
  203. {
  204. return _config.GetConfiguration<TvdbOptions>("tvdb");
  205. }
  206. private readonly Task _cachedTask = Task.FromResult(true);
  207. internal Task EnsureSeriesInfo(string seriesId, string preferredMetadataLanguage, CancellationToken cancellationToken)
  208. {
  209. var seriesDataPath = GetSeriesDataPath(_config.ApplicationPaths, seriesId);
  210. _fileSystem.CreateDirectory(seriesDataPath);
  211. var files = _fileSystem.GetFiles(seriesDataPath)
  212. .ToList();
  213. var seriesXmlFilename = preferredMetadataLanguage + ".xml";
  214. var download = false;
  215. var automaticUpdatesEnabled = GetTvDbOptions().EnableAutomaticUpdates;
  216. const int cacheDays = 2;
  217. var seriesFile = files.FirstOrDefault(i => string.Equals(seriesXmlFilename, i.Name, StringComparison.OrdinalIgnoreCase));
  218. // No need to check age if automatic updates are enabled
  219. if (seriesFile == null || !seriesFile.Exists || (!automaticUpdatesEnabled && (DateTime.UtcNow - _fileSystem.GetLastWriteTimeUtc(seriesFile)).TotalDays > cacheDays))
  220. {
  221. download = true;
  222. }
  223. var actorsXml = files.FirstOrDefault(i => string.Equals("actors.xml", i.Name, StringComparison.OrdinalIgnoreCase));
  224. // No need to check age if automatic updates are enabled
  225. if (actorsXml == null || !actorsXml.Exists || (!automaticUpdatesEnabled && (DateTime.UtcNow - _fileSystem.GetLastWriteTimeUtc(actorsXml)).TotalDays > cacheDays))
  226. {
  227. download = true;
  228. }
  229. var bannersXml = files.FirstOrDefault(i => string.Equals("banners.xml", i.Name, StringComparison.OrdinalIgnoreCase));
  230. // No need to check age if automatic updates are enabled
  231. if (bannersXml == null || !bannersXml.Exists || (!automaticUpdatesEnabled && (DateTime.UtcNow - _fileSystem.GetLastWriteTimeUtc(bannersXml)).TotalDays > cacheDays))
  232. {
  233. download = true;
  234. }
  235. // Only download if not already there
  236. // The post-scan task will take care of updates so we don't need to re-download here
  237. if (download)
  238. {
  239. return DownloadSeriesZip(seriesId, seriesDataPath, null, preferredMetadataLanguage, cancellationToken);
  240. }
  241. return _cachedTask;
  242. }
  243. /// <summary>
  244. /// Finds the series.
  245. /// </summary>
  246. /// <param name="name">The name.</param>
  247. /// <param name="language">The language.</param>
  248. /// <param name="cancellationToken">The cancellation token.</param>
  249. /// <returns>Task{System.String}.</returns>
  250. private async Task<IEnumerable<RemoteSearchResult>> FindSeries(string name, string language, CancellationToken cancellationToken)
  251. {
  252. var results = (await FindSeriesInternal(name, language, cancellationToken).ConfigureAwait(false)).ToList();
  253. if (results.Count == 0)
  254. {
  255. var parsedName = _libraryManager.ParseName(name);
  256. var nameWithoutYear = parsedName.Name;
  257. if (!string.IsNullOrWhiteSpace(nameWithoutYear) && !string.Equals(nameWithoutYear, name, StringComparison.OrdinalIgnoreCase))
  258. {
  259. results = (await FindSeriesInternal(nameWithoutYear, language, cancellationToken).ConfigureAwait(false)).ToList();
  260. }
  261. }
  262. return results;
  263. }
  264. private async Task<IEnumerable<RemoteSearchResult>> FindSeriesInternal(string name, string language, CancellationToken cancellationToken)
  265. {
  266. var url = string.Format(SeriesSearchUrl, WebUtility.UrlEncode(name), language.ToLower());
  267. var doc = new XmlDocument();
  268. using (var results = await _httpClient.Get(new HttpRequestOptions
  269. {
  270. Url = url,
  271. ResourcePool = TvDbResourcePool,
  272. CancellationToken = cancellationToken
  273. }).ConfigureAwait(false))
  274. {
  275. doc.Load(results);
  276. }
  277. var searchResults = new List<RemoteSearchResult>();
  278. if (doc.HasChildNodes)
  279. {
  280. var nodes = doc.SelectNodes("//Series");
  281. var comparableName = GetComparableName(name);
  282. if (nodes != null)
  283. {
  284. foreach (XmlNode node in nodes)
  285. {
  286. var searchResult = new RemoteSearchResult
  287. {
  288. SearchProviderName = Name
  289. };
  290. var titles = new List<string>();
  291. var nameNode = node.SelectSingleNode("./SeriesName");
  292. if (nameNode != null)
  293. {
  294. titles.Add(GetComparableName(nameNode.InnerText));
  295. }
  296. var aliasNode = node.SelectSingleNode("./AliasNames");
  297. if (aliasNode != null)
  298. {
  299. var alias = aliasNode.InnerText.Split('|').Select(GetComparableName);
  300. titles.AddRange(alias);
  301. }
  302. var imdbIdNode = node.SelectSingleNode("./IMDB_ID");
  303. if (imdbIdNode != null)
  304. {
  305. var val = imdbIdNode.InnerText;
  306. if (!string.IsNullOrWhiteSpace(val))
  307. {
  308. searchResult.SetProviderId(MetadataProviders.Imdb, val);
  309. }
  310. }
  311. var bannerNode = node.SelectSingleNode("./banner");
  312. if (bannerNode != null)
  313. {
  314. var val = bannerNode.InnerText;
  315. if (!string.IsNullOrWhiteSpace(val))
  316. {
  317. searchResult.ImageUrl = TVUtils.BannerUrl + val;
  318. }
  319. }
  320. var airDateNode = node.SelectSingleNode("./FirstAired");
  321. if (airDateNode != null)
  322. {
  323. var val = airDateNode.InnerText;
  324. if (!string.IsNullOrWhiteSpace(val))
  325. {
  326. DateTime date;
  327. if (DateTime.TryParse(val, out date))
  328. {
  329. searchResult.ProductionYear = date.Year;
  330. }
  331. }
  332. }
  333. foreach (var title in titles)
  334. {
  335. if (string.Equals(title, comparableName, StringComparison.OrdinalIgnoreCase))
  336. {
  337. var id = node.SelectSingleNode("./seriesid") ??
  338. node.SelectSingleNode("./id");
  339. if (id != null)
  340. {
  341. searchResult.Name = title;
  342. searchResult.SetProviderId(MetadataProviders.Tvdb, id.InnerText);
  343. searchResults.Add(searchResult);
  344. }
  345. break;
  346. }
  347. _logger.Info("TVDb Provider - " + title + " did not match " + comparableName);
  348. }
  349. }
  350. }
  351. }
  352. if (searchResults.Count == 0)
  353. {
  354. _logger.Info("TVDb Provider - Could not find " + name + ". Check name on Thetvdb.org.");
  355. }
  356. return searchResults;
  357. }
  358. /// <summary>
  359. /// The remove
  360. /// </summary>
  361. const string remove = "\"'!`?";
  362. /// <summary>
  363. /// The spacers
  364. /// </summary>
  365. const string spacers = "/,.:;\\(){}[]+-_=–*"; // (there are not actually two - in the they are different char codes)
  366. /// <summary>
  367. /// Gets the name of the comparable.
  368. /// </summary>
  369. /// <param name="name">The name.</param>
  370. /// <returns>System.String.</returns>
  371. internal static string GetComparableName(string name)
  372. {
  373. name = name.ToLower();
  374. name = name.Normalize(NormalizationForm.FormKD);
  375. var sb = new StringBuilder();
  376. foreach (var c in name)
  377. {
  378. if ((int)c >= 0x2B0 && (int)c <= 0x0333)
  379. {
  380. // skip char modifier and diacritics
  381. }
  382. else if (remove.IndexOf(c) > -1)
  383. {
  384. // skip chars we are removing
  385. }
  386. else if (spacers.IndexOf(c) > -1)
  387. {
  388. sb.Append(" ");
  389. }
  390. else if (c == '&')
  391. {
  392. sb.Append(" and ");
  393. }
  394. else
  395. {
  396. sb.Append(c);
  397. }
  398. }
  399. name = sb.ToString();
  400. name = name.Replace(", the", "");
  401. name = name.Replace("the ", " ");
  402. name = name.Replace(" the ", " ");
  403. string prevName;
  404. do
  405. {
  406. prevName = name;
  407. name = name.Replace(" ", " ");
  408. } while (name.Length != prevName.Length);
  409. return name.Trim();
  410. }
  411. private void FetchSeriesInfo(Series item, string seriesXmlPath, CancellationToken cancellationToken)
  412. {
  413. var settings = new XmlReaderSettings
  414. {
  415. CheckCharacters = false,
  416. IgnoreProcessingInstructions = true,
  417. IgnoreComments = true,
  418. ValidationType = ValidationType.None
  419. };
  420. var episiodeAirDates = new List<DateTime>();
  421. using (var streamReader = new StreamReader(seriesXmlPath, Encoding.UTF8))
  422. {
  423. // Use XmlReader for best performance
  424. using (var reader = XmlReader.Create(streamReader, settings))
  425. {
  426. reader.MoveToContent();
  427. // Loop through each element
  428. while (reader.Read())
  429. {
  430. cancellationToken.ThrowIfCancellationRequested();
  431. if (reader.NodeType == XmlNodeType.Element)
  432. {
  433. switch (reader.Name)
  434. {
  435. case "Series":
  436. {
  437. using (var subtree = reader.ReadSubtree())
  438. {
  439. FetchDataFromSeriesNode(item, subtree, cancellationToken);
  440. }
  441. break;
  442. }
  443. case "Episode":
  444. {
  445. using (var subtree = reader.ReadSubtree())
  446. {
  447. var date = GetFirstAiredDateFromEpisodeNode(subtree, cancellationToken);
  448. if (date.HasValue)
  449. {
  450. episiodeAirDates.Add(date.Value);
  451. }
  452. }
  453. break;
  454. }
  455. default:
  456. reader.Skip();
  457. break;
  458. }
  459. }
  460. }
  461. }
  462. }
  463. if (item.Status.HasValue && item.Status.Value == SeriesStatus.Ended && episiodeAirDates.Count > 0)
  464. {
  465. item.EndDate = episiodeAirDates.Max();
  466. }
  467. }
  468. private DateTime? GetFirstAiredDateFromEpisodeNode(XmlReader reader, CancellationToken cancellationToken)
  469. {
  470. DateTime? airDate = null;
  471. int? seasonNumber = null;
  472. reader.MoveToContent();
  473. // Loop through each element
  474. while (reader.Read())
  475. {
  476. cancellationToken.ThrowIfCancellationRequested();
  477. if (reader.NodeType == XmlNodeType.Element)
  478. {
  479. switch (reader.Name)
  480. {
  481. case "FirstAired":
  482. {
  483. var val = reader.ReadElementContentAsString();
  484. if (!string.IsNullOrWhiteSpace(val))
  485. {
  486. DateTime date;
  487. if (DateTime.TryParse(val, out date))
  488. {
  489. airDate = date.ToUniversalTime();
  490. }
  491. }
  492. break;
  493. }
  494. case "SeasonNumber":
  495. {
  496. var val = reader.ReadElementContentAsString();
  497. if (!string.IsNullOrWhiteSpace(val))
  498. {
  499. int rval;
  500. // int.TryParse is local aware, so it can be probamatic, force us culture
  501. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
  502. {
  503. seasonNumber = rval;
  504. }
  505. }
  506. break;
  507. }
  508. default:
  509. reader.Skip();
  510. break;
  511. }
  512. }
  513. }
  514. if (seasonNumber.HasValue && seasonNumber.Value != 0)
  515. {
  516. return airDate;
  517. }
  518. return null;
  519. }
  520. /// <summary>
  521. /// Fetches the actors.
  522. /// </summary>
  523. /// <param name="result">The result.</param>
  524. /// <param name="actorsXmlPath">The actors XML path.</param>
  525. private void FetchActors(MetadataResult<Series> result, string actorsXmlPath)
  526. {
  527. var settings = new XmlReaderSettings
  528. {
  529. CheckCharacters = false,
  530. IgnoreProcessingInstructions = true,
  531. IgnoreComments = true,
  532. ValidationType = ValidationType.None
  533. };
  534. using (var streamReader = new StreamReader(actorsXmlPath, Encoding.UTF8))
  535. {
  536. // Use XmlReader for best performance
  537. using (var reader = XmlReader.Create(streamReader, settings))
  538. {
  539. reader.MoveToContent();
  540. // Loop through each element
  541. while (reader.Read())
  542. {
  543. if (reader.NodeType == XmlNodeType.Element)
  544. {
  545. switch (reader.Name)
  546. {
  547. case "Actor":
  548. {
  549. using (var subtree = reader.ReadSubtree())
  550. {
  551. FetchDataFromActorNode(result, subtree);
  552. }
  553. break;
  554. }
  555. default:
  556. reader.Skip();
  557. break;
  558. }
  559. }
  560. }
  561. }
  562. }
  563. }
  564. /// <summary>
  565. /// Fetches the data from actor node.
  566. /// </summary>
  567. /// <param name="result">The result.</param>
  568. /// <param name="reader">The reader.</param>
  569. private void FetchDataFromActorNode(MetadataResult<Series> result, XmlReader reader)
  570. {
  571. reader.MoveToContent();
  572. var personInfo = new PersonInfo();
  573. while (reader.Read())
  574. {
  575. if (reader.NodeType == XmlNodeType.Element)
  576. {
  577. switch (reader.Name)
  578. {
  579. case "Name":
  580. {
  581. personInfo.Name = (reader.ReadElementContentAsString() ?? string.Empty).Trim();
  582. break;
  583. }
  584. case "Role":
  585. {
  586. personInfo.Role = (reader.ReadElementContentAsString() ?? string.Empty).Trim();
  587. break;
  588. }
  589. case "SortOrder":
  590. {
  591. var val = reader.ReadElementContentAsString();
  592. if (!string.IsNullOrWhiteSpace(val))
  593. {
  594. int rval;
  595. // int.TryParse is local aware, so it can be probamatic, force us culture
  596. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
  597. {
  598. personInfo.SortOrder = rval;
  599. }
  600. }
  601. break;
  602. }
  603. default:
  604. reader.Skip();
  605. break;
  606. }
  607. }
  608. }
  609. personInfo.Type = PersonType.Actor;
  610. if (!string.IsNullOrWhiteSpace(personInfo.Name))
  611. {
  612. result.AddPerson(personInfo);
  613. }
  614. }
  615. private void FetchDataFromSeriesNode(Series item, XmlReader reader, CancellationToken cancellationToken)
  616. {
  617. reader.MoveToContent();
  618. // Loop through each element
  619. while (reader.Read())
  620. {
  621. cancellationToken.ThrowIfCancellationRequested();
  622. if (reader.NodeType == XmlNodeType.Element)
  623. {
  624. switch (reader.Name)
  625. {
  626. case "SeriesName":
  627. {
  628. item.Name = (reader.ReadElementContentAsString() ?? string.Empty).Trim();
  629. break;
  630. }
  631. case "Overview":
  632. {
  633. item.Overview = (reader.ReadElementContentAsString() ?? string.Empty).Trim();
  634. break;
  635. }
  636. case "Airs_DayOfWeek":
  637. {
  638. var val = reader.ReadElementContentAsString();
  639. if (!string.IsNullOrWhiteSpace(val))
  640. {
  641. item.AirDays = TVUtils.GetAirDays(val);
  642. }
  643. break;
  644. }
  645. case "Airs_Time":
  646. {
  647. var val = reader.ReadElementContentAsString();
  648. if (!string.IsNullOrWhiteSpace(val))
  649. {
  650. item.AirTime = val;
  651. }
  652. break;
  653. }
  654. case "ContentRating":
  655. {
  656. var val = reader.ReadElementContentAsString();
  657. if (!string.IsNullOrWhiteSpace(val))
  658. {
  659. item.OfficialRating = val;
  660. }
  661. break;
  662. }
  663. case "Rating":
  664. {
  665. var val = reader.ReadElementContentAsString();
  666. if (!string.IsNullOrWhiteSpace(val))
  667. {
  668. // Only fill this if it doesn't already have a value, since we get it from imdb which has better data
  669. if (!item.CommunityRating.HasValue || string.IsNullOrWhiteSpace(item.GetProviderId(MetadataProviders.Imdb)))
  670. {
  671. float rval;
  672. // float.TryParse is local aware, so it can be probamatic, force us culture
  673. if (float.TryParse(val, NumberStyles.AllowDecimalPoint, _usCulture, out rval))
  674. {
  675. item.CommunityRating = rval;
  676. }
  677. }
  678. }
  679. break;
  680. }
  681. case "RatingCount":
  682. {
  683. var val = reader.ReadElementContentAsString();
  684. if (!string.IsNullOrWhiteSpace(val))
  685. {
  686. int rval;
  687. // int.TryParse is local aware, so it can be probamatic, force us culture
  688. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
  689. {
  690. item.VoteCount = rval;
  691. }
  692. }
  693. break;
  694. }
  695. case "IMDB_ID":
  696. {
  697. var val = reader.ReadElementContentAsString();
  698. if (!string.IsNullOrWhiteSpace(val))
  699. {
  700. item.SetProviderId(MetadataProviders.Imdb, val);
  701. }
  702. break;
  703. }
  704. case "zap2it_id":
  705. {
  706. var val = reader.ReadElementContentAsString();
  707. if (!string.IsNullOrWhiteSpace(val))
  708. {
  709. item.SetProviderId(MetadataProviders.Zap2It, val);
  710. }
  711. break;
  712. }
  713. case "Status":
  714. {
  715. var val = reader.ReadElementContentAsString();
  716. if (!string.IsNullOrWhiteSpace(val))
  717. {
  718. SeriesStatus seriesStatus;
  719. if (Enum.TryParse(val, true, out seriesStatus))
  720. item.Status = seriesStatus;
  721. }
  722. break;
  723. }
  724. case "FirstAired":
  725. {
  726. var val = reader.ReadElementContentAsString();
  727. if (!string.IsNullOrWhiteSpace(val))
  728. {
  729. DateTime date;
  730. if (DateTime.TryParse(val, out date))
  731. {
  732. date = date.ToUniversalTime();
  733. item.PremiereDate = date;
  734. item.ProductionYear = date.Year;
  735. }
  736. }
  737. break;
  738. }
  739. case "Runtime":
  740. {
  741. var val = reader.ReadElementContentAsString();
  742. if (!string.IsNullOrWhiteSpace(val))
  743. {
  744. int rval;
  745. // int.TryParse is local aware, so it can be probamatic, force us culture
  746. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
  747. {
  748. item.RunTimeTicks = TimeSpan.FromMinutes(rval).Ticks;
  749. }
  750. }
  751. break;
  752. }
  753. case "Genre":
  754. {
  755. var val = reader.ReadElementContentAsString();
  756. if (!string.IsNullOrWhiteSpace(val))
  757. {
  758. var vals = val
  759. .Split(new[] { '|' }, StringSplitOptions.RemoveEmptyEntries)
  760. .Select(i => i.Trim())
  761. .Where(i => !string.IsNullOrWhiteSpace(i))
  762. .ToList();
  763. if (vals.Count > 0)
  764. {
  765. item.Genres.Clear();
  766. foreach (var genre in vals)
  767. {
  768. item.AddGenre(genre);
  769. }
  770. }
  771. }
  772. break;
  773. }
  774. case "Network":
  775. {
  776. var val = reader.ReadElementContentAsString();
  777. if (!string.IsNullOrWhiteSpace(val))
  778. {
  779. var vals = val
  780. .Split(new[] { '|' }, StringSplitOptions.RemoveEmptyEntries)
  781. .Select(i => i.Trim())
  782. .Where(i => !string.IsNullOrWhiteSpace(i))
  783. .ToList();
  784. if (vals.Count > 0)
  785. {
  786. item.Studios.Clear();
  787. foreach (var genre in vals)
  788. {
  789. item.AddStudio(genre);
  790. }
  791. }
  792. }
  793. break;
  794. }
  795. default:
  796. reader.Skip();
  797. break;
  798. }
  799. }
  800. }
  801. }
  802. /// <summary>
  803. /// Extracts info for each episode into invididual xml files so that they can be easily accessed without having to step through the entire series xml
  804. /// </summary>
  805. /// <param name="seriesDataPath">The series data path.</param>
  806. /// <param name="xmlFile">The XML file.</param>
  807. /// <param name="lastTvDbUpdateTime">The last tv db update time.</param>
  808. /// <returns>Task.</returns>
  809. private async Task ExtractEpisodes(string seriesDataPath, string xmlFile, long? lastTvDbUpdateTime)
  810. {
  811. var settings = new XmlReaderSettings
  812. {
  813. CheckCharacters = false,
  814. IgnoreProcessingInstructions = true,
  815. IgnoreComments = true,
  816. ValidationType = ValidationType.None
  817. };
  818. using (var streamReader = new StreamReader(xmlFile, Encoding.UTF8))
  819. {
  820. // Use XmlReader for best performance
  821. using (var reader = XmlReader.Create(streamReader, settings))
  822. {
  823. reader.MoveToContent();
  824. // Loop through each element
  825. while (reader.Read())
  826. {
  827. if (reader.NodeType == XmlNodeType.Element)
  828. {
  829. switch (reader.Name)
  830. {
  831. case "Episode":
  832. {
  833. var outerXml = reader.ReadOuterXml();
  834. await SaveEpsiodeXml(seriesDataPath, outerXml, lastTvDbUpdateTime).ConfigureAwait(false);
  835. break;
  836. }
  837. default:
  838. reader.Skip();
  839. break;
  840. }
  841. }
  842. }
  843. }
  844. }
  845. }
  846. private async Task SaveEpsiodeXml(string seriesDataPath, string xml, long? lastTvDbUpdateTime)
  847. {
  848. var settings = new XmlReaderSettings
  849. {
  850. CheckCharacters = false,
  851. IgnoreProcessingInstructions = true,
  852. IgnoreComments = true,
  853. ValidationType = ValidationType.None
  854. };
  855. var seasonNumber = -1;
  856. var episodeNumber = -1;
  857. var absoluteNumber = -1;
  858. var lastUpdateString = string.Empty;
  859. using (var streamReader = new StringReader(xml))
  860. {
  861. // Use XmlReader for best performance
  862. using (var reader = XmlReader.Create(streamReader, settings))
  863. {
  864. reader.MoveToContent();
  865. // Loop through each element
  866. while (reader.Read())
  867. {
  868. if (reader.NodeType == XmlNodeType.Element)
  869. {
  870. switch (reader.Name)
  871. {
  872. case "lastupdated":
  873. {
  874. lastUpdateString = reader.ReadElementContentAsString();
  875. break;
  876. }
  877. case "EpisodeNumber":
  878. {
  879. var val = reader.ReadElementContentAsString();
  880. if (!string.IsNullOrWhiteSpace(val))
  881. {
  882. int num;
  883. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out num))
  884. {
  885. episodeNumber = num;
  886. }
  887. }
  888. break;
  889. }
  890. case "absolute_number":
  891. {
  892. var val = reader.ReadElementContentAsString();
  893. if (!string.IsNullOrWhiteSpace(val))
  894. {
  895. int num;
  896. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out num))
  897. {
  898. absoluteNumber = num;
  899. }
  900. }
  901. break;
  902. }
  903. case "SeasonNumber":
  904. {
  905. var val = reader.ReadElementContentAsString();
  906. if (!string.IsNullOrWhiteSpace(val))
  907. {
  908. int num;
  909. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out num))
  910. {
  911. seasonNumber = num;
  912. }
  913. }
  914. break;
  915. }
  916. default:
  917. reader.Skip();
  918. break;
  919. }
  920. }
  921. }
  922. }
  923. }
  924. var hasEpisodeChanged = true;
  925. if (!string.IsNullOrWhiteSpace(lastUpdateString) && lastTvDbUpdateTime.HasValue)
  926. {
  927. long num;
  928. if (long.TryParse(lastUpdateString, NumberStyles.Any, _usCulture, out num))
  929. {
  930. hasEpisodeChanged = num >= lastTvDbUpdateTime.Value;
  931. }
  932. }
  933. var file = Path.Combine(seriesDataPath, string.Format("episode-{0}-{1}.xml", seasonNumber, episodeNumber));
  934. // Only save the file if not already there, or if the episode has changed
  935. if (hasEpisodeChanged || !_fileSystem.FileExists(file))
  936. {
  937. using (var writer = XmlWriter.Create(file, new XmlWriterSettings
  938. {
  939. Encoding = Encoding.UTF8,
  940. Async = true
  941. }))
  942. {
  943. await writer.WriteRawAsync(xml).ConfigureAwait(false);
  944. }
  945. }
  946. if (absoluteNumber != -1)
  947. {
  948. file = Path.Combine(seriesDataPath, string.Format("episode-abs-{0}.xml", absoluteNumber));
  949. // Only save the file if not already there, or if the episode has changed
  950. if (hasEpisodeChanged || !_fileSystem.FileExists(file))
  951. {
  952. using (var writer = XmlWriter.Create(file, new XmlWriterSettings
  953. {
  954. Encoding = Encoding.UTF8,
  955. Async = true
  956. }))
  957. {
  958. await writer.WriteRawAsync(xml).ConfigureAwait(false);
  959. }
  960. }
  961. }
  962. }
  963. /// <summary>
  964. /// Gets the series data path.
  965. /// </summary>
  966. /// <param name="appPaths">The app paths.</param>
  967. /// <param name="seriesId">The series id.</param>
  968. /// <returns>System.String.</returns>
  969. internal static string GetSeriesDataPath(IApplicationPaths appPaths, string seriesId)
  970. {
  971. var seriesDataPath = Path.Combine(GetSeriesDataPath(appPaths), seriesId);
  972. return seriesDataPath;
  973. }
  974. /// <summary>
  975. /// Gets the series data path.
  976. /// </summary>
  977. /// <param name="appPaths">The app paths.</param>
  978. /// <returns>System.String.</returns>
  979. internal static string GetSeriesDataPath(IApplicationPaths appPaths)
  980. {
  981. var dataPath = Path.Combine(appPaths.CachePath, "tvdb");
  982. return dataPath;
  983. }
  984. private void DeleteXmlFiles(string path)
  985. {
  986. try
  987. {
  988. foreach (var file in _fileSystem.GetFilePaths(path, true)
  989. .ToList())
  990. {
  991. _fileSystem.DeleteFile(file);
  992. }
  993. }
  994. catch (DirectoryNotFoundException)
  995. {
  996. // No biggie
  997. }
  998. }
  999. /// <summary>
  1000. /// Sanitizes the XML file.
  1001. /// </summary>
  1002. /// <param name="file">The file.</param>
  1003. /// <returns>Task.</returns>
  1004. private async Task SanitizeXmlFile(string file)
  1005. {
  1006. string validXml;
  1007. using (var fileStream = _fileSystem.GetFileStream(file, FileMode.Open, FileAccess.Read, FileShare.Read, true))
  1008. {
  1009. using (var reader = new StreamReader(fileStream))
  1010. {
  1011. var xml = await reader.ReadToEndAsync().ConfigureAwait(false);
  1012. validXml = StripInvalidXmlCharacters(xml);
  1013. }
  1014. }
  1015. using (var fileStream = _fileSystem.GetFileStream(file, FileMode.Create, FileAccess.Write, FileShare.Read, true))
  1016. {
  1017. using (var writer = new StreamWriter(fileStream))
  1018. {
  1019. await writer.WriteAsync(validXml).ConfigureAwait(false);
  1020. }
  1021. }
  1022. }
  1023. /// <summary>
  1024. /// Strips the invalid XML characters.
  1025. /// </summary>
  1026. /// <param name="inString">The in string.</param>
  1027. /// <returns>System.String.</returns>
  1028. public static string StripInvalidXmlCharacters(string inString)
  1029. {
  1030. if (inString == null) return null;
  1031. var sbOutput = new StringBuilder();
  1032. char ch;
  1033. for (int i = 0; i < inString.Length; i++)
  1034. {
  1035. ch = inString[i];
  1036. if ((ch >= 0x0020 && ch <= 0xD7FF) ||
  1037. (ch >= 0xE000 && ch <= 0xFFFD) ||
  1038. ch == 0x0009 ||
  1039. ch == 0x000A ||
  1040. ch == 0x000D)
  1041. {
  1042. sbOutput.Append(ch);
  1043. }
  1044. }
  1045. return sbOutput.ToString();
  1046. }
  1047. public string Name
  1048. {
  1049. get { return "TheTVDB"; }
  1050. }
  1051. public async Task Identify(SeriesInfo info)
  1052. {
  1053. if (string.IsNullOrEmpty(info.GetProviderId(MetadataProviders.Tvdb)))
  1054. {
  1055. var srch = await FindSeries(info.Name, info.MetadataLanguage, CancellationToken.None).ConfigureAwait(false);
  1056. var entry = srch.FirstOrDefault();
  1057. if (entry != null)
  1058. {
  1059. var id = entry.GetProviderId(MetadataProviders.Tvdb);
  1060. info.SetProviderId(MetadataProviders.Tvdb, id);
  1061. }
  1062. }
  1063. }
  1064. public int Order
  1065. {
  1066. get
  1067. {
  1068. // After Omdb
  1069. return 1;
  1070. }
  1071. }
  1072. public Task<HttpResponseInfo> GetImageResponse(string url, CancellationToken cancellationToken)
  1073. {
  1074. return _httpClient.GetResponse(new HttpRequestOptions
  1075. {
  1076. CancellationToken = cancellationToken,
  1077. Url = url,
  1078. ResourcePool = TvDbResourcePool
  1079. });
  1080. }
  1081. }
  1082. public class TvdbConfigStore : IConfigurationFactory
  1083. {
  1084. public IEnumerable<ConfigurationStore> GetConfigurations()
  1085. {
  1086. return new List<ConfigurationStore>
  1087. {
  1088. new ConfigurationStore
  1089. {
  1090. Key = "tvdb",
  1091. ConfigurationType = typeof(TvdbOptions)
  1092. }
  1093. };
  1094. }
  1095. }
  1096. }