.
This commit is contained in:
parent
f2efc515cd
commit
5143593d4c
|
@ -105,6 +105,8 @@
|
||||||
errorbox.style.display = 'block'
|
errorbox.style.display = 'block'
|
||||||
canvas.style.display = 'none'
|
canvas.style.display = 'none'
|
||||||
console.error('Server reported error: ' + code + ': ' + msg)
|
console.error('Server reported error: ' + code + ': ' + msg)
|
||||||
|
} else if (type == 'ack') {
|
||||||
|
ws.send('ack')
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
ws.onerror = setup
|
ws.onerror = setup
|
||||||
|
|
68
wss.py
68
wss.py
|
@ -7,22 +7,61 @@ import random
|
||||||
import time
|
import time
|
||||||
import shutil
|
import shutil
|
||||||
import tempfile
|
import tempfile
|
||||||
|
import queue
|
||||||
from websocket_server import WebsocketServer
|
from websocket_server import WebsocketServer
|
||||||
|
|
||||||
def cycle():
|
def cycle():
|
||||||
try:
|
try:
|
||||||
changed = hcapi.get_img()
|
changed = hcapi.get_img()
|
||||||
except ImportError as e:
|
except Exception as e:
|
||||||
server.send_message_to_all('err%noconn%Server failed to capture screenshot')
|
#server.send_message_to_all('err%noconn%Server failed to capture screenshot')
|
||||||
|
for client in clients.values():
|
||||||
|
client.send('err%noconn%Server failed to capture screenshot')
|
||||||
time.sleep(3)
|
time.sleep(3)
|
||||||
return
|
return
|
||||||
|
|
||||||
for i in changed:
|
for i in changed:
|
||||||
threading.Thread(target=do_img, args=(i,)).start()
|
threading.Thread(target=do_img, args=(i,)).start()
|
||||||
|
|
||||||
|
for client in clients.values():
|
||||||
|
client.ack()
|
||||||
|
|
||||||
|
|
||||||
|
class Client:
|
||||||
|
def __init__(self, client, server):
|
||||||
|
self.client = client
|
||||||
|
self.server = server
|
||||||
|
self.queue = queue.Queue()
|
||||||
|
self.good = True
|
||||||
|
|
||||||
|
def send(self, item, name):
|
||||||
|
self.messages[name] = item
|
||||||
|
|
||||||
|
def ack(self):
|
||||||
|
self.queue.put(['ack'])
|
||||||
|
|
||||||
|
def cycle(self):
|
||||||
|
while not self.good:
|
||||||
|
pass
|
||||||
|
|
||||||
|
item = self.queue.get()
|
||||||
|
|
||||||
|
if item[0] == 'msg':
|
||||||
|
self.server.send_message(self.client, item[1])
|
||||||
|
else:
|
||||||
|
self.server.send_message(self.client, 'ack')
|
||||||
|
self.good = False
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
while True:
|
||||||
|
self.cycle()
|
||||||
|
clients = {}
|
||||||
|
|
||||||
def do_img(imgname):
|
def do_img(imgname):
|
||||||
server.send_message_to_all(img(imgname))
|
#server.send_message_to_all(img(imgname))
|
||||||
|
for client in clients.values():
|
||||||
|
print(client)
|
||||||
|
client.send(img(imgname), img)
|
||||||
|
|
||||||
|
|
||||||
def img(imgname):
|
def img(imgname):
|
||||||
|
@ -34,29 +73,38 @@ def img(imgname):
|
||||||
return response
|
return response
|
||||||
|
|
||||||
def new_client(client, server):
|
def new_client(client, server):
|
||||||
print('got client')
|
|
||||||
try:
|
try:
|
||||||
imgname = hcapi.get_full_img()
|
imgname = hcapi.get_full_img()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
server.send_message(client, 'err%noconn%Server failed to capture screenshot')
|
server.send_message(client, 'err%noconn%Server failed to capture screenshot')
|
||||||
return
|
return
|
||||||
|
|
||||||
|
clients[client['id']] = Client(client, server)
|
||||||
|
|
||||||
with open(imgname, 'rb') as f:
|
with open(imgname, 'rb') as f:
|
||||||
img = f.read()
|
img = f.read()
|
||||||
os.unlink(imgname)
|
os.unlink(imgname)
|
||||||
server.send_message(client, f'pic%0x0%data:imgage/jpeg;base64,{base64.b64encode(img).decode("utf-8")}')
|
server.send_message(client, f'pic%0x0%data:imgage/jpeg;base64,{base64.b64encode(img).decode("utf-8")}')
|
||||||
|
clients[client['id']] = Client(client, server)
|
||||||
|
clients[client['id']].ack()
|
||||||
|
threading.Thread(target=clients[client['id']].run).start()
|
||||||
|
|
||||||
def do_touch(client, server, message):
|
def do_touch(client, server, message):
|
||||||
password, x, y, w = message.split(' ')
|
action = message.split(' ', 1)[0]
|
||||||
if password == 'password':
|
if action == 'ack':
|
||||||
x, y, w = int(x), int(y), int(w)
|
clients[client['id']].good = True
|
||||||
hcapi.touch(x, y, w)
|
|
||||||
else:
|
else:
|
||||||
server.send_message(client, f'badpass')
|
_, password, x, y, w = message.split(' ')
|
||||||
|
if password == 'password':
|
||||||
|
x, y, w = int(x), int(y), int(w)
|
||||||
|
hcapi.touch(x, y, w)
|
||||||
|
else:
|
||||||
|
clients[client['id']].send(f'badpass')
|
||||||
|
|
||||||
def do_cycles():
|
def do_cycles():
|
||||||
while True:
|
while True:
|
||||||
cycle()
|
cycle()
|
||||||
|
time.sleep(0.2)
|
||||||
|
|
||||||
tmp = tempfile.mkdtemp(prefix="HCRA-")
|
tmp = tempfile.mkdtemp(prefix="HCRA-")
|
||||||
try:
|
try:
|
||||||
|
|
Loading…
Reference in New Issue
Block a user