sublime-browser-integration/browser_integration_inject.py at master · apiad/sublime-browser-integration · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
from .browser_integration import *


browser_macros_js = """
if (!window.biMacros) {
    window.biMacros = {};

    window.biMacros.createXPathFromElement = function (elm) {
        var allNodes = document.getElementsByTagName('*');

        for (segs = []; elm && elm.nodeType == 1; elm = elm.parentNode)
        {
            if (elm.hasAttribute('id')) {
                    var uniqueIdCount = 0;
                    for (var n=0;n < allNodes.length;n++) {
                        if (allNodes[n].hasAttribute('id') &&
                            allNodes[n].id == elm.id)
                            uniqueIdCount++;
                        if (uniqueIdCount > 1) break;
                    };
                    if (uniqueIdCount == 1) {
                        segs.unshift('id("' + elm.getAttribute('id') + '")');
                        return segs.join('/');
                    } else {
                        segs.unshift(elm.localName.toLowerCase() +
                            '[@id="' + elm.getAttribute('id') + '"]');
                    }
            } else if (elm.hasAttribute('class')) {
                segs.unshift(elm.localName.toLowerCase() +
                    '[@class="' + elm.getAttribute('class') + '"]');
            } else {
                for (i = 1, sib = elm.previousSibling; sib;
                     sib = sib.previousSibling) {
                    if (sib.localName == elm.localName)  i++; };
                    segs.unshift(elm.localName.toLowerCase() + '[' + i + ']');
            };
        };

        return segs.length ? '/' + segs.join('/') : null;
    };

    window.biMacros.eventHandler = function (evt) {
        if (!localStorage.__bi_tracking_events) return;

        var counter = Number(localStorage.__bi_counter);
        localStorage.__bi_counter = counter + 1;

        var time = + new Date();
        var delay = time - Number(localStorage['__bi_event_time'])

        var eventData = {
            type: evt.type,
            x: evt.offsetX,
            y: evt.offsetY,
            btn: evt.which,
            altKey: evt.altKey,
            metaKey: evt.metaKey,
            ctrlKey: evt.ctrlKey,
            shiftKey: evt.shiftKey,
            keyCode: evt.keyCode,
            charCode: evt.charCode,
            which: evt.which,
            delay: delay,
            idx: counter,
            el: window.biMacros.createXPathFromElement(evt.target),
        }

        localStorage['__bi_event_' + counter] = JSON.stringify(eventData);
        localStorage['__bi_event_time'] = time;
    }

    document.addEventListener('mousedown', window.biMacros.eventHandler);
    document.addEventListener('mouseup', window.biMacros.eventHandler);
    document.addEventListener('keypress', window.biMacros.eventHandler);
}
"""


@async
def inject_browser_macros_js():
    while True:
        if browser.connected() and browser.recording:
            browser.execute(browser_macros_js)
        else:
            return

        time.sleep(0.1)