Rocknix 中屏幕键盘的配置

#!/usr/bin/env python3
import struct
import subprocess
import time
import os
import glob
import sys

# Format: time_sec, time_usec, type, code, value
FORMAT = 'iiHHi' if struct.calcsize('l') == 4 else 'llHHi'
EVENT_SIZE = struct.calcsize(FORMAT)

# Keycodes
BTN_MODE = 316
KEY_RECORD = 167
CONFIRMED_SWAYSOCK = "/run/0-runtime-dir/sway-ipc.0.sock"

def log(msg):
    """Compact logging: writes message and flushes immediately without extra newlines."""
    sys.stdout.write(f"{msg}\n")
    sys.stdout.flush()

def get_gamepad_device():
    """Scan all input devices to find the Xbox controller dynamically."""
    for dev in glob.glob('/dev/input/event*'):
        try:
            with open(f'/sys/class/input/{os.path.basename(dev)}/device/name', 'r') as f:
                name = f.read().strip()
                if "Xbox" in name:
                    return dev
        except:
            continue
    return None

def setup_wayland_env():
    """Inject required Sway environment variables."""
    os.environ['SWAYSOCK'] = CONFIRMED_SWAYSOCK
    os.environ['XDG_RUNTIME_DIR'] = "/run/0-runtime-dir"
    os.environ['WAYLAND_DISPLAY'] = "wayland-1"

def toggle_keyboard():
    log("Combo triggered. Toggling keyboard.")
    setup_wayland_env()
    check_proc = subprocess.run(['pgrep', '-x', 'wvkbd-mobintl'], stdout=subprocess.DEVNULL)
    if check_proc.returncode == 0:
        log("Keyboard running. Killing...")
        subprocess.run(['killall', 'wvkbd-mobintl'])
    else:
        log("Keyboard not running. Launching...")
        try:
            subprocess.Popen(['swaymsg', 'exec', 'wvkbd-mobintl --output DSI-1 -L 500'])
        except Exception as e:
            log(f"Launch error: {e}")

def main():
    mode_held = False
    log("Service starting...")

    while True:
        device_path = get_gamepad_device()
        if not device_path:
            log("Searching for Xbox controller...")
            time.sleep(3)
            continue

        try:
            with open(device_path, 'rb') as f:
                log(f"Success: Hooked into {device_path}")
                while True:
                    event = f.read(EVENT_SIZE)
                    if len(event) == EVENT_SIZE:
                        _, _, ev_type, ev_code, ev_value = struct.unpack(FORMAT, event)
                        if ev_type == 1:
                            if ev_code == BTN_MODE:
                                mode_held = (ev_value == 1)
                            elif ev_code == KEY_RECORD and ev_value == 1:
                                if mode_held:
                                    toggle_keyboard()
        except Exception as e:
            log(f"Device error or lost: {e}. Re-scanning...")
            time.sleep(3)

if __name__ == '__main__':
    main()

systemctl daemon-reload
systemctl enable kb-combo.service
systemctl start kb-combo.service