agents 0.0.0-956c772 → 0.0.0-9688c15

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/src/index.ts CHANGED
@@ -12,10 +12,23 @@ import { parseCronExpression } from "cron-schedule";
12
12
  import { nanoid } from "nanoid";
13
13
 
14
14
  import { AsyncLocalStorage } from "node:async_hooks";
15
+ import { MCPClientManager } from "./mcp/client";
16
+ import {
17
+ DurableObjectOAuthClientProvider,
18
+ type AgentsOAuthProvider,
19
+ } from "./mcp/do-oauth-client-provider";
20
+ import type {
21
+ Tool,
22
+ Resource,
23
+ Prompt,
24
+ } from "@modelcontextprotocol/sdk/types.js";
15
25
 
16
- export type { Connection, WSMessage, ConnectionContext } from "partyserver";
26
+ import type { Client } from "@modelcontextprotocol/sdk/client/index.js";
27
+ import type { SSEClientTransportOptions } from "@modelcontextprotocol/sdk/client/sse.js";
17
28
 
18
- import { WorkflowEntrypoint as CFWorkflowEntrypoint } from "cloudflare:workers";
29
+ import { camelCaseToKebabCase } from "./client";
30
+
31
+ export type { Connection, WSMessage, ConnectionContext } from "partyserver";
19
32
 
20
33
  /**
21
34
  * RPC request message from client
@@ -119,11 +132,6 @@ export function unstable_callable(metadata: CallableMetadata = {}) {
119
132
  };
120
133
  }
121
134
 
122
- /**
123
- * A class for creating workflow entry points that can be used with Cloudflare Workers
124
- */
125
- export class WorkflowEntrypoint extends CFWorkflowEntrypoint {}
126
-
127
135
  /**
128
136
  * Represents a scheduled task within an Agent
129
137
  * @template T Type of the payload data
@@ -165,17 +173,78 @@ function getNextCronTime(cron: string) {
165
173
  return interval.getNextDate();
166
174
  }
167
175
 
176
+ /**
177
+ * MCP Server state update message from server -> Client
178
+ */
179
+ export type MCPServerMessage = {
180
+ type: "cf_agent_mcp_servers";
181
+ mcp: MCPServersState;
182
+ };
183
+
184
+ export type MCPServersState = {
185
+ servers: {
186
+ [id: string]: MCPServer;
187
+ };
188
+ tools: Tool[];
189
+ prompts: Prompt[];
190
+ resources: Resource[];
191
+ };
192
+
193
+ export type MCPServer = {
194
+ name: string;
195
+ server_url: string;
196
+ auth_url: string | null;
197
+ state: "authenticating" | "connecting" | "ready" | "discovering" | "failed";
198
+ };
199
+
200
+ /**
201
+ * MCP Server data stored in DO SQL for resuming MCP Server connections
202
+ */
203
+ type MCPServerRow = {
204
+ id: string;
205
+ name: string;
206
+ server_url: string;
207
+ client_id: string | null;
208
+ auth_url: string | null;
209
+ callback_url: string;
210
+ server_options: string;
211
+ };
212
+
168
213
  const STATE_ROW_ID = "cf_state_row_id";
169
214
  const STATE_WAS_CHANGED = "cf_state_was_changed";
170
215
 
171
216
  const DEFAULT_STATE = {} as unknown;
172
217
 
173
- export const unstable_context = new AsyncLocalStorage<{
218
+ const agentContext = new AsyncLocalStorage<{
174
219
  agent: Agent<unknown>;
175
220
  connection: Connection | undefined;
176
221
  request: Request | undefined;
177
222
  }>();
178
223
 
224
+ export function getCurrentAgent<
225
+ T extends Agent<unknown, unknown> = Agent<unknown, unknown>,
226
+ >(): {
227
+ agent: T | undefined;
228
+ connection: Connection | undefined;
229
+ request: Request<unknown, CfProperties<unknown>> | undefined;
230
+ } {
231
+ const store = agentContext.getStore() as
232
+ | {
233
+ agent: T;
234
+ connection: Connection | undefined;
235
+ request: Request<unknown, CfProperties<unknown>> | undefined;
236
+ }
237
+ | undefined;
238
+ if (!store) {
239
+ return {
240
+ agent: undefined,
241
+ connection: undefined,
242
+ request: undefined,
243
+ };
244
+ }
245
+ return store;
246
+ }
247
+
179
248
  /**
180
249
  * Base class for creating Agent implementations
181
250
  * @template Env Environment type containing bindings
@@ -184,6 +253,11 @@ export const unstable_context = new AsyncLocalStorage<{
184
253
  export class Agent<Env, State = unknown> extends Server<Env> {
185
254
  #state = DEFAULT_STATE as State;
186
255
 
256
+ #ParentClass: typeof Agent<Env, State> =
257
+ Object.getPrototypeOf(this).constructor;
258
+
259
+ mcp: MCPClientManager = new MCPClientManager(this.#ParentClass.name, "0.0.1");
260
+
187
261
  /**
188
262
  * Initial state for the Agent
189
263
  * Override to provide default state values
@@ -298,9 +372,49 @@ export class Agent<Env, State = unknown> extends Server<Env> {
298
372
  });
299
373
  });
300
374
 
375
+ this.sql`
376
+ CREATE TABLE IF NOT EXISTS cf_agents_mcp_servers (
377
+ id TEXT PRIMARY KEY NOT NULL,
378
+ name TEXT NOT NULL,
379
+ server_url TEXT NOT NULL,
380
+ callback_url TEXT NOT NULL,
381
+ client_id TEXT,
382
+ auth_url TEXT,
383
+ server_options TEXT
384
+ )
385
+ `;
386
+
387
+ const _onRequest = this.onRequest.bind(this);
388
+ this.onRequest = (request: Request) => {
389
+ return agentContext.run(
390
+ { agent: this, connection: undefined, request },
391
+ async () => {
392
+ if (this.mcp.isCallbackRequest(request)) {
393
+ await this.mcp.handleCallbackRequest(request);
394
+
395
+ // after the MCP connection handshake, we can send updated mcp state
396
+ this.broadcast(
397
+ JSON.stringify({
398
+ type: "cf_agent_mcp_servers",
399
+ mcp: this.#getMcpServerStateInternal(),
400
+ })
401
+ );
402
+
403
+ // We probably should let the user configure this response/redirect, but this is fine for now.
404
+ return new Response("<script>window.close();</script>", {
405
+ status: 200,
406
+ headers: { "content-type": "text/html" },
407
+ });
408
+ }
409
+
410
+ return this.#tryCatch(() => _onRequest(request));
411
+ }
412
+ );
413
+ };
414
+
301
415
  const _onMessage = this.onMessage.bind(this);
302
416
  this.onMessage = async (connection: Connection, message: WSMessage) => {
303
- return unstable_context.run(
417
+ return agentContext.run(
304
418
  { agent: this, connection, request: undefined },
305
419
  async () => {
306
420
  if (typeof message !== "string") {
@@ -378,7 +492,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
378
492
  this.onConnect = (connection: Connection, ctx: ConnectionContext) => {
379
493
  // TODO: This is a hack to ensure the state is sent after the connection is established
380
494
  // must fix this
381
- return unstable_context.run(
495
+ return agentContext.run(
382
496
  { agent: this, connection, request: ctx.request },
383
497
  async () => {
384
498
  setTimeout(() => {
@@ -390,11 +504,58 @@ export class Agent<Env, State = unknown> extends Server<Env> {
390
504
  })
391
505
  );
392
506
  }
507
+
508
+ connection.send(
509
+ JSON.stringify({
510
+ type: "cf_agent_mcp_servers",
511
+ mcp: this.#getMcpServerStateInternal(),
512
+ })
513
+ );
514
+
393
515
  return this.#tryCatch(() => _onConnect(connection, ctx));
394
516
  }, 20);
395
517
  }
396
518
  );
397
519
  };
520
+
521
+ const _onStart = this.onStart.bind(this);
522
+ this.onStart = async () => {
523
+ return agentContext.run(
524
+ { agent: this, connection: undefined, request: undefined },
525
+ async () => {
526
+ const servers = this.sql<MCPServerRow>`
527
+ SELECT id, name, server_url, client_id, auth_url, callback_url, server_options FROM cf_agents_mcp_servers;
528
+ `;
529
+
530
+ // from DO storage, reconnect to all servers using our saved auth information
531
+ await Promise.allSettled(
532
+ servers.map((server) => {
533
+ return this.#connectToMcpServerInternal(
534
+ server.name,
535
+ server.server_url,
536
+ server.callback_url,
537
+ server.server_options
538
+ ? JSON.parse(server.server_options)
539
+ : undefined,
540
+ {
541
+ id: server.id,
542
+ oauthClientId: server.client_id ?? undefined,
543
+ }
544
+ );
545
+ })
546
+ );
547
+
548
+ this.broadcast(
549
+ JSON.stringify({
550
+ type: "cf_agent_mcp_servers",
551
+ mcp: this.#getMcpServerStateInternal(),
552
+ })
553
+ );
554
+
555
+ await this.#tryCatch(() => _onStart());
556
+ }
557
+ );
558
+ };
398
559
  }
399
560
 
400
561
  #setStateInternal(state: State, source: Connection | "server" = "server") {
@@ -415,8 +576,8 @@ export class Agent<Env, State = unknown> extends Server<Env> {
415
576
  source !== "server" ? [source.id] : []
416
577
  );
417
578
  return this.#tryCatch(() => {
418
- const { connection, request } = unstable_context.getStore() || {};
419
- return unstable_context.run(
579
+ const { connection, request } = agentContext.getStore() || {};
580
+ return agentContext.run(
420
581
  { agent: this, connection, request },
421
582
  async () => {
422
583
  return this.onStateUpdate(state, source);
@@ -447,7 +608,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
447
608
  * @param email Email message to process
448
609
  */
449
610
  onEmail(email: ForwardableEmailMessage) {
450
- return unstable_context.run(
611
+ return agentContext.run(
451
612
  { agent: this, connection: undefined, request: undefined },
452
613
  async () => {
453
614
  console.error("onEmail not implemented");
@@ -680,10 +841,14 @@ export class Agent<Env, State = unknown> extends Server<Env> {
680
841
  }
681
842
 
682
843
  /**
683
- * Method called when an alarm fires
684
- * Executes any scheduled tasks that are due
844
+ * Method called when an alarm fires.
845
+ * Executes any scheduled tasks that are due.
846
+ *
847
+ * @remarks
848
+ * To schedule a task, please use the `this.schedule` method instead.
849
+ * See {@link https://developers.cloudflare.com/agents/api-reference/schedule-tasks/}
685
850
  */
686
- async alarm() {
851
+ public readonly alarm = async () => {
687
852
  const now = Math.floor(Date.now() / 1000);
688
853
 
689
854
  // Get all schedules that should be executed now
@@ -697,7 +862,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
697
862
  console.error(`callback ${row.callback} not found`);
698
863
  continue;
699
864
  }
700
- await unstable_context.run(
865
+ await agentContext.run(
701
866
  { agent: this, connection: undefined, request: undefined },
702
867
  async () => {
703
868
  try {
@@ -730,7 +895,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
730
895
 
731
896
  // Schedule the next alarm
732
897
  await this.#scheduleNextAlarm();
733
- }
898
+ };
734
899
 
735
900
  /**
736
901
  * Destroy the Agent, removing all state and scheduled tasks
@@ -739,6 +904,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
739
904
  // drop all tables
740
905
  this.sql`DROP TABLE IF EXISTS cf_agents_state`;
741
906
  this.sql`DROP TABLE IF EXISTS cf_agents_schedules`;
907
+ this.sql`DROP TABLE IF EXISTS cf_agents_mcp_servers`;
742
908
 
743
909
  // delete all alarms
744
910
  await this.ctx.storage.deleteAlarm();
@@ -753,6 +919,165 @@ export class Agent<Env, State = unknown> extends Server<Env> {
753
919
  // biome-ignore lint/complexity/noBannedTypes: <explanation>
754
920
  return callableMetadata.has(this[method as keyof this] as Function);
755
921
  }
922
+
923
+ /**
924
+ * Connect to a new MCP Server
925
+ *
926
+ * @param url MCP Server SSE URL
927
+ * @param callbackHost Base host for the agent, used for the redirect URI.
928
+ * @param agentsPrefix agents routing prefix if not using `agents`
929
+ * @param options MCP client and transport (header) options
930
+ * @returns authUrl
931
+ */
932
+ async addMcpServer(
933
+ serverName: string,
934
+ url: string,
935
+ callbackHost: string,
936
+ agentsPrefix = "agents",
937
+ options?: {
938
+ client?: ConstructorParameters<typeof Client>[1];
939
+ transport?: {
940
+ headers: HeadersInit;
941
+ };
942
+ }
943
+ ): Promise<{ id: string; authUrl: string | undefined }> {
944
+ const callbackUrl = `${callbackHost}/${agentsPrefix}/${camelCaseToKebabCase(this.#ParentClass.name)}/${this.name}/callback`;
945
+
946
+ const result = await this.#connectToMcpServerInternal(
947
+ serverName,
948
+ url,
949
+ callbackUrl,
950
+ options
951
+ );
952
+
953
+ this.broadcast(
954
+ JSON.stringify({
955
+ type: "cf_agent_mcp_servers",
956
+ mcp: this.#getMcpServerStateInternal(),
957
+ })
958
+ );
959
+
960
+ return result;
961
+ }
962
+
963
+ async #connectToMcpServerInternal(
964
+ serverName: string,
965
+ url: string,
966
+ callbackUrl: string,
967
+ // it's important that any options here are serializable because we put them into our sqlite DB for reconnection purposes
968
+ options?: {
969
+ client?: ConstructorParameters<typeof Client>[1];
970
+ /**
971
+ * We don't expose the normal set of transport options because:
972
+ * 1) we can't serialize things like the auth provider or a fetch function into the DB for reconnection purposes
973
+ * 2) We probably want these options to be agnostic to the transport type (SSE vs Streamable)
974
+ *
975
+ * This has the limitation that you can't override fetch, but I think headers should handle nearly all cases needed (i.e. non-standard bearer auth).
976
+ */
977
+ transport?: {
978
+ headers?: HeadersInit;
979
+ };
980
+ },
981
+ reconnect?: {
982
+ id: string;
983
+ oauthClientId?: string;
984
+ }
985
+ ): Promise<{ id: string; authUrl: string | undefined }> {
986
+ const authProvider = new DurableObjectOAuthClientProvider(
987
+ this.ctx.storage,
988
+ this.name,
989
+ callbackUrl
990
+ );
991
+
992
+ if (reconnect) {
993
+ authProvider.serverId = reconnect.id;
994
+ if (reconnect.oauthClientId) {
995
+ authProvider.clientId = reconnect.oauthClientId;
996
+ }
997
+ }
998
+
999
+ // allows passing through transport headers if necessary
1000
+ // this handles some non-standard bearer auth setups (i.e. MCP server behind CF access instead of OAuth)
1001
+ let headerTransportOpts: SSEClientTransportOptions = {};
1002
+ if (options?.transport?.headers) {
1003
+ headerTransportOpts = {
1004
+ eventSourceInit: {
1005
+ fetch: (url, init) =>
1006
+ fetch(url, {
1007
+ ...init,
1008
+ headers: options?.transport?.headers,
1009
+ }),
1010
+ },
1011
+ requestInit: {
1012
+ headers: options?.transport?.headers,
1013
+ },
1014
+ };
1015
+ }
1016
+
1017
+ const { id, authUrl, clientId } = await this.mcp.connect(url, {
1018
+ reconnect,
1019
+ transport: {
1020
+ ...headerTransportOpts,
1021
+ authProvider,
1022
+ },
1023
+ client: options?.client,
1024
+ });
1025
+
1026
+ this.sql`
1027
+ INSERT OR REPLACE INTO cf_agents_mcp_servers (id, name, server_url, client_id, auth_url, callback_url, server_options)
1028
+ VALUES (
1029
+ ${id},
1030
+ ${serverName},
1031
+ ${url},
1032
+ ${clientId ?? null},
1033
+ ${authUrl ?? null},
1034
+ ${callbackUrl},
1035
+ ${options ? JSON.stringify(options) : null}
1036
+ );
1037
+ `;
1038
+
1039
+ return {
1040
+ id,
1041
+ authUrl,
1042
+ };
1043
+ }
1044
+
1045
+ async removeMcpServer(id: string) {
1046
+ this.mcp.closeConnection(id);
1047
+ this.sql`
1048
+ DELETE FROM cf_agents_mcp_servers WHERE id = ${id};
1049
+ `;
1050
+ this.broadcast(
1051
+ JSON.stringify({
1052
+ type: "cf_agent_mcp_servers",
1053
+ mcp: this.#getMcpServerStateInternal(),
1054
+ })
1055
+ );
1056
+ }
1057
+
1058
+ #getMcpServerStateInternal(): MCPServersState {
1059
+ const mcpState: MCPServersState = {
1060
+ servers: {},
1061
+ tools: this.mcp.listTools(),
1062
+ prompts: this.mcp.listPrompts(),
1063
+ resources: this.mcp.listResources(),
1064
+ };
1065
+
1066
+ const servers = this.sql<MCPServerRow>`
1067
+ SELECT id, name, server_url, client_id, auth_url, callback_url, server_options FROM cf_agents_mcp_servers;
1068
+ `;
1069
+
1070
+ for (const server of servers) {
1071
+ mcpState.servers[server.id] = {
1072
+ name: server.name,
1073
+ server_url: server.server_url,
1074
+ auth_url: server.auth_url,
1075
+ state: this.mcp.mcpConnections[server.id].connectionState,
1076
+ };
1077
+ }
1078
+
1079
+ return mcpState;
1080
+ }
756
1081
  }
757
1082
 
758
1083
  /**
@@ -856,7 +1181,7 @@ export async function routeAgentEmail<Env>(
856
1181
  * @param options Options for Agent creation
857
1182
  * @returns Promise resolving to an Agent instance stub
858
1183
  */
859
- export function getAgentByName<Env, T extends Agent<Env>>(
1184
+ export async function getAgentByName<Env, T extends Agent<Env>>(
860
1185
  namespace: AgentNamespace<T>,
861
1186
  name: string,
862
1187
  options?: {
@@ -1 +0,0 @@
1
- {"version":3,"sources":["../src/index.ts"],"sourcesContent":["import {\n Server,\n routePartykitRequest,\n type PartyServerOptions,\n getServerByName,\n type Connection,\n type ConnectionContext,\n type WSMessage,\n} from \"partyserver\";\n\nimport { parseCronExpression } from \"cron-schedule\";\nimport { nanoid } from \"nanoid\";\n\nimport { AsyncLocalStorage } from \"node:async_hooks\";\n\nexport type { Connection, WSMessage, ConnectionContext } from \"partyserver\";\n\nimport { WorkflowEntrypoint as CFWorkflowEntrypoint } from \"cloudflare:workers\";\n\n/**\n * RPC request message from client\n */\nexport type RPCRequest = {\n type: \"rpc\";\n id: string;\n method: string;\n args: unknown[];\n};\n\n/**\n * State update message from client\n */\nexport type StateUpdateMessage = {\n type: \"cf_agent_state\";\n state: unknown;\n};\n\n/**\n * RPC response message to client\n */\nexport type RPCResponse = {\n type: \"rpc\";\n id: string;\n} & (\n | {\n success: true;\n result: unknown;\n done?: false;\n }\n | {\n success: true;\n result: unknown;\n done: true;\n }\n | {\n success: false;\n error: string;\n }\n);\n\n/**\n * Type guard for RPC request messages\n */\nfunction isRPCRequest(msg: unknown): msg is RPCRequest {\n return (\n typeof msg === \"object\" &&\n msg !== null &&\n \"type\" in msg &&\n msg.type === \"rpc\" &&\n \"id\" in msg &&\n typeof msg.id === \"string\" &&\n \"method\" in msg &&\n typeof msg.method === \"string\" &&\n \"args\" in msg &&\n Array.isArray((msg as RPCRequest).args)\n );\n}\n\n/**\n * Type guard for state update messages\n */\nfunction isStateUpdateMessage(msg: unknown): msg is StateUpdateMessage {\n return (\n typeof msg === \"object\" &&\n msg !== null &&\n \"type\" in msg &&\n msg.type === \"cf_agent_state\" &&\n \"state\" in msg\n );\n}\n\n/**\n * Metadata for a callable method\n */\nexport type CallableMetadata = {\n /** Optional description of what the method does */\n description?: string;\n /** Whether the method supports streaming responses */\n streaming?: boolean;\n};\n\n// biome-ignore lint/complexity/noBannedTypes: <explanation>\nconst callableMetadata = new Map<Function, CallableMetadata>();\n\n/**\n * Decorator that marks a method as callable by clients\n * @param metadata Optional metadata about the callable method\n */\nexport function unstable_callable(metadata: CallableMetadata = {}) {\n return function callableDecorator<This, Args extends unknown[], Return>(\n target: (this: This, ...args: Args) => Return,\n context: ClassMethodDecoratorContext\n ) {\n if (!callableMetadata.has(target)) {\n callableMetadata.set(target, metadata);\n }\n\n return target;\n };\n}\n\n/**\n * A class for creating workflow entry points that can be used with Cloudflare Workers\n */\nexport class WorkflowEntrypoint extends CFWorkflowEntrypoint {}\n\n/**\n * Represents a scheduled task within an Agent\n * @template T Type of the payload data\n */\nexport type Schedule<T = string> = {\n /** Unique identifier for the schedule */\n id: string;\n /** Name of the method to be called */\n callback: string;\n /** Data to be passed to the callback */\n payload: T;\n} & (\n | {\n /** Type of schedule for one-time execution at a specific time */\n type: \"scheduled\";\n /** Timestamp when the task should execute */\n time: number;\n }\n | {\n /** Type of schedule for delayed execution */\n type: \"delayed\";\n /** Timestamp when the task should execute */\n time: number;\n /** Number of seconds to delay execution */\n delayInSeconds: number;\n }\n | {\n /** Type of schedule for recurring execution based on cron expression */\n type: \"cron\";\n /** Timestamp for the next execution */\n time: number;\n /** Cron expression defining the schedule */\n cron: string;\n }\n);\n\nfunction getNextCronTime(cron: string) {\n const interval = parseCronExpression(cron);\n return interval.getNextDate();\n}\n\nconst STATE_ROW_ID = \"cf_state_row_id\";\nconst STATE_WAS_CHANGED = \"cf_state_was_changed\";\n\nconst DEFAULT_STATE = {} as unknown;\n\nexport const unstable_context = new AsyncLocalStorage<{\n agent: Agent<unknown>;\n connection: Connection | undefined;\n request: Request | undefined;\n}>();\n\n/**\n * Base class for creating Agent implementations\n * @template Env Environment type containing bindings\n * @template State State type to store within the Agent\n */\nexport class Agent<Env, State = unknown> extends Server<Env> {\n #state = DEFAULT_STATE as State;\n\n /**\n * Initial state for the Agent\n * Override to provide default state values\n */\n initialState: State = DEFAULT_STATE as State;\n\n /**\n * Current state of the Agent\n */\n get state(): State {\n if (this.#state !== DEFAULT_STATE) {\n // state was previously set, and populated internal state\n return this.#state;\n }\n // looks like this is the first time the state is being accessed\n // check if the state was set in a previous life\n const wasChanged = this.sql<{ state: \"true\" | undefined }>`\n SELECT state FROM cf_agents_state WHERE id = ${STATE_WAS_CHANGED}\n `;\n\n // ok, let's pick up the actual state from the db\n const result = this.sql<{ state: State | undefined }>`\n SELECT state FROM cf_agents_state WHERE id = ${STATE_ROW_ID}\n `;\n\n if (\n wasChanged[0]?.state === \"true\" ||\n // we do this check for people who updated their code before we shipped wasChanged\n result[0]?.state\n ) {\n const state = result[0]?.state as string; // could be null?\n\n this.#state = JSON.parse(state);\n return this.#state;\n }\n\n // ok, this is the first time the state is being accessed\n // and the state was not set in a previous life\n // so we need to set the initial state (if provided)\n if (this.initialState === DEFAULT_STATE) {\n // no initial state provided, so we return undefined\n return undefined as State;\n }\n // initial state provided, so we set the state,\n // update db and return the initial state\n this.setState(this.initialState);\n return this.initialState;\n }\n\n /**\n * Agent configuration options\n */\n static options = {\n /** Whether the Agent should hibernate when inactive */\n hibernate: true, // default to hibernate\n };\n\n /**\n * Execute SQL queries against the Agent's database\n * @template T Type of the returned rows\n * @param strings SQL query template strings\n * @param values Values to be inserted into the query\n * @returns Array of query results\n */\n sql<T = Record<string, string | number | boolean | null>>(\n strings: TemplateStringsArray,\n ...values: (string | number | boolean | null)[]\n ) {\n let query = \"\";\n try {\n // Construct the SQL query with placeholders\n query = strings.reduce(\n (acc, str, i) => acc + str + (i < values.length ? \"?\" : \"\"),\n \"\"\n );\n\n // Execute the SQL query with the provided values\n return [...this.ctx.storage.sql.exec(query, ...values)] as T[];\n } catch (e) {\n console.error(`failed to execute sql query: ${query}`, e);\n throw this.onError(e);\n }\n }\n constructor(ctx: AgentContext, env: Env) {\n super(ctx, env);\n\n this.sql`\n CREATE TABLE IF NOT EXISTS cf_agents_state (\n id TEXT PRIMARY KEY NOT NULL,\n state TEXT\n )\n `;\n\n void this.ctx.blockConcurrencyWhile(async () => {\n return this.#tryCatch(async () => {\n // Create alarms table if it doesn't exist\n this.sql`\n CREATE TABLE IF NOT EXISTS cf_agents_schedules (\n id TEXT PRIMARY KEY NOT NULL DEFAULT (randomblob(9)),\n callback TEXT,\n payload TEXT,\n type TEXT NOT NULL CHECK(type IN ('scheduled', 'delayed', 'cron')),\n time INTEGER,\n delayInSeconds INTEGER,\n cron TEXT,\n created_at INTEGER DEFAULT (unixepoch())\n )\n `;\n\n // execute any pending alarms and schedule the next alarm\n await this.alarm();\n });\n });\n\n const _onMessage = this.onMessage.bind(this);\n this.onMessage = async (connection: Connection, message: WSMessage) => {\n return unstable_context.run(\n { agent: this, connection, request: undefined },\n async () => {\n if (typeof message !== \"string\") {\n return this.#tryCatch(() => _onMessage(connection, message));\n }\n\n let parsed: unknown;\n try {\n parsed = JSON.parse(message);\n } catch (e) {\n // silently fail and let the onMessage handler handle it\n return this.#tryCatch(() => _onMessage(connection, message));\n }\n\n if (isStateUpdateMessage(parsed)) {\n this.#setStateInternal(parsed.state as State, connection);\n return;\n }\n\n if (isRPCRequest(parsed)) {\n try {\n const { id, method, args } = parsed;\n\n // Check if method exists and is callable\n const methodFn = this[method as keyof this];\n if (typeof methodFn !== \"function\") {\n throw new Error(`Method ${method} does not exist`);\n }\n\n if (!this.#isCallable(method)) {\n throw new Error(`Method ${method} is not callable`);\n }\n\n // biome-ignore lint/complexity/noBannedTypes: <explanation>\n const metadata = callableMetadata.get(methodFn as Function);\n\n // For streaming methods, pass a StreamingResponse object\n if (metadata?.streaming) {\n const stream = new StreamingResponse(connection, id);\n await methodFn.apply(this, [stream, ...args]);\n return;\n }\n\n // For regular methods, execute and send response\n const result = await methodFn.apply(this, args);\n const response: RPCResponse = {\n type: \"rpc\",\n id,\n success: true,\n result,\n done: true,\n };\n connection.send(JSON.stringify(response));\n } catch (e) {\n // Send error response\n const response: RPCResponse = {\n type: \"rpc\",\n id: parsed.id,\n success: false,\n error:\n e instanceof Error ? e.message : \"Unknown error occurred\",\n };\n connection.send(JSON.stringify(response));\n console.error(\"RPC error:\", e);\n }\n return;\n }\n\n return this.#tryCatch(() => _onMessage(connection, message));\n }\n );\n };\n\n const _onConnect = this.onConnect.bind(this);\n this.onConnect = (connection: Connection, ctx: ConnectionContext) => {\n // TODO: This is a hack to ensure the state is sent after the connection is established\n // must fix this\n return unstable_context.run(\n { agent: this, connection, request: ctx.request },\n async () => {\n setTimeout(() => {\n if (this.state) {\n connection.send(\n JSON.stringify({\n type: \"cf_agent_state\",\n state: this.state,\n })\n );\n }\n return this.#tryCatch(() => _onConnect(connection, ctx));\n }, 20);\n }\n );\n };\n }\n\n #setStateInternal(state: State, source: Connection | \"server\" = \"server\") {\n this.#state = state;\n this.sql`\n INSERT OR REPLACE INTO cf_agents_state (id, state)\n VALUES (${STATE_ROW_ID}, ${JSON.stringify(state)})\n `;\n this.sql`\n INSERT OR REPLACE INTO cf_agents_state (id, state)\n VALUES (${STATE_WAS_CHANGED}, ${JSON.stringify(true)})\n `;\n this.broadcast(\n JSON.stringify({\n type: \"cf_agent_state\",\n state: state,\n }),\n source !== \"server\" ? [source.id] : []\n );\n return this.#tryCatch(() => {\n const { connection, request } = unstable_context.getStore() || {};\n return unstable_context.run(\n { agent: this, connection, request },\n async () => {\n return this.onStateUpdate(state, source);\n }\n );\n });\n }\n\n /**\n * Update the Agent's state\n * @param state New state to set\n */\n setState(state: State) {\n this.#setStateInternal(state, \"server\");\n }\n\n /**\n * Called when the Agent's state is updated\n * @param state Updated state\n * @param source Source of the state update (\"server\" or a client connection)\n */\n onStateUpdate(state: State | undefined, source: Connection | \"server\") {\n // override this to handle state updates\n }\n\n /**\n * Called when the Agent receives an email\n * @param email Email message to process\n */\n onEmail(email: ForwardableEmailMessage) {\n return unstable_context.run(\n { agent: this, connection: undefined, request: undefined },\n async () => {\n console.error(\"onEmail not implemented\");\n }\n );\n }\n\n async #tryCatch<T>(fn: () => T | Promise<T>) {\n try {\n return await fn();\n } catch (e) {\n throw this.onError(e);\n }\n }\n\n override onError(\n connection: Connection,\n error: unknown\n ): void | Promise<void>;\n override onError(error: unknown): void | Promise<void>;\n override onError(connectionOrError: Connection | unknown, error?: unknown) {\n let theError: unknown;\n if (connectionOrError && error) {\n theError = error;\n // this is a websocket connection error\n console.error(\n \"Error on websocket connection:\",\n (connectionOrError as Connection).id,\n theError\n );\n console.error(\n \"Override onError(connection, error) to handle websocket connection errors\"\n );\n } else {\n theError = connectionOrError;\n // this is a server error\n console.error(\"Error on server:\", theError);\n console.error(\"Override onError(error) to handle server errors\");\n }\n throw theError;\n }\n\n /**\n * Render content (not implemented in base class)\n */\n render() {\n throw new Error(\"Not implemented\");\n }\n\n /**\n * Schedule a task to be executed in the future\n * @template T Type of the payload data\n * @param when When to execute the task (Date, seconds delay, or cron expression)\n * @param callback Name of the method to call\n * @param payload Data to pass to the callback\n * @returns Schedule object representing the scheduled task\n */\n async schedule<T = string>(\n when: Date | string | number,\n callback: keyof this,\n payload?: T\n ): Promise<Schedule<T>> {\n const id = nanoid(9);\n\n if (typeof callback !== \"string\") {\n throw new Error(\"Callback must be a string\");\n }\n\n if (typeof this[callback] !== \"function\") {\n throw new Error(`this.${callback} is not a function`);\n }\n\n if (when instanceof Date) {\n const timestamp = Math.floor(when.getTime() / 1000);\n this.sql`\n INSERT OR REPLACE INTO cf_agents_schedules (id, callback, payload, type, time)\n VALUES (${id}, ${callback}, ${JSON.stringify(\n payload\n )}, 'scheduled', ${timestamp})\n `;\n\n await this.#scheduleNextAlarm();\n\n return {\n id,\n callback: callback,\n payload: payload as T,\n time: timestamp,\n type: \"scheduled\",\n };\n }\n if (typeof when === \"number\") {\n const time = new Date(Date.now() + when * 1000);\n const timestamp = Math.floor(time.getTime() / 1000);\n\n this.sql`\n INSERT OR REPLACE INTO cf_agents_schedules (id, callback, payload, type, delayInSeconds, time)\n VALUES (${id}, ${callback}, ${JSON.stringify(\n payload\n )}, 'delayed', ${when}, ${timestamp})\n `;\n\n await this.#scheduleNextAlarm();\n\n return {\n id,\n callback: callback,\n payload: payload as T,\n delayInSeconds: when,\n time: timestamp,\n type: \"delayed\",\n };\n }\n if (typeof when === \"string\") {\n const nextExecutionTime = getNextCronTime(when);\n const timestamp = Math.floor(nextExecutionTime.getTime() / 1000);\n\n this.sql`\n INSERT OR REPLACE INTO cf_agents_schedules (id, callback, payload, type, cron, time)\n VALUES (${id}, ${callback}, ${JSON.stringify(\n payload\n )}, 'cron', ${when}, ${timestamp})\n `;\n\n await this.#scheduleNextAlarm();\n\n return {\n id,\n callback: callback,\n payload: payload as T,\n cron: when,\n time: timestamp,\n type: \"cron\",\n };\n }\n throw new Error(\"Invalid schedule type\");\n }\n\n /**\n * Get a scheduled task by ID\n * @template T Type of the payload data\n * @param id ID of the scheduled task\n * @returns The Schedule object or undefined if not found\n */\n async getSchedule<T = string>(id: string): Promise<Schedule<T> | undefined> {\n const result = this.sql<Schedule<string>>`\n SELECT * FROM cf_agents_schedules WHERE id = ${id}\n `;\n if (!result) {\n console.error(`schedule ${id} not found`);\n return undefined;\n }\n\n return { ...result[0], payload: JSON.parse(result[0].payload) as T };\n }\n\n /**\n * Get scheduled tasks matching the given criteria\n * @template T Type of the payload data\n * @param criteria Criteria to filter schedules\n * @returns Array of matching Schedule objects\n */\n getSchedules<T = string>(\n criteria: {\n id?: string;\n type?: \"scheduled\" | \"delayed\" | \"cron\";\n timeRange?: { start?: Date; end?: Date };\n } = {}\n ): Schedule<T>[] {\n let query = \"SELECT * FROM cf_agents_schedules WHERE 1=1\";\n const params = [];\n\n if (criteria.id) {\n query += \" AND id = ?\";\n params.push(criteria.id);\n }\n\n if (criteria.type) {\n query += \" AND type = ?\";\n params.push(criteria.type);\n }\n\n if (criteria.timeRange) {\n query += \" AND time >= ? AND time <= ?\";\n const start = criteria.timeRange.start || new Date(0);\n const end = criteria.timeRange.end || new Date(999999999999999);\n params.push(\n Math.floor(start.getTime() / 1000),\n Math.floor(end.getTime() / 1000)\n );\n }\n\n const result = this.ctx.storage.sql\n .exec(query, ...params)\n .toArray()\n .map((row) => ({\n ...row,\n payload: JSON.parse(row.payload as string) as T,\n })) as Schedule<T>[];\n\n return result;\n }\n\n /**\n * Cancel a scheduled task\n * @param id ID of the task to cancel\n * @returns true if the task was cancelled, false otherwise\n */\n async cancelSchedule(id: string): Promise<boolean> {\n this.sql`DELETE FROM cf_agents_schedules WHERE id = ${id}`;\n\n await this.#scheduleNextAlarm();\n return true;\n }\n\n async #scheduleNextAlarm() {\n // Find the next schedule that needs to be executed\n const result = this.sql`\n SELECT time FROM cf_agents_schedules \n WHERE time > ${Math.floor(Date.now() / 1000)}\n ORDER BY time ASC \n LIMIT 1\n `;\n if (!result) return;\n\n if (result.length > 0 && \"time\" in result[0]) {\n const nextTime = (result[0].time as number) * 1000;\n await this.ctx.storage.setAlarm(nextTime);\n }\n }\n\n /**\n * Method called when an alarm fires\n * Executes any scheduled tasks that are due\n */\n async alarm() {\n const now = Math.floor(Date.now() / 1000);\n\n // Get all schedules that should be executed now\n const result = this.sql<Schedule<string>>`\n SELECT * FROM cf_agents_schedules WHERE time <= ${now}\n `;\n\n for (const row of result || []) {\n const callback = this[row.callback as keyof Agent<Env>];\n if (!callback) {\n console.error(`callback ${row.callback} not found`);\n continue;\n }\n await unstable_context.run(\n { agent: this, connection: undefined, request: undefined },\n async () => {\n try {\n await (\n callback as (\n payload: unknown,\n schedule: Schedule<unknown>\n ) => Promise<void>\n ).bind(this)(JSON.parse(row.payload as string), row);\n } catch (e) {\n console.error(`error executing callback \"${row.callback}\"`, e);\n }\n }\n );\n if (row.type === \"cron\") {\n // Update next execution time for cron schedules\n const nextExecutionTime = getNextCronTime(row.cron);\n const nextTimestamp = Math.floor(nextExecutionTime.getTime() / 1000);\n\n this.sql`\n UPDATE cf_agents_schedules SET time = ${nextTimestamp} WHERE id = ${row.id}\n `;\n } else {\n // Delete one-time schedules after execution\n this.sql`\n DELETE FROM cf_agents_schedules WHERE id = ${row.id}\n `;\n }\n }\n\n // Schedule the next alarm\n await this.#scheduleNextAlarm();\n }\n\n /**\n * Destroy the Agent, removing all state and scheduled tasks\n */\n async destroy() {\n // drop all tables\n this.sql`DROP TABLE IF EXISTS cf_agents_state`;\n this.sql`DROP TABLE IF EXISTS cf_agents_schedules`;\n\n // delete all alarms\n await this.ctx.storage.deleteAlarm();\n await this.ctx.storage.deleteAll();\n }\n\n /**\n * Get all methods marked as callable on this Agent\n * @returns A map of method names to their metadata\n */\n #isCallable(method: string): boolean {\n // biome-ignore lint/complexity/noBannedTypes: <explanation>\n return callableMetadata.has(this[method as keyof this] as Function);\n }\n}\n\n/**\n * Namespace for creating Agent instances\n * @template Agentic Type of the Agent class\n */\nexport type AgentNamespace<Agentic extends Agent<unknown>> =\n DurableObjectNamespace<Agentic>;\n\n/**\n * Agent's durable context\n */\nexport type AgentContext = DurableObjectState;\n\n/**\n * Configuration options for Agent routing\n */\nexport type AgentOptions<Env> = PartyServerOptions<Env> & {\n /**\n * Whether to enable CORS for the Agent\n */\n cors?: boolean | HeadersInit | undefined;\n};\n\n/**\n * Route a request to the appropriate Agent\n * @param request Request to route\n * @param env Environment containing Agent bindings\n * @param options Routing options\n * @returns Response from the Agent or undefined if no route matched\n */\nexport async function routeAgentRequest<Env>(\n request: Request,\n env: Env,\n options?: AgentOptions<Env>\n) {\n const corsHeaders =\n options?.cors === true\n ? {\n \"Access-Control-Allow-Origin\": \"*\",\n \"Access-Control-Allow-Methods\": \"GET, POST, HEAD, OPTIONS\",\n \"Access-Control-Allow-Credentials\": \"true\",\n \"Access-Control-Max-Age\": \"86400\",\n }\n : options?.cors;\n\n if (request.method === \"OPTIONS\") {\n if (corsHeaders) {\n return new Response(null, {\n headers: corsHeaders,\n });\n }\n console.warn(\n \"Received an OPTIONS request, but cors was not enabled. Pass `cors: true` or `cors: { ...custom cors headers }` to routeAgentRequest to enable CORS.\"\n );\n }\n\n let response = await routePartykitRequest(\n request,\n env as Record<string, unknown>,\n {\n prefix: \"agents\",\n ...(options as PartyServerOptions<Record<string, unknown>>),\n }\n );\n\n if (\n response &&\n corsHeaders &&\n request.headers.get(\"upgrade\")?.toLowerCase() !== \"websocket\" &&\n request.headers.get(\"Upgrade\")?.toLowerCase() !== \"websocket\"\n ) {\n response = new Response(response.body, {\n headers: {\n ...response.headers,\n ...corsHeaders,\n },\n });\n }\n return response;\n}\n\n/**\n * Route an email to the appropriate Agent\n * @param email Email message to route\n * @param env Environment containing Agent bindings\n * @param options Routing options\n */\nexport async function routeAgentEmail<Env>(\n email: ForwardableEmailMessage,\n env: Env,\n options?: AgentOptions<Env>\n): Promise<void> {}\n\n/**\n * Get or create an Agent by name\n * @template Env Environment type containing bindings\n * @template T Type of the Agent class\n * @param namespace Agent namespace\n * @param name Name of the Agent instance\n * @param options Options for Agent creation\n * @returns Promise resolving to an Agent instance stub\n */\nexport function getAgentByName<Env, T extends Agent<Env>>(\n namespace: AgentNamespace<T>,\n name: string,\n options?: {\n jurisdiction?: DurableObjectJurisdiction;\n locationHint?: DurableObjectLocationHint;\n }\n) {\n return getServerByName<Env, T>(namespace, name, options);\n}\n\n/**\n * A wrapper for streaming responses in callable methods\n */\nexport class StreamingResponse {\n #connection: Connection;\n #id: string;\n #closed = false;\n\n constructor(connection: Connection, id: string) {\n this.#connection = connection;\n this.#id = id;\n }\n\n /**\n * Send a chunk of data to the client\n * @param chunk The data to send\n */\n send(chunk: unknown) {\n if (this.#closed) {\n throw new Error(\"StreamingResponse is already closed\");\n }\n const response: RPCResponse = {\n type: \"rpc\",\n id: this.#id,\n success: true,\n result: chunk,\n done: false,\n };\n this.#connection.send(JSON.stringify(response));\n }\n\n /**\n * End the stream and send the final chunk (if any)\n * @param finalChunk Optional final chunk of data to send\n */\n end(finalChunk?: unknown) {\n if (this.#closed) {\n throw new Error(\"StreamingResponse is already closed\");\n }\n this.#closed = true;\n const response: RPCResponse = {\n type: \"rpc\",\n id: this.#id,\n success: true,\n result: finalChunk,\n done: true,\n };\n this.#connection.send(JSON.stringify(response));\n }\n}\n"],"mappings":";;;;;;;;AAAA;AAAA,EACE;AAAA,EACA;AAAA,EAEA;AAAA,OAIK;AAEP,SAAS,2BAA2B;AACpC,SAAS,cAAc;AAEvB,SAAS,yBAAyB;AAIlC,SAAS,sBAAsB,4BAA4B;AA8C3D,SAAS,aAAa,KAAiC;AACrD,SACE,OAAO,QAAQ,YACf,QAAQ,QACR,UAAU,OACV,IAAI,SAAS,SACb,QAAQ,OACR,OAAO,IAAI,OAAO,YAClB,YAAY,OACZ,OAAO,IAAI,WAAW,YACtB,UAAU,OACV,MAAM,QAAS,IAAmB,IAAI;AAE1C;AAKA,SAAS,qBAAqB,KAAyC;AACrE,SACE,OAAO,QAAQ,YACf,QAAQ,QACR,UAAU,OACV,IAAI,SAAS,oBACb,WAAW;AAEf;AAaA,IAAM,mBAAmB,oBAAI,IAAgC;AAMtD,SAAS,kBAAkB,WAA6B,CAAC,GAAG;AACjE,SAAO,SAAS,kBACd,QACA,SACA;AACA,QAAI,CAAC,iBAAiB,IAAI,MAAM,GAAG;AACjC,uBAAiB,IAAI,QAAQ,QAAQ;AAAA,IACvC;AAEA,WAAO;AAAA,EACT;AACF;AAKO,IAAM,qBAAN,cAAiC,qBAAqB;AAAC;AAsC9D,SAAS,gBAAgB,MAAc;AACrC,QAAM,WAAW,oBAAoB,IAAI;AACzC,SAAO,SAAS,YAAY;AAC9B;AAEA,IAAM,eAAe;AACrB,IAAM,oBAAoB;AAE1B,IAAM,gBAAgB,CAAC;AAEhB,IAAM,mBAAmB,IAAI,kBAIjC;AAhLH;AAuLO,IAAM,QAAN,cAA0C,OAAY;AAAA,EAsF3D,YAAY,KAAmB,KAAU;AACvC,UAAM,KAAK,GAAG;AAvFX;AACL,+BAAS;AAMT;AAAA;AAAA;AAAA;AAAA,wBAAsB;AAkFpB,SAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAOL,SAAK,KAAK,IAAI,sBAAsB,YAAY;AAC9C,aAAO,sBAAK,+BAAL,WAAe,YAAY;AAEhC,aAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAcL,cAAM,KAAK,MAAM;AAAA,MACnB;AAAA,IACF,CAAC;AAED,UAAM,aAAa,KAAK,UAAU,KAAK,IAAI;AAC3C,SAAK,YAAY,OAAO,YAAwB,YAAuB;AACrE,aAAO,iBAAiB;AAAA,QACtB,EAAE,OAAO,MAAM,YAAY,SAAS,OAAU;AAAA,QAC9C,YAAY;AACV,cAAI,OAAO,YAAY,UAAU;AAC/B,mBAAO,sBAAK,+BAAL,WAAe,MAAM,WAAW,YAAY,OAAO;AAAA,UAC5D;AAEA,cAAI;AACJ,cAAI;AACF,qBAAS,KAAK,MAAM,OAAO;AAAA,UAC7B,SAAS,GAAG;AAEV,mBAAO,sBAAK,+BAAL,WAAe,MAAM,WAAW,YAAY,OAAO;AAAA,UAC5D;AAEA,cAAI,qBAAqB,MAAM,GAAG;AAChC,kCAAK,uCAAL,WAAuB,OAAO,OAAgB;AAC9C;AAAA,UACF;AAEA,cAAI,aAAa,MAAM,GAAG;AACxB,gBAAI;AACF,oBAAM,EAAE,IAAI,QAAQ,KAAK,IAAI;AAG7B,oBAAM,WAAW,KAAK,MAAoB;AAC1C,kBAAI,OAAO,aAAa,YAAY;AAClC,sBAAM,IAAI,MAAM,UAAU,MAAM,iBAAiB;AAAA,cACnD;AAEA,kBAAI,CAAC,sBAAK,iCAAL,WAAiB,SAAS;AAC7B,sBAAM,IAAI,MAAM,UAAU,MAAM,kBAAkB;AAAA,cACpD;AAGA,oBAAM,WAAW,iBAAiB,IAAI,QAAoB;AAG1D,kBAAI,UAAU,WAAW;AACvB,sBAAM,SAAS,IAAI,kBAAkB,YAAY,EAAE;AACnD,sBAAM,SAAS,MAAM,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC;AAC5C;AAAA,cACF;AAGA,oBAAM,SAAS,MAAM,SAAS,MAAM,MAAM,IAAI;AAC9C,oBAAM,WAAwB;AAAA,gBAC5B,MAAM;AAAA,gBACN;AAAA,gBACA,SAAS;AAAA,gBACT;AAAA,gBACA,MAAM;AAAA,cACR;AACA,yBAAW,KAAK,KAAK,UAAU,QAAQ,CAAC;AAAA,YAC1C,SAAS,GAAG;AAEV,oBAAM,WAAwB;AAAA,gBAC5B,MAAM;AAAA,gBACN,IAAI,OAAO;AAAA,gBACX,SAAS;AAAA,gBACT,OACE,aAAa,QAAQ,EAAE,UAAU;AAAA,cACrC;AACA,yBAAW,KAAK,KAAK,UAAU,QAAQ,CAAC;AACxC,sBAAQ,MAAM,cAAc,CAAC;AAAA,YAC/B;AACA;AAAA,UACF;AAEA,iBAAO,sBAAK,+BAAL,WAAe,MAAM,WAAW,YAAY,OAAO;AAAA,QAC5D;AAAA,MACF;AAAA,IACF;AAEA,UAAM,aAAa,KAAK,UAAU,KAAK,IAAI;AAC3C,SAAK,YAAY,CAAC,YAAwBA,SAA2B;AAGnE,aAAO,iBAAiB;AAAA,QACtB,EAAE,OAAO,MAAM,YAAY,SAASA,KAAI,QAAQ;AAAA,QAChD,YAAY;AACV,qBAAW,MAAM;AACf,gBAAI,KAAK,OAAO;AACd,yBAAW;AAAA,gBACT,KAAK,UAAU;AAAA,kBACb,MAAM;AAAA,kBACN,OAAO,KAAK;AAAA,gBACd,CAAC;AAAA,cACH;AAAA,YACF;AACA,mBAAO,sBAAK,+BAAL,WAAe,MAAM,WAAW,YAAYA,IAAG;AAAA,UACxD,GAAG,EAAE;AAAA,QACP;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EA1MA,IAAI,QAAe;AACjB,QAAI,mBAAK,YAAW,eAAe;AAEjC,aAAO,mBAAK;AAAA,IACd;AAGA,UAAM,aAAa,KAAK;AAAA,uDAC2B,iBAAiB;AAAA;AAIpE,UAAM,SAAS,KAAK;AAAA,qDAC6B,YAAY;AAAA;AAG7D,QACE,WAAW,CAAC,GAAG,UAAU;AAAA,IAEzB,OAAO,CAAC,GAAG,OACX;AACA,YAAM,QAAQ,OAAO,CAAC,GAAG;AAEzB,yBAAK,QAAS,KAAK,MAAM,KAAK;AAC9B,aAAO,mBAAK;AAAA,IACd;AAKA,QAAI,KAAK,iBAAiB,eAAe;AAEvC,aAAO;AAAA,IACT;AAGA,SAAK,SAAS,KAAK,YAAY;AAC/B,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAiBA,IACE,YACG,QACH;AACA,QAAI,QAAQ;AACZ,QAAI;AAEF,cAAQ,QAAQ;AAAA,QACd,CAAC,KAAK,KAAK,MAAM,MAAM,OAAO,IAAI,OAAO,SAAS,MAAM;AAAA,QACxD;AAAA,MACF;AAGA,aAAO,CAAC,GAAG,KAAK,IAAI,QAAQ,IAAI,KAAK,OAAO,GAAG,MAAM,CAAC;AAAA,IACxD,SAAS,GAAG;AACV,cAAQ,MAAM,gCAAgC,KAAK,IAAI,CAAC;AACxD,YAAM,KAAK,QAAQ,CAAC;AAAA,IACtB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAmKA,SAAS,OAAc;AACrB,0BAAK,uCAAL,WAAuB,OAAO;AAAA,EAChC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,cAAc,OAA0B,QAA+B;AAAA,EAEvE;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,QAAQ,OAAgC;AACtC,WAAO,iBAAiB;AAAA,MACtB,EAAE,OAAO,MAAM,YAAY,QAAW,SAAS,OAAU;AAAA,MACzD,YAAY;AACV,gBAAQ,MAAM,yBAAyB;AAAA,MACzC;AAAA,IACF;AAAA,EACF;AAAA,EAeS,QAAQ,mBAAyC,OAAiB;AACzE,QAAI;AACJ,QAAI,qBAAqB,OAAO;AAC9B,iBAAW;AAEX,cAAQ;AAAA,QACN;AAAA,QACC,kBAAiC;AAAA,QAClC;AAAA,MACF;AACA,cAAQ;AAAA,QACN;AAAA,MACF;AAAA,IACF,OAAO;AACL,iBAAW;AAEX,cAAQ,MAAM,oBAAoB,QAAQ;AAC1C,cAAQ,MAAM,iDAAiD;AAAA,IACjE;AACA,UAAM;AAAA,EACR;AAAA;AAAA;AAAA;AAAA,EAKA,SAAS;AACP,UAAM,IAAI,MAAM,iBAAiB;AAAA,EACnC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,SACJ,MACA,UACA,SACsB;AACtB,UAAM,KAAK,OAAO,CAAC;AAEnB,QAAI,OAAO,aAAa,UAAU;AAChC,YAAM,IAAI,MAAM,2BAA2B;AAAA,IAC7C;AAEA,QAAI,OAAO,KAAK,QAAQ,MAAM,YAAY;AACxC,YAAM,IAAI,MAAM,QAAQ,QAAQ,oBAAoB;AAAA,IACtD;AAEA,QAAI,gBAAgB,MAAM;AACxB,YAAM,YAAY,KAAK,MAAM,KAAK,QAAQ,IAAI,GAAI;AAClD,WAAK;AAAA;AAAA,kBAEO,EAAE,KAAK,QAAQ,KAAK,KAAK;AAAA,QACjC;AAAA,MACF,CAAC,kBAAkB,SAAS;AAAA;AAG9B,YAAM,sBAAK,wCAAL;AAEN,aAAO;AAAA,QACL;AAAA,QACA;AAAA,QACA;AAAA,QACA,MAAM;AAAA,QACN,MAAM;AAAA,MACR;AAAA,IACF;AACA,QAAI,OAAO,SAAS,UAAU;AAC5B,YAAM,OAAO,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,GAAI;AAC9C,YAAM,YAAY,KAAK,MAAM,KAAK,QAAQ,IAAI,GAAI;AAElD,WAAK;AAAA;AAAA,kBAEO,EAAE,KAAK,QAAQ,KAAK,KAAK;AAAA,QACjC;AAAA,MACF,CAAC,gBAAgB,IAAI,KAAK,SAAS;AAAA;AAGrC,YAAM,sBAAK,wCAAL;AAEN,aAAO;AAAA,QACL;AAAA,QACA;AAAA,QACA;AAAA,QACA,gBAAgB;AAAA,QAChB,MAAM;AAAA,QACN,MAAM;AAAA,MACR;AAAA,IACF;AACA,QAAI,OAAO,SAAS,UAAU;AAC5B,YAAM,oBAAoB,gBAAgB,IAAI;AAC9C,YAAM,YAAY,KAAK,MAAM,kBAAkB,QAAQ,IAAI,GAAI;AAE/D,WAAK;AAAA;AAAA,kBAEO,EAAE,KAAK,QAAQ,KAAK,KAAK;AAAA,QACjC;AAAA,MACF,CAAC,aAAa,IAAI,KAAK,SAAS;AAAA;AAGlC,YAAM,sBAAK,wCAAL;AAEN,aAAO;AAAA,QACL;AAAA,QACA;AAAA,QACA;AAAA,QACA,MAAM;AAAA,QACN,MAAM;AAAA,QACN,MAAM;AAAA,MACR;AAAA,IACF;AACA,UAAM,IAAI,MAAM,uBAAuB;AAAA,EACzC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,YAAwB,IAA8C;AAC1E,UAAM,SAAS,KAAK;AAAA,qDAC6B,EAAE;AAAA;AAEnD,QAAI,CAAC,QAAQ;AACX,cAAQ,MAAM,YAAY,EAAE,YAAY;AACxC,aAAO;AAAA,IACT;AAEA,WAAO,EAAE,GAAG,OAAO,CAAC,GAAG,SAAS,KAAK,MAAM,OAAO,CAAC,EAAE,OAAO,EAAO;AAAA,EACrE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,aACE,WAII,CAAC,GACU;AACf,QAAI,QAAQ;AACZ,UAAM,SAAS,CAAC;AAEhB,QAAI,SAAS,IAAI;AACf,eAAS;AACT,aAAO,KAAK,SAAS,EAAE;AAAA,IACzB;AAEA,QAAI,SAAS,MAAM;AACjB,eAAS;AACT,aAAO,KAAK,SAAS,IAAI;AAAA,IAC3B;AAEA,QAAI,SAAS,WAAW;AACtB,eAAS;AACT,YAAM,QAAQ,SAAS,UAAU,SAAS,oBAAI,KAAK,CAAC;AACpD,YAAM,MAAM,SAAS,UAAU,OAAO,oBAAI,KAAK,eAAe;AAC9D,aAAO;AAAA,QACL,KAAK,MAAM,MAAM,QAAQ,IAAI,GAAI;AAAA,QACjC,KAAK,MAAM,IAAI,QAAQ,IAAI,GAAI;AAAA,MACjC;AAAA,IACF;AAEA,UAAM,SAAS,KAAK,IAAI,QAAQ,IAC7B,KAAK,OAAO,GAAG,MAAM,EACrB,QAAQ,EACR,IAAI,CAAC,SAAS;AAAA,MACb,GAAG;AAAA,MACH,SAAS,KAAK,MAAM,IAAI,OAAiB;AAAA,IAC3C,EAAE;AAEJ,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,eAAe,IAA8B;AACjD,SAAK,iDAAiD,EAAE;AAExD,UAAM,sBAAK,wCAAL;AACN,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAsBA,MAAM,QAAQ;AACZ,UAAM,MAAM,KAAK,MAAM,KAAK,IAAI,IAAI,GAAI;AAGxC,UAAM,SAAS,KAAK;AAAA,wDACgC,GAAG;AAAA;AAGvD,eAAW,OAAO,UAAU,CAAC,GAAG;AAC9B,YAAM,WAAW,KAAK,IAAI,QAA4B;AACtD,UAAI,CAAC,UAAU;AACb,gBAAQ,MAAM,YAAY,IAAI,QAAQ,YAAY;AAClD;AAAA,MACF;AACA,YAAM,iBAAiB;AAAA,QACrB,EAAE,OAAO,MAAM,YAAY,QAAW,SAAS,OAAU;AAAA,QACzD,YAAY;AACV,cAAI;AACF,kBACE,SAIA,KAAK,IAAI,EAAE,KAAK,MAAM,IAAI,OAAiB,GAAG,GAAG;AAAA,UACrD,SAAS,GAAG;AACV,oBAAQ,MAAM,6BAA6B,IAAI,QAAQ,KAAK,CAAC;AAAA,UAC/D;AAAA,QACF;AAAA,MACF;AACA,UAAI,IAAI,SAAS,QAAQ;AAEvB,cAAM,oBAAoB,gBAAgB,IAAI,IAAI;AAClD,cAAM,gBAAgB,KAAK,MAAM,kBAAkB,QAAQ,IAAI,GAAI;AAEnE,aAAK;AAAA,kDACqC,aAAa,eAAe,IAAI,EAAE;AAAA;AAAA,MAE9E,OAAO;AAEL,aAAK;AAAA,uDAC0C,IAAI,EAAE;AAAA;AAAA,MAEvD;AAAA,IACF;AAGA,UAAM,sBAAK,wCAAL;AAAA,EACR;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,UAAU;AAEd,SAAK;AACL,SAAK;AAGL,UAAM,KAAK,IAAI,QAAQ,YAAY;AACnC,UAAM,KAAK,IAAI,QAAQ,UAAU;AAAA,EACnC;AAUF;AA3jBE;AADK;AAwNL,sBAAiB,SAAC,OAAc,SAAgC,UAAU;AACxE,qBAAK,QAAS;AACd,OAAK;AAAA;AAAA,cAEK,YAAY,KAAK,KAAK,UAAU,KAAK,CAAC;AAAA;AAEhD,OAAK;AAAA;AAAA,cAEK,iBAAiB,KAAK,KAAK,UAAU,IAAI,CAAC;AAAA;AAEpD,OAAK;AAAA,IACH,KAAK,UAAU;AAAA,MACb,MAAM;AAAA,MACN;AAAA,IACF,CAAC;AAAA,IACD,WAAW,WAAW,CAAC,OAAO,EAAE,IAAI,CAAC;AAAA,EACvC;AACA,SAAO,sBAAK,+BAAL,WAAe,MAAM;AAC1B,UAAM,EAAE,YAAY,QAAQ,IAAI,iBAAiB,SAAS,KAAK,CAAC;AAChE,WAAO,iBAAiB;AAAA,MACtB,EAAE,OAAO,MAAM,YAAY,QAAQ;AAAA,MACnC,YAAY;AACV,eAAO,KAAK,cAAc,OAAO,MAAM;AAAA,MACzC;AAAA,IACF;AAAA,EACF;AACF;AAgCM,cAAY,eAAC,IAA0B;AAC3C,MAAI;AACF,WAAO,MAAM,GAAG;AAAA,EAClB,SAAS,GAAG;AACV,UAAM,KAAK,QAAQ,CAAC;AAAA,EACtB;AACF;AA0MM,uBAAkB,iBAAG;AAEzB,QAAM,SAAS,KAAK;AAAA;AAAA,qBAEH,KAAK,MAAM,KAAK,IAAI,IAAI,GAAI,CAAC;AAAA;AAAA;AAAA;AAI9C,MAAI,CAAC,OAAQ;AAEb,MAAI,OAAO,SAAS,KAAK,UAAU,OAAO,CAAC,GAAG;AAC5C,UAAM,WAAY,OAAO,CAAC,EAAE,OAAkB;AAC9C,UAAM,KAAK,IAAI,QAAQ,SAAS,QAAQ;AAAA,EAC1C;AACF;AAAA;AAAA;AAAA;AAAA;AAwEA,gBAAW,SAAC,QAAyB;AAEnC,SAAO,iBAAiB,IAAI,KAAK,MAAoB,CAAa;AACpE;AAAA;AAAA;AAAA;AA3jBW,MAuDJ,UAAU;AAAA;AAAA,EAEf,WAAW;AAAA;AACb;AAiiBF,eAAsB,kBACpB,SACA,KACA,SACA;AACA,QAAM,cACJ,SAAS,SAAS,OACd;AAAA,IACE,+BAA+B;AAAA,IAC/B,gCAAgC;AAAA,IAChC,oCAAoC;AAAA,IACpC,0BAA0B;AAAA,EAC5B,IACA,SAAS;AAEf,MAAI,QAAQ,WAAW,WAAW;AAChC,QAAI,aAAa;AACf,aAAO,IAAI,SAAS,MAAM;AAAA,QACxB,SAAS;AAAA,MACX,CAAC;AAAA,IACH;AACA,YAAQ;AAAA,MACN;AAAA,IACF;AAAA,EACF;AAEA,MAAI,WAAW,MAAM;AAAA,IACnB;AAAA,IACA;AAAA,IACA;AAAA,MACE,QAAQ;AAAA,MACR,GAAI;AAAA,IACN;AAAA,EACF;AAEA,MACE,YACA,eACA,QAAQ,QAAQ,IAAI,SAAS,GAAG,YAAY,MAAM,eAClD,QAAQ,QAAQ,IAAI,SAAS,GAAG,YAAY,MAAM,aAClD;AACA,eAAW,IAAI,SAAS,SAAS,MAAM;AAAA,MACrC,SAAS;AAAA,QACP,GAAG,SAAS;AAAA,QACZ,GAAG;AAAA,MACL;AAAA,IACF,CAAC;AAAA,EACH;AACA,SAAO;AACT;AAQA,eAAsB,gBACpB,OACA,KACA,SACe;AAAC;AAWX,SAAS,eACd,WACA,MACA,SAIA;AACA,SAAO,gBAAwB,WAAW,MAAM,OAAO;AACzD;AAn2BA;AAw2BO,IAAM,oBAAN,MAAwB;AAAA,EAK7B,YAAY,YAAwB,IAAY;AAJhD;AACA;AACA,gCAAU;AAGR,uBAAK,aAAc;AACnB,uBAAK,KAAM;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,KAAK,OAAgB;AACnB,QAAI,mBAAK,UAAS;AAChB,YAAM,IAAI,MAAM,qCAAqC;AAAA,IACvD;AACA,UAAM,WAAwB;AAAA,MAC5B,MAAM;AAAA,MACN,IAAI,mBAAK;AAAA,MACT,SAAS;AAAA,MACT,QAAQ;AAAA,MACR,MAAM;AAAA,IACR;AACA,uBAAK,aAAY,KAAK,KAAK,UAAU,QAAQ,CAAC;AAAA,EAChD;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,IAAI,YAAsB;AACxB,QAAI,mBAAK,UAAS;AAChB,YAAM,IAAI,MAAM,qCAAqC;AAAA,IACvD;AACA,uBAAK,SAAU;AACf,UAAM,WAAwB;AAAA,MAC5B,MAAM;AAAA,MACN,IAAI,mBAAK;AAAA,MACT,SAAS;AAAA,MACT,QAAQ;AAAA,MACR,MAAM;AAAA,IACR;AACA,uBAAK,aAAY,KAAK,KAAK,UAAU,QAAQ,CAAC;AAAA,EAChD;AACF;AA7CE;AACA;AACA;","names":["ctx"]}