godot-webrtc-mplayer-testing/multiplayer_client.gd

extends Node

"""
This module sets up WebRTC peer connections.
"""

var multiplayer_url = "ws://localhost:8888"
var webrtc_ice_servers = [
	{ "urls": ["stun:stun.l.google.com:19302"] }
]

const SignallerClient = preload("signaller_client.gd")

onready var mp = WebRTCMultiplayer.new()
onready var signaller = SignallerClient.new()

func _ready():
	# signaller.connect("websocket_connected", self, "connected")
	# signaller.connect("websocket_disconnected", self, "disconnected")

	signaller.connect("offer_received", self, "offer_received")
	signaller.connect("answer_received", self, "answer_received")
	signaller.connect("candidate_received", self, "candidate_received")

	signaller.connect("lobby_joined", self, "_lobby_joined")
	# connect("lobby_sealed", self, "lobby_sealed")
	signaller.connect("peer_joined", self, "_peer_joined")
	# connect("peer_disconnected", self, "peer_disconnected")
	add_child(signaller)

func close():
	mp.close()
	signaller.close()

func connect_to_signaller():
	close()
	signaller.connect_to_websocket_signaller(multiplayer_url)

func _create_peer(peer_id):
	var peer: WebRTCPeerConnection = WebRTCPeerConnection.new()
	print("New Net Peer %s with peer_id %d" % [peer, peer_id])
	var _peer_init_results = peer.initialize({"iceServers": webrtc_ice_servers})
	peer.connect("session_description_created", self, "_offer_created", [peer_id])
	peer.connect("ice_candidate_created", self, "_new_ice_candidate", [peer_id])
	mp.add_peer(peer, peer_id)
	var uid = mp.get_unique_id()
	if peer_id > uid:
		var offer_result = peer.create_offer()
		print("Create Offer Result: ", 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_name, index_name, sdp_name, peer_id)
	signaller.send_candidate(peer_id, mid_name, index_name, sdp_name)

func _offer_created(type, data, peer_id):
	print("Offer Created", type, data, peer_id)
	if not mp.has_peer(peer_id):
		return
	mp.get_peer(peer_id).connection.set_local_description(type, data)
	if type == "offer":
		# TODO: send_offer(id, data)
		pass
	else:
		# TODO: send_answer(id, data)
		pass

func _lobby_joined(id, peer_id):
	print("Connected to lobby %s as peer %d" % [id, peer_id])
	mp.initialize(peer_id, true)

func _peer_joined(peers):
	for i in range(len(peers)):
		var peer_id = peers[i]["peerId"]
		if not mp.has_peer(peer_id):
			_create_peer(peer_id)

func peer_disconnected(id):
	print("Peer disconnected ", id)
	if mp.has_peer(id):
		mp.remove_peer(id)

func offer_received(data):
	print("Got offer: ", data)
	var id = data["peerId"]
	if mp.has_peer(id):
		mp.get_peer(id).connection.set_remote_description("offer", data["offer"])

func answer_received(data):
	print("Got answer: ", data)
	var id = data["peerId"]
	if mp.has_peer(id):
		mp.get_peer(id).connection.set_remote_description("answer", data["offer"])

func candidate_received(data):
	print("Candidate Received: ", data)
	var id = data["peerId"]
	if mp.has_peer(id):
		mp.get_peer(id).connection.add_ice_candidate(data["mid"], data["index"], data["sdp"])