FixPlaylistOwner.cs 2.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. using System;
  2. using System.Linq;
  3. using System.Threading;
  4. using Jellyfin.Data.Enums;
  5. using MediaBrowser.Controller.Entities;
  6. using MediaBrowser.Controller.Library;
  7. using MediaBrowser.Controller.Playlists;
  8. using Microsoft.Extensions.Logging;
  9. namespace Jellyfin.Server.Migrations.Routines;
  10. /// <summary>
  11. /// Properly set playlist owner.
  12. /// </summary>
  13. internal class FixPlaylistOwner : IMigrationRoutine
  14. {
  15. private readonly ILogger<RemoveDuplicateExtras> _logger;
  16. private readonly ILibraryManager _libraryManager;
  17. private readonly IPlaylistManager _playlistManager;
  18. public FixPlaylistOwner(
  19. ILogger<RemoveDuplicateExtras> logger,
  20. ILibraryManager libraryManager,
  21. IPlaylistManager playlistManager)
  22. {
  23. _logger = logger;
  24. _libraryManager = libraryManager;
  25. _playlistManager = playlistManager;
  26. }
  27. /// <inheritdoc/>
  28. public Guid Id => Guid.Parse("{615DFA9E-2497-4DBB-A472-61938B752C5B}");
  29. /// <inheritdoc/>
  30. public string Name => "FixPlaylistOwner";
  31. /// <inheritdoc/>
  32. public bool PerformOnNewInstall => false;
  33. /// <inheritdoc/>
  34. public void Perform()
  35. {
  36. var playlists = _libraryManager.GetItemList(new InternalItemsQuery
  37. {
  38. IncludeItemTypes = new[] { BaseItemKind.Playlist }
  39. })
  40. .Cast<Playlist>()
  41. .Where(x => x.OwnerUserId.Equals(Guid.Empty))
  42. .ToArray();
  43. if (playlists.Length > 0)
  44. {
  45. foreach (var playlist in playlists)
  46. {
  47. var shares = playlist.Shares;
  48. if (shares.Count > 0)
  49. {
  50. var firstEditShare = shares.First(x => x.CanEdit);
  51. if (firstEditShare is not null)
  52. {
  53. playlist.OwnerUserId = firstEditShare.UserId;
  54. playlist.Shares = shares.Where(x => x != firstEditShare).ToArray();
  55. playlist.UpdateToRepositoryAsync(ItemUpdateType.MetadataEdit, CancellationToken.None).GetAwaiter().GetResult();
  56. _playlistManager.SavePlaylistFile(playlist);
  57. }
  58. }
  59. else
  60. {
  61. playlist.OpenAccess = true;
  62. playlist.UpdateToRepositoryAsync(ItemUpdateType.MetadataEdit, CancellationToken.None).GetAwaiter().GetResult();
  63. }
  64. }
  65. }
  66. }
  67. }