Agama/main.js

// main proc for EasyDEX GUI
// this app spawns iguana in background in nontech-mode

const electron = require('electron'),
			app = electron.app,
			BrowserWindow = electron.BrowserWindow,
			path = require('path'),
			url = require('url'),
			os = require('os'),
			md5 = require('md5'),
			spawn = require('child_process').spawn,
			exec = require('child_process').exec,
			{ Menu } = require('electron'),
			portscanner = require('portscanner'),
			osPlatform = os.platform(),
			fixPath = require('fix-path');

var express = require('express'),
		bodyParser = require('body-parser'),
		fs = require('fs'),
		fsnode = require('fs'),
		fs = require('fs-extra'),
		mkdirp = require('mkdirp'),
		pm2 = require('pm2'),
		cluster = require('cluster'),
		numCPUs = require('os').cpus().length,
		ipc = require('electron').ipcMain;

Promise = require('bluebird');

// read app version
const localVersionFile = fs.readFileSync(`version`, 'utf8');
let localVersion;

if (localVersionFile.indexOf('\r\n') > -1) {
  localVersion = localVersionFile.split('\r\n');
} else {
  localVersion = localVersionFile.split('\n');
}

const appBasicInfo = {
	name: 'Agama',
	version: localVersion[0],
};

app.setName(appBasicInfo.name);
app.setVersion(appBasicInfo.version);

if (osPlatform === 'linux') {
	process.env.ELECTRON_RUN_AS_NODE = true;
	// console.log(process.env);
}

// GUI APP settings and starting gui on address http://120.0.0.1:17777
var shepherd = require('./routes/shepherd');
var guiapp = express();

shepherd.createIguanaDirs();

const appSessionHash = md5(Date.now());

shepherd.writeLog(`app init ${appSessionHash}`);
shepherd.writeLog(`app info: ${appBasicInfo.name} ${appBasicInfo.version}`);
shepherd.writeLog('sys info:');
shepherd.writeLog(`totalmem_readable: ${formatBytes(os.totalmem())}`);
shepherd.writeLog(`arch: ${os.arch()}`);
shepherd.writeLog(`cpu: ${os.cpus()[0].model}`);
shepherd.writeLog(`cpu_cores: ${os.cpus().length}`);
shepherd.writeLog(`platform: ${osPlatform}`);
shepherd.writeLog(`os_release: ${os.release()}`);
shepherd.writeLog(`os_type: ${os.type()}`);

var appConfig = shepherd.loadLocalConfig(); // load app config

shepherd.writeLog(`app started in ${(appConfig.dev ? 'dev mode' : ' user mode')}`);

shepherd.setConfKMD();

// kill rogue iguana copies on start
if (appConfig.killIguanaOnStart) {
	let iguanaGrep;

	switch (osPlatform) {
		case 'darwin':
			iguanaGrep = "ps -p $(ps -A | grep -m1 iguana | awk '{print $1}') | grep -i iguana";
			break;
		case 'linux':
			iguanaGrep = 'ps -p $(pidof iguana) | grep -i iguana';
			break;
		case 'win32':
			iguanaGrep = 'tasklist';
			break;
	}

	exec(iguanaGrep, function(error, stdout, stderr) {
		if (stdout.indexOf('iguana') > -1) {
			const pkillCmd = osPlatform === 'win32' ? 'taskkill /f /im iguana.exe' : 'pkill -15 iguana';

			console.log('found another iguana process(es)');
			shepherd.writeLog('found another iguana process(es)');

			exec(pkillCmd, function(error, stdout, stderr) {
				console.log(`${pkillCmd} is issued`);
				shepherd.writeLog(`${pkillCmd} is issued`);

				if (error !== null) {
					console.log(`${pkillCmd} exec error: ${error}`);
					shepherd.writeLog(`${pkillCmd} exec error: ${error}`);
				};
			});
		}

		if (error !== null) {
			console.log(`${iguanaGrep} exec error: ${error}`);
			shepherd.writeLog(`${iguanaGrep} exec error: ${error}`);
		};
	});
}

guiapp.use(function(req, res, next) {
	res.header('Access-Control-Allow-Origin', appConfig.dev ? '*' : 'http://127.0.0.1:3000');
	res.header('Access-Control-Allow-Headers', 'X-Requested-With');
	res.header('Access-Control-Allow-Credentials', 'true');
	res.header('Access-Control-Allow-Headers', 'Content-Type');
	res.header('Access-Control-Allow-Methods', 'PUT, GET, POST, DELETE, OPTIONS');
	next();
});

// preload.js
const _setImmediate = setImmediate,
			_clearImmediate = clearImmediate;

process.once('loaded', () => {
	global.setImmediate = _setImmediate;
	global.clearImmediate = _clearImmediate;

	if (osPlatform === 'darwin') {
		process.setFdLimit(appConfig.maxDescriptors.darwin);
		app.setAboutPanelOptions({
			applicationName: app.getName(),
			applicationVersion: app.getVersion(),
			copyright: 'Released under the MIT license',
			credits: 'SuperNET Team'
		})
	}
	if (osPlatform === 'linux') {
		process.setFdLimit(appConfig.maxDescriptors.linux);
	}
});

guiapp.use(bodyParser.json({ limit: '50mb' })); // support json encoded bodies
guiapp.use(bodyParser.urlencoded({
	limit: '50mb',
	extended: true
})); // support encoded bodies

guiapp.get('/', function (req, res) {
	res.send('Agama app server');
});

const guipath = path.join(__dirname, '/gui');
guiapp.use('/gui', express.static(guipath));
guiapp.use('/shepherd', shepherd);

const server = require('http').createServer(guiapp);
const io = require('socket.io').listen(server);
let willQuitApp = false;

// check if agama is already running
portscanner.checkPortStatus(appConfig.agamaPort, '127.0.0.1', function(error, status) {
	// Status is 'open' if currently in use or 'closed' if available
	if (status === 'closed') {
		server.listen(appConfig.agamaPort, function() {
			console.log(`guiapp and sockets.io are listening on port ${appConfig.agamaPort}`);
			shepherd.writeLog(`guiapp and sockets.io are listening on port ${appConfig.agamaPort}`);
		});
	} else {
		willQuitApp = true;
		server.listen(appConfig.agamaPort + 1, function() {
			console.log(`guiapp and sockets.io are listening on port ${appConfig.agamaPort + 1}`);
			shepherd.writeLog(`guiapp and sockets.io are listening on port ${appConfig.agamaPort + 1}`);
		});
		loadingWindow.loadURL(`http://${appConfig.host}:${appConfig.agamaPort + 1}/gui/agama-instance-error.html`);
		console.log('another agama app is already running');
	}
});

io.set('origins', appConfig.dev ? 'http://127.0.0.1:3000' : `http://127.0.0.1:${appConfig.agamaPort}`); // set origin

io.on('connection', function(client) {
	console.log('EDEX GUI is connected...');
	shepherd.writeLog('EDEX GUI is connected...');

	client.on('event', function(data) { // listen for client requests
		console.log(data);
	});
	client.on('disconnect', function(data) {
		console.log('EDEX GUI is disconnected');
	});
	client.on('join', function(data) {
		console.log(data);
		client.emit('messages', 'Sockets server is listening');
	});
});

shepherd.setIO(io); // pass sockets object to shepherd router
shepherd.setVar('appBasicInfo', appBasicInfo);
shepherd.setVar('appSessionHash', appSessionHash);

module.exports = guiapp;
var iguanaIcon;

if (os.platform() === 'linux') {
	iguanaIcon = path.join(__dirname, '/assets/icons/agama_icons/128x128.png');
}
if (os.platform() === 'win32') {
	iguanaIcon = path.join(__dirname, '/assets/icons/agama_icons/agama_app_icon.ico');
}

let mainWindow;
let loadingWindow;
let closeAppAfterLoading = false;

function createLoadingWindow() {
	mainWindow = null;

	// initialise window
	loadingWindow = new BrowserWindow({
		width: 500,
		height: 300,
		frame: false,
		icon: iguanaIcon
	});

	loadingWindow.createWindow = createWindow; // expose createWindow to front-end scripts
	loadingWindow.appConfig = appConfig;
	loadingWindow.forseCloseApp = forseCloseApp;

	// load our index.html (i.e. easyDEX GUI)
	loadingWindow.loadURL(`http://${appConfig.host}:${appConfig.agamaPort}/gui/`);
	shepherd.writeLog('show loading window');

	// DEVTOOLS - only for dev purposes - ca333
	// loadingWindow.webContents.openDevTools()

	// if window closed we kill iguana proc
	loadingWindow.on('hide', function () {
		// our app does not have multiwindow - so we dereference the window object instead of
		// putting them into an window_arr
		loadingWindow = null;
	});

  loadingWindow.on('close', (e) => {
    if (willQuitApp) {
      /* the user tried to quit the app */
      loadingWindow = null;
    } else {
      /* the user only tried to close the window */
      closeAppAfterLoading = true;
      e.preventDefault();
    }
  });
}

// close app
function forseCloseApp() {
	loadingWindow = null;
	mainWindow = null;
	app.quit();
}

app.on('ready', createLoadingWindow);

function createWindow (status) {
	if (status === 'open') {
		require(path.join(__dirname, 'private/mainmenu'));

		// initialise window
		mainWindow = new BrowserWindow({ // dirty hack to prevent main window flash on quit
			width: closeAppAfterLoading ? 1 : 1280,
			height: closeAppAfterLoading ? 1 : 800,
			icon: iguanaIcon
		});

		if (closeAppAfterLoading) {
			mainWindow = null;
			loadingWindow = null;
			pm2Exit();
		}

		const staticMenu = Menu.buildFromTemplate([ // if static
			{ role: 'copy' },
			{ type: 'separator' },
			{ role: 'selectall' },
		]);

		const editMenu = Menu.buildFromTemplate([ // if editable
			{ role: 'undo' },
			{ role: 'redo' },
			{ type: 'separator' },
			{ role: 'cut' },
			{ role: 'copy' },
			{ role: 'paste' },
			{ type: 'separator' },
			{ role: 'selectall' },
		]);

		// load our index.html (i.e. easyDEX GUI)
		if (appConfig.edexGuiOnly) {
			if (appConfig.v2) {
				shepherd.writeLog('show edex gui');
				mainWindow.appConfig = appConfig;
				mainWindow.appBasicInfo = appBasicInfo;
				mainWindow.appSessionHash = appSessionHash;

				if (appConfig.dev) {
					mainWindow.loadURL('http://127.0.0.1:3000');
				} else {
					mainWindow.loadURL(`http://${appConfig.host}:${appConfig.agamaPort}/gui/EasyDEX-GUI/react/build`);
				}
			} else {
				shepherd.writeLog('show edex gui');
				mainWindow.loadURL(`http://${appConfig.host}:${appConfig.agamaPort}/gui/EasyDEX-GUI/`);
			}
		} else {
			mainWindow.loadURL(`http://${appConfig.host}:${appConfig.agamaPort}/gui/main.html`);
		}

		mainWindow.webContents.on('context-menu', (e, params) => { // context-menu returns params
			const { selectionText, isEditable } = params; // params obj

			if (isEditable) {
				editMenu.popup(mainWindow);
			} else if (selectionText && selectionText.trim() !== '') {
				staticMenu.popup(mainWindow);
			}
		});

		// DEVTOOLS - only for dev purposes - ca333
		// mainWindow.webContents.openDevTools()

		function pm2Exit() {
			const ConnectToPm2 = function() {
				return new Promise(function(resolve, reject) {
					console.log('Closing Main Window...');
					shepherd.writeLog('exiting app...');

					shepherd.dumpCacheBeforeExit();
					shepherd.quitKomodod();

					pm2.connect(true, function(err) {
						console.log('connecting to pm2...');
						shepherd.writeLog('connecting to pm2...');

						if (err) {
							console.log(err);
						}
					});

					const result = 'Connecting To Pm2: done';

					console.log(result);
					shepherd.writeLog(result);
					resolve(result);
				})
			}

			const KillPm2 = function() {
				return new Promise(function(resolve, reject) {
					console.log('killing to pm2...');
					shepherd.writeLog('killing to pm2...');

					pm2.killDaemon(function(err) {
						pm2.disconnect();
						console.log('killed to pm2...');
						shepherd.writeLog('killed to pm2...');

						if (err)
							throw err;
					});

					const result = 'Killing Pm2: done';

					setTimeout(function() {
						console.log(result);
						shepherd.writeLog(result);

						resolve(result);
					}, 2000)
				})
			}

			const HideMainWindow = function() {
				return new Promise(function(resolve, reject) {
					console.log('Exiting App...');
					mainWindow = null;

					const result = 'Hiding Main Window: done';
					console.log(result);
					resolve(result);
				});
			}

			const QuitApp = function() {
				return new Promise(function(resolve, reject) {
					KillPm2(); // required for normal app quit in iguana-less mode
					app.quit();
					const result = 'Quiting App: done';
					console.log(result);
					resolve(result);
				});
			}

			ConnectToPm2()
			.then(function(result) {
				return KillPm2();
			})
			.then(HideMainWindow)
			.then(QuitApp);
		}

		// if window closed we kill iguana proc
		mainWindow.on('closed', function () {
			pm2Exit();
		});
	}
}

app.on('window-all-closed', function () {
	//if (os.platform() !== 'win32') { ig.kill(); }
	// in osx apps stay active in menu bar until explictly closed or quitted by CMD Q
	// so we do not kill the app --> for the case user clicks again on the iguana icon
	// we open just a new window and respawn iguana proc
	/*if (process.platform !== 'darwin' || process.platform !== 'linux' || process.platform !== 'win32') {
		app.quit()
	}*/
})

// Emitted before the application starts closing its windows.
// Calling event.preventDefault() will prevent the default behaviour, which is terminating the application.
app.on('before-quit', function (event) {
	console.log('before-quit');
	if (mainWindow === null && loadingWindow != null) { // mainWindow not intitialised and loadingWindow not dereferenced
		// loading window is still open
		console.log('before-quit prevented');
		shepherd.writeLog('quit app after loading is done');
		closeAppAfterLoading = true;
		let code = `$('#loading_status_text').html('Preparing to shutdown the wallet.<br/>Please wait while all daemons are closed...')`;
		loadingWindow.webContents.executeJavaScript(code);
		event.preventDefault();
	}
});

// Emitted when all windows have been closed and the application will quit.
// Calling event.preventDefault() will prevent the default behaviour, which is terminating the application.
app.on('will-quit', function (event) {
	if (mainWindow === null && loadingWindow != null) {
		// loading window is still open
		console.log('will-quit while loading window active');
		event.preventDefault();
	}
});

// Emitted when the application is quitting.
// Calling event.preventDefault() will prevent the default behaviour, which is terminating the application.
app.on('quit', function (event) {
	if (mainWindow === null && loadingWindow != null) {
		console.log('quit while loading window active');
		event.preventDefault();
	}
})

app.on('activate', function () {
	if (mainWindow === null) {
		// createWindow('open');
	}
});

function formatBytes(bytes, decimals) {
  if (bytes === 0)
    return '0 Bytes';

  const k = 1000,
	      dm = decimals + 1 || 3,
	      sizes = [
	        'Bytes',
	        'KB',
	        'MB',
	        'GB',
	        'TB',
	        'PB',
	        'EB',
	        'ZB',
	        'YB'
	      ],
	      i = Math.floor(Math.log(bytes) / Math.log(k));

  return `${parseFloat((bytes / Math.pow(k, i)).toFixed(dm))} ${sizes[i]}`;
}