ersatztv/ErsatzTV.Core/Scheduling/CountExpression.cs

using ErsatzTV.Core.Interfaces.Scheduling;
using NCalc;

namespace ErsatzTV.Core.Scheduling;

public static class CountExpression
{
    public static int Evaluate(
        string countExpression,
        IMediaCollectionEnumerator enumerator,
        Random random,
        CancellationToken cancellationToken)
    {
        int enumeratorCount = enumerator is PlaylistEnumerator playlistEnumerator
            ? playlistEnumerator.CountForRandom
            : enumerator.Count;
        var expression = new Expression(countExpression);
        expression.EvaluateParameter += (name, e) =>
        {
            e.Result = name switch
            {
                "count" => enumeratorCount,
                "random" => enumeratorCount > 0 ? random.Next() % enumeratorCount : 0,
                _ => e.Result
            };
        };

        object expressionResult = 0;
        try
        {
            expressionResult = expression.Evaluate(cancellationToken);
        }
        catch (Exception)
        {
            // do nothing
        }

        return expressionResult switch
        {
            double d when double.IsInfinity(d) || double.IsNaN(d) => 0,
            double doubleResult => (int)Math.Floor(doubleResult),
            int intResult => intResult,
            long longResult => (int)longResult,
            _ => 0
        };
    }
}