ersatztv/ErsatzTV.Core/Scheduling/ScriptedScheduling/ScriptedPlayoutBuilder.cs

using System.CommandLine.Parsing;
using System.IO.Abstractions;
using CliWrap;
using CliWrap.Buffered;
using ErsatzTV.Core.Domain;
using ErsatzTV.Core.Interfaces.Repositories;
using ErsatzTV.Core.Interfaces.Scheduling;
using ErsatzTV.Core.Scheduling.Engine;
using Microsoft.Extensions.Logging;

namespace ErsatzTV.Core.Scheduling.ScriptedScheduling;

public class ScriptedPlayoutBuilder(
    IConfigElementRepository configElementRepository,
    IScriptedPlayoutBuilderService scriptedPlayoutBuilderService,
    ISchedulingEngine schedulingEngine,
    IFileSystem fileSystem,
    ILogger<ScriptedPlayoutBuilder> logger)
    : IScriptedPlayoutBuilder
{
    public async Task<Either<BaseError, PlayoutBuildResult>> Build(
        DateTimeOffset start,
        Playout playout,
        PlayoutReferenceData referenceData,
        PlayoutBuildMode mode,
        CancellationToken cancellationToken)
    {
        var result = PlayoutBuildResult.Empty;

        ScriptedBuildSession session = scriptedPlayoutBuilderService.StartSession(schedulingEngine);
        Guid buildId = session.BuildId;

        var timeoutSeconds = 30;

        try
        {
            var args = CommandLineParser.SplitCommandLine(playout.ScheduleFile).ToList();

            string scriptFile = args[0];
            string[] scriptArgs = args.Skip(1).ToArray();

            if (!fileSystem.File.Exists(scriptFile))
            {
                logger.LogError(
                    "Cannot build scripted playout; schedule file {File} does not exist",
                    scriptFile);
                return BaseError.New($"Cannot build scripted playout; schedule file {scriptFile} does not exist");
            }

            var arguments = new List<string>
            {
                $"http://localhost:{Settings.UiPort}",
                buildId.ToString(),
                mode.ToString().ToLowerInvariant()
            };

            if (scriptArgs.Length > 0)
            {
                arguments.AddRange(scriptArgs);
            }

            logger.LogInformation(
                "Building scripted playout {Script} with arguments {Arguments}",
                scriptFile,
                arguments);

            int daysToBuild = await GetDaysToBuild(cancellationToken);
            DateTimeOffset finish = start.AddDays(daysToBuild);

            schedulingEngine.WithPlayoutId(playout.Id);
            schedulingEngine.WithMode(mode);
            schedulingEngine.WithSeed(playout.Seed);
            schedulingEngine.BuildBetween(start, finish);
            schedulingEngine.WithReferenceData(referenceData);

            schedulingEngine.RestoreOrReset(Optional(playout.Anchor));

            Option<int> maybeTimeoutSeconds = await configElementRepository.GetValue<int>(
                ConfigElementKey.PlayoutScriptedScheduleTimeoutSeconds,
                cancellationToken);

            foreach (int seconds in maybeTimeoutSeconds)
            {
                timeoutSeconds = seconds;
            }

            using var cts = new CancellationTokenSource(TimeSpan.FromSeconds(timeoutSeconds));
            using var linkedCts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, cts.Token);

            Command command = CreateCommand(scriptFile, arguments, session.Token);

            var commandResult = await command.ExecuteBufferedAsync(linkedCts.Token);
            if (!string.IsNullOrWhiteSpace(commandResult.StandardOutput))
            {
                logger.LogDebug("Scripted playout output: {Output}", commandResult.StandardOutput);
            }

            if (commandResult.ExitCode != 0)
            {
                logger.LogWarning(
                    "Scripted playout process exited with code {Code}: {Error}",
                    commandResult.ExitCode,
                    commandResult.StandardError);
                return BaseError.New(
                    $"Scripted playout process exited with code {commandResult.ExitCode}: {commandResult.StandardError}");
            }

            playout.Anchor = schedulingEngine.GetAnchor();

            result = MergeResult(result, schedulingEngine.GetState());
        }
        catch (OperationCanceledException)
        {
            logger.LogWarning("Scripted playout build timed out after {TimeoutSeconds} seconds", timeoutSeconds);
            return BaseError.New($"Scripted playout build timed out after {timeoutSeconds} seconds");
        }
        catch (Exception ex)
        {
            logger.LogWarning(ex, "Unexpected exception building scripted playout");
            return BaseError.New($"Unexpected exception building scripted playout: {ex}");
        }
        finally
        {
            scriptedPlayoutBuilderService.EndSession(buildId);
        }

        return result;
    }

    /// <summary>
    ///     The process launch for one build. The arguments are the published contract (host, build id, mode, then
    ///     the user's own); the build-scoped token travels in <see cref="ScriptedBuildToken.EnvironmentVariable" />
    ///     instead, because the global API authorization filter rejects an unauthenticated callback and the argument
    ///     list is logged (ersatztv#913). CliWrap layers the variable over the inherited environment.
    /// </summary>
    internal static Command CreateCommand(string scriptFile, IEnumerable<string> arguments, string buildToken) =>
        Cli.Wrap(scriptFile)
            .WithArguments(arguments)
            .WithEnvironmentVariables(env => env.Set(ScriptedBuildToken.EnvironmentVariable, buildToken))
            .WithValidation(CommandResultValidation.None);

    private async Task<int> GetDaysToBuild(CancellationToken cancellationToken) =>
        await configElementRepository
            .GetValue<int>(ConfigElementKey.PlayoutDaysToBuild, cancellationToken)
            .IfNoneAsync(2);

    private static PlayoutBuildResult MergeResult(PlayoutBuildResult result, ISchedulingEngineState state) =>
        result with
        {
            ClearItems = state.ClearItems,
            RemoveBefore = state.RemoveBefore,
            AddedItems = state.AddedItems,
            //ItemsToRemove = state.ItemsToRemove,
            AddedHistory = state.AddedHistory,
            HistoryToRemove = state.HistoryToRemove
        };
}