import time
import crypto.ecdsa

fn main() {
	iterations := 1000

	println('Benchmarking key generation...')
	mut total_gen_time := i64(0)
	for _ in 0 .. iterations {
		sw := time.new_stopwatch()
		_, _ := ecdsa.generate_key() or { panic(err) }
		elapsed := sw.elapsed().microseconds()
		total_gen_time += elapsed
	}
	avg_gen_time := total_gen_time / iterations
	println('Average key generation time: ${avg_gen_time} µs')

	pub_key, priv_key := ecdsa.generate_key() or { panic(err) }
	message := 'Benchmark message'.bytes()

	println('Benchmarking signing...')
	mut total_sign_time := i64(0)
	for _ in 0 .. iterations {
		sw := time.new_stopwatch()
		_ := priv_key.sign(message) or { panic(err) }
		elapsed := sw.elapsed().microseconds()
		total_sign_time += elapsed
	}
	avg_sign_time := total_sign_time / iterations
	println('Average sign time: ${avg_sign_time} µs')

	sig := priv_key.sign(message) or { panic(err) }

	println('Benchmarking verification...')
	mut total_verify_time := i64(0)
	for _ in 0 .. iterations {
		sw := time.new_stopwatch()
		_ := pub_key.verify(message, sig) or { panic(err) }
		elapsed := sw.elapsed().microseconds()
		total_verify_time += elapsed
	}
	avg_verify_time := total_verify_time / iterations
	println('Average verify time: ${avg_verify_time} µs')
}