o ?i@s ddlZddlmZddlmZmZddlmZddlmZddl m Z ddl m Z ddl mZeZeje d gd d gd gd ejd ed ddZedddZedddZeddedefddZedkrddlZddlZddl mZeejejede jddSdS)N)datetime)FastAPI WebSocket) JSONResponse)WebSocketDisconnect)CORSMiddleware)Config)Log*T)Z allow_originsZallow_credentialsZ allow_methodsZ allow_headers/)Zresponse_classcs ddiS)Nmessagez%Mcube Call Service Server is running!r r r homebook/main.py index_pagesrz/healthcsdtdddS)zBasic health check endpoint.healthyMcube Call Servicez1.0.0)status timestampserviceversion)rnow isoformatr r r r health_checks  rz/health/performancec sddl}z:|jdd}|}|d}|dkr |jdkr dndt||jt|j d d |jt|j d d d d d WSt y]}zdtt |dWYd}~Sd}~ww)z"Performance health check endpoint.rN)intervalr PrZwarningi@) cpu_percentZmemory_percentZmemory_available_gbZ disk_percentZ disk_free_gbr)rrsystemrerror)rrr) psutilrZvirtual_memory disk_usagepercentrrrroundZ availablefree Exceptionstr)r rZmemoryZdisker r rperformance_health_check"s.     r(z/ws/{session_id} websocket session_idc std||IdHddlm}|||IdHs5td||jdddIdHdSz:j}j d t d dffd d }d t d dffd d }dt d dffdd }t ||||}zz|g}t j|t jdIdH\} } | D]} | z| IdHWq}t jyYq}w| D]^} | r| } t| trtd|||IdHWW||jvr||IdHWdSWdStd|d| ||IdHWW||jvr||IdHWdSWdSqtd|||IdHWnDty!td|||IdHYn+tyK} ztd| ddl}|||IdHWYd} ~ nd} ~ wwW||jvr^||IdHWdSWdS||jvrp||IdHwwty} zGtd|d| ddl}|z ||IdHWnty}ztd|WYd}~n d}~wwWYd} ~ dSWYd} ~ dSd} ~ ww)z=Handle Mcube WebSocket call connection with proper isolation.z#Mcube call connected - Session ID: Nr) call_manageru&❌ Failed to create call session for izFailed to create call session)codereasondatareturnc|IdHdS)z&Handle incoming media data from Mcube.N)handle_media_eventr.sessionr rr1Oz-handle_mcube_call..handle_media_eventcr0)z1Handle call start event with full WebSocket data.N)handle_call_startr2r3r rr6Sr5z,handle_mcube_call..handle_call_startnamecs ||IdHdS)z%Handle playedStream event from Mcube.N)Zhandle_played_stream_event)r7 audio_servicer4r rhandle_played_streamWs z/handle_mcube_call..handle_played_stream)Z return_whenu9🔌 WebSocket disconnected detected in task for session zTask exception for session z: z)All tasks completed normally for session z#WebSocket disconnected for session zError in media stream handler: u6❌ Unexpected error in WebSocket handler for session u❌ Error during cleanup: )r headerZacceptZservices.call_managerr+Zcreate_call_sessionrcloseconnection_managerr9dictr&asyncioZ create_taskZreceive_from_mcubewaitZFIRST_COMPLETEDZcancelZCancelledError exception isinstancerinfoZremove_call_sessionZactive_sessionsr% traceback print_exc)r)r*r+r=r1r6r:Z mcube_taskZ tasks_to_waitZdoneZpendingZtaskrAr'rDZ cleanup_errorr r8rhandle_mcube_call?s       rF__main__z0.0.0.0)ZhostZport)r?rZfastapirrZfastapi.responsesrZfastapi.websocketsrZfastapi.middleware.corsrZconfigrZservices.log_utilsr ZappZadd_middlewaregetrrr(r)r&rF__name__ZuvicornatexitregisterZclose_log_filerunZPORTr r r rs>          Q