using linker.messenger.tunnel.stun.enums;
using System.Buffers.Binary;
using System.Security.Cryptography;

namespace linker.messenger.tunnel.stun.messages;

/// <summary>
/// https://tools.ietf.org/html/rfc5389#section-15
/// </summary>
public class StunAttribute
{
	/*
        Length 是大端
        必须4字节对齐
        对齐的字节可以是任意值
         0                   1                   2                   3
         0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
        +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
        |         Type                  |            Length             |
        +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
        |                         Value (variable)                ....
        +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
     */

	public AttributeType Type { get; set; } = AttributeType.Useless;

	public ushort Length { get; set; }

	public ushort RealLength => (ushort)(Type == AttributeType.Useless ? 0 : 4 + Length + (4 - Length % 4) % 4);

	public IStunAttributeValue Value { get; set; } = new UselessStunAttributeValue();

	public int WriteTo(Span<byte> buffer)
	{
		int length = 4 + Length;
		int n = (4 - length % 4) % 4; // 填充的字节数
		int totalLength = length + n;

		BinaryPrimitives.WriteUInt16BigEndian(buffer, (ushort)Type);
		BinaryPrimitives.WriteUInt16BigEndian(buffer[2..], Length);
		int valueLength = Value.WriteTo(buffer[4..]);

		RandomNumberGenerator.Fill(buffer.Slice(length, n));

		return totalLength;
	}

	/// <returns>
	/// Parse 成功字节，0 则表示 Parse 失败
	/// </returns>
	public int TryParse(ReadOnlySpan<byte> buffer, ReadOnlySpan<byte> magicCookieAndTransactionId)
	{
		if (buffer.Length < 4)
		{
			return default;
		}

		Type = (AttributeType)BinaryPrimitives.ReadUInt16BigEndian(buffer);

		Length = BinaryPrimitives.ReadUInt16BigEndian(buffer[2..]);

		if (buffer.Length < 4 + Length)
		{
			return default;
		}

		ReadOnlySpan<byte> value = buffer.Slice(4, Length);

		IStunAttributeValue t = Type switch
		{
			AttributeType.MappedAddress => new MappedAddressStunAttributeValue(),
			AttributeType.XorMappedAddress => new XorMappedAddressStunAttributeValue(magicCookieAndTransactionId),
			AttributeType.ResponseAddress => new ResponseAddressStunAttributeValue(),
			AttributeType.ChangeRequest => new ChangeRequestStunAttributeValue(),
			AttributeType.SourceAddress => new SourceAddressStunAttributeValue(),
			AttributeType.ChangedAddress => new ChangedAddressStunAttributeValue(),
			AttributeType.OtherAddress => new OtherAddressStunAttributeValue(),
			AttributeType.ReflectedFrom => new ReflectedFromStunAttributeValue(),
			AttributeType.ErrorCode => new ErrorCodeStunAttributeValue(),
			_ => new UselessStunAttributeValue()
		};
		if (t.TryParse(value))
		{
			Value = t;
		}

		return 4 + Length + (4 - Length % 4) % 4; // 对齐
	}
}
