|
@@ -1,19 +1,13 @@
|
|
|
-using MediaBrowser.Common.Extensions;
|
|
|
-using MediaBrowser.Controller.Configuration;
|
|
|
+using MediaBrowser.Controller.Configuration;
|
|
|
using MediaBrowser.Controller.Entities.TV;
|
|
|
using MediaBrowser.Controller.Library;
|
|
|
using MediaBrowser.Controller.Providers;
|
|
|
using MediaBrowser.Model.Entities;
|
|
|
using MediaBrowser.Model.Logging;
|
|
|
using System;
|
|
|
-using System.Collections.Generic;
|
|
|
-using System.Globalization;
|
|
|
-using System.IO;
|
|
|
using System.Linq;
|
|
|
-using System.Text;
|
|
|
using System.Threading;
|
|
|
using System.Threading.Tasks;
|
|
|
-using System.Xml;
|
|
|
|
|
|
namespace MediaBrowser.Providers.TV
|
|
|
{
|
|
@@ -100,478 +94,4 @@ namespace MediaBrowser.Providers.TV
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- class MissingEpisodeProvider
|
|
|
- {
|
|
|
- private readonly IServerConfigurationManager _config;
|
|
|
- private readonly ILogger _logger;
|
|
|
-
|
|
|
- private static readonly CultureInfo UsCulture = new CultureInfo("en-US");
|
|
|
-
|
|
|
- public MissingEpisodeProvider(ILogger logger, IServerConfigurationManager config)
|
|
|
- {
|
|
|
- _logger = logger;
|
|
|
- _config = config;
|
|
|
- }
|
|
|
-
|
|
|
- public async Task Run(IEnumerable<IGrouping<string, Series>> series, CancellationToken cancellationToken)
|
|
|
- {
|
|
|
- foreach (var seriesGroup in series)
|
|
|
- {
|
|
|
- await Run(seriesGroup, cancellationToken).ConfigureAwait(false);
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- private async Task Run(IGrouping<string, Series> group, CancellationToken cancellationToken)
|
|
|
- {
|
|
|
- var tvdbId = group.Key;
|
|
|
-
|
|
|
- var seriesDataPath = TvdbSeriesProvider.GetSeriesDataPath(_config.ApplicationPaths, tvdbId);
|
|
|
-
|
|
|
- var episodeFiles = Directory.EnumerateFiles(seriesDataPath, "*.xml", SearchOption.TopDirectoryOnly)
|
|
|
- .Select(Path.GetFileNameWithoutExtension)
|
|
|
- .Where(i => i.StartsWith("episode-", StringComparison.OrdinalIgnoreCase))
|
|
|
- .ToList();
|
|
|
-
|
|
|
- var episodeLookup = episodeFiles
|
|
|
- .Select(i =>
|
|
|
- {
|
|
|
- var parts = i.Split('-');
|
|
|
-
|
|
|
- if (parts.Length == 3)
|
|
|
- {
|
|
|
- int seasonNumber;
|
|
|
-
|
|
|
- if (int.TryParse(parts[1], NumberStyles.Integer, UsCulture, out seasonNumber))
|
|
|
- {
|
|
|
- int episodeNumber;
|
|
|
-
|
|
|
- if (int.TryParse(parts[2], NumberStyles.Integer, UsCulture, out episodeNumber))
|
|
|
- {
|
|
|
- return new Tuple<int, int>(seasonNumber, episodeNumber);
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- return new Tuple<int, int>(-1, -1);
|
|
|
- })
|
|
|
- .Where(i => i.Item1 != -1 && i.Item2 != -1)
|
|
|
- .ToList();
|
|
|
-
|
|
|
- var anySeasonsRemoved = await RemoveObsoleteOrMissingSeasons(group, episodeLookup, cancellationToken)
|
|
|
- .ConfigureAwait(false);
|
|
|
-
|
|
|
- var anyEpisodesRemoved = await RemoveObsoleteOrMissingEpisodes(group, episodeLookup, cancellationToken)
|
|
|
- .ConfigureAwait(false);
|
|
|
-
|
|
|
- var hasNewEpisodes = false;
|
|
|
- var hasNewSeasons = false;
|
|
|
-
|
|
|
- foreach (var series in group.Where(s => s.ContainsEpisodesWithoutSeasonFolders))
|
|
|
- {
|
|
|
- hasNewSeasons = await AddDummySeasonFolders(series, cancellationToken).ConfigureAwait(false);
|
|
|
- }
|
|
|
-
|
|
|
- var seriesConfig = _config.Configuration.MetadataOptions.FirstOrDefault(i => string.Equals(i.ItemType, typeof(Series).Name, StringComparison.OrdinalIgnoreCase));
|
|
|
-
|
|
|
- if (seriesConfig == null || !seriesConfig.DisabledMetadataFetchers.Contains(TvdbSeriesProvider.Current.Name, StringComparer.OrdinalIgnoreCase))
|
|
|
- {
|
|
|
- hasNewEpisodes = await AddMissingEpisodes(group.ToList(), seriesDataPath, episodeLookup, cancellationToken)
|
|
|
- .ConfigureAwait(false);
|
|
|
- }
|
|
|
-
|
|
|
- if (hasNewSeasons || hasNewEpisodes || anySeasonsRemoved || anyEpisodesRemoved)
|
|
|
- {
|
|
|
- foreach (var series in group)
|
|
|
- {
|
|
|
- await series.RefreshMetadata(new MetadataRefreshOptions
|
|
|
- {
|
|
|
- }, cancellationToken).ConfigureAwait(false);
|
|
|
-
|
|
|
- await series.ValidateChildren(new Progress<double>(), cancellationToken, new MetadataRefreshOptions(), true)
|
|
|
- .ConfigureAwait(false);
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- /// <summary>
|
|
|
- /// For series with episodes directly under the series folder, this adds dummy seasons to enable regular browsing and metadata
|
|
|
- /// </summary>
|
|
|
- /// <param name="series"></param>
|
|
|
- /// <param name="cancellationToken"></param>
|
|
|
- /// <returns></returns>
|
|
|
- private async Task<bool> AddDummySeasonFolders(Series series, CancellationToken cancellationToken)
|
|
|
- {
|
|
|
- var existingEpisodes = series.RecursiveChildren
|
|
|
- .OfType<Episode>()
|
|
|
- .ToList();
|
|
|
-
|
|
|
- var hasChanges = false;
|
|
|
-
|
|
|
- // Loop through the unique season numbers
|
|
|
- foreach (var seasonNumber in existingEpisodes.Select(i => i.ParentIndexNumber ?? -1)
|
|
|
- .Where(i => i >= 0)
|
|
|
- .Distinct()
|
|
|
- .ToList())
|
|
|
- {
|
|
|
- var hasSeason = series.Children.OfType<Season>()
|
|
|
- .Any(i => i.IndexNumber.HasValue && i.IndexNumber.Value == seasonNumber);
|
|
|
-
|
|
|
- if (!hasSeason)
|
|
|
- {
|
|
|
- await AddSeason(series, seasonNumber, cancellationToken).ConfigureAwait(false);
|
|
|
-
|
|
|
- hasChanges = true;
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- return hasChanges;
|
|
|
- }
|
|
|
-
|
|
|
- /// <summary>
|
|
|
- /// Adds the missing episodes.
|
|
|
- /// </summary>
|
|
|
- /// <param name="series">The series.</param>
|
|
|
- /// <param name="seriesDataPath">The series data path.</param>
|
|
|
- /// <param name="episodeLookup">The episode lookup.</param>
|
|
|
- /// <param name="cancellationToken">The cancellation token.</param>
|
|
|
- /// <returns>Task.</returns>
|
|
|
- private async Task<bool> AddMissingEpisodes(List<Series> series, string seriesDataPath, IEnumerable<Tuple<int, int>> episodeLookup, CancellationToken cancellationToken)
|
|
|
- {
|
|
|
- var existingEpisodes = series.SelectMany(s => s.RecursiveChildren.OfType<Episode>()).ToList();
|
|
|
-
|
|
|
- var hasChanges = false;
|
|
|
-
|
|
|
- foreach (var tuple in episodeLookup)
|
|
|
- {
|
|
|
- if (tuple.Item1 <= 0)
|
|
|
- {
|
|
|
- // Ignore season zeros
|
|
|
- continue;
|
|
|
- }
|
|
|
-
|
|
|
- if (tuple.Item2 <= 0)
|
|
|
- {
|
|
|
- // Ignore episode zeros
|
|
|
- continue;
|
|
|
- }
|
|
|
-
|
|
|
- var existingEpisode = GetExistingEpisode(existingEpisodes, tuple);
|
|
|
-
|
|
|
- if (existingEpisode != null)
|
|
|
- {
|
|
|
- continue;
|
|
|
- }
|
|
|
-
|
|
|
- var airDate = GetAirDate(seriesDataPath, tuple.Item1, tuple.Item2);
|
|
|
-
|
|
|
- if (!airDate.HasValue)
|
|
|
- {
|
|
|
- continue;
|
|
|
- }
|
|
|
- var now = DateTime.UtcNow;
|
|
|
-
|
|
|
- var targetSeries = DetermineAppropriateSeries(series, tuple.Item1);
|
|
|
-
|
|
|
- if (airDate.Value < now)
|
|
|
- {
|
|
|
- // tvdb has a lot of nearly blank episodes
|
|
|
- _logger.Info("Creating virtual missing episode {0} {1}x{2}", targetSeries.Name, tuple.Item1, tuple.Item2);
|
|
|
-
|
|
|
- await AddEpisode(targetSeries, tuple.Item1, tuple.Item2, cancellationToken).ConfigureAwait(false);
|
|
|
-
|
|
|
- hasChanges = true;
|
|
|
- }
|
|
|
- else if (airDate.Value > now)
|
|
|
- {
|
|
|
- // tvdb has a lot of nearly blank episodes
|
|
|
- _logger.Info("Creating virtual unaired episode {0} {1}x{2}", targetSeries.Name, tuple.Item1, tuple.Item2);
|
|
|
-
|
|
|
- await AddEpisode(targetSeries, tuple.Item1, tuple.Item2, cancellationToken).ConfigureAwait(false);
|
|
|
-
|
|
|
- hasChanges = true;
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- return hasChanges;
|
|
|
- }
|
|
|
-
|
|
|
- private Series DetermineAppropriateSeries(List<Series> series, int seasonNumber)
|
|
|
- {
|
|
|
- return series.FirstOrDefault(s => s.RecursiveChildren.OfType<Season>().Any(season => season.IndexNumber == seasonNumber)) ??
|
|
|
- series.FirstOrDefault(s => s.RecursiveChildren.OfType<Season>().Any(season => season.IndexNumber == 1)) ??
|
|
|
- series.OrderBy(s => s.RecursiveChildren.OfType<Season>().Select(season => season.IndexNumber).Min()).First();
|
|
|
- }
|
|
|
-
|
|
|
- /// <summary>
|
|
|
- /// Removes the virtual entry after a corresponding physical version has been added
|
|
|
- /// </summary>
|
|
|
- private async Task<bool> RemoveObsoleteOrMissingEpisodes(IEnumerable<Series> series, IEnumerable<Tuple<int, int>> episodeLookup, CancellationToken cancellationToken)
|
|
|
- {
|
|
|
- var existingEpisodes = series.SelectMany(s => s.RecursiveChildren.OfType<Episode>()).ToList();
|
|
|
-
|
|
|
- var physicalEpisodes = existingEpisodes
|
|
|
- .Where(i => i.LocationType != LocationType.Virtual)
|
|
|
- .ToList();
|
|
|
-
|
|
|
- var virtualEpisodes = existingEpisodes
|
|
|
- .Where(i => i.LocationType == LocationType.Virtual)
|
|
|
- .ToList();
|
|
|
-
|
|
|
- var episodesToRemove = virtualEpisodes
|
|
|
- .Where(i =>
|
|
|
- {
|
|
|
- if (i.IndexNumber.HasValue && i.ParentIndexNumber.HasValue)
|
|
|
- {
|
|
|
- var seasonNumber = i.ParentIndexNumber.Value;
|
|
|
- var episodeNumber = i.IndexNumber.Value;
|
|
|
-
|
|
|
- // If there's a physical episode with the same season and episode number, delete it
|
|
|
- if (physicalEpisodes.Any(p =>
|
|
|
- p.ParentIndexNumber.HasValue && p.ParentIndexNumber.Value == seasonNumber &&
|
|
|
- p.ContainsEpisodeNumber(episodeNumber)))
|
|
|
- {
|
|
|
- return true;
|
|
|
- }
|
|
|
-
|
|
|
- // If the episode no longer exists in the remote lookup, delete it
|
|
|
- if (!episodeLookup.Any(e => e.Item1 == seasonNumber && e.Item2 == episodeNumber))
|
|
|
- {
|
|
|
- return true;
|
|
|
- }
|
|
|
-
|
|
|
- return false;
|
|
|
- }
|
|
|
-
|
|
|
- return true;
|
|
|
- })
|
|
|
- .ToList();
|
|
|
-
|
|
|
- var hasChanges = false;
|
|
|
-
|
|
|
- foreach (var episodeToRemove in episodesToRemove)
|
|
|
- {
|
|
|
- _logger.Info("Removing missing/unaired episode {0} {1}x{2}", episodeToRemove.Series.Name, episodeToRemove.ParentIndexNumber, episodeToRemove.IndexNumber);
|
|
|
-
|
|
|
- await episodeToRemove.Parent.RemoveChild(episodeToRemove, cancellationToken).ConfigureAwait(false);
|
|
|
-
|
|
|
- hasChanges = true;
|
|
|
- }
|
|
|
-
|
|
|
- return hasChanges;
|
|
|
- }
|
|
|
-
|
|
|
- /// <summary>
|
|
|
- /// Removes the obsolete or missing seasons.
|
|
|
- /// </summary>
|
|
|
- /// <param name="series">The series.</param>
|
|
|
- /// <param name="episodeLookup">The episode lookup.</param>
|
|
|
- /// <param name="cancellationToken">The cancellation token.</param>
|
|
|
- /// <returns>Task{System.Boolean}.</returns>
|
|
|
- private async Task<bool> RemoveObsoleteOrMissingSeasons(IEnumerable<Series> series, IEnumerable<Tuple<int, int>> episodeLookup, CancellationToken cancellationToken)
|
|
|
- {
|
|
|
- var existingSeasons = series.SelectMany(s => s.Children.OfType<Season>()).ToList();
|
|
|
-
|
|
|
- var physicalSeasons = existingSeasons
|
|
|
- .Where(i => i.LocationType != LocationType.Virtual)
|
|
|
- .ToList();
|
|
|
-
|
|
|
- var virtualSeasons = existingSeasons
|
|
|
- .Where(i => i.LocationType == LocationType.Virtual)
|
|
|
- .ToList();
|
|
|
-
|
|
|
- var seasonsToRemove = virtualSeasons
|
|
|
- .Where(i =>
|
|
|
- {
|
|
|
- if (i.IndexNumber.HasValue)
|
|
|
- {
|
|
|
- var seasonNumber = i.IndexNumber.Value;
|
|
|
-
|
|
|
- // If there's a physical season with the same number, delete it
|
|
|
- if (physicalSeasons.Any(p => p.IndexNumber.HasValue && p.IndexNumber.Value == seasonNumber))
|
|
|
- {
|
|
|
- return true;
|
|
|
- }
|
|
|
-
|
|
|
- // If the season no longer exists in the remote lookup, delete it
|
|
|
- if (episodeLookup.All(e => e.Item1 != seasonNumber))
|
|
|
- {
|
|
|
- return true;
|
|
|
- }
|
|
|
-
|
|
|
- return false;
|
|
|
- }
|
|
|
-
|
|
|
- return true;
|
|
|
- })
|
|
|
- .ToList();
|
|
|
-
|
|
|
- var hasChanges = false;
|
|
|
-
|
|
|
- foreach (var seasonToRemove in seasonsToRemove)
|
|
|
- {
|
|
|
- _logger.Info("Removing virtual season {0} {1}", seasonToRemove.Series.Name, seasonToRemove.IndexNumber);
|
|
|
-
|
|
|
- await seasonToRemove.Parent.RemoveChild(seasonToRemove, cancellationToken).ConfigureAwait(false);
|
|
|
-
|
|
|
- hasChanges = true;
|
|
|
- }
|
|
|
-
|
|
|
- return hasChanges;
|
|
|
- }
|
|
|
-
|
|
|
- /// <summary>
|
|
|
- /// Adds the episode.
|
|
|
- /// </summary>
|
|
|
- /// <param name="series">The series.</param>
|
|
|
- /// <param name="seasonNumber">The season number.</param>
|
|
|
- /// <param name="episodeNumber">The episode number.</param>
|
|
|
- /// <param name="cancellationToken">The cancellation token.</param>
|
|
|
- /// <returns>Task.</returns>
|
|
|
- private async Task AddEpisode(Series series, int seasonNumber, int episodeNumber, CancellationToken cancellationToken)
|
|
|
- {
|
|
|
- var season = series.Children.OfType<Season>()
|
|
|
- .FirstOrDefault(i => i.IndexNumber.HasValue && i.IndexNumber.Value == seasonNumber);
|
|
|
-
|
|
|
- if (season == null)
|
|
|
- {
|
|
|
- season = await AddSeason(series, seasonNumber, cancellationToken).ConfigureAwait(false);
|
|
|
- }
|
|
|
-
|
|
|
- var name = string.Format("Episode {0}", episodeNumber.ToString(UsCulture));
|
|
|
-
|
|
|
- var episode = new Episode
|
|
|
- {
|
|
|
- Name = name,
|
|
|
- IndexNumber = episodeNumber,
|
|
|
- ParentIndexNumber = seasonNumber,
|
|
|
- Parent = season,
|
|
|
- DisplayMediaType = typeof(Episode).Name,
|
|
|
- Id = (series.Id + seasonNumber.ToString(UsCulture) + name).GetMBId(typeof(Episode))
|
|
|
- };
|
|
|
-
|
|
|
- await season.AddChild(episode, cancellationToken).ConfigureAwait(false);
|
|
|
-
|
|
|
- await episode.RefreshMetadata(new MetadataRefreshOptions
|
|
|
- {
|
|
|
- }, cancellationToken).ConfigureAwait(false);
|
|
|
- }
|
|
|
-
|
|
|
- /// <summary>
|
|
|
- /// Adds the season.
|
|
|
- /// </summary>
|
|
|
- /// <param name="series">The series.</param>
|
|
|
- /// <param name="seasonNumber">The season number.</param>
|
|
|
- /// <param name="cancellationToken">The cancellation token.</param>
|
|
|
- /// <returns>Task{Season}.</returns>
|
|
|
- private async Task<Season> AddSeason(Series series, int seasonNumber, CancellationToken cancellationToken)
|
|
|
- {
|
|
|
- _logger.Info("Creating Season {0} entry for {1}", seasonNumber, series.Name);
|
|
|
-
|
|
|
- var name = seasonNumber == 0 ? _config.Configuration.SeasonZeroDisplayName : string.Format("Season {0}", seasonNumber.ToString(UsCulture));
|
|
|
-
|
|
|
- var season = new Season
|
|
|
- {
|
|
|
- Name = name,
|
|
|
- IndexNumber = seasonNumber,
|
|
|
- Parent = series,
|
|
|
- DisplayMediaType = typeof(Season).Name,
|
|
|
- Id = (series.Id + seasonNumber.ToString(UsCulture) + name).GetMBId(typeof(Season))
|
|
|
- };
|
|
|
-
|
|
|
- await series.AddChild(season, cancellationToken).ConfigureAwait(false);
|
|
|
-
|
|
|
- await season.RefreshMetadata(new MetadataRefreshOptions
|
|
|
- {
|
|
|
- }, cancellationToken).ConfigureAwait(false);
|
|
|
-
|
|
|
- return season;
|
|
|
- }
|
|
|
-
|
|
|
- /// <summary>
|
|
|
- /// Gets the existing episode.
|
|
|
- /// </summary>
|
|
|
- /// <param name="existingEpisodes">The existing episodes.</param>
|
|
|
- /// <param name="tuple">The tuple.</param>
|
|
|
- /// <returns>Episode.</returns>
|
|
|
- private Episode GetExistingEpisode(IEnumerable<Episode> existingEpisodes, Tuple<int, int> tuple)
|
|
|
- {
|
|
|
- return existingEpisodes
|
|
|
- .FirstOrDefault(i => (i.ParentIndexNumber ?? -1) == tuple.Item1 && i.ContainsEpisodeNumber(tuple.Item2));
|
|
|
- }
|
|
|
-
|
|
|
- /// <summary>
|
|
|
- /// Gets the air date.
|
|
|
- /// </summary>
|
|
|
- /// <param name="seriesDataPath">The series data path.</param>
|
|
|
- /// <param name="seasonNumber">The season number.</param>
|
|
|
- /// <param name="episodeNumber">The episode number.</param>
|
|
|
- /// <returns>System.Nullable{DateTime}.</returns>
|
|
|
- private DateTime? GetAirDate(string seriesDataPath, int seasonNumber, int episodeNumber)
|
|
|
- {
|
|
|
- // First open up the tvdb xml file and make sure it has valid data
|
|
|
- var filename = string.Format("episode-{0}-{1}.xml", seasonNumber.ToString(UsCulture), episodeNumber.ToString(UsCulture));
|
|
|
-
|
|
|
- var xmlPath = Path.Combine(seriesDataPath, filename);
|
|
|
-
|
|
|
- DateTime? airDate = null;
|
|
|
-
|
|
|
- // It appears the best way to filter out invalid entries is to only include those with valid air dates
|
|
|
- using (var streamReader = new StreamReader(xmlPath, Encoding.UTF8))
|
|
|
- {
|
|
|
- // Use XmlReader for best performance
|
|
|
- using (var reader = XmlReader.Create(streamReader, new XmlReaderSettings
|
|
|
- {
|
|
|
- CheckCharacters = false,
|
|
|
- IgnoreProcessingInstructions = true,
|
|
|
- IgnoreComments = true,
|
|
|
- ValidationType = ValidationType.None
|
|
|
- }))
|
|
|
- {
|
|
|
- reader.MoveToContent();
|
|
|
-
|
|
|
- // Loop through each element
|
|
|
- while (reader.Read())
|
|
|
- {
|
|
|
- if (reader.NodeType == XmlNodeType.Element)
|
|
|
- {
|
|
|
- switch (reader.Name)
|
|
|
- {
|
|
|
- case "EpisodeName":
|
|
|
- {
|
|
|
- var val = reader.ReadElementContentAsString();
|
|
|
- if (string.IsNullOrWhiteSpace(val))
|
|
|
- {
|
|
|
- // Not valid, ignore these
|
|
|
- return null;
|
|
|
- }
|
|
|
- break;
|
|
|
- }
|
|
|
- case "FirstAired":
|
|
|
- {
|
|
|
- var val = reader.ReadElementContentAsString();
|
|
|
-
|
|
|
- if (!string.IsNullOrWhiteSpace(val))
|
|
|
- {
|
|
|
- DateTime date;
|
|
|
- if (DateTime.TryParse(val, out date))
|
|
|
- {
|
|
|
- airDate = date.ToUniversalTime();
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- break;
|
|
|
- }
|
|
|
-
|
|
|
- default:
|
|
|
- reader.Skip();
|
|
|
- break;
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- return airDate;
|
|
|
- }
|
|
|
- }
|
|
|
}
|