Compare commits
No commits in common. "33d57752aab9058e1d0613a3712db7fb7d83e4a3" and "6a7e500b8af063cd2dc8142d1be9853085f2f369" have entirely different histories.
33d57752aa
...
6a7e500b8a
413
ngenv
413
ngenv
|
|
@ -6,54 +6,6 @@ const { spawn, exec } = require("child_process");
|
||||||
const commandLineArgs = require("command-line-args");
|
const commandLineArgs = require("command-line-args");
|
||||||
const ngrok = require("ngrok");
|
const ngrok = require("ngrok");
|
||||||
|
|
||||||
/************************************************** Global **************************************************/
|
|
||||||
|
|
||||||
// Get the command line options
|
|
||||||
const scriptName = path.basename(process.argv[1]);
|
|
||||||
const actionVerb = process.argv[2];
|
|
||||||
const options =
|
|
||||||
(actionVerb && actionVerb !== "--help") || !actionVerb === "-h"
|
|
||||||
? getOptions()
|
|
||||||
: {};
|
|
||||||
function getOptions() {
|
|
||||||
const optionDefinitions = [
|
|
||||||
{ name: "proto", alias: "P", type: String },
|
|
||||||
{ name: "port", alias: "p", type: Number },
|
|
||||||
{ name: "env", alias: "e", type: String },
|
|
||||||
{ name: "verbose", alias: "v", type: Boolean, defaultValue: false },
|
|
||||||
{ name: "help", alias: "h", type: Boolean },
|
|
||||||
];
|
|
||||||
|
|
||||||
try {
|
|
||||||
// Parse options, skipping the first three arguments ('node', script name and verb)
|
|
||||||
const options = commandLineArgs(optionDefinitions, {
|
|
||||||
argv: process.argv.slice(3),
|
|
||||||
});
|
|
||||||
|
|
||||||
return options;
|
|
||||||
} catch (err) {
|
|
||||||
console.error(err.message);
|
|
||||||
process.exit(1);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (options.verbose) console.log("Options:", options);
|
|
||||||
|
|
||||||
// Find the .env file
|
|
||||||
const dotEnvFile = findDotEnv(options?.env ?? ".env");
|
|
||||||
if (!dotEnvFile) {
|
|
||||||
console.error(
|
|
||||||
`The specified environment file was not found: ${options?.env ?? ".env"}`
|
|
||||||
);
|
|
||||||
process.exit(1);
|
|
||||||
}
|
|
||||||
if (options.verbose) console.log(`Using environment file: ${dotEnvFile}`);
|
|
||||||
// Walk the path to find .env file...
|
|
||||||
function findDotEnv(filename) {
|
|
||||||
const currentDir = findProjectFolder(filename);
|
|
||||||
return `${currentDir}/${filename}`;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if the ngrok config file exists
|
|
||||||
const ngrokConfigFile = getNgrokConfig();
|
const ngrokConfigFile = getNgrokConfig();
|
||||||
if (!ngrokConfigFile) {
|
if (!ngrokConfigFile) {
|
||||||
console.error(
|
console.error(
|
||||||
|
|
@ -61,66 +13,14 @@ if (!ngrokConfigFile) {
|
||||||
);
|
);
|
||||||
process.exit(1);
|
process.exit(1);
|
||||||
}
|
}
|
||||||
if (options.verbose) console.log(`Using ngrok config file: ${ngrokConfigFile}`);
|
|
||||||
function getNgrokConfig(pathOnly = false) {
|
|
||||||
const paths = [
|
|
||||||
os.homedir() + "/.ngrok2", // Newer versions of ngrok
|
|
||||||
os.homedir() + "/Library/Application Support/ngrok", // MacOS
|
|
||||||
os.homedir() + "/AppData/Local/ngrok", // Windows
|
|
||||||
os.homedir() + "/.config/ngrok", // Alternative location for linux
|
|
||||||
];
|
|
||||||
for (const p of paths) {
|
|
||||||
if (fs.existsSync(`${p}/ngrok.yml`)) {
|
|
||||||
return pathOnly ? p : `${p}/ngrok.yml`;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (pathOnly) return null;
|
const projectFolder = findProjectFolder();
|
||||||
throw new Error(
|
|
||||||
"Ngrok config file not found. Try running 'ngrok authtoken <token>' in your terminal."
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Read the ngrok config file
|
|
||||||
const authToken = readNgrokConfig(ngrokConfigFile);
|
|
||||||
if (!authToken) {
|
|
||||||
console.error(
|
|
||||||
"No ngrok authtoken found! Run 'ngrok authtoken <token>' in your terminal."
|
|
||||||
);
|
|
||||||
process.exit(1);
|
|
||||||
}
|
|
||||||
function readNgrokConfig() {
|
|
||||||
const data = fs.readFileSync(ngrokConfigFile, "utf8");
|
|
||||||
|
|
||||||
let authToken;
|
|
||||||
|
|
||||||
// Convert string to string array, split at newlines
|
|
||||||
let lines = data.split("\n"); // string[]
|
|
||||||
|
|
||||||
// find the auth token (format: authtoken: token_goes_here)
|
|
||||||
lines.forEach((element) => {
|
|
||||||
const [name, value] = element.split(": ");
|
|
||||||
if (name === "authtoken") authToken = value;
|
|
||||||
});
|
|
||||||
|
|
||||||
// No token found
|
|
||||||
if (!authToken) {
|
|
||||||
// https://dashboard.ngrok.com/get-started/your-authtoken
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
return authToken;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Setup the ngenv folder
|
|
||||||
const projectFolder = findProjectFolder(options?.env ?? ".env");
|
|
||||||
if (!projectFolder) {
|
if (!projectFolder) {
|
||||||
console.error(
|
console.error(
|
||||||
"No project folder found! Make sure there is a .env file in the project folder."
|
"No project folder found! Make sure there is a .env file in the project folder."
|
||||||
);
|
);
|
||||||
process.exit(1);
|
process.exit(1);
|
||||||
}
|
}
|
||||||
if (options.verbose) console.log(`Using project folder: ${projectFolder}`);
|
|
||||||
if (!fs.existsSync(`${projectFolder}/.ngenv`)) {
|
if (!fs.existsSync(`${projectFolder}/.ngenv`)) {
|
||||||
fs.mkdirSync(`${projectFolder}/.ngenv`);
|
fs.mkdirSync(`${projectFolder}/.ngenv`);
|
||||||
}
|
}
|
||||||
|
|
@ -128,161 +28,7 @@ const lockFile = `${projectFolder}/.ngenv/running.lock`;
|
||||||
const outLogFile = `${projectFolder}/.ngenv/out.log`;
|
const outLogFile = `${projectFolder}/.ngenv/out.log`;
|
||||||
const errLogFile = `${projectFolder}/.ngenv/err.log`;
|
const errLogFile = `${projectFolder}/.ngenv/err.log`;
|
||||||
|
|
||||||
/******************************************* Startup *******************************************/
|
let authToken = null; // string || null
|
||||||
|
|
||||||
// Show the root help
|
|
||||||
if (!actionVerb || actionVerb === "--help" || actionVerb === "-h") {
|
|
||||||
rootHelp();
|
|
||||||
process.exit(0);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Show the command specific help
|
|
||||||
if (options.help) {
|
|
||||||
commandHelp(actionVerb);
|
|
||||||
process.exit(0);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Process command line arguments
|
|
||||||
switch (actionVerb) {
|
|
||||||
case "start":
|
|
||||||
// Spawn the background process
|
|
||||||
startBackgroundProcess();
|
|
||||||
break;
|
|
||||||
case "stop":
|
|
||||||
// Stop the background process
|
|
||||||
stopBackgroundProcess();
|
|
||||||
break;
|
|
||||||
case "clear":
|
|
||||||
// Clear the log files
|
|
||||||
clearLogs();
|
|
||||||
break;
|
|
||||||
case "logs":
|
|
||||||
// Display the log files
|
|
||||||
displayLogs();
|
|
||||||
break;
|
|
||||||
case "run":
|
|
||||||
case "background": {
|
|
||||||
// Set up the background process
|
|
||||||
if (actionVerb === "background") {
|
|
||||||
// Start-up for the background process
|
|
||||||
process.on("SIGTERM", () => {
|
|
||||||
console.log("Stopping background process...");
|
|
||||||
removeLockFile();
|
|
||||||
process.exit(0);
|
|
||||||
});
|
|
||||||
process.on("exit", (code) => {
|
|
||||||
console.log(`About to exit with code: ${code}`);
|
|
||||||
// Perform cleanup or final operations here
|
|
||||||
removeLockFile();
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Start the main process
|
|
||||||
main();
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
console.log(`Unknown command: ${actionVerb}`);
|
|
||||||
console.log(`Run '${scriptName} --help' for usage.`);
|
|
||||||
process.exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Process command line arguments
|
|
||||||
if (actionVerb === "start") {
|
|
||||||
} else if (actionVerb === "stop") {
|
|
||||||
// Stop the background process
|
|
||||||
stopBackgroundProcess();
|
|
||||||
} else if (actionVerb === "clear") {
|
|
||||||
// Clear the log files
|
|
||||||
clearLogs();
|
|
||||||
} else if (actionVerb === "logs") {
|
|
||||||
// Display the log files
|
|
||||||
displayLogs();
|
|
||||||
// tailLogs();
|
|
||||||
} else if (actionVerb === "run" || actionVerb === "background") {
|
|
||||||
// Set up the background process
|
|
||||||
if (actionVerb === "background") {
|
|
||||||
// Start-up for the background process
|
|
||||||
process.on("SIGTERM", () => {
|
|
||||||
console.log("Stopping background process...");
|
|
||||||
removeLockFile();
|
|
||||||
process.exit(0);
|
|
||||||
});
|
|
||||||
process.on("exit", (code) => {
|
|
||||||
console.log(`About to exit with code: ${code}`);
|
|
||||||
// Perform cleanup or final operations here
|
|
||||||
removeLockFile();
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Start the main process
|
|
||||||
main();
|
|
||||||
} else {
|
|
||||||
rootHelp();
|
|
||||||
}
|
|
||||||
|
|
||||||
function rootHelp() {
|
|
||||||
console.log(`Usage: ${scriptName} [run|start|stop|logs|clear] [options]`);
|
|
||||||
console.log(`\nCommands:`);
|
|
||||||
console.log(` run\t\t\tRun the tunnel in the foreground.`);
|
|
||||||
console.log(` start\t\t\tStart the tunnel in the background.`);
|
|
||||||
console.log(` stop\t\t\tStop the background process if it's running.`);
|
|
||||||
console.log(` logs\t\t\tDisplay the output and error logs.`);
|
|
||||||
console.log(` clear\t\t\tClear the content of the output and error logs.`);
|
|
||||||
console.log(`\nOptions:`);
|
|
||||||
console.log(` -h, --help\t\tDisplay this help message.`);
|
|
||||||
console.log(`\nExample:`);
|
|
||||||
console.log(` ${scriptName} start\t\t# Starts the background process`);
|
|
||||||
}
|
|
||||||
|
|
||||||
function commandHelp(command) {
|
|
||||||
console.log(`Usage: ${scriptName} ${command} [options]`);
|
|
||||||
console.log(`\nOptions:`);
|
|
||||||
if (command === "start" || command === "run") {
|
|
||||||
console.log(
|
|
||||||
` -P, --proto\t\tProtocol to use (http|tcp|tls). Default is http.`
|
|
||||||
);
|
|
||||||
console.log(` -p, --port\t\tPort to use. Default is 3000.`);
|
|
||||||
console.log(` -e, --env\t\tEnvironment file to use. Default is .env.`);
|
|
||||||
console.log(` -v, --verbose\t\tShow verbose output.`);
|
|
||||||
}
|
|
||||||
console.log(` -h, --help\t\tDisplay this help message.`);
|
|
||||||
}
|
|
||||||
|
|
||||||
/******************************************* Main *******************************************/
|
|
||||||
|
|
||||||
async function main() {
|
|
||||||
// Get the ngrok config file path
|
|
||||||
let ngrokConfig = getNgrokConfig();
|
|
||||||
|
|
||||||
// Check if the ngrok config file exists
|
|
||||||
if (!fs.existsSync(ngrokConfig)) {
|
|
||||||
throw new Error("No ngrok config file found");
|
|
||||||
}
|
|
||||||
|
|
||||||
// Start ngrok
|
|
||||||
const proto = options?.proto ?? "http";
|
|
||||||
const addr = options?.port ? options.port : 3000;
|
|
||||||
try {
|
|
||||||
const url = await ngrok.connect({
|
|
||||||
authtoken: authToken,
|
|
||||||
proto,
|
|
||||||
addr,
|
|
||||||
onLogEvent: (data) => {
|
|
||||||
// console.log(data);
|
|
||||||
},
|
|
||||||
});
|
|
||||||
writeDotEnv(url); // Sync the .env file
|
|
||||||
console.log(`Started ngrok: protocol '${proto}', addr '${addr}'`);
|
|
||||||
console.log(`Your current ngrok url is ${url}.`);
|
|
||||||
console.log(`Your .env file has been updated.`);
|
|
||||||
} catch (err) {
|
|
||||||
console.error(err.message);
|
|
||||||
process.exit(1);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/******************************************* Functions *******************************************/
|
|
||||||
|
|
||||||
function isAlreadyRunning() {
|
function isAlreadyRunning() {
|
||||||
return fs.existsSync(lockFile);
|
return fs.existsSync(lockFile);
|
||||||
|
|
@ -385,6 +131,151 @@ function clearLogs() {
|
||||||
console.log("Log files cleared.");
|
console.log("Log files cleared.");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function getOptions() {
|
||||||
|
const optionDefinitions = [
|
||||||
|
{ name: "proto", alias: "P", type: String },
|
||||||
|
{ name: "port", alias: "p", type: Number },
|
||||||
|
{ name: "env", alias: "e", type: String },
|
||||||
|
];
|
||||||
|
|
||||||
|
// Parse options, skipping the first three arguments ('node', script name and verb)
|
||||||
|
const options = commandLineArgs(optionDefinitions, {
|
||||||
|
argv: process.argv.slice(3),
|
||||||
|
});
|
||||||
|
|
||||||
|
return options;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Process command line arguments
|
||||||
|
if (process.argv[2] === "start") {
|
||||||
|
// The result is only used in the background process, but the check is also done here, before the
|
||||||
|
// background process is started to prevent bad arguments from being passed to the background process.
|
||||||
|
getOptions();
|
||||||
|
// Spawn the background process
|
||||||
|
startBackgroundProcess();
|
||||||
|
} else if (process.argv[2] === "stop") {
|
||||||
|
// Stop the background process
|
||||||
|
stopBackgroundProcess();
|
||||||
|
} else if (process.argv[2] === "clear") {
|
||||||
|
// Clear the log files
|
||||||
|
clearLogs();
|
||||||
|
} else if (process.argv[2] === "logs") {
|
||||||
|
// Display the log files
|
||||||
|
displayLogs();
|
||||||
|
// tailLogs();
|
||||||
|
} else if (process.argv[2] === "run" || process.argv[2] === "background") {
|
||||||
|
if (process.argv[2] === "background") {
|
||||||
|
// Start-up for the background process
|
||||||
|
process.on("SIGTERM", () => {
|
||||||
|
console.log("Stopping background process...");
|
||||||
|
removeLockFile();
|
||||||
|
process.exit(0);
|
||||||
|
});
|
||||||
|
process.on("exit", (code) => {
|
||||||
|
console.log(`About to exit with code: ${code}`);
|
||||||
|
// Perform cleanup or final operations here
|
||||||
|
removeLockFile();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
const options = getOptions();
|
||||||
|
backgroundMain(options);
|
||||||
|
} else {
|
||||||
|
const scriptName = path.basename(process.argv[1]);
|
||||||
|
console.log(`Usage: node ${scriptName} [run|start|stop|logs|clear]`);
|
||||||
|
console.log(`\nCommands:`);
|
||||||
|
console.log(
|
||||||
|
` start\t\tStart the background process if it's not already running.`
|
||||||
|
);
|
||||||
|
console.log(` stop\t\tStop the background process if it's running.`);
|
||||||
|
console.log(` logs\t\tDisplay the output and error logs.`);
|
||||||
|
console.log(` clear\t\tClear the content of the output and error logs.`);
|
||||||
|
console.log(`\nExample:`);
|
||||||
|
console.log(` node ${scriptName} start\t\t# Starts the background process`);
|
||||||
|
}
|
||||||
|
|
||||||
|
async function backgroundMain(options) {
|
||||||
|
// Get the ngrok config file path
|
||||||
|
let ngrokConfig = getNgrokConfig();
|
||||||
|
|
||||||
|
// Check if the ngrok config file exists
|
||||||
|
if (!fs.existsSync(ngrokConfig)) {
|
||||||
|
throw new Error("No ngrok config file found");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read the ngrok config file
|
||||||
|
try {
|
||||||
|
const { authToken } = readNgrokConfig(ngrokConfig);
|
||||||
|
global.authToken = authToken;
|
||||||
|
} catch (err) {
|
||||||
|
console.error(err);
|
||||||
|
process.exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start ngrok
|
||||||
|
const proto = options?.proto ?? "http";
|
||||||
|
const addr = options?.port ? options.port : 3000;
|
||||||
|
try {
|
||||||
|
const url = await ngrok.connect({
|
||||||
|
authtoken: authToken,
|
||||||
|
proto,
|
||||||
|
addr,
|
||||||
|
onLogEvent: (data) => {
|
||||||
|
// console.log(data);
|
||||||
|
},
|
||||||
|
});
|
||||||
|
writeDotEnv(url); // Sync the .env file
|
||||||
|
console.log(`Started ngrok: protocol '${proto}', addr '${addr}'`);
|
||||||
|
console.log(`Your current ngrok url is ${url}.`);
|
||||||
|
console.log(`Your .env file has been updated.`);
|
||||||
|
} catch (err) {
|
||||||
|
console.error(err.message);
|
||||||
|
process.exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function getNgrokConfig(pathOnly = false) {
|
||||||
|
const paths = [
|
||||||
|
os.homedir() + "/.ngrok2", // Newer versions of ngrok
|
||||||
|
os.homedir() + "/Library/Application Support/ngrok", // MacOS
|
||||||
|
os.homedir() + "/AppData/Local/ngrok", // Windows
|
||||||
|
os.homedir() + "/.config/ngrok", // Alternative location for linux
|
||||||
|
];
|
||||||
|
for (const p of paths) {
|
||||||
|
if (fs.existsSync(`${p}/ngrok.yml`)) {
|
||||||
|
return pathOnly ? p : `${p}/ngrok.yml`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pathOnly) return null;
|
||||||
|
throw new Error(
|
||||||
|
"Ngrok config file not found. Try running 'ngrok authtoken <token>' in your terminal."
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
function readNgrokConfig(ngrokConfig) {
|
||||||
|
const data = fs.readFileSync(ngrokConfig, "utf8");
|
||||||
|
|
||||||
|
let authToken;
|
||||||
|
|
||||||
|
// Convert string to string array, split at newlines
|
||||||
|
let lines = data.split("\n"); // string[]
|
||||||
|
|
||||||
|
// find the auth token (format: authtoken: token_goes_here)
|
||||||
|
lines.forEach((element) => {
|
||||||
|
const [name, value] = element.split(": ");
|
||||||
|
if (name === "authtoken") authToken = value;
|
||||||
|
});
|
||||||
|
|
||||||
|
// No token found
|
||||||
|
if (!authToken) {
|
||||||
|
// https://dashboard.ngrok.com/get-started/your-authtoken
|
||||||
|
throw new Error("No ngrok authtoken found in the config file");
|
||||||
|
}
|
||||||
|
|
||||||
|
return { authToken };
|
||||||
|
}
|
||||||
|
|
||||||
function findProjectFolder(filename = ".env") {
|
function findProjectFolder(filename = ".env") {
|
||||||
// Walk the path to find .env file... options?.env
|
// Walk the path to find .env file... options?.env
|
||||||
let currentDir = __dirname;
|
let currentDir = __dirname;
|
||||||
|
|
@ -406,6 +297,12 @@ function findProjectFolder(filename = ".env") {
|
||||||
return currentDir;
|
return currentDir;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Walk the path to find .env file...
|
||||||
|
function findDotEnv(filename = ".env") {
|
||||||
|
const currentDir = findProjectFolder(filename);
|
||||||
|
return `${currentDir}/${filename}`;
|
||||||
|
}
|
||||||
|
|
||||||
function readDotEnv(path) {
|
function readDotEnv(path) {
|
||||||
const data = fs.readFileSync(path, "utf8");
|
const data = fs.readFileSync(path, "utf8");
|
||||||
|
|
||||||
|
|
@ -417,7 +314,7 @@ function readDotEnv(path) {
|
||||||
|
|
||||||
function writeDotEnv(url) {
|
function writeDotEnv(url) {
|
||||||
// Read the .env file
|
// Read the .env file
|
||||||
let lines = readDotEnv(dotEnvFile);
|
let lines = readDotEnv(findDotEnv());
|
||||||
|
|
||||||
// Rebuild lines with the new url
|
// Rebuild lines with the new url
|
||||||
let found = false;
|
let found = false;
|
||||||
|
|
@ -443,5 +340,5 @@ function writeDotEnv(url) {
|
||||||
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.writeFileSync(dotEnvFile, writeData);
|
fs.writeFileSync(findDotEnv(), writeData);
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue