poe-code 4.0.21 → 4.0.22

package/packages/toolcraft/dist/http.js

@@ -2,6 +2,7 @@ import "./node-require-shim.js";
 import { createHttpServer } from "tiny-http-mcp-server/server";
 import { createMCPServerForTransport } from "./mcp.js";
 import { enableSourceMaps } from "./stack-trim.js";
+import { isHostedOAuthConfiguration, prepareHostedOAuthRuntime } from "./http-hosted-oauth.js";
 function toVerifiedAccessToken(token, issuer, verified) {
     return {
         token,
@@ -62,7 +63,8 @@ function createTransportOptions(options, serverOptions) {
         requestIdGenerator: options.requestIdGenerator,
         observability: options.observability,
         trustedProxy: options.trustedProxy,
-        oauth: options.oauth
+        oauth: options.oauth,
+        requestHandler: options.requestHandler
     };
 }
 function createListenOptions(options) {
@@ -77,20 +79,50 @@ function createListenOptions(options) {
     };
 }
 export async function createHTTPMCPServer(roots, options) {
+    let resolvedOptions = options;
+    let hostedMcpPath;
+    if (isHostedOAuthConfiguration(options.oauth)) {
+        const runtime = await prepareHostedOAuthRuntime(options.oauth);
+        const applicationServices = options.requestServices;
+        hostedMcpPath = runtime.mcpPath;
+        resolvedOptions = {
+            ...options,
+            oauth: runtime.oauth,
+            sessionIdGenerator: undefined,
+            enableJsonResponse: true,
+            requestServices: async (context) => {
+                const subject = context.auth?.subject;
+                if (subject === undefined)
+                    throw new Error("Hosted OAuth request is missing a verified subject.");
+                return {
+                    ...(applicationServices === undefined ? {} : await applicationServices(context)),
+                    ...(await runtime.requestServices(subject))
+                };
+            },
+            requestHandler: runtime.requestHandler
+        };
+    }
     let server;
-    await createMCPServerForTransport(roots, options, {
+    await createMCPServerForTransport(roots, resolvedOptions, {
         createServer(serverOptions) {
-            server = createHttpServer(createTransportOptions(options, serverOptions));
+            server = createHttpServer(createTransportOptions(resolvedOptions, serverOptions));
             return server;
         },
         getRequestContext() {
             return server?.getRequestContext();
         },
-        requestServices: options.requestServices
+        requestServices: resolvedOptions.requestServices
     });
     if (server === undefined) {
         throw new Error("Toolcraft HTTP MCP server was not created.");
     }
+    if (hostedMcpPath !== undefined) {
+        const listenHttp = server.listenHttp.bind(server);
+        server.listenHttp = (listenOptions = {}) => listenHttp({
+            path: hostedMcpPath,
+            ...listenOptions
+        });
+    }
     return server;
 }
 export async function runHTTPMCP(roots, options) {