using MessagePack;

namespace ZB.MOM.WW.OtOpcUa.Driver.Historian.Wonderware.Client.Ipc;

/// <summary>
///     Reads length-prefixed, kind-tagged frames from a stream. Single-consumer — do not call
///     <see cref="ReadFrameAsync"/> from multiple threads against the same instance. Mirror of
///     the sidecar's <c>FrameReader</c>; kept byte-identical so the wire protocol stays stable.
/// </summary>
public sealed class FrameReader : IDisposable
{
    private readonly Stream _stream;
    private readonly bool _leaveOpen;

    public FrameReader(Stream stream, bool leaveOpen = false)
    {
        _stream = stream ?? throw new ArgumentNullException(nameof(stream));
        _leaveOpen = leaveOpen;
    }

    public async Task<(MessageKind Kind, byte[] Body)?> ReadFrameAsync(CancellationToken ct)
    {
        var lengthPrefix = new byte[Framing.LengthPrefixSize];
        if (!await ReadExactAsync(lengthPrefix, ct).ConfigureAwait(false))
            return null; // clean EOF on frame boundary

        var length = (lengthPrefix[0] << 24) | (lengthPrefix[1] << 16) | (lengthPrefix[2] << 8) | lengthPrefix[3];
        if (length < 0 || length > Framing.MaxFrameBodyBytes)
            throw new InvalidDataException($"Sidecar IPC frame length {length} out of range.");

        var kindByte = _stream.ReadByte();
        if (kindByte < 0) throw new EndOfStreamException("EOF after length prefix, before kind byte.");

        var body = new byte[length];
        if (!await ReadExactAsync(body, ct).ConfigureAwait(false))
            throw new EndOfStreamException("EOF mid-frame.");

        return ((MessageKind)(byte)kindByte, body);
    }

    public static T Deserialize<T>(byte[] body) => MessagePackSerializer.Deserialize<T>(body);

    private async Task<bool> ReadExactAsync(byte[] buffer, CancellationToken ct)
    {
        var offset = 0;
        while (offset < buffer.Length)
        {
            var read = await _stream.ReadAsync(buffer.AsMemory(offset, buffer.Length - offset), ct).ConfigureAwait(false);
            if (read == 0)
            {
                if (offset == 0) return false;
                throw new EndOfStreamException($"Stream ended after reading {offset} of {buffer.Length} bytes.");
            }
            offset += read;
        }
        return true;
    }

    public void Dispose()
    {
        if (!_leaveOpen) _stream.Dispose();
    }
}