agents 0.0.79 → 0.0.81

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
@@ -13,6 +13,20 @@ import { nanoid } from "nanoid";
13
13
 
14
14
  import { AsyncLocalStorage } from "node:async_hooks";
15
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";
25
+
26
+ import type { Client } from "@modelcontextprotocol/sdk/client/index.js";
27
+ import type { SSEClientTransportOptions } from "@modelcontextprotocol/sdk/client/sse.js";
28
+
29
+ import { camelCaseToKebabCase } from "./client";
16
30
 
17
31
  export type { Connection, WSMessage, ConnectionContext } from "partyserver";
18
32
 
@@ -159,6 +173,43 @@ function getNextCronTime(cron: string) {
159
173
  return interval.getNextDate();
160
174
  }
161
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
+
162
213
  const STATE_ROW_ID = "cf_state_row_id";
163
214
  const STATE_WAS_CHANGED = "cf_state_was_changed";
164
215
 
@@ -200,12 +251,12 @@ export function getCurrentAgent<
200
251
  * @template State State type to store within the Agent
201
252
  */
202
253
  export class Agent<Env, State = unknown> extends Server<Env> {
203
- #state = DEFAULT_STATE as State;
254
+ private _state = DEFAULT_STATE as State;
204
255
 
205
- #ParentClass: typeof Agent<Env, State> =
256
+ private _ParentClass: typeof Agent<Env, State> =
206
257
  Object.getPrototypeOf(this).constructor;
207
258
 
208
- mcp: MCPClientManager = new MCPClientManager(this.#ParentClass.name, "0.0.1");
259
+ mcp: MCPClientManager = new MCPClientManager(this._ParentClass.name, "0.0.1");
209
260
 
210
261
  /**
211
262
  * Initial state for the Agent
@@ -217,9 +268,9 @@ export class Agent<Env, State = unknown> extends Server<Env> {
217
268
  * Current state of the Agent
218
269
  */
219
270
  get state(): State {
220
- if (this.#state !== DEFAULT_STATE) {
271
+ if (this._state !== DEFAULT_STATE) {
221
272
  // state was previously set, and populated internal state
222
- return this.#state;
273
+ return this._state;
223
274
  }
224
275
  // looks like this is the first time the state is being accessed
225
276
  // check if the state was set in a previous life
@@ -239,8 +290,8 @@ export class Agent<Env, State = unknown> extends Server<Env> {
239
290
  ) {
240
291
  const state = result[0]?.state as string; // could be null?
241
292
 
242
- this.#state = JSON.parse(state);
243
- return this.#state;
293
+ this._state = JSON.parse(state);
294
+ return this._state;
244
295
  }
245
296
 
246
297
  // ok, this is the first time the state is being accessed
@@ -301,7 +352,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
301
352
  `;
302
353
 
303
354
  void this.ctx.blockConcurrencyWhile(async () => {
304
- return this.#tryCatch(async () => {
355
+ return this._tryCatch(async () => {
305
356
  // Create alarms table if it doesn't exist
306
357
  this.sql`
307
358
  CREATE TABLE IF NOT EXISTS cf_agents_schedules (
@@ -321,12 +372,42 @@ export class Agent<Env, State = unknown> extends Server<Env> {
321
372
  });
322
373
  });
323
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
+
324
387
  const _onRequest = this.onRequest.bind(this);
325
388
  this.onRequest = (request: Request) => {
326
389
  return agentContext.run(
327
390
  { agent: this, connection: undefined, request },
328
391
  async () => {
329
- return this.#tryCatch(() => _onRequest(request));
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));
330
411
  }
331
412
  );
332
413
  };
@@ -337,7 +418,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
337
418
  { agent: this, connection, request: undefined },
338
419
  async () => {
339
420
  if (typeof message !== "string") {
340
- return this.#tryCatch(() => _onMessage(connection, message));
421
+ return this._tryCatch(() => _onMessage(connection, message));
341
422
  }
342
423
 
343
424
  let parsed: unknown;
@@ -345,11 +426,11 @@ export class Agent<Env, State = unknown> extends Server<Env> {
345
426
  parsed = JSON.parse(message);
346
427
  } catch (e) {
347
428
  // silently fail and let the onMessage handler handle it
348
- return this.#tryCatch(() => _onMessage(connection, message));
429
+ return this._tryCatch(() => _onMessage(connection, message));
349
430
  }
350
431
 
351
432
  if (isStateUpdateMessage(parsed)) {
352
- this.#setStateInternal(parsed.state as State, connection);
433
+ this._setStateInternal(parsed.state as State, connection);
353
434
  return;
354
435
  }
355
436
 
@@ -363,7 +444,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
363
444
  throw new Error(`Method ${method} does not exist`);
364
445
  }
365
446
 
366
- if (!this.#isCallable(method)) {
447
+ if (!this._isCallable(method)) {
367
448
  throw new Error(`Method ${method} is not callable`);
368
449
  }
369
450
 
@@ -402,7 +483,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
402
483
  return;
403
484
  }
404
485
 
405
- return this.#tryCatch(() => _onMessage(connection, message));
486
+ return this._tryCatch(() => _onMessage(connection, message));
406
487
  }
407
488
  );
408
489
  };
@@ -423,15 +504,65 @@ export class Agent<Env, State = unknown> extends Server<Env> {
423
504
  })
424
505
  );
425
506
  }
426
- return this.#tryCatch(() => _onConnect(connection, ctx));
507
+
508
+ connection.send(
509
+ JSON.stringify({
510
+ type: "cf_agent_mcp_servers",
511
+ mcp: this.#getMcpServerStateInternal(),
512
+ })
513
+ );
514
+
515
+ return this._tryCatch(() => _onConnect(connection, ctx));
427
516
  }, 20);
428
517
  }
429
518
  );
430
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
+ };
431
559
  }
432
560
 
433
- #setStateInternal(state: State, source: Connection | "server" = "server") {
434
- this.#state = state;
561
+ private _setStateInternal(
562
+ state: State,
563
+ source: Connection | "server" = "server"
564
+ ) {
565
+ this._state = state;
435
566
  this.sql`
436
567
  INSERT OR REPLACE INTO cf_agents_state (id, state)
437
568
  VALUES (${STATE_ROW_ID}, ${JSON.stringify(state)})
@@ -447,7 +578,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
447
578
  }),
448
579
  source !== "server" ? [source.id] : []
449
580
  );
450
- return this.#tryCatch(() => {
581
+ return this._tryCatch(() => {
451
582
  const { connection, request } = agentContext.getStore() || {};
452
583
  return agentContext.run(
453
584
  { agent: this, connection, request },
@@ -463,7 +594,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
463
594
  * @param state New state to set
464
595
  */
465
596
  setState(state: State) {
466
- this.#setStateInternal(state, "server");
597
+ this._setStateInternal(state, "server");
467
598
  }
468
599
 
469
600
  /**
@@ -488,7 +619,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
488
619
  );
489
620
  }
490
621
 
491
- async #tryCatch<T>(fn: () => T | Promise<T>) {
622
+ private async _tryCatch<T>(fn: () => T | Promise<T>) {
492
623
  try {
493
624
  return await fn();
494
625
  } catch (e) {
@@ -562,7 +693,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
562
693
  )}, 'scheduled', ${timestamp})
563
694
  `;
564
695
 
565
- await this.#scheduleNextAlarm();
696
+ await this._scheduleNextAlarm();
566
697
 
567
698
  return {
568
699
  id,
@@ -583,7 +714,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
583
714
  )}, 'delayed', ${when}, ${timestamp})
584
715
  `;
585
716
 
586
- await this.#scheduleNextAlarm();
717
+ await this._scheduleNextAlarm();
587
718
 
588
719
  return {
589
720
  id,
@@ -605,7 +736,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
605
736
  )}, 'cron', ${when}, ${timestamp})
606
737
  `;
607
738
 
608
- await this.#scheduleNextAlarm();
739
+ await this._scheduleNextAlarm();
609
740
 
610
741
  return {
611
742
  id,
@@ -692,11 +823,11 @@ export class Agent<Env, State = unknown> extends Server<Env> {
692
823
  async cancelSchedule(id: string): Promise<boolean> {
693
824
  this.sql`DELETE FROM cf_agents_schedules WHERE id = ${id}`;
694
825
 
695
- await this.#scheduleNextAlarm();
826
+ await this._scheduleNextAlarm();
696
827
  return true;
697
828
  }
698
829
 
699
- async #scheduleNextAlarm() {
830
+ private async _scheduleNextAlarm() {
700
831
  // Find the next schedule that needs to be executed
701
832
  const result = this.sql`
702
833
  SELECT time FROM cf_agents_schedules
@@ -766,7 +897,7 @@ export class Agent<Env, State = unknown> extends Server<Env> {
766
897
  }
767
898
 
768
899
  // Schedule the next alarm
769
- await this.#scheduleNextAlarm();
900
+ await this._scheduleNextAlarm();
770
901
  };
771
902
 
772
903
  /**
@@ -776,20 +907,176 @@ export class Agent<Env, State = unknown> extends Server<Env> {
776
907
  // drop all tables
777
908
  this.sql`DROP TABLE IF EXISTS cf_agents_state`;
778
909
  this.sql`DROP TABLE IF EXISTS cf_agents_schedules`;
910
+ this.sql`DROP TABLE IF EXISTS cf_agents_mcp_servers`;
779
911
 
780
912
  // delete all alarms
781
913
  await this.ctx.storage.deleteAlarm();
782
914
  await this.ctx.storage.deleteAll();
783
915
  }
784
916
 
785
- /**
786
- * Get all methods marked as callable on this Agent
787
- * @returns A map of method names to their metadata
788
- */
789
- #isCallable(method: string): boolean {
917
+ private _isCallable(method: string): boolean {
790
918
  // biome-ignore lint/complexity/noBannedTypes: <explanation>
791
919
  return callableMetadata.has(this[method as keyof this] as Function);
792
920
  }
921
+
922
+ /**
923
+ * Connect to a new MCP Server
924
+ *
925
+ * @param url MCP Server SSE URL
926
+ * @param callbackHost Base host for the agent, used for the redirect URI.
927
+ * @param agentsPrefix agents routing prefix if not using `agents`
928
+ * @param options MCP client and transport (header) options
929
+ * @returns authUrl
930
+ */
931
+ async addMcpServer(
932
+ serverName: string,
933
+ url: string,
934
+ callbackHost: string,
935
+ agentsPrefix = "agents",
936
+ options?: {
937
+ client?: ConstructorParameters<typeof Client>[1];
938
+ transport?: {
939
+ headers: HeadersInit;
940
+ };
941
+ }
942
+ ): Promise<{ id: string; authUrl: string | undefined }> {
943
+ const callbackUrl = `${callbackHost}/${agentsPrefix}/${camelCaseToKebabCase(this._ParentClass.name)}/${this.name}/callback`;
944
+
945
+ const result = await this.#connectToMcpServerInternal(
946
+ serverName,
947
+ url,
948
+ callbackUrl,
949
+ options
950
+ );
951
+
952
+ this.broadcast(
953
+ JSON.stringify({
954
+ type: "cf_agent_mcp_servers",
955
+ mcp: this.#getMcpServerStateInternal(),
956
+ })
957
+ );
958
+
959
+ return result;
960
+ }
961
+
962
+ async #connectToMcpServerInternal(
963
+ serverName: string,
964
+ url: string,
965
+ callbackUrl: string,
966
+ // it's important that any options here are serializable because we put them into our sqlite DB for reconnection purposes
967
+ options?: {
968
+ client?: ConstructorParameters<typeof Client>[1];
969
+ /**
970
+ * We don't expose the normal set of transport options because:
971
+ * 1) we can't serialize things like the auth provider or a fetch function into the DB for reconnection purposes
972
+ * 2) We probably want these options to be agnostic to the transport type (SSE vs Streamable)
973
+ *
974
+ * 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).
975
+ */
976
+ transport?: {
977
+ headers?: HeadersInit;
978
+ };
979
+ },
980
+ reconnect?: {
981
+ id: string;
982
+ oauthClientId?: string;
983
+ }
984
+ ): Promise<{ id: string; authUrl: string | undefined }> {
985
+ const authProvider = new DurableObjectOAuthClientProvider(
986
+ this.ctx.storage,
987
+ this.name,
988
+ callbackUrl
989
+ );
990
+
991
+ if (reconnect) {
992
+ authProvider.serverId = reconnect.id;
993
+ if (reconnect.oauthClientId) {
994
+ authProvider.clientId = reconnect.oauthClientId;
995
+ }
996
+ }
997
+
998
+ // allows passing through transport headers if necessary
999
+ // this handles some non-standard bearer auth setups (i.e. MCP server behind CF access instead of OAuth)
1000
+ let headerTransportOpts: SSEClientTransportOptions = {};
1001
+ if (options?.transport?.headers) {
1002
+ headerTransportOpts = {
1003
+ eventSourceInit: {
1004
+ fetch: (url, init) =>
1005
+ fetch(url, {
1006
+ ...init,
1007
+ headers: options?.transport?.headers,
1008
+ }),
1009
+ },
1010
+ requestInit: {
1011
+ headers: options?.transport?.headers,
1012
+ },
1013
+ };
1014
+ }
1015
+
1016
+ const { id, authUrl, clientId } = await this.mcp.connect(url, {
1017
+ reconnect,
1018
+ transport: {
1019
+ ...headerTransportOpts,
1020
+ authProvider,
1021
+ },
1022
+ client: options?.client,
1023
+ });
1024
+
1025
+ this.sql`
1026
+ INSERT OR REPLACE INTO cf_agents_mcp_servers (id, name, server_url, client_id, auth_url, callback_url, server_options)
1027
+ VALUES (
1028
+ ${id},
1029
+ ${serverName},
1030
+ ${url},
1031
+ ${clientId ?? null},
1032
+ ${authUrl ?? null},
1033
+ ${callbackUrl},
1034
+ ${options ? JSON.stringify(options) : null}
1035
+ );
1036
+ `;
1037
+
1038
+ return {
1039
+ id,
1040
+ authUrl,
1041
+ };
1042
+ }
1043
+
1044
+ async removeMcpServer(id: string) {
1045
+ this.mcp.closeConnection(id);
1046
+ this.sql`
1047
+ DELETE FROM cf_agents_mcp_servers WHERE id = ${id};
1048
+ `;
1049
+ this.broadcast(
1050
+ JSON.stringify({
1051
+ type: "cf_agent_mcp_servers",
1052
+ mcp: this.#getMcpServerStateInternal(),
1053
+ })
1054
+ );
1055
+ }
1056
+
1057
+ #getMcpServerStateInternal(): MCPServersState {
1058
+ const mcpState: MCPServersState = {
1059
+ servers: {},
1060
+ tools: this.mcp.listTools(),
1061
+ prompts: this.mcp.listPrompts(),
1062
+ resources: this.mcp.listResources(),
1063
+ };
1064
+
1065
+ const servers = this.sql<MCPServerRow>`
1066
+ SELECT id, name, server_url, client_id, auth_url, callback_url, server_options FROM cf_agents_mcp_servers;
1067
+ `;
1068
+
1069
+ for (const server of servers) {
1070
+ mcpState.servers[server.id] = {
1071
+ name: server.name,
1072
+ server_url: server.server_url,
1073
+ auth_url: server.auth_url,
1074
+ state: this.mcp.mcpConnections[server.id].connectionState,
1075
+ };
1076
+ }
1077
+
1078
+ return mcpState;
1079
+ }
793
1080
  }
794
1081
 
795
1082
  /**
@@ -908,13 +1195,13 @@ export async function getAgentByName<Env, T extends Agent<Env>>(
908
1195
  * A wrapper for streaming responses in callable methods
909
1196
  */
910
1197
  export class StreamingResponse {
911
- #connection: Connection;
912
- #id: string;
913
- #closed = false;
1198
+ private _connection: Connection;
1199
+ private _id: string;
1200
+ private _closed = false;
914
1201
 
915
1202
  constructor(connection: Connection, id: string) {
916
- this.#connection = connection;
917
- this.#id = id;
1203
+ this._connection = connection;
1204
+ this._id = id;
918
1205
  }
919
1206
 
920
1207
  /**
@@ -922,17 +1209,17 @@ export class StreamingResponse {
922
1209
  * @param chunk The data to send
923
1210
  */
924
1211
  send(chunk: unknown) {
925
- if (this.#closed) {
1212
+ if (this._closed) {
926
1213
  throw new Error("StreamingResponse is already closed");
927
1214
  }
928
1215
  const response: RPCResponse = {
929
1216
  type: "rpc",
930
- id: this.#id,
1217
+ id: this._id,
931
1218
  success: true,
932
1219
  result: chunk,
933
1220
  done: false,
934
1221
  };
935
- this.#connection.send(JSON.stringify(response));
1222
+ this._connection.send(JSON.stringify(response));
936
1223
  }
937
1224
 
938
1225
  /**
@@ -940,17 +1227,17 @@ export class StreamingResponse {
940
1227
  * @param finalChunk Optional final chunk of data to send
941
1228
  */
942
1229
  end(finalChunk?: unknown) {
943
- if (this.#closed) {
1230
+ if (this._closed) {
944
1231
  throw new Error("StreamingResponse is already closed");
945
1232
  }
946
- this.#closed = true;
1233
+ this._closed = true;
947
1234
  const response: RPCResponse = {
948
1235
  type: "rpc",
949
- id: this.#id,
1236
+ id: this._id,
950
1237
  success: true,
951
1238
  result: finalChunk,
952
1239
  done: true,
953
1240
  };
954
- this.#connection.send(JSON.stringify(response));
1241
+ this._connection.send(JSON.stringify(response));
955
1242
  }
956
1243
  }
@@ -1 +0,0 @@
1
- {"version":3,"sources":["../src/mcp/sse-edge.ts","../src/mcp/client-connection.ts","../src/mcp/client.ts"],"sourcesContent":["import {\n SSEClientTransport,\n type SSEClientTransportOptions,\n} from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport type { OAuthClientProvider } from \"@modelcontextprotocol/sdk/client/auth.js\";\n\nexport class SSEEdgeClientTransport extends SSEClientTransport {\n private authProvider: OAuthClientProvider | undefined;\n /**\n * Creates a new EdgeSSEClientTransport, which overrides fetch to be compatible with the CF workers environment\n */\n constructor(url: URL, options: SSEClientTransportOptions) {\n const fetchOverride: typeof fetch = async (\n fetchUrl: RequestInfo | URL,\n fetchInit: RequestInit = {}\n ) => {\n // add auth headers\n const headers = await this.authHeaders();\n const workerOptions = {\n ...fetchInit,\n headers: {\n ...options.requestInit?.headers,\n ...fetchInit?.headers,\n ...headers,\n },\n };\n\n // Remove unsupported properties\n // biome-ignore lint/performance/noDelete: workaround for workers environment\n delete workerOptions.mode;\n\n // Call the original fetch with fixed options\n return (\n (options.eventSourceInit?.fetch?.(\n fetchUrl as URL | string,\n // @ts-expect-error Expects FetchLikeInit from EventSource but is compatible with RequestInit\n workerOptions\n ) as Promise<Response>) || fetch(fetchUrl, workerOptions)\n );\n };\n\n super(url, {\n ...options,\n eventSourceInit: {\n ...options.eventSourceInit,\n fetch: fetchOverride,\n },\n });\n this.authProvider = options.authProvider;\n }\n\n async authHeaders() {\n if (this.authProvider) {\n const tokens = await this.authProvider.tokens();\n if (tokens) {\n return {\n Authorization: `Bearer ${tokens.access_token}`,\n };\n }\n }\n }\n}\n","import { SSEEdgeClientTransport } from \"./sse-edge\";\n\nimport {\n ToolListChangedNotificationSchema,\n type ClientCapabilities,\n type Resource,\n type Tool,\n type Prompt,\n ResourceListChangedNotificationSchema,\n PromptListChangedNotificationSchema,\n type ListToolsResult,\n type ListResourcesResult,\n type ListPromptsResult,\n type ServerCapabilities,\n type ResourceTemplate,\n type ListResourceTemplatesResult,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport type { SSEClientTransportOptions } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport type { AgentsOAuthProvider } from \"./do-oauth-client-provider\";\n\nexport class MCPClientConnection {\n client: Client;\n connectionState:\n | \"authenticating\"\n | \"connecting\"\n | \"ready\"\n | \"discovering\"\n | \"failed\" = \"connecting\";\n instructions?: string;\n tools: Tool[] = [];\n prompts: Prompt[] = [];\n resources: Resource[] = [];\n resourceTemplates: ResourceTemplate[] = [];\n serverCapabilities: ServerCapabilities | undefined;\n\n constructor(\n public url: URL,\n info: ConstructorParameters<typeof Client>[0],\n public options: {\n transport: SSEClientTransportOptions & {\n authProvider?: AgentsOAuthProvider;\n };\n client: ConstructorParameters<typeof Client>[1];\n capabilities: ClientCapabilities;\n } = { transport: {}, client: {}, capabilities: {} }\n ) {\n this.client = new Client(info, options.client);\n this.client.registerCapabilities(options.capabilities);\n }\n\n /**\n * Initialize a client connection\n *\n * @param code Optional OAuth code to initialize the connection with if auth hasn't been initialized\n * @returns\n */\n async init(code?: string, clientId?: string) {\n try {\n const transport = new SSEEdgeClientTransport(\n this.url,\n this.options.transport\n );\n if (code) {\n await transport.finishAuth(code);\n }\n\n await this.client.connect(transport);\n // biome-ignore lint/suspicious/noExplicitAny: allow for the error check here\n } catch (e: any) {\n if (e.toString().includes(\"Unauthorized\")) {\n // unauthorized, we should wait for the user to authenticate\n this.connectionState = \"authenticating\";\n return;\n }\n this.connectionState = \"failed\";\n throw e;\n }\n\n this.connectionState = \"discovering\";\n\n this.serverCapabilities = await this.client.getServerCapabilities();\n if (!this.serverCapabilities) {\n throw new Error(\"The MCP Server failed to return server capabilities\");\n }\n\n const [instructions, tools, resources, prompts, resourceTemplates] =\n await Promise.all([\n this.client.getInstructions(),\n this.registerTools(),\n this.registerResources(),\n this.registerPrompts(),\n this.registerResourceTemplates(),\n ]);\n\n this.instructions = instructions;\n this.tools = tools;\n this.resources = resources;\n this.prompts = prompts;\n this.resourceTemplates = resourceTemplates;\n\n this.connectionState = \"ready\";\n }\n\n /**\n * Notification handler registration\n */\n async registerTools(): Promise<Tool[]> {\n if (!this.serverCapabilities || !this.serverCapabilities.tools) {\n return [];\n }\n\n if (this.serverCapabilities.tools.listChanged) {\n this.client.setNotificationHandler(\n ToolListChangedNotificationSchema,\n async (_notification) => {\n this.tools = await this.fetchTools();\n }\n );\n }\n\n return this.fetchTools();\n }\n\n async registerResources(): Promise<Resource[]> {\n if (!this.serverCapabilities || !this.serverCapabilities.resources) {\n return [];\n }\n\n if (this.serverCapabilities.resources.listChanged) {\n this.client.setNotificationHandler(\n ResourceListChangedNotificationSchema,\n async (_notification) => {\n this.resources = await this.fetchResources();\n }\n );\n }\n\n return this.fetchResources();\n }\n\n async registerPrompts(): Promise<Prompt[]> {\n if (!this.serverCapabilities || !this.serverCapabilities.prompts) {\n return [];\n }\n\n if (this.serverCapabilities.prompts.listChanged) {\n this.client.setNotificationHandler(\n PromptListChangedNotificationSchema,\n async (_notification) => {\n this.prompts = await this.fetchPrompts();\n }\n );\n }\n\n return this.fetchPrompts();\n }\n\n async registerResourceTemplates(): Promise<ResourceTemplate[]> {\n if (!this.serverCapabilities || !this.serverCapabilities.resources) {\n return [];\n }\n\n return this.fetchResourceTemplates();\n }\n\n async fetchTools() {\n let toolsAgg: Tool[] = [];\n let toolsResult: ListToolsResult = { tools: [] };\n do {\n toolsResult = await this.client\n .listTools({\n cursor: toolsResult.nextCursor,\n })\n .catch(capabilityErrorHandler({ tools: [] }, \"tools/list\"));\n toolsAgg = toolsAgg.concat(toolsResult.tools);\n } while (toolsResult.nextCursor);\n return toolsAgg;\n }\n\n async fetchResources() {\n let resourcesAgg: Resource[] = [];\n let resourcesResult: ListResourcesResult = { resources: [] };\n do {\n resourcesResult = await this.client\n .listResources({\n cursor: resourcesResult.nextCursor,\n })\n .catch(capabilityErrorHandler({ resources: [] }, \"resources/list\"));\n resourcesAgg = resourcesAgg.concat(resourcesResult.resources);\n } while (resourcesResult.nextCursor);\n return resourcesAgg;\n }\n\n async fetchPrompts() {\n let promptsAgg: Prompt[] = [];\n let promptsResult: ListPromptsResult = { prompts: [] };\n do {\n promptsResult = await this.client\n .listPrompts({\n cursor: promptsResult.nextCursor,\n })\n .catch(capabilityErrorHandler({ prompts: [] }, \"prompts/list\"));\n promptsAgg = promptsAgg.concat(promptsResult.prompts);\n } while (promptsResult.nextCursor);\n return promptsAgg;\n }\n\n async fetchResourceTemplates() {\n let templatesAgg: ResourceTemplate[] = [];\n let templatesResult: ListResourceTemplatesResult = {\n resourceTemplates: [],\n };\n do {\n templatesResult = await this.client\n .listResourceTemplates({\n cursor: templatesResult.nextCursor,\n })\n .catch(\n capabilityErrorHandler(\n { resourceTemplates: [] },\n \"resources/templates/list\"\n )\n );\n templatesAgg = templatesAgg.concat(templatesResult.resourceTemplates);\n } while (templatesResult.nextCursor);\n return templatesAgg;\n }\n}\n\nfunction capabilityErrorHandler<T>(empty: T, method: string) {\n return (e: { code: number }) => {\n // server is badly behaved and returning invalid capabilities. This commonly occurs for resource templates\n if (e.code === -32601) {\n console.error(\n `The server advertised support for the capability ${method.split(\"/\")[0]}, but returned \"Method not found\" for '${method}'.`\n );\n return empty;\n }\n throw e;\n };\n}\n","import { MCPClientConnection } from \"./client-connection\";\n\nimport type {\n ClientCapabilities,\n CallToolRequest,\n CallToolResultSchema,\n CompatibilityCallToolResultSchema,\n ReadResourceRequest,\n GetPromptRequest,\n Tool,\n Resource,\n Prompt,\n ResourceTemplate,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport type { SSEClientTransportOptions } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport type { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport type { RequestOptions } from \"@modelcontextprotocol/sdk/shared/protocol.js\";\nimport type { AgentsOAuthProvider } from \"./do-oauth-client-provider\";\nimport { jsonSchema, type ToolSet } from \"ai\";\nimport { nanoid } from \"nanoid\";\n\n/**\n * Utility class that aggregates multiple MCP clients into one\n */\nexport class MCPClientManager {\n public mcpConnections: Record<string, MCPClientConnection> = {};\n private callbackUrls: string[] = [];\n\n /**\n * @param name Name of the MCP client\n * @param version Version of the MCP Client\n * @param auth Auth paramters if being used to create a DurableObjectOAuthClientProvider\n */\n constructor(\n private name: string,\n private version: string\n ) {}\n\n /**\n * Connect to and register an MCP server\n *\n * @param transportConfig Transport config\n * @param clientConfig Client config\n * @param capabilities Client capabilities (i.e. if the client supports roots/sampling)\n */\n async connect(\n url: string,\n options: {\n // Allows you to reconnect to a server (in the case of a auth reconnect)\n // Doesn't handle session reconnection\n reconnect?: {\n id: string;\n oauthClientId?: string;\n oauthCode?: string;\n };\n // we're overriding authProvider here because we want to be able to access the auth URL\n transport?: SSEClientTransportOptions & {\n authProvider?: AgentsOAuthProvider;\n };\n client?: ConstructorParameters<typeof Client>[1];\n capabilities?: ClientCapabilities;\n } = {}\n ): Promise<{ id: string; authUrl: string | undefined }> {\n const id = options.reconnect?.id ?? nanoid(8);\n\n if (!options.transport?.authProvider) {\n console.warn(\n \"No authProvider provided in the transport options. This client will only support unauthenticated remote MCP Servers\"\n );\n } else {\n options.transport.authProvider.serverId = id;\n }\n\n this.mcpConnections[id] = new MCPClientConnection(\n new URL(url),\n {\n name: this.name,\n version: this.version,\n },\n {\n transport: options.transport ?? {},\n client: options.client ?? {},\n capabilities: options.client ?? {},\n }\n );\n\n await this.mcpConnections[id].init(\n options.reconnect?.oauthCode,\n options.reconnect?.oauthClientId\n );\n\n const authUrl = options.transport?.authProvider?.authUrl;\n if (authUrl && options.transport?.authProvider?.redirectUrl) {\n this.callbackUrls.push(\n options.transport.authProvider.redirectUrl.toString()\n );\n }\n\n return {\n id,\n authUrl,\n };\n }\n\n isCallbackRequest(req: Request): boolean {\n return (\n req.method === \"GET\" &&\n !!this.callbackUrls.find((url) => {\n return req.url.startsWith(url);\n })\n );\n }\n\n async handleCallbackRequest(req: Request) {\n const url = new URL(req.url);\n const urlMatch = this.callbackUrls.find((url) => {\n return req.url.startsWith(url);\n });\n if (!urlMatch) {\n throw new Error(\n `No callback URI match found for the request url: ${req.url}. Was the request matched with \\`isCallbackRequest()\\`?`\n );\n }\n const code = url.searchParams.get(\"code\");\n const clientId = url.searchParams.get(\"state\");\n const urlParams = urlMatch.split(\"/\");\n const serverId = urlParams[urlParams.length - 1];\n if (!code) {\n throw new Error(\"Unauthorized: no code provided\");\n }\n if (!clientId) {\n throw new Error(\"Unauthorized: no state provided\");\n }\n\n if (this.mcpConnections[serverId] === undefined) {\n throw new Error(`Could not find serverId: ${serverId}`);\n }\n\n if (this.mcpConnections[serverId].connectionState !== \"authenticating\") {\n throw new Error(\n \"Failed to authenticate: the client isn't in the `authenticating` state\"\n );\n }\n\n const conn = this.mcpConnections[serverId];\n if (!conn.options.transport.authProvider) {\n throw new Error(\n \"Trying to finalize authentication for a server connection without an authProvider\"\n );\n }\n\n conn.options.transport.authProvider.clientId = clientId;\n conn.options.transport.authProvider.serverId = serverId;\n\n // reconnect to server with authorization\n const serverUrl = conn.url.toString();\n await this.connect(serverUrl, {\n reconnect: {\n id: serverId,\n oauthClientId: clientId,\n oauthCode: code,\n },\n ...conn.options,\n });\n\n if (this.mcpConnections[serverId].connectionState === \"authenticating\") {\n throw new Error(\"Failed to authenticate: client failed to initialize\");\n }\n\n return { serverId };\n }\n\n /**\n * @returns namespaced list of tools\n */\n listTools(): NamespacedData[\"tools\"] {\n return getNamespacedData(this.mcpConnections, \"tools\");\n }\n\n /**\n * @returns a set of tools that you can use with the AI SDK\n */\n unstable_getAITools(): ToolSet {\n return Object.fromEntries(\n getNamespacedData(this.mcpConnections, \"tools\").map((tool) => {\n return [\n `${tool.serverId}_${tool.name}`,\n {\n parameters: jsonSchema(tool.inputSchema),\n description: tool.description,\n execute: async (args) => {\n const result = await this.callTool({\n name: tool.name,\n arguments: args,\n serverId: tool.serverId,\n });\n if (result.isError) {\n // @ts-expect-error TODO we should fix this\n throw new Error(result.content[0].text);\n }\n return result;\n },\n },\n ];\n })\n );\n }\n\n /**\n * Closes all connections to MCP servers\n */\n async closeAllConnections() {\n return Promise.all(\n Object.values(this.mcpConnections).map(async (connection) => {\n await connection.client.close();\n })\n );\n }\n\n /**\n * Closes a connection to an MCP server\n * @param id The id of the connection to close\n */\n async closeConnection(id: string) {\n if (!this.mcpConnections[id]) {\n throw new Error(`Connection with id \"${id}\" does not exist.`);\n }\n await this.mcpConnections[id].client.close();\n }\n\n /**\n * @returns namespaced list of prompts\n */\n listPrompts(): NamespacedData[\"prompts\"] {\n return getNamespacedData(this.mcpConnections, \"prompts\");\n }\n\n /**\n * @returns namespaced list of tools\n */\n listResources(): NamespacedData[\"resources\"] {\n return getNamespacedData(this.mcpConnections, \"resources\");\n }\n\n /**\n * @returns namespaced list of resource templates\n */\n listResourceTemplates(): NamespacedData[\"resourceTemplates\"] {\n return getNamespacedData(this.mcpConnections, \"resourceTemplates\");\n }\n\n /**\n * Namespaced version of callTool\n */\n callTool(\n params: CallToolRequest[\"params\"] & { serverId: string },\n resultSchema?:\n | typeof CallToolResultSchema\n | typeof CompatibilityCallToolResultSchema,\n options?: RequestOptions\n ) {\n const unqualifiedName = params.name.replace(`${params.serverId}.`, \"\");\n return this.mcpConnections[params.serverId].client.callTool(\n {\n ...params,\n name: unqualifiedName,\n },\n resultSchema,\n options\n );\n }\n\n /**\n * Namespaced version of readResource\n */\n readResource(\n params: ReadResourceRequest[\"params\"] & { serverId: string },\n options: RequestOptions\n ) {\n return this.mcpConnections[params.serverId].client.readResource(\n params,\n options\n );\n }\n\n /**\n * Namespaced version of getPrompt\n */\n getPrompt(\n params: GetPromptRequest[\"params\"] & { serverId: string },\n options: RequestOptions\n ) {\n return this.mcpConnections[params.serverId].client.getPrompt(\n params,\n options\n );\n }\n}\n\ntype NamespacedData = {\n tools: (Tool & { serverId: string })[];\n prompts: (Prompt & { serverId: string })[];\n resources: (Resource & { serverId: string })[];\n resourceTemplates: (ResourceTemplate & { serverId: string })[];\n};\n\nexport function getNamespacedData<T extends keyof NamespacedData>(\n mcpClients: Record<string, MCPClientConnection>,\n type: T\n): NamespacedData[T] {\n const sets = Object.entries(mcpClients).map(([name, conn]) => {\n return { name, data: conn[type] };\n });\n\n const namespacedData = sets.flatMap(({ name: serverId, data }) => {\n return data.map((item) => {\n return {\n ...item,\n // we add a serverId so we can easily pull it out and send the tool call to the right server\n serverId,\n };\n });\n });\n\n return namespacedData as NamespacedData[T]; // Type assertion needed due to TS limitations with conditional return types\n}\n"],"mappings":";AAAA;AAAA,EACE;AAAA,OAEK;AAGA,IAAM,yBAAN,cAAqC,mBAAmB;AAAA;AAAA;AAAA;AAAA,EAK7D,YAAY,KAAU,SAAoC;AACxD,UAAM,gBAA8B,OAClC,UACA,YAAyB,CAAC,MACvB;AAEH,YAAM,UAAU,MAAM,KAAK,YAAY;AACvC,YAAM,gBAAgB;AAAA,QACpB,GAAG;AAAA,QACH,SAAS;AAAA,UACP,GAAG,QAAQ,aAAa;AAAA,UACxB,GAAG,WAAW;AAAA,UACd,GAAG;AAAA,QACL;AAAA,MACF;AAIA,aAAO,cAAc;AAGrB,aACG,QAAQ,iBAAiB;AAAA,QACxB;AAAA;AAAA,QAEA;AAAA,MACF,KAA2B,MAAM,UAAU,aAAa;AAAA,IAE5D;AAEA,UAAM,KAAK;AAAA,MACT,GAAG;AAAA,MACH,iBAAiB;AAAA,QACf,GAAG,QAAQ;AAAA,QACX,OAAO;AAAA,MACT;AAAA,IACF,CAAC;AACD,SAAK,eAAe,QAAQ;AAAA,EAC9B;AAAA,EAEA,MAAM,cAAc;AAClB,QAAI,KAAK,cAAc;AACrB,YAAM,SAAS,MAAM,KAAK,aAAa,OAAO;AAC9C,UAAI,QAAQ;AACV,eAAO;AAAA,UACL,eAAe,UAAU,OAAO,YAAY;AAAA,QAC9C;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;;;AC3DA;AAAA,EACE;AAAA,EAKA;AAAA,EACA;AAAA,OAOK;AACP,SAAS,cAAc;AAIhB,IAAM,sBAAN,MAA0B;AAAA,EAe/B,YACS,KACP,MACO,UAMH,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAC,GAAG,cAAc,CAAC,EAAE,GAClD;AATO;AAEA;AAhBT,2BAKe;AAEf,iBAAgB,CAAC;AACjB,mBAAoB,CAAC;AACrB,qBAAwB,CAAC;AACzB,6BAAwC,CAAC;AAcvC,SAAK,SAAS,IAAI,OAAO,MAAM,QAAQ,MAAM;AAC7C,SAAK,OAAO,qBAAqB,QAAQ,YAAY;AAAA,EACvD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,KAAK,MAAe,UAAmB;AAC3C,QAAI;AACF,YAAM,YAAY,IAAI;AAAA,QACpB,KAAK;AAAA,QACL,KAAK,QAAQ;AAAA,MACf;AACA,UAAI,MAAM;AACR,cAAM,UAAU,WAAW,IAAI;AAAA,MACjC;AAEA,YAAM,KAAK,OAAO,QAAQ,SAAS;AAAA,IAErC,SAAS,GAAQ;AACf,UAAI,EAAE,SAAS,EAAE,SAAS,cAAc,GAAG;AAEzC,aAAK,kBAAkB;AACvB;AAAA,MACF;AACA,WAAK,kBAAkB;AACvB,YAAM;AAAA,IACR;AAEA,SAAK,kBAAkB;AAEvB,SAAK,qBAAqB,MAAM,KAAK,OAAO,sBAAsB;AAClE,QAAI,CAAC,KAAK,oBAAoB;AAC5B,YAAM,IAAI,MAAM,qDAAqD;AAAA,IACvE;AAEA,UAAM,CAAC,cAAc,OAAO,WAAW,SAAS,iBAAiB,IAC/D,MAAM,QAAQ,IAAI;AAAA,MAChB,KAAK,OAAO,gBAAgB;AAAA,MAC5B,KAAK,cAAc;AAAA,MACnB,KAAK,kBAAkB;AAAA,MACvB,KAAK,gBAAgB;AAAA,MACrB,KAAK,0BAA0B;AAAA,IACjC,CAAC;AAEH,SAAK,eAAe;AACpB,SAAK,QAAQ;AACb,SAAK,YAAY;AACjB,SAAK,UAAU;AACf,SAAK,oBAAoB;AAEzB,SAAK,kBAAkB;AAAA,EACzB;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,gBAAiC;AACrC,QAAI,CAAC,KAAK,sBAAsB,CAAC,KAAK,mBAAmB,OAAO;AAC9D,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,KAAK,mBAAmB,MAAM,aAAa;AAC7C,WAAK,OAAO;AAAA,QACV;AAAA,QACA,OAAO,kBAAkB;AACvB,eAAK,QAAQ,MAAM,KAAK,WAAW;AAAA,QACrC;AAAA,MACF;AAAA,IACF;AAEA,WAAO,KAAK,WAAW;AAAA,EACzB;AAAA,EAEA,MAAM,oBAAyC;AAC7C,QAAI,CAAC,KAAK,sBAAsB,CAAC,KAAK,mBAAmB,WAAW;AAClE,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,KAAK,mBAAmB,UAAU,aAAa;AACjD,WAAK,OAAO;AAAA,QACV;AAAA,QACA,OAAO,kBAAkB;AACvB,eAAK,YAAY,MAAM,KAAK,eAAe;AAAA,QAC7C;AAAA,MACF;AAAA,IACF;AAEA,WAAO,KAAK,eAAe;AAAA,EAC7B;AAAA,EAEA,MAAM,kBAAqC;AACzC,QAAI,CAAC,KAAK,sBAAsB,CAAC,KAAK,mBAAmB,SAAS;AAChE,aAAO,CAAC;AAAA,IACV;AAEA,QAAI,KAAK,mBAAmB,QAAQ,aAAa;AAC/C,WAAK,OAAO;AAAA,QACV;AAAA,QACA,OAAO,kBAAkB;AACvB,eAAK,UAAU,MAAM,KAAK,aAAa;AAAA,QACzC;AAAA,MACF;AAAA,IACF;AAEA,WAAO,KAAK,aAAa;AAAA,EAC3B;AAAA,EAEA,MAAM,4BAAyD;AAC7D,QAAI,CAAC,KAAK,sBAAsB,CAAC,KAAK,mBAAmB,WAAW;AAClE,aAAO,CAAC;AAAA,IACV;AAEA,WAAO,KAAK,uBAAuB;AAAA,EACrC;AAAA,EAEA,MAAM,aAAa;AACjB,QAAI,WAAmB,CAAC;AACxB,QAAI,cAA+B,EAAE,OAAO,CAAC,EAAE;AAC/C,OAAG;AACD,oBAAc,MAAM,KAAK,OACtB,UAAU;AAAA,QACT,QAAQ,YAAY;AAAA,MACtB,CAAC,EACA,MAAM,uBAAuB,EAAE,OAAO,CAAC,EAAE,GAAG,YAAY,CAAC;AAC5D,iBAAW,SAAS,OAAO,YAAY,KAAK;AAAA,IAC9C,SAAS,YAAY;AACrB,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,iBAAiB;AACrB,QAAI,eAA2B,CAAC;AAChC,QAAI,kBAAuC,EAAE,WAAW,CAAC,EAAE;AAC3D,OAAG;AACD,wBAAkB,MAAM,KAAK,OAC1B,cAAc;AAAA,QACb,QAAQ,gBAAgB;AAAA,MAC1B,CAAC,EACA,MAAM,uBAAuB,EAAE,WAAW,CAAC,EAAE,GAAG,gBAAgB,CAAC;AACpE,qBAAe,aAAa,OAAO,gBAAgB,SAAS;AAAA,IAC9D,SAAS,gBAAgB;AACzB,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,eAAe;AACnB,QAAI,aAAuB,CAAC;AAC5B,QAAI,gBAAmC,EAAE,SAAS,CAAC,EAAE;AACrD,OAAG;AACD,sBAAgB,MAAM,KAAK,OACxB,YAAY;AAAA,QACX,QAAQ,cAAc;AAAA,MACxB,CAAC,EACA,MAAM,uBAAuB,EAAE,SAAS,CAAC,EAAE,GAAG,cAAc,CAAC;AAChE,mBAAa,WAAW,OAAO,cAAc,OAAO;AAAA,IACtD,SAAS,cAAc;AACvB,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,yBAAyB;AAC7B,QAAI,eAAmC,CAAC;AACxC,QAAI,kBAA+C;AAAA,MACjD,mBAAmB,CAAC;AAAA,IACtB;AACA,OAAG;AACD,wBAAkB,MAAM,KAAK,OAC1B,sBAAsB;AAAA,QACrB,QAAQ,gBAAgB;AAAA,MAC1B,CAAC,EACA;AAAA,QACC;AAAA,UACE,EAAE,mBAAmB,CAAC,EAAE;AAAA,UACxB;AAAA,QACF;AAAA,MACF;AACF,qBAAe,aAAa,OAAO,gBAAgB,iBAAiB;AAAA,IACtE,SAAS,gBAAgB;AACzB,WAAO;AAAA,EACT;AACF;AAEA,SAAS,uBAA0B,OAAU,QAAgB;AAC3D,SAAO,CAAC,MAAwB;AAE9B,QAAI,EAAE,SAAS,QAAQ;AACrB,cAAQ;AAAA,QACN,oDAAoD,OAAO,MAAM,GAAG,EAAE,CAAC,CAAC,0CAA0C,MAAM;AAAA,MAC1H;AACA,aAAO;AAAA,IACT;AACA,UAAM;AAAA,EACR;AACF;;;AC/NA,SAAS,kBAAgC;AACzC,SAAS,cAAc;AAKhB,IAAM,mBAAN,MAAuB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAS5B,YACU,MACA,SACR;AAFQ;AACA;AAVV,SAAO,iBAAsD,CAAC;AAC9D,SAAQ,eAAyB,CAAC;AAAA,EAU/B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASH,MAAM,QACJ,KACA,UAcI,CAAC,GACiD;AACtD,UAAM,KAAK,QAAQ,WAAW,MAAM,OAAO,CAAC;AAE5C,QAAI,CAAC,QAAQ,WAAW,cAAc;AACpC,cAAQ;AAAA,QACN;AAAA,MACF;AAAA,IACF,OAAO;AACL,cAAQ,UAAU,aAAa,WAAW;AAAA,IAC5C;AAEA,SAAK,eAAe,EAAE,IAAI,IAAI;AAAA,MAC5B,IAAI,IAAI,GAAG;AAAA,MACX;AAAA,QACE,MAAM,KAAK;AAAA,QACX,SAAS,KAAK;AAAA,MAChB;AAAA,MACA;AAAA,QACE,WAAW,QAAQ,aAAa,CAAC;AAAA,QACjC,QAAQ,QAAQ,UAAU,CAAC;AAAA,QAC3B,cAAc,QAAQ,UAAU,CAAC;AAAA,MACnC;AAAA,IACF;AAEA,UAAM,KAAK,eAAe,EAAE,EAAE;AAAA,MAC5B,QAAQ,WAAW;AAAA,MACnB,QAAQ,WAAW;AAAA,IACrB;AAEA,UAAM,UAAU,QAAQ,WAAW,cAAc;AACjD,QAAI,WAAW,QAAQ,WAAW,cAAc,aAAa;AAC3D,WAAK,aAAa;AAAA,QAChB,QAAQ,UAAU,aAAa,YAAY,SAAS;AAAA,MACtD;AAAA,IACF;AAEA,WAAO;AAAA,MACL;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,kBAAkB,KAAuB;AACvC,WACE,IAAI,WAAW,SACf,CAAC,CAAC,KAAK,aAAa,KAAK,CAAC,QAAQ;AAChC,aAAO,IAAI,IAAI,WAAW,GAAG;AAAA,IAC/B,CAAC;AAAA,EAEL;AAAA,EAEA,MAAM,sBAAsB,KAAc;AACxC,UAAM,MAAM,IAAI,IAAI,IAAI,GAAG;AAC3B,UAAM,WAAW,KAAK,aAAa,KAAK,CAACA,SAAQ;AAC/C,aAAO,IAAI,IAAI,WAAWA,IAAG;AAAA,IAC/B,CAAC;AACD,QAAI,CAAC,UAAU;AACb,YAAM,IAAI;AAAA,QACR,oDAAoD,IAAI,GAAG;AAAA,MAC7D;AAAA,IACF;AACA,UAAM,OAAO,IAAI,aAAa,IAAI,MAAM;AACxC,UAAM,WAAW,IAAI,aAAa,IAAI,OAAO;AAC7C,UAAM,YAAY,SAAS,MAAM,GAAG;AACpC,UAAM,WAAW,UAAU,UAAU,SAAS,CAAC;AAC/C,QAAI,CAAC,MAAM;AACT,YAAM,IAAI,MAAM,gCAAgC;AAAA,IAClD;AACA,QAAI,CAAC,UAAU;AACb,YAAM,IAAI,MAAM,iCAAiC;AAAA,IACnD;AAEA,QAAI,KAAK,eAAe,QAAQ,MAAM,QAAW;AAC/C,YAAM,IAAI,MAAM,4BAA4B,QAAQ,EAAE;AAAA,IACxD;AAEA,QAAI,KAAK,eAAe,QAAQ,EAAE,oBAAoB,kBAAkB;AACtE,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,UAAM,OAAO,KAAK,eAAe,QAAQ;AACzC,QAAI,CAAC,KAAK,QAAQ,UAAU,cAAc;AACxC,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,SAAK,QAAQ,UAAU,aAAa,WAAW;AAC/C,SAAK,QAAQ,UAAU,aAAa,WAAW;AAG/C,UAAM,YAAY,KAAK,IAAI,SAAS;AACpC,UAAM,KAAK,QAAQ,WAAW;AAAA,MAC5B,WAAW;AAAA,QACT,IAAI;AAAA,QACJ,eAAe;AAAA,QACf,WAAW;AAAA,MACb;AAAA,MACA,GAAG,KAAK;AAAA,IACV,CAAC;AAED,QAAI,KAAK,eAAe,QAAQ,EAAE,oBAAoB,kBAAkB;AACtE,YAAM,IAAI,MAAM,qDAAqD;AAAA,IACvE;AAEA,WAAO,EAAE,SAAS;AAAA,EACpB;AAAA;AAAA;AAAA;AAAA,EAKA,YAAqC;AACnC,WAAO,kBAAkB,KAAK,gBAAgB,OAAO;AAAA,EACvD;AAAA;AAAA;AAAA;AAAA,EAKA,sBAA+B;AAC7B,WAAO,OAAO;AAAA,MACZ,kBAAkB,KAAK,gBAAgB,OAAO,EAAE,IAAI,CAAC,SAAS;AAC5D,eAAO;AAAA,UACL,GAAG,KAAK,QAAQ,IAAI,KAAK,IAAI;AAAA,UAC7B;AAAA,YACE,YAAY,WAAW,KAAK,WAAW;AAAA,YACvC,aAAa,KAAK;AAAA,YAClB,SAAS,OAAO,SAAS;AACvB,oBAAM,SAAS,MAAM,KAAK,SAAS;AAAA,gBACjC,MAAM,KAAK;AAAA,gBACX,WAAW;AAAA,gBACX,UAAU,KAAK;AAAA,cACjB,CAAC;AACD,kBAAI,OAAO,SAAS;AAElB,sBAAM,IAAI,MAAM,OAAO,QAAQ,CAAC,EAAE,IAAI;AAAA,cACxC;AACA,qBAAO;AAAA,YACT;AAAA,UACF;AAAA,QACF;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,sBAAsB;AAC1B,WAAO,QAAQ;AAAA,MACb,OAAO,OAAO,KAAK,cAAc,EAAE,IAAI,OAAO,eAAe;AAC3D,cAAM,WAAW,OAAO,MAAM;AAAA,MAChC,CAAC;AAAA,IACH;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,gBAAgB,IAAY;AAChC,QAAI,CAAC,KAAK,eAAe,EAAE,GAAG;AAC5B,YAAM,IAAI,MAAM,uBAAuB,EAAE,mBAAmB;AAAA,IAC9D;AACA,UAAM,KAAK,eAAe,EAAE,EAAE,OAAO,MAAM;AAAA,EAC7C;AAAA;AAAA;AAAA;AAAA,EAKA,cAAyC;AACvC,WAAO,kBAAkB,KAAK,gBAAgB,SAAS;AAAA,EACzD;AAAA;AAAA;AAAA;AAAA,EAKA,gBAA6C;AAC3C,WAAO,kBAAkB,KAAK,gBAAgB,WAAW;AAAA,EAC3D;AAAA;AAAA;AAAA;AAAA,EAKA,wBAA6D;AAC3D,WAAO,kBAAkB,KAAK,gBAAgB,mBAAmB;AAAA,EACnE;AAAA;AAAA;AAAA;AAAA,EAKA,SACE,QACA,cAGA,SACA;AACA,UAAM,kBAAkB,OAAO,KAAK,QAAQ,GAAG,OAAO,QAAQ,KAAK,EAAE;AACrE,WAAO,KAAK,eAAe,OAAO,QAAQ,EAAE,OAAO;AAAA,MACjD;AAAA,QACE,GAAG;AAAA,QACH,MAAM;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,aACE,QACA,SACA;AACA,WAAO,KAAK,eAAe,OAAO,QAAQ,EAAE,OAAO;AAAA,MACjD;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,UACE,QACA,SACA;AACA,WAAO,KAAK,eAAe,OAAO,QAAQ,EAAE,OAAO;AAAA,MACjD;AAAA,MACA;AAAA,IACF;AAAA,EACF;AACF;AASO,SAAS,kBACd,YACA,MACmB;AACnB,QAAM,OAAO,OAAO,QAAQ,UAAU,EAAE,IAAI,CAAC,CAAC,MAAM,IAAI,MAAM;AAC5D,WAAO,EAAE,MAAM,MAAM,KAAK,IAAI,EAAE;AAAA,EAClC,CAAC;AAED,QAAM,iBAAiB,KAAK,QAAQ,CAAC,EAAE,MAAM,UAAU,KAAK,MAAM;AAChE,WAAO,KAAK,IAAI,CAAC,SAAS;AACxB,aAAO;AAAA,QACL,GAAG;AAAA;AAAA,QAEH;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AAED,SAAO;AACT;","names":["url"]}