guanana / pikvm-lib Goto Github PK

import ssl, time
import random
import json
import os

host = os.environ.get('HOST')
user = os.environ.get('USER', 'admin')
password = os.environ.get('PASSWORD', 'admin')

screen_width = 1920
screen_height = 1080

uri = f'wss://{host}/api/ws?stream=0'
print(f'connecting to {user}@{host} uri: {uri}')

headers = {"X-KVMD-User": user, "X-KVMD-Passwd": password}
print('headers', headers)

ws = websocket.WebSocket(sslopt={"cert_reqs": ssl.CERT_NONE})
ws.connect(uri, header=headers)

def scale_mouse_xy_to_i16(screenx, screeny, width=1920, height=1080):
    # Map from (0, width) to (i16)
    hid_x_i16 = int((screenx / width) * 0xffff - 0x8000)

    # Map from (0, height) to (i16)
    hid_y_i16 = int((screeny / height) * 0xffff - 0x8000)

    return (hid_x_i16, hid_y_i16)

def mouse_jump_to(screenx, screeny, width=1920, height=1080):
    print('mouse_move', screenx, screeny)

    kvmx, kvmy = scale_mouse_xy_to_i16(screenx, screeny)

    msg = json.dumps({
            "event_type": "mouse_move",
            "event": {
                "to": {
                    "x": kvmx,
                    "y": kvmy
                },
            }
        })
    ws.send(msg)

while True:
    for y in range(0, screen_height, 10):
        for x in range(0, screen_width, 19):
            mouse_jump_to(x, y)
            time.sleep(0.1)

ws.close()