OKey.py

import os
import evdev
os.environ["PYGAME_HIDE_SUPPORT_PROMPT"] = "1"
import pygame

import threading
import sounddevice as sd

from consts import SETTINGS, CONFIG_FILE, TERMINAL
import face
import mic
import twitch
ctrl, mod, alt, shift = False, False, False, False
mouse_clicked = False
specials = False
frames_left_to_show_text = 0
frame_time = 0
events = ""
specials = False
stream = None

def display(text: str):
    global frames_left_to_show_text, ctrl, alt, mod, shift, events, specials
    if "ctrl" in text.lower():
        if text.endswith("down"):
            ctrl = True
        elif text.endswith("up"):
            ctrl = False
        return
    elif "shift" in text.lower():
        if text.endswith("down"):
            shift = True
        elif text.endswith("up"):
            shift = False
        return
    elif "alt" in text.lower():
        if text.endswith("down"):
            alt = True
        elif text.endswith("up"):
            alt = False
        return
    elif "meta" in text.lower():
        if text.endswith("down"):
            mod = True
        elif text.endswith("up"):
            mod = False
        return

    if not text.endswith("down"):
        return
    frames_left_to_show_text = SETTINGS["clear_text_frames"]
    text = text[text.index("(")+1:text.index(")")].removeprefix("KEY_").lower()
    if SETTINGS["change"] is not None and text in SETTINGS["change"]:
        text = SETTINGS["change"][text]
    if shift:
        if SETTINGS["shift_change"] is not None and text in SETTINGS["shift_change"]:
            text = SETTINGS["shift_change"][text]
        else:
            text = text.upper()
    was_special = specials
    specials = ctrl or mod or alt
    if specials: 
        text = f"{SETTINGS['ctrl_text'] if ctrl else ""}{SETTINGS['mod_text'] if mod else ""}{SETTINGS['alt_text'] if alt else ""}{text}"
        events = text
    else:
        if was_special:
            events = ""
        events += text
    if len(events)>SETTINGS["text_len"]:
        events = events[-SETTINGS["text_len"]:]
    if SETTINGS["shift_change"] is not None and events in SETTINGS["workflow_change"]:
        events = SETTINGS["workflow_change"][events]

def render(text):
    global frames_left_to_show_text
    if twitch.message_frames>0:
        return twitch.twitch_msg.center(SETTINGS["text_len"]) 
    if frames_left_to_show_text>0:
        return text.center(SETTINGS["text_len"]) 
    return face.face(frame_time+1, mouse_clicked, mic.talking_frames>0)

pygame.init()
screen = pygame.display.set_mode(SETTINGS["window_size"])
pygame.display.set_caption(SETTINGS["window_name"])
if SETTINGS["mic_on"]:
    stream = sd.InputStream(callback=mic.callback)

PYGAME_FONT = pygame.font.Font(SETTINGS["font_path"], SETTINGS["font_size"])

clock = pygame.time.Clock()


keyboard = mouse = None
devices = [evdev.InputDevice(path) for path in evdev.list_devices()]
if len(devices)==0:
    print(f"{TERMINAL['error']}ERROR: No devices are found :( This is probably caused by this user not being added to the input group{TERMINAL['normal']}")
if SETTINGS["show_basic_settings_on_start"]:
    print(f"Tracking mouse = {SETTINGS["mouse_on"]}")
    print(f"Tracking whether talking = {SETTINGS["mic_on"]}")
    print(f"Checking twitch chat with iirc = {SETTINGS["using_twitch"]}")
    print()

for device in devices:
    if SETTINGS["keyboard_name"].lower() in device.name.lower() and keyboard is None:
        keyboard = evdev.InputDevice(device.path)
        if SETTINGS["show_basic_settings_on_start"]:
            print(f"{TERMINAL['green']} Keyboard: {device.name}{TERMINAL['normal']}")
        if SETTINGS["keyboard_name"] != device.name:
            print(f"{TERMINAL['warn']}WARNING: '{device.name}' isn't '{SETTINGS["keyboard_name"]}', if this is the correct keyboard, set keyboard_name to '{device.name}'.{TERMINAL['normal']}")
        continue
    elif SETTINGS["mouse_name"].lower() in device.name.lower() and mouse is None:
        mouse = evdev.InputDevice(device.path)
        if SETTINGS["show_basic_settings_on_start"]:
            print(f"{TERMINAL['green']}󰍽 Mouse: {device.name}{TERMINAL['normal']}")
        if SETTINGS["mouse_name"] != device.name:
            print(f"{TERMINAL['warn']}WARNING: '{device.name}' isn't '{SETTINGS["mouse_name"]}', if this is the correct mouse, set mouse_name to '{device.name}'.{TERMINAL['normal']}")
        continue
    if SETTINGS["show_basic_settings_on_start"]:
        print(device.name)

def listen_keyboard():
    global events
    if keyboard is None:
        print(f"{TERMINAL['error']}ERROR: Couldn't find matching keyboard input for '{SETTINGS["keyboard_name"]}' :c Try setting the keyboard_name in {CONFIG_FILE} to one of the printed devices on startup{TERMINAL['normal']}")
        return
    for event in keyboard.read_loop():
        if event.type == evdev.ecodes.EV_KEY:
            key_event = str(evdev.categorize(event))
            display(key_event)
def listen_mouse():
    global events, mouse_clicked
    if mouse is None:
        print(f"{TERMINAL['error']}ERROR: Couldn't find matching mouse input for '{SETTINGS['mouse_name']}' :c Try setting the mouse_name in {CONFIG_FILE} to one of the printed devices on startup, or have mouse_on set to false{TERMINAL['normal']}")
        return
    for event in mouse.read_loop():
        if event.type == evdev.ecodes.EV_KEY:
            key_event = str(evdev.categorize(event))
            if key_event.endswith("down"):
                mouse_clicked = True
            elif key_event.endswith("up"):
                mouse_clicked = False

# Start threads to listen on keyboard and mouse
threading.Thread(target=listen_keyboard, daemon=True).start()
if SETTINGS["mouse_on"]:
    threading.Thread(target=listen_mouse, daemon=True).start()
if SETTINGS["using_twitch"]:
    threading.Thread(target=twitch.start, daemon=True).start()

running = True
if stream is not None:
    stream.start()
while running:
    if twitch.message_frames>0:
        screen.fill(SETTINGS["twitch_color"])
    else:
        screen.fill(SETTINGS["bg_color"])

    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False

    text_surface = PYGAME_FONT.render(render(events), SETTINGS["text_antialias"], SETTINGS["text_color"])
    screen.blit(text_surface, SETTINGS["text_pos"])

    pygame.display.flip()
    if frames_left_to_show_text>0:
        frames_left_to_show_text -= 1
    else:
        events = ""

    if mic.talking_frames>0:
        mic.talking_frames -= 1
    if twitch.message_frames>0:
        twitch.message_frames -= 1

    frame_time = (frame_time+1)%face.FRAMES_FOR_ONE_MOVE
    clock.tick(SETTINGS["fps"])

pygame.quit()
if stream is not None:
    stream.stop()
    stream.close()