From 6503857d0ce73d09242cb2908fb8bb659a3313ac Mon Sep 17 00:00:00 2001
From: stelcheck <mtrudel@wizcorp.jp>
Date: Sat, 29 Jul 2017 14:02:39 +0900
Subject: [PATCH] Configurable socket path

Fixes #6
---
 Readme.md | 16 ++++++++++++++
 index.js  | 62 ++++++++++++++++++++++++++++++++++++++++---------------
 2 files changed, 61 insertions(+), 17 deletions(-)

diff --git a/Readme.md b/Readme.md
index 165e184..9d946a7 100755
--- a/Readme.md
+++ b/Readme.md
@@ -66,6 +66,22 @@ If your project is a TypeScript project, set the value to `ts-mage-console` inst
 
 Then simply run `npm run develop` like you would normally do.
 
+Configuration
+--------------
+
+You may configure `mage-console` by adding configuration entries into your MAGE
+configuration. See [MAGE's documentation](https://mage.github.io/mage/#configuration) for more details.
+
+> config/default.yaml
+
+```yaml
+external:
+  mage-console:
+    # Where to put the socket file for the console.
+    # Default: [project-dir]/node_modules/mage-console/mage-console.sock
+    sockfile: /tmp/mage-console.sock
+```
+
 Debugging
 ---------
 
diff --git a/index.js b/index.js
index 39739e4..ec7fbd9 100755
--- a/index.js
+++ b/index.js
@@ -14,6 +14,18 @@ var APP_LIB_PATH = path.join(process.cwd(), 'lib');
 
 var debug = require('./debug');
 
+function watchFiles(onChange) {
+	watch(APP_LIB_PATH, {
+		recursive: true
+	}, function (event, name) {
+		if (name.split(path.sep).pop()[0] === '.') {
+			return;
+		}
+
+		onChange(event, name);
+	});
+}
+
 /**
  * exports.eval can be used to customise how the code and commands
  * will be interpreted in the REPL interface - null means default node
@@ -35,13 +47,7 @@ exports.debugFlag = '--debug';
  * Boot mage
  */
 var mage = require(APP_LIB_PATH);
-
-process.on('uncaughtException', function (error) {
-	if (!mage.logger) {
-		console.error(error);
-		process.exit(-1);
-	}
-});
+var processManager = mage.core.processManager;
 
 mage.boot();
 
@@ -56,24 +62,27 @@ if (clusterConfiguration !== 1) {
 	process.exit(-1);
 }
 
+var logger = mage.core.logger.context('REPL');
+
 function crash(error) {
-	console.error(error);
-	mage.quit(-1);
+	mage.logger.error(error);
+	mage.quit();
 }
 
-var logger = mage.core.logger.context('REPL');
-
 /**
  * @summary Retrieve the IPC path on which to listen/connect to
  * @returns {String} path the master process will listen on
  */
 function getIPCPath() {
+	// See https://github.com/nodejs/node/issues/13670 for more details
+	const defaultFilepath = path.relative(process.cwd(), path.join(__dirname, 'mage-console.sock'));
+	const filepath = mage.core.config.get('external.mage-console.sockfile') || defaultFilepath;
+
 	if (process.platform === 'win32') {
-		return path.join('\\\\.\\pipe', __dirname, 'mage-console.sock');
+		return path.join('\\\\.\\pipe', filepath);
 	}
 
-	// See https://github.com/nodejs/node/issues/13670 for more details
-	return path.relative(process.cwd(), path.join(__dirname, 'mage-console.sock'));
+	return filepath;
 }
 
 var ipcPath = getIPCPath();
@@ -189,9 +198,7 @@ function connect() {
 		});
 
 		// Watch the lib folder for changes
-		watch(APP_LIB_PATH, {
-			recursive: true
-		}, function (event, name) {
+		watchFiles(function (event, name) {
 			logger.debug('File ' + name + ' was ' + event + 'd, reloading');
 			saveHistory(repl.history);
 			process.send('reload');
@@ -215,6 +222,27 @@ if (cluster.isWorker) {
 // to connect to, and patches stdin/stdout/stderr into
 // the connection
 
+cluster.removeAllListeners('exit');
+cluster.on('exit', function (worker) {
+	// check if the worker was supposed to die
+
+	if (!worker._mageManagedExit) {
+		// this exit was not supposed to happen!
+		// spawn a new worker to replace the dead one
+
+		processManager.emit('workerOffline', worker.id);
+		watchFiles(function (event, name) {
+			logger.debug('File ' + name + ' was ' + event + 'd, reloading');
+			processManager.createWorker();
+		});
+	} else {
+		if (!processManager.getNumWorkers()) {
+			logger.emergency('All workers have shut down, shutting down master now.');
+			process.exit(0);
+		}
+	}
+});
+
 // Clean up old lingering sockets
 try {
 	fs.unlinkSync(ipcPath);