From b4839dc676c8b89a3f51d9b7a30ea6066639deed Mon Sep 17 00:00:00 2001
From: parthlambdatest <parthkirsan@lambdatest.com>
Date: Thu, 19 Sep 2024 18:08:06 +0530
Subject: [PATCH] code clean up

---
 src/lib/ctx.ts             |  4 ++--
 src/lib/processSnapshot.ts | 17 +++++++++--------
 src/types.ts               |  2 +-
 3 files changed, 12 insertions(+), 11 deletions(-)

diff --git a/src/lib/ctx.ts b/src/lib/ctx.ts
index f790f3c..6fbc097 100644
--- a/src/lib/ctx.ts
+++ b/src/lib/ctx.ts
@@ -11,8 +11,8 @@ export default (options: Record<string, string>): Context => {
     let env: Env = getEnv();
     let webConfig: WebConfig;
     let mobileConfig: MobileConfig;
-    let basicAuthObj: basicAuth
-    let requestHeaderObj: authToken
+    let basicAuthObj: basicAuth;
+    let requestHeaderObj: authToken;
     let config = constants.DEFAULT_CONFIG;
     let port: number;
     let resolutionOff: boolean;
diff --git a/src/lib/processSnapshot.ts b/src/lib/processSnapshot.ts
index 1921ac1..09745a2 100644
--- a/src/lib/processSnapshot.ts
+++ b/src/lib/processSnapshot.ts
@@ -146,18 +146,19 @@ async function processSnapshot(snapshot: Snapshot, ctx: Context): Promise<Record
             // handle discovery config
             ctx.config.allowedHostnames.push(new URL(snapshot.url).hostname);
             if (ctx.config.enableJavaScript) ALLOWED_RESOURCES.push('script');
-            
+
             if (ctx.config.basicAuthorization || ctx.config.requestHeaders) {
-                ctx.log.debug(`Adding basic authorization to the headers for root url`);
-                let token;
-                if(ctx.config.basicAuthorization){
-                    token = Buffer.from(`${ctx.config.basicAuthorization.username}:${ctx.config.basicAuthorization.password}`).toString('base64');
-                } else {
-                    token = ctx.config.requestHeaders.auth
+                ctx.log.debug(`Adding basic authorization to the headers for url ${requestUrl}`);
+                let basicToken;
+                if (ctx.config.basicAuthorization){
+                    let token = Buffer.from(`${ctx.config.basicAuthorization.username}:${ctx.config.basicAuthorization.password}`).toString('base64');
+                    basicToken = `Basic ${token}`
+                } else if (ctx.config.requestHeaders) {
+                    basicToken = ctx.config.requestHeaders.auth
                 }
                 requestOptions.headers = {
                     ...request.headers(),
-                    Authorization: `Basic ${token}`
+                    Authorization: basicToken
                 };
             }
 
diff --git a/src/types.ts b/src/types.ts
index 544f2c0..46b3c24 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -24,7 +24,7 @@ export interface Context {
         scrollTime: number;
         allowedHostnames: Array<string>;
         basicAuthorization: basicAuth | undefined;
-        requestHeaders: authToken | undefined
+        requestHeaders: authToken | undefined;
 
     };
     uploadFilePath: string;