building out cli arguments

ngrok.js

@@ -1,61 +1,74 @@
-const ngrok = require('ngrok');
+const ngrok = require("ngrok");
-const fs = require('fs');
+const fs = require("fs");
 const { exec } = require("child_process");
 
-const dotenvFile = '../.env'
+const dotenvFile = "../.env";
+
+const optionDefinitions = [
+  { name: "proto", type: String },
+  { name: "addr", type: Number },
+];
+
+const commandLineArgs = require("command-line-args");
+const options = commandLineArgs(optionDefinitions);
+
+console.log(options);
 
 // Read the .env file
-fs.readFile(dotenvFile, 'utf8', (err, data) => {
+fs.readFile(dotenvFile, "utf8", (err, data) => {
-	if (err) {
+  if (err) {
-		console.error(err);
+    console.error(err);
-		return;
+    return;
-	}
+  }
 
-    // Convert string to string array, split at newlines
+  // Convert string to string array, split at newlines
-    let lines = data.split('\n') // string[]
+  let lines = data.split("\n"); // string[]
-    let token = null; // string || null
+  let authtoken = null; // string || null
 
-    // find the auth token
+  // find the auth token
-    lines.forEach(element => {
+  lines.forEach((element) => {
-        const [name, value] = element.split('=');
+    const [name, value] = element.split("=");
-        if (name === 'NGROK_AUTHTOKEN')
+    if (name === "NGROK_AUTHTOKEN") authtoken = value;
-            token = value;
+  });
-    });
 
-    // No token found
+  // No token found
-    if (!token) {
+  if (!authtoken) {
-        console.error("Setup NGROK_AUTHTOKEN in your .env file")
+    console.error("Setup NGROK_AUTHTOKEN in your .env file");
-        return;
+    return;
-    }
+  }
+
+  // Start ngrok
+  (async function () {
+    const proto = options?.proto ?? "http";
+    const addr = options?.addr ? options.addr : 3000;
 
     // Start ngrok
-	(async function () {
+    const url = await ngrok.connect({ authtoken, proto, addr });
-		const url = await ngrok.connect({ authtoken: token, proto: 'http', addr: 3000 });
 
-        // Rebuild lines with the new url
+    // Rebuild lines with the new url
-        lines = lines.map(element => {
+    lines = lines.map((element) => {
-            const [name] = element.split('=');
+      const [name] = element.split("=");
-            if (name === 'NGROK_SERVERHOST')
+      if (name === "NGROK_SERVERHOST") return `${name}=${url}`;
-                return `${name}=${url}`
+      return element;
-            return element;
+    });
-        })
 
-        // convert back to string format
+    // convert back to string format
-        let writeData = ''
+    let writeData = "";
-        lines.forEach(element => {
+    lines.forEach((element) => {
-            writeData += element + '\n'
+      writeData += element + "\n";
-        })
+    });
-        writeData = writeData.slice(0, writeData.length - 1);
+    writeData = writeData.slice(0, writeData.length - 1);
 
-        // Write the new .env file
+    // Write the new .env file
-        fs.writeFile(dotenvFile, writeData, (err) => {
+    fs.writeFile(dotenvFile, writeData, (err) => {
-            if (err) {
+      if (err) {
-                console.error(err);
+        console.error(err);
-                return;
+        return;
-            }
+      }
-            console.clear();
+      console.clear();
-            console.log(`Your current ngrok url is ${url}.`);
+      console.log(`Started ngrok: protocol '${proto}', addr '${addr}'`)
-            console.log(`Your .env file has been updated.`);            
+      console.log(`Your current ngrok url is ${url}.`);
-          });		
+      console.log(`Your .env file has been updated.`);
-	})();
+    });
+  })();
 });

package.json

@@ -4,12 +4,13 @@
   "description": "ngrok start-up wrapper that uses a dotenv file to get the authtoken and save the https address",
   "main": "ngrok.js",
   "license": "MIT",
-	"scripts": {
+  "scripts": {
-		"test": "echo \"Error: no test specified\" && exit 1",
+    "test": "echo \"Error: no test specified\" && exit 1",
-		"debug": "node --inspect ngrok.js",
+    "debug": "node --inspect ngrok.js",
-		"start": "node ngrok"
+    "start": "node ngrok"
-	},  
+  },
   "dependencies": {
+    "command-line-args": "^5.2.0",
     "ngrok": "^4.2.2"
   }
 }