final commit for start stop server

Author: parthlambdatest

src/commander/ping.ts

@@ -19,7 +19,7 @@ command
             console.log(chalk.yellow(`Pinging server at ${serverAddress} from terminal...`));
 
             // Send GET request to the /ping endpoint
-            const response = await axios.get(`${serverAddress}/ping`);
+            const response = await axios.get(`${serverAddress}/ping`, { timeout: 15000 });
 
             // Log the response from the server
             if (response.status === 200) {
@@ -30,7 +30,11 @@ command
             }
         } catch (error: any) {
             // Handle any errors during the HTTP request
-            console.error(chalk.red('SmartUI server is not running'));
+            if (error.code === 'ECONNABORTED') {
+                console.error(chalk.red('Error: SmartUI server did not respond in 15 seconds'));
+            } else {
+                console.error(chalk.red('SmartUI server is not running'));
+            }
         }
     });
 

src/commander/server.ts

@@ -7,7 +7,7 @@ import ctxInit from '../lib/ctx.js';
 import getGitInfo from '../tasks/getGitInfo.js';
 import createBuild from '../tasks/createBuild.js';
 import snapshotQueue from '../lib/snapshotQueue.js';
-import { startPolling } from '../lib/utils.js';
+import { startPolling, startPingPolling } from '../lib/utils.js';
 
 const command = new Command();
 
@@ -26,6 +26,7 @@ command
         let ctx: Context = ctxInit(command.optsWithGlobals());
         ctx.snapshotQueue = new snapshotQueue(ctx);
         ctx.totalSnapshots = 0
+        ctx.isStartExec = true
 
         let tasks = new Listr<Context>(
             [
@@ -49,6 +50,7 @@ command
 
         try {
             await tasks.run(ctx);
+            startPingPolling(ctx);
             if (ctx.options.fetchResults) {
                 startPolling(ctx);
             }

src/lib/ctx.ts

@@ -126,6 +126,7 @@ export default (options: Record<string, string>): Context => {
             fetchResultsFileName: fetchResultsFileObj,
         },
         cliVersion: version,
-        totalSnapshots: -1
+        totalSnapshots: -1,
+        isStartExec: false
     }
 }

src/lib/httpClient.ts

@@ -84,14 +84,15 @@ export default class httpClient {
         }
     }
 
-    createBuild(git: Git, config: any, log: Logger, buildName: string) {
+    createBuild(git: Git, config: any, log: Logger, buildName: string, isStartExec: boolean) {
         return this.request({
             url: '/build',
             method: 'POST',
             data: {
                 git,
                 config,
-                buildName
+                buildName,
+                isStartExec
             }
         }, log)
     }
@@ -102,7 +103,18 @@ export default class httpClient {
             method: 'GET',
             params: { buildId, baseline }
         }, log);
-    }    
+    }
+    
+    ping(buildId: string, log: Logger) {
+        return this.request({
+            url: '/build/ping',
+            method: 'POST',
+            data: {
+                buildId: buildId
+            }
+        }, log);
+    }
+    
 
     finalizeBuild(buildId: string, totalSnapshots: number, log: Logger) {
         let params: Record<string, string | number> = {buildId};

src/lib/utils.ts

@@ -307,4 +307,25 @@ export async function startPolling(ctx: Context): Promise<void> {
             isPollingActive = false;
         }
     }, 5000);
-}
+}
+
+export async function startPingPolling(ctx: Context): Promise<void> {
+    try {
+        ctx.log.debug('Sending initial ping to server...');
+        await ctx.client.ping(ctx.build.id, ctx.log);
+        ctx.log.debug('Initial ping sent successfully.');
+    } catch (error: any) {
+        ctx.log.error(`Error during initial ping: ${error.message}`);
+    }
+
+    setInterval(async () => {
+        try {
+            ctx.log.debug('Sending ping to server...');
+            await ctx.client.ping(ctx.build.id, ctx.log);
+            ctx.log.debug('Ping sent successfully.');
+        } catch (error: any) {
+            ctx.log.error(`Error during ping polling: ${error.message}`);
+        }
+    }, 10 * 60 * 1000);
+}
+

src/tasks/createBuild.ts

@@ -10,7 +10,7 @@ export default (ctx: Context): ListrTask<Context, ListrRendererFactory, ListrRen
             updateLogContext({task: 'createBuild'});
 
             try {
-                let resp = await ctx.client.createBuild(ctx.git, ctx.config, ctx.log, ctx.build.name);
+                let resp = await ctx.client.createBuild(ctx.git, ctx.config, ctx.log, ctx.build.name, ctx.isStartExec);
                 ctx.build = {
                     id: resp.data.buildId,
                     name: resp.data.buildName,

src/types.ts

@@ -52,6 +52,7 @@ export interface Context {
     totalSnapshots: number;
     figmaDesignConfig?: FigmaDesignConfig;
     testType?: string;
+    isStartExec ?: boolean;
 }
 
 export interface Env {