#!/usr/bin/env python

import os
import re
import subprocess

from gi.repository import GLib
from dbus import SessionBus
from dbus.mainloop.glib import DBusGMainLoop

previous_xrandr = 'xrandr --output DSI-1 --auto'
xinput_regex = re.compile(r'.*Capacitive TouchScreen\s+id=(\d+).*')


def get_current_xrandr() -> str:
    proc = subprocess.Popen(['unxrandr'], stdout=subprocess.PIPE)
    return proc.communicate()[0].decode().strip()


def get_touchscreen_id() -> int:
    proc = subprocess.Popen(['xinput'], stdout=subprocess.PIPE)
    for line in proc.communicate()[0].decode().split('\n'):
        if m := xinput_regex.match(line):
            return int(m.group(1))

    assert 'Touchscreen device not found'


def is_display_on() -> bool:
    proc = subprocess.Popen(['xrandr', '-q'], stdout=subprocess.PIPE)
    out = proc.communicate()[0].decode().strip()
    lines = [
        line for line in out.split('\n')
        if line.startswith('DSI-1 ')
    ]

    assert lines, 'Display DSI-1 not found'
    m = re.search(r'^DSI-1\s+(connected\s+)?(primary\s+)?\d+x\d+\+\d+\+\d+', lines[0])
    return m is not None


def signal_handler(name, _):
    global previous_xrandr
    if name != 'button':
        return

    is_on = is_display_on()
    touchscreen_id = get_touchscreen_id()

    if is_on:
        previous_xrandr = get_current_xrandr()
        os.system('xrandr --output DSI-1 --off')
        os.system(f'xinput disable {touchscreen_id}')
    else:
        os.system(previous_xrandr)
        os.system(f'xinput enable {touchscreen_id}')


def main():
    loop = DBusGMainLoop()
    bus = SessionBus(mainloop=loop)
    bus.add_signal_receiver(
        signal_handler, 'updateEvent',
        'io.cutiepi.interface', path='/mcu'
    )

    loop = GLib.MainLoop()

    try:
        loop.run()
    finally:
        loop.quit()


if __name__ == '__main__':
    main()