MeshCentral/agents/modules_meshcore/monitor-border.js
2018-08-29 18:47:22 -07:00

320 lines
20 KiB
JavaScript

var red = 0xFF;
var yellow = 0xFFFF;
var GXxor = 0x6; // src XOR dst
var GXclear = 0x0;
var ExposureMask = (1 << 15);
function windows_monitorborder()
{
this._ObjectID = 'monitor-info';
var info = require('monitor-info');
var gm = require('_GenericMarshal');
var user32 = gm.CreateNativeProxy('user32.dll');
info.monitors = [];
user32.CreateMethod('GetDC');
user32.CreateMethod('ReleaseDC');
user32.CreateMethod('FillRect');
user32.CreateMethod('InvalidateRect');
var gdi32 = gm.CreateNativeProxy('gdi32.dll');
gdi32.CreateMethod('CreateSolidBrush');
var redBrush = gdi32.CreateSolidBrush(red);
var yellowBrush = gdi32.CreateSolidBrush(yellow);
require('events').EventEmitter.call(this);
this.on('~', function () { this.Stop(); });
this.Stop = function Stop()
{
info.redInterval = null;
var drawRect = gm.CreateVariable(16);
var drawRectBuffer = drawRect.toBuffer();
for (var i in info.monitors)
{
// Top
drawRectBuffer.writeInt32LE(info.monitors[i].left, 0);
drawRectBuffer.writeInt32LE(info.monitors[i].top, 4);
drawRectBuffer.writeInt32LE(info.monitors[i].left + (info.monitors[i].right - info.monitors[i].left), 8);
drawRectBuffer.writeInt32LE(info.monitors[i].bottom - info.monitors[i].top, 12);
user32.InvalidateRect(0, drawRect, 0);
}
}
this.Start = function Start()
{
info.getInfo().then(function (mon)
{
var drawRect = gm.CreateVariable(16);
info.monitors = mon;
info.dc = user32.GetDC(0);
info.state = 0;
info.redInterval = setInterval(function ()
{
info.state = (info.state + 1) % 8;
var drawRectBuffer = drawRect.toBuffer();
for(var i in info.monitors)
{
drawRectBuffer.writeInt32LE(info.monitors[i].left, 0);
drawRectBuffer.writeInt32LE(info.monitors[i].top, 4);
drawRectBuffer.writeInt32LE(info.monitors[i].left + (info.monitors[i].right - info.monitors[i].left)/2, 8);
drawRectBuffer.writeInt32LE(5, 12);
user32.FillRect(info.dc, drawRect, (info.state == 0 || info.state == 4) ? yellowBrush : redBrush);
drawRectBuffer.writeInt32LE(info.monitors[i].left + (info.monitors[i].right - info.monitors[i].left) / 2, 0);
drawRectBuffer.writeInt32LE(info.monitors[i].top, 4);
drawRectBuffer.writeInt32LE(info.monitors[i].right, 8);
drawRectBuffer.writeInt32LE(5, 12);
user32.FillRect(info.dc, drawRect, (info.state == 1 || info.state == 5) ? yellowBrush : redBrush);
drawRectBuffer.writeInt32LE(info.monitors[i].right - 5, 0);
drawRectBuffer.writeInt32LE(info.monitors[i].top, 4);
drawRectBuffer.writeInt32LE(info.monitors[i].right, 8);
drawRectBuffer.writeInt32LE(info.monitors[i].top + (info.monitors[i].bottom - info.monitors[i].top)/2, 12);
user32.FillRect(info.dc, drawRect, (info.state == 2 || info.state == 6) ? yellowBrush : redBrush);
drawRectBuffer.writeInt32LE(info.monitors[i].right - 5, 0);
drawRectBuffer.writeInt32LE(info.monitors[i].top + (info.monitors[i].bottom - info.monitors[i].top) / 2, 4);
drawRectBuffer.writeInt32LE(info.monitors[i].right, 8);
drawRectBuffer.writeInt32LE(info.monitors[i].bottom, 12);
user32.FillRect(info.dc, drawRect, (info.state == 3 || info.state == 7) ? yellowBrush : redBrush);
drawRectBuffer.writeInt32LE(info.monitors[i].left + (info.monitors[i].right - info.monitors[i].left) / 2, 0);
drawRectBuffer.writeInt32LE(info.monitors[i].bottom - 5, 4);
drawRectBuffer.writeInt32LE(info.monitors[i].right, 8);
drawRectBuffer.writeInt32LE(info.monitors[i].bottom, 12);
user32.FillRect(info.dc, drawRect, (info.state == 4 || info.state == 0) ? yellowBrush : redBrush);
drawRectBuffer.writeInt32LE(info.monitors[i].left, 0);
drawRectBuffer.writeInt32LE(info.monitors[i].bottom - 5, 4);
drawRectBuffer.writeInt32LE(info.monitors[i].left + (info.monitors[i].right - info.monitors[i].left) / 2, 8);
drawRectBuffer.writeInt32LE(info.monitors[i].bottom, 12);
user32.FillRect(info.dc, drawRect, (info.state == 5 || info.state == 1) ? yellowBrush : redBrush);
drawRectBuffer.writeInt32LE(info.monitors[i].left, 0);
drawRectBuffer.writeInt32LE(info.monitors[i].top + (info.monitors[i].bottom - info.monitors[i].top) / 2, 4);
drawRectBuffer.writeInt32LE(info.monitors[i].left + 5, 8);
drawRectBuffer.writeInt32LE(info.monitors[i].bottom, 12);
user32.FillRect(info.dc, drawRect, (info.state == 6 || info.state == 2) ? yellowBrush : redBrush);
drawRectBuffer.writeInt32LE(info.monitors[i].left, 0);
drawRectBuffer.writeInt32LE(info.monitors[i].top, 4);
drawRectBuffer.writeInt32LE(info.monitors[i].left + 5, 8);
drawRectBuffer.writeInt32LE(info.monitors[i].top + (info.monitors[i].bottom - info.monitors[i].top) / 2, 12);
user32.FillRect(info.dc, drawRect, (info.state == 7 || info.state == 3) ? yellowBrush : redBrush);
}
}, 450);
});
}
}
function linux_monitorborder()
{
var self = this;
this.displays = [];
this._ObjectID = 'monitor-info';
this._info = require('monitor-info');
this._isUnity = this._info.isUnity();
console.log('isUnity = ' + this._isUnity);
require('events').EventEmitter.call(this);
this.on('~', function () { this.Stop(); });
this.Stop = function Stop()
{
this._timeout = null;
if(!this._isUnity)
{
for(var i=0; i < this.displays.length; ++i)
{
if(this.displays[i].GC1 && this.displays[i].rootWindow)
{
self._info._X11.XSetFunction(self.displays[i].display, self.displays[i].GC1, GXclear);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].rootWindow, self.displays[i].GC1, 0, 0, self.displays[i].right, 0);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].rootWindow, self.displays[i].GC1, self.displays[i].right, 0, self.displays[i].right, self.displays[i].bottom);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].rootWindow, self.displays[i].GC1, 0, self.displays[i].bottom, self.displays[i].right, self.displays[i].bottom);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].rootWindow, self.displays[i].GC1, 0, 0, 0, self.displays[i].bottom);
this._info._X11.XFlush(this.displays[i].display);
}
}
}
}
this.Start = function Start()
{
this._info.getInfo().then(function (mon)
{
self.displays = mon;
console.log(mon.length + ' displays');
for(var i = 0; i<mon.length; ++i)
{
console.log('Width: ' + mon[i].right + ', Height: ' + mon[i].bottom);
mon[i].rootWindow = self._info._X11.XRootWindow(mon[i].display, mon[i].screenId);
if (self._isUnity)
{
// We are unity, so we have to fake the borders with borderless windows
var white = self._info._X11.XWhitePixel(mon[i].display, mon[i].screenId).Val;
// Top
mon[i].window_top = self._info._X11.XCreateSimpleWindow(mon[i].display, mon[i].rootWindow, 0, 0, mon[i].right, 5, 0, white, white);
mon[i].window_top.gc = self._info._X11.XCreateGC(mon[i].display, mon[i].window_top, 0, 0);
self._info._X11.XSetLineAttributes(mon[i].display, mon[i].window_top.gc, 10, 0, 1, 1);
self._info._X11.XSetSubwindowMode(mon[i].display, mon[i].window_top.gc, 1);
self._info.unDecorateWindow(mon[i].display, mon[i].window_top);
self._info.setWindowSizeHints(mon[i].display, mon[i].window_top, 0, 0, mon[i].right, 5);
// Right
mon[i].window_right = self._info._X11.XCreateSimpleWindow(mon[i].display, mon[i].rootWindow, mon[i].right - 5, 0, 5, mon[i].bottom, 0, white, white);
mon[i].window_right.gc = self._info._X11.XCreateGC(mon[i].display, mon[i].window_right, 0, 0);
self._info._X11.XSetLineAttributes(mon[i].display, mon[i].window_right.gc, 10, 0, 1, 1);
self._info._X11.XSetSubwindowMode(mon[i].display, mon[i].window_right.gc, 1);
self._info.unDecorateWindow(mon[i].display, mon[i].window_right);
self._info.setWindowSizeHints(mon[i].display, mon[i].window_right, mon[i].right - 5, 0, 5, mon[i].bottom);
// Left
mon[i].window_left = self._info._X11.XCreateSimpleWindow(mon[i].display, mon[i].rootWindow, 0, 0, 5, mon[i].bottom, 0, white, white);
mon[i].window_left.gc = self._info._X11.XCreateGC(mon[i].display, mon[i].window_left, 0, 0);
self._info._X11.XSetLineAttributes(mon[i].display, mon[i].window_left.gc, 10, 0, 1, 1);
self._info._X11.XSetSubwindowMode(mon[i].display, mon[i].window_left.gc, 1);
self._info.unDecorateWindow(mon[i].display, mon[i].window_left);
self._info.setWindowSizeHints(mon[i].display, mon[i].window_left, 0, 0, 5, mon[i].bottom);
// Bottom
mon[i].window_bottom = self._info._X11.XCreateSimpleWindow(mon[i].display, mon[i].rootWindow, 0, mon[i].bottom - 5, mon[i].right, 5, 0, white, white);
mon[i].window_bottom.gc = self._info._X11.XCreateGC(mon[i].display, mon[i].window_bottom, 0, 0);
self._info._X11.XSetLineAttributes(mon[i].display, mon[i].window_bottom.gc, 10, 0, 1, 1);
self._info._X11.XSetSubwindowMode(mon[i].display, mon[i].window_bottom.gc, 1);
self._info.unDecorateWindow(mon[i].display, mon[i].window_bottom);
self._info.setWindowSizeHints(mon[i].display, mon[i].window_bottom, 0, mon[i].bottom - 5, mon[i].right, 5);
self._info._X11.XMapWindow(mon[i].display, mon[i].window_top);
self._info._X11.XMapWindow(mon[i].display, mon[i].window_right);
self._info._X11.XMapWindow(mon[i].display, mon[i].window_left);
self._info._X11.XMapWindow(mon[i].display, mon[i].window_bottom);
self._info.setAlwaysOnTop(mon[i].display, mon[i].rootWindow, mon[i].window_top);
self._info.hideWindowIcon(mon[i].display, mon[i].rootWindow, mon[i].window_top);
self._info.setAlwaysOnTop(mon[i].display, mon[i].rootWindow, mon[i].window_right);
self._info.hideWindowIcon(mon[i].display, mon[i].rootWindow, mon[i].window_right);
self._info.setAlwaysOnTop(mon[i].display, mon[i].rootWindow, mon[i].window_left);
self._info.hideWindowIcon(mon[i].display, mon[i].rootWindow, mon[i].window_left);
self._info.setAlwaysOnTop(mon[i].display, mon[i].rootWindow, mon[i].window_bottom);
self._info.hideWindowIcon(mon[i].display, mon[i].rootWindow, mon[i].window_bottom);
self._info._X11.XFlush(mon[i].display);
mon[i].borderState = 0;
}
else
{
// If we aren't unity, then we can just draw
mon[i].GC1 = self._info._X11.XCreateGC(mon[i].display, mon[i].rootWindow, 0, 0);
mon[i].borderState = 0;
self._info._X11.XSetForeground(mon[i].display, mon[i].GC1, self._info._X11.XWhitePixel(mon[i].display, mon[i].screenId).Val); // White
self._info._X11.XSetLineAttributes(mon[i].display, mon[i].GC1, 10, 0, 1, 1);
self._info._X11.XSetSubwindowMode(mon[i].display, mon[i].GC1, 1);
}
}
self._info._XEvent = self._info._gm.CreateVariable(192);
self._timeout = setTimeout(self._isUnity ? self.unity_drawBorder : self.timeoutHandler, 250);
});
}
this.timeoutHandler = function()
{
for (var i = 0; i < self.displays.length; ++i) {
self.displays[i].borderState = (self.displays[i].borderState + 1) % 8;
// Top
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].GC1, (self.displays[i].borderState == 0 || self.displays[i].borderState == 4) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].rootWindow, self.displays[i].GC1, 0, 0, self.displays[i].right / 2, 0);
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].GC1, (self.displays[i].borderState == 1 || self.displays[i].borderState == 5) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].rootWindow, self.displays[i].GC1, self.displays[i].right / 2, 0, self.displays[i].right, 0);
// Right
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].GC1, (self.displays[i].borderState == 2 || self.displays[i].borderState == 6) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].rootWindow, self.displays[i].GC1, self.displays[i].right, 0, self.displays[i].right, self.displays[i].bottom / 2);
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].GC1, (self.displays[i].borderState == 3 || self.displays[i].borderState == 7) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].rootWindow, self.displays[i].GC1, self.displays[i].right, self.displays[i].bottom / 2, self.displays[i].right, self.displays[i].bottom);
// Bottom
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].GC1, (self.displays[i].borderState == 5 || self.displays[i].borderState == 1) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].rootWindow, self.displays[i].GC1, 0, self.displays[i].bottom, self.displays[i].right / 2, self.displays[i].bottom);
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].GC1, (self.displays[i].borderState == 4 || self.displays[i].borderState == 0) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].rootWindow, self.displays[i].GC1, self.displays[i].right / 2, self.displays[i].bottom, self.displays[i].right, self.displays[i].bottom);
// Left
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].GC1, (self.displays[i].borderState == 7 || self.displays[i].borderState == 3) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].rootWindow, self.displays[i].GC1, 0, 0, 0, self.displays[i].bottom / 2);
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].GC1, (self.displays[i].borderState == 6 || self.displays[i].borderState == 2) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].rootWindow, self.displays[i].GC1, 0, self.displays[i].bottom / 2, 0, self.displays[i].bottom);
self._info._X11.XFlush(self.displays[i].display);
}
self._timeout = setTimeout(self._isUnity ? self.unity_drawBorder : self.timeoutHandler, 400);
}
this.unity_drawBorder = function unity_drawBorder()
{
for (var i = 0; i < self.displays.length; ++i)
{
self.displays[i].borderState = (self.displays[i].borderState + 1) % 8;
// Top
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].window_top.gc, (self.displays[i].borderState == 0 || self.displays[i].borderState == 4) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].window_top, self.displays[i].window_top.gc, 0, 0, self.displays[i].right / 2, 0);
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].window_top.gc, (self.displays[i].borderState == 1 || self.displays[i].borderState == 5) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].window_top, self.displays[i].window_top.gc, self.displays[i].right / 2, 0, self.displays[i].right, 0);
self._info._X11.XFlush(self.displays[i].display);
// Right
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].window_right.gc, (self.displays[i].borderState == 2 || self.displays[i].borderState == 6) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].window_right, self.displays[i].window_right.gc, 0, 0, 0, self.displays[i].bottom / 2);
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].window_right.gc, (self.displays[i].borderState == 3 || self.displays[i].borderState == 7) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].window_right, self.displays[i].window_right.gc, 0, self.displays[i].bottom / 2, 0, self.displays[i].bottom);
self._info._X11.XFlush(self.displays[i].display);
// Bottom
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].window_bottom.gc, (self.displays[i].borderState == 5 || self.displays[i].borderState == 1) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].window_bottom, self.displays[i].window_bottom.gc, 0, 0, self.displays[i].right / 2, 0);
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].window_bottom.gc, (self.displays[i].borderState == 4 || self.displays[i].borderState == 0) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].window_bottom, self.displays[i].window_bottom.gc, self.displays[i].right / 2, 0, self.displays[i].right, 0);
self._info._X11.XFlush(self.displays[i].display);
// Left
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].window_left.gc, (self.displays[i].borderState == 7 || self.displays[i].borderState == 3) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].window_left, self.displays[i].window_left.gc, 0, 0, 0, self.displays[i].bottom / 2);
self._info._X11.XSetForeground(self.displays[i].display, self.displays[i].window_left.gc, (self.displays[i].borderState == 6 || self.displays[i].borderState == 2) ? 0xffff00 : 0xff0000);
self._info._X11.XDrawLine(self.displays[i].display, self.displays[i].window_left, self.displays[i].window_left.gc, 0, self.displays[i].bottom / 2, 0, self.displays[i].bottom);
self._info._X11.XFlush(self.displays[i].display);
}
self._timeout = setTimeout(self._isUnity ? self.unity_drawBorder : self.timeoutHandler, 400);
}
}
switch(process.platform)
{
case 'win32':
module.exports = new windows_monitorborder();
break;
case 'linux':
module.exports = new linux_monitorborder();
break;
default:
break;
}