feat: select audio streams for next engine during playout build (#2866)

ErsatzTV.Application/Playouts/Commands/SyncNextPlayoutHandler.cs

@@ -6,7 +6,9 @@ using CliWrap;
 using ErsatzTV.Core;
 using ErsatzTV.Core.Domain;
 using ErsatzTV.Core.Extensions;
+using ErsatzTV.Core.FFmpeg;
 using ErsatzTV.Core.Interfaces.Emby;
+using ErsatzTV.Core.Interfaces.FFmpeg;
 using ErsatzTV.Core.Interfaces.Jellyfin;
 using ErsatzTV.Core.Interfaces.Metadata;
 using ErsatzTV.Core.Interfaces.Plex;
@@ -14,7 +16,6 @@ using ErsatzTV.Infrastructure.Data;
 using ErsatzTV.Infrastructure.Extensions;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.Extensions.Logging;
-using PlayoutItem = ErsatzTV.Core.Domain.PlayoutItem;
 
 namespace ErsatzTV.Application.Playouts;
 
@@ -24,6 +25,8 @@ public partial class SyncNextPlayoutHandler(
     IPlexPathReplacementService plexPathReplacementService,
     IJellyfinPathReplacementService jellyfinPathReplacementService,
     IEmbyPathReplacementService embyPathReplacementService,
+    ICustomStreamSelector customStreamSelector,
+    IFFmpegStreamSelector ffmpegStreamSelector,
     IDbContextFactory<TvContext> dbContextFactory,
     ILogger<SyncNextPlayoutHandler> logger)
     : IRequestHandler<SyncNextPlayout>
@@ -223,6 +226,21 @@ public partial class SyncNextPlayoutHandler(
                     };
                 }
 
+                maybeChannel = await dbContext.Channels
+                    .AsNoTracking()
+                    .SingleOrDefaultAsync(c => c.Number == channelNumber, cancellationToken);
+                foreach (Channel channel in maybeChannel)
+                {
+                    var audioVersion = new MediaItemAudioVersion(playoutItem.MediaItem, headVersion);
+                    await SelectTracks(
+                        channel,
+                        audioVersion,
+                        nextPlayoutItem,
+                        playoutItem.PreferredAudioLanguageCode ?? channel.PreferredAudioLanguageCode,
+                        playoutItem.PreferredAudioTitle ?? channel.PreferredAudioTitle,
+                        cancellationToken);
+                }
+
                 playout.Items.Add(nextPlayoutItem);
             }
 
@@ -230,6 +248,63 @@ public partial class SyncNextPlayoutHandler(
         }
     }
 
+    private async Task SelectTracks(
+        Channel channel,
+        MediaItemAudioVersion audioVersion,
+        Core.Next.PlayoutItem nextPlayoutItem,
+        string preferredAudioLanguage,
+        string preferredAudioTitle,
+        CancellationToken cancellationToken)
+    {
+        // TODO: NEXT: support subtitles
+        List<Subtitle> allSubtitles = [];
+
+        Option<MediaStream> maybeAudioStream = Option<MediaStream>.None;
+        //Option<Subtitle> maybeSubtitle = Option<Subtitle>.None;
+
+        if (channel.StreamSelectorMode is ChannelStreamSelectorMode.Custom)
+        {
+            StreamSelectorResult result = await customStreamSelector.SelectStreams(
+                channel,
+                nextPlayoutItem.Start,
+                audioVersion,
+                allSubtitles);
+            maybeAudioStream = result.AudioStream;
+            //maybeSubtitle = result.Subtitle;
+        }
+
+        if (channel.StreamSelectorMode is ChannelStreamSelectorMode.Default || maybeAudioStream.IsNone)
+        {
+            maybeAudioStream =
+                await ffmpegStreamSelector.SelectAudioStream(
+                    audioVersion,
+                    channel.StreamingMode,
+                    channel,
+                    preferredAudioLanguage,
+                    preferredAudioTitle,
+                    shouldLogMessages: false,
+                    cancellationToken);
+
+            // maybeSubtitle =
+            //     await ffmpegStreamSelector.SelectSubtitleStream(
+            //         allSubtitles.ToImmutableList(),
+            //         channel,
+            //         preferredSubtitleLanguage,
+            //         subtitleMode,
+            //         cancellationToken);
+        }
+
+        foreach (MediaStream audioStream in maybeAudioStream)
+        {
+            if (nextPlayoutItem.Tracks?.Audio?.StreamIndex is null)
+            {
+                nextPlayoutItem.Tracks ??= new Core.Next.PlayoutItemTracks();
+                nextPlayoutItem.Tracks.Audio ??= new Core.Next.TrackSelection();
+                nextPlayoutItem.Tracks.Audio.StreamIndex = audioStream.Index;
+            }
+        }
+    }
+
     private async Task<Option<Core.Next.Source>> SourceForItem(
         PlayoutItem playoutItem,
         CancellationToken cancellationToken)

ErsatzTV.Core.Tests/FFmpeg/FFmpegStreamSelectorTests.cs

@@ -72,6 +72,7 @@ public class FFmpegStreamSelectorTests
                 channel,
                 "jpn",
                 "Whatever",
+                shouldLogMessages: false,
                 cancellationToken);
             selectedStream.IsSome.ShouldBeTrue();
             foreach (MediaStream stream in selectedStream)
@@ -134,6 +135,7 @@ public class FFmpegStreamSelectorTests
                 channel,
                 null,
                 channel.PreferredAudioTitle,
+                shouldLogMessages: false,
                 cancellationToken);
             selectedStream.IsSome.ShouldBeTrue();
             foreach (MediaStream stream in selectedStream)

ErsatzTV.Core/FFmpeg/FFmpegLibraryProcessService.cs

@@ -153,6 +153,7 @@ public class FFmpegLibraryProcessService : IFFmpegProcessService
                     channel,
                     preferredAudioLanguage,
                     preferredAudioTitle,
+                    shouldLogMessages: true,
                     cancellationToken);
 
             maybeSubtitle =

ErsatzTV.Core/FFmpeg/FFmpegStreamSelector.cs

@@ -47,29 +47,42 @@ public class FFmpegStreamSelector : IFFmpegStreamSelector
         Channel channel,
         string preferredAudioLanguage,
         string preferredAudioTitle,
+        bool shouldLogMessages,
         CancellationToken cancellationToken)
     {
         if (streamingMode == StreamingMode.HttpLiveStreamingDirect &&
             string.IsNullOrWhiteSpace(preferredAudioLanguage) && string.IsNullOrWhiteSpace(preferredAudioTitle))
+        {
+            if (shouldLogMessages)
             {
                 _logger.LogDebug(
                     "Channel {Number} is HLS Direct with no preferred audio language or title; using all audio streams",
                     channel.Number);
+            }
+
             return None;
         }
 
         string language = (preferredAudioLanguage ?? string.Empty).ToLowerInvariant();
         if (string.IsNullOrWhiteSpace(language))
+        {
+            if (shouldLogMessages)
             {
                 _logger.LogDebug("Channel {Number} has no preferred audio language code", channel.Number);
+            }
+
             Option<string> maybeDefaultLanguage = await _configElementRepository.GetValue<string>(
                 ConfigElementKey.FFmpegPreferredLanguageCode,
                 cancellationToken);
             maybeDefaultLanguage.Match(
                 lang => language = lang.ToLowerInvariant(),
                 () =>
+                {
+                    if (shouldLogMessages)
                     {
                         _logger.LogDebug("FFmpeg has no preferred audio language code; falling back to {Code}", "eng");
+                    }
+
                     language = "eng";
                 });
         }
@@ -77,9 +90,12 @@ public class FFmpegStreamSelector : IFFmpegStreamSelector
         List<string> allLanguageCodes =
             GetTwoAndThreeLetterLanguageCodes(_languageCodeService.GetAllLanguageCodes([language]));
         if (allLanguageCodes.Count > 1)
+        {
+            if (shouldLogMessages)
             {
                 _logger.LogDebug("Preferred audio language has multiple codes {Codes}", allLanguageCodes);
             }
+        }
 
         try
         {
@@ -93,7 +109,11 @@ public class FFmpegStreamSelector : IFFmpegStreamSelector
                         version.MediaItem.Id,
                         version.MediaVersion);
                     sw.Stop();
+                    if (shouldLogMessages)
+                    {
                         _logger.LogDebug("SelectAudioStream duration: {Duration}", sw.Elapsed);
+                    }
+
                     if (result.IsSome)
                     {
                         return result;
@@ -108,7 +128,11 @@ public class FFmpegStreamSelector : IFFmpegStreamSelector
                         version.MediaItem.Id,
                         version.MediaVersion);
                     sw2.Stop();
+                    if (shouldLogMessages)
+                    {
                         _logger.LogDebug("SelectAudioStream duration: {Duration}", sw2.Elapsed);
+                    }
+
                     if (result2.IsSome)
                     {
                         return result2;

ErsatzTV.Core/Interfaces/FFmpeg/IFFmpegStreamSelector.cs

@@ -14,6 +14,7 @@ public interface IFFmpegStreamSelector
         Channel channel,
         string preferredAudioLanguage,
         string preferredAudioTitle,
+        bool shouldLogMessages,
         CancellationToken cancellationToken);
 
     Task<Option<Subtitle>> SelectSubtitleStream(

ErsatzTV.Scanner.Tests/Core/FFmpeg/TranscodingTests.cs

@@ -1186,6 +1186,7 @@ public class TranscodingTests
             Channel channel,
             string preferredAudioLanguage,
             string preferredAudioTitle,
+            bool shouldLogMessages,
             CancellationToken cancellationToken) =>
             Optional(version.MediaVersion.Streams.FirstOrDefault(s => s.MediaStreamKind == MediaStreamKind.Audio))
                 .AsTask();