keeweb/app/scripts/auto-type/emitter/auto-type-emitter-win32.js

'use strict';

var Launcher = require('../../comp/launcher');

// https://msdn.microsoft.com/en-us/library/windows/desktop/dd375731(v=vs.85).aspx
var KeyMap = {
    tab: '{tab}', enter: '{enter}', space: '{space}',
    up: '{up}', down: '{down}', left: '{left}', right: '{right}', home: '{home}', end: '{end}', pgup: '{pgup}', pgdn: '{pgdn}',
    ins: '{ins}', del: '{del}', bs: '{bs}', esc: '{esc}',
    win: 0x5B, rwin: 0x5C,
    f1: '{f1}', f2: '{f2}', f3: '{f3}', f4: '{f4}', f5: '{f5}', f6: '{f6}', f7: '{f7}', f8: '{f8}', f9: '{f9}',
    f10: '{f10}', f11: '{f11}', f12: '{f12}', f13: '{f13}', f14: '{f14}', f15: '{f15}', f16: '{f16}',
    add: '{add}', subtract: '{subtract}', multiply: '{multiply}', divide: '{divide}',
    n0: '0', n1: '1', n2: '2', n3: '3', n4: '4',
    n5: '5', n6: '6', n7: '7', n8: '8', n9: '9'
};

var ModMap = {
    '^': '^',
    '+': '+',
    '%': '%',
    '^^': '^'
};

var ModVk = {
    '+': 0x10,
    '^': 0x11,
    '%': 0x12
};

var TextReplaceRegex = /[\(\)\{}\[\]\+\^%~]/g;

var AutoTypeEmitter = function(callback) {
    this.callback = callback;
    this.mod = {};
    this.pendingScript = [];
};

AutoTypeEmitter.prototype.setMod = function(mod, enabled) {
    if (enabled) {
        this.mod[ModMap[mod]] = true;
    } else {
        delete this.mod[ModMap[mod]];
    }
};

AutoTypeEmitter.prototype.text = function(text) {
    text = this.addMod(this.escapeText(text.replace(TextReplaceRegex, function(match) { return '{' + match + '}'; })));
    this.pendingScript.push('[System.Windows.Forms.SendKeys]::SendWait("' + text + '")');
    this.callback();
};

AutoTypeEmitter.prototype.key = function(key) {
    if (typeof key !== 'number') {
        key = KeyMap[key];
        if (!key) {
            return this.callback('Bad key: ' + key);
        }
    }
    if (typeof key === 'number') {
        Object.keys(this.mod).forEach(function(mod) { this.pendingScript.push('[KwHelper]::Down(' + ModVk[mod] + ')'); }, this);
        this.pendingScript.push('[KwHelper]::Press(' + key + ')');
        Object.keys(this.mod).forEach(function(mod) { this.pendingScript.push('[KwHelper]::Up(' + ModVk[mod] + ')'); }, this);
    } else {
        var text = this.addMod(key);
        this.pendingScript.push('[System.Windows.Forms.SendKeys]::SendWait("' + text + '")');
    }
    this.callback();
};

AutoTypeEmitter.prototype.copyPaste = function(text) {
    this.pendingScript.push('[System.Threading.Thread]::Sleep(500)');
    this.pendingScript.push('[System.Windows.Forms.Clipboard]::SetText("' + this.escapeText(text) + '")');
    this.pendingScript.push('[System.Threading.Thread]::Sleep(500)');
    this.pendingScript.push('[System.Windows.Forms.SendKeys]::SendWait("+{ins}")');
    this.pendingScript.push('[System.Threading.Thread]::Sleep(500)');
    this.callback();
};

AutoTypeEmitter.prototype.wait = function(time) {
    this.pendingScript.push('[System.Threading.Thread]::Sleep(' + time + ')');
    this.callback();
};

AutoTypeEmitter.prototype.waitComplete = function() {
    if (this.pendingScript.length) {
        var script = this.pendingScript.join('\n');
        this.pendingScript.length = 0;
        this.runScript(script);
    } else {
        this.callback();
    }
};

AutoTypeEmitter.prototype.setDelay = function(delay) {
    this.delay = delay || 0;
    this.callback('Not implemented');
};

AutoTypeEmitter.prototype.escapeText = function(text) {
    return text.replace(/"/g, '`"').replace(/`/g, '``').replace(/\n/g, '``n');
};

AutoTypeEmitter.prototype.addMod = function(text) {
    var keys = Object.keys(this.mod);
    if (!keys.length || !text) {
        return text;
    }
    return keys.join('') + (text.length > 1 ? '(' + text + ')' : text);
};

AutoTypeEmitter.prototype.runScript = function(script) {
    script =
'Add-Type @"\n' +
'using System;\n' +
'using System.Runtime.InteropServices;\n' +
'public static class KwHelper {\n' +
    '[DllImport("user32.dll")]\n' +
    'static extern void keybd_event(byte bVk, byte bScan, uint dwFlags, UIntPtr dwExtraInfo);\n' +
    'public static void Down(byte code) { keybd_event(code, 0, 1, UIntPtr.Zero); }\n' +
    'public static void Up(byte code) { keybd_event(code, 0, 3, UIntPtr.Zero); }\n' +
    'public static void Press(byte code) { Down(code); Up(code); }\n' +
'}\n' +
'"@\n\n' +
'[Console]::InputEncoding = [System.Text.Encoding]::UTF8\n' +
'[Console]::OutputEncoding = [System.Text.Encoding]::UTF8\n' +
'[System.Reflection.Assembly]::LoadWithPartialName("System.Windows.Forms")\n' +
script;

    Launcher.spawn({
        cmd: 'powershell',
        args: ['-Command', '-'],
        data: script,
        complete: this.callback
    });
};

module.exports = AutoTypeEmitter;