Merge pull request #26 from pinanks/DOT-2159

Add totalSnapshots in param in build finalize

Author: sushobhit-lt

src/commander/capture.ts

@@ -24,7 +24,7 @@ command
                 getGitInfo(ctx),
                 createBuild(ctx),
                 captureScreenshots(ctx),
-                finalizeBuild(ctx, 0)
+                finalizeBuild(ctx)
             ],
             {
                 rendererOptions: {

src/commander/exec.ts

@@ -24,6 +24,7 @@ command
             return
         }
         ctx.args.execCommand = execCommand
+        ctx.totalSnapshots = 0
 
         let tasks = new Listr<Context>(
             [
@@ -32,7 +33,7 @@ command
                 getGitInfo(ctx),
                 createBuild(ctx),
                 exec(ctx),
-                finalizeBuild(ctx, 30000)
+                finalizeBuild(ctx)
             ],
             {
                 rendererOptions: {

src/lib/ctx.ts

@@ -47,6 +47,7 @@ export default (options: Record<string, string>): Context => {
             projectId: '',
         },
         args: {},
-        cliVersion: version
+        cliVersion: version,
+        totalSnapshots: -1
     }
 }

src/lib/httpClient.ts

@@ -65,13 +65,14 @@ export default class httpClient {
         }, log)
     }
 
-    finalizeBuild(buildId: string, log: Logger) {
+    finalizeBuild(buildId: string, totalSnapshots: number, log: Logger) {
+        let params: Record<string, string | number> = {buildId};
+        if (totalSnapshots > -1) params.totalSnapshots = totalSnapshots;
+
         return this.request({
             url: '/build',
             method: 'DELETE',
-            params: {
-                buildId: buildId
-            }
+            params: params
         }, log)
     }
 

src/lib/server.ts

@@ -29,6 +29,8 @@ export default async (ctx: Context): Promise<FastifyInstance<Server, IncomingMes
 		} catch (error: any) {
 			reply.code(500).send({ error: { message: error.message}})
 		}
+
+		ctx.totalSnapshots++
 		reply.code(200).send({data: { message: "success" }});
 	});
 

src/tasks/finalizeBuild.ts

@@ -2,15 +2,13 @@ import { ListrTask, ListrRendererFactory } from 'listr2';
 import { Context } from '../types.js'
 import chalk from 'chalk';
 
-export default (ctx: Context, waitTime: number): ListrTask<Context, ListrRendererFactory, ListrRendererFactory>  =>  {
+export default (ctx: Context): ListrTask<Context, ListrRendererFactory, ListrRendererFactory>  =>  {
     return {
         title: `Finalizing build`,
         task: async (ctx, task): Promise<void> => {
             try {
-                if (waitTime > 0) {
-                    await new Promise(resolve => (setTimeout(resolve, waitTime)));
-                    await ctx.client.finalizeBuild(ctx.build.id, ctx.log);
-                }
+                await new Promise(resolve => (setTimeout(resolve, 2000)));
+                await ctx.client.finalizeBuild(ctx.build.id, ctx.totalSnapshots, ctx.log);
                 task.output = chalk.gray(`build url: ${ctx.build.url}`);
                 task.title = 'Finalized build'
             } catch (error) {

src/types.ts

@@ -20,6 +20,7 @@ export interface Context {
         execCommand?: Array<string>
     }
     cliVersion: string;
+    totalSnapshots: number;
 
 }