EpisodeResolver.cs 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. using System;
  2. using System.Linq;
  3. using MediaBrowser.Controller.Entities.TV;
  4. using MediaBrowser.Controller.Library;
  5. using MediaBrowser.Model.Entities;
  6. using MediaBrowser.Model.IO;
  7. namespace Emby.Server.Implementations.Library.Resolvers.TV
  8. {
  9. /// <summary>
  10. /// Class EpisodeResolver
  11. /// </summary>
  12. public class EpisodeResolver : BaseVideoResolver<Episode>
  13. {
  14. /// <summary>
  15. /// Resolves the specified args.
  16. /// </summary>
  17. /// <param name="args">The args.</param>
  18. /// <returns>Episode.</returns>
  19. protected override Episode Resolve(ItemResolveArgs args)
  20. {
  21. var parent = args.Parent;
  22. if (parent == null)
  23. {
  24. return null;
  25. }
  26. var season = parent as Season;
  27. // Just in case the user decided to nest episodes.
  28. // Not officially supported but in some cases we can handle it.
  29. if (season == null)
  30. {
  31. season = parent.GetParents().OfType<Season>().FirstOrDefault();
  32. }
  33. // If the parent is a Season or Series, then this is an Episode if the VideoResolver returns something
  34. // Also handle flat tv folders
  35. if (season != null ||
  36. string.Equals(args.GetCollectionType(), CollectionType.TvShows, StringComparison.OrdinalIgnoreCase) ||
  37. args.HasParent<Series>())
  38. {
  39. var episode = ResolveVideo<Episode>(args, false);
  40. if (episode != null)
  41. {
  42. var series = parent as Series;
  43. if (series == null)
  44. {
  45. series = parent.GetParents().OfType<Series>().FirstOrDefault();
  46. }
  47. if (series != null)
  48. {
  49. episode.SeriesId = series.Id;
  50. episode.SeriesName = series.Name;
  51. }
  52. if (season != null)
  53. {
  54. episode.SeasonId = season.Id;
  55. episode.SeasonName = season.Name;
  56. }
  57. // Assume season 1 if there's no season folder and a season number could not be determined
  58. if (season == null && !episode.ParentIndexNumber.HasValue && (episode.IndexNumber.HasValue || episode.PremiereDate.HasValue))
  59. {
  60. episode.ParentIndexNumber = 1;
  61. }
  62. }
  63. return episode;
  64. }
  65. return null;
  66. }
  67. public EpisodeResolver(ILibraryManager libraryManager, IFileSystem fileSystem) : base(libraryManager, fileSystem)
  68. {
  69. }
  70. }
  71. }