67 lines
1.8 KiB
GDScript
67 lines
1.8 KiB
GDScript
extends HTTPRequest
|
|
|
|
var socket = WebSocketPeer.new()
|
|
|
|
func InitWebsocket():
|
|
print("Connecting websocket ")
|
|
|
|
socket.connect_to_url($"/root/Main".websocket_url, TLSOptions.client_unsafe())
|
|
while socket.get_ready_state() != WebSocketPeer.STATE_OPEN:
|
|
socket.poll()
|
|
|
|
var init_message = {
|
|
"register": {
|
|
"version": 1,
|
|
"auth_key": $"/root/Main".auth_key
|
|
}
|
|
}
|
|
socket.send_text(str(init_message))
|
|
print("Connected")
|
|
while true:
|
|
await PollWebsocket()
|
|
RenderingServer.force_draw()
|
|
DisplayServer.process_events()
|
|
|
|
|
|
func PollWebsocket():
|
|
var main = $"/root/Main"
|
|
socket.poll()
|
|
var state = socket.get_ready_state()
|
|
if state == WebSocketPeer.STATE_OPEN:
|
|
while socket.get_available_packet_count():
|
|
var packet = socket.get_packet()
|
|
var json = JSON.parse_string(packet.get_string_from_utf8())
|
|
print("Packet: ", packet.slice(0,5), "...")
|
|
if "welcome" in json:
|
|
print("Registered as client %s" % json["welcome"]["clientId"])
|
|
if "job" in json:
|
|
var job = json["job"]
|
|
print("Recevied Job ", job["jobId"])
|
|
main.rendering = true
|
|
var result = await $"../Renderer".RenderJob(job)
|
|
var response = {
|
|
"result": {
|
|
"jobId": job["jobId"],
|
|
"path": $"/root/Main".public_path + result
|
|
}
|
|
}
|
|
socket.send_text(str(response))
|
|
print("Sent result ", result)
|
|
main.rendering = false
|
|
elif state == WebSocketPeer.STATE_CLOSING:
|
|
# Keep polling to achieve proper close.
|
|
pass
|
|
elif state == WebSocketPeer.STATE_CLOSED:
|
|
var code = socket.get_close_code()
|
|
var reason = socket.get_close_reason()
|
|
print("WebSocket closed with code: %d, reason %s. Clean: %s" % [code, reason, code != -1])
|
|
print("Attempting reconnect in 10")
|
|
set_process(false) # Stop processing.
|
|
await get_tree().create_timer(10).timeout
|
|
InitWebsocket()
|
|
|
|
func _process(_delta):
|
|
pass
|
|
|
|
|