orbit-rpc 1.0.0 → 1.2.0

package/dist/index.d.mts

@@ -1,3 +1,5 @@
+import { Context } from "hono";
+import { AsyncLocalStorage } from "node:async_hooks";
 import { Plugin } from "vite";
 
 //#region src/plugin.d.ts
@@ -9,4 +11,45 @@ interface OrbitRpcConfig {
 }
 declare function orbitRpc(config?: OrbitRpcConfig): Plugin[];
 //#endregion
-export { type OrbitRpcConfig, orbitRpc };
+//#region src/context.d.ts
+declare const contextStorage: AsyncLocalStorage<Context<any, any, {}>>;
+/**
+ * アプリ側で環境変数の型を登録するためのインターフェース。
+ *
+ * ```ts
+ * // src/lib/context.ts
+ * declare module "orbit-rpc" {
+ *   interface Register {
+ *     env: { DB: D1Database; ADMIN_PASSWORD: string };
+ *   }
+ * }
+ * ```
+ */
+interface Register {}
+/** Register.env が登録済みなら使い、未登録なら Record<string, unknown> にフォールバック */
+type RegisteredEnv = Register extends {
+  env: infer T;
+} ? T : Record<string, unknown>;
+/**
+ * RPC ハンドラ内で Hono Context を取得する。
+ *
+ * server.ts の関数内から呼ぶと、現在のリクエストに紐づく Hono Context が返る。
+ * Cloudflare Workers のバインディング（D1, KV 等）や Cookie へのアクセスに使う。
+ *
+ * ```ts
+ * // src/lib/context.ts で型を1回宣言すれば:
+ * declare module "orbit-rpc" {
+ *   interface Register { env: { DB: D1Database } }
+ * }
+ *
+ * // server.ts では型パラメータ不要で使える:
+ * import { getContext } from "orbit-rpc";
+ * const c = getContext();
+ * c.env.DB // ← 型補完が効く
+ * ```
+ */
+declare function getContext<E extends Record<string, unknown> = RegisteredEnv>(): Context<{
+  Bindings: E;
+}>;
+//#endregion
+export { type OrbitRpcConfig, type Register, contextStorage, getContext, orbitRpc };

package/dist/index.mjs

@@ -1,5 +1,34 @@
 import path from "node:path";
+import { Readable } from "node:stream";
+import { Hono } from "hono";
+import { AsyncLocalStorage } from "node:async_hooks";
 import fs from "node:fs";
+//#region src/context.ts
+const contextStorage = new AsyncLocalStorage();
+/**
+* RPC ハンドラ内で Hono Context を取得する。
+*
+* server.ts の関数内から呼ぶと、現在のリクエストに紐づく Hono Context が返る。
+* Cloudflare Workers のバインディング（D1, KV 等）や Cookie へのアクセスに使う。
+*
+* ```ts
+* // src/lib/context.ts で型を1回宣言すれば:
+* declare module "orbit-rpc" {
+*   interface Register { env: { DB: D1Database } }
+* }
+*
+* // server.ts では型パラメータ不要で使える:
+* import { getContext } from "orbit-rpc";
+* const c = getContext();
+* c.env.DB // ← 型補完が効く
+* ```
+*/
+function getContext() {
+	const ctx = contextStorage.getStore();
+	if (!ctx) throw new Error("getContext() must be called within an RPC handler. Make sure you are calling it inside a function exported from server.ts.");
+	return ctx;
+}
+//#endregion
 //#region src/scanner.ts
 /**
 * routes ディレクトリから server.ts ファイルをスキャンし、
@@ -159,7 +188,7 @@ function dirToRoutePrefix(relativePath) {
 *
 * 2つの仕事をする:
 * 1. クライアント側: server.ts の import を HTTP fetch スタブに差し替え
-* 2. dev サーバー: /rpc/* リクエストを受けて server.ts の関数を実行
+* 2. dev サーバー / 本番ビルド: Hono アプリで RPC リクエストを処理
 *
 * schema.ts に Zod スキーマがあれば、自動でバリデーションを適用する。
 */
@@ -215,17 +244,18 @@ function orbitRpc(config = {}) {
 			server.watcher.on("add", onFileChange);
 			server.watcher.on("change", onFileChange);
 			server.watcher.on("unlink", onFileChange);
+			const devApp = createDevRpcApp(server, rpcBase, () => serverModules);
 			server.middlewares.use(async (req, res, next) => {
 				if (!req.url?.startsWith(rpcBase)) return next();
 				try {
-					const result = await handleRpcRequest(server, req, rpcBase, serverModules);
-					res.setHeader("Content-Type", "application/json");
-					res.end(JSON.stringify(result ?? null));
+					const webReq = toWebRequest(req);
+					await writeWebResponse(await devApp.fetch(webReq), res);
 				} catch (err) {
-					const message = err instanceof Error ? err.message : "Internal Server Error";
-					res.statusCode = err instanceof RpcError ? err.status : 500;
-					res.setHeader("Content-Type", "application/json");
-					res.end(JSON.stringify({ error: message }));
+					if (!res.headersSent) {
+						res.statusCode = 500;
+						res.setHeader("Content-Type", "application/json");
+						res.end(JSON.stringify({ error: "Internal Server Error" }));
+					}
 				}
 			});
 		}
@@ -267,51 +297,50 @@ function generateClientStub(mod, rpcBase) {
 	}
 	return lines.join("\n");
 }
-var RpcError = class extends Error {
-	constructor(message, status) {
-		super(message);
-		this.status = status;
-	}
-};
-/**
-* dev サーバーで RPC リクエストを処理する。
-*
-* URL: POST /rpc/{routePrefix}/{functionName}
-* Body: JSON（関数の引数）
-*
-* schema.ts に Zod スキーマがあれば、引数をバリデーションしてから関数を実行する。
-*/
-async function handleRpcRequest(server, req, rpcBase, modules) {
-	if (req.method !== "POST") throw new RpcError("Method not allowed", 405);
-	const rpcPath = req.url.slice(rpcBase.length);
-	const lastSlash = rpcPath.lastIndexOf("/");
-	if (lastSlash === -1) throw new RpcError("Invalid RPC path", 400);
-	const routePrefix = rpcPath.slice(0, lastSlash) || "";
-	const functionName = decodeURIComponent(rpcPath.slice(lastSlash + 1));
-	const mod = modules.find((m) => m.routePrefix === routePrefix);
-	if (!mod) throw new RpcError(`Module not found: ${routePrefix}`, 404);
-	const fnDef = mod.functions.find((f) => f.name === functionName);
-	if (!fnDef) throw new RpcError(`Function not found: ${functionName}`, 404);
-	const fn = (await server.ssrLoadModule(mod.filePath))[functionName];
-	if (typeof fn !== "function") throw new RpcError(`${functionName} is not a function`, 500);
-	const body = await readBody(req);
-	let args;
-	try {
-		args = body ? JSON.parse(body) : [];
-	} catch {
-		throw new RpcError("Invalid JSON in request body", 400);
-	}
-	if (!Array.isArray(args)) throw new RpcError("Request body must be a JSON array", 400);
-	return fn(...await validateArgs(args, fnDef, mod, server));
+function createDevRpcApp(server, rpcBase, getModules) {
+	const app = new Hono();
+	app.post(`${rpcBase}/*`, async (c) => {
+		const rpcPath = c.req.path.slice(rpcBase.length);
+		const lastSlash = rpcPath.lastIndexOf("/");
+		if (lastSlash === -1) return c.json({ error: "Invalid RPC path" }, 400);
+		const routePrefix = rpcPath.slice(0, lastSlash) || "";
+		const functionName = decodeURIComponent(rpcPath.slice(lastSlash + 1));
+		const mod = getModules().find((m) => m.routePrefix === routePrefix);
+		if (!mod) return c.json({ error: `Module not found: ${routePrefix}` }, 404);
+		const fnDef = mod.functions.find((f) => f.name === functionName);
+		if (!fnDef) return c.json({ error: `Function not found: ${functionName}` }, 404);
+		const fn = (await server.ssrLoadModule(mod.filePath))[functionName];
+		if (typeof fn !== "function") return c.json({ error: `${functionName} is not a function` }, 500);
+		if (Number(c.req.header("content-length") ?? 0) > 1048576) return c.json({ error: "Payload too large" }, 413);
+		const body = await c.req.text();
+		if (body.length > 1048576) return c.json({ error: "Payload too large" }, 413);
+		let args;
+		try {
+			args = body ? JSON.parse(body) : [];
+		} catch {
+			return c.json({ error: "Invalid JSON in request body" }, 400);
+		}
+		if (!Array.isArray(args)) return c.json({ error: "Request body must be a JSON array" }, 400);
+		let validatedArgs;
+		try {
+			validatedArgs = await validateArgs(args, fnDef, mod, (p) => server.ssrLoadModule(p));
+		} catch (err) {
+			if (err instanceof RpcValidationError) return c.json({ error: err.message }, 400);
+			throw err;
+		}
+		const result = await contextStorage.run(c, () => fn(...validatedArgs));
+		return c.json(result ?? null);
+	});
+	app.all(`${rpcBase}/*`, (c) => c.json({ error: "Method not allowed" }, 405));
+	return app;
 }
 /**
 * 各引数を対応する Zod スキーマでバリデーションする。
 * スキーマが紐付いていない引数はそのまま通す。
-* 引数が不足していてもスキーマがあればバリデーションを実行する（Zod が required エラーを出す）。
 */
-async function validateArgs(args, fnDef, mod, server) {
+async function validateArgs(args, fnDef, mod, loadModule) {
 	if (!fnDef.params.some((p) => p.schemaName) || !mod.schemaFilePath) return args;
-	const schemaModule = await server.ssrLoadModule(mod.schemaFilePath);
+	const schemaModule = await loadModule(mod.schemaFilePath);
 	const validated = [...args];
 	for (let i = 0; i < fnDef.params.length; i++) {
 		const param = fnDef.params[i];
@@ -321,13 +350,18 @@ async function validateArgs(args, fnDef, mod, server) {
 			const result = schema.safeParse(args[i]);
 			if (!result.success) {
 				const issues = result.error.issues.map((issue) => issue.path.length > 0 ? `${issue.path.join(".")}: ${issue.message}` : issue.message).join(", ");
-				throw new RpcError(`Validation error on "${param.name}": ${issues}`, 400);
+				throw new RpcValidationError(`Validation error on "${param.name}": ${issues}`);
 			}
 			validated[i] = result.data;
 		}
 	}
 	return validated;
 }
+var RpcValidationError = class extends Error {
+	constructor(message) {
+		super(message);
+	}
+};
 /**
 * 本番用 Hono アプリのコードを生成する。
 *
@@ -339,6 +373,7 @@ async function validateArgs(args, fnDef, mod, server) {
 function generateHonoApp(modules, root, rpcBase) {
 	const lines = [];
 	lines.push(`import { Hono } from "hono";`);
+	lines.push(`import { contextStorage } from "orbit-rpc";`);
 	lines.push(``);
 	const validModules = [];
 	for (const [i, mod] of modules.entries()) {
@@ -361,6 +396,8 @@ function generateHonoApp(modules, root, rpcBase) {
 		const endpoint = `${rpcBase}${mod.routePrefix}/${fn.name}`;
 		lines.push(`app.post("${endpoint}", async (c) => {`);
 		lines.push(`  try {`);
+		lines.push(`    const cl = Number(c.req.header("content-length") ?? 0);`);
+		lines.push(`    if (cl > 1048576) return c.json({ error: "Payload too large" }, 413);`);
 		lines.push(`    const body = await c.req.text();`);
 		lines.push(`    if (body.length > 1048576) return c.json({ error: "Payload too large" }, 413);`);
 		lines.push(`    let args;`);
@@ -376,7 +413,7 @@ function generateHonoApp(modules, root, rpcBase) {
 				lines.push(`    }`);
 			}
 		}
-		lines.push(`    const result = await mod${i}.${fn.name}(...args);`);
+		lines.push(`    const result = await contextStorage.run(c, () => mod${i}.${fn.name}(...args));`);
 		lines.push(`    return c.json(result ?? null);`);
 		lines.push(`  } catch (err) {`);
 		lines.push(`    console.error(err);`);
@@ -388,23 +425,41 @@ function generateHonoApp(modules, root, rpcBase) {
 	lines.push(`export default app;`);
 	return lines.join("\n");
 }
-const MAX_BODY_SIZE = 1024 * 1024;
-function readBody(req) {
-	return new Promise((resolve, reject) => {
-		let data = "";
-		let size = 0;
-		req.on("data", (chunk) => {
-			size += chunk.length;
-			if (size > MAX_BODY_SIZE) {
-				req.destroy?.();
-				reject(new RpcError("Request body too large", 413));
-				return;
-			}
-			data += chunk.toString();
-		});
-		req.on("end", () => resolve(data));
-		req.on("error", reject);
+/**
+* Node.js IncomingMessage → Web Standard Request に変換する。
+* Vite の connect middleware から Hono に橋渡しするために使う。
+*/
+function toWebRequest(nodeReq) {
+	const url = new URL(nodeReq.url, `http://${nodeReq.headers.host || "localhost"}`);
+	const headers = new Headers();
+	for (const [key, value] of Object.entries(nodeReq.headers)) {
+		if (value === void 0) continue;
+		if (Array.isArray(value)) for (const v of value) headers.append(key, v);
+		else headers.set(key, value);
+	}
+	const hasBody = nodeReq.method === "POST" || nodeReq.method === "PUT" || nodeReq.method === "PATCH";
+	return new Request(url, {
+		method: nodeReq.method,
+		headers,
+		...hasBody ? {
+			body: Readable.toWeb(nodeReq),
+			duplex: "half"
+		} : {}
 	});
 }
+/**
+* Web Standard Response → Node.js ServerResponse に書き戻す。
+*/
+async function writeWebResponse(webRes, nodeRes) {
+	nodeRes.statusCode = webRes.status;
+	for (const [key, value] of webRes.headers) {
+		if (key.toLowerCase() === "set-cookie") continue;
+		nodeRes.setHeader(key, value);
+	}
+	const cookies = webRes.headers.getSetCookie?.();
+	if (cookies?.length) nodeRes.setHeader("set-cookie", cookies);
+	const body = await webRes.arrayBuffer();
+	nodeRes.end(Buffer.from(body));
+}
 //#endregion
-export { orbitRpc };
+export { contextStorage, getContext, orbitRpc };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "orbit-rpc",
-  "version": "1.0.0",
+  "version": "1.2.0",
   "description": "RPC layer for Orbit — auto-converts server.ts functions to Hono routes",
   "keywords": [
     "fullstack",