RemoteSeriesProvider.cs 47 KB

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