EpisodeFileOrganizer.cs 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839
  1. using MediaBrowser.Controller.Configuration;
  2. using MediaBrowser.Controller.Entities.TV;
  3. using MediaBrowser.Controller.FileOrganization;
  4. using MediaBrowser.Controller.Library;
  5. using MediaBrowser.Controller.Providers;
  6. using MediaBrowser.Model.Entities;
  7. using MediaBrowser.Model.Extensions;
  8. using MediaBrowser.Model.FileOrganization;
  9. using MediaBrowser.Model.Logging;
  10. using System;
  11. using System.Collections.Generic;
  12. using System.Globalization;
  13. using System.IO;
  14. using System.Linq;
  15. using System.Threading;
  16. using System.Threading.Tasks;
  17. using Emby.Server.Implementations.Library;
  18. using MediaBrowser.Common.IO;
  19. using MediaBrowser.Controller.Entities;
  20. using MediaBrowser.Controller.IO;
  21. using MediaBrowser.Model.IO;
  22. using MediaBrowser.Naming.TV;
  23. using EpisodeInfo = MediaBrowser.Controller.Providers.EpisodeInfo;
  24. namespace Emby.Server.Implementations.FileOrganization
  25. {
  26. public class EpisodeFileOrganizer
  27. {
  28. private readonly ILibraryMonitor _libraryMonitor;
  29. private readonly ILibraryManager _libraryManager;
  30. private readonly ILogger _logger;
  31. private readonly IFileSystem _fileSystem;
  32. private readonly IFileOrganizationService _organizationService;
  33. private readonly IServerConfigurationManager _config;
  34. private readonly IProviderManager _providerManager;
  35. private readonly CultureInfo _usCulture = new CultureInfo("en-US");
  36. public EpisodeFileOrganizer(IFileOrganizationService organizationService, IServerConfigurationManager config, IFileSystem fileSystem, ILogger logger, ILibraryManager libraryManager, ILibraryMonitor libraryMonitor, IProviderManager providerManager)
  37. {
  38. _organizationService = organizationService;
  39. _config = config;
  40. _fileSystem = fileSystem;
  41. _logger = logger;
  42. _libraryManager = libraryManager;
  43. _libraryMonitor = libraryMonitor;
  44. _providerManager = providerManager;
  45. }
  46. public async Task<FileOrganizationResult> OrganizeEpisodeFile(string path, AutoOrganizeOptions options, bool overwriteExisting, CancellationToken cancellationToken)
  47. {
  48. _logger.Info("Sorting file {0}", path);
  49. var result = new FileOrganizationResult
  50. {
  51. Date = DateTime.UtcNow,
  52. OriginalPath = path,
  53. OriginalFileName = Path.GetFileName(path),
  54. Type = FileOrganizerType.Episode,
  55. FileSize = _fileSystem.GetFileInfo(path).Length
  56. };
  57. try
  58. {
  59. if (_libraryMonitor.IsPathLocked(path))
  60. {
  61. result.Status = FileSortingStatus.Failure;
  62. result.StatusMessage = "Path is locked by other processes. Please try again later.";
  63. return result;
  64. }
  65. var namingOptions = ((LibraryManager)_libraryManager).GetNamingOptions();
  66. var resolver = new EpisodeResolver(namingOptions, new NullLogger());
  67. var episodeInfo = resolver.Resolve(path, false) ??
  68. new MediaBrowser.Naming.TV.EpisodeInfo();
  69. var seriesName = episodeInfo.SeriesName;
  70. if (!string.IsNullOrEmpty(seriesName))
  71. {
  72. var seasonNumber = episodeInfo.SeasonNumber;
  73. result.ExtractedSeasonNumber = seasonNumber;
  74. // Passing in true will include a few extra regex's
  75. var episodeNumber = episodeInfo.EpisodeNumber;
  76. result.ExtractedEpisodeNumber = episodeNumber;
  77. var premiereDate = episodeInfo.IsByDate ?
  78. new DateTime(episodeInfo.Year.Value, episodeInfo.Month.Value, episodeInfo.Day.Value) :
  79. (DateTime?)null;
  80. if (episodeInfo.IsByDate || (seasonNumber.HasValue && episodeNumber.HasValue))
  81. {
  82. if (episodeInfo.IsByDate)
  83. {
  84. _logger.Debug("Extracted information from {0}. Series name {1}, Date {2}", path, seriesName, premiereDate.Value);
  85. }
  86. else
  87. {
  88. _logger.Debug("Extracted information from {0}. Series name {1}, Season {2}, Episode {3}", path, seriesName, seasonNumber, episodeNumber);
  89. }
  90. var endingEpisodeNumber = episodeInfo.EndingEpsiodeNumber;
  91. result.ExtractedEndingEpisodeNumber = endingEpisodeNumber;
  92. await OrganizeEpisode(path,
  93. seriesName,
  94. seasonNumber,
  95. episodeNumber,
  96. endingEpisodeNumber,
  97. premiereDate,
  98. options,
  99. overwriteExisting,
  100. false,
  101. result,
  102. cancellationToken).ConfigureAwait(false);
  103. }
  104. else
  105. {
  106. var msg = string.Format("Unable to determine episode number from {0}", path);
  107. result.Status = FileSortingStatus.Failure;
  108. result.StatusMessage = msg;
  109. _logger.Warn(msg);
  110. }
  111. }
  112. else
  113. {
  114. var msg = string.Format("Unable to determine series name from {0}", path);
  115. result.Status = FileSortingStatus.Failure;
  116. result.StatusMessage = msg;
  117. _logger.Warn(msg);
  118. }
  119. var previousResult = _organizationService.GetResultBySourcePath(path);
  120. if (previousResult != null)
  121. {
  122. // Don't keep saving the same result over and over if nothing has changed
  123. if (previousResult.Status == result.Status && previousResult.StatusMessage == result.StatusMessage && result.Status != FileSortingStatus.Success)
  124. {
  125. return previousResult;
  126. }
  127. }
  128. await _organizationService.SaveResult(result, CancellationToken.None).ConfigureAwait(false);
  129. }
  130. catch (Exception ex)
  131. {
  132. result.Status = FileSortingStatus.Failure;
  133. result.StatusMessage = ex.Message;
  134. }
  135. return result;
  136. }
  137. public async Task<FileOrganizationResult> OrganizeWithCorrection(EpisodeFileOrganizationRequest request, AutoOrganizeOptions options, CancellationToken cancellationToken)
  138. {
  139. var result = _organizationService.GetResult(request.ResultId);
  140. try
  141. {
  142. Series series = null;
  143. if (request.NewSeriesProviderIds.Count > 0)
  144. {
  145. // We're having a new series here
  146. SeriesInfo seriesRequest = new SeriesInfo();
  147. seriesRequest.ProviderIds = request.NewSeriesProviderIds;
  148. var refreshOptions = new MetadataRefreshOptions(_fileSystem);
  149. series = new Series();
  150. series.Id = Guid.NewGuid();
  151. series.Name = request.NewSeriesName;
  152. int year;
  153. if (int.TryParse(request.NewSeriesYear, out year))
  154. {
  155. series.ProductionYear = year;
  156. }
  157. var seriesFolderName = series.Name;
  158. if (series.ProductionYear.HasValue)
  159. {
  160. seriesFolderName = string.Format("{0} ({1})", seriesFolderName, series.ProductionYear);
  161. }
  162. series.Path = Path.Combine(request.TargetFolder, seriesFolderName);
  163. series.ProviderIds = request.NewSeriesProviderIds;
  164. await series.RefreshMetadata(refreshOptions, cancellationToken);
  165. }
  166. if (series == null)
  167. {
  168. // Existing Series
  169. series = (Series)_libraryManager.GetItemById(new Guid(request.SeriesId));
  170. }
  171. await OrganizeEpisode(result.OriginalPath,
  172. series,
  173. request.SeasonNumber,
  174. request.EpisodeNumber,
  175. request.EndingEpisodeNumber,
  176. null,
  177. options,
  178. true,
  179. request.RememberCorrection,
  180. result,
  181. cancellationToken).ConfigureAwait(false);
  182. await _organizationService.SaveResult(result, CancellationToken.None).ConfigureAwait(false);
  183. }
  184. catch (Exception ex)
  185. {
  186. result.Status = FileSortingStatus.Failure;
  187. result.StatusMessage = ex.Message;
  188. }
  189. return result;
  190. }
  191. private Task OrganizeEpisode(string sourcePath,
  192. string seriesName,
  193. int? seasonNumber,
  194. int? episodeNumber,
  195. int? endingEpiosdeNumber,
  196. DateTime? premiereDate,
  197. AutoOrganizeOptions options,
  198. bool overwriteExisting,
  199. bool rememberCorrection,
  200. FileOrganizationResult result,
  201. CancellationToken cancellationToken)
  202. {
  203. var series = GetMatchingSeries(seriesName, result, options);
  204. if (series == null)
  205. {
  206. var msg = string.Format("Unable to find series in library matching name {0}", seriesName);
  207. result.Status = FileSortingStatus.Failure;
  208. result.StatusMessage = msg;
  209. _logger.Warn(msg);
  210. return Task.FromResult(true);
  211. }
  212. return OrganizeEpisode(sourcePath,
  213. series,
  214. seasonNumber,
  215. episodeNumber,
  216. endingEpiosdeNumber,
  217. premiereDate,
  218. options,
  219. overwriteExisting,
  220. rememberCorrection,
  221. result,
  222. cancellationToken);
  223. }
  224. private async Task OrganizeEpisode(string sourcePath,
  225. Series series,
  226. int? seasonNumber,
  227. int? episodeNumber,
  228. int? endingEpiosdeNumber,
  229. DateTime? premiereDate,
  230. AutoOrganizeOptions options,
  231. bool overwriteExisting,
  232. bool rememberCorrection,
  233. FileOrganizationResult result,
  234. CancellationToken cancellationToken)
  235. {
  236. _logger.Info("Sorting file {0} into series {1}", sourcePath, series.Path);
  237. var originalExtractedSeriesString = result.ExtractedName;
  238. bool isNew = string.IsNullOrWhiteSpace(result.Id);
  239. if (isNew)
  240. {
  241. await _organizationService.SaveResult(result, cancellationToken);
  242. }
  243. if (!_organizationService.AddToInProgressList(result, isNew))
  244. {
  245. throw new Exception("File is currently processed otherwise. Please try again later.");
  246. }
  247. try
  248. {
  249. // Proceed to sort the file
  250. var newPath = await GetNewPath(sourcePath, series, seasonNumber, episodeNumber, endingEpiosdeNumber, premiereDate, options.TvOptions, cancellationToken).ConfigureAwait(false);
  251. if (string.IsNullOrEmpty(newPath))
  252. {
  253. var msg = string.Format("Unable to sort {0} because target path could not be determined.", sourcePath);
  254. throw new Exception(msg);
  255. }
  256. _logger.Info("Sorting file {0} to new path {1}", sourcePath, newPath);
  257. result.TargetPath = newPath;
  258. var fileExists = _fileSystem.FileExists(result.TargetPath);
  259. var otherDuplicatePaths = GetOtherDuplicatePaths(result.TargetPath, series, seasonNumber, episodeNumber, endingEpiosdeNumber);
  260. if (!overwriteExisting)
  261. {
  262. if (options.TvOptions.CopyOriginalFile && fileExists && IsSameEpisode(sourcePath, newPath))
  263. {
  264. var msg = string.Format("File '{0}' already copied to new path '{1}', stopping organization", sourcePath, newPath);
  265. _logger.Info(msg);
  266. result.Status = FileSortingStatus.SkippedExisting;
  267. result.StatusMessage = msg;
  268. return;
  269. }
  270. if (fileExists)
  271. {
  272. var msg = string.Format("File '{0}' already exists as '{1}', stopping organization", sourcePath, newPath);
  273. _logger.Info(msg);
  274. result.Status = FileSortingStatus.SkippedExisting;
  275. result.StatusMessage = msg;
  276. result.TargetPath = newPath;
  277. return;
  278. }
  279. if (otherDuplicatePaths.Count > 0)
  280. {
  281. var msg = string.Format("File '{0}' already exists as these:'{1}'. Stopping organization", sourcePath, string.Join("', '", otherDuplicatePaths));
  282. _logger.Info(msg);
  283. result.Status = FileSortingStatus.SkippedExisting;
  284. result.StatusMessage = msg;
  285. result.DuplicatePaths = otherDuplicatePaths;
  286. return;
  287. }
  288. }
  289. PerformFileSorting(options.TvOptions, result);
  290. if (overwriteExisting)
  291. {
  292. var hasRenamedFiles = false;
  293. foreach (var path in otherDuplicatePaths)
  294. {
  295. _logger.Debug("Removing duplicate episode {0}", path);
  296. _libraryMonitor.ReportFileSystemChangeBeginning(path);
  297. var renameRelatedFiles = !hasRenamedFiles &&
  298. string.Equals(Path.GetDirectoryName(path), Path.GetDirectoryName(result.TargetPath), StringComparison.OrdinalIgnoreCase);
  299. if (renameRelatedFiles)
  300. {
  301. hasRenamedFiles = true;
  302. }
  303. try
  304. {
  305. DeleteLibraryFile(path, renameRelatedFiles, result.TargetPath);
  306. }
  307. catch (IOException ex)
  308. {
  309. _logger.ErrorException("Error removing duplicate episode", ex, path);
  310. }
  311. finally
  312. {
  313. _libraryMonitor.ReportFileSystemChangeComplete(path, true);
  314. }
  315. }
  316. }
  317. }
  318. catch (Exception ex)
  319. {
  320. result.Status = FileSortingStatus.Failure;
  321. result.StatusMessage = ex.Message;
  322. _logger.Warn(ex.Message);
  323. return;
  324. }
  325. finally
  326. {
  327. _organizationService.RemoveFromInprogressList(result);
  328. }
  329. if (rememberCorrection)
  330. {
  331. SaveSmartMatchString(originalExtractedSeriesString, series, options);
  332. }
  333. }
  334. private void SaveSmartMatchString(string matchString, Series series, AutoOrganizeOptions options)
  335. {
  336. if (string.IsNullOrEmpty(matchString) || matchString.Length < 3)
  337. {
  338. return;
  339. }
  340. SmartMatchInfo info = options.SmartMatchInfos.FirstOrDefault(i => string.Equals(i.ItemName, series.Name, StringComparison.OrdinalIgnoreCase));
  341. if (info == null)
  342. {
  343. info = new SmartMatchInfo();
  344. info.ItemName = series.Name;
  345. info.OrganizerType = FileOrganizerType.Episode;
  346. info.DisplayName = series.Name;
  347. var list = options.SmartMatchInfos.ToList();
  348. list.Add(info);
  349. options.SmartMatchInfos = list.ToArray();
  350. }
  351. if (!info.MatchStrings.Contains(matchString, StringComparer.OrdinalIgnoreCase))
  352. {
  353. var list = info.MatchStrings.ToList();
  354. list.Add(matchString);
  355. info.MatchStrings = list.ToArray();
  356. _config.SaveAutoOrganizeOptions(options);
  357. }
  358. }
  359. private void DeleteLibraryFile(string path, bool renameRelatedFiles, string targetPath)
  360. {
  361. _fileSystem.DeleteFile(path);
  362. if (!renameRelatedFiles)
  363. {
  364. return;
  365. }
  366. // Now find other files
  367. var originalFilenameWithoutExtension = Path.GetFileNameWithoutExtension(path);
  368. var directory = Path.GetDirectoryName(path);
  369. if (!string.IsNullOrWhiteSpace(originalFilenameWithoutExtension) && !string.IsNullOrWhiteSpace(directory))
  370. {
  371. // Get all related files, e.g. metadata, images, etc
  372. var files = _fileSystem.GetFilePaths(directory)
  373. .Where(i => (Path.GetFileNameWithoutExtension(i) ?? string.Empty).StartsWith(originalFilenameWithoutExtension, StringComparison.OrdinalIgnoreCase))
  374. .ToList();
  375. var targetFilenameWithoutExtension = Path.GetFileNameWithoutExtension(targetPath);
  376. foreach (var file in files)
  377. {
  378. directory = Path.GetDirectoryName(file);
  379. var filename = Path.GetFileName(file);
  380. filename = filename.Replace(originalFilenameWithoutExtension, targetFilenameWithoutExtension,
  381. StringComparison.OrdinalIgnoreCase);
  382. var destination = Path.Combine(directory, filename);
  383. _fileSystem.MoveFile(file, destination);
  384. }
  385. }
  386. }
  387. private List<string> GetOtherDuplicatePaths(string targetPath,
  388. Series series,
  389. int? seasonNumber,
  390. int? episodeNumber,
  391. int? endingEpisodeNumber)
  392. {
  393. // TODO: Support date-naming?
  394. if (!seasonNumber.HasValue || !episodeNumber.HasValue)
  395. {
  396. return new List<string>();
  397. }
  398. var episodePaths = series.GetRecursiveChildren()
  399. .OfType<Episode>()
  400. .Where(i =>
  401. {
  402. var locationType = i.LocationType;
  403. // Must be file system based and match exactly
  404. if (locationType != LocationType.Remote &&
  405. locationType != LocationType.Virtual &&
  406. i.ParentIndexNumber.HasValue &&
  407. i.ParentIndexNumber.Value == seasonNumber &&
  408. i.IndexNumber.HasValue &&
  409. i.IndexNumber.Value == episodeNumber)
  410. {
  411. if (endingEpisodeNumber.HasValue || i.IndexNumberEnd.HasValue)
  412. {
  413. return endingEpisodeNumber.HasValue && i.IndexNumberEnd.HasValue &&
  414. endingEpisodeNumber.Value == i.IndexNumberEnd.Value;
  415. }
  416. return true;
  417. }
  418. return false;
  419. })
  420. .Select(i => i.Path)
  421. .ToList();
  422. var folder = Path.GetDirectoryName(targetPath);
  423. var targetFileNameWithoutExtension = _fileSystem.GetFileNameWithoutExtension(targetPath);
  424. try
  425. {
  426. var filesOfOtherExtensions = _fileSystem.GetFilePaths(folder)
  427. .Where(i => _libraryManager.IsVideoFile(i) && string.Equals(_fileSystem.GetFileNameWithoutExtension(i), targetFileNameWithoutExtension, StringComparison.OrdinalIgnoreCase));
  428. episodePaths.AddRange(filesOfOtherExtensions);
  429. }
  430. catch (IOException)
  431. {
  432. // No big deal. Maybe the season folder doesn't already exist.
  433. }
  434. return episodePaths.Where(i => !string.Equals(i, targetPath, StringComparison.OrdinalIgnoreCase))
  435. .Distinct(StringComparer.OrdinalIgnoreCase)
  436. .ToList();
  437. }
  438. private void PerformFileSorting(TvFileOrganizationOptions options, FileOrganizationResult result)
  439. {
  440. // We should probably handle this earlier so that we never even make it this far
  441. if (string.Equals(result.OriginalPath, result.TargetPath, StringComparison.OrdinalIgnoreCase))
  442. {
  443. return;
  444. }
  445. _libraryMonitor.ReportFileSystemChangeBeginning(result.TargetPath);
  446. _fileSystem.CreateDirectory(Path.GetDirectoryName(result.TargetPath));
  447. var targetAlreadyExists = _fileSystem.FileExists(result.TargetPath);
  448. try
  449. {
  450. if (targetAlreadyExists || options.CopyOriginalFile)
  451. {
  452. _fileSystem.CopyFile(result.OriginalPath, result.TargetPath, true);
  453. }
  454. else
  455. {
  456. _fileSystem.MoveFile(result.OriginalPath, result.TargetPath);
  457. }
  458. result.Status = FileSortingStatus.Success;
  459. result.StatusMessage = string.Empty;
  460. }
  461. catch (Exception ex)
  462. {
  463. var errorMsg = string.Format("Failed to move file from {0} to {1}: {2}", result.OriginalPath, result.TargetPath, ex.Message);
  464. result.Status = FileSortingStatus.Failure;
  465. result.StatusMessage = errorMsg;
  466. _logger.ErrorException(errorMsg, ex);
  467. return;
  468. }
  469. finally
  470. {
  471. _libraryMonitor.ReportFileSystemChangeComplete(result.TargetPath, true);
  472. }
  473. if (targetAlreadyExists && !options.CopyOriginalFile)
  474. {
  475. try
  476. {
  477. _fileSystem.DeleteFile(result.OriginalPath);
  478. }
  479. catch (Exception ex)
  480. {
  481. _logger.ErrorException("Error deleting {0}", ex, result.OriginalPath);
  482. }
  483. }
  484. }
  485. private Series GetMatchingSeries(string seriesName, FileOrganizationResult result, AutoOrganizeOptions options)
  486. {
  487. var parsedName = _libraryManager.ParseName(seriesName);
  488. var yearInName = parsedName.Year;
  489. var nameWithoutYear = parsedName.Name;
  490. result.ExtractedName = nameWithoutYear;
  491. result.ExtractedYear = yearInName;
  492. var series = _libraryManager.GetItemList(new InternalItemsQuery
  493. {
  494. IncludeItemTypes = new[] { typeof(Series).Name },
  495. Recursive = true
  496. })
  497. .Cast<Series>()
  498. .Select(i => NameUtils.GetMatchScore(nameWithoutYear, yearInName, i))
  499. .Where(i => i.Item2 > 0)
  500. .OrderByDescending(i => i.Item2)
  501. .Select(i => i.Item1)
  502. .FirstOrDefault();
  503. if (series == null)
  504. {
  505. SmartMatchInfo info = options.SmartMatchInfos.FirstOrDefault(e => e.MatchStrings.Contains(nameWithoutYear, StringComparer.OrdinalIgnoreCase));
  506. if (info != null)
  507. {
  508. series = _libraryManager.GetItemList(new InternalItemsQuery
  509. {
  510. IncludeItemTypes = new[] { typeof(Series).Name },
  511. Recursive = true,
  512. Name = info.ItemName
  513. }).Cast<Series>().FirstOrDefault();
  514. }
  515. }
  516. return series;
  517. }
  518. /// <summary>
  519. /// Gets the new path.
  520. /// </summary>
  521. /// <param name="sourcePath">The source path.</param>
  522. /// <param name="series">The series.</param>
  523. /// <param name="seasonNumber">The season number.</param>
  524. /// <param name="episodeNumber">The episode number.</param>
  525. /// <param name="endingEpisodeNumber">The ending episode number.</param>
  526. /// <param name="premiereDate">The premiere date.</param>
  527. /// <param name="options">The options.</param>
  528. /// <param name="cancellationToken">The cancellation token.</param>
  529. /// <returns>System.String.</returns>
  530. private async Task<string> GetNewPath(string sourcePath,
  531. Series series,
  532. int? seasonNumber,
  533. int? episodeNumber,
  534. int? endingEpisodeNumber,
  535. DateTime? premiereDate,
  536. TvFileOrganizationOptions options,
  537. CancellationToken cancellationToken)
  538. {
  539. var episodeInfo = new EpisodeInfo
  540. {
  541. IndexNumber = episodeNumber,
  542. IndexNumberEnd = endingEpisodeNumber,
  543. MetadataCountryCode = series.GetPreferredMetadataCountryCode(),
  544. MetadataLanguage = series.GetPreferredMetadataLanguage(),
  545. ParentIndexNumber = seasonNumber,
  546. SeriesProviderIds = series.ProviderIds,
  547. PremiereDate = premiereDate
  548. };
  549. var searchResults = await _providerManager.GetRemoteSearchResults<Episode, EpisodeInfo>(new RemoteSearchQuery<EpisodeInfo>
  550. {
  551. SearchInfo = episodeInfo
  552. }, cancellationToken).ConfigureAwait(false);
  553. var episode = searchResults.FirstOrDefault();
  554. if (episode == null)
  555. {
  556. var msg = string.Format("No provider metadata found for {0} season {1} episode {2}", series.Name, seasonNumber, episodeNumber);
  557. _logger.Warn(msg);
  558. throw new Exception(msg);
  559. }
  560. var episodeName = episode.Name;
  561. //if (string.IsNullOrWhiteSpace(episodeName))
  562. //{
  563. // var msg = string.Format("No provider metadata found for {0} season {1} episode {2}", series.Name, seasonNumber, episodeNumber);
  564. // _logger.Warn(msg);
  565. // return null;
  566. //}
  567. seasonNumber = seasonNumber ?? episode.ParentIndexNumber;
  568. episodeNumber = episodeNumber ?? episode.IndexNumber;
  569. var newPath = GetSeasonFolderPath(series, seasonNumber.Value, options);
  570. // MAX_PATH - trailing <NULL> charachter - drive component: 260 - 1 - 3 = 256
  571. // Usually newPath would include the drive component, but use 256 to be sure
  572. var maxFilenameLength = 256 - newPath.Length;
  573. if (!newPath.EndsWith(@"\"))
  574. {
  575. // Remove 1 for missing backslash combining path and filename
  576. maxFilenameLength--;
  577. }
  578. // Remove additional 4 chars to prevent PathTooLongException for downloaded subtitles (eg. filename.ext.eng.srt)
  579. maxFilenameLength -= 4;
  580. var episodeFileName = GetEpisodeFileName(sourcePath, series.Name, seasonNumber.Value, episodeNumber.Value, endingEpisodeNumber, episodeName, options, maxFilenameLength);
  581. if (string.IsNullOrEmpty(episodeFileName))
  582. {
  583. // cause failure
  584. return string.Empty;
  585. }
  586. newPath = Path.Combine(newPath, episodeFileName);
  587. return newPath;
  588. }
  589. /// <summary>
  590. /// Gets the season folder path.
  591. /// </summary>
  592. /// <param name="series">The series.</param>
  593. /// <param name="seasonNumber">The season number.</param>
  594. /// <param name="options">The options.</param>
  595. /// <returns>System.String.</returns>
  596. private string GetSeasonFolderPath(Series series, int seasonNumber, TvFileOrganizationOptions options)
  597. {
  598. // If there's already a season folder, use that
  599. var season = series
  600. .GetRecursiveChildren(i => i is Season && i.LocationType == LocationType.FileSystem && i.IndexNumber.HasValue && i.IndexNumber.Value == seasonNumber)
  601. .FirstOrDefault();
  602. if (season != null)
  603. {
  604. return season.Path;
  605. }
  606. var path = series.Path;
  607. if (series.ContainsEpisodesWithoutSeasonFolders)
  608. {
  609. return path;
  610. }
  611. if (seasonNumber == 0)
  612. {
  613. return Path.Combine(path, _fileSystem.GetValidFilename(options.SeasonZeroFolderName));
  614. }
  615. var seasonFolderName = options.SeasonFolderPattern
  616. .Replace("%s", seasonNumber.ToString(_usCulture))
  617. .Replace("%0s", seasonNumber.ToString("00", _usCulture))
  618. .Replace("%00s", seasonNumber.ToString("000", _usCulture));
  619. return Path.Combine(path, _fileSystem.GetValidFilename(seasonFolderName));
  620. }
  621. private string GetEpisodeFileName(string sourcePath, string seriesName, int seasonNumber, int episodeNumber, int? endingEpisodeNumber, string episodeTitle, TvFileOrganizationOptions options, int? maxLength)
  622. {
  623. seriesName = _fileSystem.GetValidFilename(seriesName).Trim();
  624. if (string.IsNullOrWhiteSpace(episodeTitle))
  625. {
  626. episodeTitle = string.Empty;
  627. }
  628. else
  629. {
  630. episodeTitle = _fileSystem.GetValidFilename(episodeTitle).Trim();
  631. }
  632. var sourceExtension = (Path.GetExtension(sourcePath) ?? string.Empty).TrimStart('.');
  633. var pattern = endingEpisodeNumber.HasValue ? options.MultiEpisodeNamePattern : options.EpisodeNamePattern;
  634. if (string.IsNullOrWhiteSpace(pattern))
  635. {
  636. throw new Exception("GetEpisodeFileName: Configured episode name pattern is empty!");
  637. }
  638. var result = pattern.Replace("%sn", seriesName)
  639. .Replace("%s.n", seriesName.Replace(" ", "."))
  640. .Replace("%s_n", seriesName.Replace(" ", "_"))
  641. .Replace("%s", seasonNumber.ToString(_usCulture))
  642. .Replace("%0s", seasonNumber.ToString("00", _usCulture))
  643. .Replace("%00s", seasonNumber.ToString("000", _usCulture))
  644. .Replace("%ext", sourceExtension)
  645. .Replace("%en", "%#1")
  646. .Replace("%e.n", "%#2")
  647. .Replace("%e_n", "%#3");
  648. if (endingEpisodeNumber.HasValue)
  649. {
  650. result = result.Replace("%ed", endingEpisodeNumber.Value.ToString(_usCulture))
  651. .Replace("%0ed", endingEpisodeNumber.Value.ToString("00", _usCulture))
  652. .Replace("%00ed", endingEpisodeNumber.Value.ToString("000", _usCulture));
  653. }
  654. result = result.Replace("%e", episodeNumber.ToString(_usCulture))
  655. .Replace("%0e", episodeNumber.ToString("00", _usCulture))
  656. .Replace("%00e", episodeNumber.ToString("000", _usCulture));
  657. if (maxLength.HasValue && result.Contains("%#"))
  658. {
  659. // Substract 3 for the temp token length (%#1, %#2 or %#3)
  660. int maxRemainingTitleLength = maxLength.Value - result.Length + 3;
  661. string shortenedEpisodeTitle = string.Empty;
  662. if (maxRemainingTitleLength > 5)
  663. {
  664. // A title with fewer than 5 letters wouldn't be of much value
  665. shortenedEpisodeTitle = episodeTitle.Substring(0, Math.Min(maxRemainingTitleLength, episodeTitle.Length));
  666. }
  667. result = result.Replace("%#1", shortenedEpisodeTitle)
  668. .Replace("%#2", shortenedEpisodeTitle.Replace(" ", "."))
  669. .Replace("%#3", shortenedEpisodeTitle.Replace(" ", "_"));
  670. }
  671. if (maxLength.HasValue && result.Length > maxLength.Value)
  672. {
  673. // There may be cases where reducing the title length may still not be sufficient to
  674. // stay below maxLength
  675. var msg = string.Format("Unable to generate an episode file name shorter than {0} characters to constrain to the max path limit", maxLength);
  676. throw new Exception(msg);
  677. }
  678. return result;
  679. }
  680. private bool IsSameEpisode(string sourcePath, string newPath)
  681. {
  682. try
  683. {
  684. var sourceFileInfo = _fileSystem.GetFileInfo(sourcePath);
  685. var destinationFileInfo = _fileSystem.GetFileInfo(newPath);
  686. if (sourceFileInfo.Length == destinationFileInfo.Length)
  687. {
  688. return true;
  689. }
  690. }
  691. catch (FileNotFoundException)
  692. {
  693. return false;
  694. }
  695. catch (IOException)
  696. {
  697. return false;
  698. }
  699. return false;
  700. }
  701. }
  702. }