Musify/backend/Musify.Application/Tracks/GetTrackStream.cs

using ErrorOr;
using Mediator;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using Musify.Application.Configuration;
using Musify.Application.Contracts;
using Musify.Application.Tracks.Responses;
using Musify.Domain.Entities;
using Musify.Domain.ValueObjects;

namespace Musify.Application.Tracks
{
    public record GetTrackStreamQuery(Guid TrackId, long UserId)
        : IQuery<ErrorOr<TrackStreamResponse>>;

    public class GetTrackStreamQueryHandler(
        IDatabase database,
        IStreamTicketService ticketService,
        TrackConfiguration trackConfiguration,
        StreamGatewayConfiguration streamGatewayConfiguration,
        ILogger<GetTrackStreamQueryHandler> logger)
        : IQueryHandler<GetTrackStreamQuery, ErrorOr<TrackStreamResponse>>
    {
        public async ValueTask<ErrorOr<TrackStreamResponse>> Handle(GetTrackStreamQuery request, CancellationToken cancellationToken)
        {
            var track = await database.Tracks.AsNoTracking()
                .Where(t => t.Id == request.TrackId)
                .Select(t => new { t.Id, t.AudioTranscodeProcessingStatus, t.AudioFolderName } )
                .SingleOrDefaultAsync(cancellationToken);

            if (track == null)
                return Error.NotFound();

            if (string.IsNullOrWhiteSpace(track.AudioFolderName)
                || track.AudioTranscodeProcessingStatus != ProcessingStatus.Completed)
            {
                logger.LogInformation("Stream requested for track {TrackId} but audio is not ready", request.TrackId);
                return Error.Conflict(description: "Track audio is not available for streaming yet.");
            }

            var folderPath = trackConfiguration.Routes.BuildProcessedAudioPath(track.AudioFolderName);
            var keyPrefix = $"{folderPath}/";

            var ticket = ticketService.IssueTicket(request.UserId, keyPrefix);

            var manifestUrl = string.Join('/',
                streamGatewayConfiguration.PublicBaseUrl.TrimEnd('/'),
                "media",
                folderPath,
                streamGatewayConfiguration.AudioFileName);

            logger.LogInformation("Issued stream ticket for track {TrackId} to user {UserId}", request.TrackId, request.UserId);

            var newListeningHistory = new ListeningHistory
            {
                UserId = request.UserId,
                TrackId = request.TrackId,
            };
            await database.ListeningHistories.AddAsync(newListeningHistory, cancellationToken);
            await database.SaveChangesAsync(cancellationToken);

            logger.LogDebug("Listening history added for track {TrackId} by user {UserId}", request.TrackId, request.UserId);

            return new TrackStreamResponse(manifestUrl, ticket.Token, ticket.ExpiresInSeconds);
        }
    }
}