class_name WebSocketServer
extends Node
signal message_received(peer_id: int, message: String)
signal client_connected(peer_id: int)
signal client_disconnected(peer_id: int)
@export var handshake_headers := PackedStringArray()
@export var supported_protocols := PackedStringArray()
@export var handshake_timout := 3000
@export var use_tls := false
@export var tls_cert: X509Certificate
@export var tls_key: CryptoKey
@export var refuse_new_connections := false:
set(refuse):
if refuse:
pending_peers.clear()
class PendingPeer:
var connect_time: int
var tcp: StreamPeerTCP
var connection: StreamPeer
var ws: WebSocketPeer
func _init(p_tcp: StreamPeerTCP) -> void:
tcp = p_tcp
connection = p_tcp
connect_time = Time.get_ticks_msec()
var tcp_server := TCPServer.new()
var pending_peers: Array[PendingPeer] = []
var peers: Dictionary
func listen(port: int) -> int:
assert(not tcp_server.is_listening())
return tcp_server.listen(port)
func stop() -> void:
tcp_server.stop()
pending_peers.clear()
peers.clear()
func send(peer_id: int, message: String) -> int:
var type := typeof(message)
if peer_id <= 0:
# Send to multiple peers, (zero = broadcast, negative = exclude one).
for id: int in peers:
if id == -peer_id:
continue
if type == TYPE_STRING:
peers[id].send_text(message)
else:
peers[id].put_packet(message)
return OK
assert(peers.has(peer_id))
var socket: WebSocketPeer = peers[peer_id]
if type == TYPE_STRING:
return socket.send_text(message)
return socket.send(var_to_bytes(message))
func get_message(peer_id: int) -> Variant:
assert(peers.has(peer_id))
var socket: WebSocketPeer = peers[peer_id]
if socket.get_available_packet_count() < 1:
return null
var pkt: PackedByteArray = socket.get_packet()
if socket.was_string_packet():
return pkt.get_string_from_utf8()
return bytes_to_var(pkt)
func has_message(peer_id: int) -> bool:
assert(peers.has(peer_id))
return peers[peer_id].get_available_packet_count() > 0
func _create_peer() -> WebSocketPeer:
var ws := WebSocketPeer.new()
ws.supported_protocols = supported_protocols
ws.handshake_headers = handshake_headers
return ws
func poll() -> void:
if not tcp_server.is_listening():
return
while not refuse_new_connections and tcp_server.is_connection_available():
var conn: StreamPeerTCP = tcp_server.take_connection()
assert(conn != null)
pending_peers.append(PendingPeer.new(conn))
var to_remove := []
for p in pending_peers:
if not _connect_pending(p):
if p.connect_time + handshake_timout < Time.get_ticks_msec():
# Timeout.
to_remove.append(p)
continue # Still pending.
to_remove.append(p)
for r: RefCounted in to_remove:
pending_peers.erase(r)
to_remove.clear()
for id: int in peers:
var p: WebSocketPeer = peers[id]
p.poll()
if p.get_ready_state() != WebSocketPeer.STATE_OPEN:
client_disconnected.emit(id)
to_remove.append(id)
continue
while p.get_available_packet_count():
message_received.emit(id, get_message(id))
for r: int in to_remove:
peers.erase(r)
to_remove.clear()
func _connect_pending(p: PendingPeer) -> bool:
if p.ws != null:
# Poll websocket client if doing handshake.
p.ws.poll()
var state := p.ws.get_ready_state()
if state == WebSocketPeer.STATE_OPEN:
var id := randi_range(2, 1 << 30)
peers[id] = p.ws
client_connected.emit(id)
return true # Success.
elif state != WebSocketPeer.STATE_CONNECTING:
return true # Failure.
return false # Still connecting.
elif p.tcp.get_status() != StreamPeerTCP.STATUS_CONNECTED:
return true # TCP disconnected.
elif not use_tls:
# TCP is ready, create WS peer.
p.ws = _create_peer()
p.ws.accept_stream(p.tcp)
return false # WebSocketPeer connection is pending.
else:
if p.connection == p.tcp:
assert(tls_key != null and tls_cert != null)
var tls := StreamPeerTLS.new()
tls.accept_stream(p.tcp, TLSOptions.server(tls_key, tls_cert))
p.connection = tls
p.connection.poll()
var status: StreamPeerTLS.Status = p.connection.get_status()
if status == StreamPeerTLS.STATUS_CONNECTED:
p.ws = _create_peer()
p.ws.accept_stream(p.connection)
return false # WebSocketPeer connection is pending.
if status != StreamPeerTLS.STATUS_HANDSHAKING:
return true # Failure.
return false
func _process(_delta: float) -> void:
poll()
' href='#n10'>10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
name: "Main Area"
panels {
name: "GEMSTONE"
path: "Panels/Main/panel_1"
clue: "gemstone"
answer: "ruby"
symbols: EXAMPLE
proxies { answer: "jade" path: "Panels/J/panel_1" }
}
panels {
name: "PULL"
path: "Panels/Main/panel_2"
clue: "pull"
answer: "yank"
symbols: PYRAMID
proxies { answer: "jerk" path: "Panels/J/panel_2" }
}
panels {
name: "LOIN"
path: "Panels/Main/panel_3"
clue: "loin"
answer: "coin"
symbols: ZERO
proxies { answer: "join" path: "Panels/J/panel_3" }
}
panels {
name: "SMALL"
path: "Panels/Main/panel_4"
clue: "small"
answer: "large"
symbols: SUN
proxies { answer: "jumbo" path: "Panels/J/panel_4" }
}
panels {
name: "HOP"
path: "Panels/Main/panel_5"
clue: "hop"
answer: "leap"
symbols: PYRAMID
proxies { answer: "jump" path: "Panels/J/panel_5" }
}
panels {
name: "UNFAIR"
path: "Panels/Main/panel_6"
clue: "unfair"
answer: "fair"
symbols: SUN
proxies { answer: "just" path: "Panels/J/panel_6" }
}
panels {
name: "SPRINT"
path: "Panels/Main/panel_7"
clue: "sprint"
answer: "run"
symbols: PYRAMID
proxies { answer: "jog" path: "Panels/J/panel_7" }
}
panels {
name: "MINOR"
path: "Panels/Main/panel_8"
clue: "minor"
answer: "adult"
symbols: SUN
proxies { answer: "major" path: "Panels/J/panel_8" }
}
panels {
name: "BIRD"
path: "Panels/Main/panel_9"
clue: "bird"
answer: "owl"
symbols: EXAMPLE
proxies { answer: "jay" path: "Panels/J/panel_9" }
}
panels {
name: "TREE"
path: "Panels/Main/panel_10"
clue: "tree"
answer: "forest"
symbols: BOXES
proxies { answer: "jungle" path: "Panels/J/panel_10" }
}
panels {
name: "ORANGE"
path: "Panels/Main/panel_11"
clue: "orange"
answer: "fruit"
symbols: EXAMPLE
proxies { answer: "juice" path: "Panels/J/panel_11" }
}
panels {
name: "QUEEN"
path: "Panels/Main/panel_12"
clue: "queen"
answer: "king"
symbols: EXAMPLE
proxies { answer: "jack" path: "Panels/J/panel_12" }
}
ports {
name: "GREAT"
path: "Components/Warps/worldport"
destination { x: -3 y: 0 z: 9 }
rotation: 270
}
|