index.js

/* global process, require, __filename, __dirname, Promise */
"use strict";

const cluster = require('cluster');
const path = require('path');

const _ = require(path.resolve(__dirname,'./lib/underscore-with-mixins'));
const coreClient = require(path.resolve(__dirname,'./lib/core-client'));
const coreHost = require(path.resolve(__dirname,'./lib/core-host'));
const coreProcessManager = require(path.resolve(__dirname,'./lib/core-process-manager'));

module.exports = exports = (target)=>{
	if(_.isObject(target) && 'constructor' in target && 'client' in target.constructor && target.constructor.client instanceof coreClient.requireWorkerClient){
		return target.constructor.client;
	}
	if(_.isString(target))try{ target = require.resolve(target); }catch(err){}
	if(coreHost.hostsMap.has(target)) return coreHost.hostsMap.get(target);
	if(coreClient.clientsMap.size===0 && coreHost.hostsMap.size>0) throw Error("first argument must be a valid require-worker host or file path");
	if(!coreClient.clientsMap.has(target)) throw Error("first argument must be a valid require-worker client or file path");
	return coreClient.clientsMap.get(target);
};

var requireWorkerObj = { exports, __filename, coreClient, coreHost, coreProcessManager, cluster };

coreClient.setRequireWorker(requireWorkerObj);
exports.coreClient = coreClient;
exports.preConfiguredProxy = coreClient.preConfiguredProxy;

coreHost.setRequireWorker(requireWorkerObj);
exports.coreHost = coreHost;

coreProcessManager.setRequireWorker(requireWorkerObj);
exports.coreProcessManager = coreProcessManager;
exports.prepareProcesses = coreProcessManager.prepareProcesses;
exports.destroyPreparedProcesses = coreProcessManager.destroyPreparedProcesses;
exports.getPreparedProcessesCount = coreProcessManager.getPreparedProcessesCount;

exports.require = (path,options)=>new coreClient.requireWorkerClient(path,options);

requireWorkerObj.getStackFiles = function getStackFiles(){
	let opst = Error.prepareStackTrace, thisError, result = [];
	Error.prepareStackTrace = (errStackStr,cssfArr)=>cssfArr;
	thisError = new Error();
	Error.captureStackTrace(thisError,getStackFiles); // https://nodejs.org/api/errors.html#errors_new_error_message
	let cssfArr = thisError.stack;
	Error.prepareStackTrace = opst;
	for(let i=0,l=cssfArr.length; i<l; i++){
		let cssf = cssfArr[i]; // https://github.com/v8/v8/wiki/Stack-Trace-API
		let file = cssf.getFileName();
		if(file===__filename || file===coreClient.__filename || file===coreHost.__filename || file===coreProcessManager.__filename) continue;
		result.push(file);
	}
	return result; 
};

// Check if the current process is a require-worker host
if(require.main===module && process.argv.length>=4 && process.argv[2]==='--rwProcess'){
	if(cluster.isWorker && process.argv.length>=5) process.chdir(process.argv[4]);
	coreHost.initHostProcess({ ipcTransportID:process.argv[3] });
}