TvdbSeriesProvider.cs 50 KB

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