Allow instantiating an api Agent with a string or URL (bluesky-social…

…#2917) * Allow instantiating an api Agent with a string or URL * changeset * Update .changeset/stupid-spiders-buy.md Co-authored-by: Bigint <[email protected]> --------- Co-authored-by: Bigint <[email protected]>
rcg4u · Nov 7, 2024 · 839202a · 839202a
1 parent 1982693
commit 839202a
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 4 deletions.
diff --git a/.changeset/stupid-spiders-buy.md b/.changeset/stupid-spiders-buy.md
@@ -0,0 +1,5 @@
+---
+"@atproto/api": patch
+---
+
+Allow instantiating an API Agent with a string or URL
diff --git a/packages/api/src/agent.ts b/packages/api/src/agent.ts
@@ -1,6 +1,11 @@
 import { TID } from '@atproto/common-web'
 import { AtUri, ensureValidDid } from '@atproto/syntax'
-import { buildFetchHandler, FetchHandler, XrpcClient } from '@atproto/xrpc'
+import {
+  buildFetchHandler,
+  BuildFetchHandlerOptions,
+  FetchHandler,
+  XrpcClient,
+} from '@atproto/xrpc'
 import AwaitLock from 'await-lock'
 import {
   AppBskyActorDefs,
@@ -105,8 +110,16 @@ export class Agent extends XrpcClient {
     return this
   }
 
-  constructor(readonly sessionManager: SessionManager) {
-    const fetchHandler = buildFetchHandler(sessionManager)
+  readonly sessionManager: SessionManager
+
+  constructor(options: string | URL | SessionManager) {
+    const sessionManager: SessionManager =
+      typeof options === 'string' || options instanceof URL
+        ? {
+            did: undefined,
+            fetchHandler: buildFetchHandler(options),
+          }
+        : options
 
     super((url, init) => {
       const headers = new Headers(init?.headers)
@@ -128,8 +141,10 @@ export class Agent extends XrpcClient {
           .join(', '),
       )
 
-      return fetchHandler(url, { ...init, headers })
+      return this.sessionManager.fetchHandler(url, { ...init, headers })
     }, schemas)
+
+    this.sessionManager = sessionManager
   }
 
   //#region Cloning utilities