MeshCentral/MeshAgent
1
0
Fork 0
mirror of https://github.com/Ylianst/MeshAgent synced 2026-01-04 17:43:45 +00:00
Code Issues Releases Wiki Activity

Update documentation

Browse Source
This commit is contained in:
Bryan Roe
2022-10-11 00:14:24 -07:00
parent f811d1fc93
commit f5283e30ae
3 changed files with 81 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
modules/toaster.js
View File

@@ -14,10 +14,19 @@ See the License for the specific language governing permissions and
limitations under the License.
*/
//
// Toaster.js provides functionality to be able to pop up a toast notification in a platform agnostic fashion
//
var promise = require('promise');
if (process.platform == 'linux' || process.platform == 'darwin' || process.platform == 'freebsd')
{
//
// Helper function to find the binary path for the specified application, using the system utility 'whereis'
//
function findPath(app)
{
var child = require('child_process').execFile('/bin/sh', ['sh']);
@@ -41,10 +50,15 @@ if (process.platform == 'linux' || process.platform == 'darwin' || process.platf
function Toaster()
{
this._ObjectID = 'toaster';
//
// Pops a toast, with the specified titale and caption, using the specified sid
// Returns a promise that resolves when the toast is dismissed
//
this.Toast = function Toast(title, caption, tsid)
{
var retVal = new promise(function (res, rej) { this._res = res; this._rej = rej; });
if (title == 'MeshCentral') { try { title = require('MeshAgent').displayName; } catch (x) { } }
if (title == 'MeshCentral') { try { title = require('MeshAgent').displayName; } catch (x) { } } // IF the tital is the default 'MeshCentral', try to switch it for the agent's displayName
retVal.title = title;
retVal.caption = caption;
@@ -53,8 +67,13 @@ function Toaster()
{
case 'win32':
{
//
// For Windows, we will use powershell to display the toast. We tried using Shell Notify Icon, but ran into stability issues with it.
// The Powershell interface for it seemed much more reliable
//
var cid;
retVal.options = { env: { _title: title, _caption: caption } };
retVal.options = { env: { _title: title, _caption: caption } }; // We are putting these values into the environment, becuase Powershell has an issue with passing UTF8 values otherwise
for (var c1e in process.env)
{
retVal.options.env[c1e] = process.env[c1e];
@@ -82,26 +101,29 @@ function Toaster()
return (retVal);
}
// Spawn a powershell process so we can setup everything to display the toast
retVal.child = require('child_process').execFile(process.env['windir'] + '\\System32\\WindowsPowerShell\\v1.0\\powershell.exe', ['powershell', '-noprofile', '-nologo', '-command', '-'], retVal.options);
retVal.child.descriptorMetadata = 'toaster';
retVal.child.toast = retVal;
retVal.child.stdout.stdin = retVal.child.stdin;
retVal.child.stderr.stdin = retVal.child.stdin;
retVal.child.stdout.on('data', function (c) { if (c.toString().includes('<DISMISSED>')) { this.stdin.write('exit\n'); } });
retVal.child.stdout.on('data', function (c) { if (c.toString().includes('<DISMISSED>')) { this.stdin.write('exit\n'); } }); // When the toast is dismissed, exit the process
retVal.child.stderr.once('data', function (c) { this.stdin.write('$objBalloon.dispose();exit\n'); });
retVal.child.stdin.write('[System.Reflection.Assembly]::LoadWithPartialName("System.Windows.Forms")\r\n');
retVal.child.stdin.write('$objBalloon = New-Object System.Windows.Forms.NotifyIcon\r\n');
retVal.child.stdin.write('$objBalloon.Icon = [System.Drawing.SystemIcons]::Information\r\n');
retVal.child.stdin.write('[System.Reflection.Assembly]::LoadWithPartialName("System.Windows.Forms")\r\n'); // Init
retVal.child.stdin.write('$objBalloon = New-Object System.Windows.Forms.NotifyIcon\r\n'); // Create the NotifyIcon object
retVal.child.stdin.write('$objBalloon.Icon = [System.Drawing.SystemIcons]::Information\r\n'); // Set the icon type
retVal.child.stdin.write('$objBalloon.Visible = $True\r\n');
retVal.child.stdin.write('Register-ObjectEvent -InputObject $objBalloon -EventName BalloonTipClosed -Action { $objBalloon.dispose();Write-Host "<`DISMISSED`>" }')
retVal.child.stdin.write('$objBalloon.ShowBalloonTip(10000, $env:_title, $env:_caption, 0)\r\n');
retVal.child.stdin.write('Register-ObjectEvent -InputObject $objBalloon -EventName BalloonTipClosed -Action { $objBalloon.dispose();Write-Host "<`DISMISSED`>" }') // Set an event handler for when the toast is dismissed
retVal.child.stdin.write('$objBalloon.ShowBalloonTip(10000, $env:_title, $env:_caption, 0)\r\n'); // Show the toast
retVal.child.timeout = setTimeout(function (c)
{
// Set a timeout to cleanup after 10 seconds. This will cause the process to cleanup if the user doesn't interact with the toast
c.timeout = null;
c.stdin.write('$objBalloon.dispose();exit\n');
}, 10000, retVal.child);
retVal.child.on('exit', function ()
{
// Handler that is called when the powershell process has exited
if (this.timeout != null) { clearTimeout(this.timeout); }
this.toast._res('DISMISSED');
});
@@ -114,6 +136,9 @@ function Toaster()
{
try
{
//
// Fetch some necessary configuration information about the currently logged in session, so that we can display the toast in the correct desktop session
//
retVal.consoleUid = require('user-sessions').consoleUid();
retVal.xinfo = require('monitor-info').getXInfo(retVal.consoleUid);
retVal.username = require('user-sessions').getUsername(retVal.consoleUid);
@@ -124,7 +149,7 @@ function Toaster()
return (retVal);
}
if (require('message-box').zenity)
if (require('message-box').zenity) // Check to see if ZENITY is installed on the system
{
if (process.platform == 'linux' && !require('linux-dbus').hasService('org.freedesktop.Notifications'))
{
@@ -144,8 +169,7 @@ function Toaster()
}
else if (require('message-box').zenity.broken || require('message-box').zenity.version[0] < 3 || (require('message-box').zenity.version[0] == 3 && require('message-box').zenity.version[1] < 10))
{
// ZENITY Notification is broken
// ZENITY Notification is broken for this version
if (require('message-box').notifysend)
{
// Using notify-send
@@ -201,6 +225,8 @@ function Toaster()
}
else
{
// This platform doesn't have ZENITY, so lets check other system utilities we can use
util = findPath('kdialog');
if (util)
{
@@ -240,6 +266,9 @@ function Toaster()
}
else if (require('message-box').xmessage)
{
//
// XMESSAGE is very rudimentary, so this is a last resort if nothing else exists.
//
retVal._mb = require('message-box').create(title, caption, 5, 'OK');
retVal._mb.ret = retVal;
retVal._mb.then(function () { this.ret._res('DISMISSED'); }, function () { this.ret._res('DISMISSED'); });
@@ -253,6 +282,9 @@ function Toaster()
}
break;
case 'darwin':
//
// For macOS we implemented this in message-box, so let's use that
//
retVal._toast = require('message-box').notify(title, caption);
retVal._toast.parent = retVal;
retVal._toast.then(function (v) { this.parent._res(v); }, function (e) { this.parent._rej(e); });
@@ -263,6 +295,9 @@ function Toaster()
};
if(process.platform == 'win32')
{
//
// Old way to display toast, using NOTIFY_ShellIcon
//
this._containerToast = function _containerToast(caption, title)
{
var toast;
@@ -292,6 +327,10 @@ function Toaster()
}
module.exports = new Toaster();
//
// Helper method on Linux to check if a dbus service exists
//
if (process.platform == 'linux' && !require('linux-dbus').hasService)
{
require('linux-dbus').hasService = function hasService(name)