Add tests for AE, and support rate limiting in the Vite plugin (#9803)

* Add tests for AE, and support rate limiting in the Vite plugin

* changeset

Author: penalosa

.changeset/lemon-bobcats-lose.md

@@ -0,0 +1,5 @@
+---
+"@cloudflare/vite-plugin": minor
+---
+
+Support Workers Analytics Engine & Rate Limiting bindings

packages/vite-plugin-cloudflare/playground/bindings/__tests__/worker.spec.ts

@@ -10,3 +10,13 @@ test("unsafe_hello_world support", async () => {
 	const response = await getTextResponse("/hello-world");
 	expect(response).toBe("Hello World binding works");
 });
+
+test("analytics_engine support", async () => {
+	const response = await getTextResponse("/ae");
+	expect(response).toBe("AE binding works");
+});
+
+test("ratelimit support", async () => {
+	const response = await getTextResponse("/rate-limit");
+	expect(response).toBe("Rate limit binding works: first: true, second: false");
+});

packages/vite-plugin-cloudflare/playground/bindings/src/index.ts

@@ -18,6 +18,28 @@ export default {
 					status: 200,
 				});
 			}
+			case "/ae": {
+				await env.WAE.writeDataPoint({ doubles: [2, 3] });
+
+				return new Response("AE binding works", {
+					status: 200,
+				});
+			}
+			case "/rate-limit": {
+				const { success: first } = await env.RATE_LIMITER.limit({
+					key: "shared-key",
+				});
+				const { success: second } = await env.RATE_LIMITER.limit({
+					key: "shared-key",
+				});
+
+				return new Response(
+					`Rate limit binding works: first: ${first}, second: ${second}`,
+					{
+						status: 200,
+					}
+				);
+			}
 			case "/hello-world": {
 				const value = Math.floor(Date.now() * Math.random()).toString(36);
 				await env.HELLO_WORLD.set(value);

packages/vite-plugin-cloudflare/playground/bindings/worker-configuration.d.ts

@@ -1,9 +1,11 @@
 /* eslint-disable */
-// Generated by Wrangler by running `wrangler types --include-runtime=false` (hash: 2aff1a83eda1c659fd494a8ca649c27a)
+// Generated by Wrangler by running `wrangler types --no-include-runtime` (hash: c9b4a08f8a0baf679382de06c1dd7d1e)
 declare namespace Cloudflare {
 	interface Env {
 		KV: KVNamespace;
 		HELLO_WORLD: HelloWorldBinding;
+		WAE: AnalyticsEngineDataset;
+		RATE_LIMITER: RateLimit;
 	}
 }
 interface Env extends Cloudflare.Env {}

packages/vite-plugin-cloudflare/playground/bindings/wrangler.jsonc

@@ -1,4 +1,5 @@
 {
+	"$schema": "./node_modules/wrangler/config-schema.json",
 	"name": "worker",
 	"main": "./src/index.ts",
 	"compatibility_date": "2024-12-30",
@@ -13,4 +14,23 @@
 			"binding": "HELLO_WORLD",
 		},
 	],
+	"analytics_engine_datasets": [
+		{
+			"dataset": "test",
+			"binding": "WAE",
+		},
+	],
+	"unsafe": {
+		"bindings": [
+			{
+				"name": "RATE_LIMITER",
+				"type": "ratelimit",
+				"namespace_id": "1001",
+				"simple": {
+					"limit": 1,
+					"period": 60,
+				},
+			},
+		],
+	},
 }

packages/vite-plugin-cloudflare/src/miniflare-options.ts

@@ -398,8 +398,7 @@ export async function getDevMiniflareOptions(
 
 							const { externalWorkers } = miniflareWorkerOptions;
 
-							const { ratelimits, ...workerOptions } =
-								miniflareWorkerOptions.workerOptions;
+							const workerOptions = miniflareWorkerOptions.workerOptions;
 
 							return {
 								externalWorkers,
@@ -700,7 +699,7 @@ export async function getPreviewMiniflareOptions(
 
 				const { externalWorkers } = miniflareWorkerOptions;
 
-				const { ratelimits, modulesRules, ...workerOptions } =
+				const { modulesRules, ...workerOptions } =
 					miniflareWorkerOptions.workerOptions;
 
 				logUnknownTails(