using System.Globalization;
using System.Text.Json;
using Opc.Ua;
using Shouldly;
using Xunit;
using ZB.MOM.WW.OtOpcUa.OpcUaServer;
using ZB.MOM.WW.OtOpcUa.OpcUaServer.Compatibility.Aggregates;

namespace ZB.MOM.WW.OtOpcUa.OpcUaServer.IntegrationTests;

/// <summary>Compares all preserved aggregate factories with independently generated SDK 1.5.378.182 vectors.</summary>
public sealed class Sdk2AggregateCompatibilityTests
{
    [Theory]
    [InlineData(false)]
    [InlineData(true)]
    public async Task All_37_factories_preserve_the_6144_baseline_vectors(bool throughRuntimeManager)
    {
        await using var rig = throughRuntimeManager
            ? await ServerEventRig.StartAsync("AggregateBaseline", encrypted: false, TestContext.Current.CancellationToken)
            : null;
        var baseline = JsonSerializer.Deserialize<Baseline>(File.ReadAllText(Path.Combine(AppContext.BaseDirectory, "Fixtures", "opcua-1.5.378.182-aggregates.json")), new JsonSerializerOptions { PropertyNameCaseInsensitive = true })!;
        var telemetry = DefaultTelemetry.Create(_ => { });
        var epoch = new DateTime(baseline.EpochTicks, DateTimeKind.Utc);
        var failures = new List<string>();
        foreach (var vector in baseline.Vectors)
        {
            var id = (NodeId)typeof(ObjectIds).GetProperty(vector.Aggregate)!.GetValue(null)!;
            var start = epoch.AddSeconds(vector.Reverse ? 70 : 0); var end = epoch.AddSeconds(vector.Reverse ? 0 : 70);
            var profile = vector.Configuration;
            var config = new AggregateConfiguration
            {
                TreatUncertainAsBad = profile?.TreatUncertainAsBad ?? vector.UncertainAsBad,
                PercentDataBad = profile?.PercentDataBad ?? 100, PercentDataGood = profile?.PercentDataGood ?? 100,
                UseSlopedExtrapolation = profile?.UseSlopedExtrapolation ?? false,
                UseServerCapabilitiesDefaults = profile?.UseServerCapabilitiesDefaults ?? false,
            };
            var calculator = throughRuntimeManager
                ? rig!.Server.CurrentInstance.AggregateManager.CreateCalculator(id, start, end, baseline.IntervalMilliseconds, vector.Stepped, config)
                : CompatibilityAggregateFactory.Create(id, start, end, baseline.IntervalMilliseconds, vector.Stepped, config, telemetry);
            calculator.ShouldNotBeNull($"Runtime registration must serve {vector.Aggregate}.");
            var output = new List<Value>(); var accepted = new List<bool>(); string? error = null;
            try
            {
                var samples = baseline.Data[vector.DataSet].OrderBy(x => x.Seconds);
                foreach (var sample in vector.Reverse ? samples.Reverse() : samples)
                {
                    var time = epoch.AddSeconds(sample.Seconds);
                    accepted.Add(calculator.QueueRawValue(new DataValue(SdkValueBoundary.FromClr(Payload(sample)), sample.Status, time, time.AddTicks(sample.ServerOffsetTicks), sample.SourcePicoseconds, sample.ServerPicoseconds)));
                    if (vector.Incremental) Drain(false);
                }
                Drain(true);
            }
            catch (Exception ex) { error = ex.GetType().FullName; }
            if (error != vector.Exception || !output.SequenceEqual(vector.Output) || (vector.QueueAccepted is { } expectedAccepted && !accepted.SequenceEqual(expectedAccepted)))
            {
                failures.Add($"{vector.Aggregate}/{vector.DataSet}/rev={vector.Reverse}/step={vector.Stepped}/uncertainBad={vector.UncertainAsBad}/incremental={vector.Incremental}: error {vector.Exception} vs {error}; expected {JsonSerializer.Serialize(vector.Output)} actual {JsonSerializer.Serialize(output)}");
            }
            void Drain(bool partial)
            {
                for (var guard = 0; guard < 100; guard++)
                {
                    if (!calculator.TryGetProcessedValue(partial, out var value)) return;
                    var payload = SdkValueBoundary.ToClr(value.WrappedValue);
                    output.Add(new(partial, payload?.GetType().Name, Format(payload), value.StatusCode.Code, LegacyTimestampTicks(value.SourceTimestamp), LegacyTimestampTicks(value.ServerTimestamp), value.SourcePicoseconds, value.ServerPicoseconds));
                }
                throw new InvalidOperationException("Baseline calculator did not terminate.");
            }
        }
        baseline.Vectors.Length.ShouldBe(6144);
        baseline.Vectors.Select(x => x.Aggregate).Distinct().Count().ShouldBe(37);
        failures.ShouldBeEmpty("The application must retain baseline values, CLR types, complete status bits, timestamps and incremental processing.");
    }

    // UA FILETIME zero is the old CLR unset sentinel, not a different historical instant.
    private static long LegacyTimestampTicks(DateTimeUtc time) => time.IsNull ? DateTime.MinValue.Ticks : ((DateTime)time).Ticks;
    static object? Payload(Sample sample) => sample.Type switch { "SByte" => sbyte.Parse(sample.Payload!, CultureInfo.InvariantCulture),
        "Byte" => byte.Parse(sample.Payload!, CultureInfo.InvariantCulture),
        "Int16" => short.Parse(sample.Payload!, CultureInfo.InvariantCulture),
        "UInt16" => ushort.Parse(sample.Payload!, CultureInfo.InvariantCulture),
        "Int32" => int.Parse(sample.Payload!, CultureInfo.InvariantCulture),
        "UInt32" => uint.Parse(sample.Payload!, CultureInfo.InvariantCulture),
        "UInt64" => ulong.Parse(sample.Payload!, CultureInfo.InvariantCulture),
        "Single" => float.Parse(sample.Payload!, CultureInfo.InvariantCulture),
        "Double" => double.Parse(sample.Payload!, CultureInfo.InvariantCulture), "Int64" => long.Parse(sample.Payload!, CultureInfo.InvariantCulture), "Boolean" => bool.Parse(sample.Payload!), _ => null };
    static string? Format(object? value) => value switch { null => null, double number => number.ToString("R", CultureInfo.InvariantCulture), float number => number.ToString("R", CultureInfo.InvariantCulture), IFormattable f => f.ToString(null, CultureInfo.InvariantCulture), _ => value.ToString() };
    record Baseline(long EpochTicks, int IntervalMilliseconds, Dictionary<string,Sample[]> Data, Vector[] Vectors);
    record Sample(double Seconds,string Type,string? Payload,uint Status,ushort SourcePicoseconds = 0,ushort ServerPicoseconds = 0,long ServerOffsetTicks = 0);
    record Value(bool PartialRequested,string? Type,string? Payload,uint Status,long SourceTicks,long ServerTicks,ushort SourcePicoseconds,ushort ServerPicoseconds);
    record Vector(string Aggregate,string DataSet,bool Reverse,bool Stepped,bool UncertainAsBad,bool Incremental,Value[] Output,string? Exception, Configuration? Configuration = null,bool[]? QueueAccepted = null);
    record Configuration(bool TreatUncertainAsBad,byte PercentDataBad,byte PercentDataGood,bool UseSlopedExtrapolation,bool UseServerCapabilitiesDefaults);
}