add X-Sequence-Token-Key header support in http fetch clients (#432)

0xsequence · Oct 13, 2023 · 8df5dc6 · 8df5dc6
1 parent 1ca333d
commit 8df5dc6
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 8 deletions.
diff --git a/packages/api/src/index.ts b/packages/api/src/index.ts
@@ -7,7 +7,10 @@ const fetch = typeof global === 'object' ? global.fetch : window.fetch
 export class SequenceAPIClient extends ApiRpc {
   constructor(
     hostname: string,
-    public jwtAuth?: string
+    public authorization?: {
+      jwtAuth?: string
+      tokenKey?: string
+    }
   ) {
     super(hostname.endsWith('/') ? hostname.slice(0, -1) : hostname, fetch)
     this.fetch = this._fetch
@@ -17,8 +20,15 @@ export class SequenceAPIClient extends ApiRpc {
     // automatically include jwt auth header to requests
     // if its been set on the api client
     const headers: { [key: string]: any } = {}
-    if (this.jwtAuth && this.jwtAuth.length > 0) {
-      headers['Authorization'] = `BEARER ${this.jwtAuth}`
+
+    const {jwtAuth, tokenKey} = this.authorization || {}
+
+    if (jwtAuth && jwtAuth.length > 0) {
+      headers['Authorization'] = `BEARER ${jwtAuth}`
+    }
+
+    if (tokenKey && tokenKey.length > 0) {
+      headers['X-Sequence-Token-Key'] = `${tokenKey}`
     }
 
     // before the request is made

diff --git a/packages/auth/src/services.ts b/packages/auth/src/services.ts
@@ -114,7 +114,9 @@ export class Services {
         }
       }
 
-      return new SequenceAPIClient(url, jwtAuth)
+      return new SequenceAPIClient(url, {
+        jwtAuth
+      })
     })()
 
     return this._initialAuthRequest
@@ -221,7 +223,7 @@ export class Services {
       if (!url) throw Error('No sequence api url')
 
       const jwtAuth = (await this.getJWT(tryAuth)).token
-      this.apiClient = new SequenceAPIClient(url, jwtAuth)
+      this.apiClient = new SequenceAPIClient(url, { jwtAuth })
     }
 
     return this.apiClient

diff --git a/packages/indexer/src/index.ts b/packages/indexer/src/index.ts
@@ -29,7 +29,10 @@ const fetch = typeof global === 'object' ? global.fetch : window.fetch
 export class SequenceIndexerClient extends IndexerRpc {
   constructor(
     hostname: string,
-    public jwtAuth?: string
+    public authorization?: {
+      jwtAuth?: string
+      tokenKey?: string
+    }
   ) {
     super(hostname.endsWith('/') ? hostname.slice(0, -1) : hostname, fetch)
     this.fetch = this._fetch
@@ -39,8 +42,15 @@ export class SequenceIndexerClient extends IndexerRpc {
     // automatically include jwt auth header to requests
     // if its been set on the api client
     const headers: { [key: string]: any } = {}
-    if (this.jwtAuth && this.jwtAuth.length > 0) {
-      headers['Authorization'] = `BEARER ${this.jwtAuth}`
+
+    const {jwtAuth, tokenKey} = this.authorization || {}
+
+    if (jwtAuth && jwtAuth.length > 0) {
+      headers['Authorization'] = `BEARER ${jwtAuth}`
+    }
+
+    if (tokenKey && tokenKey.length > 0) {
+      headers['X-Sequence-Token-Key'] = `${tokenKey}`
     }
 
     // before the request is made