TvdbEpisodeProvider.cs 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977
  1. using MediaBrowser.Common.Net;
  2. using MediaBrowser.Controller.Configuration;
  3. using MediaBrowser.Controller.Entities;
  4. using MediaBrowser.Controller.Entities.TV;
  5. using MediaBrowser.Controller.Providers;
  6. using MediaBrowser.Model.Entities;
  7. using MediaBrowser.Model.Logging;
  8. using MediaBrowser.Model.Providers;
  9. using System;
  10. using System.Collections.Generic;
  11. using System.Globalization;
  12. using System.IO;
  13. using System.Linq;
  14. using System.Text;
  15. using System.Text.RegularExpressions;
  16. using System.Threading;
  17. using System.Threading.Tasks;
  18. using System.Xml;
  19. using CommonIO;
  20. namespace MediaBrowser.Providers.TV
  21. {
  22. /// <summary>
  23. /// Class RemoteEpisodeProvider
  24. /// </summary>
  25. class TvdbEpisodeProvider : IRemoteMetadataProvider<Episode, EpisodeInfo>, IItemIdentityProvider<EpisodeInfo>, IHasItemChangeMonitor
  26. {
  27. private static readonly string FullIdKey = MetadataProviders.Tvdb + "-Full";
  28. internal static TvdbEpisodeProvider Current;
  29. private readonly IFileSystem _fileSystem;
  30. private readonly IServerConfigurationManager _config;
  31. private readonly IHttpClient _httpClient;
  32. private readonly ILogger _logger;
  33. public TvdbEpisodeProvider(IFileSystem fileSystem, IServerConfigurationManager config, IHttpClient httpClient, ILogger logger)
  34. {
  35. _fileSystem = fileSystem;
  36. _config = config;
  37. _httpClient = httpClient;
  38. _logger = logger;
  39. Current = this;
  40. }
  41. public async Task<IEnumerable<RemoteSearchResult>> GetSearchResults(EpisodeInfo searchInfo, CancellationToken cancellationToken)
  42. {
  43. var list = new List<RemoteSearchResult>();
  44. // The search query must either provide an episode number or date
  45. if (!searchInfo.IndexNumber.HasValue && !searchInfo.PremiereDate.HasValue)
  46. {
  47. return list;
  48. }
  49. if (TvdbSeriesProvider.IsValidSeries(searchInfo.SeriesProviderIds))
  50. {
  51. var seriesDataPath = TvdbSeriesProvider.GetSeriesDataPath(_config.ApplicationPaths, searchInfo.SeriesProviderIds);
  52. var searchNumbers = new EpisodeNumbers();
  53. if (searchInfo.IndexNumber.HasValue) {
  54. searchNumbers.EpisodeNumber = searchInfo.IndexNumber.Value;
  55. }
  56. searchNumbers.SeasonNumber = searchInfo.ParentIndexNumber;
  57. searchNumbers.EpisodeNumberEnd = searchInfo.IndexNumberEnd ?? searchNumbers.EpisodeNumber;
  58. try
  59. {
  60. var metadataResult = FetchEpisodeData(searchInfo, searchNumbers, seriesDataPath, cancellationToken);
  61. if (metadataResult.HasMetadata)
  62. {
  63. var item = metadataResult.Item;
  64. list.Add(new RemoteSearchResult
  65. {
  66. IndexNumber = item.IndexNumber,
  67. Name = item.Name,
  68. ParentIndexNumber = item.ParentIndexNumber,
  69. PremiereDate = item.PremiereDate,
  70. ProductionYear = item.ProductionYear,
  71. ProviderIds = item.ProviderIds,
  72. SearchProviderName = Name,
  73. IndexNumberEnd = item.IndexNumberEnd
  74. });
  75. }
  76. }
  77. catch (FileNotFoundException)
  78. {
  79. // Don't fail the provider because this will just keep on going and going.
  80. }
  81. catch (DirectoryNotFoundException)
  82. {
  83. // Don't fail the provider because this will just keep on going and going.
  84. }
  85. }
  86. return list;
  87. }
  88. public string Name
  89. {
  90. get { return "TheTVDB"; }
  91. }
  92. public async Task<MetadataResult<Episode>> GetMetadata(EpisodeInfo searchInfo, CancellationToken cancellationToken)
  93. {
  94. var result = new MetadataResult<Episode>();
  95. if (TvdbSeriesProvider.IsValidSeries(searchInfo.SeriesProviderIds) &&
  96. (searchInfo.IndexNumber.HasValue || searchInfo.PremiereDate.HasValue))
  97. {
  98. await TvdbSeriesProvider.Current.EnsureSeriesInfo(searchInfo.SeriesProviderIds, searchInfo.MetadataLanguage, cancellationToken).ConfigureAwait(false);
  99. var seriesDataPath = TvdbSeriesProvider.GetSeriesDataPath(_config.ApplicationPaths, searchInfo.SeriesProviderIds);
  100. var searchNumbers = new EpisodeNumbers();
  101. if (searchInfo.IndexNumber.HasValue) {
  102. searchNumbers.EpisodeNumber = searchInfo.IndexNumber.Value;
  103. }
  104. searchNumbers.SeasonNumber = searchInfo.ParentIndexNumber;
  105. searchNumbers.EpisodeNumberEnd = searchInfo.IndexNumberEnd ?? searchNumbers.EpisodeNumber;
  106. try
  107. {
  108. result = FetchEpisodeData(searchInfo, searchNumbers, seriesDataPath, cancellationToken);
  109. }
  110. catch (FileNotFoundException)
  111. {
  112. // Don't fail the provider because this will just keep on going and going.
  113. }
  114. catch (DirectoryNotFoundException)
  115. {
  116. // Don't fail the provider because this will just keep on going and going.
  117. }
  118. }
  119. else
  120. {
  121. _logger.Debug("No series identity found for {0}", searchInfo.Name);
  122. }
  123. return result;
  124. }
  125. public bool HasChanged(IHasMetadata item, IDirectoryService directoryService)
  126. {
  127. if (!TvdbSeriesProvider.Current.GetTvDbOptions().EnableAutomaticUpdates)
  128. {
  129. return false;
  130. }
  131. var episode = (Episode)item;
  132. var series = episode.Series;
  133. if (series != null && TvdbSeriesProvider.IsValidSeries(series.ProviderIds))
  134. {
  135. // Process images
  136. var seriesXmlPath = TvdbSeriesProvider.Current.GetSeriesXmlPath(series.ProviderIds, series.GetPreferredMetadataLanguage());
  137. return _fileSystem.GetLastWriteTimeUtc(seriesXmlPath) > item.DateLastRefreshed;
  138. }
  139. return false;
  140. }
  141. /// <summary>
  142. /// Gets the episode XML files.
  143. /// </summary>
  144. /// <param name="seriesDataPath">The series data path.</param>
  145. /// <param name="searchInfo">The search information.</param>
  146. /// <returns>List{FileInfo}.</returns>
  147. internal List<XmlReader> GetEpisodeXmlNodes(string seriesDataPath, EpisodeInfo searchInfo)
  148. {
  149. var seriesXmlPath = TvdbSeriesProvider.Current.GetSeriesXmlPath (searchInfo.SeriesProviderIds, searchInfo.MetadataLanguage);
  150. try
  151. {
  152. return GetXmlNodes(seriesXmlPath, searchInfo);
  153. }
  154. catch (DirectoryNotFoundException)
  155. {
  156. return new List<XmlReader> ();
  157. }
  158. catch (FileNotFoundException)
  159. {
  160. return new List<XmlReader> ();
  161. }
  162. }
  163. private class EpisodeNumbers
  164. {
  165. public int EpisodeNumber;
  166. public int? SeasonNumber;
  167. public int EpisodeNumberEnd;
  168. }
  169. /// <summary>
  170. /// Fetches the episode data.
  171. /// </summary>
  172. /// <param name="id">The identifier.</param>
  173. /// <param name="searchNumbers">The search numbers.</param>
  174. /// <param name="seriesDataPath">The series data path.</param>
  175. /// <param name="cancellationToken">The cancellation token.</param>
  176. /// <returns>Task{System.Boolean}.</returns>
  177. private MetadataResult<Episode> FetchEpisodeData(EpisodeInfo id, EpisodeNumbers searchNumbers, string seriesDataPath, CancellationToken cancellationToken)
  178. {
  179. var result = new MetadataResult<Episode>()
  180. {
  181. Item = new Episode
  182. {
  183. IndexNumber = id.IndexNumber,
  184. ParentIndexNumber = id.ParentIndexNumber,
  185. IndexNumberEnd = id.IndexNumberEnd
  186. }
  187. };
  188. var xmlNodes = GetEpisodeXmlNodes (seriesDataPath, id);
  189. if (xmlNodes.Count > 0) {
  190. FetchMainEpisodeInfo(result, xmlNodes[0], cancellationToken);
  191. result.HasMetadata = true;
  192. }
  193. foreach (var node in xmlNodes.Skip(1)) {
  194. FetchAdditionalPartInfo(result, node, cancellationToken);
  195. }
  196. return result;
  197. }
  198. private List<XmlReader> GetXmlNodes(string xmlFile, EpisodeInfo searchInfo)
  199. {
  200. var list = new List<XmlReader> ();
  201. if (searchInfo.IndexNumber.HasValue)
  202. {
  203. var files = GetEpisodeXmlFiles (searchInfo.ParentIndexNumber, searchInfo.IndexNumber, searchInfo.IndexNumberEnd, Path.GetDirectoryName (xmlFile));
  204. list = files.Select (GetXmlReader).ToList ();
  205. }
  206. if (list.Count == 0 && searchInfo.PremiereDate.HasValue) {
  207. list = GetXmlNodesByPremiereDate (xmlFile, searchInfo.PremiereDate.Value);
  208. }
  209. return list;
  210. }
  211. private List<FileSystemMetadata> GetEpisodeXmlFiles(int? seasonNumber, int? episodeNumber, int? endingEpisodeNumber, string seriesDataPath)
  212. {
  213. var files = new List<FileSystemMetadata>();
  214. if (episodeNumber == null)
  215. {
  216. return files;
  217. }
  218. var usingAbsoluteData = false;
  219. if (seasonNumber.HasValue)
  220. {
  221. var file = Path.Combine(seriesDataPath, string.Format("episode-{0}-{1}.xml", seasonNumber.Value, episodeNumber));
  222. var fileInfo = _fileSystem.GetFileInfo(file);
  223. if (fileInfo.Exists)
  224. {
  225. files.Add(fileInfo);
  226. }
  227. }
  228. else
  229. {
  230. usingAbsoluteData = true;
  231. var file = Path.Combine(seriesDataPath, string.Format("episode-abs-{0}.xml", episodeNumber));
  232. var fileInfo = _fileSystem.GetFileInfo(file);
  233. if (fileInfo.Exists)
  234. {
  235. files.Add(fileInfo);
  236. }
  237. }
  238. var end = endingEpisodeNumber ?? episodeNumber;
  239. episodeNumber++;
  240. while (episodeNumber <= end)
  241. {
  242. string file;
  243. if (usingAbsoluteData)
  244. {
  245. file = Path.Combine(seriesDataPath, string.Format("episode-abs-{0}.xml", episodeNumber));
  246. }
  247. else
  248. {
  249. file = Path.Combine(seriesDataPath, string.Format("episode-{0}-{1}.xml", seasonNumber.Value, episodeNumber));
  250. }
  251. var fileInfo = _fileSystem.GetFileInfo(file);
  252. if (fileInfo.Exists)
  253. {
  254. files.Add(fileInfo);
  255. }
  256. else
  257. {
  258. break;
  259. }
  260. episodeNumber++;
  261. }
  262. return files;
  263. }
  264. private XmlReader GetXmlReader(FileSystemMetadata xmlFile)
  265. {
  266. return GetXmlReader (_fileSystem.ReadAllText(xmlFile.FullName, Encoding.UTF8));
  267. }
  268. private XmlReader GetXmlReader(String xml)
  269. {
  270. var streamReader = new StringReader (xml);
  271. return XmlReader.Create (streamReader, new XmlReaderSettings {
  272. CheckCharacters = false,
  273. IgnoreProcessingInstructions = true,
  274. IgnoreComments = true,
  275. ValidationType = ValidationType.None
  276. });
  277. }
  278. private List<XmlReader> GetXmlNodesByPremiereDate(string xmlFile, DateTime premiereDate)
  279. {
  280. var list = new List<XmlReader> ();
  281. using (var streamReader = new StreamReader (xmlFile, Encoding.UTF8)) {
  282. // Use XmlReader for best performance
  283. using (var reader = XmlReader.Create (streamReader, new XmlReaderSettings {
  284. CheckCharacters = false,
  285. IgnoreProcessingInstructions = true,
  286. IgnoreComments = true,
  287. ValidationType = ValidationType.None
  288. }))
  289. {
  290. reader.MoveToContent();
  291. // Loop through each element
  292. while (reader.Read())
  293. {
  294. if (reader.NodeType == XmlNodeType.Element)
  295. {
  296. switch (reader.Name)
  297. {
  298. case "Episode":
  299. {
  300. var outerXml = reader.ReadOuterXml();
  301. var airDate = GetEpisodeAirDate (outerXml);
  302. if (airDate.HasValue && premiereDate.Date == airDate.Value.Date)
  303. {
  304. list.Add (GetXmlReader(outerXml));
  305. return list;
  306. }
  307. break;
  308. }
  309. default:
  310. reader.Skip();
  311. break;
  312. }
  313. }
  314. }
  315. }
  316. }
  317. return list;
  318. }
  319. private DateTime? GetEpisodeAirDate(string xml)
  320. {
  321. using (var streamReader = new StringReader (xml))
  322. {
  323. // Use XmlReader for best performance
  324. using (var reader = XmlReader.Create (streamReader, new XmlReaderSettings {
  325. CheckCharacters = false,
  326. IgnoreProcessingInstructions = true,
  327. IgnoreComments = true,
  328. ValidationType = ValidationType.None
  329. }))
  330. {
  331. reader.MoveToContent ();
  332. // Loop through each element
  333. while (reader.Read ()) {
  334. if (reader.NodeType == XmlNodeType.Element) {
  335. switch (reader.Name) {
  336. case "FirstAired":
  337. {
  338. var val = reader.ReadElementContentAsString ();
  339. if (!string.IsNullOrWhiteSpace (val)) {
  340. DateTime date;
  341. if (DateTime.TryParse (val, out date)) {
  342. date = date.ToUniversalTime ();
  343. return date;
  344. }
  345. }
  346. break;
  347. }
  348. default:
  349. reader.Skip ();
  350. break;
  351. }
  352. }
  353. }
  354. }
  355. }
  356. return null;
  357. }
  358. private readonly CultureInfo _usCulture = new CultureInfo("en-US");
  359. private void FetchMainEpisodeInfo(MetadataResult<Episode> result, XmlReader reader, CancellationToken cancellationToken)
  360. {
  361. var item = result.Item;
  362. // Use XmlReader for best performance
  363. using (reader)
  364. {
  365. reader.MoveToContent();
  366. result.ResetPeople();
  367. // Loop through each element
  368. while (reader.Read())
  369. {
  370. cancellationToken.ThrowIfCancellationRequested();
  371. if (reader.NodeType == XmlNodeType.Element)
  372. {
  373. switch (reader.Name)
  374. {
  375. case "id":
  376. {
  377. var val = reader.ReadElementContentAsString();
  378. if (!string.IsNullOrWhiteSpace(val))
  379. {
  380. item.SetProviderId(MetadataProviders.Tvdb, val);
  381. }
  382. break;
  383. }
  384. case "IMDB_ID":
  385. {
  386. var val = reader.ReadElementContentAsString();
  387. if (!string.IsNullOrWhiteSpace(val))
  388. {
  389. item.SetProviderId(MetadataProviders.Imdb, val);
  390. }
  391. break;
  392. }
  393. case "DVD_episodenumber":
  394. {
  395. var val = reader.ReadElementContentAsString();
  396. if (!string.IsNullOrWhiteSpace(val))
  397. {
  398. float num;
  399. if (float.TryParse(val, NumberStyles.Any, _usCulture, out num))
  400. {
  401. item.DvdEpisodeNumber = num;
  402. }
  403. }
  404. break;
  405. }
  406. case "DVD_season":
  407. {
  408. var val = reader.ReadElementContentAsString();
  409. if (!string.IsNullOrWhiteSpace(val))
  410. {
  411. float num;
  412. if (float.TryParse(val, NumberStyles.Any, _usCulture, out num))
  413. {
  414. item.DvdSeasonNumber = Convert.ToInt32(num);
  415. }
  416. }
  417. break;
  418. }
  419. case "EpisodeNumber":
  420. {
  421. if (!item.IndexNumber.HasValue)
  422. {
  423. var val = reader.ReadElementContentAsString();
  424. if (!string.IsNullOrWhiteSpace(val))
  425. {
  426. int rval;
  427. // int.TryParse is local aware, so it can be probamatic, force us culture
  428. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
  429. {
  430. item.IndexNumber = rval;
  431. }
  432. }
  433. }
  434. break;
  435. }
  436. case "SeasonNumber":
  437. {
  438. if (!item.ParentIndexNumber.HasValue)
  439. {
  440. var val = reader.ReadElementContentAsString();
  441. if (!string.IsNullOrWhiteSpace(val))
  442. {
  443. int rval;
  444. // int.TryParse is local aware, so it can be probamatic, force us culture
  445. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
  446. {
  447. item.ParentIndexNumber = rval;
  448. }
  449. }
  450. }
  451. break;
  452. }
  453. case "absolute_number":
  454. {
  455. var val = reader.ReadElementContentAsString();
  456. if (!string.IsNullOrWhiteSpace(val))
  457. {
  458. int rval;
  459. // int.TryParse is local aware, so it can be probamatic, force us culture
  460. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
  461. {
  462. item.AbsoluteEpisodeNumber = rval;
  463. }
  464. }
  465. break;
  466. }
  467. case "airsbefore_episode":
  468. {
  469. var val = reader.ReadElementContentAsString();
  470. if (!string.IsNullOrWhiteSpace(val))
  471. {
  472. int rval;
  473. // int.TryParse is local aware, so it can be probamatic, force us culture
  474. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
  475. {
  476. item.AirsBeforeEpisodeNumber = rval;
  477. }
  478. }
  479. break;
  480. }
  481. case "airsafter_season":
  482. {
  483. var val = reader.ReadElementContentAsString();
  484. if (!string.IsNullOrWhiteSpace(val))
  485. {
  486. int rval;
  487. // int.TryParse is local aware, so it can be probamatic, force us culture
  488. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
  489. {
  490. item.AirsAfterSeasonNumber = rval;
  491. }
  492. }
  493. break;
  494. }
  495. case "airsbefore_season":
  496. {
  497. var val = reader.ReadElementContentAsString();
  498. if (!string.IsNullOrWhiteSpace(val))
  499. {
  500. int rval;
  501. // int.TryParse is local aware, so it can be probamatic, force us culture
  502. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
  503. {
  504. item.AirsBeforeSeasonNumber = rval;
  505. }
  506. }
  507. break;
  508. }
  509. case "EpisodeName":
  510. {
  511. if (!item.LockedFields.Contains(MetadataFields.Name))
  512. {
  513. var val = reader.ReadElementContentAsString();
  514. if (!string.IsNullOrWhiteSpace(val))
  515. {
  516. item.Name = val;
  517. }
  518. }
  519. break;
  520. }
  521. case "Overview":
  522. {
  523. if (!item.LockedFields.Contains(MetadataFields.Overview))
  524. {
  525. var val = reader.ReadElementContentAsString();
  526. if (!string.IsNullOrWhiteSpace(val))
  527. {
  528. item.Overview = val;
  529. }
  530. }
  531. break;
  532. }
  533. case "Rating":
  534. {
  535. var val = reader.ReadElementContentAsString();
  536. if (!string.IsNullOrWhiteSpace(val))
  537. {
  538. float rval;
  539. // float.TryParse is local aware, so it can be probamatic, force us culture
  540. if (float.TryParse(val, NumberStyles.AllowDecimalPoint, _usCulture, out rval))
  541. {
  542. item.CommunityRating = rval;
  543. }
  544. }
  545. break;
  546. }
  547. case "RatingCount":
  548. {
  549. var val = reader.ReadElementContentAsString();
  550. if (!string.IsNullOrWhiteSpace(val))
  551. {
  552. int rval;
  553. // int.TryParse is local aware, so it can be probamatic, force us culture
  554. if (int.TryParse(val, NumberStyles.Integer, _usCulture, out rval))
  555. {
  556. item.VoteCount = rval;
  557. }
  558. }
  559. break;
  560. }
  561. case "FirstAired":
  562. {
  563. var val = reader.ReadElementContentAsString();
  564. if (!string.IsNullOrWhiteSpace(val))
  565. {
  566. DateTime date;
  567. if (DateTime.TryParse(val, out date))
  568. {
  569. date = date.ToUniversalTime();
  570. item.PremiereDate = date;
  571. item.ProductionYear = date.Year;
  572. }
  573. }
  574. break;
  575. }
  576. case "Director":
  577. {
  578. var val = reader.ReadElementContentAsString();
  579. if (!string.IsNullOrWhiteSpace(val))
  580. {
  581. if (!item.LockedFields.Contains(MetadataFields.Cast))
  582. {
  583. AddPeople(result, val, PersonType.Director);
  584. }
  585. }
  586. break;
  587. }
  588. case "GuestStars":
  589. {
  590. var val = reader.ReadElementContentAsString();
  591. if (!string.IsNullOrWhiteSpace(val))
  592. {
  593. if (!item.LockedFields.Contains(MetadataFields.Cast))
  594. {
  595. AddGuestStars(result, val);
  596. }
  597. }
  598. break;
  599. }
  600. case "Writer":
  601. {
  602. var val = reader.ReadElementContentAsString();
  603. if (!string.IsNullOrWhiteSpace(val))
  604. {
  605. if (!item.LockedFields.Contains(MetadataFields.Cast))
  606. {
  607. AddPeople(result, val, PersonType.Writer);
  608. }
  609. }
  610. break;
  611. }
  612. default:
  613. reader.Skip();
  614. break;
  615. }
  616. }
  617. }
  618. }
  619. }
  620. private void AddPeople<T>(MetadataResult<T> result, string val, string personType)
  621. {
  622. // Sometimes tvdb actors have leading spaces
  623. foreach (var person in val.Split(new[] { '|', ',' }, StringSplitOptions.RemoveEmptyEntries)
  624. .Where(i => !string.IsNullOrWhiteSpace(i))
  625. .Select(str => new PersonInfo { Type = personType, Name = str.Trim() }))
  626. {
  627. result.AddPerson(person);
  628. }
  629. }
  630. private void AddGuestStars<T>(MetadataResult<T> result, string val)
  631. where T : BaseItem
  632. {
  633. // Sometimes tvdb actors have leading spaces
  634. //Regex Info:
  635. //The first block are the posible delimitators (open-parentheses should be there cause if dont the next block will fail)
  636. //The second block Allow the delimitators to be part of the text if they're inside parentheses
  637. var persons = Regex.Matches(val, @"(?<delimitators>([^|,(])|(?<ignoreinParentheses>\([^)]*\)*))+")
  638. .Cast<Match>()
  639. .Select(m => m.Value)
  640. .Where(i => !string.IsNullOrWhiteSpace(i) && !string.IsNullOrEmpty(i));
  641. foreach (var person in persons.Select(str =>
  642. {
  643. var nameGroup = str.Split(new[] { '(' }, 2, StringSplitOptions.RemoveEmptyEntries);
  644. var name = nameGroup[0].Trim();
  645. var roles = nameGroup.Count() > 1 ? nameGroup[1].Trim() : null;
  646. if (roles != null)
  647. roles = roles.EndsWith(")") ? roles.Substring(0, roles.Length - 1) : roles;
  648. return new PersonInfo { Type = PersonType.GuestStar, Name = name, Role = roles };
  649. }))
  650. {
  651. if (!string.IsNullOrWhiteSpace(person.Name))
  652. {
  653. result.AddPerson(person);
  654. }
  655. }
  656. }
  657. private void FetchAdditionalPartInfo(MetadataResult<Episode> result, XmlReader reader, CancellationToken cancellationToken)
  658. {
  659. var item = result.Item;
  660. // Use XmlReader for best performance
  661. using (reader)
  662. {
  663. reader.MoveToContent();
  664. // Loop through each element
  665. while (reader.Read())
  666. {
  667. cancellationToken.ThrowIfCancellationRequested();
  668. if (reader.NodeType == XmlNodeType.Element)
  669. {
  670. switch (reader.Name)
  671. {
  672. case "EpisodeName":
  673. {
  674. if (!item.LockedFields.Contains(MetadataFields.Name))
  675. {
  676. var val = reader.ReadElementContentAsString();
  677. if (!string.IsNullOrWhiteSpace(val))
  678. {
  679. item.Name += ", " + val;
  680. }
  681. }
  682. break;
  683. }
  684. case "Overview":
  685. {
  686. if (!item.LockedFields.Contains(MetadataFields.Overview))
  687. {
  688. var val = reader.ReadElementContentAsString();
  689. if (!string.IsNullOrWhiteSpace(val))
  690. {
  691. item.Overview += Environment.NewLine + Environment.NewLine + val;
  692. }
  693. }
  694. break;
  695. }
  696. case "Director":
  697. {
  698. var val = reader.ReadElementContentAsString();
  699. if (!string.IsNullOrWhiteSpace(val))
  700. {
  701. if (!item.LockedFields.Contains(MetadataFields.Cast))
  702. {
  703. AddPeople(result, val, PersonType.Director);
  704. }
  705. }
  706. break;
  707. }
  708. case "GuestStars":
  709. {
  710. var val = reader.ReadElementContentAsString();
  711. if (!string.IsNullOrWhiteSpace(val))
  712. {
  713. if (!item.LockedFields.Contains(MetadataFields.Cast))
  714. {
  715. AddGuestStars(result, val);
  716. }
  717. }
  718. break;
  719. }
  720. case "Writer":
  721. {
  722. var val = reader.ReadElementContentAsString();
  723. if (!string.IsNullOrWhiteSpace(val))
  724. {
  725. if (!item.LockedFields.Contains(MetadataFields.Cast))
  726. {
  727. AddPeople(result, val, PersonType.Writer);
  728. }
  729. }
  730. break;
  731. }
  732. default:
  733. reader.Skip();
  734. break;
  735. }
  736. }
  737. }
  738. }
  739. }
  740. public Task<HttpResponseInfo> GetImageResponse(string url, CancellationToken cancellationToken)
  741. {
  742. return _httpClient.GetResponse(new HttpRequestOptions
  743. {
  744. CancellationToken = cancellationToken,
  745. Url = url,
  746. ResourcePool = TvdbSeriesProvider.Current.TvDbResourcePool
  747. });
  748. }
  749. public Task Identify(EpisodeInfo info)
  750. {
  751. if (info.ProviderIds.ContainsKey(FullIdKey))
  752. {
  753. return Task.FromResult<object>(null);
  754. }
  755. string seriesTvdbId;
  756. info.SeriesProviderIds.TryGetValue(MetadataProviders.Tvdb.ToString(), out seriesTvdbId);
  757. if (string.IsNullOrEmpty(seriesTvdbId) || info.IndexNumber == null)
  758. {
  759. return Task.FromResult<object>(null);
  760. }
  761. var id = new Identity(seriesTvdbId, info.ParentIndexNumber, info.IndexNumber.Value, info.IndexNumberEnd);
  762. info.SetProviderId(FullIdKey, id.ToString());
  763. return Task.FromResult(id);
  764. }
  765. public int Order { get { return 0; } }
  766. public struct Identity
  767. {
  768. public string SeriesId { get; private set; }
  769. public int? SeasonIndex { get; private set; }
  770. public int EpisodeNumber { get; private set; }
  771. public int? EpisodeNumberEnd { get; private set; }
  772. public Identity(string id)
  773. : this()
  774. {
  775. this = ParseIdentity(id).Value;
  776. }
  777. public Identity(string seriesId, int? seasonIndex, int episodeNumber, int? episodeNumberEnd)
  778. : this()
  779. {
  780. SeriesId = seriesId;
  781. SeasonIndex = seasonIndex;
  782. EpisodeNumber = episodeNumber;
  783. EpisodeNumberEnd = episodeNumberEnd;
  784. }
  785. public override string ToString()
  786. {
  787. return string.Format("{0}:{1}:{2}",
  788. SeriesId,
  789. SeasonIndex != null ? SeasonIndex.Value.ToString() : "A",
  790. EpisodeNumber + (EpisodeNumberEnd != null ? "-" + EpisodeNumberEnd.Value.ToString() : ""));
  791. }
  792. public static Identity? ParseIdentity(string id)
  793. {
  794. if (string.IsNullOrEmpty(id))
  795. return null;
  796. try {
  797. var parts = id.Split(':');
  798. var series = parts[0];
  799. var season = parts[1] != "A" ? (int?)int.Parse(parts[1]) : null;
  800. int index;
  801. int? indexEnd;
  802. if (parts[2].Contains("-")) {
  803. var split = parts[2].IndexOf("-", StringComparison.OrdinalIgnoreCase);
  804. index = int.Parse(parts[2].Substring(0, split));
  805. indexEnd = int.Parse(parts[2].Substring(split + 1));
  806. } else {
  807. index = int.Parse(parts[2]);
  808. indexEnd = null;
  809. }
  810. return new Identity(series, season, index, indexEnd);
  811. } catch {
  812. return null;
  813. }
  814. }
  815. }
  816. }
  817. }