benches and generic infection

rewriter/native/Cargo.toml

@@ -12,3 +12,8 @@ urlencoding = "2.1.3"
 
 [dev-dependencies]
 boa_engine = "0.20.0"
+criterion = "0.5.1"
+
+[[bench]]
+name = "samples"
+harness = false

rewriter/native/benches/samples.rs

@@ -0,0 +1,62 @@
+use criterion::{criterion_group, criterion_main, BatchSize, BenchmarkId, Criterion};
+use rewriter::{cfg::Config, rewrite};
+use std::str::FromStr;
+use url::Url;
+use urlencoding::encode;
+
+fn encode_string(str: String) -> String {
+	encode(&str).to_string()
+}
+
+pub fn bench(c: &mut Criterion) {
+	let discord = include_str!("../sample/discord.js");
+	let google = include_str!("../sample/google.js");
+
+	let cfg = Config {
+		prefix: "/scrammedjet/".to_string(),
+		encoder: Box::new(encode_string),
+
+		base: Url::from_str("https://google.com/glorngle/si.js").expect("invalid base"),
+		sourcetag: "glongle1".to_string(),
+
+		wrapfn: "$wrap".to_string(),
+		wrapthisfn: "$gwrap".to_string(),
+		importfn: "$import".to_string(),
+		rewritefn: "$rewrite".to_string(),
+		metafn: "$meta".to_string(),
+		setrealmfn: "$setrealm".to_string(),
+		pushsourcemapfn: "$pushsourcemap".to_string(),
+
+		capture_errors: true,
+		do_sourcemaps: true,
+		scramitize: false,
+		strict_rewrites: true,
+	};
+
+	c.bench_with_input(
+		BenchmarkId::new("rewrite/samples", "discord"),
+		&(discord, cfg.clone()),
+		|b, input| {
+			b.iter_batched(
+				|| input.clone(),
+				|x| rewrite(x.0, x.1),
+				BatchSize::SmallInput,
+			)
+		},
+	);
+
+	c.bench_with_input(
+		BenchmarkId::new("rewrite/samples", "google"),
+		&(google, cfg.clone()),
+		|b, input| {
+			b.iter_batched(
+				|| input.clone(),
+				|x| rewrite(x.0, x.1),
+				BatchSize::SmallInput,
+			)
+		},
+	);
+}
+
+criterion_group!(samples, bench);
+criterion_main!(samples);