74 lines
1.8 KiB
Python
74 lines
1.8 KiB
Python
|
import os
|
||
|
|
import base64
|
||
|
|
import hcapi
|
||
|
|
import threading
|
||
|
|
import logging
|
||
|
|
import random
|
||
|
|
import time
|
||
|
|
import shutil
|
||
|
|
import tempfile
|
||
|
|
from websocket_server import WebsocketServer
|
||
|
|
|
||
|
|
def cycle():
|
||
|
|
try:
|
||
|
|
changed = hcapi.get_img()
|
||
|
|
except ImportError as e:
|
||
|
|
server.send_message_to_all('err%noconn%Server failed to capture screenshot')
|
||
|
|
time.sleep(3)
|
||
|
|
return
|
||
|
|
|
||
|
|
for i in changed:
|
||
|
|
threading.Thread(target=do_img, args=(i,)).start()
|
||
|
|
|
||
|
|
|
||
|
|
def do_img(imgname):
|
||
|
|
server.send_message_to_all(img(imgname))
|
||
|
|
|
||
|
|
|
||
|
|
def img(imgname):
|
||
|
|
with open(f'pieces/{imgname}.jpg', 'rb') as f:
|
||
|
|
img = f.read()
|
||
|
|
|
||
|
|
response = f'pic%{imgname}%data:imgage/jpeg;base64,{base64.b64encode(img).decode("utf-8")}'
|
||
|
|
|
||
|
|
return response
|
||
|
|
|
||
|
|
def new_client(client, server):
|
||
|
|
print('got client')
|
||
|
|
try:
|
||
|
|
imgname = hcapi.get_full_img()
|
||
|
|
except Exception as e:
|
||
|
|
server.send_message(client, 'err%noconn%Server failed to capture screenshot')
|
||
|
|
return
|
||
|
|
|
||
|
|
with open(imgname, 'rb') as f:
|
||
|
|
img = f.read()
|
||
|
|
os.unlink(imgname)
|
||
|
|
server.send_message(client, f'pic%0x0%data:imgage/jpeg;base64,{base64.b64encode(img).decode("utf-8")}')
|
||
|
|
|
||
|
|
def do_touch(client, server, message):
|
||
|
|
password, x, y, w = message.split(' ')
|
||
|
|
if password == 'password':
|
||
|
|
x, y, w = int(x), int(y), int(w)
|
||
|
|
hcapi.touch(x, y, w)
|
||
|
|
else:
|
||
|
|
server.send_message(client, f'badpass')
|
||
|
|
|
||
|
|
def do_cycles():
|
||
|
|
while True:
|
||
|
|
cycle()
|
||
|
|
|
||
|
|
tmp = tempfile.mkdtemp(prefix="HCRA-")
|
||
|
|
try:
|
||
|
|
shutil.copy('crops.json', tmp)
|
||
|
|
os.chdir(tmp)
|
||
|
|
os.mkdir('pieces')
|
||
|
|
server = WebsocketServer(1234, host='0.0.0.0', loglevel=logging.INFO)
|
||
|
|
threading.Thread(target=do_cycles).start()
|
||
|
|
server.set_fn_new_client(new_client)
|
||
|
|
server.set_fn_message_received(do_touch)
|
||
|
|
server.run_forever()
|
||
|
|
finally:
|
||
|
|
os.chdir('/')
|
||
|
|
shutil.rmtree(tmp)
|