Add authentication_token

pinax-network · Feb 1, 2023 · 59e5eb5 · 59e5eb5
1 parent 1eadea5
commit 59e5eb5
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 4 deletions.
diff --git a/example.js b/example.js
@@ -10,7 +10,7 @@ const stopBlockNum = "+10";
 const substreams = new Substreams(outputModule, {
     startBlockNum,
     stopBlockNum,
-    authorization: process.env.SUBSTREAMS_API_TOKEN
+    authorization: "<STREAMINGFAST_KEY or SUBSTREAMS_API_TOKEN>"
 });
 
 (async () => {

diff --git a/examples/subtivity.ts b/examples/subtivity.ts
@@ -10,7 +10,7 @@ const stopBlockNum = "+10";
 const substreams = new Substreams(outputModule, {
     startBlockNum,
     stopBlockNum,
-    authorization: process.env.SUBSTREAMS_API_TOKEN
+    authorization: process.env.STREAMINGFAST_KEY
 });
 
 (async () => {

diff --git a/src/index.ts b/src/index.ts
@@ -20,7 +20,7 @@ export * from "./generated/sf/substreams/v1/substreams"
 export * from "./utils";
 
 // Utils
-import { parseBlockData, parseStopBlock } from './utils';
+import { parseAuthorization, parseBlockData, parseStopBlock } from './utils';
 
 interface MapOutput extends ModuleOutput {
     data: {
@@ -88,7 +88,9 @@ export class Substreams extends (EventEmitter as new () => TypedEmitter<MessageE
 
         // Credentials
         const metadata = new Metadata();
-        if ( options.authorization ) metadata.add('authorization', options.authorization);
+        if ( options.authorization ) parseAuthorization(options.authorization).then( token => {
+            metadata.add('authorization', token);
+        })
         const creds = credentials.combineChannelCredentials(
             credentials.createSsl(),
             credentials.createFromMetadataGenerator((_, callback) => callback(null, metadata)),

diff --git a/src/utils.ts b/src/utils.ts
@@ -29,6 +29,28 @@ export function getSeconds( clock?: Clock ) {
     return Number(clock?.timestamp?.seconds);
 }
 
+interface Token {
+    token: string;
+    expires_at: number;
+}
+
+export async function authentication_token(api_key: string) {
+    const url = 'https://auth.streamingfast.io/v1/auth/issue';
+    const response = await fetch(url, {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({api_key})
+    })
+    return response.json() as Promise<Token>;
+}
+
+export async function parseAuthorization(authorization: string ) {
+    if ( authorization.includes("server_") ) {
+        return (await authentication_token(authorization)).token;
+    }
+    return authorization;
+}
+
 export const isIpfs = ( str: string ) => /^Qm[1-9A-Za-z]{44}$/.test(str);
 
 export function parseStopBlock( startBlock: string, stopBlock?: string ) {