| | | 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; |
| | | 8 | | using Jellyfin.Database.Implementations.Entities; |
| | | 9 | | using Jellyfin.Extensions; |
| | | 10 | | using MediaBrowser.Controller.Dto; |
| | | 11 | | using MediaBrowser.Controller.Entities; |
| | | 12 | | using MediaBrowser.Controller.Library; |
| | | 13 | | using MediaBrowser.Controller.Persistence; |
| | | 14 | | using MediaBrowser.Model.Querying; |
| | | 15 | | using MediaBrowser.Model.Search; |
| | | 16 | | using Microsoft.EntityFrameworkCore; |
| | | 17 | | using Microsoft.Extensions.Logging; |
| | | 18 | | |
| | | 19 | | namespace Emby.Server.Implementations.Library.Search; |
| | | 20 | | |
| | | 21 | | /// <summary> |
| | | 22 | | /// Manages search providers and orchestrates search operations. |
| | | 23 | | /// </summary> |
| | | 24 | | public class SearchManager : ISearchManager |
| | | 25 | | { |
| | | 26 | | private readonly ILibraryManager _libraryManager; |
| | | 27 | | private readonly IUserManager _userManager; |
| | | 28 | | private readonly IDbContextFactory<JellyfinDbContext> _dbProvider; |
| | | 29 | | private readonly IItemQueryHelpers _queryHelpers; |
| | | 30 | | private readonly ILogger<SearchManager> _logger; |
| | 27 | 31 | | private IExternalSearchProvider[] _externalProviders = []; |
| | 27 | 32 | | private IInternalSearchProvider[] _internalProviders = []; |
| | | 33 | | |
| | | 34 | | /// <summary> |
| | | 35 | | /// Initializes a new instance of the <see cref="SearchManager"/> class. |
| | | 36 | | /// </summary> |
| | | 37 | | /// <param name="libraryManager">The library manager.</param> |
| | | 38 | | /// <param name="userManager">The user manager.</param> |
| | | 39 | | /// <param name="dbProvider">The database context factory.</param> |
| | | 40 | | /// <param name="queryHelpers">The shared item query helpers.</param> |
| | | 41 | | /// <param name="logger">The logger.</param> |
| | | 42 | | public SearchManager( |
| | | 43 | | ILibraryManager libraryManager, |
| | | 44 | | IUserManager userManager, |
| | | 45 | | IDbContextFactory<JellyfinDbContext> dbProvider, |
| | | 46 | | IItemQueryHelpers queryHelpers, |
| | | 47 | | ILogger<SearchManager> logger) |
| | | 48 | | { |
| | 27 | 49 | | _libraryManager = libraryManager; |
| | 27 | 50 | | _userManager = userManager; |
| | 27 | 51 | | _dbProvider = dbProvider; |
| | 27 | 52 | | _queryHelpers = queryHelpers; |
| | 27 | 53 | | _logger = logger; |
| | 27 | 54 | | } |
| | | 55 | | |
| | | 56 | | /// <inheritdoc/> |
| | | 57 | | public void AddParts(IEnumerable<ISearchProvider> providers) |
| | | 58 | | { |
| | 27 | 59 | | var allProviders = providers.OrderBy(p => p.Priority).ToArray(); |
| | | 60 | | |
| | 27 | 61 | | _externalProviders = allProviders.OfType<IExternalSearchProvider>().ToArray(); |
| | 27 | 62 | | _internalProviders = allProviders.OfType<IInternalSearchProvider>().ToArray(); |
| | | 63 | | |
| | 27 | 64 | | _logger.LogInformation( |
| | 27 | 65 | | "Registered {ExternalCount} external search providers: {ExternalProviders}. Fallback providers: {FallbackPro |
| | 27 | 66 | | _externalProviders.Length, |
| | 27 | 67 | | string.Join(", ", _externalProviders.Select(p => $"{p.Name} (priority {p.Priority})")), |
| | 27 | 68 | | string.Join(", ", _internalProviders.Select(p => $"{p.Name} (priority {p.Priority})"))); |
| | 27 | 69 | | } |
| | | 70 | | |
| | | 71 | | /// <inheritdoc/> |
| | | 72 | | public IReadOnlyList<ISearchProvider> GetProviders() |
| | | 73 | | { |
| | 0 | 74 | | return [.. _externalProviders, .. _internalProviders]; |
| | | 75 | | } |
| | | 76 | | |
| | | 77 | | /// <inheritdoc/> |
| | | 78 | | public async Task<IReadOnlyList<SearchResult>> GetSearchResultsAsync( |
| | | 79 | | SearchProviderQuery query, |
| | | 80 | | CancellationToken cancellationToken = default) |
| | | 81 | | { |
| | 0 | 82 | | ArgumentNullException.ThrowIfNull(query); |
| | 0 | 83 | | ArgumentException.ThrowIfNullOrWhiteSpace(query.SearchTerm); |
| | | 84 | | |
| | 0 | 85 | | var searchTerm = query.SearchTerm.Trim().RemoveDiacritics(); |
| | | 86 | | |
| | 0 | 87 | | var externalTask = CollectFromProvidersAsync(_externalProviders, query, searchTerm, cancellationToken); |
| | 0 | 88 | | var internalTask = _internalProviders.Length > 0 |
| | 0 | 89 | | ? CollectFromProvidersAsync(_internalProviders, query, searchTerm, cancellationToken) |
| | 0 | 90 | | : Task.FromResult<IReadOnlyList<SearchResult>>([]); |
| | | 91 | | |
| | 0 | 92 | | await Task.WhenAll(externalTask, internalTask).ConfigureAwait(false); |
| | | 93 | | |
| | 0 | 94 | | var externalResults = await externalTask.ConfigureAwait(false); |
| | | 95 | | |
| | | 96 | | // Internal providers apply user-access filtering inline in their queries. External |
| | | 97 | | // providers don't know about user permissions, so they may return IDs from hidden |
| | | 98 | | // libraries or items the user is otherwise blocked from. Filter them here to close |
| | | 99 | | // that gap. The Items controller's second roundtrip via folder.GetItems applies most |
| | | 100 | | // of these again, but it does not restrict by TopParentIds when ItemIds is set. |
| | 0 | 101 | | if (externalResults.Count > 0 && query.UserId.HasValue && !query.UserId.Value.IsEmpty()) |
| | | 102 | | { |
| | 0 | 103 | | var user = _userManager.GetUserById(query.UserId.Value); |
| | 0 | 104 | | if (user is not null) |
| | | 105 | | { |
| | 0 | 106 | | externalResults = await FilterByUserAccessAsync(externalResults, user, query, cancellationToken).Configu |
| | | 107 | | } |
| | | 108 | | } |
| | | 109 | | |
| | 0 | 110 | | if (externalResults.Count > 0) |
| | | 111 | | { |
| | 0 | 112 | | return externalResults; |
| | | 113 | | } |
| | | 114 | | |
| | 0 | 115 | | if (_internalProviders.Length > 0) |
| | | 116 | | { |
| | 0 | 117 | | _logger.LogDebug("No results from external providers, using internal provider results"); |
| | | 118 | | } |
| | | 119 | | |
| | 0 | 120 | | return await internalTask.ConfigureAwait(false); |
| | 0 | 121 | | } |
| | | 122 | | |
| | | 123 | | private async Task<IReadOnlyList<SearchResult>> FilterByUserAccessAsync( |
| | | 124 | | IReadOnlyList<SearchResult> candidates, |
| | | 125 | | User user, |
| | | 126 | | SearchProviderQuery query, |
| | | 127 | | CancellationToken cancellationToken) |
| | | 128 | | { |
| | | 129 | | // SetUser populates parental rating + blocked/allowed tags, Build populates TopParentIds |
| | | 130 | | // for the user's accessible libraries. The candidate ids are applied to the query below |
| | | 131 | | // rather than to the filter because LibraryManager.AddUserToQuery skips TopParentIds when |
| | | 132 | | // ItemIds is non-empty. |
| | 0 | 133 | | var accessFilter = SearchQueryAccessFilter.Build(user, query, _libraryManager); |
| | | 134 | | |
| | 0 | 135 | | Guid[] candidateIds = [.. candidates.Select(c => c.ItemId)]; |
| | | 136 | | |
| | 0 | 137 | | var dbContext = await _dbProvider.CreateDbContextAsync(cancellationToken).ConfigureAwait(false); |
| | 0 | 138 | | await using (dbContext.ConfigureAwait(false)) |
| | | 139 | | { |
| | 0 | 140 | | var baseQuery = dbContext.BaseItems |
| | 0 | 141 | | .AsNoTracking() |
| | 0 | 142 | | .WhereOneOrMany(candidateIds, e => e.Id); |
| | | 143 | | |
| | 0 | 144 | | baseQuery = _queryHelpers.ApplyAccessFiltering(dbContext, baseQuery, accessFilter); |
| | | 145 | | |
| | 0 | 146 | | var allowed = await baseQuery |
| | 0 | 147 | | .Select(e => new { e.Id, e.PrimaryVersionId }) |
| | 0 | 148 | | .ToListAsync(cancellationToken) |
| | 0 | 149 | | .ConfigureAwait(false); |
| | | 150 | | |
| | 0 | 151 | | var allowedIds = allowed.Select(e => e.Id).ToHashSet(); |
| | | 152 | | |
| | | 153 | | // A provider can return both an alternate version and the primary it belongs to, and the |
| | | 154 | | // two are one item to the user. |
| | 0 | 155 | | allowedIds.ExceptWith(allowed |
| | 0 | 156 | | .Where(e => e.PrimaryVersionId.HasValue && allowedIds.Contains(e.PrimaryVersionId.Value)) |
| | 0 | 157 | | .Select(e => e.Id)); |
| | | 158 | | |
| | 0 | 159 | | if (allowedIds.Count == candidates.Count) |
| | | 160 | | { |
| | 0 | 161 | | return candidates; |
| | | 162 | | } |
| | | 163 | | |
| | 0 | 164 | | var filtered = candidates.Where(c => allowedIds.Contains(c.ItemId)).ToList(); |
| | 0 | 165 | | if (filtered.Count < candidates.Count) |
| | | 166 | | { |
| | 0 | 167 | | _logger.LogDebug( |
| | 0 | 168 | | "Dropped {Dropped} of {Total} search candidates due to user access filtering", |
| | 0 | 169 | | candidates.Count - filtered.Count, |
| | 0 | 170 | | candidates.Count); |
| | | 171 | | } |
| | | 172 | | |
| | 0 | 173 | | return filtered; |
| | | 174 | | } |
| | 0 | 175 | | } |
| | | 176 | | |
| | | 177 | | /// <inheritdoc/> |
| | | 178 | | public async Task<QueryResult<SearchHintInfo>> GetSearchHintsAsync(SearchQuery query, CancellationToken cancellation |
| | | 179 | | { |
| | 0 | 180 | | ArgumentNullException.ThrowIfNull(query); |
| | 0 | 181 | | ArgumentException.ThrowIfNullOrWhiteSpace(query.SearchTerm); |
| | | 182 | | |
| | 0 | 183 | | var providerQuery = BuildProviderQuery(query); |
| | 0 | 184 | | var candidates = await GetSearchResultsAsync(providerQuery, cancellationToken).ConfigureAwait(false); |
| | 0 | 185 | | if (candidates.Count == 0) |
| | | 186 | | { |
| | 0 | 187 | | return new QueryResult<SearchHintInfo>(); |
| | | 188 | | } |
| | | 189 | | |
| | 0 | 190 | | var candidateScores = BuildScoreLookup(candidates); |
| | 0 | 191 | | var user = query.UserId.IsEmpty() ? null : _userManager.GetUserById(query.UserId); |
| | | 192 | | |
| | 0 | 193 | | var excludeItemTypes = BuildExcludeItemTypes(query); |
| | 0 | 194 | | var includeItemTypes = BuildIncludeItemTypes(query); |
| | | 195 | | |
| | 0 | 196 | | var internalQuery = new InternalItemsQuery(user) |
| | 0 | 197 | | { |
| | 0 | 198 | | ItemIds = candidateScores.Keys.ToArray(), |
| | 0 | 199 | | ExcludeItemTypes = excludeItemTypes.ToArray(), |
| | 0 | 200 | | IncludeItemTypes = includeItemTypes.Count > 0 ? includeItemTypes.ToArray() : [], |
| | 0 | 201 | | MediaTypes = query.MediaTypes.ToArray(), |
| | 0 | 202 | | IncludeItemsByName = !query.ParentId.HasValue, |
| | 0 | 203 | | ParentId = query.ParentId ?? Guid.Empty, |
| | 0 | 204 | | Recursive = true, |
| | 0 | 205 | | IsKids = query.IsKids, |
| | 0 | 206 | | IsMovie = query.IsMovie, |
| | 0 | 207 | | IsNews = query.IsNews, |
| | 0 | 208 | | IsSeries = query.IsSeries, |
| | 0 | 209 | | IsSports = query.IsSports, |
| | 0 | 210 | | DtoOptions = new DtoOptions |
| | 0 | 211 | | { |
| | 0 | 212 | | Fields = |
| | 0 | 213 | | [ |
| | 0 | 214 | | ItemFields.AirTime, |
| | 0 | 215 | | ItemFields.DateCreated, |
| | 0 | 216 | | ItemFields.ChannelInfo, |
| | 0 | 217 | | ItemFields.ParentId |
| | 0 | 218 | | ] |
| | 0 | 219 | | } |
| | 0 | 220 | | }; |
| | | 221 | | |
| | | 222 | | // MusicArtist items are "ItemsByName" entities - virtual items that aggregate content by artist name |
| | | 223 | | // rather than being stored as regular library items. They require special handling: |
| | | 224 | | // 1. Convert ParentId to AncestorIds (to filter by library folder) |
| | | 225 | | // 2. Set IncludeItemsByName = true (to include these virtual items in results) |
| | | 226 | | // 3. Clear IncludeItemTypes (GetAllArtists handles type filtering internally) |
| | | 227 | | // 4. Use GetAllArtists() instead of GetItemList() to query the artist index |
| | | 228 | | IReadOnlyList<BaseItem> items; |
| | 0 | 229 | | if (internalQuery.IncludeItemTypes.Length == 1 && internalQuery.IncludeItemTypes[0] == BaseItemKind.MusicArtist) |
| | | 230 | | { |
| | 0 | 231 | | if (!internalQuery.ParentId.IsEmpty()) |
| | | 232 | | { |
| | 0 | 233 | | internalQuery.AncestorIds = [internalQuery.ParentId]; |
| | 0 | 234 | | internalQuery.ParentId = Guid.Empty; |
| | | 235 | | } |
| | | 236 | | |
| | 0 | 237 | | internalQuery.IncludeItemsByName = true; |
| | 0 | 238 | | internalQuery.IncludeItemTypes = []; |
| | 0 | 239 | | items = _libraryManager.GetAllArtists(internalQuery).Items.Select(i => i.Item).ToList(); |
| | | 240 | | } |
| | | 241 | | else |
| | | 242 | | { |
| | 0 | 243 | | items = _libraryManager.GetItemList(internalQuery); |
| | | 244 | | } |
| | | 245 | | |
| | 0 | 246 | | var orderedResults = items |
| | 0 | 247 | | .Select(item => new SearchHintInfo { Item = item }) |
| | 0 | 248 | | .OrderByDescending(hint => candidateScores.GetValueOrDefault(hint.Item.Id, 0f)) |
| | 0 | 249 | | .ToList(); |
| | | 250 | | |
| | 0 | 251 | | var totalCount = orderedResults.Count; |
| | | 252 | | |
| | 0 | 253 | | if (query.StartIndex.HasValue) |
| | | 254 | | { |
| | 0 | 255 | | orderedResults = orderedResults.Skip(query.StartIndex.Value).ToList(); |
| | | 256 | | } |
| | | 257 | | |
| | 0 | 258 | | if (query.Limit.HasValue) |
| | | 259 | | { |
| | 0 | 260 | | orderedResults = orderedResults.Take(query.Limit.Value).ToList(); |
| | | 261 | | } |
| | | 262 | | |
| | 0 | 263 | | return new QueryResult<SearchHintInfo>(query.StartIndex, totalCount, orderedResults); |
| | 0 | 264 | | } |
| | | 265 | | |
| | | 266 | | private async Task<IReadOnlyList<SearchResult>> CollectFromProvidersAsync( |
| | | 267 | | IEnumerable<ISearchProvider> providers, |
| | | 268 | | SearchProviderQuery providerQuery, |
| | | 269 | | string searchTerm, |
| | | 270 | | CancellationToken cancellationToken) |
| | | 271 | | { |
| | 0 | 272 | | var requestedLimit = providerQuery.Limit ?? 100; |
| | 0 | 273 | | var applicable = providers.Where(p => p.CanSearch(providerQuery)).ToArray(); |
| | 0 | 274 | | if (applicable.Length == 0) |
| | | 275 | | { |
| | 0 | 276 | | return []; |
| | | 277 | | } |
| | | 278 | | |
| | 0 | 279 | | var perProvider = await Task.WhenAll( |
| | 0 | 280 | | applicable.Select(p => CollectFromProviderAsync(p, providerQuery, searchTerm, requestedLimit, cancellationTo |
| | 0 | 281 | | .ConfigureAwait(false); |
| | | 282 | | |
| | 0 | 283 | | var bestScores = new Dictionary<Guid, float>(); |
| | 0 | 284 | | foreach (var providerResults in perProvider) |
| | | 285 | | { |
| | 0 | 286 | | foreach (var result in providerResults) |
| | | 287 | | { |
| | 0 | 288 | | UpdateBestScore(bestScores, result); |
| | | 289 | | } |
| | | 290 | | } |
| | | 291 | | |
| | 0 | 292 | | return bestScores |
| | 0 | 293 | | .Select(kvp => new SearchResult(kvp.Key, kvp.Value)) |
| | 0 | 294 | | .OrderByDescending(r => r.Score) |
| | 0 | 295 | | .Take(requestedLimit) |
| | 0 | 296 | | .ToList(); |
| | 0 | 297 | | } |
| | | 298 | | |
| | | 299 | | private async Task<IReadOnlyList<SearchResult>> CollectFromProviderAsync( |
| | | 300 | | ISearchProvider provider, |
| | | 301 | | SearchProviderQuery providerQuery, |
| | | 302 | | string searchTerm, |
| | | 303 | | int requestedLimit, |
| | | 304 | | CancellationToken cancellationToken) |
| | | 305 | | { |
| | | 306 | | try |
| | | 307 | | { |
| | 0 | 308 | | var results = provider is IExternalSearchProvider externalProvider |
| | 0 | 309 | | ? await CollectFromExternalProviderAsync(externalProvider, providerQuery, requestedLimit, cancellationTo |
| | 0 | 310 | | : await provider.SearchAsync(providerQuery, cancellationToken).ConfigureAwait(false); |
| | | 311 | | |
| | 0 | 312 | | _logger.LogDebug( |
| | 0 | 313 | | "Provider {Provider} returned {Count} candidates for search term '{SearchTerm}'", |
| | 0 | 314 | | provider.Name, |
| | 0 | 315 | | results.Count, |
| | 0 | 316 | | searchTerm); |
| | 0 | 317 | | return results; |
| | | 318 | | } |
| | 0 | 319 | | catch (Exception ex) |
| | | 320 | | { |
| | 0 | 321 | | _logger.LogWarning(ex, "Search provider {Provider} failed for term '{SearchTerm}'", provider.Name, searchTer |
| | 0 | 322 | | return []; |
| | | 323 | | } |
| | 0 | 324 | | } |
| | | 325 | | |
| | | 326 | | private static async Task<IReadOnlyList<SearchResult>> CollectFromExternalProviderAsync( |
| | | 327 | | IExternalSearchProvider provider, |
| | | 328 | | SearchProviderQuery providerQuery, |
| | | 329 | | int requestedLimit, |
| | | 330 | | CancellationToken cancellationToken) |
| | | 331 | | { |
| | 0 | 332 | | var results = new List<SearchResult>(); |
| | 0 | 333 | | await foreach (var result in provider.SearchAsync(providerQuery, cancellationToken).ConfigureAwait(false)) |
| | | 334 | | { |
| | 0 | 335 | | results.Add(result); |
| | 0 | 336 | | if (results.Count >= requestedLimit) |
| | | 337 | | { |
| | | 338 | | break; |
| | | 339 | | } |
| | | 340 | | } |
| | | 341 | | |
| | 0 | 342 | | return results; |
| | 0 | 343 | | } |
| | | 344 | | |
| | | 345 | | private static void UpdateBestScore(Dictionary<Guid, float> bestScores, SearchResult result) |
| | | 346 | | { |
| | 0 | 347 | | if (!bestScores.TryGetValue(result.ItemId, out var existingScore) || result.Score > existingScore) |
| | | 348 | | { |
| | 0 | 349 | | bestScores[result.ItemId] = result.Score; |
| | | 350 | | } |
| | 0 | 351 | | } |
| | | 352 | | |
| | | 353 | | private static Dictionary<Guid, float> BuildScoreLookup(IReadOnlyList<SearchResult> results) |
| | | 354 | | { |
| | 0 | 355 | | var lookup = new Dictionary<Guid, float>(results.Count); |
| | 0 | 356 | | foreach (var result in results) |
| | | 357 | | { |
| | 0 | 358 | | lookup[result.ItemId] = result.Score; |
| | | 359 | | } |
| | | 360 | | |
| | 0 | 361 | | return lookup; |
| | | 362 | | } |
| | | 363 | | |
| | | 364 | | private static SearchProviderQuery BuildProviderQuery(SearchQuery query) |
| | | 365 | | { |
| | 0 | 366 | | var excludeItemTypes = BuildExcludeItemTypes(query); |
| | 0 | 367 | | var includeItemTypes = BuildIncludeItemTypes(query); |
| | | 368 | | |
| | | 369 | | // Remove any excluded types from includes |
| | 0 | 370 | | if (includeItemTypes.Count > 0 && excludeItemTypes.Count > 0) |
| | | 371 | | { |
| | 0 | 372 | | includeItemTypes.RemoveAll(excludeItemTypes.Contains); |
| | | 373 | | } |
| | | 374 | | |
| | 0 | 375 | | return new SearchProviderQuery |
| | 0 | 376 | | { |
| | 0 | 377 | | SearchTerm = query.SearchTerm, |
| | 0 | 378 | | UserId = query.UserId.IsEmpty() ? null : query.UserId, |
| | 0 | 379 | | IncludeItemTypes = includeItemTypes.ToArray(), |
| | 0 | 380 | | ExcludeItemTypes = excludeItemTypes.ToArray(), |
| | 0 | 381 | | MediaTypes = query.MediaTypes.ToArray(), |
| | 0 | 382 | | Limit = query.Limit, |
| | 0 | 383 | | ParentId = query.ParentId |
| | 0 | 384 | | }; |
| | | 385 | | } |
| | | 386 | | |
| | | 387 | | private static List<BaseItemKind> BuildExcludeItemTypes(SearchQuery query) |
| | | 388 | | { |
| | 0 | 389 | | var excludeItemTypes = query.ExcludeItemTypes.ToList(); |
| | | 390 | | |
| | 0 | 391 | | excludeItemTypes.Add(BaseItemKind.Year); |
| | 0 | 392 | | excludeItemTypes.Add(BaseItemKind.Folder); |
| | 0 | 393 | | excludeItemTypes.Add(BaseItemKind.CollectionFolder); |
| | | 394 | | |
| | 0 | 395 | | if (!query.IncludeGenres) |
| | | 396 | | { |
| | 0 | 397 | | AddIfMissing(excludeItemTypes, BaseItemKind.Genre); |
| | 0 | 398 | | AddIfMissing(excludeItemTypes, BaseItemKind.MusicGenre); |
| | | 399 | | } |
| | | 400 | | |
| | 0 | 401 | | if (!query.IncludePeople) |
| | | 402 | | { |
| | 0 | 403 | | AddIfMissing(excludeItemTypes, BaseItemKind.Person); |
| | | 404 | | } |
| | | 405 | | |
| | 0 | 406 | | if (!query.IncludeStudios) |
| | | 407 | | { |
| | 0 | 408 | | AddIfMissing(excludeItemTypes, BaseItemKind.Studio); |
| | | 409 | | } |
| | | 410 | | |
| | 0 | 411 | | if (!query.IncludeArtists) |
| | | 412 | | { |
| | 0 | 413 | | AddIfMissing(excludeItemTypes, BaseItemKind.MusicArtist); |
| | | 414 | | } |
| | | 415 | | |
| | 0 | 416 | | return excludeItemTypes; |
| | | 417 | | } |
| | | 418 | | |
| | | 419 | | private static List<BaseItemKind> BuildIncludeItemTypes(SearchQuery query) |
| | | 420 | | { |
| | 0 | 421 | | var includeItemTypes = query.IncludeItemTypes.ToList(); |
| | 0 | 422 | | if (query.IncludeMedia) |
| | | 423 | | { |
| | 0 | 424 | | return includeItemTypes; |
| | | 425 | | } |
| | | 426 | | |
| | 0 | 427 | | if (query.IncludeGenres && IsEmptyOrContains(includeItemTypes, BaseItemKind.Genre)) |
| | | 428 | | { |
| | 0 | 429 | | AddIfMissing(includeItemTypes, BaseItemKind.Genre); |
| | 0 | 430 | | AddIfMissing(includeItemTypes, BaseItemKind.MusicGenre); |
| | | 431 | | } |
| | | 432 | | |
| | 0 | 433 | | if (query.IncludePeople && IsEmptyOrContains(includeItemTypes, BaseItemKind.Person)) |
| | | 434 | | { |
| | 0 | 435 | | AddIfMissing(includeItemTypes, BaseItemKind.Person); |
| | | 436 | | } |
| | | 437 | | |
| | 0 | 438 | | if (query.IncludeStudios && IsEmptyOrContains(includeItemTypes, BaseItemKind.Studio)) |
| | | 439 | | { |
| | 0 | 440 | | AddIfMissing(includeItemTypes, BaseItemKind.Studio); |
| | | 441 | | } |
| | | 442 | | |
| | 0 | 443 | | if (query.IncludeArtists && IsEmptyOrContains(includeItemTypes, BaseItemKind.MusicArtist)) |
| | | 444 | | { |
| | 0 | 445 | | AddIfMissing(includeItemTypes, BaseItemKind.MusicArtist); |
| | | 446 | | } |
| | | 447 | | |
| | 0 | 448 | | return includeItemTypes; |
| | | 449 | | } |
| | | 450 | | |
| | | 451 | | private static bool IsEmptyOrContains(List<BaseItemKind> list, BaseItemKind value) |
| | 0 | 452 | | => list.Count == 0 || list.Contains(value); |
| | | 453 | | |
| | | 454 | | private static void AddIfMissing(List<BaseItemKind> list, BaseItemKind value) |
| | | 455 | | { |
| | 0 | 456 | | if (!list.Contains(value)) |
| | | 457 | | { |
| | 0 | 458 | | list.Add(value); |
| | | 459 | | } |
| | 0 | 460 | | } |
| | | 461 | | } |