100 lines
3.8 KiB
C#
100 lines
3.8 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Jellyfin.Data.Enums;
|
|
using Jellyfin.Plugin.JRay.Models;
|
|
using Jellyfin.Plugin.JRay.Services;
|
|
using Jellyfin.Plugin.JRay.Services.Interfaces;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Entities.TV;
|
|
using MediaBrowser.Controller.Library;
|
|
using Microsoft.AspNetCore.Authorization;
|
|
using Microsoft.AspNetCore.Http;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
|
|
namespace Jellyfin.Plugin.JRay.Controllers;
|
|
|
|
/// <summary>
|
|
/// Lets a remote extraction worker discover which library items still need
|
|
/// to be processed.
|
|
/// </summary>
|
|
[ApiController]
|
|
[Route("Plugins/JRay/Tasks")]
|
|
[Authorize(Roles = "Administrator")]
|
|
public class TasksController : ControllerBase
|
|
{
|
|
private const int DefaultLimit = 10;
|
|
private const int MaxLimit = 100;
|
|
|
|
private readonly ILibraryManager _libraryManager;
|
|
private readonly ITruthDataService _truthDataService;
|
|
private readonly IMediaPolicyStore _policyStore;
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="TasksController"/> class.
|
|
/// </summary>
|
|
/// <param name="libraryManager">The Jellyfin library manager.</param>
|
|
/// <param name="truthDataService">The truth data service.</param>
|
|
/// <param name="policyStore">The prioritise/ignore policy store.</param>
|
|
public TasksController(ILibraryManager libraryManager, ITruthDataService truthDataService, IMediaPolicyStore policyStore)
|
|
{
|
|
_libraryManager = libraryManager;
|
|
_truthDataService = truthDataService;
|
|
_policyStore = policyStore;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets a random sample of movies/episodes that have no truth data yet.
|
|
/// </summary>
|
|
/// <param name="limit">The maximum number of items to return (default 10, max 100).</param>
|
|
/// <returns>Up to <paramref name="limit"/> items with no truth data, in random order.</returns>
|
|
[HttpGet("Pending")]
|
|
[ProducesResponseType(StatusCodes.Status200OK)]
|
|
public ActionResult<IEnumerable<PendingExtractionItem>> GetPending([FromQuery] int limit = DefaultLimit)
|
|
{
|
|
var effectiveLimit = Math.Clamp(limit, 1, MaxLimit);
|
|
var rules = _policyStore.GetRules();
|
|
|
|
var items = _libraryManager.GetItemList(new InternalItemsQuery
|
|
{
|
|
IncludeItemTypes = new[] { BaseItemKind.Movie, BaseItemKind.Episode },
|
|
IsVirtualItem = false,
|
|
Recursive = true,
|
|
});
|
|
|
|
// Keep only items that still need truth data, then apply the policy:
|
|
// drop anything ignored, and order prioritised items ahead of the rest.
|
|
// Randomise within each tier so the backlog still spreads across workers.
|
|
var pending = items
|
|
.Where(item => !string.IsNullOrEmpty(item.Path) && !_truthDataService.HasTruth(item.Id, item.Path))
|
|
.Select(item => new
|
|
{
|
|
Item = item,
|
|
Action = PolicyResolver.Resolve(rules, item.Id, GetSeriesId(item), item.Genres)
|
|
})
|
|
.Where(x => x.Action != PolicyAction.Ignore)
|
|
.OrderByDescending(x => x.Action == PolicyAction.Prioritise)
|
|
.ThenBy(_ => Random.Shared.Next())
|
|
.Take(effectiveLimit)
|
|
.Select(x => new PendingExtractionItem
|
|
{
|
|
ItemId = x.Item.Id,
|
|
Path = x.Item.Path,
|
|
Name = x.Item.Name
|
|
});
|
|
|
|
return Ok(pending);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the series id for an episode, or <see cref="Guid.Empty"/> for any
|
|
/// other item type (so series rules only ever match episodes).
|
|
/// </summary>
|
|
/// <param name="item">The library item.</param>
|
|
/// <returns>The owning series id, or empty.</returns>
|
|
internal static Guid GetSeriesId(BaseItem item)
|
|
{
|
|
return item is Episode episode ? episode.SeriesId : Guid.Empty;
|
|
}
|
|
}
|