@wakeshift/sdk 0.5.2 → 0.6.0

package/dist/toolcall.js

@@ -18,11 +18,12 @@
 import { z } from "zod";
 import * as z3 from "zod/v3";
 import * as z4 from "zod/v4";
-import { getObjectShape, isZ4Schema, normalizeObjectSchema, safeParseAsync } from "@modelcontextprotocol/sdk/server/zod-compat.js";
+import * as z4mini from "zod/v4-mini";
+import { getObjectShape, isSchema, isZ4Schema, normalizeObjectSchema } from "./zod.js";
 import { WakeshiftError } from "./errors.js";
 import { EXTENSION_KEY, dialectOf } from "./spec.js";
 import { takesTasks } from "./discover.js";
-import { packCall } from "./wakeshift.js";
+import { packCall, userAgentOf } from "./wakeshift.js";
 // Wakeshift's three arguments (wakeshift.ts's SESSION_ARG, HARNESS_ARG and
 // CONVERSATION_ARG, repeated: that module imports this one).
 const SESSION_ARG = "session_key";
@@ -66,12 +67,26 @@ const byVersion = {
     },
 };
 // fieldsFor is Wakeshift's fields in the zod API shape's own fields use
-// (the default import's for an empty shape).
-function fieldsFor(shape, kind) {
+// (for an empty shape, the default import's, or zod 4's for the MCP SDK
+// 2.x, which takes only zod 4).
+function fieldsFor(shape, kind, host) {
     const first = Object.values(shape ?? {})[0];
-    const fields = !first ? (kind === "inputs" ? wakeInputs : wakeOutput) : byVersion[isZ4Schema(first) ? "v4" : "v3"][kind];
+    const version = first ? (isZ4Schema(first) ? "v4" : "v3") : host?.version === 2 ? "v4" : null;
+    const fields = version ? byVersion[version][kind] : kind === "inputs" ? wakeInputs : wakeOutput;
     return fields;
 }
+/** @internal Wakeshift's three arguments as attach's own tools take them, in the zod the host's MCP SDK takes. */
+export function wakeInputsFor(host) {
+    return host.version === 2 ? byVersion.v4.inputs : wakeInputs;
+}
+// withFields is an object schema with fields added, in the schema's own zod
+// API (zod 3, zod 4 classic or mini).
+function withFields(schema, fields) {
+    const s = schema;
+    if (typeof s.extend === "function")
+        return s.extend(fields);
+    return z4mini.extend(schema, fields);
+}
 const WAKE_KEYS = [SESSION_ARG, HARNESS_ARG, CONVERSATION_ARG];
 /**
  * @internal The tools/call _meta keys a harness names its conversation in
@@ -209,12 +224,6 @@ function sameJSON(a, b) {
 function detailOf(err) {
     return err instanceof WakeshiftError ? err.detail : String(err?.message ?? err);
 }
-function registeredTool(server, name) {
-    const tools = server._registeredTools;
-    if (!tools || typeof tools !== "object")
-        return null; // an MCP SDK that keeps them elsewhere
-    return tools[name];
-}
 /** One tools/call of a wrapped tool: what its handler subscribed, for the block and the rollback. */
 class CallScope {
     at;
@@ -252,17 +261,18 @@ class CallScope {
     // User-Agent, the agent's harness and conversation_id arguments, the
     // conversation's session, and who's calling.
     wire(session = this.session) {
-        const meta = this.extra._meta;
+        const req = this.at.host.req(this.extra);
+        const meta = req.meta;
         return packCall({
             meta,
             clientName: this.at.clientOf(meta, this.identified),
             user: this.user,
             session: session ?? undefined,
-            userAgent: userAgentOf(this.extra),
+            userAgent: userAgentOf(req),
             harness: this.args.harness,
             conversation: this.args.conversation,
             tool: this.tool,
-            tasks: takesTasks(meta, this.extra),
+            tasks: takesTasks(meta, req),
         });
     }
     // request runs one request of the call to Wakeshift, after any before it
@@ -415,19 +425,12 @@ class CallScope {
 function recognizedOf(a) {
     return { harness: a?.harness ?? null, via: a?.via ?? null, claimed: a?.claimed ?? null };
 }
-function headerOf(extra, name) {
-    const h = extra?.requestInfo?.headers;
-    if (!h)
-        return "";
-    const v = h[name] ?? h[name.toLowerCase()];
-    return (Array.isArray(v) ? v[0] : v) ?? "";
-}
-function userAgentOf(extra) {
-    return headerOf(extra, "user-agent") || headerOf(extra, "User-Agent");
-}
 function hasKey(o, key) {
     return !!o && typeof o === "object" && !Array.isArray(o) && o[key] !== undefined;
 }
+function isRecord(v) {
+    return !!v && typeof v === "object" && !Array.isArray(v);
+}
 /**
  * @internal Whether a call comes from a client that subscribes to the app's
  * events itself, on stateless servers too (which never see initialize), in
@@ -438,15 +441,16 @@ function hasKey(o, key) {
  * _meta's). conversation is ChatGPT's conversation, when it names one.
  */
 export function nativeOf(at, extra, identified) {
-    const meta = (extra._meta ?? {});
+    const req = at.host.req(extra);
+    const meta = (req.meta ?? {});
     const conversation = typeof meta["openai/session"] === "string" && meta["openai/session"] ? meta["openai/session"] : null;
     if (at.events.native?.(extra))
         return { native: true, conversation };
     if (conversation)
         return { native: true, conversation };
-    if (dialectOf(at.clientOf(meta, identified), userAgentOf(extra)) === "chatgpt")
+    if (dialectOf(at.clientOf(meta, identified), userAgentOf(req)) === "chatgpt")
         return { native: true, conversation: null };
-    const caps = at.server.server.getClientCapabilities();
+    const caps = at.host.clientCapabilities();
     if (hasKey(caps?.extensions, EXTENSION_KEY))
         return { native: true, conversation: null };
     const metaCaps = meta["io.modelcontextprotocol/clientCapabilities"];
@@ -457,7 +461,9 @@ export function nativeOf(at, extra, identified) {
 /**
  * Wakeshift attached to an MCP server: the handle `wake.attach` returns.
  * Wrap a tool's callback with events, and give its input schema
- * Wakeshift's three arguments with inputs.
+ * Wakeshift's three arguments with inputs. C is your MCP SDK's request
+ * context (2.x's ctx, 1.x's extra), which a wrapped call carries as
+ * `call.extra`.
  */
 export class Attached {
     at;
@@ -467,29 +473,33 @@ export class Attached {
     constructor(at) {
         this.at = at;
     }
-    /**
-     * shape plus Wakeshift's three optional arguments (session_key, harness
-     * and conversation_id), for a wrapped tool's inputSchema:
-     * `inputSchema: attached.inputs({ table: z.string() })`.
-     */
     inputs(shape) {
+        if (isSchema(shape)) {
+            const own = getObjectShape(shape);
+            if (!own)
+                throw new Error("wakeshift: attached.inputs takes a zod object or a shape");
+            for (const k of WAKE_KEYS)
+                if (k in own)
+                    throw new Error(`wakeshift: attached.inputs adds ${k} itself: leave it out of the schema`);
+            return withFields(shape, byVersion[isZ4Schema(shape) ? "v4" : "v3"].inputs);
+        }
         for (const k of WAKE_KEYS)
             if (k in (shape ?? {}))
                 throw new Error(`wakeshift: attached.inputs adds ${k} itself: leave it out of the shape`);
-        return { ...shape, ...fieldsFor(shape, "inputs") };
+        return { ...shape, ...fieldsFor(shape, "inputs", this.at.host) };
     }
-    /**
-     * shape plus an optional wakeshift string, for a wrapped tool's
-     * outputSchema. Optional: every structured result of a wrapped tool gets
-     * the block's text (without the beacon line) in wakeshift, for hosts that
-     * show the model only structuredContent, and tools/list names it in the
-     * tool's output schema; this declares it in your own schema and types.
-     * The block stays in the content too.
-     */
     output(shape) {
+        if (isSchema(shape)) {
+            const own = getObjectShape(shape);
+            if (!own)
+                throw new Error("wakeshift: attached.output takes a zod object or a shape");
+            if (WAKESHIFT_FIELD in own)
+                throw new Error("wakeshift: attached.output adds wakeshift itself: leave it out of the schema");
+            return withFields(shape, byVersion[isZ4Schema(shape) ? "v4" : "v3"].output);
+        }
         if ("wakeshift" in (shape ?? {}))
             throw new Error("wakeshift: attached.output adds wakeshift itself: leave it out of the shape");
-        return { ...shape, ...fieldsFor(shape, "output") };
+        return { ...shape, ...fieldsFor(shape, "output", this.at.host) };
     }
     /**
      * Wraps a tool's callback, for `server.registerTool(name, …)` with the
@@ -509,7 +519,8 @@ export class Attached {
         return cb;
     }
     async run(name, cb, handler, input, extra) {
-        const tool = registeredTool(this.at.server, name);
+        const host = this.at.host;
+        const tool = host.registeredTool(name);
         if (tool === undefined || (tool && (tool.handler ?? tool.callback) !== cb)) {
             throw new Error(`wakeshift: attached.events("${name}", …) isn't the tool named ${name} on the server passed to attach: register it with the same name, on that server`);
         }
@@ -547,7 +558,7 @@ export class Attached {
             await call.rollback();
             return result;
         }
-        if ((typeof result === "object" && result.isError) || extra?.signal?.aborted || !(await validOutput(outputSchema, result))) {
+        if ((typeof result === "object" && result.isError) || host.req(extra).signal?.aborted || !(await host.validOutput(outputSchema, result))) {
             await call.rollback();
             return typeof result === "string" ? { content: [{ type: "text", text: result }] } : result;
         }
@@ -563,7 +574,7 @@ export class Attached {
             return { isError: true, content: [{ type: "text", text: `Wakeshift couldn't finish setting up this conversation's wakes (${detailOf(err)}). Try again.` }] };
         }
         const out = mergeBlock(result, add ?? {});
-        if (out.structuredContent && WAKESHIFT_FIELD in out.structuredContent && !(await validOutput(outputSchema, out))) {
+        if (isRecord(out.structuredContent) && WAKESHIFT_FIELD in out.structuredContent && !(await host.validOutput(outputSchema, out))) {
             // A schema that refuses fields it doesn't name (a strict object):
             // MCP would refuse the result, so it goes without the field.
             const { [WAKESHIFT_FIELD]: _dropped, ...own } = out.structuredContent;
@@ -579,8 +590,8 @@ export class Attached {
      * attach, so a later set is wrapped too (as tasks.ts does).
      */
     amendToolsList() {
-        const handlers = this.at.server.server._requestHandlers;
-        if (!(handlers instanceof Map))
+        const handlers = this.at.host.handlers();
+        if (!handlers)
             return; // a future SDK: schemas stay as registered
         const wrap = (h) => async (request, extra) => {
             const out = await h(request, extra);
@@ -605,16 +616,4 @@ export class Attached {
         }
     }
 }
-// validOutput mirrors McpServer's own check of a result against the tool's
-// outputSchema (validateToolOutput): one it would refuse fails the call.
-async function validOutput(outputSchema, result) {
-    if (!outputSchema)
-        return true;
-    if (typeof result === "string" || !result.structuredContent)
-        return false;
-    const obj = normalizeObjectSchema(outputSchema);
-    if (!obj)
-        return true;
-    return (await safeParseAsync(obj, result.structuredContent)).success;
-}
 //# sourceMappingURL=toolcall.js.map

package/dist/toolcall.js.map

@@ -1 +1 @@
-{"version":3,"file":"toolcall.js","sourceRoot":"","sources":["../src/toolcall.ts"],"names":[],"mappings":"AAAA,0EAA0E;AAC1E,yEAAyE;AACzE,wEAAwE;AACxE,2EAA2E;AAC3E,2EAA2E;AAC3E,4EAA4E;AAC5E,yEAAyE;AACzE,yDAAyD;AACzD,EAAE;AACF,6FAA6F;AAC7F,gCAAgC;AAChC,iFAAiF;AACjF,2DAA2D;AAC3D,6FAA6F;AAC7F,qDAAqD;AACrD,4CAA4C;AAC5C,UAAU;AAEV,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,MAAM,QAAQ,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,QAAQ,CAAC;AAI7B,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,qBAAqB,EAAE,cAAc,EAA8E,MAAM,gDAAgD,CAAC;AAC/M,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAE3C,OAAO,EAAE,QAAQ,EAA6D,MAAM,gBAAgB,CAAC;AAErG,2EAA2E;AAC3E,6DAA6D;AAC7D,MAAM,WAAW,GAAG,aAAa,CAAC;AAClC,MAAM,WAAW,GAAG,SAAS,CAAC;AAC9B,MAAM,gBAAgB,GAAG,iBAAiB,CAAC;AAK3C,wEAAwE;AACxE,2BAA2B;AAC3B,MAAM,uBAAuB,GAC3B,mKAAmK,CAAC;AACtK,MAAM,mBAAmB,GACvB,6QAA6Q,CAAC;AAChR,MAAM,wBAAwB,GAC5B,4MAA4M,CAAC;AAC/M,MAAM,4BAA4B,GAAG,gFAAgF,CAAC;AACtH,mIAAmI;AACnI,MAAM,CAAC,MAAM,qBAAqB,GAAG,EAAE,WAAW,EAAE,uBAAuB,EAAE,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,wBAAwB,EAAW,CAAC;AAEhK,MAAM,UAAU,GAAG;IACjB,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;IACtE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;IAClE,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;CAC7E,CAAC;AACF,yIAAyI;AACzI,MAAM,CAAC,MAAM,QAAQ,GAAG,UAAU,CAAC;AACnC,MAAM,UAAU,GAAG,EAAE,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC,EAAE,CAAC;AAE/F,2EAA2E;AAC3E,4EAA4E;AAC5E,uCAAuC;AACvC,MAAM,SAAS,GAAG;IAChB,EAAE,EAAE;QACF,MAAM,EAAE;YACN,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;YACvE,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;YACnE,CAAC,gBAAgB,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;SAC9E;QACD,MAAM,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC,EAAE;KACrF;IACD,EAAE,EAAE;QACF,MAAM,EAAE;YACN,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;YACvE,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;YACnE,CAAC,gBAAgB,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;SAC9E;QACD,MAAM,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC,EAAE;KACrF;CACF,CAAC;AAEF,wEAAwE;AACxE,6CAA6C;AAC7C,SAAS,SAAS,CAAgC,KAAwB,EAAE,IAAO;IACjF,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5C,MAAM,MAAM,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC;IACzH,OAAO,MAAa,CAAC;AACvB,CAAC;AAeD,MAAM,SAAS,GAAuB,CAAC,WAAW,EAAE,WAAW,EAAE,gBAAgB,CAAC,CAAC;AAEnF;;;;GAIG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAsB;IAClD,kBAAkB;IAClB,oCAAoC;IACpC,uBAAuB;IACvB,wBAAwB;IACxB,iCAAiC;CAClC,CAAC;AAyEF,kDAAkD;AAClD,MAAM,OAAO,SAAU,SAAQ,KAAK;IAClC;QACE,KAAK,CAAC,0EAA0E,CAAC,CAAC;QAClF,IAAI,CAAC,IAAI,GAAG,WAAW,CAAC;IAC1B,CAAC;CACF;AAaD,yEAAyE;AACzE,iEAAiE;AACjE,MAAM,kBAAkB,GAAG,yEAAyE,CAAC;AACrG,MAAM,kBAAkB,GAAG,qBAAqB,CAAC;AACjD,sDAAsD;AACtD,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAW5B,mEAAmE;AACnE,qEAAqE;AACrE,iBAAiB;AACjB,SAAS,WAAW,CAAC,OAAkB;IACrC,MAAM,IAAI,GAAG,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5B,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM;QAAE,OAAO,EAAE,CAAC;IAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;IAClD,MAAM,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IACjC,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAC/C,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;QAAE,OAAO,EAAE,CAAC;IAC3E,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACvE,IAAI,IAAI,KAAK,EAAE;QAAE,OAAO,CAAC,GAAG,EAAE,CAAC;;QAC1B,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IAC3D,OAAO,IAAI,CAAC;AACd,CAAC;AAED,2HAA2H;AAC3H,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC;AAE3C;;;;;;GAMG;AACH,MAAM,UAAU,kBAAkB,CAAC,MAAe;IAChD,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO,MAAM,CAAC;IAClF,MAAM,CAAC,GAAG,MAAiC,CAAC;IAC5C,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC;IACvC,MAAM,KAAK,GAAG,CAAC,CAAC,UAAU,CAAC;IAC3B,IAAI,KAAK,KAAK,SAAS,IAAI,CAAC,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAAE,OAAO,MAAM,CAAC;IACxG,IAAI,KAAK,IAAI,eAAe,IAAK,KAAgB;QAAE,OAAO,MAAM,CAAC;IACjE,OAAO,EAAE,GAAG,CAAC,EAAE,UAAU,EAAE,EAAE,GAAG,CAAE,KAAgB,IAAI,EAAE,CAAC,EAAE,CAAC,eAAe,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,4BAA4B,EAAE,EAAE,EAAE,CAAC;AAClJ,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,UAAU,CAAC,MAAqB,EAAE,GAAiB;IACjE,MAAM,QAAQ,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC;IAC5C,MAAM,IAAI,GAAmB,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;IAC/F,MAAM,GAAG,GAAmB,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;IAC5E,MAAM,IAAI,GAAG,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACrC,MAAM,UAAU,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,IAAI,kBAAkB,IAAI,GAAG,CAAC,KAAK,CAAC;IAClE,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC1C,IAAI,IAAI;QAAE,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,UAAU,IAAI,QAAQ,KAAK,EAAE,CAAC;IAC1C,IAAI,QAAQ;QAAE,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;SAC5D,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,OAAO;QAAE,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;IACpF,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE,CAAC;QACpD,IAAI,CAAC,KAAK,kBAAkB,IAAI,GAAG;YAAE,SAAS;QAC9C,GAAG,CAAC,KAAK,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;IAC/C,CAAC;IACD,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;QAChG,GAAG,CAAC,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7D,CAAC;IACD,IAAI,IAAI,IAAI,GAAG,CAAC,iBAAiB,IAAI,OAAO,GAAG,CAAC,iBAAiB,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACxH,GAAG,CAAC,iBAAiB,GAAG,EAAE,GAAG,GAAG,CAAC,iBAAiB,EAAE,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,CAAC;IAChF,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,0EAA0E;AAC1E,4EAA4E;AAC5E,SAAS,UAAU,CAAC,IAAyC,EAAE,GAAW;IACxE,IAAI,CAAC,IAAI;QAAE,OAAO,EAAE,CAAC;IACrB,MAAM,KAAK,GAAG,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAClF,IAAI,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAC3D,IAAI,GAAG,GAAY,IAAI,CAAC;IACxB,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;QAClC,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC;YAAE,OAAO,EAAE,CAAC;QACrE,GAAG,GAAI,GAA+B,CAAC,IAAI,CAAC,CAAC;IAC/C,CAAC;IACD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC;AACpB,CAAC;AAED,mIAAmI;AACnI,MAAM,UAAU,iBAAiB,CAAC,IAAyC;IACzE,OAAO,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;AACnE,CAAC;AAED,SAAS,QAAQ,CAAC,CAAU,EAAE,CAAU;IACtC,MAAM,IAAI,GAAG,CAAC,CAAU,EAAW,EAAE,CACnC,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,SAAS;QAC3B,CAAC,CAAC,EAAE;QACJ,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;YAChB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC;YACb,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ;gBACrB,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAW,CAAC,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAE,CAA6B,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9G,CAAC,CAAC,CAAC,CAAC;IACZ,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;AAC7D,CAAC;AAED,SAAS,QAAQ,CAAC,GAAY;IAC5B,OAAO,GAAG,YAAY,cAAc,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAE,GAAa,EAAE,OAAO,IAAI,GAAG,CAAC,CAAC;AAC7F,CAAC;AAYD,SAAS,cAAc,CAAC,MAAiB,EAAE,IAAY;IACrD,MAAM,KAAK,GAAI,MAAmF,CAAC,gBAAgB,CAAC;IACpH,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC,CAAC,uCAAuC;IAC7F,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC;AACrB,CAAC;AA0BD,qGAAqG;AACrG,MAAM,SAAS;IAoBM,EAAE;IAGF,IAAI;IAtBvB,IAAI,CAAqB;IACzB,IAAI,CAAS;IACb,MAAM,GAAG,KAAK,CAAC;IACf,kBAAkB,GAAkB,IAAI,CAAC;IACzC,OAAO,CAAgB;IACvB,UAAU,GAAsB,IAAI,CAAC;IACrC,KAAK,CAAgB;IAErB,KAAK,GAAG,KAAK,CAAC;IACd,OAAO,GAAG,EAAE,CAAC;IACb,KAAK,GAAG,KAAK,CAAC;IACd,OAAO,GAAiB,EAAE,CAAC;IAC3B,WAAW,GAAiB,EAAE,CAAC;IAC/B,QAAQ,GAAG,IAAI,GAAG,EAAoB,CAAC;IAC/B,KAAK,GAAqB,OAAO,CAAC,OAAO,EAAE,CAAC;IACpD,kEAAkE;IACjD,UAAU,CAAqB;IAEhD,YACmB,EAAiB,EAClC,IAAY,EACZ,KAAoB,EACH,IAAmE;kBAHnE,EAAE;oBAGF,IAAI;QAErB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,MAAM,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC9B,IAAI,CAAC,IAAI,GAAG,EAAE,EAAE,IAAI,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,EAAE,EAAE,MAAM,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC;QACpC,MAAM,CAAC,GAAG,QAAQ,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;QACvB,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC,YAAY,CAAC;IAC3C,CAAC;IAED,0EAA0E;IAC1E,qEAAqE;IACrE,6CAA6C;IAC7C,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAA4C,CAAC;QACrE,OAAO,QAAQ,CAAC;YACd,IAAI;YACJ,UAAU,EAAE,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC;YACnD,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,OAAO,EAAE,OAAO,IAAI,SAAS;YAC7B,SAAS,EAAE,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC;YAClC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO;YAC1B,YAAY,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY;YACpC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC;SACpC,CAAC,CAAC;IACL,CAAC;IAED,yEAAyE;IACzE,oEAAoE;IAC5D,OAAO,CAAI,EAAoB;QACrC,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC,CAAC;QACvD,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACzF,IAAI,CAAC,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACxC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACvB,GAAG,CAAC,IAAI,CACN,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,EAC/B,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAChC,CAAC;QACF,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,UAAU,CAAC,KAA0B;QAC3C,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,CAAC,KAAK,KAAK,UAAU,IAAI,KAAK,CAAC,KAAK,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YAClF,MAAM,IAAI,cAAc,CAAC,GAAG,EAAE,eAAe,EAAE,GAAG,KAAK,EAAE,IAAI,IAAI,MAAM,oDAAoD,CAAC,CAAC;QAC/H,CAAC;IACH,CAAC;IAED,SAAS,CAAsB,KAAwB,EAAE,IAAoB,EAAE,IAAI,GAAyB,EAAE;QAC5G,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE;YAC7B,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACvB,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gBAClC,OAAO,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;YAC/I,CAAC;YACD,IAAI,KAAK,CAAC,QAAQ,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACjC,MAAM,IAAI,cAAc,CAAC,GAAG,EAAE,eAAe,EAAE,GAAG,KAAK,CAAC,IAAI,gGAAgG,CAAC,CAAC;YAChK,CAAC;YACD,MAAM,IAAI,GAA4B,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC;YACtH,IAAI,IAAI,CAAC,OAAO;gBAAE,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;YAC9C,IAAI,IAAI,CAAC,GAAG,KAAK,SAAS;gBAAE,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;YAChD,IAAI,IAAI,CAAC,MAAM;gBAAE,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;YAC3C,IAAI,IAAI,CAAC,KAAK;gBAAE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;YACxC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAKtC,MAAM,EAAE,sBAAsB,EAAE,IAAI,CAAC,CAAC;YACzC,IAAI,GAAG,CAAC,OAAO;gBAAE,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;YAC5C,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;YAClB,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;YAC3B,MAAM,GAAG,GAAG,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACpC,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC;YACtB,MAAM,EAAE,GAAG,GAAG,CAAC,kBAAkB,CAAC;YAClC,IAAI,EAAE,CAAC,OAAO;gBAAE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;YACtI,OAAO,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;QACtI,CAAC,CAAC,CAAC;IACL,CAAC;IAED,WAAW,CAAsB,KAAwB,EAAE,IAAoB;QAC7E,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE;YAC7B,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACvB,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,IAAI,CAAC,MAAM;gBAAE,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAC1D,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAsB,MAAM,EAAE,kCAAkC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;YACtK,OAAO,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC;QACrB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE;YAC7B,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO;gBAAE,OAAO,IAAI,CAAC,OAAO,CAAC;YACtD,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAoE,MAAM,EAAE,gBAAgB,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC9J,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;YAC3B,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;YAC3B,IAAI,CAAC,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC1C,OAAO,GAAG,CAAC,OAAO,CAAC;QACrB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,iFAAiF;IACzE,KAAK,CAAC,SAAS,CAAC,IAAY,EAAE,IAAa;QACjD,IAAI,CAAC,IAAI,CAAC,IAAI;YAAE,OAAO,KAAK,CAAC;QAC7B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAA+F,MAAM,EAAE,2BAA2B,EAAE;YAC3K,IAAI,EAAE,IAAI,CAAC,IAAI;SAChB,CAAC,CAAC;QACH,IAAI,KAAK,GAAG,KAAK,CAAC;QAClB,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,aAAa,IAAI,EAAE,EAAE,CAAC;YACxC,IAAI,CAAC,CAAC,KAAK,KAAK,IAAI,IAAI,CAAC,CAAC,WAAW,KAAK,SAAS,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,EAAE,IAAI,CAAC;gBAAE,SAAS;YAC9F,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAsB,MAAM,EAAE,0BAA0B,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAChI,KAAK,GAAG,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QAC7B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,8DAA8D;IACtD,SAAS,CAAC,IAAY,EAAE,IAAa;QAC3C,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,IAAI,QAAQ,CAAC,CAAC,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAAE,OAAO;QACzF,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,KAAK,gBAAgB,EAAE,CAAC;YACjD,IAAI,CAAC,EAAE,CAAC,QAAQ,CACd,SAAS,GAAG,IAAI,CAAC,IAAI,EACrB,+BAA+B,IAAI,CAAC,IAAI,4CAA4C,gBAAgB,6BAA6B,gBAAgB,YAAY,CAC9J,CAAC;YACF,OAAO;QACT,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,CAAC;IAED,oEAAoE;IACpE,2EAA2E;IAC3E,KAAK,CAAC,GAAG;QACP,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,EAAE,CAAC,QAAQ,CACd,WAAW,GAAG,IAAI,CAAC,IAAI,EACvB,+BAA+B,IAAI,CAAC,IAAI,+IAA+I,CACxL,CAAC;YACF,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC/C,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;IACpB,CAAC;IAED,sEAAsE;IACtE,uEAAuE;IACvE,yEAAyE;IACzE,yBAAyB;IACzB,KAAK,CAAC,QAAQ;QACZ,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,kCAAkC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,SAAS,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YAC7L,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,8DAA8D,IAAI,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,MAAM,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC7H,CAAC;QACH,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACpB,CAAC;IAED,sEAAsE;IACtE,mEAAmE;IACnE,sEAAsE;IACtE,qEAAqE;IACrE,yEAAyE;IACzE,0EAA0E;IAC1E,8DAA8D;IAC9D,KAAK,CAAC,QAAQ;QACZ,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QACzB,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC;QACxI,MAAM,IAAI,GAA4B,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QACrD,IAAI,IAAI,CAAC,OAAO;YAAE,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC9C,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC;YAAE,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC;QAChE,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAe,MAAM,EAAE,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACtF,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC;gBAAE,MAAM,GAAG,CAAC;YACzD,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,kDAAkD,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACnG,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;CACF;AAED,SAAS,YAAY,CAAC,CAAkC;IACtD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,IAAI,IAAI,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,IAAI,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,IAAI,IAAI,EAAE,CAAC;AAC3F,CAAC;AAED,SAAS,QAAQ,CAAC,KAAoB,EAAE,IAAY;IAClD,MAAM,CAAC,GAAG,KAAK,EAAE,WAAW,EAAE,OAAoE,CAAC;IACnG,IAAI,CAAC,CAAC;QAAE,OAAO,EAAE,CAAC;IAClB,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IAC3C,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;AAC7C,CAAC;AAED,SAAS,WAAW,CAAC,KAAoB;IACvC,OAAO,QAAQ,CAAC,KAAK,EAAE,YAAY,CAAC,IAAI,QAAQ,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;AACxE,CAAC;AAED,SAAS,MAAM,CAAC,CAAU,EAAE,GAAW;IACrC,OAAO,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAK,CAA6B,CAAC,GAAG,CAAC,KAAK,SAAS,CAAC;AAChH,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,QAAQ,CACtB,EAAyD,EACzD,KAAoB,EACpB,UAAmB;IAEnB,MAAM,IAAI,GAAG,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAA4B,CAAC;IAC5D,MAAM,YAAY,GAAG,OAAO,IAAI,CAAC,gBAAgB,CAAC,KAAK,QAAQ,IAAI,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAE,IAAI,CAAC,gBAAgB,CAAY,CAAC,CAAC,CAAC,IAAI,CAAC;IACtI,IAAI,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;IACrE,IAAI,YAAY;QAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;IACxD,IAAI,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC,KAAK,SAAS;QAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;IAC5H,MAAM,IAAI,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,qBAAqB,EAAyC,CAAC;IAC7F,IAAI,MAAM,CAAC,IAAI,EAAE,UAAU,EAAE,aAAa,CAAC;QAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;IACzF,MAAM,QAAQ,GAAG,IAAI,CAAC,4CAA4C,CAAwC,CAAC;IAC3G,IAAI,MAAM,CAAC,QAAQ,EAAE,UAAU,EAAE,aAAa,CAAC;QAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;IAC7F,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;AAC/C,CAAC;AAED;;;;GAIG;AACH,MAAM,OAAO,QAAQ;IAKU,EAAE;IAJ/B,gBAAgB;IACC,OAAO,GAAG,IAAI,GAAG,EAAmB,CAAC;IAEtD,iCAAiC;IACjC,YAA6B,EAAiB;kBAAjB,EAAE;IAAkB,CAAC;IAElD;;;;OAIG;IACH,MAAM,CAA8B,KAAQ;QAC1C,KAAK,MAAM,CAAC,IAAI,SAAS;YAAE,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,oCAAoC,CAAC,CAAC;QAC7I,OAAO,EAAE,GAAG,KAAK,EAAE,GAAG,SAAS,CAAC,KAAK,EAAE,QAAQ,CAAC,EAAE,CAAC;IACrD,CAAC;IAED;;;;;;;OAOG;IACH,MAAM,CAA8B,KAAQ;QAC1C,IAAI,WAAW,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,6EAA6E,CAAC,CAAC;QACjI,OAAO,EAAE,GAAG,KAAK,EAAE,GAAG,SAAS,CAAC,KAAK,EAAE,QAAQ,CAAC,EAAE,CAAC;IACrD,CAAC;IAED;;;;;;;OAOG;IACH,MAAM,CAAsB,IAAY,EAAE,OAAqF;QAC7H,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;QAC1H,IAAI,OAAO,OAAO,KAAK,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QACjG,MAAM,EAAE,GAAG,CAAC,KAAK,EAAE,IAA6B,EAAE,KAAoB,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,EAAE,OAAc,EAAE,IAAI,IAAI,EAAE,EAAE,KAAK,CAAC,CAA+B,CAAC;QAChK,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC3B,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,KAAK,CAAC,GAAG,CACf,IAAY,EACZ,EAAW,EACX,OAAkG,EAClG,KAA8B,EAC9B,KAAoB;QAEpB,MAAM,IAAI,GAAG,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAClD,IAAI,IAAI,KAAK,SAAS,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC;YAC3E,MAAM,IAAI,KAAK,CAAC,+BAA+B,IAAI,8BAA8B,IAAI,iFAAiF,CAAC,CAAC;QAC1K,CAAC;QACD,IAAI,YAAmC,CAAC;QACxC,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,KAAK,GAAG,cAAc,CAAC,qBAAqB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;YACtE,IAAI,CAAC,KAAK,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,EAAE,CAAC;gBACnD,MAAM,IAAI,KAAK,CAAC,uBAAuB,IAAI,mFAAmF,CAAC,CAAC;YAClI,CAAC;YACD,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,mIAAmI,CAAC,CAAC;QACrK,CAAC;QACD,MAAM,GAAG,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAK,KAAK,CAAC,CAAC,CAAY,CAAC,CAAC,CAAE,KAAK,CAAC,CAAC,CAAY,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACrH,MAAM,IAAI,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,WAAW,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,WAAW,CAAC,EAAE,YAAY,EAAE,GAAG,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;QAChJ,MAAM,OAAO,GAA4B,EAAE,CAAC;QAC5C,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC;YAAE,IAAI,CAAE,SAA+B,CAAC,QAAQ,CAAC,CAAC,CAAC;gBAAE,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAE9G,IAAI,MAAqB,CAAC;QAC1B,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACxC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtB,MAAM,GAAG,CAAC;QACZ,CAAC;QACD,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;QACjB,sEAAsE;QACtE,oEAAoE;QACpE,4CAA4C;QAC5C,IAAI,MAAM,KAAK,IAAI,IAAI,MAAM,KAAK,SAAS,IAAI,CAAC,OAAO,MAAM,KAAK,QAAQ,IAAI,OAAO,MAAM,KAAK,QAAQ,CAAC,EAAE,CAAC;YAC1G,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtB,OAAO,MAAmC,CAAC;QAC7C,CAAC;QACD,IAAI,CAAC,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,IAAI,CAAC,CAAC,MAAM,WAAW,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC,EAAE,CAAC;YAC3H,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtB,OAAO,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;QAC7F,CAAC;QACD,IAAI,GAAwB,CAAC;QAC7B,IAAI,CAAC;YACH,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC9B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,oEAAoE;YACpE,oDAAoD;YACpD,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtB,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,gDAAgD,IAAI,0DAA0D,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACjJ,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mEAAmE,QAAQ,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC,EAAE,CAAC;QAC/J,CAAC;QACD,MAAM,GAAG,GAAG,UAAU,CAAC,MAAM,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;QAC1C,IAAI,GAAG,CAAC,iBAAiB,IAAI,eAAe,IAAI,GAAG,CAAC,iBAAiB,IAAI,CAAC,CAAC,MAAM,WAAW,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;YACjH,kEAAkE;YAClE,6DAA6D;YAC7D,MAAM,EAAE,CAAC,eAAe,CAAC,EAAE,QAAQ,EAAE,GAAG,GAAG,EAAE,GAAG,GAAG,CAAC,iBAAiB,CAAC;YACtE,GAAG,CAAC,iBAAiB,GAAG,GAAG,CAAC;YAC5B,IAAI,CAAC,EAAE,CAAC,QAAQ,CACd,SAAS,GAAG,IAAI,EAChB,+BAA+B,IAAI,6JAA6J,CACjM,CAAC;QACJ,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED;;;;;OAKG;IACH,cAAc;QAEZ,MAAM,QAAQ,GAAI,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,MAAc,CAAC,gBAAoD,CAAC;QACrG,IAAI,CAAC,CAAC,QAAQ,YAAY,GAAG,CAAC;YAAE,OAAO,CAAC,2CAA2C;QACnF,MAAM,IAAI,GACR,CAAC,CAAU,EAAW,EAAE,CACxB,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;YACvB,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACpC,IAAI,GAAG,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC;gBAAE,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAChE,OAAO,GAAG,CAAC;QACb,CAAC,CAAC;QACJ,MAAM,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACrC,IAAI,CAAC;YAAE,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,QAAQ,CAAC,GAAG,GAAG,CAAC,MAAc,EAAE,CAAU,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,KAAK,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACpG,CAAC;IAEO,UAAU,CAAC,KAAgB;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAsD,CAAC;YACxE,IAAI,CAAC,CAAC,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,YAAY;gBAAE,SAAS;YACxH,MAAM,MAAM,GAAG,kBAAkB,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;YAClD,IAAI,MAAM,KAAK,CAAC,CAAC,YAAY;gBAAE,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC;QAC3E,CAAC;IACH,CAAC;CACF;AAED,2EAA2E;AAC3E,yEAAyE;AACzE,KAAK,UAAU,WAAW,CAAC,YAAmC,EAAE,MAAqB;IACnF,IAAI,CAAC,YAAY;QAAE,OAAO,IAAI,CAAC;IAC/B,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,iBAAiB;QAAE,OAAO,KAAK,CAAC;IAC1E,MAAM,GAAG,GAAG,qBAAqB,CAAC,YAAY,CAAC,CAAC;IAChD,IAAI,CAAC,GAAG;QAAE,OAAO,IAAI,CAAC;IACtB,OAAO,CAAC,MAAM,cAAc,CAAC,GAAG,EAAE,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,OAAO,CAAC;AACvE,CAAC","sourcesContent":["// One way to subscribe from a tool (docs/subscribe-from-tools.md): a tool\n// callback wrapped by the handle attach returns (attached.events) gets a\n// ToolCall, and subscribes the calling conversation with call.subscribe\n// (ends it with call.unsubscribe). The wrapper adds one Wakeshift block to\n// whatever the handler returns, so a conversation can't be left unwakeable\n// by a forgotten argument, a wrong tool name or an unreturned result; and a\n// call that fails ends the subscriptions it made. A port of the Go SDK's\n// appkit.Events (pkg/appkit/toolcall.go, attachcall.go).\n//\n//   const attached = wake.attach(server, { identify: (extra) => ({ user: userOf(extra) }) })\n//   server.registerTool(\"join\",\n//     { description: \"…\", inputSchema: attached.inputs({ table: z.string() }) },\n//     attached.events(\"join\", async ({ table }, call) => {\n//       const sub = await call.subscribe(turn, { table }, { reason: `Playing at ${table}` })\n//       await casino.seat(call.user!, table, sub.id)\n//       return `You're seated at ${table}.`\n//     }))\n\nimport { z } from \"zod\";\nimport * as z3 from \"zod/v3\";\nimport * as z4 from \"zod/v4\";\nimport type { CallToolResult, ServerNotification, ServerRequest } from \"@modelcontextprotocol/sdk/types.js\";\nimport type { McpServer, ToolCallback } from \"@modelcontextprotocol/sdk/server/mcp.js\";\nimport type { RequestHandlerExtra } from \"@modelcontextprotocol/sdk/shared/protocol.js\";\nimport { getObjectShape, isZ4Schema, normalizeObjectSchema, safeParseAsync, type AnySchema, type SchemaInput, type ShapeOutput, type ZodRawShapeCompat } from \"@modelcontextprotocol/sdk/server/zod-compat.js\";\nimport { WakeshiftError } from \"./errors.js\";\nimport { EXTENSION_KEY, dialectOf } from \"./spec.js\";\nimport { takesTasks } from \"./discover.js\";\nimport type { Identity, WakeEvent } from \"./events.js\";\nimport { packCall, type Call, type Duration, type Recognized, type Wakeshift } from \"./wakeshift.js\";\n\n// Wakeshift's three arguments (wakeshift.ts's SESSION_ARG, HARNESS_ARG and\n// CONVERSATION_ARG, repeated: that module imports this one).\nconst SESSION_ARG = \"session_key\";\nconst HARNESS_ARG = \"harness\";\nconst CONVERSATION_ARG = \"conversation_id\";\n\n/** MCP's own extra for a tool call: the request's signal, auth, _meta, HTTP request. */\nexport type ToolCallExtra = RequestHandlerExtra<ServerRequest, ServerNotification>;\n\n// What the agent is told each of Wakeshift's arguments is: the Go SDK's\n// WakeArgs, word for word.\nconst SESSION_KEY_DESCRIPTION =\n  \"This conversation's session_key, from an earlier result (a secret token, or a handle such as claude-code#2). Keep it to this conversation; don't share or log it.\";\nconst HARNESS_DESCRIPTION =\n  \"The program you're running in: codex, claude-code, cursor (local CLI or IDE), cursor-cloud (Cursor cloud agent), grok-bot, copilot-cli, opencode, antigravity, openclaw (even when it runs your turns in Codex), hermes, chatgpt, or your program's name if it isn't listed\";\nconst CONVERSATION_DESCRIPTION =\n  \"Your own ID for this conversation, if your program gives you one. Cursor and Grok Bot: the output of `printenv CURSOR_CONVERSATION_ID`. Codex and Antigravity send it automatically; others leave it empty\";\nconst WAKESHIFT_OUTPUT_DESCRIPTION = \"What Wakeshift adds for this conversation: how it's woken by this app's events\";\n/** @internal Wakeshift's three arguments' descriptions, for a tool whose input schema is written out (the wait tool, tasks.ts). */\nexport const WAKE_ARG_DESCRIPTIONS = { session_key: SESSION_KEY_DESCRIPTION, harness: HARNESS_DESCRIPTION, conversation_id: CONVERSATION_DESCRIPTION } as const;\n\nconst wakeInputs = {\n  [SESSION_ARG]: z.string().optional().describe(SESSION_KEY_DESCRIPTION),\n  [HARNESS_ARG]: z.string().optional().describe(HARNESS_DESCRIPTION),\n  [CONVERSATION_ARG]: z.string().optional().describe(CONVERSATION_DESCRIPTION),\n};\n/** @internal Wakeshift's three arguments as attach's own tools take them: worded as a wrapped tool's (test/events-list.json pins it). */\nexport const toolArgs = wakeInputs;\nconst wakeOutput = { wakeshift: z.string().optional().describe(WAKESHIFT_OUTPUT_DESCRIPTION) };\n\n// The same in each of zod's two APIs: MCP refuses a shape that mixes them,\n// so a shape written with zod/v3 (or zod/v4) gets Wakeshift's fields in its\n// own (zod 3.25 and later carry both).\nconst byVersion = {\n  v3: {\n    inputs: {\n      [SESSION_ARG]: z3.string().optional().describe(SESSION_KEY_DESCRIPTION),\n      [HARNESS_ARG]: z3.string().optional().describe(HARNESS_DESCRIPTION),\n      [CONVERSATION_ARG]: z3.string().optional().describe(CONVERSATION_DESCRIPTION),\n    },\n    output: { wakeshift: z3.string().optional().describe(WAKESHIFT_OUTPUT_DESCRIPTION) },\n  },\n  v4: {\n    inputs: {\n      [SESSION_ARG]: z4.string().optional().describe(SESSION_KEY_DESCRIPTION),\n      [HARNESS_ARG]: z4.string().optional().describe(HARNESS_DESCRIPTION),\n      [CONVERSATION_ARG]: z4.string().optional().describe(CONVERSATION_DESCRIPTION),\n    },\n    output: { wakeshift: z4.string().optional().describe(WAKESHIFT_OUTPUT_DESCRIPTION) },\n  },\n};\n\n// fieldsFor is Wakeshift's fields in the zod API shape's own fields use\n// (the default import's for an empty shape).\nfunction fieldsFor<K extends \"inputs\" | \"output\">(shape: ZodRawShapeCompat, kind: K): K extends \"inputs\" ? WakeInputs : WakeOutputShape {\n  const first = Object.values(shape ?? {})[0];\n  const fields = !first ? (kind === \"inputs\" ? wakeInputs : wakeOutput) : byVersion[isZ4Schema(first) ? \"v4\" : \"v3\"][kind];\n  return fields as any;\n}\n\n/** Wakeshift's three arguments, as attached.inputs adds them. */\nexport type WakeInputs = typeof wakeInputs;\n/** The optional wakeshift string attached.output adds. */\nexport type WakeOutputShape = typeof wakeOutput;\n/** The names of Wakeshift's three arguments. */\nexport type WakeKey = typeof SESSION_ARG | typeof HARNESS_ARG | typeof CONVERSATION_ARG;\n/** An input shape with Wakeshift's three arguments in it (attached.inputs'). */\nexport type WakeShape = ZodRawShapeCompat & { [K in WakeKey]: AnySchema };\n/** What a wrapped handler gets: the tool's own arguments, without Wakeshift's three. */\nexport type AppArgs<S extends ZodRawShapeCompat> = Omit<ShapeOutput<S>, WakeKey>;\n/** What a wrapped handler returns: a string (one text block) or a CallToolResult. */\nexport type WrappedResult = string | CallToolResult;\n\nconst WAKE_KEYS: readonly WakeKey[] = [SESSION_ARG, HARNESS_ARG, CONVERSATION_ARG];\n\n/**\n * @internal The tools/call _meta keys a harness names its conversation in\n * (the core's harnesses ConversationMeta, dotted for nested objects):\n * test/conversation-meta.json, which a Go test keeps current.\n */\nexport const CONVERSATION_META: readonly string[] = [\n  \"agent-session-id\",\n  \"antigravity.google/conversation_id\",\n  \"vscode.conversationId\",\n  \"wakeshift/conversation\",\n  \"x-codex-turn-metadata.thread_id\",\n];\n\n/**\n * The calling conversation's subscription, made or renewed by\n * call.subscribe. id and user address it later: emit's\n * `to: [{ user, subscription: id }]`, or `event.end(user, id)`.\n */\nexport interface Subscription {\n  id: string;\n  user: string;\n  /** False for a renewal. */\n  created: boolean;\n  /** The conversation's session. A secret: don't show, store or log it. */\n  session: string;\n  /** What Wakeshift recognized the agent's program as. */\n  recognized: Recognized;\n  /** The caller subscribes itself (call.native): nothing was made, id is \"\", and the block tells it what to subscribe to. */\n  native: boolean;\n}\n\n/** How call.subscribe subscribes. */\nexport interface ToolSubscribeOptions {\n  /** End after this long without a delivery or a fresh subscribe (default 30 days). */\n  ttl?: Duration;\n  /** Why the app subscribes the conversation, shown in the console (at most 200 characters). */\n  reason?: string;\n  /** Names a new session (agent@label), for an app acting for its agents. */\n  label?: string;\n}\n\n/**\n * One call of a wrapped tool (attached.events): who's calling, and the\n * calls that subscribe the calling conversation. Only the wrapper makes\n * one, and it works until the handler returns (then CallEnded).\n */\nexport interface ToolCall {\n  /** Who attach's identify says is calling (undefined for none). */\n  readonly user: string | undefined;\n  /** The tool being called. */\n  readonly tool: string;\n  /**\n   * The client subscribes to the app's events itself (MCP Events: ChatGPT,\n   * the house agents). call.subscribe then makes no conversation\n   * subscription, and the block tells it what to subscribe to.\n   */\n  readonly native: boolean;\n  /** A native client's conversation, when it names one (ChatGPT's _meta[\"openai/session\"]). */\n  readonly nativeConversation: string | null;\n  /** The session_key passed; after a subscribe or resolveSession, the conversation's session. */\n  readonly session: string | null;\n  /** What Wakeshift recognized the agent's program as: null until a subscribe or resolveSession. */\n  readonly recognized: Recognized | null;\n  /** MCP's own extra for the call. */\n  readonly extra: ToolCallExtra;\n  /**\n   * Subscribes the calling conversation to event with args, or renews its\n   * subscription. The reply then carries what the conversation needs to be\n   * woken (one block for the call, however many subscribes it makes), and\n   * if the call fails, the subscription is ended. For a native caller\n   * nothing is made. Your tool decides who may: authorize isn't consulted.\n   * A personal event needs a user (identify's).\n   */\n  subscribe<A extends AnySchema>(event: WakeEvent<A, any>, args: SchemaInput<A>, opts?: ToolSubscribeOptions): Promise<Subscription>;\n  /**\n   * Ends the calling conversation's subscription to event with args,\n   * resolving to whether one ended. For a native caller it ends the user's\n   * own (webhook) subscriptions to the event with those arguments.\n   */\n  unsubscribe<A extends AnySchema>(event: WakeEvent<A, any>, args: SchemaInput<A>): Promise<boolean>;\n  /** Finds or starts the calling conversation's session without subscribing it (it doesn't make it wakeable). */\n  resolveSession(): Promise<string>;\n}\n\n/** A ToolCall used after its handler returned. */\nexport class CallEnded extends Error {\n  constructor() {\n    super(\"wakeshift: the tool call has ended: subscribe before the handler returns\");\n    this.name = \"CallEnded\";\n  }\n}\n\n/** attach's event options. */\nexport interface EventsOptions {\n  /**\n   * Asked first whether a wrapped tool's call comes from a client that\n   * subscribes to the app's events itself (call.native), for callers the\n   * SDK can't tell (an app's own house agents). False leaves it to the\n   * SDK's own checks.\n   */\n  native?: (extra: ToolCallExtra) => boolean;\n}\n\n// The line a text-carried beacon is on (the core's beacon.Text), which a\n// harness's hook reads only as the last line of a result's text.\nconst BEACON_LINE_PREFIX = \"(For Wakeshift's hook on this computer, not for you: wakeshift-beacon: \";\nconst CONNECTOR_META_KEY = \"wakeshift/connector\";\n// The most native lines Wakeshift takes for one call.\nconst MAX_NATIVE_LINES = 20;\n\n/** /calls/addition's answer. */\nexport interface CallAddition {\n  text?: string;\n  trailer?: string;\n  meta?: Record<string, unknown>;\n}\n\ntype Content = CallToolResult[\"content\"][number];\n\n// takeTrailer takes a beacon line off the end of the result's text\n// (dropping its block if nothing else is in it), returning it (\"\" if\n// there's none).\nfunction takeTrailer(content: Content[]): string {\n  const last = content.at(-1);\n  if (!last || last.type !== \"text\") return \"\";\n  const text = last.text.replace(/[ \\t\\r\\n]+$/, \"\");\n  const i = text.lastIndexOf(\"\\n\");\n  const line = i >= 0 ? text.slice(i + 1) : text;\n  if (!line.startsWith(BEACON_LINE_PREFIX) || !line.endsWith(\")\")) return \"\";\n  const rest = i >= 0 ? text.slice(0, i).replace(/[ \\t\\r\\n]+$/, \"\") : \"\";\n  if (rest === \"\") content.pop();\n  else content[content.length - 1] = { ...last, text: rest };\n  return line;\n}\n\n/** The structuredContent field a wrapped tool's structured result carries Wakeshift's block in (the same in every SDK). */\nexport const WAKESHIFT_FIELD = \"wakeshift\";\n\n/**\n * @internal An object output schema (JSON Schema, as tools/list gives it)\n * with an optional wakeshift string added, as a copy, so a client that\n * checks structuredContent against it takes the block. Anything else, or a\n * schema that already names it, comes back as it was.\n * test/wrapped-block.json (output_schemas) pins it.\n */\nexport function withWakeshiftField(schema: unknown): unknown {\n  if (!schema || typeof schema !== \"object\" || Array.isArray(schema)) return schema;\n  const s = schema as Record<string, unknown>;\n  if (s.type !== \"object\") return schema;\n  const props = s.properties;\n  if (props !== undefined && (!props || typeof props !== \"object\" || Array.isArray(props))) return schema;\n  if (props && WAKESHIFT_FIELD in (props as object)) return schema;\n  return { ...s, properties: { ...((props as object) ?? {}), [WAKESHIFT_FIELD]: { type: \"string\", description: WAKESHIFT_OUTPUT_DESCRIPTION } } };\n}\n\n/**\n * @internal Adds the call's block to a wrapped handler's result: Wakeshift's\n * text as a block after the app's content, then a text-carried beacon last,\n * and _meta; a string result stays one text block. A result that already\n * carries a beacon keeps it, last, and gets no second one. A structured\n * result (structuredContent an object) gets the block's text in its\n * wakeshift field too. test/wrapped-block.json is the shape, shared with\n * the other SDKs.\n */\nexport function mergeBlock(result: WrappedResult, add: CallAddition): CallToolResult {\n  const isString = typeof result === \"string\";\n  const base: CallToolResult = isString ? { content: [{ type: \"text\", text: result }] } : result;\n  const out: CallToolResult = { ...base, content: [...(base.content ?? [])] };\n  const text = (add.text ?? \"\").trim();\n  const metaBeacon = !!out._meta && CONNECTOR_META_KEY in out._meta;\n  const existing = takeTrailer(out.content);\n  if (text) out.content.push({ type: \"text\", text });\n  const has = metaBeacon || existing !== \"\";\n  if (existing) out.content.push({ type: \"text\", text: existing });\n  else if (!has && add.trailer) out.content.push({ type: \"text\", text: add.trailer });\n  for (const [k, v] of Object.entries(add.meta ?? {})) {\n    if (k === CONNECTOR_META_KEY && has) continue;\n    out._meta = { ...(out._meta ?? {}), [k]: v };\n  }\n  if (isString) {\n    const texts = out.content.map((c) => (c.type === \"text\" ? c.text : \"\")).filter((t) => t !== \"\");\n    out.content = [{ type: \"text\", text: texts.join(\"\\n\\n\") }];\n  }\n  if (text && out.structuredContent && typeof out.structuredContent === \"object\" && !Array.isArray(out.structuredContent)) {\n    out.structuredContent = { ...out.structuredContent, [WAKESHIFT_FIELD]: text };\n  }\n  return out;\n}\n\n// metaString is the string at key in a tool call's _meta: the key itself,\n// else a dotted path into nested objects (the core's harnesses.metaString).\nfunction metaString(meta: Record<string, unknown> | undefined, key: string): string {\n  if (!meta) return \"\";\n  const limit = (s: unknown) => (typeof s === \"string\" && s.length <= 200 ? s : \"\");\n  if (typeof meta[key] === \"string\") return limit(meta[key]);\n  let cur: unknown = meta;\n  for (const part of key.split(\".\")) {\n    if (!cur || typeof cur !== \"object\" || Array.isArray(cur)) return \"\";\n    cur = (cur as Record<string, unknown>)[part];\n  }\n  return limit(cur);\n}\n\n/** @internal Whether a tool call's _meta names its conversation, the way a harness puts it there (test/conversation-meta.json). */\nexport function namesConversation(meta: Record<string, unknown> | undefined): boolean {\n  return CONVERSATION_META.some((k) => metaString(meta, k) !== \"\");\n}\n\nfunction sameJSON(a: unknown, b: unknown): boolean {\n  const norm = (v: unknown): unknown =>\n    v === null || v === undefined\n      ? {}\n      : Array.isArray(v)\n        ? v.map(norm)\n        : typeof v === \"object\"\n          ? Object.fromEntries(Object.keys(v as object).sort().map((k) => [k, norm((v as Record<string, unknown>)[k])]))\n          : v;\n  return JSON.stringify(norm(a)) === JSON.stringify(norm(b));\n}\n\nfunction detailOf(err: unknown): string {\n  return err instanceof WakeshiftError ? err.detail : String((err as Error)?.message ?? err);\n}\n\n// The parts of an McpServer the wrapper reads that aren't public: its\n// registered tools (the handler, input and output schemas). A unit test pins\n// them against the supported MCP SDK versions.\ninterface RegisteredToolInternal {\n  inputSchema?: AnySchema;\n  outputSchema?: AnySchema;\n  handler?: unknown;\n  callback?: unknown;\n}\n\nfunction registeredTool(server: McpServer, name: string): RegisteredToolInternal | undefined | null {\n  const tools = (server as unknown as { _registeredTools?: Record<string, RegisteredToolInternal> })._registeredTools;\n  if (!tools || typeof tools !== \"object\") return null; // an MCP SDK that keeps them elsewhere\n  return tools[name];\n}\n\ninterface CreatedSub {\n  name: string;\n  arguments: unknown;\n  session: string;\n  pending: string; // the call's pending-addition record, which made it\n}\n\ninterface NativeLine {\n  name: string;\n  arguments: unknown;\n}\n\n/** @internal What the wrapper needs of attach: the client, the server, who's calling, the event options. */\nexport interface AttachContext {\n  wake: Wakeshift;\n  server: McpServer;\n  /** Who attach's identify says is calling: its user and client. */\n  identify(extra: ToolCallExtra): Identity | undefined;\n  /** The client's name: the request's (initialize's, else _meta's), else identified (identify's), else \"\". */\n  clientOf(meta: Record<string, unknown> | undefined, identified?: string): string;\n  events: EventsOptions;\n  warnOnce(key: string, message: string): void;\n}\n\n/** One tools/call of a wrapped tool: what its handler subscribed, for the block and the rollback. */\nclass CallScope implements ToolCall {\n  user: string | undefined;\n  tool: string;\n  native = false;\n  nativeConversation: string | null = null;\n  session: string | null;\n  recognized: Recognized | null = null;\n  extra: ToolCallExtra;\n\n  ended = false;\n  pending = \"\";\n  armed = false;\n  created: CreatedSub[] = [];\n  nativeLines: NativeLine[] = [];\n  inflight = new Set<Promise<unknown>>();\n  private queue: Promise<unknown> = Promise.resolve();\n  // The client's name identify gave, for a request that names none.\n  private readonly identified: string | undefined;\n\n  constructor(\n    private readonly at: AttachContext,\n    tool: string,\n    extra: ToolCallExtra,\n    private readonly args: { session?: string; harness?: string; conversation?: string },\n  ) {\n    this.tool = tool;\n    this.extra = extra;\n    const id = at.identify(extra);\n    this.user = id?.user;\n    this.identified = id?.client;\n    this.session = args.session || null;\n    const n = nativeOf(at, extra, this.identified);\n    this.native = n.native;\n    this.nativeConversation = n.conversation;\n  }\n\n  // wire is the call as Wakeshift takes it: the request's client, _meta and\n  // User-Agent, the agent's harness and conversation_id arguments, the\n  // conversation's session, and who's calling.\n  wire(session = this.session): Call {\n    const meta = this.extra._meta as Record<string, unknown> | undefined;\n    return packCall({\n      meta,\n      clientName: this.at.clientOf(meta, this.identified),\n      user: this.user,\n      session: session ?? undefined,\n      userAgent: userAgentOf(this.extra),\n      harness: this.args.harness,\n      conversation: this.args.conversation,\n      tool: this.tool,\n      tasks: takesTasks(meta, this.extra),\n    });\n  }\n\n  // request runs one request of the call to Wakeshift, after any before it\n  // (one at a time per call), refusing once the handler has returned.\n  private request<T>(fn: () => Promise<T>): Promise<T> {\n    if (this.ended) return Promise.reject(new CallEnded());\n    const run = this.queue.then(() => (this.ended ? Promise.reject(new CallEnded()) : fn()));\n    this.queue = run.catch(() => undefined);\n    this.inflight.add(run);\n    run.then(\n      () => this.inflight.delete(run),\n      () => this.inflight.delete(run),\n    );\n    return run;\n  }\n\n  private checkEvent(event: WakeEvent<any, any>): void {\n    if (!event || typeof event.owner !== \"function\" || event.owner() !== this.at.wake) {\n      throw new WakeshiftError(404, \"unknown_event\", `${event?.name ?? \"that\"} is another app's event: this app doesn't offer it`);\n    }\n  }\n\n  subscribe<A extends AnySchema>(event: WakeEvent<A, any>, args: SchemaInput<A>, opts: ToolSubscribeOptions = {}): Promise<Subscription> {\n    return this.request(async () => {\n      this.checkEvent(event);\n      const known = event.parseArgs(args);\n      if (this.native) {\n        this.addNative(event.name, known);\n        return { id: \"\", user: this.user ?? \"\", created: false, session: \"\", recognized: { harness: null, via: null, claimed: null }, native: true };\n      }\n      if (event.personal && !this.user) {\n        throw new WakeshiftError(400, \"bad_arguments\", `${event.name} is personal: call.subscribe needs a user (attach's identify), who the subscription belongs to`);\n      }\n      const body: Record<string, unknown> = { call: this.wire(), name: event.name, arguments: known, defer_addition: true };\n      if (this.pending) body.pending = this.pending;\n      if (opts.ttl !== undefined) body.ttl = opts.ttl;\n      if (opts.reason) body.reason = opts.reason;\n      if (opts.label) body.label = opts.label;\n      const out = await this.at.wake.requestApp<{\n        event_subscription: { id: string; user?: string; created: boolean };\n        session: string;\n        agent?: Partial<Recognized>;\n        pending?: string;\n      }>(\"POST\", \"/event-subscriptions\", body);\n      if (out.pending) this.pending = out.pending;\n      this.armed = true;\n      this.session = out.session;\n      const rec = recognizedOf(out.agent);\n      this.recognized = rec;\n      const es = out.event_subscription;\n      if (es.created) this.created.push({ name: event.name, arguments: known, session: out.session, pending: out.pending ?? this.pending });\n      return { id: es.id, user: es.user ?? this.user ?? \"\", created: !!es.created, session: out.session, recognized: rec, native: false };\n    });\n  }\n\n  unsubscribe<A extends AnySchema>(event: WakeEvent<A, any>, args: SchemaInput<A>): Promise<boolean> {\n    return this.request(async () => {\n      this.checkEvent(event);\n      const known = event.parseArgs(args);\n      if (this.native) return this.endNative(event.name, known);\n      const out = await this.at.wake.requestApp<{ ended?: boolean }>(\"POST\", \"/event-subscriptions/unsubscribe\", { call: this.wire(), name: event.name, arguments: known });\n      return !!out.ended;\n    });\n  }\n\n  resolveSession(): Promise<string> {\n    return this.request(async () => {\n      if (this.pending && this.session) return this.session;\n      const out = await this.at.wake.requestApp<{ session: string; pending: string; agent?: Partial<Recognized> }>(\"POST\", \"/calls/session\", { call: this.wire() });\n      this.pending = out.pending;\n      this.session = out.session;\n      this.recognized = recognizedOf(out.agent);\n      return out.session;\n    });\n  }\n\n  // endNative ends the user's webhook subscriptions to event with these arguments.\n  private async endNative(name: string, args: unknown): Promise<boolean> {\n    if (!this.user) return false;\n    const out = await this.at.wake.requestApp<{ subscriptions?: { id: string; event: string; arguments: unknown; destination: string }[] }>(\"POST\", \"/event-subscriptions/list\", {\n      user: this.user,\n    });\n    let ended = false;\n    for (const s of out.subscriptions ?? []) {\n      if (s.event !== name || s.destination !== \"webhook\" || !sameJSON(s.arguments, args)) continue;\n      const r = await this.at.wake.requestApp<{ ended?: boolean }>(\"POST\", \"/event-subscriptions/end\", { user: this.user, id: s.id });\n      ended = ended || !!r.ended;\n    }\n    return ended;\n  }\n\n  // addNative adds a native line, once per event and arguments.\n  private addNative(name: string, args: unknown): void {\n    if (this.nativeLines.some((l) => l.name === name && sameJSON(l.arguments, args))) return;\n    if (this.nativeLines.length === MAX_NATIVE_LINES) {\n      this.at.warnOnce(\n        \"native:\" + this.tool,\n        `wakeshift: the wrapped tool ${this.tool} subscribed a native client to more than ${MAX_NATIVE_LINES} events in one call; only ${MAX_NATIVE_LINES} are named`,\n      );\n      return;\n    }\n    this.nativeLines.push({ name, arguments: args });\n  }\n\n  // end closes the call to further requests, after waiting for any in\n  // flight (warning once per tool: the handler should have waited for them).\n  async end(): Promise<void> {\n    if (this.inflight.size > 0) {\n      this.at.warnOnce(\n        \"inflight:\" + this.tool,\n        `wakeshift: the wrapped tool ${this.tool} returned while a subscribe was in flight; it was waited for: await call.subscribe, call.unsubscribe and call.resolveSession before returning`,\n      );\n      await Promise.allSettled([...this.inflight]);\n    }\n    this.ended = true;\n  }\n\n  // rollback ends every subscription the call created and no other call\n  // has renewed since (the core checks the pending record that made it),\n  // with reason rolled_back (audited as such, no \"ended\" notice), whatever\n  // became of the request.\n  async rollback(): Promise<void> {\n    for (const c of this.created) {\n      try {\n        await this.at.wake.requestApp(\"POST\", \"/event-subscriptions/unsubscribe\", { call: this.wire(c.session), name: c.name, arguments: c.arguments, reason: \"rolled_back\", pending: c.pending });\n      } catch (err) {\n        this.at.wake.warn(`wakeshift: rolling back a failed tool call's subscription (${this.tool}, ${c.name}): ${detailOf(err)}`);\n      }\n    }\n    this.created = [];\n  }\n\n  // addition asks Wakeshift for the call's block: what the conversation\n  // needs to be woken (from the call's pending-addition record, if a\n  // subscribe or resolveSession made one; otherwise what any other tool\n  // result of the app's carries), and a line per event a native caller\n  // should subscribe to. A call that names no conversation, and subscribed\n  // nothing, gets nothing (null). Only a failure for a call that subscribed\n  // is thrown: without a subscription, the block is a courtesy.\n  async addition(): Promise<CallAddition | null> {\n    const wire = this.wire();\n    if (!this.pending && this.nativeLines.length === 0 && !wire.session && !wire.conversation && !namesConversation(wire.meta)) return null;\n    const body: Record<string, unknown> = { call: wire };\n    if (this.pending) body.pending = this.pending;\n    if (this.nativeLines.length > 0) body.native = this.nativeLines;\n    try {\n      return await this.at.wake.requestApp<CallAddition>(\"POST\", \"/calls/addition\", body);\n    } catch (err) {\n      if (this.armed || this.nativeLines.length > 0) throw err;\n      this.at.wake.warn(`wakeshift: Wakeshift's addition to a result of ${this.tool}: ${detailOf(err)}`);\n      return null;\n    }\n  }\n}\n\nfunction recognizedOf(a: Partial<Recognized> | undefined): Recognized {\n  return { harness: a?.harness ?? null, via: a?.via ?? null, claimed: a?.claimed ?? null };\n}\n\nfunction headerOf(extra: ToolCallExtra, name: string): string {\n  const h = extra?.requestInfo?.headers as Record<string, string | string[] | undefined> | undefined;\n  if (!h) return \"\";\n  const v = h[name] ?? h[name.toLowerCase()];\n  return (Array.isArray(v) ? v[0] : v) ?? \"\";\n}\n\nfunction userAgentOf(extra: ToolCallExtra): string {\n  return headerOf(extra, \"user-agent\") || headerOf(extra, \"User-Agent\");\n}\n\nfunction hasKey(o: unknown, key: string): boolean {\n  return !!o && typeof o === \"object\" && !Array.isArray(o) && (o as Record<string, unknown>)[key] !== undefined;\n}\n\n/**\n * @internal Whether a call comes from a client that subscribes to the app's\n * events itself, on stateless servers too (which never see initialize), in\n * this order: the app's override (events.native); ChatGPT's\n * _meta[\"openai/session\"]; a client name (initialize's, or _meta's, else\n * identify's) or User-Agent of a known native client; a client that\n * declares the MCP Events extension (initialize's capabilities, or\n * _meta's). conversation is ChatGPT's conversation, when it names one.\n */\nexport function nativeOf(\n  at: Pick<AttachContext, \"server\" | \"clientOf\" | \"events\">,\n  extra: ToolCallExtra,\n  identified?: string,\n): { native: boolean; conversation: string | null } {\n  const meta = (extra._meta ?? {}) as Record<string, unknown>;\n  const conversation = typeof meta[\"openai/session\"] === \"string\" && meta[\"openai/session\"] ? (meta[\"openai/session\"] as string) : null;\n  if (at.events.native?.(extra)) return { native: true, conversation };\n  if (conversation) return { native: true, conversation };\n  if (dialectOf(at.clientOf(meta, identified), userAgentOf(extra)) === \"chatgpt\") return { native: true, conversation: null };\n  const caps = at.server.server.getClientCapabilities() as Record<string, unknown> | undefined;\n  if (hasKey(caps?.extensions, EXTENSION_KEY)) return { native: true, conversation: null };\n  const metaCaps = meta[\"io.modelcontextprotocol/clientCapabilities\"] as Record<string, unknown> | undefined;\n  if (hasKey(metaCaps?.extensions, EXTENSION_KEY)) return { native: true, conversation: null };\n  return { native: false, conversation: null };\n}\n\n/**\n * Wakeshift attached to an MCP server: the handle `wake.attach` returns.\n * Wrap a tool's callback with events, and give its input schema\n * Wakeshift's three arguments with inputs.\n */\nexport class Attached {\n  /** @internal */\n  private readonly wrapped = new Map<string, unknown>();\n\n  /** @internal Use wake.attach. */\n  constructor(private readonly at: AttachContext) {}\n\n  /**\n   * shape plus Wakeshift's three optional arguments (session_key, harness\n   * and conversation_id), for a wrapped tool's inputSchema:\n   * `inputSchema: attached.inputs({ table: z.string() })`.\n   */\n  inputs<S extends ZodRawShapeCompat>(shape: S): S & WakeInputs {\n    for (const k of WAKE_KEYS) if (k in (shape ?? {})) throw new Error(`wakeshift: attached.inputs adds ${k} itself: leave it out of the shape`);\n    return { ...shape, ...fieldsFor(shape, \"inputs\") };\n  }\n\n  /**\n   * shape plus an optional wakeshift string, for a wrapped tool's\n   * outputSchema. Optional: every structured result of a wrapped tool gets\n   * the block's text (without the beacon line) in wakeshift, for hosts that\n   * show the model only structuredContent, and tools/list names it in the\n   * tool's output schema; this declares it in your own schema and types.\n   * The block stays in the content too.\n   */\n  output<S extends ZodRawShapeCompat>(shape: S): S & WakeOutputShape {\n    if (\"wakeshift\" in (shape ?? {})) throw new Error(\"wakeshift: attached.output adds wakeshift itself: leave it out of the shape\");\n    return { ...shape, ...fieldsFor(shape, \"output\") };\n  }\n\n  /**\n   * Wraps a tool's callback, for `server.registerTool(name, …)` with the\n   * same name (MCP doesn't tell a callback its tool's name): the handler\n   * gets the tool's own arguments and a ToolCall, and returns a string or\n   * a CallToolResult; the reply gets Wakeshift's block. The tool's\n   * inputSchema must come from attached.inputs (a type error otherwise).\n   * Register it on the server passed to attach.\n   */\n  events<S extends WakeShape>(name: string, handler: (args: AppArgs<S>, call: ToolCall) => WrappedResult | Promise<WrappedResult>): ToolCallback<S> {\n    if (!name || typeof name !== \"string\") throw new Error(\"wakeshift: attached.events needs the tool's name, as registered\");\n    if (typeof handler !== \"function\") throw new Error(\"wakeshift: attached.events needs a handler\");\n    const cb = (async (args: Record<string, unknown>, extra: ToolCallExtra) => this.run(name, cb, handler as any, args ?? {}, extra)) as unknown as ToolCallback<S>;\n    this.wrapped.set(name, cb);\n    return cb;\n  }\n\n  private async run(\n    name: string,\n    cb: unknown,\n    handler: (args: Record<string, unknown>, call: ToolCall) => WrappedResult | Promise<WrappedResult>,\n    input: Record<string, unknown>,\n    extra: ToolCallExtra,\n  ): Promise<CallToolResult> {\n    const tool = registeredTool(this.at.server, name);\n    if (tool === undefined || (tool && (tool.handler ?? tool.callback) !== cb)) {\n      throw new Error(`wakeshift: attached.events(\"${name}\", …) isn't the tool named ${name} on the server passed to attach: register it with the same name, on that server`);\n    }\n    let outputSchema: AnySchema | undefined;\n    if (tool) {\n      const shape = getObjectShape(normalizeObjectSchema(tool.inputSchema));\n      if (!shape || WAKE_KEYS.some((k) => !(k in shape))) {\n        throw new Error(`wakeshift: the tool ${name} doesn't take Wakeshift's arguments: make its inputSchema with attached.inputs(…)`);\n      }\n      outputSchema = tool.outputSchema;\n    } else {\n      this.at.warnOnce(\"internals\", \"wakeshift: this MCP SDK keeps its tools where Wakeshift can't read them, so wrapped tools' registration and output aren't checked\");\n    }\n    const str = (k: string) => (typeof input[k] === \"string\" && (input[k] as string) ? (input[k] as string) : undefined);\n    const call = new CallScope(this.at, name, extra, { session: str(SESSION_ARG), harness: str(HARNESS_ARG), conversation: str(CONVERSATION_ARG) });\n    const appArgs: Record<string, unknown> = {};\n    for (const [k, v] of Object.entries(input)) if (!(WAKE_KEYS as readonly string[]).includes(k)) appArgs[k] = v;\n\n    let result: WrappedResult;\n    try {\n      result = await handler(appArgs, call);\n    } catch (err) {\n      await call.end();\n      await call.rollback();\n      throw err;\n    }\n    await call.end();\n    // Rollback is decided from the final result, the one the client would\n    // receive: MCP validates the output after this callback returns, so\n    // the wrapper checks it the same way first.\n    if (result === null || result === undefined || (typeof result !== \"string\" && typeof result !== \"object\")) {\n      await call.rollback();\n      return result as unknown as CallToolResult;\n    }\n    if ((typeof result === \"object\" && result.isError) || extra?.signal?.aborted || !(await validOutput(outputSchema, result))) {\n      await call.rollback();\n      return typeof result === \"string\" ? { content: [{ type: \"text\", text: result }] } : result;\n    }\n    let add: CallAddition | null;\n    try {\n      add = await call.addition();\n    } catch (err) {\n      // The conversation is subscribed but wouldn't learn how it's woken:\n      // the call fails, as one whose handler failed does.\n      await call.rollback();\n      this.at.wake.warn(`wakeshift: Wakeshift's block for a result of ${name} failed, so the call's subscriptions were rolled back: ${detailOf(err)}`);\n      return { isError: true, content: [{ type: \"text\", text: `Wakeshift couldn't finish setting up this conversation's wakes (${detailOf(err)}). Try again.` }] };\n    }\n    const out = mergeBlock(result, add ?? {});\n    if (out.structuredContent && WAKESHIFT_FIELD in out.structuredContent && !(await validOutput(outputSchema, out))) {\n      // A schema that refuses fields it doesn't name (a strict object):\n      // MCP would refuse the result, so it goes without the field.\n      const { [WAKESHIFT_FIELD]: _dropped, ...own } = out.structuredContent;\n      out.structuredContent = own;\n      this.at.warnOnce(\n        \"output:\" + name,\n        `wakeshift: the wrapped tool ${name}'s outputSchema refuses a wakeshift field, so hosts that show only structuredContent won't see how this conversation is woken: make it with attached.output`,\n      );\n    }\n    return out;\n  }\n\n  /**\n   * @internal Wraps the server's tools/list handler so each wrapped tool's\n   * object output schema names wakeshift (withWakeshiftField). McpServer\n   * sets the handler when the first tool is registered, perhaps after\n   * attach, so a later set is wrapped too (as tasks.ts does).\n   */\n  amendToolsList(): void {\n    type Handler = (request: any, extra: any) => Promise<any>;\n    const handlers = (this.at.server.server as any)._requestHandlers as Map<string, Handler> | undefined;\n    if (!(handlers instanceof Map)) return; // a future SDK: schemas stay as registered\n    const wrap =\n      (h: Handler): Handler =>\n      async (request, extra) => {\n        const out = await h(request, extra);\n        if (out && Array.isArray(out.tools)) this.amendTools(out.tools);\n        return out;\n      };\n    const h = handlers.get(\"tools/list\");\n    if (h) Map.prototype.set.call(handlers, \"tools/list\", wrap(h));\n    const set = handlers.set.bind(handlers);\n    handlers.set = (method: string, h: Handler) => set(method, method === \"tools/list\" ? wrap(h) : h);\n  }\n\n  private amendTools(tools: unknown[]): void {\n    for (let i = 0; i < tools.length; i++) {\n      const t = tools[i] as { name?: unknown; outputSchema?: unknown } | null;\n      if (!t || typeof t !== \"object\" || typeof t.name !== \"string\" || !this.wrapped.has(t.name) || !t.outputSchema) continue;\n      const schema = withWakeshiftField(t.outputSchema);\n      if (schema !== t.outputSchema) tools[i] = { ...t, outputSchema: schema };\n    }\n  }\n}\n\n// validOutput mirrors McpServer's own check of a result against the tool's\n// outputSchema (validateToolOutput): one it would refuse fails the call.\nasync function validOutput(outputSchema: AnySchema | undefined, result: WrappedResult): Promise<boolean> {\n  if (!outputSchema) return true;\n  if (typeof result === \"string\" || !result.structuredContent) return false;\n  const obj = normalizeObjectSchema(outputSchema);\n  if (!obj) return true;\n  return (await safeParseAsync(obj, result.structuredContent)).success;\n}\n"]}
+{"version":3,"file":"toolcall.js","sourceRoot":"","sources":["../src/toolcall.ts"],"names":[],"mappings":"AAAA,0EAA0E;AAC1E,yEAAyE;AACzE,wEAAwE;AACxE,2EAA2E;AAC3E,2EAA2E;AAC3E,4EAA4E;AAC5E,yEAAyE;AACzE,yDAAyD;AACzD,EAAE;AACF,6FAA6F;AAC7F,gCAAgC;AAChC,iFAAiF;AACjF,2DAA2D;AAC3D,6FAA6F;AAC7F,qDAAqD;AACrD,4CAA4C;AAC5C,UAAU;AAEV,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,MAAM,QAAQ,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,QAAQ,CAAC;AAC7B,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AAEtC,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,UAAU,EAAE,qBAAqB,EAA8E,MAAM,UAAU,CAAC;AAEnK,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAE3C,OAAO,EAAE,QAAQ,EAAE,WAAW,EAA6D,MAAM,gBAAgB,CAAC;AAElH,2EAA2E;AAC3E,6DAA6D;AAC7D,MAAM,WAAW,GAAG,aAAa,CAAC;AAClC,MAAM,WAAW,GAAG,SAAS,CAAC;AAC9B,MAAM,gBAAgB,GAAG,iBAAiB,CAAC;AAI3C,wEAAwE;AACxE,2BAA2B;AAC3B,MAAM,uBAAuB,GAC3B,mKAAmK,CAAC;AACtK,MAAM,mBAAmB,GACvB,6QAA6Q,CAAC;AAChR,MAAM,wBAAwB,GAC5B,4MAA4M,CAAC;AAC/M,MAAM,4BAA4B,GAAG,gFAAgF,CAAC;AACtH,mIAAmI;AACnI,MAAM,CAAC,MAAM,qBAAqB,GAAG,EAAE,WAAW,EAAE,uBAAuB,EAAE,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,wBAAwB,EAAW,CAAC;AAEhK,MAAM,UAAU,GAAG;IACjB,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;IACtE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;IAClE,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;CAC7E,CAAC;AACF,yIAAyI;AACzI,MAAM,CAAC,MAAM,QAAQ,GAAG,UAAU,CAAC;AACnC,MAAM,UAAU,GAAG,EAAE,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC,EAAE,CAAC;AAE/F,2EAA2E;AAC3E,4EAA4E;AAC5E,uCAAuC;AACvC,MAAM,SAAS,GAAG;IAChB,EAAE,EAAE;QACF,MAAM,EAAE;YACN,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;YACvE,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;YACnE,CAAC,gBAAgB,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;SAC9E;QACD,MAAM,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC,EAAE;KACrF;IACD,EAAE,EAAE;QACF,MAAM,EAAE;YACN,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;YACvE,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;YACnE,CAAC,gBAAgB,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;SAC9E;QACD,MAAM,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC,EAAE;KACrF;CACF,CAAC;AAEF,wEAAwE;AACxE,wEAAwE;AACxE,gCAAgC;AAChC,SAAS,SAAS,CAAgC,KAAwB,EAAE,IAAO,EAAE,IAAW;IAC9F,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5C,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;IAC9F,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC;IAChG,OAAO,MAAa,CAAC;AACvB,CAAC;AAED,kHAAkH;AAClH,MAAM,UAAU,aAAa,CAAC,IAAU;IACtC,OAAO,IAAI,CAAC,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC;AAC/D,CAAC;AAED,4EAA4E;AAC5E,sCAAsC;AACtC,SAAS,UAAU,CAAC,MAAiB,EAAE,MAAyB;IAC9D,MAAM,CAAC,GAAG,MAA0D,CAAC;IACrE,IAAI,OAAO,CAAC,CAAC,MAAM,KAAK,UAAU;QAAE,OAAO,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC5D,OAAO,MAAM,CAAC,MAAM,CAAC,MAAa,EAAE,MAAa,CAAc,CAAC;AAClE,CAAC;AA+BD,MAAM,SAAS,GAAuB,CAAC,WAAW,EAAE,WAAW,EAAE,gBAAgB,CAAC,CAAC;AAEnF;;;;GAIG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAsB;IAClD,kBAAkB;IAClB,oCAAoC;IACpC,uBAAuB;IACvB,wBAAwB;IACxB,iCAAiC;CAClC,CAAC;AAyEF,kDAAkD;AAClD,MAAM,OAAO,SAAU,SAAQ,KAAK;IAClC;QACE,KAAK,CAAC,0EAA0E,CAAC,CAAC;QAClF,IAAI,CAAC,IAAI,GAAG,WAAW,CAAC;IAC1B,CAAC;CACF;AAaD,yEAAyE;AACzE,iEAAiE;AACjE,MAAM,kBAAkB,GAAG,yEAAyE,CAAC;AACrG,MAAM,kBAAkB,GAAG,qBAAqB,CAAC;AACjD,sDAAsD;AACtD,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAW5B,mEAAmE;AACnE,qEAAqE;AACrE,iBAAiB;AACjB,SAAS,WAAW,CAAC,OAAkB;IACrC,MAAM,IAAI,GAAG,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5B,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM;QAAE,OAAO,EAAE,CAAC;IAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;IAClD,MAAM,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IACjC,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAC/C,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;QAAE,OAAO,EAAE,CAAC;IAC3E,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACvE,IAAI,IAAI,KAAK,EAAE;QAAE,OAAO,CAAC,GAAG,EAAE,CAAC;;QAC1B,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IAC3D,OAAO,IAAI,CAAC;AACd,CAAC;AAED,2HAA2H;AAC3H,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC;AAE3C;;;;;;GAMG;AACH,MAAM,UAAU,kBAAkB,CAAC,MAAe;IAChD,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO,MAAM,CAAC;IAClF,MAAM,CAAC,GAAG,MAAiC,CAAC;IAC5C,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC;IACvC,MAAM,KAAK,GAAG,CAAC,CAAC,UAAU,CAAC;IAC3B,IAAI,KAAK,KAAK,SAAS,IAAI,CAAC,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAAE,OAAO,MAAM,CAAC;IACxG,IAAI,KAAK,IAAI,eAAe,IAAK,KAAgB;QAAE,OAAO,MAAM,CAAC;IACjE,OAAO,EAAE,GAAG,CAAC,EAAE,UAAU,EAAE,EAAE,GAAG,CAAE,KAAgB,IAAI,EAAE,CAAC,EAAE,CAAC,eAAe,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,4BAA4B,EAAE,EAAE,EAAE,CAAC;AAClJ,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,UAAU,CAAC,MAAqB,EAAE,GAAiB;IACjE,MAAM,QAAQ,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC;IAC5C,MAAM,IAAI,GAAmB,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;IAC/F,MAAM,GAAG,GAAmB,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;IAC5E,MAAM,IAAI,GAAG,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACrC,MAAM,UAAU,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,IAAI,kBAAkB,IAAI,GAAG,CAAC,KAAK,CAAC;IAClE,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC1C,IAAI,IAAI;QAAE,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,UAAU,IAAI,QAAQ,KAAK,EAAE,CAAC;IAC1C,IAAI,QAAQ;QAAE,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;SAC5D,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,OAAO;QAAE,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;IACpF,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE,CAAC;QACpD,IAAI,CAAC,KAAK,kBAAkB,IAAI,GAAG;YAAE,SAAS;QAC9C,GAAG,CAAC,KAAK,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;IAC/C,CAAC;IACD,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;QAChG,GAAG,CAAC,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7D,CAAC;IACD,IAAI,IAAI,IAAI,GAAG,CAAC,iBAAiB,IAAI,OAAO,GAAG,CAAC,iBAAiB,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACxH,GAAG,CAAC,iBAAiB,GAAG,EAAE,GAAI,GAAG,CAAC,iBAA6C,EAAE,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,CAAC;IAC7G,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,0EAA0E;AAC1E,4EAA4E;AAC5E,SAAS,UAAU,CAAC,IAAyC,EAAE,GAAW;IACxE,IAAI,CAAC,IAAI;QAAE,OAAO,EAAE,CAAC;IACrB,MAAM,KAAK,GAAG,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAClF,IAAI,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAC3D,IAAI,GAAG,GAAY,IAAI,CAAC;IACxB,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;QAClC,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC;YAAE,OAAO,EAAE,CAAC;QACrE,GAAG,GAAI,GAA+B,CAAC,IAAI,CAAC,CAAC;IAC/C,CAAC;IACD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC;AACpB,CAAC;AAED,mIAAmI;AACnI,MAAM,UAAU,iBAAiB,CAAC,IAAyC;IACzE,OAAO,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;AACnE,CAAC;AAED,SAAS,QAAQ,CAAC,CAAU,EAAE,CAAU;IACtC,MAAM,IAAI,GAAG,CAAC,CAAU,EAAW,EAAE,CACnC,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,SAAS;QAC3B,CAAC,CAAC,EAAE;QACJ,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;YAChB,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC;YACb,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ;gBACrB,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAW,CAAC,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAE,CAA6B,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9G,CAAC,CAAC,CAAC,CAAC;IACZ,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;AAC7D,CAAC;AAED,SAAS,QAAQ,CAAC,GAAY;IAC5B,OAAO,GAAG,YAAY,cAAc,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAE,GAAa,EAAE,OAAO,IAAI,GAAG,CAAC,CAAC;AAC7F,CAAC;AA2BD,qGAAqG;AACrG,MAAM,SAAS;IAoBM,EAAE;IAGF,IAAI;IAtBvB,IAAI,CAAqB;IACzB,IAAI,CAAS;IACb,MAAM,GAAG,KAAK,CAAC;IACf,kBAAkB,GAAkB,IAAI,CAAC;IACzC,OAAO,CAAgB;IACvB,UAAU,GAAsB,IAAI,CAAC;IACrC,KAAK,CAAI;IAET,KAAK,GAAG,KAAK,CAAC;IACd,OAAO,GAAG,EAAE,CAAC;IACb,KAAK,GAAG,KAAK,CAAC;IACd,OAAO,GAAiB,EAAE,CAAC;IAC3B,WAAW,GAAiB,EAAE,CAAC;IAC/B,QAAQ,GAAG,IAAI,GAAG,EAAoB,CAAC;IAC/B,KAAK,GAAqB,OAAO,CAAC,OAAO,EAAE,CAAC;IACpD,kEAAkE;IACjD,UAAU,CAAqB;IAEhD,YACmB,EAAiB,EAClC,IAAY,EACZ,KAAQ,EACS,IAAmE;kBAHnE,EAAE;oBAGF,IAAI;QAErB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,MAAM,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC9B,IAAI,CAAC,IAAI,GAAG,EAAE,EAAE,IAAI,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,EAAE,EAAE,MAAM,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC;QACpC,MAAM,CAAC,GAAG,QAAQ,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;QACvB,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC,YAAY,CAAC;IAC3C,CAAC;IAED,0EAA0E;IAC1E,qEAAqE;IACrE,6CAA6C;IAC7C,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO;QACzB,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;QACtB,OAAO,QAAQ,CAAC;YACd,IAAI;YACJ,UAAU,EAAE,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC;YACnD,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,OAAO,EAAE,OAAO,IAAI,SAAS;YAC7B,SAAS,EAAE,WAAW,CAAC,GAAG,CAAC;YAC3B,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO;YAC1B,YAAY,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY;YACpC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC;SAC7B,CAAC,CAAC;IACL,CAAC;IAED,yEAAyE;IACzE,oEAAoE;IAC5D,OAAO,CAAI,EAAoB;QACrC,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC,CAAC;QACvD,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACzF,IAAI,CAAC,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACxC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACvB,GAAG,CAAC,IAAI,CACN,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,EAC/B,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAChC,CAAC;QACF,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,UAAU,CAAC,KAA0B;QAC3C,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,CAAC,KAAK,KAAK,UAAU,IAAI,KAAK,CAAC,KAAK,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YAClF,MAAM,IAAI,cAAc,CAAC,GAAG,EAAE,eAAe,EAAE,GAAG,KAAK,EAAE,IAAI,IAAI,MAAM,oDAAoD,CAAC,CAAC;QAC/H,CAAC;IACH,CAAC;IAED,SAAS,CAAsB,KAAwB,EAAE,IAAoB,EAAE,IAAI,GAAyB,EAAE;QAC5G,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE;YAC7B,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACvB,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gBAClC,OAAO,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;YAC/I,CAAC;YACD,IAAI,KAAK,CAAC,QAAQ,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACjC,MAAM,IAAI,cAAc,CAAC,GAAG,EAAE,eAAe,EAAE,GAAG,KAAK,CAAC,IAAI,gGAAgG,CAAC,CAAC;YAChK,CAAC;YACD,MAAM,IAAI,GAA4B,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC;YACtH,IAAI,IAAI,CAAC,OAAO;gBAAE,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;YAC9C,IAAI,IAAI,CAAC,GAAG,KAAK,SAAS;gBAAE,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;YAChD,IAAI,IAAI,CAAC,MAAM;gBAAE,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;YAC3C,IAAI,IAAI,CAAC,KAAK;gBAAE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;YACxC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAKtC,MAAM,EAAE,sBAAsB,EAAE,IAAI,CAAC,CAAC;YACzC,IAAI,GAAG,CAAC,OAAO;gBAAE,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;YAC5C,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;YAClB,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;YAC3B,MAAM,GAAG,GAAG,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACpC,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC;YACtB,MAAM,EAAE,GAAG,GAAG,CAAC,kBAAkB,CAAC;YAClC,IAAI,EAAE,CAAC,OAAO;gBAAE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;YACtI,OAAO,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;QACtI,CAAC,CAAC,CAAC;IACL,CAAC;IAED,WAAW,CAAsB,KAAwB,EAAE,IAAoB;QAC7E,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE;YAC7B,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACvB,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,IAAI,CAAC,MAAM;gBAAE,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAC1D,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAsB,MAAM,EAAE,kCAAkC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;YACtK,OAAO,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC;QACrB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE;YAC7B,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO;gBAAE,OAAO,IAAI,CAAC,OAAO,CAAC;YACtD,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAoE,MAAM,EAAE,gBAAgB,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC9J,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;YAC3B,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;YAC3B,IAAI,CAAC,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC1C,OAAO,GAAG,CAAC,OAAO,CAAC;QACrB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,iFAAiF;IACzE,KAAK,CAAC,SAAS,CAAC,IAAY,EAAE,IAAa;QACjD,IAAI,CAAC,IAAI,CAAC,IAAI;YAAE,OAAO,KAAK,CAAC;QAC7B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAA+F,MAAM,EAAE,2BAA2B,EAAE;YAC3K,IAAI,EAAE,IAAI,CAAC,IAAI;SAChB,CAAC,CAAC;QACH,IAAI,KAAK,GAAG,KAAK,CAAC;QAClB,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,aAAa,IAAI,EAAE,EAAE,CAAC;YACxC,IAAI,CAAC,CAAC,KAAK,KAAK,IAAI,IAAI,CAAC,CAAC,WAAW,KAAK,SAAS,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,EAAE,IAAI,CAAC;gBAAE,SAAS;YAC9F,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAsB,MAAM,EAAE,0BAA0B,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAChI,KAAK,GAAG,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QAC7B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,8DAA8D;IACtD,SAAS,CAAC,IAAY,EAAE,IAAa;QAC3C,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,IAAI,QAAQ,CAAC,CAAC,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAAE,OAAO;QACzF,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,KAAK,gBAAgB,EAAE,CAAC;YACjD,IAAI,CAAC,EAAE,CAAC,QAAQ,CACd,SAAS,GAAG,IAAI,CAAC,IAAI,EACrB,+BAA+B,IAAI,CAAC,IAAI,4CAA4C,gBAAgB,6BAA6B,gBAAgB,YAAY,CAC9J,CAAC;YACF,OAAO;QACT,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,CAAC;IAED,oEAAoE;IACpE,2EAA2E;IAC3E,KAAK,CAAC,GAAG;QACP,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,EAAE,CAAC,QAAQ,CACd,WAAW,GAAG,IAAI,CAAC,IAAI,EACvB,+BAA+B,IAAI,CAAC,IAAI,+IAA+I,CACxL,CAAC;YACF,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC/C,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;IACpB,CAAC;IAED,sEAAsE;IACtE,uEAAuE;IACvE,yEAAyE;IACzE,yBAAyB;IACzB,KAAK,CAAC,QAAQ;QACZ,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC7B,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,kCAAkC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,SAAS,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YAC7L,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,8DAA8D,IAAI,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,MAAM,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC7H,CAAC;QACH,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;IACpB,CAAC;IAED,sEAAsE;IACtE,mEAAmE;IACnE,sEAAsE;IACtE,qEAAqE;IACrE,yEAAyE;IACzE,0EAA0E;IAC1E,8DAA8D;IAC9D,KAAK,CAAC,QAAQ;QACZ,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QACzB,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC;QACxI,MAAM,IAAI,GAA4B,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QACrD,IAAI,IAAI,CAAC,OAAO;YAAE,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC9C,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC;YAAE,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC;QAChE,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAe,MAAM,EAAE,iBAAiB,EAAE,IAAI,CAAC,CAAC;QACtF,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC;gBAAE,MAAM,GAAG,CAAC;YACzD,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,kDAAkD,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACnG,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;CACF;AAED,SAAS,YAAY,CAAC,CAAkC;IACtD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,IAAI,IAAI,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,IAAI,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,IAAI,IAAI,EAAE,CAAC;AAC3F,CAAC;AAED,SAAS,MAAM,CAAC,CAAU,EAAE,GAAW;IACrC,OAAO,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAK,CAA6B,CAAC,GAAG,CAAC,KAAK,SAAS,CAAC;AAChH,CAAC;AAED,SAAS,QAAQ,CAAC,CAAU;IAC1B,OAAO,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AAC3D,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,QAAQ,CAAC,EAAuD,EAAE,KAAc,EAAE,UAAmB;IACnH,MAAM,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,MAAM,IAAI,GAAG,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAA4B,CAAC;IACzD,MAAM,YAAY,GAAG,OAAO,IAAI,CAAC,gBAAgB,CAAC,KAAK,QAAQ,IAAI,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAE,IAAI,CAAC,gBAAgB,CAAY,CAAC,CAAC,CAAC,IAAI,CAAC;IACtI,IAAI,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;IACrE,IAAI,YAAY;QAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;IACxD,IAAI,SAAS,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,KAAK,SAAS;QAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;IAC1H,MAAM,IAAI,GAAG,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC1C,IAAI,MAAM,CAAC,IAAI,EAAE,UAAU,EAAE,aAAa,CAAC;QAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;IACzF,MAAM,QAAQ,GAAG,IAAI,CAAC,4CAA4C,CAAwC,CAAC;IAC3G,IAAI,MAAM,CAAC,QAAQ,EAAE,UAAU,EAAE,aAAa,CAAC;QAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;IAC7F,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;AAC/C,CAAC;AAED;;;;;;GAMG;AACH,MAAM,OAAO,QAAQ;IAKU,EAAE;IAJ/B,gBAAgB;IACC,OAAO,GAAG,IAAI,GAAG,EAAmB,CAAC;IAEtD,iCAAiC;IACjC,YAA6B,EAAiB;kBAAjB,EAAE;IAAkB,CAAC;IAYlD,MAAM,CAAC,KAAoC;QACzC,IAAI,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACpB,MAAM,GAAG,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;YAClC,IAAI,CAAC,GAAG;gBAAE,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;YACtF,KAAK,MAAM,CAAC,IAAI,SAAS;gBAAE,IAAI,CAAC,IAAI,GAAG;oBAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,qCAAqC,CAAC,CAAC;YACpI,OAAO,UAAU,CAAC,KAAK,EAAE,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC;QAC9E,CAAC;QACD,KAAK,MAAM,CAAC,IAAI,SAAS;YAAE,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,oCAAoC,CAAC,CAAC;QAC7I,OAAO,EAAE,GAAG,KAAK,EAAE,GAAG,SAAS,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;IACnE,CAAC;IAcD,MAAM,CAAC,KAAoC;QACzC,IAAI,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACpB,MAAM,GAAG,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;YAClC,IAAI,CAAC,GAAG;gBAAE,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;YACtF,IAAI,eAAe,IAAI,GAAG;gBAAE,MAAM,IAAI,KAAK,CAAC,8EAA8E,CAAC,CAAC;YAC5H,OAAO,UAAU,CAAC,KAAK,EAAE,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC;QAC9E,CAAC;QACD,IAAI,WAAW,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,6EAA6E,CAAC,CAAC;QACjI,OAAO,EAAE,GAAG,KAAK,EAAE,GAAG,SAAS,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;IACnE,CAAC;IAED;;;;;;;OAOG;IACH,MAAM,CAAqB,IAAY,EAAE,OAA8F;QACrI,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;QAC1H,IAAI,OAAO,OAAO,KAAK,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QACjG,MAAM,EAAE,GAAG,CAAC,KAAK,EAAE,IAA6B,EAAE,KAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,EAAE,OAAc,EAAE,IAAI,IAAI,EAAE,EAAE,KAAK,CAAC,CAAqC,CAAC;QAC1J,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC3B,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,KAAK,CAAC,GAAG,CACf,IAAY,EACZ,EAAW,EACX,OAAqG,EACrG,KAA8B,EAC9B,KAAQ;QAER,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;QAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,IAAI,KAAK,SAAS,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC;YAC3E,MAAM,IAAI,KAAK,CAAC,+BAA+B,IAAI,8BAA8B,IAAI,iFAAiF,CAAC,CAAC;QAC1K,CAAC;QACD,IAAI,YAAqB,CAAC;QAC1B,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,KAAK,GAAG,cAAc,CAAC,qBAAqB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;YACtE,IAAI,CAAC,KAAK,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,EAAE,CAAC;gBACnD,MAAM,IAAI,KAAK,CAAC,uBAAuB,IAAI,mFAAmF,CAAC,CAAC;YAClI,CAAC;YACD,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,WAAW,EAAE,mIAAmI,CAAC,CAAC;QACrK,CAAC;QACD,MAAM,GAAG,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAK,KAAK,CAAC,CAAC,CAAY,CAAC,CAAC,CAAE,KAAK,CAAC,CAAC,CAAY,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACrH,MAAM,IAAI,GAAG,IAAI,SAAS,CAAI,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,WAAW,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,WAAW,CAAC,EAAE,YAAY,EAAE,GAAG,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;QACnJ,MAAM,OAAO,GAA4B,EAAE,CAAC;QAC5C,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC;YAAE,IAAI,CAAE,SAA+B,CAAC,QAAQ,CAAC,CAAC,CAAC;gBAAE,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAE9G,IAAI,MAAqB,CAAC;QAC1B,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACxC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;YACjB,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtB,MAAM,GAAG,CAAC;QACZ,CAAC;QACD,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;QACjB,sEAAsE;QACtE,oEAAoE;QACpE,4CAA4C;QAC5C,IAAI,MAAM,KAAK,IAAI,IAAI,MAAM,KAAK,SAAS,IAAI,CAAC,OAAO,MAAM,KAAK,QAAQ,IAAI,OAAO,MAAM,KAAK,QAAQ,CAAC,EAAE,CAAC;YAC1G,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtB,OAAO,MAAmC,CAAC;QAC7C,CAAC;QACD,IAAI,CAAC,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,MAAM,EAAE,OAAO,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC,EAAE,CAAC;YACzI,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtB,OAAO,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;QAC7F,CAAC;QACD,IAAI,GAAwB,CAAC;QAC7B,IAAI,CAAC;YACH,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC9B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,oEAAoE;YACpE,oDAAoD;YACpD,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtB,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,gDAAgD,IAAI,0DAA0D,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACjJ,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mEAAmE,QAAQ,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC,EAAE,CAAC;QAC/J,CAAC;QACD,MAAM,GAAG,GAAG,UAAU,CAAC,MAAM,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;QAC1C,IAAI,QAAQ,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,eAAe,IAAI,GAAG,CAAC,iBAAiB,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;YAChI,kEAAkE;YAClE,6DAA6D;YAC7D,MAAM,EAAE,CAAC,eAAe,CAAC,EAAE,QAAQ,EAAE,GAAG,GAAG,EAAE,GAAG,GAAG,CAAC,iBAAiB,CAAC;YACtE,GAAG,CAAC,iBAAiB,GAAG,GAAG,CAAC;YAC5B,IAAI,CAAC,EAAE,CAAC,QAAQ,CACd,SAAS,GAAG,IAAI,EAChB,+BAA+B,IAAI,6JAA6J,CACjM,CAAC;QACJ,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED;;;;;OAKG;IACH,cAAc;QAEZ,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACzC,IAAI,CAAC,QAAQ;YAAE,OAAO,CAAC,2CAA2C;QAClE,MAAM,IAAI,GACR,CAAC,CAAU,EAAW,EAAE,CACxB,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;YACvB,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACpC,IAAI,GAAG,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC;gBAAE,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAChE,OAAO,GAAG,CAAC;QACb,CAAC,CAAC;QACJ,MAAM,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACrC,IAAI,CAAC;YAAE,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,QAAQ,CAAC,GAAG,GAAG,CAAC,MAAc,EAAE,CAAU,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,KAAK,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACpG,CAAC;IAEO,UAAU,CAAC,KAAgB;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAsD,CAAC;YACxE,IAAI,CAAC,CAAC,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,YAAY;gBAAE,SAAS;YACxH,MAAM,MAAM,GAAG,kBAAkB,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;YAClD,IAAI,MAAM,KAAK,CAAC,CAAC,YAAY;gBAAE,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC;QAC3E,CAAC;IACH,CAAC;CACF","sourcesContent":["// One way to subscribe from a tool (docs/subscribe-from-tools.md): a tool\n// callback wrapped by the handle attach returns (attached.events) gets a\n// ToolCall, and subscribes the calling conversation with call.subscribe\n// (ends it with call.unsubscribe). The wrapper adds one Wakeshift block to\n// whatever the handler returns, so a conversation can't be left unwakeable\n// by a forgotten argument, a wrong tool name or an unreturned result; and a\n// call that fails ends the subscriptions it made. A port of the Go SDK's\n// appkit.Events (pkg/appkit/toolcall.go, attachcall.go).\n//\n//   const attached = wake.attach(server, { identify: (extra) => ({ user: userOf(extra) }) })\n//   server.registerTool(\"join\",\n//     { description: \"…\", inputSchema: attached.inputs({ table: z.string() }) },\n//     attached.events(\"join\", async ({ table }, call) => {\n//       const sub = await call.subscribe(turn, { table }, { reason: `Playing at ${table}` })\n//       await casino.seat(call.user!, table, sub.id)\n//       return `You're seated at ${table}.`\n//     }))\n\nimport { z } from \"zod\";\nimport * as z3 from \"zod/v3\";\nimport * as z4 from \"zod/v4\";\nimport * as z4mini from \"zod/v4-mini\";\nimport type { CallToolResult, ToolCallExtra } from \"./mcp.js\";\nimport { getObjectShape, isSchema, isZ4Schema, normalizeObjectSchema, type AnySchema, type SchemaInput, type ShapeOutput, type ZodRawShapeCompat } from \"./zod.js\";\nimport type { Host } from \"./host.js\";\nimport { WakeshiftError } from \"./errors.js\";\nimport { EXTENSION_KEY, dialectOf } from \"./spec.js\";\nimport { takesTasks } from \"./discover.js\";\nimport type { Identity, WakeEvent } from \"./events.js\";\nimport { packCall, userAgentOf, type Call, type Duration, type Recognized, type Wakeshift } from \"./wakeshift.js\";\n\n// Wakeshift's three arguments (wakeshift.ts's SESSION_ARG, HARNESS_ARG and\n// CONVERSATION_ARG, repeated: that module imports this one).\nconst SESSION_ARG = \"session_key\";\nconst HARNESS_ARG = \"harness\";\nconst CONVERSATION_ARG = \"conversation_id\";\n\nexport type { ToolCallExtra };\n\n// What the agent is told each of Wakeshift's arguments is: the Go SDK's\n// WakeArgs, word for word.\nconst SESSION_KEY_DESCRIPTION =\n  \"This conversation's session_key, from an earlier result (a secret token, or a handle such as claude-code#2). Keep it to this conversation; don't share or log it.\";\nconst HARNESS_DESCRIPTION =\n  \"The program you're running in: codex, claude-code, cursor (local CLI or IDE), cursor-cloud (Cursor cloud agent), grok-bot, copilot-cli, opencode, antigravity, openclaw (even when it runs your turns in Codex), hermes, chatgpt, or your program's name if it isn't listed\";\nconst CONVERSATION_DESCRIPTION =\n  \"Your own ID for this conversation, if your program gives you one. Cursor and Grok Bot: the output of `printenv CURSOR_CONVERSATION_ID`. Codex and Antigravity send it automatically; others leave it empty\";\nconst WAKESHIFT_OUTPUT_DESCRIPTION = \"What Wakeshift adds for this conversation: how it's woken by this app's events\";\n/** @internal Wakeshift's three arguments' descriptions, for a tool whose input schema is written out (the wait tool, tasks.ts). */\nexport const WAKE_ARG_DESCRIPTIONS = { session_key: SESSION_KEY_DESCRIPTION, harness: HARNESS_DESCRIPTION, conversation_id: CONVERSATION_DESCRIPTION } as const;\n\nconst wakeInputs = {\n  [SESSION_ARG]: z.string().optional().describe(SESSION_KEY_DESCRIPTION),\n  [HARNESS_ARG]: z.string().optional().describe(HARNESS_DESCRIPTION),\n  [CONVERSATION_ARG]: z.string().optional().describe(CONVERSATION_DESCRIPTION),\n};\n/** @internal Wakeshift's three arguments as attach's own tools take them: worded as a wrapped tool's (test/events-list.json pins it). */\nexport const toolArgs = wakeInputs;\nconst wakeOutput = { wakeshift: z.string().optional().describe(WAKESHIFT_OUTPUT_DESCRIPTION) };\n\n// The same in each of zod's two APIs: MCP refuses a shape that mixes them,\n// so a shape written with zod/v3 (or zod/v4) gets Wakeshift's fields in its\n// own (zod 3.25 and later carry both).\nconst byVersion = {\n  v3: {\n    inputs: {\n      [SESSION_ARG]: z3.string().optional().describe(SESSION_KEY_DESCRIPTION),\n      [HARNESS_ARG]: z3.string().optional().describe(HARNESS_DESCRIPTION),\n      [CONVERSATION_ARG]: z3.string().optional().describe(CONVERSATION_DESCRIPTION),\n    },\n    output: { wakeshift: z3.string().optional().describe(WAKESHIFT_OUTPUT_DESCRIPTION) },\n  },\n  v4: {\n    inputs: {\n      [SESSION_ARG]: z4.string().optional().describe(SESSION_KEY_DESCRIPTION),\n      [HARNESS_ARG]: z4.string().optional().describe(HARNESS_DESCRIPTION),\n      [CONVERSATION_ARG]: z4.string().optional().describe(CONVERSATION_DESCRIPTION),\n    },\n    output: { wakeshift: z4.string().optional().describe(WAKESHIFT_OUTPUT_DESCRIPTION) },\n  },\n};\n\n// fieldsFor is Wakeshift's fields in the zod API shape's own fields use\n// (for an empty shape, the default import's, or zod 4's for the MCP SDK\n// 2.x, which takes only zod 4).\nfunction fieldsFor<K extends \"inputs\" | \"output\">(shape: ZodRawShapeCompat, kind: K, host?: Host): K extends \"inputs\" ? WakeInputs : WakeOutputShape {\n  const first = Object.values(shape ?? {})[0];\n  const version = first ? (isZ4Schema(first) ? \"v4\" : \"v3\") : host?.version === 2 ? \"v4\" : null;\n  const fields = version ? byVersion[version][kind] : kind === \"inputs\" ? wakeInputs : wakeOutput;\n  return fields as any;\n}\n\n/** @internal Wakeshift's three arguments as attach's own tools take them, in the zod the host's MCP SDK takes. */\nexport function wakeInputsFor(host: Host): ZodRawShapeCompat {\n  return host.version === 2 ? byVersion.v4.inputs : wakeInputs;\n}\n\n// withFields is an object schema with fields added, in the schema's own zod\n// API (zod 3, zod 4 classic or mini).\nfunction withFields(schema: AnySchema, fields: ZodRawShapeCompat): AnySchema {\n  const s = schema as { extend?: (f: ZodRawShapeCompat) => AnySchema };\n  if (typeof s.extend === \"function\") return s.extend(fields);\n  return z4mini.extend(schema as any, fields as any) as AnySchema;\n}\n\n/** Wakeshift's three arguments, as attached.inputs adds them. */\nexport type WakeInputs = typeof wakeInputs;\n/** The optional wakeshift string attached.output adds. */\nexport type WakeOutputShape = typeof wakeOutput;\n/** Wakeshift's three arguments in zod 4, as attached.inputs adds them to a zod 4 object. */\nexport type WakeInputsV4 = typeof byVersion.v4.inputs;\n/** Wakeshift's three arguments in zod 3, as attached.inputs adds them to a zod 3 object. */\nexport type WakeInputsV3 = typeof byVersion.v3.inputs;\n/** The optional wakeshift string in zod 4, as attached.output adds it to a zod 4 object. */\nexport type WakeOutputV4 = typeof byVersion.v4.output;\n/** The optional wakeshift string in zod 3, as attached.output adds it to a zod 3 object. */\nexport type WakeOutputV3 = typeof byVersion.v3.output;\n/** What a wrapped tool's arguments must include: Wakeshift's three (attached.inputs adds them). */\nexport type WakeArgs = { [K in WakeKey]?: string | undefined };\n/**\n * The callback attached.events returns, for `server.registerTool`: it takes\n * the tool's arguments (A, Wakeshift's three included) and your MCP SDK's\n * request context (C).\n */\nexport type WrappedCallback<A, C> = (args: A, extra: C) => Promise<any>;\n/** The names of Wakeshift's three arguments. */\nexport type WakeKey = typeof SESSION_ARG | typeof HARNESS_ARG | typeof CONVERSATION_ARG;\n/** An input shape with Wakeshift's three arguments in it (attached.inputs'). */\nexport type WakeShape = ZodRawShapeCompat & { [K in WakeKey]: AnySchema };\n/** What a wrapped handler gets: the tool's own arguments, without Wakeshift's three. */\nexport type AppArgs<S extends ZodRawShapeCompat> = Omit<ShapeOutput<S>, WakeKey>;\n/** What a wrapped handler returns: a string (one text block) or a CallToolResult. */\nexport type WrappedResult = string | CallToolResult;\n\nconst WAKE_KEYS: readonly WakeKey[] = [SESSION_ARG, HARNESS_ARG, CONVERSATION_ARG];\n\n/**\n * @internal The tools/call _meta keys a harness names its conversation in\n * (the core's harnesses ConversationMeta, dotted for nested objects):\n * test/conversation-meta.json, which a Go test keeps current.\n */\nexport const CONVERSATION_META: readonly string[] = [\n  \"agent-session-id\",\n  \"antigravity.google/conversation_id\",\n  \"vscode.conversationId\",\n  \"wakeshift/conversation\",\n  \"x-codex-turn-metadata.thread_id\",\n];\n\n/**\n * The calling conversation's subscription, made or renewed by\n * call.subscribe. id and user address it later: emit's\n * `to: [{ user, subscription: id }]`, or `event.end(user, id)`.\n */\nexport interface Subscription {\n  id: string;\n  user: string;\n  /** False for a renewal. */\n  created: boolean;\n  /** The conversation's session. A secret: don't show, store or log it. */\n  session: string;\n  /** What Wakeshift recognized the agent's program as. */\n  recognized: Recognized;\n  /** The caller subscribes itself (call.native): nothing was made, id is \"\", and the block tells it what to subscribe to. */\n  native: boolean;\n}\n\n/** How call.subscribe subscribes. */\nexport interface ToolSubscribeOptions {\n  /** End after this long without a delivery or a fresh subscribe (default 30 days). */\n  ttl?: Duration;\n  /** Why the app subscribes the conversation, shown in the console (at most 200 characters). */\n  reason?: string;\n  /** Names a new session (agent@label), for an app acting for its agents. */\n  label?: string;\n}\n\n/**\n * One call of a wrapped tool (attached.events): who's calling, and the\n * calls that subscribe the calling conversation. Only the wrapper makes\n * one, and it works until the handler returns (then CallEnded).\n */\nexport interface ToolCall<C = any> {\n  /** Who attach's identify says is calling (undefined for none). */\n  readonly user: string | undefined;\n  /** The tool being called. */\n  readonly tool: string;\n  /**\n   * The client subscribes to the app's events itself (MCP Events: ChatGPT,\n   * the house agents). call.subscribe then makes no conversation\n   * subscription, and the block tells it what to subscribe to.\n   */\n  readonly native: boolean;\n  /** A native client's conversation, when it names one (ChatGPT's _meta[\"openai/session\"]). */\n  readonly nativeConversation: string | null;\n  /** The session_key passed; after a subscribe or resolveSession, the conversation's session. */\n  readonly session: string | null;\n  /** What Wakeshift recognized the agent's program as: null until a subscribe or resolveSession. */\n  readonly recognized: Recognized | null;\n  /** The call's request context, as your MCP SDK passes it to the tool's callback: 2.x's ctx, 1.x's extra. */\n  readonly extra: C;\n  /**\n   * Subscribes the calling conversation to event with args, or renews its\n   * subscription. The reply then carries what the conversation needs to be\n   * woken (one block for the call, however many subscribes it makes), and\n   * if the call fails, the subscription is ended. For a native caller\n   * nothing is made. Your tool decides who may: authorize isn't consulted.\n   * A personal event needs a user (identify's).\n   */\n  subscribe<A extends AnySchema>(event: WakeEvent<A, any>, args: SchemaInput<A>, opts?: ToolSubscribeOptions): Promise<Subscription>;\n  /**\n   * Ends the calling conversation's subscription to event with args,\n   * resolving to whether one ended. For a native caller it ends the user's\n   * own (webhook) subscriptions to the event with those arguments.\n   */\n  unsubscribe<A extends AnySchema>(event: WakeEvent<A, any>, args: SchemaInput<A>): Promise<boolean>;\n  /** Finds or starts the calling conversation's session without subscribing it (it doesn't make it wakeable). */\n  resolveSession(): Promise<string>;\n}\n\n/** A ToolCall used after its handler returned. */\nexport class CallEnded extends Error {\n  constructor() {\n    super(\"wakeshift: the tool call has ended: subscribe before the handler returns\");\n    this.name = \"CallEnded\";\n  }\n}\n\n/** attach's event options. */\nexport interface EventsOptions<C = any> {\n  /**\n   * Asked first whether a wrapped tool's call comes from a client that\n   * subscribes to the app's events itself (call.native), for callers the\n   * SDK can't tell (an app's own house agents). False leaves it to the\n   * SDK's own checks. It gets the call's request context, as identify does.\n   */\n  native?(extra: C): boolean;\n}\n\n// The line a text-carried beacon is on (the core's beacon.Text), which a\n// harness's hook reads only as the last line of a result's text.\nconst BEACON_LINE_PREFIX = \"(For Wakeshift's hook on this computer, not for you: wakeshift-beacon: \";\nconst CONNECTOR_META_KEY = \"wakeshift/connector\";\n// The most native lines Wakeshift takes for one call.\nconst MAX_NATIVE_LINES = 20;\n\n/** /calls/addition's answer. */\nexport interface CallAddition {\n  text?: string;\n  trailer?: string;\n  meta?: Record<string, unknown>;\n}\n\ntype Content = CallToolResult[\"content\"][number];\n\n// takeTrailer takes a beacon line off the end of the result's text\n// (dropping its block if nothing else is in it), returning it (\"\" if\n// there's none).\nfunction takeTrailer(content: Content[]): string {\n  const last = content.at(-1);\n  if (!last || last.type !== \"text\") return \"\";\n  const text = last.text.replace(/[ \\t\\r\\n]+$/, \"\");\n  const i = text.lastIndexOf(\"\\n\");\n  const line = i >= 0 ? text.slice(i + 1) : text;\n  if (!line.startsWith(BEACON_LINE_PREFIX) || !line.endsWith(\")\")) return \"\";\n  const rest = i >= 0 ? text.slice(0, i).replace(/[ \\t\\r\\n]+$/, \"\") : \"\";\n  if (rest === \"\") content.pop();\n  else content[content.length - 1] = { ...last, text: rest };\n  return line;\n}\n\n/** The structuredContent field a wrapped tool's structured result carries Wakeshift's block in (the same in every SDK). */\nexport const WAKESHIFT_FIELD = \"wakeshift\";\n\n/**\n * @internal An object output schema (JSON Schema, as tools/list gives it)\n * with an optional wakeshift string added, as a copy, so a client that\n * checks structuredContent against it takes the block. Anything else, or a\n * schema that already names it, comes back as it was.\n * test/wrapped-block.json (output_schemas) pins it.\n */\nexport function withWakeshiftField(schema: unknown): unknown {\n  if (!schema || typeof schema !== \"object\" || Array.isArray(schema)) return schema;\n  const s = schema as Record<string, unknown>;\n  if (s.type !== \"object\") return schema;\n  const props = s.properties;\n  if (props !== undefined && (!props || typeof props !== \"object\" || Array.isArray(props))) return schema;\n  if (props && WAKESHIFT_FIELD in (props as object)) return schema;\n  return { ...s, properties: { ...((props as object) ?? {}), [WAKESHIFT_FIELD]: { type: \"string\", description: WAKESHIFT_OUTPUT_DESCRIPTION } } };\n}\n\n/**\n * @internal Adds the call's block to a wrapped handler's result: Wakeshift's\n * text as a block after the app's content, then a text-carried beacon last,\n * and _meta; a string result stays one text block. A result that already\n * carries a beacon keeps it, last, and gets no second one. A structured\n * result (structuredContent an object) gets the block's text in its\n * wakeshift field too. test/wrapped-block.json is the shape, shared with\n * the other SDKs.\n */\nexport function mergeBlock(result: WrappedResult, add: CallAddition): CallToolResult {\n  const isString = typeof result === \"string\";\n  const base: CallToolResult = isString ? { content: [{ type: \"text\", text: result }] } : result;\n  const out: CallToolResult = { ...base, content: [...(base.content ?? [])] };\n  const text = (add.text ?? \"\").trim();\n  const metaBeacon = !!out._meta && CONNECTOR_META_KEY in out._meta;\n  const existing = takeTrailer(out.content);\n  if (text) out.content.push({ type: \"text\", text });\n  const has = metaBeacon || existing !== \"\";\n  if (existing) out.content.push({ type: \"text\", text: existing });\n  else if (!has && add.trailer) out.content.push({ type: \"text\", text: add.trailer });\n  for (const [k, v] of Object.entries(add.meta ?? {})) {\n    if (k === CONNECTOR_META_KEY && has) continue;\n    out._meta = { ...(out._meta ?? {}), [k]: v };\n  }\n  if (isString) {\n    const texts = out.content.map((c) => (c.type === \"text\" ? c.text : \"\")).filter((t) => t !== \"\");\n    out.content = [{ type: \"text\", text: texts.join(\"\\n\\n\") }];\n  }\n  if (text && out.structuredContent && typeof out.structuredContent === \"object\" && !Array.isArray(out.structuredContent)) {\n    out.structuredContent = { ...(out.structuredContent as Record<string, unknown>), [WAKESHIFT_FIELD]: text };\n  }\n  return out;\n}\n\n// metaString is the string at key in a tool call's _meta: the key itself,\n// else a dotted path into nested objects (the core's harnesses.metaString).\nfunction metaString(meta: Record<string, unknown> | undefined, key: string): string {\n  if (!meta) return \"\";\n  const limit = (s: unknown) => (typeof s === \"string\" && s.length <= 200 ? s : \"\");\n  if (typeof meta[key] === \"string\") return limit(meta[key]);\n  let cur: unknown = meta;\n  for (const part of key.split(\".\")) {\n    if (!cur || typeof cur !== \"object\" || Array.isArray(cur)) return \"\";\n    cur = (cur as Record<string, unknown>)[part];\n  }\n  return limit(cur);\n}\n\n/** @internal Whether a tool call's _meta names its conversation, the way a harness puts it there (test/conversation-meta.json). */\nexport function namesConversation(meta: Record<string, unknown> | undefined): boolean {\n  return CONVERSATION_META.some((k) => metaString(meta, k) !== \"\");\n}\n\nfunction sameJSON(a: unknown, b: unknown): boolean {\n  const norm = (v: unknown): unknown =>\n    v === null || v === undefined\n      ? {}\n      : Array.isArray(v)\n        ? v.map(norm)\n        : typeof v === \"object\"\n          ? Object.fromEntries(Object.keys(v as object).sort().map((k) => [k, norm((v as Record<string, unknown>)[k])]))\n          : v;\n  return JSON.stringify(norm(a)) === JSON.stringify(norm(b));\n}\n\nfunction detailOf(err: unknown): string {\n  return err instanceof WakeshiftError ? err.detail : String((err as Error)?.message ?? err);\n}\n\ninterface CreatedSub {\n  name: string;\n  arguments: unknown;\n  session: string;\n  pending: string; // the call's pending-addition record, which made it\n}\n\ninterface NativeLine {\n  name: string;\n  arguments: unknown;\n}\n\n/** @internal What the wrapper needs of attach: the client, the server, who's calling, the event options. */\nexport interface AttachContext {\n  wake: Wakeshift;\n  /** The server, through what attach needs of its MCP SDK (host.ts). */\n  host: Host;\n  /** Who attach's identify says is calling: its user and client (raw: the MCP SDK's own request context). */\n  identify(raw: unknown): Identity | undefined;\n  /** The client's name: the request's (initialize's, else _meta's), else identified (identify's), else \"\". */\n  clientOf(meta: Record<string, unknown> | undefined, identified?: string): string;\n  events: EventsOptions<any>;\n  warnOnce(key: string, message: string): void;\n}\n\n/** One tools/call of a wrapped tool: what its handler subscribed, for the block and the rollback. */\nclass CallScope<C> implements ToolCall<C> {\n  user: string | undefined;\n  tool: string;\n  native = false;\n  nativeConversation: string | null = null;\n  session: string | null;\n  recognized: Recognized | null = null;\n  extra: C;\n\n  ended = false;\n  pending = \"\";\n  armed = false;\n  created: CreatedSub[] = [];\n  nativeLines: NativeLine[] = [];\n  inflight = new Set<Promise<unknown>>();\n  private queue: Promise<unknown> = Promise.resolve();\n  // The client's name identify gave, for a request that names none.\n  private readonly identified: string | undefined;\n\n  constructor(\n    private readonly at: AttachContext,\n    tool: string,\n    extra: C,\n    private readonly args: { session?: string; harness?: string; conversation?: string },\n  ) {\n    this.tool = tool;\n    this.extra = extra;\n    const id = at.identify(extra);\n    this.user = id?.user;\n    this.identified = id?.client;\n    this.session = args.session || null;\n    const n = nativeOf(at, extra, this.identified);\n    this.native = n.native;\n    this.nativeConversation = n.conversation;\n  }\n\n  // wire is the call as Wakeshift takes it: the request's client, _meta and\n  // User-Agent, the agent's harness and conversation_id arguments, the\n  // conversation's session, and who's calling.\n  wire(session = this.session): Call {\n    const req = this.at.host.req(this.extra);\n    const meta = req.meta;\n    return packCall({\n      meta,\n      clientName: this.at.clientOf(meta, this.identified),\n      user: this.user,\n      session: session ?? undefined,\n      userAgent: userAgentOf(req),\n      harness: this.args.harness,\n      conversation: this.args.conversation,\n      tool: this.tool,\n      tasks: takesTasks(meta, req),\n    });\n  }\n\n  // request runs one request of the call to Wakeshift, after any before it\n  // (one at a time per call), refusing once the handler has returned.\n  private request<T>(fn: () => Promise<T>): Promise<T> {\n    if (this.ended) return Promise.reject(new CallEnded());\n    const run = this.queue.then(() => (this.ended ? Promise.reject(new CallEnded()) : fn()));\n    this.queue = run.catch(() => undefined);\n    this.inflight.add(run);\n    run.then(\n      () => this.inflight.delete(run),\n      () => this.inflight.delete(run),\n    );\n    return run;\n  }\n\n  private checkEvent(event: WakeEvent<any, any>): void {\n    if (!event || typeof event.owner !== \"function\" || event.owner() !== this.at.wake) {\n      throw new WakeshiftError(404, \"unknown_event\", `${event?.name ?? \"that\"} is another app's event: this app doesn't offer it`);\n    }\n  }\n\n  subscribe<A extends AnySchema>(event: WakeEvent<A, any>, args: SchemaInput<A>, opts: ToolSubscribeOptions = {}): Promise<Subscription> {\n    return this.request(async () => {\n      this.checkEvent(event);\n      const known = event.parseArgs(args);\n      if (this.native) {\n        this.addNative(event.name, known);\n        return { id: \"\", user: this.user ?? \"\", created: false, session: \"\", recognized: { harness: null, via: null, claimed: null }, native: true };\n      }\n      if (event.personal && !this.user) {\n        throw new WakeshiftError(400, \"bad_arguments\", `${event.name} is personal: call.subscribe needs a user (attach's identify), who the subscription belongs to`);\n      }\n      const body: Record<string, unknown> = { call: this.wire(), name: event.name, arguments: known, defer_addition: true };\n      if (this.pending) body.pending = this.pending;\n      if (opts.ttl !== undefined) body.ttl = opts.ttl;\n      if (opts.reason) body.reason = opts.reason;\n      if (opts.label) body.label = opts.label;\n      const out = await this.at.wake.requestApp<{\n        event_subscription: { id: string; user?: string; created: boolean };\n        session: string;\n        agent?: Partial<Recognized>;\n        pending?: string;\n      }>(\"POST\", \"/event-subscriptions\", body);\n      if (out.pending) this.pending = out.pending;\n      this.armed = true;\n      this.session = out.session;\n      const rec = recognizedOf(out.agent);\n      this.recognized = rec;\n      const es = out.event_subscription;\n      if (es.created) this.created.push({ name: event.name, arguments: known, session: out.session, pending: out.pending ?? this.pending });\n      return { id: es.id, user: es.user ?? this.user ?? \"\", created: !!es.created, session: out.session, recognized: rec, native: false };\n    });\n  }\n\n  unsubscribe<A extends AnySchema>(event: WakeEvent<A, any>, args: SchemaInput<A>): Promise<boolean> {\n    return this.request(async () => {\n      this.checkEvent(event);\n      const known = event.parseArgs(args);\n      if (this.native) return this.endNative(event.name, known);\n      const out = await this.at.wake.requestApp<{ ended?: boolean }>(\"POST\", \"/event-subscriptions/unsubscribe\", { call: this.wire(), name: event.name, arguments: known });\n      return !!out.ended;\n    });\n  }\n\n  resolveSession(): Promise<string> {\n    return this.request(async () => {\n      if (this.pending && this.session) return this.session;\n      const out = await this.at.wake.requestApp<{ session: string; pending: string; agent?: Partial<Recognized> }>(\"POST\", \"/calls/session\", { call: this.wire() });\n      this.pending = out.pending;\n      this.session = out.session;\n      this.recognized = recognizedOf(out.agent);\n      return out.session;\n    });\n  }\n\n  // endNative ends the user's webhook subscriptions to event with these arguments.\n  private async endNative(name: string, args: unknown): Promise<boolean> {\n    if (!this.user) return false;\n    const out = await this.at.wake.requestApp<{ subscriptions?: { id: string; event: string; arguments: unknown; destination: string }[] }>(\"POST\", \"/event-subscriptions/list\", {\n      user: this.user,\n    });\n    let ended = false;\n    for (const s of out.subscriptions ?? []) {\n      if (s.event !== name || s.destination !== \"webhook\" || !sameJSON(s.arguments, args)) continue;\n      const r = await this.at.wake.requestApp<{ ended?: boolean }>(\"POST\", \"/event-subscriptions/end\", { user: this.user, id: s.id });\n      ended = ended || !!r.ended;\n    }\n    return ended;\n  }\n\n  // addNative adds a native line, once per event and arguments.\n  private addNative(name: string, args: unknown): void {\n    if (this.nativeLines.some((l) => l.name === name && sameJSON(l.arguments, args))) return;\n    if (this.nativeLines.length === MAX_NATIVE_LINES) {\n      this.at.warnOnce(\n        \"native:\" + this.tool,\n        `wakeshift: the wrapped tool ${this.tool} subscribed a native client to more than ${MAX_NATIVE_LINES} events in one call; only ${MAX_NATIVE_LINES} are named`,\n      );\n      return;\n    }\n    this.nativeLines.push({ name, arguments: args });\n  }\n\n  // end closes the call to further requests, after waiting for any in\n  // flight (warning once per tool: the handler should have waited for them).\n  async end(): Promise<void> {\n    if (this.inflight.size > 0) {\n      this.at.warnOnce(\n        \"inflight:\" + this.tool,\n        `wakeshift: the wrapped tool ${this.tool} returned while a subscribe was in flight; it was waited for: await call.subscribe, call.unsubscribe and call.resolveSession before returning`,\n      );\n      await Promise.allSettled([...this.inflight]);\n    }\n    this.ended = true;\n  }\n\n  // rollback ends every subscription the call created and no other call\n  // has renewed since (the core checks the pending record that made it),\n  // with reason rolled_back (audited as such, no \"ended\" notice), whatever\n  // became of the request.\n  async rollback(): Promise<void> {\n    for (const c of this.created) {\n      try {\n        await this.at.wake.requestApp(\"POST\", \"/event-subscriptions/unsubscribe\", { call: this.wire(c.session), name: c.name, arguments: c.arguments, reason: \"rolled_back\", pending: c.pending });\n      } catch (err) {\n        this.at.wake.warn(`wakeshift: rolling back a failed tool call's subscription (${this.tool}, ${c.name}): ${detailOf(err)}`);\n      }\n    }\n    this.created = [];\n  }\n\n  // addition asks Wakeshift for the call's block: what the conversation\n  // needs to be woken (from the call's pending-addition record, if a\n  // subscribe or resolveSession made one; otherwise what any other tool\n  // result of the app's carries), and a line per event a native caller\n  // should subscribe to. A call that names no conversation, and subscribed\n  // nothing, gets nothing (null). Only a failure for a call that subscribed\n  // is thrown: without a subscription, the block is a courtesy.\n  async addition(): Promise<CallAddition | null> {\n    const wire = this.wire();\n    if (!this.pending && this.nativeLines.length === 0 && !wire.session && !wire.conversation && !namesConversation(wire.meta)) return null;\n    const body: Record<string, unknown> = { call: wire };\n    if (this.pending) body.pending = this.pending;\n    if (this.nativeLines.length > 0) body.native = this.nativeLines;\n    try {\n      return await this.at.wake.requestApp<CallAddition>(\"POST\", \"/calls/addition\", body);\n    } catch (err) {\n      if (this.armed || this.nativeLines.length > 0) throw err;\n      this.at.wake.warn(`wakeshift: Wakeshift's addition to a result of ${this.tool}: ${detailOf(err)}`);\n      return null;\n    }\n  }\n}\n\nfunction recognizedOf(a: Partial<Recognized> | undefined): Recognized {\n  return { harness: a?.harness ?? null, via: a?.via ?? null, claimed: a?.claimed ?? null };\n}\n\nfunction hasKey(o: unknown, key: string): boolean {\n  return !!o && typeof o === \"object\" && !Array.isArray(o) && (o as Record<string, unknown>)[key] !== undefined;\n}\n\nfunction isRecord(v: unknown): v is Record<string, unknown> {\n  return !!v && typeof v === \"object\" && !Array.isArray(v);\n}\n\n/**\n * @internal Whether a call comes from a client that subscribes to the app's\n * events itself, on stateless servers too (which never see initialize), in\n * this order: the app's override (events.native); ChatGPT's\n * _meta[\"openai/session\"]; a client name (initialize's, or _meta's, else\n * identify's) or User-Agent of a known native client; a client that\n * declares the MCP Events extension (initialize's capabilities, or\n * _meta's). conversation is ChatGPT's conversation, when it names one.\n */\nexport function nativeOf(at: Pick<AttachContext, \"host\" | \"clientOf\" | \"events\">, extra: unknown, identified?: string): { native: boolean; conversation: string | null } {\n  const req = at.host.req(extra);\n  const meta = (req.meta ?? {}) as Record<string, unknown>;\n  const conversation = typeof meta[\"openai/session\"] === \"string\" && meta[\"openai/session\"] ? (meta[\"openai/session\"] as string) : null;\n  if (at.events.native?.(extra)) return { native: true, conversation };\n  if (conversation) return { native: true, conversation };\n  if (dialectOf(at.clientOf(meta, identified), userAgentOf(req)) === \"chatgpt\") return { native: true, conversation: null };\n  const caps = at.host.clientCapabilities();\n  if (hasKey(caps?.extensions, EXTENSION_KEY)) return { native: true, conversation: null };\n  const metaCaps = meta[\"io.modelcontextprotocol/clientCapabilities\"] as Record<string, unknown> | undefined;\n  if (hasKey(metaCaps?.extensions, EXTENSION_KEY)) return { native: true, conversation: null };\n  return { native: false, conversation: null };\n}\n\n/**\n * Wakeshift attached to an MCP server: the handle `wake.attach` returns.\n * Wrap a tool's callback with events, and give its input schema\n * Wakeshift's three arguments with inputs. C is your MCP SDK's request\n * context (2.x's ctx, 1.x's extra), which a wrapped call carries as\n * `call.extra`.\n */\nexport class Attached<C = any> {\n  /** @internal */\n  private readonly wrapped = new Map<string, unknown>();\n\n  /** @internal Use wake.attach. */\n  constructor(private readonly at: AttachContext) {}\n\n  /**\n   * A tool's input schema plus Wakeshift's three optional arguments\n   * (session_key, harness and conversation_id), for a wrapped tool's\n   * inputSchema: a zod object, `inputSchema: attached.inputs(z.object({ table: z.string() }))`\n   * (as @modelcontextprotocol/server 2.x takes it), or a shape,\n   * `inputSchema: attached.inputs({ table: z.string() })`.\n   */\n  inputs<O extends z4.ZodObject>(schema: O): z4.ZodObject<O[\"shape\"] & WakeInputsV4, O[\"_zod\"][\"config\"]>;\n  inputs<O extends z3.AnyZodObject>(schema: O): z3.ZodObject<O[\"shape\"] & WakeInputsV3, O[\"_def\"][\"unknownKeys\"], O[\"_def\"][\"catchall\"]>;\n  inputs<S extends ZodRawShapeCompat>(shape: S): S & WakeInputs;\n  inputs(shape: ZodRawShapeCompat | AnySchema): unknown {\n    if (isSchema(shape)) {\n      const own = getObjectShape(shape);\n      if (!own) throw new Error(\"wakeshift: attached.inputs takes a zod object or a shape\");\n      for (const k of WAKE_KEYS) if (k in own) throw new Error(`wakeshift: attached.inputs adds ${k} itself: leave it out of the schema`);\n      return withFields(shape, byVersion[isZ4Schema(shape) ? \"v4\" : \"v3\"].inputs);\n    }\n    for (const k of WAKE_KEYS) if (k in (shape ?? {})) throw new Error(`wakeshift: attached.inputs adds ${k} itself: leave it out of the shape`);\n    return { ...shape, ...fieldsFor(shape, \"inputs\", this.at.host) };\n  }\n\n  /**\n   * A tool's output schema plus an optional wakeshift string, for a wrapped\n   * tool's outputSchema (a zod object or a shape, as inputs takes).\n   * Optional: every structured result of a wrapped tool gets the block's\n   * text (without the beacon line) in wakeshift, for hosts that show the\n   * model only structuredContent, and tools/list names it in the tool's\n   * output schema; this declares it in your own schema and types. The\n   * block stays in the content too.\n   */\n  output<O extends z4.ZodObject>(schema: O): z4.ZodObject<O[\"shape\"] & WakeOutputV4, O[\"_zod\"][\"config\"]>;\n  output<O extends z3.AnyZodObject>(schema: O): z3.ZodObject<O[\"shape\"] & WakeOutputV3, O[\"_def\"][\"unknownKeys\"], O[\"_def\"][\"catchall\"]>;\n  output<S extends ZodRawShapeCompat>(shape: S): S & WakeOutputShape;\n  output(shape: ZodRawShapeCompat | AnySchema): unknown {\n    if (isSchema(shape)) {\n      const own = getObjectShape(shape);\n      if (!own) throw new Error(\"wakeshift: attached.output takes a zod object or a shape\");\n      if (WAKESHIFT_FIELD in own) throw new Error(\"wakeshift: attached.output adds wakeshift itself: leave it out of the schema\");\n      return withFields(shape, byVersion[isZ4Schema(shape) ? \"v4\" : \"v3\"].output);\n    }\n    if (\"wakeshift\" in (shape ?? {})) throw new Error(\"wakeshift: attached.output adds wakeshift itself: leave it out of the shape\");\n    return { ...shape, ...fieldsFor(shape, \"output\", this.at.host) };\n  }\n\n  /**\n   * Wraps a tool's callback, for `server.registerTool(name, …)` with the\n   * same name (MCP doesn't tell a callback its tool's name): the handler\n   * gets the tool's own arguments and a ToolCall, and returns a string or\n   * a CallToolResult; the reply gets Wakeshift's block. The tool's\n   * inputSchema must come from attached.inputs (a type error otherwise).\n   * Register it on the server passed to attach.\n   */\n  events<A extends WakeArgs>(name: string, handler: (args: Omit<A, WakeKey>, call: ToolCall<C>) => WrappedResult | Promise<WrappedResult>): WrappedCallback<A, C> {\n    if (!name || typeof name !== \"string\") throw new Error(\"wakeshift: attached.events needs the tool's name, as registered\");\n    if (typeof handler !== \"function\") throw new Error(\"wakeshift: attached.events needs a handler\");\n    const cb = (async (args: Record<string, unknown>, extra: C) => this.run(name, cb, handler as any, args ?? {}, extra)) as unknown as WrappedCallback<A, C>;\n    this.wrapped.set(name, cb);\n    return cb;\n  }\n\n  private async run(\n    name: string,\n    cb: unknown,\n    handler: (args: Record<string, unknown>, call: ToolCall<C>) => WrappedResult | Promise<WrappedResult>,\n    input: Record<string, unknown>,\n    extra: C,\n  ): Promise<CallToolResult> {\n    const host = this.at.host;\n    const tool = host.registeredTool(name);\n    if (tool === undefined || (tool && (tool.handler ?? tool.callback) !== cb)) {\n      throw new Error(`wakeshift: attached.events(\"${name}\", …) isn't the tool named ${name} on the server passed to attach: register it with the same name, on that server`);\n    }\n    let outputSchema: unknown;\n    if (tool) {\n      const shape = getObjectShape(normalizeObjectSchema(tool.inputSchema));\n      if (!shape || WAKE_KEYS.some((k) => !(k in shape))) {\n        throw new Error(`wakeshift: the tool ${name} doesn't take Wakeshift's arguments: make its inputSchema with attached.inputs(…)`);\n      }\n      outputSchema = tool.outputSchema;\n    } else {\n      this.at.warnOnce(\"internals\", \"wakeshift: this MCP SDK keeps its tools where Wakeshift can't read them, so wrapped tools' registration and output aren't checked\");\n    }\n    const str = (k: string) => (typeof input[k] === \"string\" && (input[k] as string) ? (input[k] as string) : undefined);\n    const call = new CallScope<C>(this.at, name, extra, { session: str(SESSION_ARG), harness: str(HARNESS_ARG), conversation: str(CONVERSATION_ARG) });\n    const appArgs: Record<string, unknown> = {};\n    for (const [k, v] of Object.entries(input)) if (!(WAKE_KEYS as readonly string[]).includes(k)) appArgs[k] = v;\n\n    let result: WrappedResult;\n    try {\n      result = await handler(appArgs, call);\n    } catch (err) {\n      await call.end();\n      await call.rollback();\n      throw err;\n    }\n    await call.end();\n    // Rollback is decided from the final result, the one the client would\n    // receive: MCP validates the output after this callback returns, so\n    // the wrapper checks it the same way first.\n    if (result === null || result === undefined || (typeof result !== \"string\" && typeof result !== \"object\")) {\n      await call.rollback();\n      return result as unknown as CallToolResult;\n    }\n    if ((typeof result === \"object\" && result.isError) || host.req(extra).signal?.aborted || !(await host.validOutput(outputSchema, result))) {\n      await call.rollback();\n      return typeof result === \"string\" ? { content: [{ type: \"text\", text: result }] } : result;\n    }\n    let add: CallAddition | null;\n    try {\n      add = await call.addition();\n    } catch (err) {\n      // The conversation is subscribed but wouldn't learn how it's woken:\n      // the call fails, as one whose handler failed does.\n      await call.rollback();\n      this.at.wake.warn(`wakeshift: Wakeshift's block for a result of ${name} failed, so the call's subscriptions were rolled back: ${detailOf(err)}`);\n      return { isError: true, content: [{ type: \"text\", text: `Wakeshift couldn't finish setting up this conversation's wakes (${detailOf(err)}). Try again.` }] };\n    }\n    const out = mergeBlock(result, add ?? {});\n    if (isRecord(out.structuredContent) && WAKESHIFT_FIELD in out.structuredContent && !(await host.validOutput(outputSchema, out))) {\n      // A schema that refuses fields it doesn't name (a strict object):\n      // MCP would refuse the result, so it goes without the field.\n      const { [WAKESHIFT_FIELD]: _dropped, ...own } = out.structuredContent;\n      out.structuredContent = own;\n      this.at.warnOnce(\n        \"output:\" + name,\n        `wakeshift: the wrapped tool ${name}'s outputSchema refuses a wakeshift field, so hosts that show only structuredContent won't see how this conversation is woken: make it with attached.output`,\n      );\n    }\n    return out;\n  }\n\n  /**\n   * @internal Wraps the server's tools/list handler so each wrapped tool's\n   * object output schema names wakeshift (withWakeshiftField). McpServer\n   * sets the handler when the first tool is registered, perhaps after\n   * attach, so a later set is wrapped too (as tasks.ts does).\n   */\n  amendToolsList(): void {\n    type Handler = (request: any, extra: any) => Promise<any>;\n    const handlers = this.at.host.handlers();\n    if (!handlers) return; // a future SDK: schemas stay as registered\n    const wrap =\n      (h: Handler): Handler =>\n      async (request, extra) => {\n        const out = await h(request, extra);\n        if (out && Array.isArray(out.tools)) this.amendTools(out.tools);\n        return out;\n      };\n    const h = handlers.get(\"tools/list\");\n    if (h) Map.prototype.set.call(handlers, \"tools/list\", wrap(h));\n    const set = handlers.set.bind(handlers);\n    handlers.set = (method: string, h: Handler) => set(method, method === \"tools/list\" ? wrap(h) : h);\n  }\n\n  private amendTools(tools: unknown[]): void {\n    for (let i = 0; i < tools.length; i++) {\n      const t = tools[i] as { name?: unknown; outputSchema?: unknown } | null;\n      if (!t || typeof t !== \"object\" || typeof t.name !== \"string\" || !this.wrapped.has(t.name) || !t.outputSchema) continue;\n      const schema = withWakeshiftField(t.outputSchema);\n      if (schema !== t.outputSchema) tools[i] = { ...t, outputSchema: schema };\n    }\n  }\n}\n"]}

package/dist/wakeshift.d.ts

@@ -1,5 +1,5 @@
-import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import type { AnySchema } from "@modelcontextprotocol/sdk/server/zod-compat.js";
+import type { AnySchema } from "./zod.js";
+import type { HostContext, McpServerLike } from "./mcp.js";
 import { WakeEvent, type AttachOptions, type EventOptions, type EventTypeManifest } from "./events.js";
 import type { Attached } from "./toolcall.js";
 /** Wakeshift's production core, the default for every key but an older wst_test_ one. Override with `core` or WAKESHIFT_CORE. */
@@ -231,8 +231,12 @@ export declare class Wakeshift {
      * It returns the handle your own tools use to subscribe the conversation
      * calling them: wrap a tool's callback with attached.events, and give it
      * Wakeshift's arguments with attached.inputs.
+     *
+     * server is an McpServer from @modelcontextprotocol/server 2.x or
+     * @modelcontextprotocol/sdk 1.x; identify gets that SDK's own request
+     * context (2.x's ctx, 1.x's extra).
      */
-    attach(server: McpServer, opts?: AttachOptions): Attached;
+    attach<S extends McpServerLike>(server: S, opts?: AttachOptions<HostContext<S>>): Attached<HostContext<S>>;
     /**
      * The workspace Wakeshift keeps for one of the app's users (user: the
      * app's own ID for them, as attach's identify gives it), created on first