fetch results support and rebased code

Author: parthlambdatest

src/lib/ctx.ts

@@ -132,7 +132,7 @@ export default (options: Record<string, string>): Context => {
         cliVersion: version,
         totalSnapshots: -1,
         isStartExec: false,
-        isSnapshotCaptured: false
+        isSnapshotCaptured: false,
         sessionCapabilitiesMap: new Map<string, any[]>(),
         sessionToBuildMap: new Map<string, string>(),
         buildToSnapshotCountMap: new Map<string, number>(),

src/lib/httpClient.ts

@@ -255,6 +255,28 @@ export default class httpClient {
             }
         }, ctx.log)
     }
+
+    processSnapshotCaps(ctx: Context, snapshot: ProcessedSnapshot, snapshotUuid: string, capsBuildId: string, capsProjectToken: string) {
+        return this.request({
+            url: `/build/${capsBuildId}/snapshot`,
+            method: 'POST',
+            headers: { 
+                'Content-Type': 'application/json',
+                projectToken: capsProjectToken !== '' ? capsProjectToken : this.projectToken
+            },
+            data: {
+                name: snapshot.name,
+                url: snapshot.url,
+                snapshotUuid: snapshotUuid,
+                test: {
+                    type: ctx.testType,
+                    source: 'cli'
+                },
+                async: false,
+            }
+        }, ctx.log)
+    }
+
     uploadSnapshotForCaps(ctx: Context, snapshot: ProcessedSnapshot, capsBuildId: string, capsProjectToken: string) {
         // Use capsBuildId if provided, otherwise fallback to ctx.build.id
         const buildId = capsBuildId !== '' ? capsBuildId : ctx.build.id;
@@ -371,6 +393,22 @@ export default class httpClient {
         }, ctx.log)
     }
 
+    getS3PresignedURLForSnapshotUploadCaps(ctx: Context, snapshotName: string, snapshotUuid: string, capsBuildId: string, capsProjectToken: string) {
+        return this.request({
+            url: `/snapshotuploadurl`,
+            method: 'POST',
+            headers: { 
+                'Content-Type': 'application/json',
+                projectToken: capsProjectToken !== '' ? capsProjectToken : this.projectToken
+            },
+            data: {
+                buildId: capsBuildId,
+                snapshotName: snapshotName,
+                snapshotUuid: snapshotUuid
+            }
+        }, ctx.log)
+    }
+
     uploadLogs(ctx: Context, uploadURL: string) {
         const fileStream = fs.createReadStream(constants.LOG_FILE_PATH);
         const { size } = fs.statSync(constants.LOG_FILE_PATH);
@@ -401,6 +439,20 @@ export default class httpClient {
         }, ctx.log)
     }
 
+    uploadSnapshotToS3Caps(ctx: Context, uploadURL: string, snapshot: Snapshot, capsProjectToken: string) {
+        return this.request({
+            url: uploadURL,
+            method: 'PUT',
+            headers: {
+                'Content-Type': 'application/json',
+                projectToken: capsProjectToken !== '' ? capsProjectToken : this.projectToken
+            },
+            data: snapshot,
+            maxBodyLength: Infinity, // prevent axios from limiting the body size
+            maxContentLength: Infinity, // prevent axios from limiting the content size
+        }, ctx.log)
+    }
+
     processWebFigma(requestBody: any, log: Logger) {
         return this.request({
             url: "figma-web/upload",

src/lib/schemaValidation.ts

@@ -168,6 +168,7 @@ const ConfigSchema = {
         useGlobalCache: {
             type: "boolean",
             errorMessage: "Invalid config; useGlobalCache must be true/false"
+        },
         skipBuildCreation: {
             type: "boolean",
             errorMessage: "Invalid config; skipBuildCreation must be true/false"

src/lib/server.ts

@@ -72,6 +72,7 @@ export default async (ctx: Context): Promise<FastifyInstance<Server, IncomingMes
 								startPolling(ctx, capsBuildId, is_baseline, fetchedCapabilitiesResp.projectToken)
 							}
 						}
+						console.log(`fetch caps for sessionId: ${sessionId} are ${JSON.stringify(fetchedCapabilitiesResp)}`)
 					} catch (error: any) {
 						// console.log(`Failed to fetch capabilities for sessionId ${sessionId}: ${error.message}`);
 					}