1
0
mirror of https://github.com/Ylianst/MeshAgent synced 2025-12-06 00:13:33 +00:00
Files
MeshAgent/modules/notifybar-desktop.js
2020-05-22 12:18:46 -07:00

372 lines
15 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.
*/
var ptrsize = require('_GenericMarshal').PointerSize;
var ClientMessage = 33;
function windows_notifybar_check(title, tsid)
{
if(require('user-sessions').getProcessOwnerName(process.pid).tsid == 0)
{
return (windows_notifybar_system(title, tsid));
}
else
{
return (windows_notifybar_local(title));
}
}
function windows_notifybar_system(title, tsid)
{
var ret = {};
var script = Buffer.from("require('notifybar-desktop')('" + title + "').on('close', function(){process._exit();});").toString('base64');
require('events').EventEmitter.call(ret, true)
.createEvent('close')
.addMethod('close', function close() { this.child.kill(); });
ret.child = require('child_process').execFile(process.execPath, [process.execPath.split('\\').pop(), '-b64exec', script], { type: 1, uid: tsid });
ret.child.descriptorMetadata = 'notifybar-desktop';
ret.child.parent = ret;
ret.child.stdout.on('data', function (c) { });
ret.child.stderr.on('data', function (c) { });
ret.child.on('exit', function (code) { this.parent.emit('close', code); });
return (ret);
}
function windows_notifybar_local(title)
{
var MessagePump;
var ret;
MessagePump = require('win-message-pump');
ret = { _ObjectID: 'notifybar-desktop.Windows', title: title, _pumps: [], _promise: require('monitor-info').getInfo() };
ret._promise.notifybar = ret;
require('events').EventEmitter.call(ret, true)
.createEvent('close')
.addMethod('close', function close()
{
for (var i = 0; i < this._pumps.length; ++i)
{
this._pumps[i].removeAllListeners('exit');
this._pumps[i].close();
}
this._pumps = [];
});
ret._promise.then(function (m)
{
var offset;
var barWidth, monWidth, offset, barHeight, monHeight;
for (var i in m)
{
//console.log('Monitor: ' + i + ' = Width[' + (m[i].right - m[i].left) + ']');
monWidth = (m[i].right - m[i].left);
monHeight = (m[i].bottom - m[i].top);
barWidth = Math.floor(monWidth * 0.30);
barHeight = Math.floor(monHeight * 0.035);
offset = Math.floor(monWidth * 0.50) - Math.floor(barWidth * 0.50);
start = m[i].left + offset;
var options =
{
window:
{
winstyles: MessagePump.WindowStyles.WS_VISIBLE | MessagePump.WindowStyles.WS_BORDER | MessagePump.WindowStyles.WS_CAPTION | MessagePump.WindowStyles.WS_SYSMENU,
x: start, y: m[i].top, left: m[i].left, right: m[i].right, width: barWidth, height: barHeight, title: this.notifybar.title
}
};
this.notifybar._pumps.push(new MessagePump(options));
this.notifybar._pumps.peek().notifybar = this.notifybar;
this.notifybar._pumps.peek().on('hwnd', function (h)
{
this._HANDLE = h;
});
this.notifybar._pumps.peek().on('exit', function (h)
{
for (var i = 0; i < this.notifybar._pumps.length; ++i)
{
this.notifybar._pumps[i].removeAllListeners('exit');
this.notifybar._pumps[i].close();
}
this.notifybar.emit('close');
this.notifybar._pumps = [];
});
this.notifybar._pumps.peek().on('message', function onWindowsMessage(msg)
{
if (msg.message == 133)
{
//console.log("WM_NCPAINT");
}
if (msg.message == 70) // We are intercepting WM_WINDOWPOSCHANGING to DISABLE moving the window
{
if (this._HANDLE)
{
var flags = 0;
switch (ptrsize)
{
case 4:
flags = msg.lparam_raw.Deref(24, 4).toBuffer().readUInt32LE() | 0x0002; // Set SWP_NOMOVE
if (msg.lparam_raw.Deref(8, 4).toBuffer().readInt32LE() < this._options.window.left ||
(msg.lparam_raw.Deref(8, 4).toBuffer().readInt32LE() + this._options.window.width) >= this._options.window.right)
{
// Disallow this move, because it will go out of bounds of the current monitor
msg.lparam_raw.Deref(24, 4).toBuffer().writeUInt32LE(flags);
}
else
{
// Allow the move, but only on the X-axis
msg.lparam_raw.Deref(12, 4).toBuffer().writeInt32LE(this._options.window.y);
}
break;
case 8:
flags = msg.lparam_raw.Deref(32, 4).toBuffer().readUInt32LE() | 0x0002 // Set SWP_NOMOVE
if (msg.lparam_raw.Deref(16, 4).toBuffer().readInt32LE() < this._options.window.left ||
(msg.lparam_raw.Deref(16, 4).toBuffer().readInt32LE() + this._options.window.width) >= this._options.window.right)
{
// Disallow this move, because it will go out of bounds of the current monitor
msg.lparam_raw.Deref(32, 4).toBuffer().writeUInt32LE(flags);
}
else
{
// Allow the move, but only on the X-axis
msg.lparam_raw.Deref(20, 4).toBuffer().writeInt32LE(this._options.window.y);
}
break;
}
}
}
});
}
});
return (ret);
}
function x_notifybar_check(title)
{
var script = Buffer.from("require('notifybar-desktop')('" + title + "').on('close', function(){process.exit();});").toString('base64');
var min = require('user-sessions').minUid();
var uid = require('user-sessions').consoleUid();
var self = require('user-sessions').Self();
if (self != 0 || uid == 0)
{
return (x_notifybar(title)); // No Dispatching necessary
}
else
{
// We are root, so we should try to spawn a child into the user's desktop
if (uid < min && uid != 0)
{
// Lets hook login event, so we can respawn the bars later
var ret = { min: min };
require('events').EventEmitter.call(ret, true)
.createEvent('close')
.addMethod('close', function close()
{
require('user-sessions').removeListener('changed', this._changed);
this._close2();
});
ret._changed = function _changed()
{
var that = _changed.self;
var uid = require('user-sessions').consoleUid();
if (uid >= that.min)
{
that._close2 = function ()
{
this.child.kill();
};
var xinfo = require('monitor-info').getXInfo(uid);
that.child = require('child_process').execFile(process.execPath, [process.execPath.split('/').pop(), '-b64exec', script], { uid: uid, env: xinfo.exportEnv() });
that.child.descriptorMetadata = 'notifybar-desktop';
that.child.parent = that;
that.child.stdout.on('data', function (c) { });
that.child.stderr.on('data', function (c) { });
that.child.on('exit', function (code) { this.parent.emit('close', code); });
}
};
ret._changed.self = ret;
require('user-sessions').on('changed', ret._changed);
ret._close2 = function _close2()
{
this.emit('close');
};
return (ret);
}
var xinfo = require('monitor-info').getXInfo(uid);
if (!xinfo)
{
throw('XServer Initialization Error')
}
var ret = {};
require('events').EventEmitter.call(ret, true)
.createEvent('close')
.addMethod('close', function close() { this.child.kill(); });
ret.child = require('child_process').execFile(process.execPath, [process.execPath.split('/').pop(), '-b64exec', script], { uid: uid, env: xinfo.exportEnv() });
ret.child.descriptorMetadata = 'notifybar-desktop';
ret.child.parent = ret;
ret.child.stdout.on('data', function (c) { });
ret.child.stderr.on('data', function (c) { });
ret.child.on('exit', function (code) { this.parent.emit('close', code); });
return (ret);
}
}
function x_notifybar(title)
{
ret = { _ObjectID: 'notifybar-desktop.X', title: title, _windows: [], _promise: require('monitor-info').getInfo(), monitors: [], workspaces: {} };
ret._promise.notifybar = ret;
require('events').EventEmitter.call(ret, true)
.createEvent('close')
.addMethod('close', function close()
{
});
ret._promise.createBars = function (m)
{
for (var i in m)
{
monWidth = (m[i].right - m[i].left);
monHeight = (m[i].bottom - m[i].top);
barWidth = Math.floor(monWidth * 0.30);
barHeight = Math.floor(monHeight * 0.035);
offset = Math.floor(monWidth * 0.50) - Math.floor(barWidth * 0.50);
start = m[i].left + offset;
var white = require('monitor-info')._X11.XWhitePixel(m[i].display, m[i].screenId).Val;
this.notifybar._windows.push({
root: require('monitor-info')._X11.XRootWindow(m[i].display, m[i].screenId),
display: m[i].display, id: m[i].screedId
});
this.notifybar._windows.peek().notifybar = require('monitor-info')._X11.XCreateSimpleWindow(m[i].display, this.notifybar._windows.peek().root, start, 0, barWidth, 1, 0, white, white);
require('monitor-info')._X11.XStoreName(m[i].display, this.notifybar._windows.peek().notifybar, require('_GenericMarshal').CreateVariable(this.notifybar.title));
require('monitor-info').setWindowSizeHints(m[i].display, this.notifybar._windows.peek().notifybar, start, 0, barWidth, 1, barWidth, 1, barWidth, 1);
require('monitor-info').hideWindowIcon(m[i].display, this.notifybar._windows.peek().root, this.notifybar._windows.peek().notifybar);
require('monitor-info').setAllowedActions(m[i].display, this.notifybar._windows.peek().notifybar, require('monitor-info').MOTIF_FLAGS.MWM_FUNC_CLOSE);
require('monitor-info').setAlwaysOnTop(m[i].display, this.notifybar._windows.peek().root, this.notifybar._windows.peek().notifybar);
var wm_delete_window_atom = require('monitor-info')._X11.XInternAtom(m[i].display, require('_GenericMarshal').CreateVariable('WM_DELETE_WINDOW'), 0).Val;
var atoms = require('_GenericMarshal').CreateVariable(4);
atoms.toBuffer().writeUInt32LE(wm_delete_window_atom);
require('monitor-info')._X11.XSetWMProtocols(m[i].display, this.notifybar._windows.peek().notifybar, atoms, 1);
require('monitor-info')._X11.XMapWindow(m[i].display, this.notifybar._windows.peek().notifybar);
require('monitor-info')._X11.XFlush(m[i].display);
this.notifybar._windows.peek().DescriptorEvent = require('DescriptorEvents').addDescriptor(require('monitor-info')._X11.XConnectionNumber(m[i].display).Val, { readset: true });
this.notifybar._windows.peek().DescriptorEvent.atom = wm_delete_window_atom;
this.notifybar._windows.peek().DescriptorEvent.ret = this.notifybar;
this.notifybar._windows.peek().DescriptorEvent._display = m[i].display;
this.notifybar._windows.peek().DescriptorEvent.on('readset', function (fd)
{
var XE = require('_GenericMarshal').CreateVariable(1024);
while (require('monitor-info')._X11.XPending(this._display).Val)
{
require('monitor-info')._X11.XNextEventSync(this._display, XE);
if (XE.Deref(0, 4).toBuffer().readUInt32LE() == ClientMessage)
{
var clientType = XE.Deref(require('_GenericMarshal').PointerSize == 8 ? 56 : 28, 4).toBuffer().readUInt32LE();
if (clientType == this.atom)
{
require('DescriptorEvents').removeDescriptor(fd);
require('monitor-info')._X11.XCloseDisplay(this._display);
ret.emit('close');
ret._windows.clear();
break;
}
}
}
});
}
};
ret._promise.then(function (m)
{
var offset;
var barWidth, monWidth, offset, barHeight, monHeight;
this.notifybar.monitors = m;
if (m.length > 0)
{
var ws = 0;
try
{
ws = m[0].display.getCurrentWorkspace();
this.notifybar.workspaces[ws] = true;
this.createBars(m);
}
catch(wex)
{
}
m[0].display._notifyBar = this.notifybar;
m[0].display.on('workspaceChanged', function (w)
{
if(!this._notifyBar.workspaces[w])
{
this._notifyBar.workspaces[w] = true;
this._notifyBar._promise.createBars(this._notifyBar.monitors);
}
});
}
});
return (ret);
}
function macos_messagebox(title)
{
var ret = {};
require('events').EventEmitter.call(ret, true)
.createEvent('close')
.addMethod('close', function close() { this._messageBox.close(); });
ret._messageBox = require('message-box').create('', title, 0, ['Disconnect']);
ret._messageBox.that = ret;
ret._messageBox.then(function () { this.that.emit('close'); }, function () { this.that.emit('close'); });
return (ret);
}
switch(process.platform)
{
case 'win32':
module.exports = windows_notifybar_check;
module.exports.system = windows_notifybar_system;
break;
case 'linux':
case 'freebsd':
module.exports = x_notifybar_check;
break;
case 'darwin':
module.exports = macos_messagebox;
break;
}