orbit-rpc 0.0.1 → 0.2.0

package/dist/index.d.mts

@@ -0,0 +1,12 @@
+import { Plugin } from "vite";
+
+//#region src/plugin.d.ts
+interface OrbitRpcConfig {
+  /** routes ディレクトリのパス（デフォルト: "src/routes"） */
+  routesDir?: string;
+  /** RPC エンドポイントのプレフィックス（デフォルト: "/rpc"） */
+  rpcBase?: string;
+}
+declare function orbitRpc(config?: OrbitRpcConfig): Plugin[];
+//#endregion
+export { type OrbitRpcConfig, orbitRpc };

package/dist/index.mjs

@@ -0,0 +1,410 @@
+import path from "node:path";
+import fs from "node:fs";
+//#region src/scanner.ts
+/**
+* routes ディレクトリから server.ts ファイルをスキャンし、
+* エクスポートされた関数名と引数の型情報を抽出する。
+* 隣接する schema.ts があれば Zod スキーマとの対応も解決する。
+*/
+async function scanServerModules(root, routesDir) {
+	const absoluteRoutesDir = path.resolve(root, routesDir);
+	if (!fs.existsSync(absoluteRoutesDir)) return [];
+	const modules = [];
+	await walk(absoluteRoutesDir, absoluteRoutesDir, modules);
+	return modules;
+}
+async function walk(dir, routesRoot, modules) {
+	const entries = await fs.promises.readdir(dir, { withFileTypes: true });
+	const serverFile = entries.find((e) => e.isFile() && /^server\.ts$/.test(e.name));
+	if (serverFile) {
+		const filePath = path.join(dir, serverFile.name);
+		const routePrefix = dirToRoutePrefix(path.relative(routesRoot, dir));
+		const schemaPath = path.join(dir, "schema.ts");
+		const hasSchema = entries.some((e) => e.isFile() && e.name === "schema.ts");
+		const functions = extractExportedFunctions(filePath, hasSchema ? extractSchemaMap(schemaPath) : /* @__PURE__ */ new Map());
+		if (functions.length > 0) modules.push({
+			filePath,
+			schemaFilePath: hasSchema ? schemaPath : void 0,
+			routePrefix,
+			functions
+		});
+	}
+	for (const entry of entries) if (entry.isDirectory() && !entry.name.startsWith("_")) await walk(path.join(dir, entry.name), routesRoot, modules);
+}
+/**
+* schema.ts から「型名 → Zod スキーマ名」のマップを抽出する。
+*
+* 認識するパターン:
+*   export type TaskForm = z.infer<typeof taskFormSchema>;
+*   → Map { "TaskForm" => "taskFormSchema" }
+*/
+function extractSchemaMap(filePath) {
+	const content = fs.readFileSync(filePath, "utf-8");
+	const map = /* @__PURE__ */ new Map();
+	for (const match of content.matchAll(/export\s+type\s+(\w+)\s*=\s*z\.infer\s*<\s*typeof\s+(\w+)\s*>/g)) map.set(match[1], match[2]);
+	return map;
+}
+/**
+* server.ts から export された関数を抽出する。
+* 各引数の型名を解析し、schemaMap にマッチすれば schemaName を紐付ける。
+*/
+function extractExportedFunctions(filePath, schemaMap) {
+	const content = fs.readFileSync(filePath, "utf-8");
+	const functions = [];
+	for (const match of content.matchAll(/export\s+(?:async\s+)?function\s+(\w+)\s*\(/g)) {
+		const name = match[1];
+		if (functions.some((f) => f.name === name)) continue;
+		const params = parseParams(extractBalancedParens(content, match.index + match[0].length - 1), schemaMap);
+		functions.push({
+			name,
+			params
+		});
+	}
+	for (const match of content.matchAll(/export\s+const\s+(\w+)\s*=\s*(?:async\s+)?\(/g)) {
+		const name = match[1];
+		if (functions.some((f) => f.name === name)) continue;
+		const params = parseParams(extractBalancedParens(content, match.index + match[0].length - 1), schemaMap);
+		functions.push({
+			name,
+			params
+		});
+	}
+	return functions;
+}
+/**
+* 開き括弧の位置から対応する閉じ括弧までの中身を返す。
+* ネストされた括弧（デフォルト引数内の関数呼び出し等）に対応。
+*/
+function extractBalancedParens(content, openIndex) {
+	if (content[openIndex] !== "(") return "";
+	let depth = 0;
+	for (let i = openIndex; i < content.length; i++) if (content[i] === "(") depth++;
+	else if (content[i] === ")") {
+		depth--;
+		if (depth === 0) return content.slice(openIndex + 1, i);
+	}
+	return "";
+}
+/**
+* 関数の引数文字列をパースして FunctionParam 配列にする。
+* デフォルト値内のカンマやネストに対応するため、括弧の深さを追跡する。
+*
+* 例: "input: TaskForm, signal?: AbortSignal"
+* → [{ name: "input", typeName: "TaskForm", schemaName: "taskFormSchema" },
+*    { name: "signal", typeName: "AbortSignal" }]
+*/
+function parseParams(paramsStr, schemaMap) {
+	if (!paramsStr.trim()) return [];
+	const parts = splitTopLevelCommas(paramsStr);
+	const params = [];
+	for (const part of parts) {
+		const withoutDefault = part.replace(/\s*=\s*[\s\S]*$/, "").trim();
+		if (!withoutDefault) continue;
+		const paramMatch = withoutDefault.match(/^(\w+)\??\s*:\s*(\w+)/);
+		if (paramMatch) {
+			const name = paramMatch[1];
+			const typeName = paramMatch[2];
+			const schemaName = schemaMap.get(typeName);
+			params.push({
+				name,
+				typeName,
+				schemaName
+			});
+		} else {
+			const nameOnly = withoutDefault.match(/^(\w+)/);
+			if (nameOnly) params.push({ name: nameOnly[1] });
+		}
+	}
+	return params;
+}
+/**
+* トップレベルのカンマで文字列を分割する。
+* 括弧 (), {}, <> 内のカンマは分割しない。
+*/
+function splitTopLevelCommas(str) {
+	const parts = [];
+	let current = "";
+	let depth = 0;
+	for (const ch of str) if (ch === "(" || ch === "{" || ch === "<") {
+		depth++;
+		current += ch;
+	} else if (ch === ")" || ch === "}" || ch === ">") {
+		depth--;
+		current += ch;
+	} else if (ch === "," && depth === 0) {
+		parts.push(current);
+		current = "";
+	} else current += ch;
+	if (current.trim()) parts.push(current);
+	return parts;
+}
+/**
+* ディレクトリの相対パスをルートプレフィックスに変換する。
+*  ""           → ""
+*  "tasks"      → "/tasks"
+*  "users/[id]" → "/users/:id"
+*/
+function dirToRoutePrefix(relativePath) {
+	if (relativePath === "") return "";
+	const segments = relativePath.split(path.sep).filter((seg) => !/^\(.+\)$/.test(seg)).map((seg) => {
+		const match = seg.match(/^\[(.+)]$/);
+		return match ? `:${match[1]}` : seg;
+	});
+	return segments.length === 0 ? "" : `/${segments.join("/")}`;
+}
+//#endregion
+//#region src/plugin.ts
+/**
+* Orbit RPC Vite プラグイン。
+*
+* 2つの仕事をする:
+* 1. クライアント側: server.ts の import を HTTP fetch スタブに差し替え
+* 2. dev サーバー: /rpc/* リクエストを受けて server.ts の関数を実行
+*
+* schema.ts に Zod スキーマがあれば、自動でバリデーションを適用する。
+*/
+const VIRTUAL_SERVER_ID = "virtual:orbit-rpc/server";
+const RESOLVED_VIRTUAL_SERVER_ID = `\0${VIRTUAL_SERVER_ID}`;
+function orbitRpc(config = {}) {
+	const routesDir = config.routesDir ?? "src/routes";
+	const rpcBase = config.rpcBase ?? "/rpc";
+	let root;
+	let serverModules = [];
+	let scanned = false;
+	return [{
+		name: "orbit-rpc:transform",
+		configResolved(resolvedConfig) {
+			root = resolvedConfig.root;
+		},
+		async buildStart() {
+			serverModules = await scanServerModules(root, routesDir);
+			scanned = true;
+		},
+		resolveId(id) {
+			if (id === VIRTUAL_SERVER_ID) return RESOLVED_VIRTUAL_SERVER_ID;
+		},
+		async load(id) {
+			if (id === RESOLVED_VIRTUAL_SERVER_ID) {
+				if (!scanned) serverModules = await scanServerModules(root, routesDir);
+				return generateHonoApp(serverModules, root, rpcBase);
+			}
+		},
+		async transform(code, id) {
+			if (!id.endsWith("/server.ts")) return null;
+			const routesPath = path.resolve(root, routesDir);
+			if (!id.startsWith(routesPath)) return null;
+			if (this.environment?.name === "ssr") return null;
+			const mod = serverModules.find((m) => m.filePath === id);
+			if (!mod) {
+				serverModules = await scanServerModules(root, routesDir);
+				const freshMod = serverModules.find((m) => m.filePath === id);
+				if (!freshMod) return null;
+				return generateClientStub(freshMod, rpcBase);
+			}
+			return generateClientStub(mod, rpcBase);
+		}
+	}, {
+		name: "orbit-rpc:dev-server",
+		configureServer(server) {
+			const routesPath = path.resolve(root, routesDir);
+			const onFileChange = async (file) => {
+				if (!file.startsWith(routesPath)) return;
+				if (!file.endsWith("/server.ts") && !file.endsWith("/schema.ts")) return;
+				serverModules = await scanServerModules(root, routesDir);
+			};
+			server.watcher.on("add", onFileChange);
+			server.watcher.on("change", onFileChange);
+			server.watcher.on("unlink", onFileChange);
+			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));
+				} 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 }));
+				}
+			});
+		}
+	}];
+}
+/**
+* server.ts の中身を RPC スタブに差し替える。
+*
+* 例: routePrefix="/tasks", rpcBase="/rpc"
+*   export async function getTasks(signal) { ... }
+* → export async function getTasks(signal) {
+*     const res = await fetch("/rpc/tasks/getTasks", { method: "POST", signal });
+*     ...
+*   }
+*/
+function generateClientStub(mod, rpcBase) {
+	const lines = [];
+	for (const fn of mod.functions) {
+		const endpoint = `${rpcBase}${mod.routePrefix}/${fn.name}`;
+		lines.push(`export async function ${fn.name}(...args) {`);
+		lines.push(`  const signal = args[args.length - 1] instanceof AbortSignal ? args.pop() : undefined;`);
+		lines.push(`  const hasArgs = args.length > 0;`);
+		lines.push(`  const res = await fetch("${endpoint}", {`);
+		lines.push(`    method: "POST",`);
+		lines.push(`    signal,`);
+		lines.push(`    ...(hasArgs ? {`);
+		lines.push(`      headers: { "Content-Type": "application/json" },`);
+		lines.push(`      body: JSON.stringify(args),`);
+		lines.push(`    } : {}),`);
+		lines.push(`  });`);
+		lines.push(`  if (!res.ok) {`);
+		lines.push(`    const body = await res.json().catch(() => ({}));`);
+		lines.push(`    throw new Error(body.error || \`RPC error: \${res.status}\`);`);
+		lines.push(`  }`);
+		lines.push(`  const text = await res.text();`);
+		lines.push(`  return text ? JSON.parse(text) : undefined;`);
+		lines.push(`}`);
+		lines.push(``);
+	}
+	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));
+}
+/**
+* 各引数を対応する Zod スキーマでバリデーションする。
+* スキーマが紐付いていない引数はそのまま通す。
+* 引数が不足していてもスキーマがあればバリデーションを実行する（Zod が required エラーを出す）。
+*/
+async function validateArgs(args, fnDef, mod, server) {
+	if (!fnDef.params.some((p) => p.schemaName) || !mod.schemaFilePath) return args;
+	const schemaModule = await server.ssrLoadModule(mod.schemaFilePath);
+	const validated = [...args];
+	for (let i = 0; i < fnDef.params.length; i++) {
+		const param = fnDef.params[i];
+		if (!param.schemaName) continue;
+		const schema = schemaModule[param.schemaName];
+		if (schema && typeof schema.safeParse === "function") {
+			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);
+			}
+			validated[i] = result.data;
+		}
+	}
+	return validated;
+}
+/**
+* 本番用 Hono アプリのコードを生成する。
+*
+* import app from "virtual:orbit-rpc/server" で使える。
+* Cloudflare Workers の場合は export default app; するだけ。
+*
+* schema.ts に Zod スキーマがあれば、.parse() によるバリデーションを含む。
+*/
+function generateHonoApp(modules, root, rpcBase) {
+	const lines = [];
+	lines.push(`import { Hono } from "hono";`);
+	lines.push(``);
+	const validModules = [];
+	for (const [i, mod] of modules.entries()) {
+		if (mod.routePrefix.includes(":")) {
+			console.warn(`[orbit-rpc] Dynamic route prefix "${mod.routePrefix}" is not supported for RPC. Skipping.`);
+			continue;
+		}
+		validModules.push([i, mod]);
+		const importPath = mod.filePath.split(path.sep).join("/");
+		lines.push(`import * as mod${i} from "${importPath}";`);
+		if (mod.schemaFilePath) {
+			const schemaPath = mod.schemaFilePath.split(path.sep).join("/");
+			lines.push(`import * as schema${i} from "${schemaPath}";`);
+		}
+	}
+	lines.push(``);
+	lines.push(`const app = new Hono();`);
+	lines.push(``);
+	for (const [i, mod] of validModules) for (const fn of mod.functions) {
+		const endpoint = `${rpcBase}${mod.routePrefix}/${fn.name}`;
+		lines.push(`app.post("${endpoint}", async (c) => {`);
+		lines.push(`  try {`);
+		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;`);
+		lines.push(`    try { args = body ? JSON.parse(body) : []; } catch { return c.json({ error: "Invalid JSON in request body" }, 400); }`);
+		lines.push(`    if (!Array.isArray(args)) return c.json({ error: "Request body must be a JSON array" }, 400);`);
+		if (mod.schemaFilePath) for (let pi = 0; pi < fn.params.length; pi++) {
+			const param = fn.params[pi];
+			if (param.schemaName) {
+				lines.push(`    {`);
+				lines.push(`      const r = schema${i}.${param.schemaName}.safeParse(args[${pi}]);`);
+				lines.push(`      if (!r.success) { const msg = r.error.issues.map(i => i.path.length ? i.path.join(".") + ": " + i.message : i.message).join(", "); return c.json({ error: "Validation error on \\"${param.name}\\": " + msg }, 400); }`);
+				lines.push(`      args[${pi}] = r.data;`);
+				lines.push(`    }`);
+			}
+		}
+		lines.push(`    const result = await mod${i}.${fn.name}(...args);`);
+		lines.push(`    return c.json(result ?? null);`);
+		lines.push(`  } catch (err) {`);
+		lines.push(`    console.error(err);`);
+		lines.push(`    return c.json({ error: "Internal Server Error" }, 500);`);
+		lines.push(`  }`);
+		lines.push(`});`);
+		lines.push(``);
+	}
+	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);
+	});
+}
+//#endregion
+export { orbitRpc };

package/package.json

@@ -1,11 +1,51 @@
 {
   "name": "orbit-rpc",
-  "version": "0.0.1",
+  "version": "0.2.0",
   "description": "RPC layer for Orbit — auto-converts server.ts functions to Hono routes",
+  "keywords": [
+    "fullstack",
+    "hono",
+    "react",
+    "rpc",
+    "vite"
+  ],
   "license": "MIT",
-  "type": "module",
-  "main": "index.js",
+  "author": "ashunar0",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/ashunar0/orbit-router",
+    "directory": "packages/orbit-rpc"
+  },
   "files": [
-    "index.js"
-  ]
+    "dist"
+  ],
+  "type": "module",
+  "types": "./dist/index.d.mts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.mts",
+      "import": "./dist/index.mjs"
+    },
+    "./package.json": "./package.json"
+  },
+  "dependencies": {
+    "hono": "^4.7.11"
+  },
+  "devDependencies": {
+    "@types/node": "^25.5.0",
+    "bumpp": "^11.0.1",
+    "typescript": "^5.9.3",
+    "vite": "npm:@voidzero-dev/vite-plus-core@0.1.12",
+    "vite-plus": "0.1.12"
+  },
+  "peerDependencies": {
+    "vite": ">=6.0.0"
+  },
+  "scripts": {
+    "build": "vp pack",
+    "dev": "vp pack --watch",
+    "test": "vp test",
+    "typecheck": "tsc --noEmit",
+    "release": "bumpp"
+  }
 }

package/index.js

@@ -1 +0,0 @@
-