feat: addition of cookies and basic auth - 9

Author: parthlambdatest

src/lib/ctx.ts

@@ -1,4 +1,4 @@
-import { Context, Env, WebConfig, MobileConfig } from '../types.js'
+import { Context, Env, WebConfig, MobileConfig, basicAuth } from '../types.js'
 import constants from './constants.js'
 import { version } from '../../package.json'
 import { validateConfig } from './schemaValidation.js'
@@ -11,6 +11,7 @@ export default (options: Record<string, string>): Context => {
     let env: Env = getEnv();
     let webConfig: WebConfig;
     let mobileConfig: MobileConfig;
+    let basicAuthObj: basicAuth
     let config = constants.DEFAULT_CONFIG;
     let port: number;
     let resolutionOff: boolean;
@@ -57,6 +58,12 @@ export default (options: Record<string, string>): Context => {
         }
     }
 
+    console.dir(basicAuthObj)
+    console.log("**************")
+    if (config.basicAuthorization){
+        basicAuthObj = config.basicAuthorization
+    }
+
     return {
         env: env,
         log: logger,
@@ -70,7 +77,7 @@ export default (options: Record<string, string>): Context => {
             cliEnableJavaScript: config.cliEnableJavaScript || true,
             scrollTime: config.scrollTime || constants.DEFAULT_SCROLL_TIME,
             allowedHostnames: config.allowedHostnames || [],
-            basicAuthorization: config.basicAuthorization || {username: '', password:''}
+            basicAuthorization: basicAuthObj
         },
         uploadFilePath: '',
         webStaticConfig: [],

src/lib/processSnapshot.ts

@@ -130,8 +130,8 @@ async function processSnapshot(snapshot: Snapshot, ctx: Context): Promise<Record
             let requestOptions: Record<string, any> = {
                 timeout: REQUEST_TIMEOUT
             }
-            if (requestUrl === snapshot.url && ctx.config.basicAuthorization.username !== '' ) {
-                ctx.log.debug(`Adding basic authorization to the headers for root url`);
+            if (requestUrl === snapshot.url && ctx.config.basicAuthorization ) {
+                ctx.log.debug(`Adding basic authorization to the headers for root url with username: ${ctx.config.basicAuthorization.username}`);
                 let token = Buffer.from(`${ctx.config.basicAuthorization.username}:${ctx.config.basicAuthorization.password}`).toString('base64');
                 requestOptions.headers = {
                     ...request.headers(),

src/types.ts

@@ -23,7 +23,7 @@ export interface Context {
         cliEnableJavaScript: boolean;
         scrollTime: number;
         allowedHostnames: Array<string>;
-        basicAuthorization: {username: string, password: string}
+        basicAuthorization: basicAuth | undefined
     };
     uploadFilePath: string;
     webStaticConfig: WebStaticConfig;
@@ -142,3 +142,8 @@ export type FigmaDesignConfig = {
     depth: number;
     figma_config: FigmaConfigItem[];
 };
+
+export interface basicAuth {
+    username: string;
+    password: string;
+}