o i0@sddlmZddlZddlZddlZddlZddlmZmZddl m Z m Z m Z m Z mZmZddlZddlmZddlmZmZddlmZdd lmZdd lmZdd lmZdd lm Z dd l!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*m+Z+ddl,m-Z-m.Z.m/Z/ddl0m1Z1m2Z2ddl3m4Z4m5Z5ddl6m7Z7ddl8m9Z9m:Z:m;Z;meGdddZ?eGdddZ@eGdddZAeGd d!d!ZBGd"d#d#eCZDGd$d%d%ee=ZEdS)&) annotationsN) dataclassfield)CallableDictLiteralOptionalcastMapping) EventEmitter) FfiClient FfiHandle)ffi_pb2)participant_pb2)room_pb2) stats_pb2)DisconnectReason)ConnectionState) TrackKind)RpcMethodInvocationEvent)BroadcastQueue) E2EEManager E2EEOptions)LocalParticipant ParticipantRemoteParticipant)RemoteAudioTrackRemoteVideoTrack)RemoteTrackPublicationTrackPublication)TranscriptionSegment)TextStreamReaderByteStreamReaderTextStreamHandlerByteStreamHandler) participant_connectedparticipant_disconnectedlocal_track_publishedlocal_track_unpublishedlocal_track_subscribedtrack_publishedtrack_unpublishedtrack_subscribedtrack_unsubscribedtrack_subscription_failed track_muted track_unmutedactive_speakers_changedroom_metadata_changedparticipant_metadata_changedparticipant_name_changedparticipant_attributes_changedconnection_quality_changed%participant_encryption_status_changedparticipant_permissions_changed data_receivedsip_dtmf_receivedtranscription_receivede2ee_state_changedconnection_state_changed connected disconnected reconnecting reconnected room_updatedmovedtoken_refreshedc@sDeZdZUejjZded< ejj Z ded< e e dZ ded<dS) RtcConfigurationz%proto_room.IceTransportType.ValueTypeice_transport_typez-proto_room.ContinualGatheringPolicy.ValueTypecontinual_gathering_policy)default_factoryzlist[proto_room.IceServer] ice_serversN)__name__ __module__ __qualname__ proto_roomIceTransportType TRANSPORT_ALLrG__annotations__ContinualGatheringPolicyGATHER_CONTINUALLYrHrlistrJrUrUR/var/www/html/livekit_bhavya/venv/lib/python3.10/site-packages/livekit/rtc/room.pyrFTs   rFc@sleZdZUdZded< dZded<dZded< dZded < dZd ed < dZ d ed < dZ ded<dS) RoomOptionsTboolauto_subscribeFdynacastNzE2EEOptions | Nonee2ee encryptionzRtcConfiguration | None rtc_configz float | Noneconnect_timeoutz bool | Nonesingle_peer_connection) rKrLrMrYrQrZr[r\r]r^r_rUrUrUrVrWcs       rWc@s8eZdZUded< ded< ded< dZded <dS) DataPacketbytesdataz#proto_room.DataPacketKind.ValueTypekindRemoteParticipant | None participantNz str | Nonetopic)rKrLrMrQrfrUrUrUrVr`ts  r`c@s.eZdZUded< ded< dZded<dS)SipDTMFintcodestrdigitNrdre)rKrLrMrQrerUrUrUrVrgs  rgc@seZdZUded<ded<dS)RtcStatszlist[proto_stats.RtcStats]publisher_statssubscriber_statsN)rKrLrMrQrUrUrUrVrls  rlc@seZdZdddZdS) ConnectErrormessagerjcCs ||_dSN)rp)selfrprUrUrV__init__s zConnectError.__init__N)rprj)rKrLrMrsrUrUrUrVrosrocseZdZ dudvfdd Zdwdd Zedxd d ZedyddZedzddZed{ddZ edxddZ edxddZ ed|ddZ ed}ddZ ed}d d!Zed~d#d$Zedd&d'Zedd)d*Zedd+d,Zdd-d.Zdudfd4d5 Zefdd:d;Zdd=d>ZddBdCZddDdEZddGdHZddIdJZdwdKdLZdwdMdNZddQdRZddTdUZddYdZZ dd]d^Z!ddadbZ"dwdcddZ#dwdedfZ$ddidjZ%ddldmZ&ddqdrZ'dxdsdtZ(Z)S)RoomNloop#Optional[asyncio.AbstractEventLoop]returnNonecstd|_|p t|_ttj|_ t |_ t |_t |_i|_tj|_tjt|_d|_i|_i|_i|_i|_d|_d|_dS)zInitializes a new Room instance. Parameters: loop (Optional[asyncio.AbstractEventLoop]): The event loop to use. If not provided, the default event loop is used. N)superrs _ffi_handleasyncioget_event_loop_loopr proto_ffiFfiEvent _room_queuerNRoomInfo_infoset_rpc_invocation_tasks_data_stream_tasks_remote_participantsrCONN_DISCONNECTED_connection_stateFuturerj_first_sid_future_local_participant_text_stream_readers_byte_stream_readers_text_stream_handlers_byte_stream_handlers_token _server_url)rrru __class__rUrVrss"   z Room.__init__cCs"|jdurtjj|jdSdSrq)rzr instancequeue unsubscribe _ffi_queuerrrUrUrV__del__s z Room.__del__rjcs|jjr |jjS|jIdHS)zAsynchronously retrieves the session ID (SID) of the room. Returns: str: The session ID of the room. N)rsidrrrUrUrVrs zRoom.sidrcCs|jdur td|jS)zGets the local participant in the room. Returns: LocalParticipant: The local participant in the room. Nz1cannot access local participant before connecting)r ExceptionrrUrUrVlocal_participants zRoom.local_participantConnectionState.ValueTypecC|jS)zGets the connection state of the room. Returns: ConnectionState: The connection state of the room. )rrrUrUrVconnection_statezRoom.connection_stateMapping[str, RemoteParticipant]cCr)zGets the remote participants in the room. Returns: dict[str, RemoteParticipant]: A dictionary of remote participants indexed by their identity. )rrrUrUrVremote_participantsszRoom.remote_participantscC|jjS)z\Gets the name of the room. Returns: str: The name of the room. )rnamerrUrUrVrz Room.namecCr)zqGets the metadata associated with the room. Returns: str: The metadata of the room. )rmetadatarrUrUrVrrz Room.metadatarcCr)zGets the end-to-end encryption (E2EE) manager for the room. Returns: E2EEManager: The E2EE manager instance. ) _e2ee_managerrrUrUrV e2ee_managerrzRoom.e2ee_managerrhcCr)zGets the number of participants in the room. This value is updated periodically, and is eventually consistent. Returns: int: The number of participants in the room. )rnum_participantsrrUrUrVrzRoom.num_participantscCr)zGets the number of publishers in the room. This value is updated periodically, and is eventually consistent. Returns: int: The number of publishers in the room. )rnum_publishersrrUrUrVr rzRoom.num_publishersdatetime.datetimecCstj|jjdtjjS)zxTime when the room was created. Returns: datetime.datetime: The creation time of the room. )datetime fromtimestampr creation_timetimezoneutcrrUrUrVrszRoom.creation_timerXcCr)zWhether the room is actively recording. Returns: bool: True if actively recording, False otherwise. )ractive_recordingrrUrUrV is_recordingrzRoom.is_recordingfloatcC t|jjS)zAmount of time to hold the room open after the last standard participant leaves. Returns: float: The departure timeout of the room. )rrdeparture_timeoutrrUrUrVr' zRoom.departure_timeoutcCr)zAmount of time to keep the room open if no participants join. Returns: float: The empty timeout of the room. )rr empty_timeoutrrUrUrVr0rzRoom.empty_timeoutcCs|jduo |jtjkS)z{Checks if the room is currently connected. Returns: bool: True if connected, False otherwise. N)rzrrrrrUrUrV isconnected9s  zRoom.isconnectedevent EventTypescallbackOptional[Callable]rcst||S)aRegisters an event handler for a specific event type. Parameters: event (EventTypes): The name of the event to listen for. callback (Callable): The function to call when the event occurs. Returns: Callable: The registered callback function. Available events: - **"participant_connected"**: Called when a new participant joins the room. - Arguments: `participant` (RemoteParticipant) - **"participant_disconnected"**: Called when a participant leaves the room. - Arguments: `participant` (RemoteParticipant) - **"local_track_published"**: Called when a local track is published. - Arguments: `publication` (LocalTrackPublication), `track` (Track) - **"local_track_unpublished"**: Called when a local track is unpublished. - Arguments: `publication` (LocalTrackPublication) - **"local_track_subscribed"**: Called when a local track is subscribed. - Arguments: `track` (Track) - **"track_published"**: Called when a remote participant publishes a track. - Arguments: `publication` (RemoteTrackPublication), `participant` (RemoteParticipant) - **"track_unpublished"**: Called when a remote participant unpublishes a track. - Arguments: `publication` (RemoteTrackPublication), `participant` (RemoteParticipant) - **"track_subscribed"**: Called when a track is subscribed. - Arguments: `track` (Track), `publication` (RemoteTrackPublication), `participant` (RemoteParticipant) - **"track_unsubscribed"**: Called when a track is unsubscribed. - Arguments: `track` (Track), `publication` (RemoteTrackPublication), `participant` (RemoteParticipant) - **"track_subscription_failed"**: Called when a track subscription fails. - Arguments: `participant` (RemoteParticipant), `track_sid` (str), `error` (str) - **"track_muted"**: Called when a track is muted. - Arguments: `participant` (Participant), `publication` (TrackPublication) - **"track_unmuted"**: Called when a track is unmuted. - Arguments: `participant` (Participant), `publication` (TrackPublication) - **"active_speakers_changed"**: Called when the list of active speakers changes. - Arguments: `speakers` (list[Participant]) - **"room_metadata_changed"**: Called when the room's metadata is updated. - Arguments: `old_metadata` (str), `new_metadata` (str) - **"participant_metadata_changed"**: Called when a participant's metadata is updated. - Arguments: `participant` (Participant), `old_metadata` (str), `new_metadata` (str) - **"participant_name_changed"**: Called when a participant's name is changed. - Arguments: `participant` (Participant), `old_name` (str), `new_name` (str) - **"participant_attributes_changed"**: Called when a participant's attributes change. - Arguments: `changed_attributes` (dict), `participant` (Participant) - **"participant_encryption_status_changed"**: Called when a participant's encryption status changes. - Arguments `is_encrypted` (bool), `participant` (Participant) - **"connection_quality_changed"**: Called when a participant's connection quality changes. - Arguments: `participant` (Participant), `quality` (ConnectionQuality) - **"transcription_received"**: Called when a transcription is received. - Arguments: `segments` (list[TranscriptionSegment]), `participant` (Participant), `publication` (TrackPublication) - **"data_received"**: Called when data is received. - Arguments: `data_packet` (DataPacket) - **"sip_dtmf_received"**: Called when a SIP DTMF signal is received. - Arguments: `sip_dtmf` (SipDTMF) - **"e2ee_state_changed"**: Called when a participant's E2EE state changes. - Arguments: `participant` (Participant), `state` (EncryptionState) - **"connection_state_changed"**: Called when the room's connection state changes. - Arguments: `connection_state` (ConnectionState) - **"connected"**: Called when the room is successfully connected. - Arguments: None - **"disconnected"**: Called when the room is disconnected. - Arguments: `reason` (DisconnectReason) - **"reconnecting"**: Called when the room is attempting to reconnect. - Arguments: None - **"reconnected"**: Called when the room has successfully reconnected. - Arguments: None - **"room_updated"**: Called when any information about the room is updated. - Arguments: None - **"moved"**: Called when the participant has been moved to another room. - Arguments: None Example: ```python def on_participant_connected(participant): print(f"Participant connected: {participant.identity}") room.on("participant_connected", on_participant_connected) ``` )ryon)rrrrrrUrVrDsPzRoom.onurltokenoptionsrWc s||_||_t}||j_||j_|j|jj_|j |jj_ |j dur.t |j d|jj_ |j dur9|j |jj_ |jrttjdtdd|jj|jjj_|jjj|jjjj_|jjj|jjjj_|jjj|jjjj_|jjj|jjjj_|jr|jj|jjj_|jjj|jjjj_|jjj|jjjj_|jjj|jjjj_|jjj|jjjj_|jr|jj|jjj_|jj|jjj_|jjjj|jjtjj !|j"|_#tjj !}ztj$||%fddIdH}Wtjj &|ntjj &|w|jj'r tjj &|j#t(|jj't)|jj*j+j,j-|_.t/|j.j,|jp|j|_0|jj*j+j1|_2t3j4|_5t6|j7|jj*j8|_9|jj*j:D]}|;|j<}|j=D] } t>| } | |j?| j@<qGq<|j"A|B|_CdS)aConnects to a LiveKit room using the specified URL and token. Parameters: url (str): The WebSocket URL of the LiveKit server to connect to. token (str): The access token for authentication and authorization. options (RoomOptions, optional): Additional options for the room connection. Raises: ConnectError: If the connection fails. Example: ```python room = Room() # Listen for events before connecting to the room @room.on("participant_connected") def on_participant_connected(participant): print(f"Participant connected: {participant.identity}") await room.connect("ws://localhost:7880", "your_token") ``` Nrz:options.e2ee is deprecated, use options.encryption instead) stacklevelc|jjjjkSrq)connectasync_ideresprUrVzRoom.connect..)Drrr~ FfiRequestrrrrYrrZr^rhconnect_timeout_msr_r[warningswarnDeprecationWarningencryption_typekey_provider_options shared_key ratchet_saltfailure_toleranceratchet_window_sizer\r]rGrHrJextendr rr subscriber}rrequestwait_forrerrorrorresultroomhandleidrzrrinforrCONN_CONNECTEDrrrrr participants_create_remote_participantre publicationsr_track_publicationsr create_task _listen_task_task) rrrrrreqrcbptrpowned_publication_info publicationrUrrVrs                        z Room.connectrlc s|s tdt}|jj|j_tj j }ztj || fddIdH}Wtj j |ntj j |w|jjrGt|jjt|jjj}t|jjj}t||dS)Nzthe room isn't connectedcrrq)get_session_statsrrrrUrVrrz$Room.get_rtc_stats..)rmrn)r RuntimeErrorr~rrzrr room_handler rrrrrrrrTrrmrnrl)rrrrrrmrnrUrrV get_rtc_statss"       zRoom.get_rtc_statsrfhandlerr%cC.|j|}|dur||j|<dStd|)Nz.byte stream handler for topic '%s' already set)rget ValueErrorrrrfrexisting_handlerrUrUrVregister_byte_stream_handler*  z!Room.register_byte_stream_handlercC |j|r|j|dSdSrq)rrpoprrrfrUrUrVunregister_byte_stream_handler1 z#Room.unregister_byte_stream_handlerr$cCr)Nz.text stream handler for topic '%s' already set)rrrrrUrUrVregister_text_stream_handler5rz!Room.register_text_stream_handlercCrrq)rrrrrUrUrVunregister_text_stream_handler<rz#Room.unregister_text_stream_handlerc s|sdS|IdH|IdHt}|jj|j_t j j }zt j ||fddIdHWt j j |nt j j |w|jIdHt j j |j|jtjkrwtj|jj_tj|_|d|j|dtjdSdS)zDisconnects from the room.Ncrrq) disconnectrrrrUrVrMrz!Room.disconnect..r>r@)r_drain_rpc_invocation_tasks_drain_data_stream_tasksr~rrzrrrr rrrrrrrrrrrrCLIENT_INITIATEDrrdisconnect_reasonemit)rrrrrUrrVr@s(       zRoom.disconnectc s |jIdH}|ddkr||jn-|jj|jjkrE|j dr'n-z| |jWnt yDt d|jd|jYnw|j||jIdHq|IdH|IdHdS)NTrprpc_method_invocationeosz&error running user callback for %s: %s)rr WhichOneof_on_rpc_method_invocationr  room_eventrrzrHasField_on_room_eventrlogging exceptionr put_nowaitjoinrr)rrrrUrUrVr\s,    zRoom._listen_taskrpc_invocationrc Csn|jdurdS|j|jjjkr5|j|j|j|j|j |j |j |j d}|j |||j jdSdS)Ng@@)rlocal_participant_handlerzrr}r_handle_rpc_method_invocation invocation_idmethod request_idcaller_identitypayloadresponse_timeout_msraddadd_done_callbackdiscard)rrrtaskrUrUrVr xs  zRoom._on_rpc_method_invocationproto_room.RoomEventc$ Cs |d}|dkr||jj}|d|dS|dkr4|jj}|j|}|jj |j _ |d|dS|dkrN|j j }|j j|}|j}|d||dS|dkrd|jj}|j j|}|d|dS|dkr|jj }|j j|}|jd|d|jdS|dkr|j|jj}t|jj}||j|j<|d||dS|dkr|j|jj}|j|jj}|d||dS|d kr |jj} | j} |j|jj}|j| j}d |_| jtjkrt | } | |_!|d | ||dS| jtj"krt#| } | |_!|d | ||dSdS|d kr2|j$j}|j|}|j|j$j }|j} d|_!d |_|d | ||dS|d krP|j%j}|j|}|j%j&}|d ||j%j |dS|dkr|j'j}|(|}t)|t*sfJ|j|j'j }d |j _+|jrzd |jj _+|d||dS|dkr|j,j}|(|}t)|t*sJ|j|j,j }d |j _+|jrd |jj _+|d||dS|dkrg}|j-j.D]}|(|}t)|t*sJ|/|q|d|dS|dkr|j0}|j1j0|j _0|d||j0dS|dkr|j js |j2|j3j|j3j|j _dS|dkr;|j4j}|(|}t)|t*s'J|j0}|j4j0|j _0|d|||j0dS|dkre|j5j}|(|}t)|t*sQJ|j6}|j5j6|j _6|d|||j6dS|dkr|j7j}|j7j8}t9dd|j7j:D}|(|}t)|t*sJ|j j8;|j j8dS|dkr|j?j}|(|}t)|t*sJ|j j@A|j?j@|d||jBdS|dkr|jCj}|(|}|d||jCjDdS|dkr,|jE}dd|jFD}|(|j}d}|r"|jG|j }|d|||dS|dkr|jH}|d }|d!krz|jIjJ}|jJ}tKL|jMtKNtKjO|jPjQ}tR|}tS|jTjUVtLtW|X|j}|d"tY||j||jIjZd#dS|d$krtLtW|X|j}|d%t[|j\j]|j\j^|d&dSdS|d'kr|j_j}|j_j`} |d'|(|| dS|d(kr|jaj`}!|!|_b|d(|!dS|d)kr|d)|jcjddS|d*kr|d*dS|d+kr|d+dS|d,kr|e|jfjg|jfjdS|d-krthi|j|jkjl}"|jmn|"|"o|jmjpdS|d.kr s  z&Room._on_room_event..csr$rqr%r(rUrUrVr+sr8r9r7r<c Ss,g|]}t|j|j|j|j|j|jdqS))rtextfinal start_timeend_timelanguage)r!rr,r-r.r/r0)r)srUrUrV 6s z'Room._on_room_event..data_packet_receivedr'userr:)rbrcrerfsip_dtmfr;)rirkrer=r>r@rArBstream_header_receivedstream_chunk_receivedstream_trailer_receivedrCrDparticipants_updatedrE)|r rr&rr r'participant_identityrrrrr( track_sidrtrack_publicationstrackr)publication_sidr*_first_subscription set_resultr+rrrrr,r- _subscribedrcr KIND_VIDEOr_track KIND_AUDIOrr.r/rr0_retrieve_participant isinstancermutedr1r2participant_identitiesappendrr3rr#r4r5rr6 attributesdictchanged_attributesclearupdater8 is_encryptedparticipant_permission_changed permissionCopyFrom permissionsr7qualityr<segmentsrr3r4rbctypesr data_ptrPOINTERc_bytedata_lencontentsrarrrdisposer_retrieve_remote_participantr`rfrgr5rirkr=stater>rr@reason_handle_stream_headerr6headerr{r_handle_stream_chunkr7chunkrrrr _handle_stream_trailerr8trailerrCrDr9ridentityrErr)$rrrwhich rparticipantrfr lpublicationltrack rpublicationowned_track_info track_inforemote_video_trackremote_audio_trackrtrackrrerspeakers old_metadataold_namerJrL transcriptionrUpartpubpacket which_valowned_buffer_info buffer_info native_datarb e2ee_staterr!rrUrUrVrs                                                                       zRoom._on_room_eventraproto_room.DataStream.Headerr:cCs|d}|dkr.|j|j}|durtd|jdSt|}||j|j<|||dS|dkrW|j |j}|durFtd|jdSt |}||j |j<|||dSt d|dS)Ncontent_header text_headerz:ignoring text stream with topic '%s', no callback attached byte_headerz:ignoring byte stream with topic '%s', no callback attachedz received unknown header type, %s) r rrrfrrr"r stream_idrr#rwarning)rrrar: stream_typetext_stream_handler text_readerbyte_stream_handler byte_readerrUrUrVr`s6      zRoom._handle_stream_headerrcproto_room.DataStream.ChunkcsR|j|j}|j|j}|r||IdHdS|r'||IdHdSdSrq)rrrr_on_chunk_update)rrrcr file_readerrUrUrVrbszRoom._handle_stream_chunkreproto_room.DataStream.Trailercsn|j|j}|j|j}|r"||IdH|j|jdS|r5||IdH|j|jdSdSrq)rrrr_on_stream_closer)rrrerrrUrUrVrdszRoom._handle_stream_trailerc<|jr|jD]}|qtj|jddiIdHdSdSNreturn_exceptionsT)rcancelr{gatherrrr!rUrUrVr   z Room._drain_rpc_invocation_taskscrr)rrr{rrrUrUrVrrzRoom._drain_data_stream_tasksrfOptional[RemoteParticipant]cCs|j|dS)z)Retrieve a remote participant by identityN)rrrrrfrUrUrVr]sz!Room._retrieve_remote_participantOptional[Participant]cCs |r ||jjkr |jS||S)z2Retrieve a local or remote participant by identity)rrfr]rrUrUrVrEs zRoom._retrieve_participant owned_info"proto_participant.OwnedParticipantrcCs.|jj|jvr tdt|}||j|j<|S)Nzparticipant already exists)rrfrrr)rrrrerUrUrVrs  zRoom._create_remote_participantc CsBd}|jr |j}d|d|jd|jdt|jd S)Nunknownz rtc.Room(sid=z, name=z , metadata=z, connection_state=))rdonerrrrNamer)rrrrUrUrV__repr__s  *z Room.__repr__rq)rurvrwrx)rwrx)rwrj)rwr)rwr)rwr)rwr)rwrh)rwr)rwrX)rwr)rrrrrwr)rrjrrjrrWrwrx)rwrl)rfrjrr%)rfrj)rfrjrr$)rr)rr")rar}r:rj)rcr)rer)rfrjrwr)rfrjrwr)rrrwr)*rKrLrMrsrpropertyrrrrrrrrrrrrrrrrWrrrrrrrrr rr`rbrdrrr]rErr __classcell__rUrUrrVrtsj                R |              rt)F __future__rrr{rVr dataclassesrrtypingrrrrr r r event_emitterr _ffi_clientr r_protorr~rproto_participantrrNr proto_stats_proto.participant_pb2r_proto.room_pb2r_proto.track_pb2r_proto.rpc_pb2r_utilsrr[rrrerrrr=rrtrack_publicationrr rtr! data_streamr"r#r$r%rrFrWr`rgrlrrortrUrUrUrVsP             $