108 lines
3.7 KiB
GDScript
108 lines
3.7 KiB
GDScript
extends Node
|
|
|
|
"""
|
|
This module sets up WebRTC peer connections.
|
|
"""
|
|
|
|
onready var webrtc = WebRTCMultiplayer.new()
|
|
|
|
var signaller_client = null
|
|
var ice_servers = null
|
|
|
|
func _init(_ice_servers, _signaller_client):
|
|
ice_servers = _ice_servers
|
|
signaller_client = _signaller_client
|
|
|
|
func _ready():
|
|
signaller_client.connect("offer_received", self, "_offer_received")
|
|
signaller_client.connect("answer_received", self, "_answer_received")
|
|
signaller_client.connect("candidate_received", self, "_candidate_received")
|
|
signaller_client.connect("connected", self, "_signaller_connected")
|
|
# signaller_client.connect("websocket_disconnected", self, "disconnected")
|
|
|
|
signaller_client.connect("peer_id_set", self, "_peer_id_set")
|
|
signaller_client.connect("peer_joined", self, "_peer_joined")
|
|
# webrtc.connect("peer_disconnected", self, "peer_disconnected")
|
|
# connect("lobby_sealed", self, "lobby_sealed")
|
|
add_child(self.signaller_client)
|
|
|
|
func _signaller_connected():
|
|
print("Connected to signaller. Will be using the following ICE servers: %s", ice_servers)
|
|
|
|
func close():
|
|
webrtc.close()
|
|
signaller_client.close()
|
|
|
|
func connect_to_signaller():
|
|
signaller_client.connect_websocket()
|
|
|
|
func _create_peer(peer_id):
|
|
# if peer_id == signaller_client.peer_id: return
|
|
var peer: WebRTCPeerConnection = WebRTCPeerConnection.new()
|
|
print("New Net Peer with peer_id %d" % [peer_id])
|
|
var _peer_init_results = peer.initialize({"iceServers": ice_servers})
|
|
var _result = peer.connect("session_description_created", self, "_offer_created", [peer_id])
|
|
_result = peer.connect("ice_candidate_created", self, "_new_ice_candidate", [peer_id])
|
|
webrtc.add_peer(peer, peer_id)
|
|
var uid = webrtc.get_unique_id()
|
|
if peer_id > uid:
|
|
var offer_result = peer.create_offer()
|
|
print("Create Offer Result: %s" % offer_result)
|
|
else:
|
|
print("Did not create offer: %d <= %d" % [peer_id, uid])
|
|
return peer
|
|
|
|
func _new_ice_candidate(mid_name, index_name, sdp_name, peer_id):
|
|
print("New ICE Candidate - MID: %s, Index: %s, SDP: %s, PeerID: %s" % [mid_name, index_name, sdp_name, peer_id])
|
|
signaller_client.send_candidate(peer_id, mid_name, index_name, sdp_name)
|
|
|
|
func _offer_created(type, data, peer_id):
|
|
print("Offer created - Type: %s, PeerID: %s, Data: %s" % [type, peer_id, JSON.print(data)])
|
|
if not webrtc.has_peer(peer_id):
|
|
return
|
|
webrtc.get_peer(peer_id).connection.set_local_description(type, data)
|
|
if type == "offer":
|
|
signaller_client.send_offer(peer_id, data)
|
|
else:
|
|
signaller_client.send_answer(peer_id, data)
|
|
|
|
func _peer_id_set(peer_id):
|
|
print("Peer ID received %s - setting up to peer..." % [peer_id])
|
|
webrtc.initialize(peer_id, true)
|
|
get_tree().network_peer = webrtc
|
|
|
|
func _peer_joined(peers):
|
|
for i in range(len(peers)):
|
|
var peer = peers[i]
|
|
if peer.has("peerId"):
|
|
var peer_id = peer["peerId"]
|
|
if not webrtc.has_peer(peer_id):
|
|
_create_peer(peer_id)
|
|
|
|
func peer_disconnected(id):
|
|
print("Peer %s disconnected" % id)
|
|
if webrtc.has_peer(id):
|
|
webrtc.remove_peer(id)
|
|
|
|
func offer_received(data):
|
|
var id = data["peerId"]
|
|
if webrtc.has_peer(id):
|
|
print("Setting offer remote description: %s" % JSON.print(data))
|
|
webrtc.get_peer(id).connection.set_remote_description("offer", data["offer"])
|
|
else:
|
|
print("Received an offer for a peer with ID %s that hasn't been added" % id)
|
|
|
|
func answer_received(data):
|
|
var id = data["peerId"]
|
|
if webrtc.has_peer(id):
|
|
print("Setting answer remote description: %s" % JSON.print(data))
|
|
webrtc.get_peer(id).connection.set_remote_description("answer", data["answer"])
|
|
|
|
func candidate_received(data):
|
|
var id = data["peerId"]
|
|
if webrtc.has_peer(id):
|
|
print("Adding ice candidate: %s" % JSON.print(data))
|
|
webrtc.get_peer(id).connection.add_ice_candidate(data["mid"], data["index"], data["sdp"])
|
|
else:
|
|
print("Received candidate for non-existant peer %s" % id)
|