@zerothreatai/cli 1.0.0 → 3.0.0

package/dist/index.js

@@ -8,17 +8,12 @@ const commander_1 = require("commander");
 const menu_js_1 = __importDefault(require("./src/menu.js"));
 const start_setup_js_1 = require("./src/commands/start-setup.js");
 const dotenv_1 = require("dotenv");
-(0, dotenv_1.config)({
-    path: !process.env.WORKING_ENVIRONMENT ? `.env.prod` : `.env`
-});
-commander_1.program
-    .name("zt")
-    .description("ZEROTHREAT AI CLI")
-    .version("1.0.0");
-commander_1.program
-    .command("menu")
-    .description("Show main menu")
-    .action(menu_js_1.default);
+if (process.env.WORKING_ENVIRONMENT)
+    (0, dotenv_1.config)({
+        path: `.env`,
+    });
+commander_1.program.name("zt").description("ZEROTHREAT AI CLI").version("1.0.0");
+commander_1.program.command("menu").description("Show main menu").action(menu_js_1.default);
 commander_1.program.command("start-setup").action(start_setup_js_1.startSetup);
 if (!process.argv.slice(2).length) {
     (0, menu_js_1.default)();

package/dist/src/actions/license-service.js

@@ -57,7 +57,6 @@ async function runCompose(args) {
     });
 }
 async function firstIgnition(licenseKey, emailId) {
-    console.log(">> Running first docker setup ...");
     let token = '';
     const acrTokenService = new acr_token_service_1.default();
     const machineId = (0, get_mac_1.getMachineId)();
@@ -68,6 +67,7 @@ async function firstIgnition(licenseKey, emailId) {
         if (app_constants_1.dockerComposeAcr) {
             COMPOSE_FILE = app_constants_1.dockerComposeAcr;
         }
+        console.log(">> Setting up application ...");
         await ensureNetwork();
         await pullImages();
         await runCompose(["up", "-d"]);

package/dist/src/commands/install-docker.js

@@ -0,0 +1,57 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.installDocker = installDocker;
+const child_process_1 = require("child_process");
+const util_1 = require("util");
+const ask_que_1 = require("../utils/ask-que");
+const chalk_1 = __importDefault(require("chalk"));
+const execAsync = (0, util_1.promisify)(child_process_1.exec);
+async function installDocker() {
+    try {
+        // Check OS
+        const platform = process.platform;
+        if (platform !== 'linux') {
+            console.log('Error: Linux/Ubuntu is required for this installation.');
+            return;
+        }
+        // Check if Docker is available
+        try {
+            await execAsync('docker --version');
+            console.log(chalk_1.default.greenBright.bold('✓ System requirements are met. Docker is already installed.'));
+            return;
+        }
+        catch {
+            console.log(chalk_1.default.red('Docker is not installed.'));
+        }
+        // Ask user permission to install Docker
+        const answer = await (0, ask_que_1.ask)(chalk_1.default.yellowBright.bold('Do you want to install Docker? (yes/no): '));
+        if (answer.toLowerCase() !== 'yes' && answer.toLowerCase() !== 'y') {
+            console.log('Installation cancelled.');
+            return;
+        }
+        console.log(chalk_1.default.bold('Installing Docker...'));
+        const commands = [
+            'sudo apt update',
+            'sudo apt install apt-transport-https ca-certificates curl software-properties-common -y',
+            'curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /usr/share/keyrings/docker-archive-keyring.gpg',
+            'echo "deb [arch=$(dpkg --print-architecture) signed-by=/usr/share/keyrings/docker-archive-keyring.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null',
+            'sudo apt update',
+            'sudo apt install docker-ce -y',
+            'sudo usermod -aG docker ${USER}',
+        ];
+        for (const cmd of commands) {
+            console.log(`Running: ${cmd}`);
+            await execAsync(cmd);
+        }
+        // Check if Docker is available
+        await execAsync('docker --version');
+        console.log(chalk_1.default.greenBright.bold('✓ System requirements are met. ✓ Docker installed successfully.'));
+    }
+    catch (error) {
+        console.error(chalk_1.default.redBright.bold('Error during system check:', error));
+    }
+}
+;

package/dist/src/config/api-config.js

@@ -2,9 +2,17 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.API_CONFIG = void 0;
 const API_CONFIG = () => {
-    return {
-        licenseApi: process.env.LICENSE_API,
-        onPremLicenseCloudeApi: process.env.CLOUD_API,
-    };
+    if (!process.env.WORKING_ENVIRONMENT) {
+        return {
+            licenseApi: `http://localhost:3201/api/license`,
+            onPremLicenseCloudeApi: `https://api.zerothreat.ai/api/on-prem/license`,
+        };
+    }
+    else {
+        return {
+            licenseApi: process.env.LICENSE_API,
+            onPremLicenseCloudeApi: process.env.CLOUD_API,
+        };
+    }
 };
 exports.API_CONFIG = API_CONFIG;

package/dist/src/menu.js

@@ -8,6 +8,7 @@ const inquirer_1 = __importDefault(require("inquirer"));
 const chalk_1 = __importDefault(require("chalk"));
 const start_setup_1 = require("./commands/start-setup");
 const deactivate_1 = __importDefault(require("./commands/deactivate"));
+const install_docker_1 = require("./commands/install-docker");
 async function showMenu() {
     console.clear();
     // Header with description
@@ -22,7 +23,7 @@ async function showMenu() {
     console.log(chalk_1.default.gray('║') + chalk_1.default.gray('  modern web apps & APIs through continuous pentesting, actionable            ') + chalk_1.default.gray('║'));
     console.log(chalk_1.default.gray('║') + chalk_1.default.gray('  insights, and coverage for 40,000+ vulnerabilities.                         ') + chalk_1.default.gray('║'));
     console.log(chalk_1.default.gray('║') + ' '.repeat(78) + chalk_1.default.gray('║'));
-    console.log(chalk_1.default.gray('║') + chalk_1.default.bold.green('  🛡️  On-Premise Installation Tool CLI                                         ') + chalk_1.default.gray('║'));
+    console.log(chalk_1.default.gray('║') + chalk_1.default.bold.green('  🛡️  On-Premise Installation Tool                                         ') + chalk_1.default.gray('║'));
     console.log(chalk_1.default.gray('║') + ' '.repeat(78) + chalk_1.default.gray('║'));
     console.log(chalk_1.default.gray('╚' + '═'.repeat(78) + '╝'));
     console.log();
@@ -35,6 +36,10 @@ async function showMenu() {
             name: "action",
             message: chalk_1.default.bold.cyan("Select an action:"),
             choices: [
+                {
+                    name: chalk_1.default.green("⚙️  Install Prerequisite Software for Linux Environment"),
+                    value: "System Check"
+                },
                 {
                     name: chalk_1.default.green("🔑 Activate License & Setup"),
                     value: "Start Setup"
@@ -65,6 +70,9 @@ async function showMenu() {
         case "Deactivate License":
             await (0, deactivate_1.default)();
             break;
+        case "System Check":
+            await (0, install_docker_1.installDocker)();
+            break;
         default:
             console.log("Exiting...");
             console.clear();

package/dist/src/services/api-service.js

@@ -5,6 +5,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 const axios_1 = __importDefault(require("axios"));
 const https_1 = __importDefault(require("https"));
+const network_error_1 = __importDefault(require("../utils/network-error"));
 class ApiService {
     client;
     constructor(config) {
@@ -42,7 +43,7 @@ class ApiService {
             return new Error(`${error.response.data?.message || 'Unknown error'}`);
         }
         if (error.request) {
-            return new Error('Network Error: No response received');
+            return new network_error_1.default('Network Error: No response received');
         }
         return new Error(`Request Error: ${error.message}`);
     }

package/dist/src/services/license-api-service.js

@@ -6,6 +6,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.LicenseType = exports.LicenseStatus = void 0;
 const api_service_1 = __importDefault(require("./api-service"));
 const api_config_1 = require("../config/api-config");
+const network_error_1 = __importDefault(require("../utils/network-error"));
 var LicenseStatus;
 (function (LicenseStatus) {
     LicenseStatus[LicenseStatus["Active"] = 1] = "Active";
@@ -53,7 +54,7 @@ class LicenseApiService extends api_service_1.default {
             return res;
         }
         catch (error) {
-            if (error.code === 'ECONNREFUSED') {
+            if (error.code === 'ECONNREFUSED' || error instanceof network_error_1.default) {
                 const fallbackService = new api_service_1.default({
                     baseURL: (0, api_config_1.API_CONFIG)().onPremLicenseCloudeApi,
                     timeout: 15000,

package/dist/src/utils/network-error.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+class NetworkError extends Error {
+    constructor(message) {
+        super(message);
+        Object.setPrototypeOf(this, new.target.prototype);
+    }
+}
+exports.default = NetworkError;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zerothreatai/cli",
-  "version": "1.0.0",
+  "version": "3.0.0",
   "main": "dist/index.js",
   "scripts": {
     "build": "node build.js",
@@ -26,8 +26,7 @@
     "node-machine-id": "^1.1.12"
   },
   "files": [
-    "dist",
-    ".env.prod"
+    "dist"
   ],
   "bin": {
     "zt": "./dist/index.js"

package/.env.prod

@@ -1,2 +0,0 @@
-LICENSE_API=http://localhost:3201/api/license
-CLOUD_API=https://api.zerothreat.ai/api/on-prem/license