SyncJobProcessor.cs 38 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987
  1. using MediaBrowser.Common.Configuration;
  2. using MediaBrowser.Common.IO;
  3. using MediaBrowser.Common.Progress;
  4. using MediaBrowser.Controller.Entities;
  5. using MediaBrowser.Controller.Entities.Audio;
  6. using MediaBrowser.Controller.Entities.TV;
  7. using MediaBrowser.Controller.Library;
  8. using MediaBrowser.Controller.MediaEncoding;
  9. using MediaBrowser.Controller.Sync;
  10. using MediaBrowser.Controller.TV;
  11. using MediaBrowser.Model.Dlna;
  12. using MediaBrowser.Model.Dto;
  13. using MediaBrowser.Model.Entities;
  14. using MediaBrowser.Model.Logging;
  15. using MediaBrowser.Model.MediaInfo;
  16. using MediaBrowser.Model.Querying;
  17. using MediaBrowser.Model.Session;
  18. using MediaBrowser.Model.Sync;
  19. using System;
  20. using System.Collections.Generic;
  21. using System.Globalization;
  22. using System.IO;
  23. using System.Linq;
  24. using System.Threading;
  25. using System.Threading.Tasks;
  26. using CommonIO;
  27. using MediaBrowser.Model.Extensions;
  28. using MediaBrowser.Model.IO;
  29. namespace MediaBrowser.Server.Implementations.Sync
  30. {
  31. public class SyncJobProcessor
  32. {
  33. private readonly ILibraryManager _libraryManager;
  34. private readonly ISyncRepository _syncRepo;
  35. private readonly SyncManager _syncManager;
  36. private readonly ILogger _logger;
  37. private readonly IUserManager _userManager;
  38. private readonly ITVSeriesManager _tvSeriesManager;
  39. private readonly IMediaEncoder _mediaEncoder;
  40. private readonly ISubtitleEncoder _subtitleEncoder;
  41. private readonly IConfigurationManager _config;
  42. private readonly IFileSystem _fileSystem;
  43. private readonly IMediaSourceManager _mediaSourceManager;
  44. public SyncJobProcessor(ILibraryManager libraryManager, ISyncRepository syncRepo, SyncManager syncManager, ILogger logger, IUserManager userManager, ITVSeriesManager tvSeriesManager, IMediaEncoder mediaEncoder, ISubtitleEncoder subtitleEncoder, IConfigurationManager config, IFileSystem fileSystem, IMediaSourceManager mediaSourceManager)
  45. {
  46. _libraryManager = libraryManager;
  47. _syncRepo = syncRepo;
  48. _syncManager = syncManager;
  49. _logger = logger;
  50. _userManager = userManager;
  51. _tvSeriesManager = tvSeriesManager;
  52. _mediaEncoder = mediaEncoder;
  53. _subtitleEncoder = subtitleEncoder;
  54. _config = config;
  55. _fileSystem = fileSystem;
  56. _mediaSourceManager = mediaSourceManager;
  57. }
  58. public async Task EnsureJobItems(SyncJob job)
  59. {
  60. var user = _userManager.GetUserById(job.UserId);
  61. if (user == null)
  62. {
  63. throw new InvalidOperationException("Cannot proceed with sync because user no longer exists.");
  64. }
  65. var items = (await GetItemsForSync(job.Category, job.ParentId, job.RequestedItemIds, user, job.UnwatchedOnly).ConfigureAwait(false))
  66. .ToList();
  67. var jobItems = _syncManager.GetJobItems(new SyncJobItemQuery
  68. {
  69. JobId = job.Id,
  70. AddMetadata = false
  71. }).Items.ToList();
  72. foreach (var item in items)
  73. {
  74. // Respect ItemLimit, if set
  75. if (job.ItemLimit.HasValue)
  76. {
  77. if (jobItems.Count(j => j.Status != SyncJobItemStatus.RemovedFromDevice && j.Status != SyncJobItemStatus.Failed) >= job.ItemLimit.Value)
  78. {
  79. break;
  80. }
  81. }
  82. var itemId = item.Id.ToString("N");
  83. var jobItem = jobItems.FirstOrDefault(i => string.Equals(i.ItemId, itemId, StringComparison.OrdinalIgnoreCase));
  84. if (jobItem != null)
  85. {
  86. continue;
  87. }
  88. var index = jobItems.Count == 0 ?
  89. 0 :
  90. jobItems.Select(i => i.JobItemIndex).Max() + 1;
  91. jobItem = new SyncJobItem
  92. {
  93. Id = Guid.NewGuid().ToString("N"),
  94. ItemId = itemId,
  95. ItemName = GetSyncJobItemName(item),
  96. JobId = job.Id,
  97. TargetId = job.TargetId,
  98. DateCreated = DateTime.UtcNow,
  99. JobItemIndex = index
  100. };
  101. await _syncRepo.Create(jobItem).ConfigureAwait(false);
  102. _syncManager.OnSyncJobItemCreated(jobItem);
  103. jobItems.Add(jobItem);
  104. }
  105. jobItems = jobItems
  106. .OrderBy(i => i.DateCreated)
  107. .ToList();
  108. await UpdateJobStatus(job, jobItems).ConfigureAwait(false);
  109. }
  110. private string GetSyncJobItemName(BaseItem item)
  111. {
  112. var name = item.Name;
  113. var episode = item as Episode;
  114. if (episode != null)
  115. {
  116. if (episode.IndexNumber.HasValue)
  117. {
  118. name = "E" + episode.IndexNumber.Value.ToString(CultureInfo.InvariantCulture) + " - " + name;
  119. }
  120. if (episode.ParentIndexNumber.HasValue)
  121. {
  122. name = "S" + episode.ParentIndexNumber.Value.ToString(CultureInfo.InvariantCulture) + ", " + name;
  123. }
  124. }
  125. return name;
  126. }
  127. public Task UpdateJobStatus(string id)
  128. {
  129. var job = _syncRepo.GetJob(id);
  130. if (job == null)
  131. {
  132. return Task.FromResult(true);
  133. }
  134. var result = _syncManager.GetJobItems(new SyncJobItemQuery
  135. {
  136. JobId = job.Id,
  137. AddMetadata = false
  138. });
  139. return UpdateJobStatus(job, result.Items.ToList());
  140. }
  141. private async Task UpdateJobStatus(SyncJob job, List<SyncJobItem> jobItems)
  142. {
  143. job.ItemCount = jobItems.Count;
  144. double pct = 0;
  145. foreach (var item in jobItems)
  146. {
  147. if (item.Status == SyncJobItemStatus.Failed || item.Status == SyncJobItemStatus.Synced || item.Status == SyncJobItemStatus.RemovedFromDevice || item.Status == SyncJobItemStatus.Cancelled)
  148. {
  149. pct += 100;
  150. }
  151. else
  152. {
  153. pct += item.Progress ?? 0;
  154. }
  155. }
  156. if (job.ItemCount > 0)
  157. {
  158. pct /= job.ItemCount;
  159. job.Progress = pct;
  160. }
  161. else
  162. {
  163. job.Progress = null;
  164. }
  165. if (jobItems.Any(i => i.Status == SyncJobItemStatus.Transferring))
  166. {
  167. job.Status = SyncJobStatus.Transferring;
  168. }
  169. else if (jobItems.Any(i => i.Status == SyncJobItemStatus.Converting))
  170. {
  171. job.Status = SyncJobStatus.Converting;
  172. }
  173. else if (jobItems.All(i => i.Status == SyncJobItemStatus.Failed))
  174. {
  175. job.Status = SyncJobStatus.Failed;
  176. }
  177. else if (jobItems.All(i => i.Status == SyncJobItemStatus.Cancelled))
  178. {
  179. job.Status = SyncJobStatus.Cancelled;
  180. }
  181. else if (jobItems.All(i => i.Status == SyncJobItemStatus.ReadyToTransfer))
  182. {
  183. job.Status = SyncJobStatus.ReadyToTransfer;
  184. }
  185. else if (jobItems.All(i => i.Status == SyncJobItemStatus.Cancelled || i.Status == SyncJobItemStatus.Failed || i.Status == SyncJobItemStatus.Synced || i.Status == SyncJobItemStatus.RemovedFromDevice))
  186. {
  187. if (jobItems.Any(i => i.Status == SyncJobItemStatus.Failed))
  188. {
  189. job.Status = SyncJobStatus.CompletedWithError;
  190. }
  191. else
  192. {
  193. job.Status = SyncJobStatus.Completed;
  194. }
  195. }
  196. else
  197. {
  198. job.Status = SyncJobStatus.Queued;
  199. }
  200. await _syncRepo.Update(job).ConfigureAwait(false);
  201. _syncManager.OnSyncJobUpdated(job);
  202. }
  203. public async Task<IEnumerable<BaseItem>> GetItemsForSync(SyncCategory? category, string parentId, IEnumerable<string> itemIds, User user, bool unwatchedOnly)
  204. {
  205. var list = new List<BaseItem>();
  206. if (category.HasValue)
  207. {
  208. list = (await GetItemsForSync(category.Value, parentId, user).ConfigureAwait(false)).ToList();
  209. }
  210. else
  211. {
  212. foreach (var itemId in itemIds)
  213. {
  214. var subList = await GetItemsForSync(itemId, user).ConfigureAwait(false);
  215. list.AddRange(subList);
  216. }
  217. }
  218. IEnumerable<BaseItem> items = list;
  219. items = items.Where(_syncManager.SupportsSync);
  220. if (unwatchedOnly)
  221. {
  222. // Avoid implicitly captured closure
  223. var currentUser = user;
  224. items = items.Where(i =>
  225. {
  226. var video = i as Video;
  227. if (video != null)
  228. {
  229. return !video.IsPlayed(currentUser);
  230. }
  231. return true;
  232. });
  233. }
  234. return items.DistinctBy(i => i.Id);
  235. }
  236. private async Task<IEnumerable<BaseItem>> GetItemsForSync(SyncCategory category, string parentId, User user)
  237. {
  238. var parent = string.IsNullOrWhiteSpace(parentId)
  239. ? user.RootFolder
  240. : (Folder)_libraryManager.GetItemById(parentId);
  241. InternalItemsQuery query;
  242. switch (category)
  243. {
  244. case SyncCategory.Latest:
  245. query = new InternalItemsQuery
  246. {
  247. IsFolder = false,
  248. SortBy = new[] { ItemSortBy.DateCreated, ItemSortBy.SortName },
  249. SortOrder = SortOrder.Descending,
  250. Recursive = true
  251. };
  252. break;
  253. case SyncCategory.Resume:
  254. query = new InternalItemsQuery
  255. {
  256. IsFolder = false,
  257. SortBy = new[] { ItemSortBy.DatePlayed, ItemSortBy.SortName },
  258. SortOrder = SortOrder.Descending,
  259. Recursive = true,
  260. IsResumable = true,
  261. MediaTypes = new[] { MediaType.Video }
  262. };
  263. break;
  264. case SyncCategory.NextUp:
  265. return _tvSeriesManager.GetNextUp(new NextUpQuery
  266. {
  267. ParentId = parentId,
  268. UserId = user.Id.ToString("N")
  269. }).Items;
  270. default:
  271. throw new ArgumentException("Unrecognized category: " + category);
  272. }
  273. if (parent == null)
  274. {
  275. return new List<BaseItem>();
  276. }
  277. query.User = user;
  278. var result = await parent.GetItems(query).ConfigureAwait(false);
  279. return result.Items;
  280. }
  281. private async Task<List<BaseItem>> GetItemsForSync(string id, User user)
  282. {
  283. var item = _libraryManager.GetItemById(id);
  284. if (item == null)
  285. {
  286. return new List<BaseItem>();
  287. }
  288. var itemByName = item as IItemByName;
  289. if (itemByName != null)
  290. {
  291. return itemByName.GetTaggedItems(new InternalItemsQuery(user)
  292. {
  293. IsFolder = false,
  294. Recursive = true
  295. }).ToList();
  296. }
  297. if (item.IsFolder)
  298. {
  299. var folder = (Folder)item;
  300. var itemsResult = await folder.GetItems(new InternalItemsQuery(user)
  301. {
  302. Recursive = true,
  303. IsFolder = false
  304. }).ConfigureAwait(false);
  305. var items = itemsResult.Items;
  306. if (!folder.IsPreSorted)
  307. {
  308. items = _libraryManager.Sort(items, user, new[] { ItemSortBy.SortName }, SortOrder.Ascending)
  309. .ToArray();
  310. }
  311. return items.ToList();
  312. }
  313. return new List<BaseItem> { item };
  314. }
  315. private async Task EnsureSyncJobItems(string targetId, CancellationToken cancellationToken)
  316. {
  317. var jobResult = _syncRepo.GetJobs(new SyncJobQuery
  318. {
  319. SyncNewContent = true,
  320. TargetId = targetId
  321. });
  322. foreach (var job in jobResult.Items)
  323. {
  324. cancellationToken.ThrowIfCancellationRequested();
  325. if (job.SyncNewContent)
  326. {
  327. await EnsureJobItems(job).ConfigureAwait(false);
  328. }
  329. }
  330. }
  331. public async Task Sync(IProgress<double> progress, CancellationToken cancellationToken)
  332. {
  333. await EnsureSyncJobItems(null, cancellationToken).ConfigureAwait(false);
  334. // Look job items that are supposedly transfering, but need to be requeued because the synced files have been deleted somehow
  335. await HandleDeletedSyncFiles(cancellationToken).ConfigureAwait(false);
  336. // If it already has a converting status then is must have been aborted during conversion
  337. var result = _syncManager.GetJobItems(new SyncJobItemQuery
  338. {
  339. Statuses = new[] { SyncJobItemStatus.Queued, SyncJobItemStatus.Converting },
  340. AddMetadata = false
  341. });
  342. await SyncJobItems(result.Items, true, progress, cancellationToken).ConfigureAwait(false);
  343. CleanDeadSyncFiles();
  344. }
  345. private async Task HandleDeletedSyncFiles(CancellationToken cancellationToken)
  346. {
  347. // Look job items that are supposedly transfering, but need to be requeued because the synced files have been deleted somehow
  348. var result = _syncManager.GetJobItems(new SyncJobItemQuery
  349. {
  350. Statuses = new[] { SyncJobItemStatus.ReadyToTransfer, SyncJobItemStatus.Transferring },
  351. AddMetadata = false
  352. });
  353. foreach (var item in result.Items)
  354. {
  355. cancellationToken.ThrowIfCancellationRequested();
  356. if (string.IsNullOrWhiteSpace(item.OutputPath) || !_fileSystem.FileExists(item.OutputPath))
  357. {
  358. item.Status = SyncJobItemStatus.Queued;
  359. await _syncManager.UpdateSyncJobItemInternal(item).ConfigureAwait(false);
  360. await UpdateJobStatus(item.JobId).ConfigureAwait(false);
  361. }
  362. }
  363. }
  364. private void CleanDeadSyncFiles()
  365. {
  366. // TODO
  367. // Clean files in sync temp folder that are not linked to any sync jobs
  368. }
  369. public async Task SyncJobItems(string targetId, bool enableConversion, IProgress<double> progress,
  370. CancellationToken cancellationToken)
  371. {
  372. await EnsureSyncJobItems(targetId, cancellationToken).ConfigureAwait(false);
  373. // If it already has a converting status then is must have been aborted during conversion
  374. var result = _syncManager.GetJobItems(new SyncJobItemQuery
  375. {
  376. Statuses = new[] { SyncJobItemStatus.Queued, SyncJobItemStatus.Converting },
  377. TargetId = targetId,
  378. AddMetadata = false
  379. });
  380. await SyncJobItems(result.Items, enableConversion, progress, cancellationToken).ConfigureAwait(false);
  381. }
  382. public async Task SyncJobItems(SyncJobItem[] items, bool enableConversion, IProgress<double> progress, CancellationToken cancellationToken)
  383. {
  384. if (items.Length > 0)
  385. {
  386. if (!SyncRegistrationInfo.Instance.IsRegistered)
  387. {
  388. _logger.Debug("Cancelling sync job processing. Please obtain a supporter membership.");
  389. return;
  390. }
  391. }
  392. var numComplete = 0;
  393. foreach (var item in items)
  394. {
  395. cancellationToken.ThrowIfCancellationRequested();
  396. double percentPerItem = 1;
  397. percentPerItem /= items.Length;
  398. var startingPercent = numComplete * percentPerItem * 100;
  399. var innerProgress = new ActionableProgress<double>();
  400. innerProgress.RegisterAction(p => progress.Report(startingPercent + percentPerItem * p));
  401. // Pull it fresh from the db just to make sure it wasn't deleted or cancelled while another item was converting
  402. var jobItem = enableConversion ? _syncRepo.GetJobItem(item.Id) : item;
  403. if (jobItem != null)
  404. {
  405. if (jobItem.Status != SyncJobItemStatus.Cancelled)
  406. {
  407. await ProcessJobItem(jobItem, enableConversion, innerProgress, cancellationToken).ConfigureAwait(false);
  408. }
  409. await UpdateJobStatus(jobItem.JobId).ConfigureAwait(false);
  410. }
  411. numComplete++;
  412. double percent = numComplete;
  413. percent /= items.Length;
  414. progress.Report(100 * percent);
  415. }
  416. }
  417. private async Task ProcessJobItem(SyncJobItem jobItem, bool enableConversion, IProgress<double> progress, CancellationToken cancellationToken)
  418. {
  419. if (jobItem == null)
  420. {
  421. throw new ArgumentNullException("jobItem");
  422. }
  423. var item = _libraryManager.GetItemById(jobItem.ItemId);
  424. if (item == null)
  425. {
  426. jobItem.Status = SyncJobItemStatus.Failed;
  427. _logger.Error("Unable to locate library item for JobItem {0}, ItemId {1}", jobItem.Id, jobItem.ItemId);
  428. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  429. return;
  430. }
  431. jobItem.Progress = 0;
  432. var syncOptions = _config.GetSyncOptions();
  433. var job = _syncManager.GetJob(jobItem.JobId);
  434. var user = _userManager.GetUserById(job.UserId);
  435. if (user == null)
  436. {
  437. jobItem.Status = SyncJobItemStatus.Failed;
  438. _logger.Error("User not found. Cannot complete the sync job.");
  439. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  440. return;
  441. }
  442. // See if there's already another active job item for the same target
  443. var existingJobItems = _syncManager.GetJobItems(new SyncJobItemQuery
  444. {
  445. AddMetadata = false,
  446. ItemId = jobItem.ItemId,
  447. TargetId = jobItem.TargetId,
  448. Statuses = new[] { SyncJobItemStatus.Converting, SyncJobItemStatus.Queued, SyncJobItemStatus.ReadyToTransfer, SyncJobItemStatus.Synced, SyncJobItemStatus.Transferring }
  449. });
  450. var duplicateJobItems = existingJobItems.Items
  451. .Where(i => !string.Equals(i.Id, jobItem.Id, StringComparison.OrdinalIgnoreCase))
  452. .ToList();
  453. if (duplicateJobItems.Count > 0)
  454. {
  455. var syncProvider = _syncManager.GetSyncProvider(jobItem) as IHasDuplicateCheck;
  456. if (!duplicateJobItems.Any(i => AllowDuplicateJobItem(syncProvider, i, jobItem)))
  457. {
  458. _logger.Debug("Cancelling sync job item because there is already another active job for the same target.");
  459. jobItem.Status = SyncJobItemStatus.Cancelled;
  460. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  461. return;
  462. }
  463. }
  464. var video = item as Video;
  465. if (video != null)
  466. {
  467. await Sync(jobItem, video, user, enableConversion, syncOptions, progress, cancellationToken).ConfigureAwait(false);
  468. }
  469. else if (item is Audio)
  470. {
  471. await Sync(jobItem, (Audio)item, user, enableConversion, syncOptions, progress, cancellationToken).ConfigureAwait(false);
  472. }
  473. else if (item is Photo)
  474. {
  475. await Sync(jobItem, (Photo)item, cancellationToken).ConfigureAwait(false);
  476. }
  477. else
  478. {
  479. await SyncGeneric(jobItem, item, cancellationToken).ConfigureAwait(false);
  480. }
  481. }
  482. private bool AllowDuplicateJobItem(IHasDuplicateCheck provider, SyncJobItem original, SyncJobItem duplicate)
  483. {
  484. if (provider != null)
  485. {
  486. return provider.AllowDuplicateJobItem(original, duplicate);
  487. }
  488. return true;
  489. }
  490. private async Task Sync(SyncJobItem jobItem, Video item, User user, bool enableConversion, SyncOptions syncOptions, IProgress<double> progress, CancellationToken cancellationToken)
  491. {
  492. var job = _syncManager.GetJob(jobItem.JobId);
  493. var jobOptions = _syncManager.GetVideoOptions(jobItem, job);
  494. var conversionOptions = new VideoOptions
  495. {
  496. Profile = jobOptions.DeviceProfile
  497. };
  498. conversionOptions.DeviceId = jobItem.TargetId;
  499. conversionOptions.Context = EncodingContext.Static;
  500. conversionOptions.ItemId = item.Id.ToString("N");
  501. conversionOptions.MediaSources = _mediaSourceManager.GetStaticMediaSources(item, false, user).ToList();
  502. var streamInfo = new StreamBuilder(_mediaEncoder, _logger).BuildVideoItem(conversionOptions);
  503. var mediaSource = streamInfo.MediaSource;
  504. // No sense creating external subs if we're already burning one into the video
  505. var externalSubs = streamInfo.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Encode ?
  506. new List<SubtitleStreamInfo>() :
  507. streamInfo.GetExternalSubtitles(false, true, null, null);
  508. // Mark as requiring conversion if transcoding the video, or if any subtitles need to be extracted
  509. var requiresVideoTranscoding = streamInfo.PlayMethod == PlayMethod.Transcode && jobOptions.IsConverting;
  510. var requiresConversion = requiresVideoTranscoding || externalSubs.Any(i => RequiresExtraction(i, mediaSource));
  511. if (requiresConversion && !enableConversion)
  512. {
  513. return;
  514. }
  515. jobItem.MediaSourceId = streamInfo.MediaSourceId;
  516. jobItem.TemporaryPath = GetTemporaryPath(jobItem);
  517. if (requiresConversion)
  518. {
  519. jobItem.Status = SyncJobItemStatus.Converting;
  520. }
  521. if (requiresVideoTranscoding)
  522. {
  523. // Save the job item now since conversion could take a while
  524. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  525. await UpdateJobStatus(jobItem.JobId).ConfigureAwait(false);
  526. try
  527. {
  528. var lastJobUpdate = DateTime.MinValue;
  529. var innerProgress = new ActionableProgress<double>();
  530. innerProgress.RegisterAction(async pct =>
  531. {
  532. progress.Report(pct);
  533. if ((DateTime.UtcNow - lastJobUpdate).TotalSeconds >= DatabaseProgressUpdateIntervalSeconds)
  534. {
  535. jobItem.Progress = pct / 2;
  536. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  537. await UpdateJobStatus(jobItem.JobId).ConfigureAwait(false);
  538. }
  539. });
  540. jobItem.OutputPath = await _mediaEncoder.EncodeVideo(new EncodingJobOptions(streamInfo, conversionOptions.Profile)
  541. {
  542. OutputDirectory = jobItem.TemporaryPath,
  543. CpuCoreLimit = syncOptions.TranscodingCpuCoreLimit,
  544. ReadInputAtNativeFramerate = !syncOptions.EnableFullSpeedTranscoding
  545. }, innerProgress, cancellationToken);
  546. jobItem.ItemDateModifiedTicks = item.DateModified.Ticks;
  547. _syncManager.OnConversionComplete(jobItem);
  548. }
  549. catch (OperationCanceledException)
  550. {
  551. jobItem.Status = SyncJobItemStatus.Queued;
  552. jobItem.Progress = 0;
  553. }
  554. catch (Exception ex)
  555. {
  556. jobItem.Status = SyncJobItemStatus.Failed;
  557. _logger.ErrorException("Error during sync transcoding", ex);
  558. }
  559. if (jobItem.Status == SyncJobItemStatus.Failed || jobItem.Status == SyncJobItemStatus.Queued)
  560. {
  561. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  562. return;
  563. }
  564. jobItem.MediaSource = await GetEncodedMediaSource(jobItem.OutputPath, user, true).ConfigureAwait(false);
  565. }
  566. else
  567. {
  568. if (mediaSource.Protocol == MediaProtocol.File)
  569. {
  570. jobItem.OutputPath = mediaSource.Path;
  571. }
  572. else if (mediaSource.Protocol == MediaProtocol.Http)
  573. {
  574. jobItem.OutputPath = await DownloadFile(jobItem, mediaSource, cancellationToken).ConfigureAwait(false);
  575. }
  576. else
  577. {
  578. throw new InvalidOperationException(string.Format("Cannot direct stream {0} protocol", mediaSource.Protocol));
  579. }
  580. jobItem.ItemDateModifiedTicks = item.DateModified.Ticks;
  581. jobItem.MediaSource = mediaSource;
  582. }
  583. jobItem.MediaSource.SupportsTranscoding = false;
  584. if (externalSubs.Count > 0)
  585. {
  586. // Save the job item now since conversion could take a while
  587. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  588. await ConvertSubtitles(jobItem, externalSubs, streamInfo, cancellationToken).ConfigureAwait(false);
  589. }
  590. jobItem.Progress = 50;
  591. jobItem.Status = SyncJobItemStatus.ReadyToTransfer;
  592. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  593. }
  594. private bool RequiresExtraction(SubtitleStreamInfo stream, MediaSourceInfo mediaSource)
  595. {
  596. var originalStream = mediaSource.MediaStreams.FirstOrDefault(i => i.Type == MediaStreamType.Subtitle && i.Index == stream.Index);
  597. return originalStream != null && !originalStream.IsExternal;
  598. }
  599. private async Task ConvertSubtitles(SyncJobItem jobItem,
  600. IEnumerable<SubtitleStreamInfo> subtitles,
  601. StreamInfo streamInfo,
  602. CancellationToken cancellationToken)
  603. {
  604. var files = new List<ItemFileInfo>();
  605. var mediaStreams = jobItem.MediaSource.MediaStreams
  606. .Where(i => i.Type != MediaStreamType.Subtitle || !i.IsExternal)
  607. .ToList();
  608. var startingIndex = mediaStreams.Count == 0 ?
  609. 0 :
  610. mediaStreams.Select(i => i.Index).Max() + 1;
  611. foreach (var subtitle in subtitles)
  612. {
  613. var fileInfo = await ConvertSubtitles(jobItem.TemporaryPath, streamInfo, subtitle, cancellationToken).ConfigureAwait(false);
  614. // Reset this to a value that will be based on the output media
  615. fileInfo.Index = startingIndex;
  616. files.Add(fileInfo);
  617. mediaStreams.Add(new MediaStream
  618. {
  619. Index = startingIndex,
  620. Codec = subtitle.Format,
  621. IsForced = subtitle.IsForced,
  622. IsExternal = true,
  623. Language = subtitle.Language,
  624. Path = fileInfo.Path,
  625. SupportsExternalStream = true,
  626. Type = MediaStreamType.Subtitle
  627. });
  628. startingIndex++;
  629. }
  630. jobItem.AdditionalFiles.AddRange(files);
  631. jobItem.MediaSource.MediaStreams = mediaStreams;
  632. }
  633. private async Task<ItemFileInfo> ConvertSubtitles(string temporaryPath, StreamInfo streamInfo, SubtitleStreamInfo subtitleStreamInfo, CancellationToken cancellationToken)
  634. {
  635. var subtitleStreamIndex = subtitleStreamInfo.Index;
  636. var filename = Guid.NewGuid() + "." + subtitleStreamInfo.Format.ToLower();
  637. var path = Path.Combine(temporaryPath, filename);
  638. _fileSystem.CreateDirectory(Path.GetDirectoryName(path));
  639. using (var stream = await _subtitleEncoder.GetSubtitles(streamInfo.ItemId, streamInfo.MediaSourceId, subtitleStreamIndex, subtitleStreamInfo.Format, 0, null, false, cancellationToken).ConfigureAwait(false))
  640. {
  641. using (var fs = _fileSystem.GetFileStream(path, FileMode.Create, FileAccess.Write, FileShare.Read, true))
  642. {
  643. await stream.CopyToAsync(fs, StreamDefaults.DefaultCopyToBufferSize, cancellationToken).ConfigureAwait(false);
  644. }
  645. }
  646. return new ItemFileInfo
  647. {
  648. Name = Path.GetFileName(path),
  649. Path = path,
  650. Type = ItemFileType.Subtitles,
  651. Index = subtitleStreamIndex
  652. };
  653. }
  654. private const int DatabaseProgressUpdateIntervalSeconds = 2;
  655. private async Task Sync(SyncJobItem jobItem, Audio item, User user, bool enableConversion, SyncOptions syncOptions, IProgress<double> progress, CancellationToken cancellationToken)
  656. {
  657. var job = _syncManager.GetJob(jobItem.JobId);
  658. var jobOptions = _syncManager.GetAudioOptions(jobItem, job);
  659. var conversionOptions = new AudioOptions
  660. {
  661. Profile = jobOptions.DeviceProfile
  662. };
  663. conversionOptions.DeviceId = jobItem.TargetId;
  664. conversionOptions.Context = EncodingContext.Static;
  665. conversionOptions.ItemId = item.Id.ToString("N");
  666. conversionOptions.MediaSources = _mediaSourceManager.GetStaticMediaSources(item, false, user).ToList();
  667. var streamInfo = new StreamBuilder(_mediaEncoder, _logger).BuildAudioItem(conversionOptions);
  668. var mediaSource = streamInfo.MediaSource;
  669. jobItem.MediaSourceId = streamInfo.MediaSourceId;
  670. jobItem.TemporaryPath = GetTemporaryPath(jobItem);
  671. if (streamInfo.PlayMethod == PlayMethod.Transcode && jobOptions.IsConverting)
  672. {
  673. if (!enableConversion)
  674. {
  675. return;
  676. }
  677. jobItem.Status = SyncJobItemStatus.Converting;
  678. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  679. await UpdateJobStatus(jobItem.JobId).ConfigureAwait(false);
  680. try
  681. {
  682. var lastJobUpdate = DateTime.MinValue;
  683. var innerProgress = new ActionableProgress<double>();
  684. innerProgress.RegisterAction(async pct =>
  685. {
  686. progress.Report(pct);
  687. if ((DateTime.UtcNow - lastJobUpdate).TotalSeconds >= DatabaseProgressUpdateIntervalSeconds)
  688. {
  689. jobItem.Progress = pct / 2;
  690. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  691. await UpdateJobStatus(jobItem.JobId).ConfigureAwait(false);
  692. }
  693. });
  694. jobItem.OutputPath = await _mediaEncoder.EncodeAudio(new EncodingJobOptions(streamInfo, conversionOptions.Profile)
  695. {
  696. OutputDirectory = jobItem.TemporaryPath,
  697. CpuCoreLimit = syncOptions.TranscodingCpuCoreLimit
  698. }, innerProgress, cancellationToken);
  699. jobItem.ItemDateModifiedTicks = item.DateModified.Ticks;
  700. _syncManager.OnConversionComplete(jobItem);
  701. }
  702. catch (OperationCanceledException)
  703. {
  704. jobItem.Status = SyncJobItemStatus.Queued;
  705. jobItem.Progress = 0;
  706. }
  707. catch (Exception ex)
  708. {
  709. jobItem.Status = SyncJobItemStatus.Failed;
  710. _logger.ErrorException("Error during sync transcoding", ex);
  711. }
  712. if (jobItem.Status == SyncJobItemStatus.Failed || jobItem.Status == SyncJobItemStatus.Queued)
  713. {
  714. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  715. return;
  716. }
  717. jobItem.MediaSource = await GetEncodedMediaSource(jobItem.OutputPath, user, false).ConfigureAwait(false);
  718. }
  719. else
  720. {
  721. if (mediaSource.Protocol == MediaProtocol.File)
  722. {
  723. jobItem.OutputPath = mediaSource.Path;
  724. }
  725. else if (mediaSource.Protocol == MediaProtocol.Http)
  726. {
  727. jobItem.OutputPath = await DownloadFile(jobItem, mediaSource, cancellationToken).ConfigureAwait(false);
  728. }
  729. else
  730. {
  731. throw new InvalidOperationException(string.Format("Cannot direct stream {0} protocol", mediaSource.Protocol));
  732. }
  733. jobItem.ItemDateModifiedTicks = item.DateModified.Ticks;
  734. jobItem.MediaSource = mediaSource;
  735. }
  736. jobItem.MediaSource.SupportsTranscoding = false;
  737. jobItem.Progress = 50;
  738. jobItem.Status = SyncJobItemStatus.ReadyToTransfer;
  739. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  740. }
  741. private async Task Sync(SyncJobItem jobItem, Photo item, CancellationToken cancellationToken)
  742. {
  743. jobItem.OutputPath = item.Path;
  744. jobItem.Progress = 50;
  745. jobItem.Status = SyncJobItemStatus.ReadyToTransfer;
  746. jobItem.ItemDateModifiedTicks = item.DateModified.Ticks;
  747. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  748. }
  749. private async Task SyncGeneric(SyncJobItem jobItem, BaseItem item, CancellationToken cancellationToken)
  750. {
  751. jobItem.OutputPath = item.Path;
  752. jobItem.Progress = 50;
  753. jobItem.Status = SyncJobItemStatus.ReadyToTransfer;
  754. jobItem.ItemDateModifiedTicks = item.DateModified.Ticks;
  755. await _syncManager.UpdateSyncJobItemInternal(jobItem).ConfigureAwait(false);
  756. }
  757. private async Task<string> DownloadFile(SyncJobItem jobItem, MediaSourceInfo mediaSource, CancellationToken cancellationToken)
  758. {
  759. // TODO: Download
  760. return mediaSource.Path;
  761. }
  762. public string GetTemporaryPath(SyncJob job)
  763. {
  764. return GetTemporaryPath(job.Id);
  765. }
  766. public string GetTemporaryPath(string jobId)
  767. {
  768. var basePath = _config.GetSyncOptions().TemporaryPath;
  769. if (string.IsNullOrWhiteSpace(basePath))
  770. {
  771. basePath = Path.Combine(_config.CommonApplicationPaths.ProgramDataPath, "sync");
  772. }
  773. return Path.Combine(basePath, jobId);
  774. }
  775. public string GetTemporaryPath(SyncJobItem jobItem)
  776. {
  777. return Path.Combine(GetTemporaryPath(jobItem.JobId), jobItem.Id);
  778. }
  779. private async Task<MediaSourceInfo> GetEncodedMediaSource(string path, User user, bool isVideo)
  780. {
  781. var item = _libraryManager.ResolvePath(_fileSystem.GetFileSystemInfo(path));
  782. await item.RefreshMetadata(CancellationToken.None).ConfigureAwait(false);
  783. var hasMediaSources = item as IHasMediaSources;
  784. var mediaSources = _mediaSourceManager.GetStaticMediaSources(hasMediaSources, false).ToList();
  785. var preferredAudio = string.IsNullOrEmpty(user.Configuration.AudioLanguagePreference)
  786. ? new string[] { }
  787. : new[] { user.Configuration.AudioLanguagePreference };
  788. var preferredSubs = string.IsNullOrEmpty(user.Configuration.SubtitleLanguagePreference)
  789. ? new List<string>() : new List<string> { user.Configuration.SubtitleLanguagePreference };
  790. foreach (var source in mediaSources)
  791. {
  792. if (isVideo)
  793. {
  794. source.DefaultAudioStreamIndex =
  795. MediaStreamSelector.GetDefaultAudioStreamIndex(source.MediaStreams, preferredAudio, user.Configuration.PlayDefaultAudioTrack);
  796. var defaultAudioIndex = source.DefaultAudioStreamIndex;
  797. var audioLangage = defaultAudioIndex == null
  798. ? null
  799. : source.MediaStreams.Where(i => i.Type == MediaStreamType.Audio && i.Index == defaultAudioIndex).Select(i => i.Language).FirstOrDefault();
  800. source.DefaultAudioStreamIndex =
  801. MediaStreamSelector.GetDefaultSubtitleStreamIndex(source.MediaStreams, preferredSubs, user.Configuration.SubtitleMode, audioLangage);
  802. }
  803. else
  804. {
  805. var audio = source.MediaStreams.FirstOrDefault(i => i.Type == MediaStreamType.Audio);
  806. if (audio != null)
  807. {
  808. source.DefaultAudioStreamIndex = audio.Index;
  809. }
  810. }
  811. }
  812. return mediaSources.FirstOrDefault();
  813. }
  814. }
  815. }