Merge pull request #100 from sushobhit-lt/DOT-5330

add tunnel support in storybook static

Author: parthlambdatest

commands/storybook.js

@@ -157,11 +157,17 @@ async function storybook(serve, options) {
                         githubURL: process.env.GITHUB_URL || '',
                     },
                     buildName: buildName,
+                    tunnel: options.tunnel || {},
                 }
 
                 // Call static render API
                 await axios.post(new URL(constants[options.env].STATIC_RENDER_PATH, constants[options.env].BASE_URL).href, payload)
                     .then(async function (response) {
+                        if (response.data && response.data.error) {
+                            console.log('[smartui] Error: ', response.data.error.message);
+                            process.exitCode = constants.ERROR_CATCHALL;
+                            return
+                        }
                         console.log('[smartui] Build URL: ', response.data.data.buildURL);
                         console.log('[smartui] Build in progress...');
                         await shortPolling(response.data.data.buildId, 0, options);

commands/utils/validate.js

@@ -140,15 +140,37 @@ function validateConfig(configFile) {
     // Sanity check waitForTimeout
     if (!Object.hasOwn(storybookConfig, 'waitForTimeout')) {
         storybookConfig.waitForTimeout = 0;
-    } else if (storybookConfig.waitForTimeout <= 0 || storybookConfig.waitForTimeout > 30000) {
-        console.log('[smartui] Warning: Invalid config, value of waitForTimeout must be > 0 and <= 30000');
+    } else if (storybookConfig.waitForTimeout <= 0 || storybookConfig.waitForTimeout > 300000) {
+        console.log('[smartui] Warning: Invalid config, value of waitForTimeout must be > 0 and <= 300000');
         console.log('[smartui] If you do not wish to include waitForTimeout parameter, remove it from the config file.');
         storybookConfig.waitForTimeout = 0;
     }
 
     return storybookConfig
 }
 
+function validateTunnel(configFile) {
+    // Verify config file exists
+    if (!fs.existsSync(configFile)) {
+        console.log(`[smartui] Error: Config file ${configFile} not found.`);
+        process.exit(constants.ERROR_CATCHALL);
+    }
+
+    let tunnelConfig;
+    try {
+        let config = JSON.parse(fs.readFileSync(configFile));
+        tunnelConfig = config.tunnel || {};
+        if (tunnelConfig &&  tunnelConfig.type != "manual")  {
+            throw new ValidationError('Invalid tunnel type. Accepted type is `manual` only');
+        }
+    } catch (error) {
+        console.log('[smartui] Error: ', error.message);
+        process.exit(constants.ERROR_CATCHALL);
+    }
+
+    return tunnelConfig
+}
+
 function validateConfigBrowsers(browsers) {
     if (browsers.length == 0) {
         throw new ValidationError('empty browsers list.');
@@ -248,5 +270,6 @@ module.exports = {
     validateConfig,
     validateConfigBrowsers,
     validateConfigResolutions,
-    validateCustomViewPorts
+    validateCustomViewPorts,
+    validateTunnel,
 };

index.js

@@ -3,7 +3,7 @@
 const { Command, Option } = require('commander');
 const program = new Command();
 const { storybook } = require('./commands/storybook');
-const { validateProjectToken, validateLatestBuild, validateConfig } = require('./commands/utils/validate');
+const { validateProjectToken, validateLatestBuild, validateConfig, validateTunnel } = require('./commands/utils/validate');
 const { createConfig } = require('./commands/config');
 const { version } = require('./package.json');
 const { checkUpdate } = require('./commands/utils/package');
@@ -50,6 +50,9 @@ program.command('storybook')
             console.log(JSON.stringify(error, null, 2));
             process.exit(1);
         }
+        if (options.config) {
+            options.tunnel = validateTunnel(options.config);
+        }
         if (options.config) {
             options.config = validateConfig(options.config);
         }