Merge pull request #2 from pinanks/DOT-822

Author: pinanks

commands/utils/constants.js

@@ -1,6 +1,5 @@
 var constants = {}
 
-constants.env = 'prod';
 constants.stage = {
     AUTH_URL: "https://stage-api.lambdatestinternal.com/storybook/auth",
     RENDER_API_URL: "https://stage-api.lambdatestinternal.com/storybook/render"

commands/utils/dom.js

@@ -48,7 +48,7 @@ async function sendDoM(storybookUrl, stories, storybookConfig, options) {
     form.append('browser', storybookConfig.browsers);
     form.append('projectToken', process.env.PROJECT_TOKEN);
     form.append('buildName', options.buildname);
-    axios.post(constants[constants.env].RENDER_API_URL, form, {
+    axios.post(constants[options.env].RENDER_API_URL, form, {
         headers: {
             ...form.getHeaders()
         }

commands/utils/validate.js

@@ -1,9 +1,9 @@
 const axios = require('axios');
 var { constants } = require('./constants');
 
-function validateProjectToken() {
+function validateProjectToken(options) {
     if (process.env.PROJECT_TOKEN) { 
-        return axios.get(constants[constants.env].AUTH_URL, {
+        return axios.get(constants[options.env].AUTH_URL, {
             headers: {
                 projectToken: process.env.PROJECT_TOKEN
             }})

index.js

@@ -1,6 +1,6 @@
 #! /usr/bin/env node
 
-const { Command } = require('commander');
+const { Command, Option } = require('commander');
 const program = new Command();
 const { storybook } = require('./commands/storybook');
 const { validateProjectToken } = require('./commands/utils/validate');
@@ -9,7 +9,8 @@ const { createConfig } = require('./commands/config');
 program
     .name('smartui')
     .description('CLI to help you run your SmartUI tests on LambdaTest platform')
-    .version('1.0.0');
+    .version('1.0.0')
+    .addOption(new Option('--env <prod|stage>', 'Runtime environment option').choices(['prod', 'stage']));
 
 const configCommand = program.command('config')
     .description('Manage LambdaTest SmartUI config')
@@ -26,8 +27,10 @@ program.command('storybook')
     .argument('<url>', 'Storybook url')
     .requiredOption('--buildname <string>', 'Build name')
     .option('-c --config <file>', 'Config file path')
+    // .option('--env <prod|stage>', 'Runtime environment option', 'prod')
     .action(async function(url, options) {
-        await validateProjectToken();
+        options.env = program.opts().env || 'prod';
+        await validateProjectToken(options);
         storybook(url, options);
     });