protocol: Add a welcoming message, removing registering as consumer requirement

Since clients know from the start their Peer ID it is fine to go ahead
and connect as a consumer without prior registration.

plugins/src/signaller/imp.rs

@@ -122,6 +122,7 @@ impl Signaller {
 
                             if let Ok(msg) = serde_json::from_str::<p::OutgoingMessage>(&msg) {
                                 match msg {
+                                    p::OutgoingMessage::Welcome { .. } => (),
                                     p::OutgoingMessage::Registered(
                                         p::RegisteredMessage::Producer { peer_id, .. },
                                     ) => {

plugins/src/webrtcsink/imp.rs

@@ -665,6 +665,7 @@ impl State {
         session: &mut Session,
         signal: bool,
     ) {
+        gst::info!(CAT, "Ending session {}", session.id);
         session.pipeline.debug_to_dot_file_with_ts(
             gst::DebugGraphDetails::all(),
             format!("removing-peer-{}-", session.peer_id,),

protocol/src/lib.rs

@@ -70,6 +70,8 @@ pub struct Peer {
 #[serde(rename_all = "camelCase")]
 /// Messages sent from the server to peers
 pub enum OutgoingMessage {
+    /// Welcoming message, sets the Peer ID linked to a new connection
+    Welcome { peer_id: String },
     /// Confirms registration
     Registered(RegisteredMessage),
     /// Confirms registration
@@ -210,6 +212,8 @@ pub struct EndSessionMessage {
 #[serde(rename_all = "camelCase")]
 /// Messages received by the server from peers
 pub enum IncomingMessage {
+    /// Register as a peer type
+    NewPeer,
     /// Register as a peer type
     Register(RegisterMessage),
     /// Unregister as a peer type

signalling/src/handlers/mod.rs

@@ -67,6 +67,16 @@ impl Handler {
         msg: p::IncomingMessage,
     ) -> Result<(), Error> {
         match msg {
+            p::IncomingMessage::NewPeer => {
+                self.items.push_back((
+                    peer_id.into(),
+                    p::OutgoingMessage::Welcome {
+                        peer_id: peer_id.to_string(),
+                    },
+                ));
+
+                Ok(())
+            }
             p::IncomingMessage::Register(message) => match message {
                 p::RegisterMessage::Producer { meta } => self.register_producer(peer_id, meta),
                 p::RegisterMessage::Consumer { meta } => self.register_consumer(peer_id, meta),
@@ -357,13 +367,6 @@ impl Handler {
     /// Start a session between two peers
     #[instrument(level = "debug", skip(self))]
     fn start_session(&mut self, producer_id: &str, consumer_id: &str) -> Result<(), Error> {
-        if !self.consumers.contains_key(consumer_id) {
-            return Err(anyhow!(
-                "Peer with id {} is not registered as a consumer",
-                consumer_id
-            ));
-        }
-
         if !self.producers.contains_key(producer_id) {
             return Err(anyhow!(
                 "Peer with id {} is not registered as a producer",

signalling/src/server/mod.rs

@@ -156,7 +156,25 @@ impl Server {
         let this_id_clone = this_id.clone();
         let state_clone = self.state.clone();
         let receive_task_handle = task::spawn(async move {
+            if let Some(tx) = tx.as_mut() {
+                if let Err(err) = tx
+                    .send((
+                        this_id_clone.clone(),
+                        Some(
+                            serde_json::json!({
+                                "type": "newPeer",
+                            })
+                            .to_string(),
+                        ),
+                    ))
+                    .await
+                {
+                    warn!(this = %this_id_clone, "Error handling message: {:?}", err);
+                }
+            }
+
             while let Some(msg) = ws_stream.next().await {
+                info!("Received message {msg:?}");
                 match msg {
                     Ok(WsMessage::Text(msg)) => {
                         if let Some(tx) = tx.as_mut() {

www/webrtc.js

@@ -220,11 +220,8 @@ function Session(our_id, peer_id, closed_callback) {
         this.ws_conn = new WebSocket(ws_url);
         /* When connected, immediately register with the server */
         this.ws_conn.addEventListener('open', (event) => {
-            this.ws_conn.send(JSON.stringify({
-                "type": "register",
-                "peerType": "consumer"
-            }));
-            this.setStatus("Registering with server");
+            this.setStatus("Connecting to the peer");
+            this.connectPeer();
         });
         this.ws_conn.addEventListener('error', this.onServerError.bind(this));
         this.ws_conn.addEventListener('message', this.onServerMessage.bind(this));
@@ -395,7 +392,9 @@ function onServerMessage(event) {
         return;
     }
 
-    if (msg.type == "registered") {
+    if (msg.type == "welcome") {
+        console.info(`Got welcomed with ID ${msg.peer_id}`);
+    } else if (msg.type == "registered") {
         ws_conn.send(JSON.stringify({
             "type": "list"
         }));