#!/usr/bin/env python3
#   input-over-ssh: Mouse and keyboard forwarding over SSH
#   Copyright © 2019  Lee Yingtong Li (RunasSudo)
#
#   This program is free software: you can redistribute it and/or modify
#   it under the terms of the GNU Affero General Public License as published by
#   the Free Software Foundation, either version 3 of the License, or
#   (at your option) any later version.
#
#   This program is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU Affero General Public License for more details.
#
#   You should have received a copy of the GNU Affero General Public License
#   along with this program.  If not, see <https://www.gnu.org/licenses/>.

MOUSEHOLD_X = 1366//2
MOUSEHOLD_Y = 728//2
MOUSEHOLD_W = 0
MOUSEHOLD_H = 0

DEVICE_MOUSE = 'Logitech USB Optical Mouse'
DEVICE_KEYBOARD = 'AT Translated Set 2 keyboard'

import asyncio
import re
import subprocess
import sys
import time
import tkinter

if MOUSEHOLD_W > 0:
	root = tkinter.Tk()
	root.geometry('{}x{}+{}+{}'.format(MOUSEHOLD_W, MOUSEHOLD_H, MOUSEHOLD_X - MOUSEHOLD_W//2, MOUSEHOLD_Y - MOUSEHOLD_H//2))
	root.update()

p_unclutter = None
async def hide_cursor():
	global p_unclutter
	p_unclutter = await asyncio.create_subprocess_exec('unclutter', '--timeout', '0', '--jitter', '9999', '--ignore-buttons', '1,2,3,4,5')
	#await p.wait()

async def mousemove():
	last_mousemove = time.time()
	
	while True:
		p = await asyncio.create_subprocess_exec('xdotool', 'getmouselocation', stdout=subprocess.PIPE)
		stdout, _ = await p.communicate()
		result = {x[:x.index(':')]: int(x[x.index(':')+1:]) for x in stdout.decode('utf-8').strip().split(' ')}
		
		dx, dy = result['x'] - MOUSEHOLD_X, result['y'] - MOUSEHOLD_Y
		if dx != 0 or dy != 0:
			print(dx, dy, file=sys.stderr)
			print('DISPLAY=:0 xdotool mousemove_relative -- {} {}'.format(dx, dy))
		
		p = await asyncio.create_subprocess_exec('xdotool', 'mousemove', str(MOUSEHOLD_X), str(MOUSEHOLD_Y))
		await p.wait()
		
		wait_s = (last_mousemove + 1/30) - time.time()
		last_mousemove = time.time()
		if wait_s > 0:
			await asyncio.sleep(wait_s)

async def mouseinp():
	# Get mouse device
	p = await asyncio.create_subprocess_exec('xinput', 'list', stdout=subprocess.PIPE)
	stdout, _ = await p.communicate()
	device = re.search(DEVICE_MOUSE + '.*?id=([0-9]+)', stdout.decode('utf-8')).group(1)
	print('Mouse device {}'.format(device), file=sys.stderr)
	
	p_test = await asyncio.create_subprocess_exec('xinput', 'test', device, stdout=subprocess.PIPE)
	while True:
		line = (await p_test.stdout.readline()).decode('utf-8').strip()
		if line.startswith('button press'):
			print(line, file=sys.stderr)
			button = line.split()[-1]
			print('DISPLAY=:0 xdotool mousedown {}'.format(button))
		elif line.startswith('button release'):
			print(line, file=sys.stderr)
			button = line.split()[-1]
			print('DISPLAY=:0 xdotool mouseup {}'.format(button))

async def keyboardinp():
	is_ctrl = False # 37
	
	# Get keyboard device
	p = await asyncio.create_subprocess_exec('xinput', 'list', stdout=subprocess.PIPE)
	stdout, _ = await p.communicate()
	device = re.search(DEVICE_KEYBOARD + '.*?id=([0-9]+)', stdout.decode('utf-8')).group(1)
	print('Keyboard device {}'.format(device), file=sys.stderr)
	
	p_test = await asyncio.create_subprocess_exec('xinput', 'test', device, stdout=subprocess.PIPE)
	while True:
		line = (await p_test.stdout.readline()).decode('utf-8').strip()
		if line.startswith('key press'):
			print(line, file=sys.stderr)
			button = line.split()[-1]
			
			if button == '37':
				is_ctrl = True
			if button == '24' and is_ctrl: # Ctrl+Q
				print('DISPLAY=:0 xdotool keyup 37') # Release Ctrl
				await p_unclutter.terminate()
				sys.exit(0)
				return
			
			# Add leading 0 to force interpretation as key code
			print('DISPLAY=:0 xdotool keydown 0{}'.format(button))
		elif line.startswith('key release'):
			print(line, file=sys.stderr)
			button = line.split()[-1]
			
			if button == '37':
				is_ctrl = False
			
			print('DISPLAY=:0 xdotool keyup 0{}'.format(button))

async def main():
	asyncio.create_task(hide_cursor())
	asyncio.create_task(mousemove())
	asyncio.create_task(mouseinp())
	await asyncio.create_task(keyboardinp())

asyncio.run(main())