forked from rotki/rotki
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.js
118 lines (98 loc) · 3.26 KB
/
main.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
const electron = require('electron');
const app = electron.app;
const ipcMain = electron.ipcMain;
const BrowserWindow = electron.BrowserWindow;
const path = require('path');
/*************************************************************
* window management
*************************************************************/
let mainWindow = null;
const createWindow = () => {
mainWindow = new BrowserWindow({width: 800, height: 600});
mainWindow.loadURL(require('url').format({
pathname: path.join(__dirname, 'ui', 'index.html'),
protocol: 'file:',
slashes: true
}));
mainWindow.webContents.openDevTools();
mainWindow.on('closed', () => {
mainWindow = null;
});
};
app.on('ready', createWindow);
app.on('window-all-closed', () => {
if (process.platform !== 'darwin') {
app.quit();
}
});
app.on('activate', () => {
if (mainWindow === null) {
createWindow();
}
});
/*************************************************************
* py process
*************************************************************/
const PY_DIST_FOLDER = 'NOTAPPPLICABLE';
const PY_FOLDER = 'rotkehlchen';
const PY_MODULE = 'server'; // without .py suffix
let pyProc = null;
let pyPort = null;
const guessPackaged = () => {
const fullPath = path.join(__dirname, PY_DIST_FOLDER);
return require('fs').existsSync(fullPath);
};
const getScriptPath = () => {
if (!guessPackaged()) {
return path.join(__dirname, PY_FOLDER, PY_MODULE + '.py');
}
if (process.platform === 'win32') {
return path.join(__dirname, PY_DIST_FOLDER, PY_MODULE, PY_MODULE + '.exe');
}
return path.join(__dirname, PY_DIST_FOLDER, PY_MODULE, PY_MODULE);
};
const selectPort = () => {
pyPort = 4242;
return pyPort;
};
var pyproc_fail_notifier = null;
// Listen for ack messages from renderer process
ipcMain.on('ack', (event, arg) => {
// when ack is received stop the pyproc fail notifier
clearInterval(pyproc_fail_notifier);
});
const createPyProc = () => {
let script = getScriptPath();
let port = '' + selectPort();
if (guessPackaged()) {
console.log("At guess packaged");
pyProc = require('child_process').execFile(script, [port]);
} else {
console.log("At not packaged: script:" + script + " port: " + port);
pyProc = require('child_process').spawn('python', ["-m", "rotkehlchen", "--zerorpc-port", port, "--ethrpc-port", "8545"]);
}
pyProc.on('error', (err) => {
console.log('Failed to start python subprocess.');
});
pyProc.on('exit', function (code, signal) {
console.log("python subprocess killed with signal " + signal + " and code " +code);
if (code != 0) {
// Notify main window every 2 seconds until it acks the notification
pyproc_fail_notifier = setInterval(function () {
mainWindow.webContents.send('failed', 'failed');
}, 2000);
}
});
if (pyProc != null) {
console.log('child process success on port ' + port);
}
console.log("CREATED PYPROCESS");
};
const exitPyProc = () => {
console.log("KILLING PYPROCESS");
pyProc.kill();
pyProc = null;
pyPort = null;
};
app.on('ready', createPyProc);
app.on('will-quit', exitPyProc);