SahlTestingFramework/TestingTools.cpp at main · islamhaqq/SahlTestingFramework · 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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
#include "TestingTools.h"

#ifdef _WIN32

#include <Windows.h>

void TestingTools::MouseClick(int x, int y) {
    SetCursorPos(x, y);

    mouse_event(MOUSEEVENTF_ABSOLUTE | MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0);
    mouse_event(MOUSEEVENTF_ABSOLUTE | MOUSEEVENTF_LEFTUP, 0, 0, 0, 0);
}

TestingTools::CursorProperties TestingTools::GetCursorPosition() {
    POINT cursorPosition;
    GetCursorPos(&cursorPosition);

    return CursorProperties{cursorPosition.x, cursorPosition.y};
}

#endif // _WIN32

#ifdef __linux__

#include <X11/extensions/XTest.h>

#include <cstring>
#include <cstdio>
#include <chrono>
#include <thread>

TestingTools::CursorProperties TestingTools::GetCursorPosition() {
    Display *x11Display = XOpenDisplay(nullptr);
    Window rootWindow = DefaultRootWindow(x11Display);

    Window rootWindowUnderMouse;
    Window childWindowUnderMouse;
    int newMouseXUnderChildWindow;
    int newMouseYUnderChildWindow;

    unsigned int bitMaskForModifierKeys;

    CursorProperties cursorProperties{};

    XQueryPointer(
            x11Display,
            rootWindow,
            &rootWindowUnderMouse,
            &childWindowUnderMouse,
            &cursorProperties.x,
            &cursorProperties.y,
            &newMouseXUnderChildWindow,
            &newMouseYUnderChildWindow,
            &bitMaskForModifierKeys
    );

    XCloseDisplay(x11Display);

    return cursorProperties;
}

Window TestingTools::FindWindowByName(Display *display, Window root, const char *name) {
    Window parent;
    Window *children;
    unsigned int numberOfChildren;
    char *windowName;

    bool baseCase = XFetchName(display, root, &windowName) && strcmp(name, windowName) == 0;
    if (baseCase) return root;

    if (XQueryTree(display, root, &root, &parent, &children, &numberOfChildren)) {
        for (unsigned int i = 0; i < numberOfChildren; i++) {
            Window result = FindWindowByName(display, children[i], name);
            if (result != None) return result;
        }
        XFree(children);
    }

    return 0;
}

XRRCrtcInfo TestingTools::GetPrimaryMonitor(Display *display) {
    Window root = DefaultRootWindow(display);

    XRRScreenResources *screenResources = XRRGetScreenResources(display, root);
    RROutput primaryOutput = XRRGetOutputPrimary(display, root);
    XRROutputInfo *outputInfo = XRRGetOutputInfo(display, screenResources, primaryOutput);

    XRRCrtcInfo primaryMonitorCrtcInfo{};

    for (int i = 0; i < screenResources->ncrtc; ++i) {
        XRRCrtcInfo *crtcInfo = XRRGetCrtcInfo(display, screenResources, screenResources->crtcs[i]);

        for (int j = 0; j < crtcInfo->noutput; ++j) {
            if (crtcInfo->outputs[j] == primaryOutput) {
                primaryMonitorCrtcInfo = *crtcInfo;
                break;
            }
        }
        XRRFreeCrtcInfo(crtcInfo);
    }

    XRRFreeOutputInfo(outputInfo);
    XRRFreeScreenResources(screenResources);

    return primaryMonitorCrtcInfo;
}

void TestingTools::MoveWindowToTopRight(Display *display, Window window, XRRCrtcInfo targetMonitor) {
    int monitorXOrigin = targetMonitor.x;
    int monitorYOrigin = targetMonitor.y;
    unsigned int monitorWidth = targetMonitor.width;

    XWindowAttributes windowAttributes;
    XGetWindowAttributes(display, window, &windowAttributes);

    XMoveWindow(display, window, monitorXOrigin + monitorWidth - windowAttributes.width, monitorYOrigin);
    XFlush(display);
    std::this_thread::sleep_for(std::chrono::milliseconds(100));
}

void TestingTools::MouseClick(int x, int y) {
    Display *x11Display = XOpenDisplay(nullptr);
    Window rootWindow = DefaultRootWindow(x11Display);
    XWarpPointer(x11Display, None, rootWindow, 0, 0, 0, 0, x, y);

    XTestFakeButtonEvent(x11Display, 1, True, 0);
    XTestFakeButtonEvent(x11Display, 1, False, 0);
    XFlush(x11Display);
}

#endif // __linux__