1
0
mirror of https://github.com/Ylianst/MeshAgent synced 2025-12-11 13:53:37 +00:00
Files
MeshAgent/modules/message-box.js
Bryan Roe dbbc47e751 1. Added work-around for Zenity timeout support detection for platforms that require DISPLAY to be set in env just to query metadata
2. Fixed bug where timeout work-around resulted in success instead of timeout
2020-04-28 03:03:18 -07:00

695 lines
28 KiB
JavaScript

/*
Copyright 2019 Intel Corporation
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
const MB_OK = 0x00000000;
const MB_OKCANCEL = 0x00000001;
const MB_ABORTRETRYIGNORE = 0x00000002;
const MB_YESNOCANCEL = 0x00000003;
const MB_YESNO = 0x00000004;
const MB_RETRYCANCEL = 0x00000005;
const MB_TOPMOST = 0x00040000;
const MB_SETFOREGROUND = 0x00010000;
const MB_DEFBUTTON1 = 0x00000000;
const MB_DEFBUTTON2 = 0x00000100;
const MB_DEFBUTTON3 = 0x00000200;
const MB_ICONHAND = 0x00000010;
const MB_ICONQUESTION = 0x00000020;
const MB_ICONEXCLAMATION = 0x00000030;
const MB_ICONASTERISK = 0x00000040;
const IDOK = 1;
const IDCANCEL = 2;
const IDABORT = 3;
const IDRETRY = 4;
const IDIGNORE = 5;
const IDYES = 6;
const IDNO = 7;
var promise = require('promise');
var childScript = "\
require('ScriptContainer').on('data', function (j)\
{\
switch(j.command)\
{\
case 'messageBox':\
if(process.platform == 'win32')\
{\
var GM = require('_GenericMarshal');\
var user32 = GM.CreateNativeProxy('user32.dll');\
user32.CreateMethod('MessageBoxA');\
user32.MessageBoxA.async(0, GM.CreateVariable(j.caption), GM.CreateVariable(j.title), j.layout).then(\
function(r)\
{\
switch(r.Val)\
{\
case " + IDOK.toString() + ":\
case " + IDCANCEL.toString() + ":\
case " + IDABORT.toString() + ":\
case " + IDRETRY.toString() + ":\
case " + IDIGNORE.toString() + ":\
case " + IDYES.toString() + ":\
require('ScriptContainer').send(r.Val);\
break;\
default:\
require('ScriptContainer').send(" + IDNO.toString() + ");\
break;\
}\
process.exit();\
});\
}\
break;\
}\
});\
";
function messageBox()
{
this._ObjectID = 'message-box';
this.create = function create(title, caption, timeout, layout)
{
if (layout == null)
{
layout = (MB_YESNO | MB_DEFBUTTON2 | MB_ICONEXCLAMATION | MB_TOPMOST);
}
else
{
layout = (MB_OK | MB_DEFBUTTON2 | MB_ICONEXCLAMATION | MB_TOPMOST);
}
var GM = require('_GenericMarshal');
var kernel32 = GM.CreateNativeProxy('kernel32.dll');
kernel32.CreateMethod('ProcessIdToSessionId');
var psid = GM.CreateVariable(4);
if (kernel32.ProcessIdToSessionId(process.pid, psid).Val == 0)
{
ret._rej('Internal Error');
return (ret);
}
if (timeout == null) { timeout = 10; }
var ret = new promise(function (res, rej) { this._res = res; this._rej = rej; });
var options = { executionTimeout: timeout };
try
{
options.sessionId = require('user-sessions').consoleUid();
if (options.sessionId == psid.toBuffer().readUInt32LE()) { delete options.sessionId; }
}
catch(ee)
{
ret._rej('No logged on users');
return (ret);
}
ret._title = title;
ret._caption = caption;
ret._container = require('ScriptContainer').Create(options);
ret._container.promise = ret;
ret._container.on('data', function (j)
{
if(j == IDYES || j == IDOK)
{
this.promise._res();
}
else
{
this.promise._rej('Denied');
}
});
ret._container.on('exit', function ()
{
this.promise._rej('Timeout');
});
ret._container.ExecuteString(childScript);
ret._container.send({ command: 'messageBox', caption: caption, title: title, layout: layout });
return (ret);
};
}
function linux_messageBox()
{
this._ObjectID = 'message-box';
Object.defineProperty(this, 'zenity',
{
value: (function ()
{
var child = require('child_process').execFile('/bin/sh', ['sh']);
child.stdout.str = ''; child.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
child.stdin.write("whereis zenity | awk '{ print $2 }'\nexit\n");
child.waitExit();
var location = child.stdout.str.trim();
if (location == '' && require('fs').existsSync('/usr/local/bin/zenity')) { location = '/usr/local/bin/zenity'; }
if (location == '') { return (null); }
var ret = { path: location, timeout: child.stdout.str.trim() == '' ? false : true };
Object.defineProperty(ret, "timeout", {
get: function ()
{
var uid, xinfo;
try
{
uid = require('user-sessions').consoleUid();
xinfo = require('monitor-info').getXInfo(uid);
}
catch (e)
{
uid = 0;
xinfo = require('monitor-info').getXInfo(0);
}
if (xinfo == null) { return (false); }
var child = require('child_process').execFile('/bin/sh', ['sh'], { uid: uid, env: { XAUTHORITY: xinfo.xauthority ? xinfo.xauthority : "", DISPLAY: xinfo.display } });
child.stdout.str = ''; child.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
child.stdin.write(location + ' --help-all | grep timeout\nexit\n');
child.stderr.on('data', function (e) { console.log(e); });
child.waitExit();
return (child.stdout.str.trim() == '' ? false : true);
}
});
Object.defineProperty(ret, "version", {
get: function ()
{
var uid, xinfo;
try
{
uid = require('user-sessions').consoleUid();
xinfo = require('monitor-info').getXInfo(uid);
}
catch (e)
{
uid = 0;
xinfo = require('monitor-info').getXInfo(0);
}
if (xinfo == null) { return (false); }
var child = require('child_process').execFile('/bin/sh', ['sh'], { uid: uid, env: { XAUTHORITY: xinfo.xauthority ? xinfo.xauthority : "", DISPLAY: xinfo.display } });
child.stdout.str = ''; child.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
child.stdin.write(location + ' --version | awk -F. \'{ printf "[%s, %s]\\n", $1, $2; } \'\nexit\n');
child.waitExit();
try
{
return (JSON.parse(child.stdout.str.trim()));
}
catch (e)
{
return ([2, 16]);
}
}
});
return (ret);
})()
});
if (!this.zenity)
{
Object.defineProperty(this, 'kdialog',
{
value: (function ()
{
var child = require('child_process').execFile('/bin/sh', ['sh']);
child.stdout.str = ''; child.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
child.stdin.write("whereis kdialog | awk '{ print $2 }'\nexit\n");
child.waitExit();
return (child.stdout.str.trim() == '' ? null : { path: child.stdout.str.trim() });
})()
});
Object.defineProperty(this, 'xmessage',
{
value: (function ()
{
var child = require('child_process').execFile('/bin/sh', ['sh']);
child.stdout.str = ''; child.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
child.stdin.write("whereis xmessage | awk '{ print $2 }'\nexit\n");
child.waitExit();
return (child.stdout.str.trim() == '' ? null : { path: child.stdout.str.trim() });
})()
});
}
else
{
Object.defineProperty(this, 'notifysend',
{
value: (function ()
{
var child = require('child_process').execFile('/bin/sh', ['sh']);
child.stdout.str = ''; child.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
child.stdin.write("whereis notify-send | awk '{ print $2 }'\nexit\n");
child.waitExit();
return (child.stdout.str.trim() == '' ? null : { path: child.stdout.str.trim() });
})()
});
}
this.create = function create(title, caption, timeout, layout)
{
if (timeout == null) { timeout = 10; }
var ret = new promise(function (res, rej) { this._res = res; this._rej = rej; });
var uid;
var xinfo;
try
{
uid = require('user-sessions').consoleUid();
xinfo = require('monitor-info').getXInfo(uid);
}
catch(e)
{
uid = 0;
xinfo = require('monitor-info').getXInfo(0);
}
if (xinfo == null)
{
ret._rej('This system cannot display a user dialog box when a user is not logged in');
return (ret);
}
if (this.zenity)
{
// GNOME/ZENITY
if (this.zenity.timeout)
{
ret.child = require('child_process').execFile(this.zenity.path, ['zenity', layout==null?'--question':'--warning', '--title=' + title, '--text=' + caption, '--timeout=' + timeout], { uid: uid, env: { XAUTHORITY: xinfo.xauthority ? xinfo.xauthority : "", DISPLAY: xinfo.display } });
}
else
{
ret.child = require('child_process').execFile(this.zenity.path, ['zenity', layout == null ? '--question' : '--warning', '--title=' + title, '--text=' + caption], { uid: uid, env: { XAUTHORITY: xinfo.xauthority ? xinfo.xauthority : "", DISPLAY: xinfo.display } });
ret.child.timeout = setTimeout(function (c)
{
c.promise._rej('timeout');
c.timeout = null;
c.kill();
}, timeout * 1000, ret.child);
}
ret.child.promise = ret;
ret.child.stderr.on('data', function (chunk) { });
ret.child.stdout.on('data', function (chunk) { });
ret.child.on('exit', function (code)
{
if (this.timeout) { clearTimeout(this.timeout); }
switch (code)
{
case 0:
this.promise._res();
break;
case 1:
this.promise._rej('denied');
break;
default:
this.promise._rej('timeout');
break;
}
});
}
else if(this.kdialog)
{
if (process.platform != 'freebsd' && process.env['DISPLAY'])
{
ret.child = require('child_process').execFile(this.kdialog.path, ['kdialog', '--title', title, layout==null?'--yesno':'--msgbox', caption]);
ret.child.promise = ret;
}
else
{
var xdg = require('user-sessions').findEnv(uid, 'XDG_RUNTIME_DIR'); if (xdg == null) { xdg = ''; }
if (!xinfo || !xinfo.display || !xinfo.xauthority) { ret._rej('Interal Error, could not determine X11/XDG env'); return (ret); }
ret.child = require('child_process').execFile(this.kdialog.path, ['kdialog', '--title', title, layout == null ? '--yesno' : '--msgbox', caption], { uid: uid, env: { DISPLAY: xinfo.display, XAUTHORITY: xinfo.xauthority, XDG_RUNTIME_DIR: xdg } });
ret.child.promise = ret;
}
ret.child.timeout = setTimeout(function (c)
{
c.timeout = null;
c.kill();
}, timeout * 1000, ret.child);
ret.child.stdout.on('data', function (chunk) { });
ret.child.stderr.on('data', function (chunk) { });
ret.child.on('exit', function (code)
{
if (this.timeout)
{
clearTimeout(this.timeout);
switch (code)
{
case 0:
this.promise._res();
break;
case 1:
this.promise._rej('denied');
break;
default:
this.promise._rej('timeout');
break;
}
}
else
{
this.promise._rej('timeout');
}
});
}
else if (this.xmessage)
{
// title, caption, timeout, layout
ret.child = require('child_process').execFile(this.xmessage.path, ['xmessage', '-center', '-buttons', layout == null ? 'No:1,Yes:2' : 'OK:2', '-timeout', timeout.toString(), '-default', layout==null?'No':'OK', '-title', title, caption], { uid: uid, env: { XAUTHORITY: xinfo.xauthority ? xinfo.xauthority : "", DISPLAY: xinfo.display } });
ret.child.stdout.on('data', function (c) { });
ret.child.stderr.on('data', function (c) { });
ret.child.promise = ret;
ret.child.on('exit', function (code)
{
switch(code)
{
case 2:
this.promise._res();
break;
case 1:
this.promise._rej('denied');
break;
default:
this.promise._rej('timeout');
break;
}
});
}
else
{
ret._rej('Unable to create dialog box');
}
return (ret);
};
}
if (process.platform == 'darwin')
{
function translateObject(obj)
{
var j = JSON.stringify(obj);
var b = Buffer.alloc(j.length + 4);
b.writeUInt32LE(j.length + 4);
Buffer.from(j).copy(b, 4);
return (b);
}
}
function macos_messageBox()
{
this._ObjectID = 'message-box';
this._initIPCBase = function _initIPCBase()
{
var ret = new promise(function (res, rej) { this._res = res; this._rej = rej; });
try
{
ret.uid = require('user-sessions').consoleUid();
}
catch (e)
{
ret._rej(e);
return (ret);
}
ret.path = '/var/tmp/' + process.execPath.split('/').pop() + '_ev';
var n;
try
{
n = require('tls').generateRandomInteger('1', '99999');
}
catch (e)
{
n = 0;
}
while (require('fs').existsSync(ret.path + n))
{
try {
n = require('tls').generateRandomInteger('1', '99999');
}
catch (e) {
++n;
}
}
ret.path = ret.path + n;
ret.tmpServiceName = 'meshNotificationServer' + n;
return (ret);
};
this.create = function create(title, caption, timeout, layout)
{
// Start Local Server
var ret = this._initIPCBase();
ret.title = title; ret.caption = caption; ret.timeout = timeout; ret.layout = layout;
ret.server = this.startMessageServer(ret);
ret.server.ret = ret;
ret.server.on('connection', function (c)
{
this._connection = c;
c.promise = this.ret;
c.on('data', function (buffer)
{
if (buffer.len < 4 || buffer.readUInt32LE(0) > buffer.len) { this.unshift(buffer); }
var p = JSON.parse(buffer.slice(4, buffer.readUInt32LE(0)).toString());
switch (p.command)
{
case 'ERROR':
this.promise._rej(p.reason);
break;
case 'DIALOG':
if (p.timeout)
{
this.promise._rej('TIMEOUT');
}
else
{
if (p.button == 'Yes' || p.button == 'OK')
{
this.promise._res(p.button);
}
else
{
this.promise._rej('denied');
}
}
break;
}
});
c.write(translateObject({ command: 'DIALOG', title: this.ret.title, caption: this.ret.caption, icon: 'caution', buttons: this.ret.layout==null?['"Yes"', '"No"']:['"OK"'], buttonDefault: this.ret.layout==null?2:1, timeout: this.ret.timeout }));
});
return (ret);
};
this.lock = function lock()
{
// Start Local Server
var ret = this._initIPCBase();
ret.server = this.startMessageServer(ret);
ret.server.ret = ret;
ret.server.on('connection', function (c)
{
this._connection = c;
c.promise = this.ret;
c.on('data', function (buffer)
{
if (buffer.len < 4 || buffer.readUInt32LE(0) > buffer.len) { this.unshift(buffer); }
var p = JSON.parse(buffer.slice(4, buffer.readUInt32LE(0)).toString());
switch (p.command)
{
case 'ERROR':
this.promise._rej(p.reason);
break;
case 'LOCK':
this.promise._res();
break;
}
});
c.write(translateObject({ command: 'LOCK' }));
});
return (ret);
};
this.notify = function notify(title, caption)
{
// Start Local Server
var ret = this._initIPCBase();
ret.title = title; ret.caption = caption;
ret.server = this.startMessageServer(ret);
ret.server.ret = ret;
ret.server.on('connection', function (c)
{
this._connection = c;
c.promise = this.ret;
c.on('data', function (buffer)
{
if (buffer.len < 4 || buffer.readUInt32LE(0) > buffer.len) { this.unshift(buffer); }
var p = JSON.parse(buffer.slice(4, buffer.readUInt32LE(0)).toString());
switch (p.command)
{
case 'ERROR':
this.promise._rej(p.reason);
break;
case 'NOTIFY':
this.promise._res();
break;
}
});
c.write(translateObject({ command: 'NOTIFY', title: this.ret.title, caption: this.ret.caption }));
});
return (ret);
};
this.startClient = function startClient(options)
{
// Create the Client
console.log('Starting Client...');
options.osversion = require('service-manager').getOSVersion();
options.uid = require('user-sessions').consoleUid();
this.client = require('net').createConnection(options);
this.client._options = options;
this.client.on('data', function (buffer)
{
if (buffer.len < 4 || buffer.readUInt32LE(0) > buffer.len) { this.unshift(buffer); }
var p = JSON.parse(buffer.slice(4, buffer.readUInt32LE(0)).toString());
switch (p.command)
{
case 'LOCK':
this._shell = require('child_process').execFile('/bin/sh', ['sh']);
this._shell.stdout.str = ''; this._shell.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
this._shell.stderr.str = ''; this._shell.stderr.on('data', function (chunk) { this.str += chunk.toString(); });
this._shell.stdin.write('/System/Library/CoreServices/Menu\\ Extras/User.menu/Contents/Resources/CGSession -suspend\nexit\n');
this._shell.waitExit();
if (this._shell.stderr.str != '')
{
this.end(translateObject({ command: 'ERROR', reason: this._shell.stderr.str }));
}
else
{
this.end(translateObject({ command: 'LOCK', status: 0 }));
}
break;
case 'NOTIFY':
this._shell = require('child_process').execFile('/bin/sh', ['sh']);
this._shell.stdout.str = ''; this._shell.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
this._shell.stderr.str = ''; this._shell.stderr.on('data', function (chunk) { this.str += chunk.toString(); });
this._shell.stdin.write('osascript -e \'tell current application to display notification "' + p.caption + '" with title "' + p.title + '"\'\nexit\n');
this._shell.waitExit();
if (this._shell.stderr.str != '')
{
this.end(translateObject({ command: 'ERROR', reason: this._shell.stderr.str }));
}
else
{
this.end(translateObject({ command: 'NOTIFY', status: 0 }));
}
break;
case 'DIALOG':
var timeout = p.timeout ? (' giving up after ' + p.timeout) : '';
var icon = p.icon ? ('with icon ' + p.icon) : '';
var buttons = p.buttons ? ('buttons {' + p.buttons.toString() + '}') : '';
if (p.buttonDefault != null)
{
buttons += (' default button ' + p.buttonDefault)
}
this._shell = require('child_process').execFile('/bin/sh', ['sh']);
this._shell.stdout.str = ''; this._shell.stdout.on('data', function (chunk) { this.str += chunk.toString(); });
this._shell.stderr.str = ''; this._shell.stderr.on('data', function (chunk) { this.str += chunk.toString(); });
this._shell.stdin.write('osascript -e \'tell current application to display dialog "' + p.caption + '" with title "' + p.title + '" ' + icon + ' ' + buttons + timeout + '\' | awk \'{ c=split($0, tokens, ","); split(tokens[1], val, ":"); if(c==1) { print val[2] } else { split(tokens[2], gu, ":"); if(gu[2]=="true") { print "_TIMEOUT_" } else { print val[2] } } }\'\nexit\n');
this._shell.waitExit();
if (this._shell.stderr.str != '' && !this._shell.stderr.str.includes('OpenGL'))
{
this.end(translateObject({ command: 'ERROR', reason: this._shell.stderr.str }));
}
else
{
if (this._shell.stdout.str.trim() == '_TIMEOUT_')
{
this.end(translateObject({ command: 'DIALOG', timeout: true }));
}
else
{
this.end(translateObject({ command: 'DIALOG', button: this._shell.stdout.str.trim() }));
}
}
break;
default:
break;
}
});
this.client.on('error', function () { this.uninstall(); }).on('end', function () { this.uninstall(); });
this.client.uninstall = function ()
{
// Need to uninstall ourselves
var child = require('child_process').execFile(process.execPath, [process.execPath.split('/').pop(), '-exec', "var s=require('service-manager').manager.getLaunchAgent('" + this._options.service + "', " + this._options.uid + "); s.unload(); require('fs').unlinkSync(s.plist);process.exit();"], { detached: true, type: require('child_process').SpawnTypes.DETACHED });
child.waitExit();
};
return (this.client);
};
this.startMessageServer = function startMessageServer(options)
{
if (require('fs').existsSync(options.path)) { require('fs').unlinkSync(options.path); }
options.writableAll = true;
var ret = require('net').createServer();
ret.uid = require('user-sessions').consoleUid();
ret.osversion = require('service-manager').getOSVersion();
ret._options = options;
ret.timer = setTimeout(function (obj)
{
obj.close();
obj._options._rej('Connection timeout');
}, 5000, ret);
ret.listen(options);
ret.on('connection', function (c)
{
clearTimeout(this.timer);
});
ret.on('~', function ()
{
require('fs').unlinkSync(this._options.path);
});
require('service-manager').manager.installLaunchAgent(
{
name: options.tmpServiceName, servicePath: process.execPath, startType: 'AUTO_START', uid: ret.uid,
sessionTypes: ['Aqua'], parameters: ['-exec', "require('message-box').startClient({ path: '" + options.path + "', service: '" + options.tmpServiceName + "' }).on('end', function () { process.exit(); }).on('error', function () { process.exit(); });"]
});
require('service-manager').manager.getLaunchAgent(options.tmpServiceName, ret.uid).load();
return (ret);
};
}
switch(process.platform)
{
case 'win32':
module.exports = new messageBox();
break;
case 'linux':
case 'freebsd':
module.exports = new linux_messageBox();
break;
case 'darwin':
module.exports = new macos_messageBox();
break;
}