Implement job assignment via WebSocket

JobServer.gd

@@ -1,8 +1,59 @@
 extends HTTPRequest
 
-func GetJob():
-	print("Fetching Job")
-	request("http://127.0.0.1:8000/template.json")
-	var res = await request_completed
-	var json = JSON.parse_string(res[3].get_string_from_utf8())
-	return json
+var socket = WebSocketPeer.new()
+
+func InitWebsocket():
+	print("Connecting websocket ")
+	var i = 0
+	while socket.get_ready_state() != WebSocketPeer.STATE_OPEN:
+		socket.connect_to_url($"/root/Main".websocket_url, TLSOptions.client_unsafe())
+		socket.poll()
+		
+		
+	socket.send_text(str({"register": 1}))
+	print("Connected")
+	while true:
+		await PollWebsocket()
+	
+
+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
+		
+