| | | 1 | | using System; |
| | | 2 | | using System.Collections.Generic; |
| | | 3 | | using System.Linq; |
| | | 4 | | using System.Threading; |
| | | 5 | | using System.Threading.Tasks; |
| | | 6 | | using Jellyfin.Data.Enums; |
| | | 7 | | using Jellyfin.Database.Implementations.Enums; |
| | | 8 | | using MediaBrowser.Controller.Collections; |
| | | 9 | | using MediaBrowser.Controller.Entities; |
| | | 10 | | using MediaBrowser.Controller.Entities.Movies; |
| | | 11 | | using MediaBrowser.Controller.Library; |
| | | 12 | | using MediaBrowser.Model.Entities; |
| | | 13 | | using MediaBrowser.Model.Querying; |
| | | 14 | | using Microsoft.Extensions.Logging; |
| | | 15 | | |
| | | 16 | | namespace Emby.Server.Implementations.Library.Validators |
| | | 17 | | { |
| | | 18 | | /// <summary> |
| | | 19 | | /// Class CollectionPostScanTask. |
| | | 20 | | /// </summary> |
| | | 21 | | public class CollectionPostScanTask : ILibraryPostScanTask |
| | | 22 | | { |
| | | 23 | | private readonly ILibraryManager _libraryManager; |
| | | 24 | | private readonly ICollectionManager _collectionManager; |
| | | 25 | | private readonly ILogger<CollectionPostScanTask> _logger; |
| | | 26 | | |
| | | 27 | | /// <summary> |
| | | 28 | | /// Initializes a new instance of the <see cref="CollectionPostScanTask" /> class. |
| | | 29 | | /// </summary> |
| | | 30 | | /// <param name="libraryManager">The library manager.</param> |
| | | 31 | | /// <param name="collectionManager">The collection manager.</param> |
| | | 32 | | /// <param name="logger">The logger.</param> |
| | | 33 | | public CollectionPostScanTask( |
| | | 34 | | ILibraryManager libraryManager, |
| | | 35 | | ICollectionManager collectionManager, |
| | | 36 | | ILogger<CollectionPostScanTask> logger) |
| | | 37 | | { |
| | 21 | 38 | | _libraryManager = libraryManager; |
| | 21 | 39 | | _collectionManager = collectionManager; |
| | 21 | 40 | | _logger = logger; |
| | 21 | 41 | | } |
| | | 42 | | |
| | | 43 | | /// <summary> |
| | | 44 | | /// Runs the specified progress. |
| | | 45 | | /// </summary> |
| | | 46 | | /// <param name="progress">The progress.</param> |
| | | 47 | | /// <param name="cancellationToken">The cancellation token.</param> |
| | | 48 | | /// <returns>Task.</returns> |
| | | 49 | | public async Task Run(IProgress<double> progress, CancellationToken cancellationToken) |
| | | 50 | | { |
| | | 51 | | var collectionNameMoviesMap = new Dictionary<string, HashSet<Guid>>(); |
| | | 52 | | |
| | | 53 | | foreach (var library in _libraryManager.RootFolder.Children) |
| | | 54 | | { |
| | | 55 | | if (!_libraryManager.GetLibraryOptions(library).AutomaticallyAddToCollection) |
| | | 56 | | { |
| | | 57 | | continue; |
| | | 58 | | } |
| | | 59 | | |
| | | 60 | | var startIndex = 0; |
| | | 61 | | var pagesize = 1000; |
| | | 62 | | |
| | | 63 | | while (true) |
| | | 64 | | { |
| | | 65 | | var movies = _libraryManager.GetItemList(new InternalItemsQuery |
| | | 66 | | { |
| | | 67 | | MediaTypes = new[] { MediaType.Video }, |
| | | 68 | | IncludeItemTypes = new[] { BaseItemKind.Movie }, |
| | | 69 | | IsVirtualItem = false, |
| | | 70 | | OrderBy = new[] { (ItemSortBy.SortName, SortOrder.Ascending) }, |
| | | 71 | | Parent = library, |
| | | 72 | | StartIndex = startIndex, |
| | | 73 | | Limit = pagesize, |
| | | 74 | | Recursive = true |
| | | 75 | | }); |
| | | 76 | | |
| | | 77 | | foreach (var m in movies) |
| | | 78 | | { |
| | | 79 | | if (m is Movie movie && !string.IsNullOrEmpty(movie.CollectionName)) |
| | | 80 | | { |
| | | 81 | | if (collectionNameMoviesMap.TryGetValue(movie.CollectionName, out var movieList)) |
| | | 82 | | { |
| | | 83 | | movieList.Add(movie.Id); |
| | | 84 | | } |
| | | 85 | | else |
| | | 86 | | { |
| | | 87 | | collectionNameMoviesMap[movie.CollectionName] = new HashSet<Guid> { movie.Id }; |
| | | 88 | | } |
| | | 89 | | } |
| | | 90 | | } |
| | | 91 | | |
| | | 92 | | if (movies.Count < pagesize) |
| | | 93 | | { |
| | | 94 | | break; |
| | | 95 | | } |
| | | 96 | | |
| | | 97 | | startIndex += pagesize; |
| | | 98 | | } |
| | | 99 | | } |
| | | 100 | | |
| | | 101 | | var numComplete = 0; |
| | | 102 | | var count = collectionNameMoviesMap.Count; |
| | | 103 | | |
| | | 104 | | if (count == 0) |
| | | 105 | | { |
| | | 106 | | progress.Report(100); |
| | | 107 | | return; |
| | | 108 | | } |
| | | 109 | | |
| | | 110 | | var boxSets = _libraryManager.GetItemList(new InternalItemsQuery |
| | | 111 | | { |
| | | 112 | | IncludeItemTypes = new[] { BaseItemKind.BoxSet }, |
| | | 113 | | CollapseBoxSetItems = false, |
| | | 114 | | Recursive = true |
| | | 115 | | }); |
| | | 116 | | |
| | | 117 | | foreach (var (collectionName, movieIds) in collectionNameMoviesMap) |
| | | 118 | | { |
| | | 119 | | try |
| | | 120 | | { |
| | | 121 | | var boxSet = boxSets.FirstOrDefault(b => b?.Name == collectionName) as BoxSet; |
| | | 122 | | if (boxSet is null) |
| | | 123 | | { |
| | | 124 | | // won't automatically create collection if only one movie in it |
| | | 125 | | if (movieIds.Count >= 2) |
| | | 126 | | { |
| | | 127 | | boxSet = await _collectionManager.CreateCollectionAsync(new CollectionCreationOptions |
| | | 128 | | { |
| | | 129 | | Name = collectionName, |
| | | 130 | | IsLocked = true |
| | | 131 | | }); |
| | | 132 | | |
| | | 133 | | await _collectionManager.AddToCollectionAsync(boxSet.Id, movieIds); |
| | | 134 | | } |
| | | 135 | | } |
| | | 136 | | else |
| | | 137 | | { |
| | | 138 | | await _collectionManager.AddToCollectionAsync(boxSet.Id, movieIds); |
| | | 139 | | } |
| | | 140 | | |
| | | 141 | | numComplete++; |
| | | 142 | | double percent = numComplete; |
| | | 143 | | percent /= count; |
| | | 144 | | percent *= 100; |
| | | 145 | | |
| | | 146 | | progress.Report(percent); |
| | | 147 | | } |
| | | 148 | | catch (Exception ex) |
| | | 149 | | { |
| | | 150 | | _logger.LogError(ex, "Error refreshing {CollectionName} with {@MovieIds}", collectionName, movieIds) |
| | | 151 | | } |
| | | 152 | | } |
| | | 153 | | |
| | | 154 | | progress.Report(100); |
| | | 155 | | } |
| | | 156 | | } |
| | | 157 | | } |