feat(kit): add context

Author: yzuyr

apps/docs/app/index.ts

@@ -18,10 +18,10 @@ const waveKitServerCode = dedent`
 const waveKitAppCode = dedent`
 	// app/index.ts
 	import { wave } from "@wavekit/wave";
-	import { WaveKitResponse } from "@wavekit/kit";
+	import type { WaveKitHandler } from "@wavekit/kit";
 
-	export function GET(req: BunRequest) {
-		return WaveKitResponse.html(
+	export const GET: WaveKitHandler = (c) => {
+		return c.html(
 			wave.article({ class: "container mx-auto" }, (article) => {
 				article
 					.h1("WaveKit is awesome.")
@@ -33,7 +33,7 @@ const waveKitAppCode = dedent`
 `;
 
 const waveKitWaveCode = dedent`
-	// src/components/form.ts
+	// src/components/form.tsx
 	import { wave } from "@wavekit/wave";
 
 	export const SignUpForm = wave.form({
@@ -104,7 +104,7 @@ export async function GET(req: BunRequest) {
 						.code("bun add @wavekit/wave @wavekit/kit")
 						.p({ style: "margin-top: 1rem;" }, "Nightly release:")
 						.code(
-							"bun add https://pkg.pr.new/getgrinta/wavekit/wave https://pkg.pr.new/getgrinta/wavekit/kit",
+							"bun add https://pkg.pr.new/getgrinta/wavekit/wave@main https://pkg.pr.new/getgrinta/wavekit/kit@main",
 						)
 						.h3("Usage")
 						.div(

packages/kit/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "@wavekit/kit",
 	"type": "module",
-	"version": "0.0.2",
+	"version": "0.0.3",
 	"module": "dist/index.js",
 	"types": "dist/index.d.ts",
 	"exports": {

packages/kit/src/index.ts

@@ -3,4 +3,8 @@
  */
 export { createWaveKit, ssgRender } from "./server";
 export { WaveKitResponse } from "./response";
-export type { SsgRenderProps, CreateWaveKitProps } from "./server";
+export type {
+	SsgRenderProps,
+	CreateWaveKitProps,
+	WaveKitHandler,
+} from "./server";

packages/kit/src/server.ts

@@ -1,5 +1,6 @@
 import path from "node:path";
-import type { RouterTypes } from "bun";
+import type { BunRequest, RouterTypes, Server } from "bun";
+import { WaveKitResponse } from "./response";
 
 const isDev = process.env.NODE_ENV === "development";
 
@@ -9,6 +10,17 @@ const defaultRoutesDir = isDev
 
 const defaultOutDir = path.join(process.cwd(), "build");
 
+type WaveKitContext = {
+	req: BunRequest;
+	res: WaveKitResponse;
+	html: typeof WaveKitResponse.html;
+	json: typeof WaveKitResponse.json;
+};
+
+export type WaveKitHandler = (
+	c: WaveKitContext,
+) => Response | Promise<Response>;
+
 type BuildRoutesProps = {
 	routes: Record<string, string>;
 };
@@ -18,14 +30,43 @@ export async function buildRoutes({
 }: BuildRoutesProps): Promise<
 	Record<string, RouterTypes.RouteHandlerObject<string>>
 > {
+	const contextStore = new Map();
 	const rawRoutes = Object.entries(routes);
 	const routesWithHandlers = rawRoutes.map(async ([path, handlerPath]) => {
 		const handler = (await import(
 			handlerPath
 		)) as RouterTypes.RouteHandlerObject<string>;
-		return [path, handler];
+		const contextHandler: RouterTypes.RouteHandlerObject<string> = {};
+		for (const method of Object.keys(handler)) {
+			const methodHandler = handler[
+				method as RouterTypes.HTTPMethod
+			] as unknown as WaveKitHandler;
+			if (!methodHandler) return;
+			contextHandler[method as RouterTypes.HTTPMethod] = (
+				req: BunRequest,
+				server: Server,
+			) => {
+				const res = new WaveKitResponse();
+				const context = {
+					req,
+					res,
+					html: WaveKitResponse.html,
+					json: WaveKitResponse.json,
+					redirect: WaveKitResponse.redirect,
+					set: contextStore.set,
+					get: contextStore.get,
+				};
+				return methodHandler(context);
+			};
+		}
+		return [path, contextHandler];
 	});
-	return Object.fromEntries(await Promise.all(routesWithHandlers));
+	return Object.fromEntries(
+		(await Promise.all(routesWithHandlers)).filter(Boolean) as [
+			string,
+			RouterTypes.RouteHandlerObject<string>,
+		][],
+	);
 }
 
 export type CreateWaveKitProps = {

packages/kit/test/app/index.test.ts

@@ -1,6 +1,6 @@
-import { WaveKitResponse } from "@wavekit/kit";
+import type { WaveKitHandler } from "@wavekit/kit";
 import { wave } from "@wavekit/wave";
 
-export function GET() {
-	return WaveKitResponse.html(wave.p("Hello World!"));
-}
+export const GET: WaveKitHandler = (c) => {
+	return c.html(wave.p("Hello World!"));
+};