[ts-sdk] Add Authorization header to manager options (#1618)

Author: wkozyra95

ts/smelter-core/src/api.ts

@@ -10,12 +10,14 @@ export type ApiRequest = {
   method: 'GET' | 'POST';
   route: string;
   body?: object;
+  headers?: Record<string, string>;
 };
 
 export type MultipartRequest = {
   method: 'POST';
   route: string;
   body: any; //FormData
+  headers?: Record<string, string>;
 };
 
 export type RegisterInputResponse = {

ts/smelter-node/src/fetch.ts

@@ -18,6 +18,7 @@ export async function sendRequest(baseUrl: string | URL, request: ApiRequest): P
     method: request.method,
     body: request.body && JSON.stringify(request.body),
     headers: {
+      ...request.headers,
       'Content-Type': 'application/json',
     },
     agent: url => (url.protocol === 'http:' ? httpAgent : httpsAgent),
@@ -39,6 +40,7 @@ export async function sendMultipartRequest(
     method: request.method,
     body: request.body as FormData,
     agent: url => (url.protocol === 'http:' ? httpAgent : httpsAgent),
+    headers: request.headers,
   });
   if (response.status >= 400) {
     const err: any = new Error(`Request to Smelter server failed.`);

ts/smelter-node/src/manager/existingInstance.ts

@@ -12,13 +12,15 @@ import { getSmelterStatus } from '../getSmelterStatus';
 
 export type ExistingInstanceOptions = {
   url: string | URL;
+  authorizationHeader?: string;
 };
 
 /**
  * SmelterManager that will connect to existing instance
  */
 class ExistingInstanceManager implements SmelterManager {
   private url: URL;
+  private authorizationHeader?: string;
   private wsConnection: WebSocketConnection;
 
   constructor(opts: ExistingInstanceOptions) {
@@ -34,10 +36,11 @@ class ExistingInstanceManager implements SmelterManager {
     }
 
     this.url = url;
+    this.authorizationHeader = opts.authorizationHeader;
 
     const wsUrl = joinUrl(url, 'ws');
     wsUrl.protocol = url.protocol === 'https:' ? 'wss:' : 'ws:';
-    this.wsConnection = new WebSocketConnection(wsUrl);
+    this.wsConnection = new WebSocketConnection(wsUrl, opts.authorizationHeader);
   }
 
   public async setupInstance(opts: SetupInstanceOptions): Promise<void> {
@@ -80,11 +83,19 @@ class ExistingInstanceManager implements SmelterManager {
   }
 
   public async sendRequest(request: ApiRequest): Promise<object> {
-    return await sendRequest(this.url, request);
+    let headers = {
+      ...request.headers,
+      ...(this.authorizationHeader ? { Authorization: this.authorizationHeader } : {}),
+    };
+    return await sendRequest(this.url, { ...request, headers });
   }
 
   async sendMultipartRequest(request: MultipartRequest): Promise<object> {
-    return await sendMultipartRequest(this.url, request);
+    let headers = {
+      ...request.headers,
+      ...(this.authorizationHeader ? { Authorization: this.authorizationHeader } : {}),
+    };
+    return await sendMultipartRequest(this.url, { ...request, headers });
   }
 
   public registerEventListener(cb: (event: object) => void): void {

ts/smelter-node/src/ws.ts

@@ -3,17 +3,21 @@ import WebSocket from 'ws';
 
 export class WebSocketConnection {
   private url: string | URL;
+  private authorizationHeader?: string;
   private listeners: Set<(event: object) => void>;
   private ws: WebSocket | null = null;
   private donePromise?: Promise<void>;
 
-  constructor(url: string | URL) {
+  constructor(url: string | URL, authorizationHeader?: string) {
     this.url = url;
+    this.authorizationHeader = authorizationHeader;
     this.listeners = new Set();
   }
 
   public async connect(logger: Logger): Promise<void> {
-    const ws = new WebSocket(this.url);
+    const ws = new WebSocket(this.url, {
+      headers: this.authorizationHeader ? { Authorization: this.authorizationHeader } : {},
+    });
 
     let connected = false;
     await new Promise<void>((res, rej) => {