agent-relay-server 0.4.8 → 0.4.9

package/codex/plugin/.codex-plugin/plugin.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-relay",
-  "version": "0.4.8",
+  "version": "0.4.9",
   "description": "Agent Relay integration for Codex sessions",
   "author": {
     "name": "Edin Mujkanovic"

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-relay-server",
-  "version": "0.4.8",
+  "version": "0.4.9",
   "description": "Lightweight HTTP message relay for inter-agent communication across machines",
   "module": "src/index.ts",
   "type": "module",

package/src/index.ts

@@ -7,7 +7,6 @@ import {
   REAP_INTERVAL_MS,
   STALE_TTL_MS,
   OFFLINE_PRUNE_MS,
-  MAX_BODY_BYTES,
   DAY_MS,
   VERSION,
 } from "./config";
@@ -57,68 +56,66 @@ function startServer(): void {
     if (pruned > 0) console.log(`pruned ${pruned} old message(s)`);
   }, DAY_MS);
 
-  const publicDir = resolve(import.meta.dir, "../public");
-  const publicDirPrefix = publicDir + sep;
-
   Bun.serve({
     port: PORT,
     hostname: HOST,
-    async fetch(req) {
-      const url = new URL(req.url);
+    fetch: createFetchHandler({ logRequests: LOG_REQUESTS }),
+  });
 
-      if (req.method === "OPTIONS") {
-        return corsPreflight(req);
-      }
-      if (!isOriginAllowed(req)) {
-        return Response.json({ error: "origin not allowed" }, { status: 403 });
-      }
+  console.log(`agent-relay ${VERSION} running on http://${HOST}:${PORT}`);
+}
 
-      // Body size guard for write methods
-      if (req.method === "POST" || req.method === "PATCH" || req.method === "PUT") {
-        const len = Number(req.headers.get("content-length") ?? 0);
-        if (len > MAX_BODY_BYTES) {
-          return Response.json(
-            { error: `request body exceeds ${MAX_BODY_BYTES} bytes` },
-            { status: 413 },
-          );
-        }
-      }
+export function createFetchHandler(
+  opts: { publicDir?: string; logRequests?: boolean } = {},
+): (req: Request) => Promise<Response> {
+  const publicDir = opts.publicDir ?? resolve(import.meta.dir, "../public");
+  const publicDirPrefix = publicDir + sep;
+  const logRequests = opts.logRequests ?? false;
 
-      // API routes
-      const matched = matchRoute(req.method, url.pathname);
-      if (matched) {
-        const integrationAuth = getIntegrationAuth(req);
-        if (!isAuthorized(req)) {
-          if (!integrationAuth || !url.pathname.startsWith("/api/integrations/")) {
-            return unauthorized(req);
-          }
-        }
-        const response = await matched.handler(req, matched.params);
-        applyCors(req, response);
-        if (LOG_REQUESTS && url.pathname.startsWith("/api/")) {
-          console.log(`${req.method} ${url.pathname} → ${response.status}`);
+  return async function fetch(req: Request): Promise<Response> {
+    const url = new URL(req.url);
+
+    if (req.method === "OPTIONS") {
+      return corsPreflight(req);
+    }
+    if (!isOriginAllowed(req)) {
+      return Response.json({ error: "origin not allowed" }, { status: 403 });
+    }
+
+    // API routes
+    const matched = matchRoute(req.method, url.pathname);
+    if (matched) {
+      const integrationAuth = getIntegrationAuth(req);
+      if (!isAuthorized(req)) {
+        if (!integrationAuth || !url.pathname.startsWith("/api/integrations/")) {
+          return unauthorized(req);
         }
-        return response;
       }
-
-      // Dashboard — serve static files, rejecting path traversal and directory requests
-      let requested = url.pathname === "/" ? "/index.html" : url.pathname;
-      if (requested.endsWith("/")) requested += "index.html";
-      const resolved = resolve(publicDir, `.${requested}`);
-      if (!resolved.startsWith(publicDirPrefix)) {
-        return Response.json({ error: "not found" }, { status: 404 });
+      const response = await matched.handler(req, matched.params);
+      applyCors(req, response);
+      if (logRequests && url.pathname.startsWith("/api/")) {
+        console.log(`${req.method} ${url.pathname} → ${response.status}`);
       }
-      const file = Bun.file(resolved);
-      if (await file.exists()) return new Response(file);
+      return response;
+    }
 
+    // Dashboard — serve static files, rejecting path traversal and directory requests
+    let requested = url.pathname === "/" ? "/index.html" : url.pathname;
+    if (requested.endsWith("/")) requested += "index.html";
+    const resolved = resolve(publicDir, `.${requested}`);
+    if (!resolved.startsWith(publicDirPrefix)) {
       return Response.json({ error: "not found" }, { status: 404 });
-    },
-  });
+    }
+    const file = Bun.file(resolved);
+    if (await file.exists()) return new Response(file);
 
-  console.log(`agent-relay ${VERSION} running on http://${HOST}:${PORT}`);
+    return Response.json({ error: "not found" }, { status: 404 });
+  };
 }
 
-main().catch((error) => {
-  console.error(error instanceof Error ? error.message : String(error));
-  process.exit(1);
-});
+if (import.meta.main) {
+  main().catch((error) => {
+    console.error(error instanceof Error ? error.message : String(error));
+    process.exit(1);
+  });
+}