import initEpoxy, { EpoxyClient, EpoxyClientOptions, EpoxyHandlers, info as epoxyInfo } from "./pkg/epoxy-bundled.js";

(async () => {
	const params = (new URL(location.href)).searchParams;
	const should_feature_test = params.has("feature_test");
	const should_multiparallel_test = params.has("multi_parallel_test");
	const should_parallel_test = params.has("parallel_test");
	const should_multiperf_test = params.has("multi_perf_test");
	const should_perf_test = params.has("perf_test");
	const should_ws_test = params.has("ws_test");
	const should_tls_test = params.has("rawtls_test");
	const should_udp_test = params.has("udp_test");
	const should_reconnect_test = params.has("reconnect_test");
	const should_perf2_test = params.has("perf2_test");
	const should_wisptransport = params.has("wisptransport");
	const test_url = params.get("url") || "https://httpbin.org/get";
	const wisp_url = params.get("wisp") || "ws://localhost:4000/";
	const wisp_v1 = params.has("v1");
	const wisp_udp = params.has("udp_extension");
	const disable_certverif = params.has("disable_certverif");
	console.log(
		"%cWASM is significantly slower with DevTools open!",
		"color:red;font-size:3rem;font-weight:bold"
	);

	const log = (str) => {
		console.log(str);
		let el = document.createElement("pre");
		el.textContent = str;
		document.getElementById("logs").appendChild(el);
		window.scrollTo(0, document.body.scrollHeight);
	}

	try {
		await initEpoxy();
		let epoxy_client_options = new EpoxyClientOptions();
		epoxy_client_options.user_agent = navigator.userAgent;
		epoxy_client_options.wisp_v2 = !wisp_v1;
		epoxy_client_options.udp_extension_required = wisp_udp;
		epoxy_client_options.disable_certificate_validation = disable_certverif;

		let epoxy_client;

		log(`connecting to "${wisp_url}"${wisp_v1 ? " with wisp v1" : ""}`);
		if (should_wisptransport) {
			log("using wisptransport with websocketstream backend");
			epoxy_client = new EpoxyClient(async () => {
				let wss = new WebSocketStream(wisp_url);
				let { readable, writable } = await wss.opened;
				return { read: readable, write: writable };
			}, epoxy_client_options);
		} else {
			epoxy_client = new EpoxyClient(wisp_url, epoxy_client_options);
		}

		const tconn0 = performance.now();
		await epoxy_client.replace_stream_provider();
		const tconn1 = performance.now();
		log(`conn establish took ${tconn1 - tconn0} ms or ${(tconn1 - tconn0) / 1000} s`);

		// epoxy classes are inspectable
		console.log(epoxy_client);
		// you can change the user agent and redirect limit in JS
		epoxy_client.redirect_limit = 15;
		console.log(epoxyInfo);

		const test_mux = async (url) => {
			const t0 = performance.now();
			await epoxy_client.fetch(url).then(r => r.text());
			const t1 = performance.now();
			return t1 - t0;
		};

		const test_native = async (url) => {
			const t0 = performance.now();
			await fetch(url, { cache: "no-store" }).then(r => r.text());
			const t1 = performance.now();
			return t1 - t0;
		};

		const readableStream = (buffer) => {
			return new ReadableStream({
				start(controller) {
					controller.enqueue(buffer);
					controller.close();
				}
			});
		};

		if (should_feature_test) {
			let formdata = new FormData();
			formdata.append("a", "b");
			for (const url of [
				["https://httpbin.org/post", { method: "POST", body: readableStream((new TextEncoder()).encode("abc")) }],
				["https://httpbin.org/get", {}],
				[new URL("https://httpbin.org/get"), {}],
				["https://httpbin.org/gzip", {}],
				["https://httpbin.org/brotli", {}],
				["https://httpbin.org/redirect/11", {}],
				["https://httpbin.org/redirect/1", { redirect: "manual" }],
				["https://httpbin.org/post", { method: "POST", body: new URLSearchParams("a=b") }],
				["https://httpbin.org/post", { method: "POST", body: formdata }],
				["https://httpbin.org/post", { method: "POST", body: "a" }],
				["https://httpbin.org/post", { method: "POST", body: (new TextEncoder()).encode("abc") }],
				["https://httpbin.org/get", { headers: { "a": "b", "b": "c" } }],
				["https://httpbin.org/get", { headers: new Headers({ "a": "b", "b": "c" }) }]
			]) {
				let resp = await epoxy_client.fetch(url[0], url[1]);
				console.warn(url, resp, Object.fromEntries(resp.headers));
				log(await resp.text());
			}
		} else if (should_multiparallel_test) {
			const num_tests = 10;
			let total_mux_minus_native = 0;
			for (const _ of Array(num_tests).keys()) {
				let total_mux = 0;
				await Promise.all([...Array(num_tests).keys()].map(async i => {
					log(`running mux test ${i}`);
					return await test_mux(test_url);
				})).then((vals) => { total_mux = vals.reduce((acc, x) => acc + x, 0) });
				total_mux = total_mux / num_tests;

				let total_native = 0;
				await Promise.all([...Array(num_tests).keys()].map(async i => {
					log(`running native test ${i}`);
					return await test_native(test_url);
				})).then((vals) => { total_native = vals.reduce((acc, x) => acc + x, 0) });
				total_native = total_native / num_tests;

				log(`avg mux (${num_tests}) took ${total_mux} ms or ${total_mux / 1000} s`);
				log(`avg native (${num_tests}) took ${total_native} ms or ${total_native / 1000} s`);
				log(`avg mux - avg native (${num_tests}): ${total_mux - total_native} ms or ${(total_mux - total_native) / 1000} s`);
				total_mux_minus_native += total_mux - total_native;
			}
			total_mux_minus_native = total_mux_minus_native / num_tests;
			log(`total mux - native (${num_tests} tests of ${num_tests} reqs): ${total_mux_minus_native} ms or ${total_mux_minus_native / 1000} s`);
		} else if (should_parallel_test) {
			const num_tests = 10;

			let total_mux = 0;
			await Promise.all([...Array(num_tests).keys()].map(async i => {
				log(`running mux test ${i}`);
				return await test_mux(test_url);
			})).then((vals) => { total_mux = vals.reduce((acc, x) => acc + x, 0) });
			total_mux = total_mux / num_tests;

			let total_native = 0;
			await Promise.all([...Array(num_tests).keys()].map(async i => {
				log(`running native test ${i}`);
				return await test_native(test_url);
			})).then((vals) => { total_native = vals.reduce((acc, x) => acc + x, 0) });
			total_native = total_native / num_tests;

			log(`avg mux (${num_tests}) took ${total_mux} ms or ${total_mux / 1000} s`);
			log(`avg native (${num_tests}) took ${total_native} ms or ${total_native / 1000} s`);
			log(`avg mux - avg native (${num_tests}): ${total_mux - total_native} ms or ${(total_mux - total_native) / 1000} s`);
		} else if (should_multiperf_test) {
			const num_tests = 10;
			let total_mux_minus_native = 0;
			for (const _ of Array(num_tests).keys()) {
				let total_mux = 0;
				for (const i of Array(num_tests).keys()) {
					log(`running mux test ${i}`);
					total_mux += await test_mux(test_url);
				}
				total_mux = total_mux / num_tests;

				let total_native = 0;
				for (const i of Array(num_tests).keys()) {
					log(`running native test ${i}`);
					total_native += await test_native(test_url);
				}
				total_native = total_native / num_tests;

				log(`avg mux (${num_tests}) took ${total_mux} ms or ${total_mux / 1000} s`);
				log(`avg native (${num_tests}) took ${total_native} ms or ${total_native / 1000} s`);
				log(`avg mux - avg native (${num_tests}): ${total_mux - total_native} ms or ${(total_mux - total_native) / 1000} s`);
				total_mux_minus_native += total_mux - total_native;
			}
			total_mux_minus_native = total_mux_minus_native / num_tests;
			log(`total mux - native (${num_tests} tests of ${num_tests} reqs): ${total_mux_minus_native} ms or ${total_mux_minus_native / 1000} s`);
		} else if (should_perf_test) {
			const num_tests = 10;

			let total_mux = 0;
			for (const i of Array(num_tests).keys()) {
				log(`running mux test ${i}`);
				total_mux += await test_mux(test_url);
			}
			total_mux = total_mux / num_tests;

			let total_native = 0;
			for (const i of Array(num_tests).keys()) {
				log(`running native test ${i}`);
				total_native += await test_native(test_url);
			}
			total_native = total_native / num_tests;

			log(`avg mux (${num_tests}) took ${total_mux} ms or ${total_mux / 1000} s`);
			log(`avg native (${num_tests}) took ${total_native} ms or ${total_native / 1000} s`);
			log(`avg mux - avg native (${num_tests}): ${total_mux - total_native} ms or ${(total_mux - total_native) / 1000} s`);
		} else if (should_ws_test) {
			let handlers = new EpoxyHandlers(
				() => log("opened"),
				() => log("closed"),
				err => console.error(err),
				msg => log(`got "${msg}"`)
			);
			let ws = await epoxy_client.connect_websocket(
				handlers,
				"wss://echo.websocket.events",
				[],
				{ "x-header": "abc" },
			);
			let i = 0;
			while (true) {
				log(`sending \`data${i}\``);
				await ws.send("data" + i);
				i++;
				await (new Promise((res, _) => setTimeout(res, 10)));
			}
		} else if (should_tls_test) {
			let decoder = new TextDecoder();
			const { read, write } = await epoxy_client.connect_tls(
				"google.com:443",
			);
			const reader = read.getReader();
			const writer = write.getWriter();

			log("opened");

			(async () => {
				while (true) {
					const { value: msg, done } = await reader.read();
					if (done || !msg) break;
					console.log(msg);
					log(decoder.decode(msg))
				}
				log("closed");
			})();

			await writer.write(new TextEncoder('utf-8').encode("GET / HTTP 1.1\r\nHost: google.com\r\n\r\n"));
			await (new Promise((res, _) => setTimeout(res, 500)));
			await writer.close();
		} else if (should_udp_test) {
			let decoder = new TextDecoder();
			// tokio example: `cargo r --example echo-udp -- 127.0.0.1:5000`
			const { read, write } = await epoxy_client.connect_udp(
				"127.0.0.1:5000",
			);

			const reader = read.getReader();
			const writer = write.getWriter();

			log("opened");

			(async () => {
				while (true) {
					const { value: msg, done } = await reader.read();
					if (done || !msg) break;
					console.log(msg);
					log(decoder.decode(msg))
				}
				log("closed");
			})();

			while (true) {
				log("sending `data`");
				await writer.write(new TextEncoder('utf-8').encode("data"));
				await (new Promise((res, _) => setTimeout(res, 100)));
			}
		} else if (should_reconnect_test) {
			while (true) {
				try {
					await epoxy_client.fetch(test_url);
				} catch (e) { console.error(e) }
				log("sent req");
				await (new Promise((res, _) => setTimeout(res, 500)));
			}
		} else if (should_perf2_test) {
			const num_outer_tests = 10;
			const num_inner_tests = 50;
			let total_mux_multi = 0;
			for (const _ of Array(num_outer_tests).keys()) {
				let total_mux = 0;
				await Promise.all([...Array(num_inner_tests).keys()].map(async i => {
					log(`running mux test ${i}`);
					return await test_mux(test_url);
				})).then((vals) => { total_mux = vals.reduce((acc, x) => acc + x, 0) });
				total_mux = total_mux / num_inner_tests;

				log(`avg mux (${num_inner_tests}) took ${total_mux} ms or ${total_mux / 1000} s`);
				total_mux_multi += total_mux;
			}
			total_mux_multi = total_mux_multi / num_outer_tests;
			log(`total avg mux (${num_outer_tests} tests of ${num_inner_tests} reqs): ${total_mux_multi} ms or ${total_mux_multi / 1000} s`);
		} else {
			const intervalStream = new ReadableStream({
				start(c) {
					let count = 0;
					const timer = setInterval(() => {
						console.log("sent!");
						c.enqueue("Hello\n");
						if (count === 5) {
							clearInterval(timer);
							c.close();
						}
						count++;
					}, 1000);
				},
			}).pipeThrough(new TextEncoderStream());

			const resp = await epoxy_client.fetch("https://full-duplex-server.deno.dev/", {
				method: "POST",
				duplex: "half",
				redirect: "manual",
				body: intervalStream,
			});

			console.log("foo");

			const reader = resp.body.pipeThrough(new TextDecoderStream()).getReader();

			while (true) {
				const { value, done } = await reader.read();
				if (done) break;
				console.log(value);
			}

			console.log("done!");
		}
		log("done");
	} catch (err) {
		console.error(err);
		log(err.stack);
	}
})();