package protocol

// A PacketNumber in QUIC
type PacketNumber int64

// InvalidPacketNumber is a packet number that is never sent.
// In QUIC, 0 is a valid packet number.
const InvalidPacketNumber PacketNumber = -1

// PacketNumberLen is the length of the packet number in bytes
type PacketNumberLen uint8

const (
	// PacketNumberLen1 is a packet number length of 1 byte
	PacketNumberLen1 PacketNumberLen = 1
	// PacketNumberLen2 is a packet number length of 2 bytes
	PacketNumberLen2 PacketNumberLen = 2
	// PacketNumberLen3 is a packet number length of 3 bytes
	PacketNumberLen3 PacketNumberLen = 3
	// PacketNumberLen4 is a packet number length of 4 bytes
	PacketNumberLen4 PacketNumberLen = 4
)

// DecodePacketNumber calculates the packet number based its length and the last seen packet number
// This function is taken from https://www.rfc-editor.org/rfc/rfc9000.html#section-a.3.
func ( PacketNumberLen,  PacketNumber,  PacketNumber) PacketNumber {
	 :=  + 1
	 := PacketNumber(1 << ( * 8))
	 :=  / 2
	 :=  - 1
	 := ( & ^) | 
	if  <= - &&  < 1<<62- {
		return  + 
	}
	if  > + &&  >=  {
		return  - 
	}
	return 
}

// PacketNumberLengthForHeader gets the length of the packet number for the public header
// it never chooses a PacketNumberLen of 1 byte, since this is too short under certain circumstances
func (,  PacketNumber) PacketNumberLen {
	var  PacketNumber
	if  == InvalidPacketNumber {
		 =  + 1
	} else {
		 =  - 
	}
	if  < 1<<(16-1) {
		return PacketNumberLen2
	}
	if  < 1<<(24-1) {
		return PacketNumberLen3
	}
	return PacketNumberLen4
}