mirror of
https://github.com/Ylianst/MeshCentral.git
synced 2024-12-22 21:31:35 +03:00
Added customization support
This commit is contained in:
Bryan Roe
parent
10b094b9eb
commit
e4a0a31f86
@ -14,63 +14,83 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
*/
|
||||
|
||||
|
||||
|
||||
// The folloing line just below with 'msh=' needs to stay exactly like this since MeshCentral will replace it with the correct settings.
|
||||
var msh = {};
|
||||
var displayName = msh.displayName ? msh.displayName : 'MeshCentral Agent';
|
||||
var s = null, buttons = ['Cancel'], skip = false;
|
||||
try { s = require('service-manager').manager.getService('meshagent'); } catch (e) { }
|
||||
var serviceName = msh.meshServiceName ? msh.meshServiceName : 'meshagent';
|
||||
|
||||
function _install(parms) {
|
||||
try { s = require('service-manager').manager.getService(serviceName); } catch (e) { }
|
||||
|
||||
var connectArgs = [process.execPath.split('/').pop(), '--no-embedded=1', '--disableUpdate=1'];
|
||||
connectArgs.push('--MeshName="' + msh.MeshName + '"');
|
||||
connectArgs.push('--MeshType="' + msh.MeshType + '"');
|
||||
connectArgs.push('--MeshID="' + msh.MeshID + '"');
|
||||
connectArgs.push('--ServerID="' + msh.ServerID + '"');
|
||||
connectArgs.push('--MeshServer="' + msh.MeshServer + '"');
|
||||
connectArgs.push('--AgentCapabilities="0x00000020"');
|
||||
if (msh.displayName) { connectArgs.push('--displayName="' + msh.displayName + '"'); }
|
||||
if (msh.agentName) { connectArgs.push('--agentName="' + msh.agentName + '"'); }
|
||||
|
||||
function _install(parms)
|
||||
{
|
||||
var mstr = require('fs').createWriteStream(process.execPath + '.msh', { flags: 'wb' });
|
||||
mstr.write('MeshName=' + msh.MeshName + '\n');
|
||||
mstr.write('MeshType=' + msh.MeshType + '\n');
|
||||
mstr.write('MeshID=' + msh.MeshID + '\n');
|
||||
mstr.write('ServerID=' + msh.ServerID + '\n');
|
||||
mstr.write('MeshServer=' + msh.MeshServer + '\n');
|
||||
if (msh.agentName) { mstr.write('agentName=' + msh.agentName + '\n'); }
|
||||
if (msh.meshServiceName) { mstr.write('meshServiceName=' + msh.meshServiceName + '\n'); }
|
||||
mstr.end();
|
||||
|
||||
|
||||
if (parms == null) { parms = []; }
|
||||
if (msh.companyName) { parms.unshift('--companyName="' + msh.companyName + '"'); }
|
||||
if (msh.displayName) { parms.unshift('--displayName="' + msh.displayName + '"'); }
|
||||
if (msh.meshServiceName) { parms.unshift('--meshServiceName="' + msh.meshServiceName + '"'); }
|
||||
parms.unshift('--copy-msh=1');
|
||||
parms.unshift('--no-embedded=1');
|
||||
parms.unshift('-fullinstall');
|
||||
parms.unshift(process.execPath.split('/').pop());
|
||||
|
||||
|
||||
global._child = require('child_process').execFile(process.execPath, parms);
|
||||
global._child.stdout.on('data', function (c) { process.stdout.write(c.toString()); });
|
||||
global._child.stderr.on('data', function (c) { process.stdout.write(c.toString()); });
|
||||
global._child.waitExit();
|
||||
}
|
||||
|
||||
function _uninstall() {
|
||||
function _uninstall()
|
||||
{
|
||||
global._child = require('child_process').execFile(process.execPath,
|
||||
[process.execPath.split('/').pop(), '-fulluninstall', '--no-embedded=1']);
|
||||
|
||||
[process.execPath.split('/').pop(), '-fulluninstall', '--no-embedded=1', '--meshServiceName="' + serviceName + '"']);
|
||||
|
||||
global._child.stdout.on('data', function (c) { process.stdout.write(c.toString()); });
|
||||
global._child.stderr.on('data', function (c) { process.stdout.write(c.toString()); });
|
||||
global._child.waitExit();
|
||||
}
|
||||
|
||||
if (msh.InstallFlags == null) {
|
||||
if (msh.InstallFlags == null)
|
||||
{
|
||||
msh.InstallFlags = 3;
|
||||
} else {
|
||||
} else
|
||||
{
|
||||
msh.InstallFlags = parseInt(msh.InstallFlags.toString());
|
||||
}
|
||||
|
||||
if (process.argv.includes('-mesh')) {
|
||||
if (process.argv.includes('-mesh'))
|
||||
{
|
||||
console.log(JSON.stringify(msh, null, 2));
|
||||
process.exit();
|
||||
}
|
||||
|
||||
if ((msh.InstallFlags & 1) == 1) {
|
||||
if ((msh.InstallFlags & 1) == 1)
|
||||
{
|
||||
buttons.unshift('Connect');
|
||||
if (process.argv.includes('-connect')) {
|
||||
global._child = require('child_process').execFile(process.execPath,
|
||||
[process.execPath.split('/').pop(), '--no-embedded=1', '--disableUpdate=1',
|
||||
'--MeshName="' + msh.MeshName + '"', '--MeshType="' + msh.MeshType + '"',
|
||||
'--MeshID="' + msh.MeshID + '"',
|
||||
'--ServerID="' + msh.ServerID + '"',
|
||||
'--MeshServer="' + msh.MeshServer + '"',
|
||||
'--AgentCapabilities="0x00000020"']);
|
||||
|
||||
if (process.argv.includes('-connect'))
|
||||
{
|
||||
global._child = require('child_process').execFile(process.execPath, connectArgs);
|
||||
global._child.stdout.on('data', function (c) { });
|
||||
global._child.stderr.on('data', function (c) { });
|
||||
global._child.on('exit', function (code) { process.exit(code); });
|
||||
@ -82,54 +102,72 @@ if ((msh.InstallFlags & 1) == 1) {
|
||||
}
|
||||
}
|
||||
|
||||
if ((!skip) && ((msh.InstallFlags & 2) == 2)) {
|
||||
if (!require('user-sessions').isRoot()) {
|
||||
if ((!skip) && ((msh.InstallFlags & 2) == 2))
|
||||
{
|
||||
if (!require('user-sessions').isRoot())
|
||||
{
|
||||
console.log('\n' + "Elevated permissions is required to install/uninstall the agent.");
|
||||
console.log("Please try again with sudo.");
|
||||
process.exit();
|
||||
}
|
||||
if (s) {
|
||||
if ((process.platform == 'darwin') || require('message-box').kdialog) {
|
||||
if (s)
|
||||
{
|
||||
if ((process.platform == 'darwin') || require('message-box').kdialog)
|
||||
{
|
||||
buttons.unshift("Setup");
|
||||
} else {
|
||||
} else
|
||||
{
|
||||
buttons.unshift("Uninstall");
|
||||
buttons.unshift("Update");
|
||||
}
|
||||
} else {
|
||||
} else
|
||||
{
|
||||
buttons.unshift("Install");
|
||||
}
|
||||
}
|
||||
|
||||
if (!skip) {
|
||||
if (process.platform != 'darwin') {
|
||||
if (process.argv.includes('-install') || process.argv.includes('-update')) {
|
||||
if (!skip)
|
||||
{
|
||||
if (process.platform != 'darwin')
|
||||
{
|
||||
if (process.argv.includes('-install') || process.argv.includes('-update'))
|
||||
{
|
||||
var p = [];
|
||||
for (var i = 0; i < process.argv.length; ++i) {
|
||||
if (process.argv[i].startsWith('--installPath=')) {
|
||||
for (var i = 0; i < process.argv.length; ++i)
|
||||
{
|
||||
if (process.argv[i].startsWith('--installPath='))
|
||||
{
|
||||
p.push('--installPath="' + process.argv[i].split('=').pop() + '"');
|
||||
}
|
||||
}
|
||||
_install(p);
|
||||
process.exit();
|
||||
}
|
||||
else if (process.argv.includes('-uninstall')) {
|
||||
else if (process.argv.includes('-uninstall'))
|
||||
{
|
||||
_uninstall();
|
||||
process.exit();
|
||||
}
|
||||
else {
|
||||
if (!require('message-box').kdialog && ((require('message-box').zenity == null) || (!require('message-box').zenity.extra))) {
|
||||
else
|
||||
{
|
||||
if (!require('message-box').kdialog && ((require('message-box').zenity == null) || (!require('message-box').zenity.extra)))
|
||||
{
|
||||
console.log('\n' + "The graphical version of this installer cannot run on this system.");
|
||||
console.log("Try installing/updating Zenity, and run again." + '\n');
|
||||
console.log("You can also run the text version from the command line with the following command(s): ");
|
||||
if ((msh.InstallFlags & 1) == 1) {
|
||||
if ((msh.InstallFlags & 1) == 1)
|
||||
{
|
||||
console.log('./' + process.execPath.split('/').pop() + ' -connect');
|
||||
}
|
||||
if ((msh.InstallFlags & 2) == 2) {
|
||||
if (s) {
|
||||
if ((msh.InstallFlags & 2) == 2)
|
||||
{
|
||||
if (s)
|
||||
{
|
||||
console.log('./' + process.execPath.split('/').pop() + ' -update');
|
||||
console.log('./' + process.execPath.split('/').pop() + ' -uninstall');
|
||||
}
|
||||
else {
|
||||
else
|
||||
{
|
||||
console.log('./' + process.execPath.split('/').pop() + ' -install');
|
||||
console.log('./' + process.execPath.split('/').pop() + ' -install --installPath="/alternate/path"');
|
||||
}
|
||||
@ -139,32 +177,40 @@ if (!skip) {
|
||||
}
|
||||
}
|
||||
}
|
||||
else {
|
||||
else
|
||||
{
|
||||
if (!require('user-sessions').isRoot()) { console.log('\n' + "This utility requires elevated permissions. Please try again with sudo."); process.exit(); }
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
if (!skip) {
|
||||
if (!s) {
|
||||
if (!skip)
|
||||
{
|
||||
if (!s)
|
||||
{
|
||||
msg = "Agent: " + "NOT INSTALLED" + '\n';
|
||||
} else {
|
||||
} else
|
||||
{
|
||||
msg = "Agent: " + (s.isRunning() ? "RUNNING" : "NOT RUNNING") + '\n';
|
||||
}
|
||||
|
||||
|
||||
msg += ("Device Group: " + msh.MeshName + '\n');
|
||||
msg += ("Server URL: " + msh.MeshServer + '\n');
|
||||
|
||||
var p = require('message-box').create("MeshCentral Agent Setup", msg, 99999, buttons);
|
||||
p.then(function (v) {
|
||||
switch (v) {
|
||||
|
||||
var p = require('message-box').create(displayName + " Setup", msg, 99999, buttons);
|
||||
p.then(function (v)
|
||||
{
|
||||
switch (v)
|
||||
{
|
||||
case "Cancel":
|
||||
process.exit();
|
||||
break;
|
||||
case 'Setup':
|
||||
var d = require('message-box').create("MeshCentral Agent", msg, 99999, ['Update', 'Uninstall', 'Cancel']);
|
||||
d.then(function (v) {
|
||||
switch (v) {
|
||||
var d = require('message-box').create(displayName, msg, 99999, ['Update', 'Uninstall', 'Cancel']);
|
||||
d.then(function (v)
|
||||
{
|
||||
switch (v)
|
||||
{
|
||||
case 'Update':
|
||||
case 'Install':
|
||||
_install();
|
||||
@ -179,28 +225,23 @@ if (!skip) {
|
||||
}).catch(function (v) { process.exit(); });
|
||||
break;
|
||||
case "Connect":
|
||||
global._child = require('child_process').execFile(process.execPath,
|
||||
[process.execPath.split('/').pop(), '--no-embedded=1', '--disableUpdate=1',
|
||||
'--MeshName="' + msh.MeshName + '"', '--MeshType="' + msh.MeshType + '"',
|
||||
'--MeshID="' + msh.MeshID + '"',
|
||||
'--ServerID="' + msh.ServerID + '"',
|
||||
'--MeshServer="' + msh.MeshServer + '"',
|
||||
'--AgentCapabilities="0x00000020"']);
|
||||
|
||||
global._child = require('child_process').execFile(process.execPath, connectArgs);
|
||||
global._child.stdout.on('data', function (c) { });
|
||||
global._child.stderr.on('data', function (c) { });
|
||||
global._child.on('exit', function (code) { process.exit(code); });
|
||||
|
||||
|
||||
msg = ("Device Group: " + msh.MeshName + '\n');
|
||||
msg += ("Server URL: " + msh.MeshServer + '\n');
|
||||
|
||||
if (process.platform != 'darwin') {
|
||||
if (!require('message-box').zenity && require('message-box').kdialog) {
|
||||
|
||||
if (process.platform != 'darwin')
|
||||
{
|
||||
if (!require('message-box').zenity && require('message-box').kdialog)
|
||||
{
|
||||
msg += ('\nPress OK to Disconnect');
|
||||
}
|
||||
}
|
||||
|
||||
var d = require('message-box').create("MeshCentral Agent", msg, 99999, ['Disconnect']);
|
||||
|
||||
var d = require('message-box').create(displayName, msg, 99999, ['Disconnect']);
|
||||
d.then(function (v) { process.exit(); }).catch(function (v) { process.exit(); });
|
||||
break;
|
||||
case "Uninstall":
|
||||
@ -217,7 +258,8 @@ if (!skip) {
|
||||
process.exit();
|
||||
break;
|
||||
}
|
||||
}).catch(function (e) {
|
||||
}).catch(function (e)
|
||||
{
|
||||
console.log(e);
|
||||
process.exit();
|
||||
});
|
||||
|
||||
Loading…
Reference in New Issue
Block a user