Merge pull request #99 from LambdaTest/stage

Added storybook cli changes for adding buildName flag

Author: sushobhit-lt

commands/storybook.js

@@ -12,6 +12,7 @@ const { shortPolling } = require('./utils/polling');
 async function storybook(serve, options) {
     let type = /^https?:\/\//.test(serve) ? 'url' : 'dir';
     let storybookConfig = options.config ? options.config : defaultSmartUIConfig.storybook;
+    const buildName = options.buildName? options.buildName : "";
 
     if (type === 'url') {
         await validateStorybookUrl(serve);
@@ -113,6 +114,29 @@ async function storybook(serve, options) {
                 }
 
                 let commit = await getLastCommit();
+                let baseLine = process.env.BASELINE_BRANCH;
+                let currentBranch = process.env.CURRENT_BRANCH;
+                if (baseLine !== null && baseLine !== undefined){
+                    if(baseLine === ''){
+                        const error = {
+                            "error": "MISSING_BRANCH_NAME",
+                            "message": "Error : The baseline branch name environment variable cannot be empty."
+                        };
+                        console.log(JSON.stringify(error, null, 2));
+                        process.exit(1);
+                    }
+                }
+                    
+                if(currentBranch !== null && currentBranch !==undefined){
+                    if(currentBranch === ''){
+                        const error = {
+                            "error": "MISSING_BRANCH_NAME",
+                            "message": "Error : The current branch name environment variable cannot be empty."
+                        };
+                        console.log(JSON.stringify(error, null, 2));
+                        process.exit(1);
+                    }
+                }
                 let payload = {
                     downloadURL: url.substring(url.search(/.com/)+5, url.search(/.zip/)+4),
                     uploadId: uploadId,
@@ -125,12 +149,14 @@ async function storybook(serve, options) {
                         customViewports: storybookConfig.customViewports
                     },
                     git: {
-                        branch: commit.branch,
-                        commitId: commit.shortHash,
-                        commitAuthor: commit.author.name,
-                        commitMessage: commit.subject,
+                        branch: currentBranch || commit.branch|| '',  
+                        baselineBranch: baseLine || '',
+                        commitId: commit.shortHash, 
+                        commitAuthor: commit.author.name, 
+                        commitMessage: commit.subject, 
                         githubURL: process.env.GITHUB_URL || '',
-                    }
+                    },
+                    buildName: buildName,
                 }
 
                 // Call static render API

index.js

@@ -34,13 +34,22 @@ program.command('storybook')
     .argument('<url|directory>', 'Storybook url or static build directory')
     .option('-c --config <file>', 'Config file path')
     .option('--force-rebuild', 'Force a rebuild of an already existing build.', false)
+    .option('--buildName <string>', 'Specify the build name for the pipeline')
     .action(async function(serve, options) {
         options.env = program.opts().env || 'prod';
 
         console.log('SmartUI Storybook CLI v' + version);
         await checkUpdate(version, options);
         console.log('\n');
-
+        // Check if buildName is undefined or empty string
+        if (options.buildName === '') {
+            const error = {
+                "error": "MISSING_BUILD_NAME",
+                "message": "The --buildName flag requires a value."
+            };
+            console.log(JSON.stringify(error, null, 2));
+            process.exit(1);
+        }
         if (options.config) {
             options.config = validateConfig(options.config);
         }

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lambdatest/smartui-storybook",
-  "version": "1.1.20",
+  "version": "1.1.21",
   "description": "LambdaTest's command-line interface (CLI) aimed to help you run your SmartUI tests on LambdaTest platform",
   "main": "index.js",
   "repository": {