import { RLP } from '@ethereumjs/rlp'
import { decodeRlp } from 'ethers'
import { bench, describe } from 'vitest'

import { rlpToBytes } from './fromRlp.js'
import { bytesToRlp } from './toRlp.js'

const generateBytes = (length: number) => {
	const bytes = new Uint8Array(length)
	for (let i = 0; i < length; i++) bytes[i] = i
	return bytes
}

const generateList = (length: number) => {
	const bytes = []
	for (let i = 0; i < length; i++) bytes[i] = generateBytes(i)
	return bytes
}

describe('rlp: prefix === 0xb8', () => {
	const bytes = bytesToRlp(generateBytes(255))

	bench('viem: `fromRlp`', () => {
		rlpToBytes(bytes)
	})

	bench('ethers: `decodeRlp`', () => {
		decodeRlp(bytes as any)
	})

	bench('@ethereumjs/rlp: `RLP.decode`', () => {
		RLP.decode(bytes)
	})
})

describe('rlp: prefix === 0xb9', () => {
	const bytes = bytesToRlp(generateBytes(65_535))

	bench('viem: `fromRlp`', () => {
		rlpToBytes(bytes)
	})

	bench('ethers: `decodeRlp`', () => {
		decodeRlp(bytes as any)
	})

	bench('@ethereumjs/rlp: `RLP.decode`', () => {
		RLP.decode(bytes)
	})
})

describe('rlp: prefix === 0xba', () => {
	const bytes = bytesToRlp(generateBytes(16_777_215))

	bench('viem: `fromRlp`', () => {
		rlpToBytes(bytes)
	})

	bench.skip('ethers: `decodeRlp`', () => {
		decodeRlp(bytes as any)
	})

	bench('@ethereumjs/rlp: `RLP.decode`', () => {
		RLP.decode(bytes)
	})
})

describe('rlp list: prefix === 0xf8', () => {
	const list = bytesToRlp(generateList(60))

	bench('viem: `fromRlp`', () => {
		rlpToBytes(list)
	})

	bench('ethers: `decodeRlp`', () => {
		decodeRlp(list as any)
	})

	bench('@ethereumjs/rlp: `RLP.decode`', () => {
		RLP.decode(list)
	})
})

describe('rlp list: prefix === 0xf8 (recursive)', () => {
	const list = bytesToRlp([
		generateList(4),
		[generateList(8), [generateList(3), generateBytes(1)]],
		[
			generateList(10),
			[
				generateList(5),
				generateBytes(2),
				[generateList(10), [generateList(20)]],
			],
		],
	])

	bench('viem: `fromRlp`', () => {
		rlpToBytes(list)
	})

	bench('ethers: `decodeRlp`', () => {
		decodeRlp(list as any)
	})

	bench('@ethereumjs/rlp: `RLP.decode`', () => {
		RLP.decode(list)
	})
})

describe('rlp: tx (2048kB - prefix: 0xfa)', () => {
	const list = bytesToRlp([
		generateBytes(1),
		generateBytes(4),
		generateBytes(8),
		generateBytes(8),
		generateBytes(4),
		generateBytes(20),
		generateBytes(8),
		generateBytes(2_048_000),
	])

	bench('viem: `fromRlp`', () => {
		rlpToBytes(list)
	})

	bench('ethers: `decodeRlp`', () => {
		decodeRlp(list as any)
	})

	bench('@ethereumjs/rlp: `RLP.decode`', () => {
		RLP.decode(list, true)
	})
})