SyncedMediaSourceProvider.cs 2.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. using MediaBrowser.Controller;
  2. using MediaBrowser.Controller.Entities;
  3. using MediaBrowser.Controller.Library;
  4. using MediaBrowser.Controller.Sync;
  5. using MediaBrowser.Model.Dto;
  6. using MediaBrowser.Model.Sync;
  7. using System;
  8. using System.Collections.Generic;
  9. using System.Linq;
  10. using System.Threading;
  11. using System.Threading.Tasks;
  12. namespace MediaBrowser.Server.Implementations.Sync
  13. {
  14. public class SyncedMediaSourceProvider : IMediaSourceProvider
  15. {
  16. private readonly SyncManager _syncManager;
  17. private readonly IServerApplicationHost _appHost;
  18. public SyncedMediaSourceProvider(ISyncManager syncManager, IServerApplicationHost appHost)
  19. {
  20. _appHost = appHost;
  21. _syncManager = (SyncManager)syncManager;
  22. }
  23. public async Task<IEnumerable<MediaSourceInfo>> GetMediaSources(IHasMediaSources item, CancellationToken cancellationToken)
  24. {
  25. var jobItemResult = _syncManager.GetJobItems(new SyncJobItemQuery
  26. {
  27. AddMetadata = false,
  28. Statuses = new List<SyncJobItemStatus> { SyncJobItemStatus.Synced },
  29. ItemId = item.Id.ToString("N")
  30. });
  31. var list = new List<MediaSourceInfo>();
  32. if (jobItemResult.Items.Length > 0)
  33. {
  34. var targets = _syncManager.ServerSyncProviders
  35. .SelectMany(i => i.GetAllSyncTargets().Select(t => new Tuple<IServerSyncProvider, SyncTarget>(i, t)))
  36. .ToList();
  37. foreach (var jobItem in jobItemResult.Items)
  38. {
  39. var targetTuple = targets.FirstOrDefault(i => string.Equals(i.Item2.Id, jobItem.TargetId, StringComparison.OrdinalIgnoreCase));
  40. if (targetTuple != null)
  41. {
  42. var syncTarget = targetTuple.Item2;
  43. var dataProvider = _syncManager.GetDataProvider(targetTuple.Item1, syncTarget);
  44. var localItemId = MediaSync.GetLocalId(_appHost.SystemId, item.Id.ToString("N"));
  45. var localItem = await dataProvider.GetCachedItem(syncTarget, localItemId).ConfigureAwait(false);
  46. if (localItem != null)
  47. {
  48. list.AddRange(localItem.Item.MediaSources);
  49. }
  50. }
  51. }
  52. }
  53. return list;
  54. }
  55. }
  56. }