add disconnect reason to Room::close

Author: theomonnom

livekit-ffi/protocol/participant.proto

@@ -85,6 +85,7 @@ enum DisconnectReason {
   SIP_TRUNK_FAILURE = 13;
   CONNECTION_TIMEOUT = 14;
   MEDIA_FAILURE = 15;
+  AGENT_ERROR = 16;
 }
 
 // copied from livekit-protocol/protocol/protobufs/livekit_models.proto and removed deprecated fields

livekit-ffi/protocol/room.proto

@@ -62,6 +62,7 @@ message ConnectCallback {
 message DisconnectRequest {
   required uint64 room_handle = 1;
   optional uint64 request_async_id = 2;
+  optional DisconnectReason reason = 3;
 }
 message DisconnectResponse { required uint64 async_id = 1; }
 message DisconnectCallback { required uint64 async_id = 1; }

livekit-ffi/src/conversion/participant.rs

@@ -96,6 +96,7 @@ impl From<DisconnectReason> for proto::DisconnectReason {
             DisconnectReason::SipTrunkFailure => proto::DisconnectReason::SipTrunkFailure,
             DisconnectReason::ConnectionTimeout => proto::DisconnectReason::ConnectionTimeout,
             DisconnectReason::MediaFailure => proto::DisconnectReason::MediaFailure,
+            DisconnectReason::AgentError => proto::DisconnectReason::AgentError,
         }
     }
 }

livekit-ffi/src/conversion/room.rs

@@ -102,6 +102,31 @@ impl From<DisconnectReason> for proto::DisconnectReason {
             DisconnectReason::SipTrunkFailure => Self::SipTrunkFailure,
             DisconnectReason::ConnectionTimeout => Self::ConnectionTimeout,
             DisconnectReason::MediaFailure => Self::MediaFailure,
+            DisconnectReason::AgentError => Self::AgentError,
+        }
+    }
+}
+
+impl From<proto::DisconnectReason> for DisconnectReason {
+    fn from(value: proto::DisconnectReason) -> Self {
+        match value {
+            proto::DisconnectReason::UnknownReason => Self::UnknownReason,
+            proto::DisconnectReason::ClientInitiated => Self::ClientInitiated,
+            proto::DisconnectReason::DuplicateIdentity => Self::DuplicateIdentity,
+            proto::DisconnectReason::ServerShutdown => Self::ServerShutdown,
+            proto::DisconnectReason::ParticipantRemoved => Self::ParticipantRemoved,
+            proto::DisconnectReason::RoomDeleted => Self::RoomDeleted,
+            proto::DisconnectReason::StateMismatch => Self::StateMismatch,
+            proto::DisconnectReason::JoinFailure => Self::JoinFailure,
+            proto::DisconnectReason::Migration => Self::Migration,
+            proto::DisconnectReason::SignalClose => Self::SignalClose,
+            proto::DisconnectReason::RoomClosed => Self::RoomClosed,
+            proto::DisconnectReason::UserUnavailable => Self::UserUnavailable,
+            proto::DisconnectReason::UserRejected => Self::UserRejected,
+            proto::DisconnectReason::SipTrunkFailure => Self::SipTrunkFailure,
+            proto::DisconnectReason::ConnectionTimeout => Self::ConnectionTimeout,
+            proto::DisconnectReason::MediaFailure => Self::MediaFailure,
+            proto::DisconnectReason::AgentError => Self::AgentError,
         }
     }
 }

livekit-ffi/src/server/mod.rs

@@ -24,6 +24,7 @@ use std::{
 
 use dashmap::{mapref::one::MappedRef, DashMap};
 use downcast_rs::{impl_downcast, Downcast};
+use livekit::prelude::DisconnectReason;
 use livekit::webrtc::{
     native::apm::AudioProcessingModule, native::audio_resampler::AudioResampler, prelude::*,
 };
@@ -169,7 +170,7 @@ impl FfiServer {
         }
 
         for room in rooms {
-            room.close(self).await;
+            room.close(self, DisconnectReason::ClientInitiated).await;
         }
 
         // Drop all handles

livekit-ffi/src/server/requests.rs

@@ -65,11 +65,16 @@ fn on_disconnect(
     disconnect: proto::DisconnectRequest,
 ) -> FfiResult<proto::DisconnectResponse> {
     let async_id = server.resolve_async_id(disconnect.request_async_id);
+    let reason = disconnect
+        .reason
+        .and_then(|r| proto::DisconnectReason::try_from(r).ok())
+        .map(DisconnectReason::from)
+        .unwrap_or(DisconnectReason::ClientInitiated);
     let handle = server.async_runtime.spawn(async move {
         let ffi_room =
             server.retrieve_handle::<room::FfiRoom>(disconnect.room_handle).unwrap().clone();
 
-        ffi_room.close(server).await;
+        ffi_room.close(server, reason).await;
 
         let _ = server.send_event(proto::DisconnectCallback { async_id }.into());
     });

livekit-ffi/src/server/room.rs

@@ -301,7 +301,7 @@ impl FfiRoom {
     }
 
     /// Close the room and stop the tasks
-    pub async fn close(&self, server: &'static FfiServer) {
+    pub async fn close(&self, server: &'static FfiServer, reason: DisconnectReason) {
         // drop associated track handles
         for (_, &handle) in self.inner.track_handle_lookup.lock().iter() {
             if server.drop_handle(handle) {
@@ -310,7 +310,7 @@ impl FfiRoom {
             }
         }
 
-        let _ = self.inner.room.close().await;
+        let _ = self.inner.room.close_with_reason(reason.into()).await;
 
         let handle = self.handle.lock().await.take();
         if let Some(handle) = handle {

livekit-protocol/protocol

@@ -1887,6 +1887,8 @@ pub enum DisconnectReason {
     ConnectionTimeout = 14,
     /// media stream failure or media timeout
     MediaFailure = 15,
+    /// agent encountered an error
+    AgentError = 16,
 }
 impl DisconnectReason {
     /// String value of the enum field names used in the ProtoBuf definition.
@@ -1911,6 +1913,7 @@ impl DisconnectReason {
             DisconnectReason::SipTrunkFailure => "SIP_TRUNK_FAILURE",
             DisconnectReason::ConnectionTimeout => "CONNECTION_TIMEOUT",
             DisconnectReason::MediaFailure => "MEDIA_FAILURE",
+            DisconnectReason::AgentError => "AGENT_ERROR",
         }
     }
     /// Creates an enum from field names used in the ProtoBuf definition.
@@ -1932,6 +1935,7 @@ impl DisconnectReason {
             "SIP_TRUNK_FAILURE" => Some(Self::SipTrunkFailure),
             "CONNECTION_TIMEOUT" => Some(Self::ConnectionTimeout),
             "MEDIA_FAILURE" => Some(Self::MediaFailure),
+            "AGENT_ERROR" => Some(Self::AgentError),
             _ => None,
         }
     }

livekit-protocol/src/livekit.rs

@@ -10412,6 +10412,7 @@ impl serde::Serialize for DisconnectReason {
             Self::SipTrunkFailure => "SIP_TRUNK_FAILURE",
             Self::ConnectionTimeout => "CONNECTION_TIMEOUT",
             Self::MediaFailure => "MEDIA_FAILURE",
+            Self::AgentError => "AGENT_ERROR",
         };
         serializer.serialize_str(variant)
     }
@@ -10439,6 +10440,7 @@ impl<'de> serde::Deserialize<'de> for DisconnectReason {
             "SIP_TRUNK_FAILURE",
             "CONNECTION_TIMEOUT",
             "MEDIA_FAILURE",
+            "AGENT_ERROR",
         ];
 
         struct GeneratedVisitor;
@@ -10495,6 +10497,7 @@ impl<'de> serde::Deserialize<'de> for DisconnectReason {
                     "SIP_TRUNK_FAILURE" => Ok(DisconnectReason::SipTrunkFailure),
                     "CONNECTION_TIMEOUT" => Ok(DisconnectReason::ConnectionTimeout),
                     "MEDIA_FAILURE" => Ok(DisconnectReason::MediaFailure),
+                    "AGENT_ERROR" => Ok(DisconnectReason::AgentError),
                     _ => Err(serde::de::Error::unknown_variant(value, FIELDS)),
                 }
             }