88 lines
3.2 KiB
C#
88 lines
3.2 KiB
C#
using System;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Jellyfin.Plugin.JRay.Models;
|
|
using Jellyfin.Plugin.JRay.Services.Interfaces;
|
|
using Microsoft.AspNetCore.Authorization;
|
|
using Microsoft.AspNetCore.Http;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
|
|
namespace Jellyfin.Plugin.JRay.Controllers;
|
|
|
|
/// <summary>
|
|
/// Exposes scene-actor-extraction "truth" data: which actors are on screen
|
|
/// at a given timestamp in a movie.
|
|
/// </summary>
|
|
[ApiController]
|
|
[Route("Plugins/JRay/Items/{itemId}")]
|
|
[Authorize]
|
|
public class ActorsController : ControllerBase
|
|
{
|
|
private readonly ITruthDataService _truthDataService;
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="ActorsController"/> class.
|
|
/// </summary>
|
|
/// <param name="truthDataService">The truth data service.</param>
|
|
public ActorsController(ITruthDataService truthDataService)
|
|
{
|
|
_truthDataService = truthDataService;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the full actor timeline (every actor with their on-screen scene windows) for a movie.
|
|
/// </summary>
|
|
/// <param name="itemId">The Jellyfin item id.</param>
|
|
/// <param name="cancellationToken">Cancellation token.</param>
|
|
/// <returns>The truth file contents, or 404 if no truth data exists for this item.</returns>
|
|
[HttpGet("Timeline")]
|
|
[ProducesResponseType(StatusCodes.Status200OK)]
|
|
[ProducesResponseType(StatusCodes.Status404NotFound)]
|
|
public async Task<ActionResult<TruthFile>> GetTimeline(Guid itemId, CancellationToken cancellationToken)
|
|
{
|
|
var truth = await _truthDataService.GetTruthAsync(itemId, cancellationToken).ConfigureAwait(false);
|
|
if (truth is null)
|
|
{
|
|
return NotFound();
|
|
}
|
|
|
|
return Ok(truth);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the JRay context (currently: on-screen actors) at a given timestamp.
|
|
/// This is an extensible envelope — future fields (locations, trivia, etc.)
|
|
/// will be added here without changing the route.
|
|
/// </summary>
|
|
/// <param name="itemId">The Jellyfin item id.</param>
|
|
/// <param name="t">The timestamp, in seconds from the start of the movie.</param>
|
|
/// <param name="cancellationToken">Cancellation token.</param>
|
|
/// <returns>The JRay context at <paramref name="t"/>, or 404 if no truth data exists for this item.</returns>
|
|
[HttpGet("jray")]
|
|
[ProducesResponseType(StatusCodes.Status200OK)]
|
|
[ProducesResponseType(StatusCodes.Status404NotFound)]
|
|
public async Task<ActionResult<JRayContext>> GetContext(Guid itemId, [FromQuery] double t, CancellationToken cancellationToken)
|
|
{
|
|
var truth = await _truthDataService.GetTruthAsync(itemId, cancellationToken).ConfigureAwait(false);
|
|
if (truth is null)
|
|
{
|
|
return NotFound();
|
|
}
|
|
|
|
var context = new JRayContext();
|
|
foreach (var actor in truth.Actors.Where(actor => actor.Scenes.Any(scene => scene.Length == 2 && scene[0] <= t && t <= scene[1])))
|
|
{
|
|
context.Actors.Add(new ActorAtTime
|
|
{
|
|
Name = actor.Name,
|
|
ImdbId = actor.ImdbId,
|
|
TmdbId = actor.TmdbId,
|
|
JellyfinId = actor.JellyfinId
|
|
});
|
|
}
|
|
|
|
return Ok(context);
|
|
}
|
|
}
|