TAp/server.py

97 lines
2.4 KiB
Python

#!/usr/bin/env python3
import string
from http.server import HTTPServer, BaseHTTPRequestHandler
from io import BytesIO
import base64
NULL_CHAR = chr(0)
last_c = None
colors = {
"Z": ["a", "a"],
"B": ["a", "b"],
"G": ["a", "c"],
"b": ["a", "d"],
"R": ["a", "e"],
"r": ["a", "f"],
"Y": ["a", "g"],
"W": ["a", "h"]
}
def write_report(report):
with open('/dev/hidg0', 'rb+') as fd:
fd.write(report.encode())
def release_keys():
write_report(NULL_CHAR*8)
def printchar(c):
if last_c == '§':
change_color(c)
if c.islower():
write_report(NULL_CHAR*2+chr(4 + ord(c) - ord('a'))+NULL_CHAR*5)
elif c.isupper():
write_report(chr(32)+NULL_CHAR+chr(4 + ord(c) - ord('A'))+NULL_CHAR*5)
elif c.isdigit():
write_report(NULL_CHAR*2+chr(0x1e + ((ord(c) - ord('0') - 1) % 10))+NULL_CHAR*5)
elif c == ' ':
write_report(NULL_CHAR*2+chr(44)+NULL_CHAR*5)
elif c == '\n':
write_report(NULL_CHAR*2+chr(40)+NULL_CHAR*5)
else:
write_report(chr(32)+NULL_CHAR+chr(0x38)+NULL_CHAR*5)
release_keys()
def change_color(c: str):
if c in colors:
# DELETE keypress
write_report(NULL_CHAR*2+chr(0x2a)+NULL_CHAR*5)
release_keys()
write_report(NULL_CHAR*2+chr(4 + ord(colors[0]) - ord('a'))+NULL_CHAR*5)
release_keys()
write_report(NULL_CHAR*2+chr(4 + ord(colors[1]) - ord('a'))+NULL_CHAR*5)
else:
write_report(chr(32)+NULL_CHAR+chr(0x38)+NULL_CHAR*5)
def printstring(s):
for c in s:
printchar(c)
last_c = c
def reset_color():
change_color('G')
release_keys()
class SimpleHTTPRequestHandler(BaseHTTPRequestHandler):
def do_GET(self):
self.send_response(200)
self.end_headers()
self.wfile.write(b'Hello, world!')
def do_POST(self):
print(self.headers)
message = base64.b64decode(self.headers['X-Messages']).decode('utf8')
self.send_response(200)
self.end_headers()
response = BytesIO()
response.write(b'This is POST request. ')
response.write(b'Received: ')
self.wfile.write(response.getvalue())
if message == 'reset':
printstring('\n' * 32)
else:
printstring(message + '\n')
reset_color()
httpd = HTTPServer(('0.0.0.0', 8000), SimpleHTTPRequestHandler)
httpd.serve_forever()