add markbaseline and baseline option

Author: parthlambdatest

src/commander/commander.ts

@@ -17,6 +17,8 @@ program
     .description('CLI to help you run your SmartUI tests on LambdaTest platform')
     .version(`v${version}`)
     .option('-c --config <filepath>', 'Config file path')
+    .option('--markBaseline', 'Mark this build baseline')
+    .option('--baselineBranch <string>', 'Mark this build baseline')
     .addCommand(exec)
     .addCommand(capture)
     .addCommand(configWeb)

src/lib/ctx.ts

@@ -144,6 +144,7 @@ export default (options: Record<string, string>): Context => {
             ignorePattern: ignoreFilePattern,
             fetchResults: fetchResultObj,
             fetchResultsFileName: fetchResultsFileObj,
+            baselineBranch: options.baselineBranch || ''
         },
         cliVersion: version,
         totalSnapshots: -1,

src/lib/git.ts

@@ -27,16 +27,27 @@ export function isGitRepo(): boolean {
 }
 
 export default (ctx: Context): Git => {
+	if (ctx.env.SMART_GIT) {
+		ctx.env.BASELINE_BRANCH = ''
+		if (ctx.options.baselineBranch !== '') {
+			ctx.env.SMART_GIT = false
+		}
+	}
 	if (ctx.env.SMARTUI_GIT_INFO_FILEPATH) {
 		let gitInfo = JSON.parse(fs.readFileSync(ctx.env.SMARTUI_GIT_INFO_FILEPATH, 'utf-8'));
 
+		if (ctx.options.markBaseline) {
+			ctx.env.BASELINE_BRANCH = ctx.env.CURRENT_BRANCH || gitInfo.branch || ''
+			ctx.env.SMART_GIT = false
+		}
+
 		return {
 			branch: ctx.env.CURRENT_BRANCH || gitInfo.branch || '',
 			commitId: gitInfo.commit_id.slice(0,6) || '',
 			commitMessage: gitInfo.commit_body || '',
 			commitAuthor: gitInfo.commit_author || '',
 			githubURL: (ctx.env.GITHUB_ACTIONS) ? `${constants.GITHUB_API_HOST}/repos/${process.env.GITHUB_REPOSITORY}/statuses/${gitInfo.commit_id}` : '',
-			baselineBranch: ctx.env.BASELINE_BRANCH || ''
+			baselineBranch: ctx.options.baselineBranch || ctx.env.BASELINE_BRANCH || ''
 		}
 	} else {
 		const splitCharacter = '<##>';
@@ -52,13 +63,18 @@ export default (ctx: Context): Git => {
 		var branch = ctx.env.CURRENT_BRANCH || branchAndTags[0];
 		var tags = branchAndTags.slice(1);
 
+		if (ctx.options.markBaseline) {
+			ctx.env.BASELINE_BRANCH = branch || ''
+			ctx.env.SMART_GIT = false
+		}
+
 		return {
 			branch: branch || '',
 			commitId: res[0] || '',
 			commitMessage: res[2] || '',
 			commitAuthor: res[7] || '',
 			githubURL: (ctx.env.GITHUB_ACTIONS) ? `${constants.GITHUB_API_HOST}/repos/${process.env.GITHUB_REPOSITORY}/statuses/${res[1]}` : '',
-			baselineBranch: ctx.env.BASELINE_BRANCH || ''
+			baselineBranch: ctx.options.baselineBranch || ctx.env.BASELINE_BRANCH || ''
 		};
 	}
 }

src/lib/httpClient.ts

@@ -202,7 +202,7 @@ export default class httpClient {
         }
     }
 
-    createBuild(git: Git, config: any, log: Logger, buildName: string, isStartExec: boolean, smartGit: boolean) {
+    createBuild(git: Git, config: any, log: Logger, buildName: string, isStartExec: boolean, smartGit: boolean, markBaseline: boolean) {
         return this.request({
             url: '/build',
             method: 'POST',
@@ -212,7 +212,8 @@ export default class httpClient {
                 buildName,
                 isStartExec,
                 packageVersion: pkgJSON.version,
-                smartGit
+                smartGit,
+                markBaseline
             }
         }, log)
     }

src/lib/snapshotQueue.ts

@@ -350,7 +350,7 @@ export default class Queue {
                     } else {
                         if (!this.ctx.build?.id) {
                             if (this.ctx.authenticatedInitially) {
-                                let resp = await this.ctx.client.createBuild(this.ctx.git, this.ctx.config, this.ctx.log, this.ctx.build.name);
+                                let resp = await this.ctx.client.createBuild(this.ctx.git, this.ctx.config, this.ctx.log, this.ctx.build.name, false, false, false);
                                 this.ctx.build = {
                                     id: resp.data.buildId,
                                     name: resp.data.buildName,

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, ctx.isStartExec, ctx.env.SMART_GIT);
+                let resp = await ctx.client.createBuild(ctx.git, ctx.config, ctx.log, ctx.build.name, ctx.isStartExec, ctx.env.SMART_GIT, ctx.options.markBaseline);
                 ctx.build = {
                     id: resp.data.buildId,
                     name: resp.data.buildName,

src/tasks/createBuildExec.ts

@@ -12,7 +12,7 @@ export default (ctx: Context): ListrTask<Context, ListrRendererFactory, ListrRen
 
             try {
                 if (ctx.authenticatedInitially && !ctx.config.skipBuildCreation) {
-                    let resp = await ctx.client.createBuild(ctx.git, ctx.config, ctx.log, ctx.build.name, ctx.isStartExec, ctx.env.SMART_GIT);
+                    let resp = await ctx.client.createBuild(ctx.git, ctx.config, ctx.log, ctx.build.name, ctx.isStartExec, ctx.env.SMART_GIT, ctx.options.markBaseline);
                     ctx.build = {
                         id: resp.data.buildId,
                         name: resp.data.buildName,

src/types.ts

@@ -59,7 +59,8 @@ export interface Context {
         stripExtension?: boolean,
         ignorePattern?: Array<string>,
         fetchResults?: boolean,
-        fetchResultsFileName?: string
+        fetchResultsFileName?: string,
+        baselineBranch?: string
     }
     cliVersion: string;
     totalSnapshots: number;