o "h]@sddlZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z m Z mZddlZddlmZz dd lmZd ZWn eyOd ZYnwdd lmZdd lmZddlmZe rjddlmZeeddZGdddZ Gdddej!j"Z#Gdddej!j$Z%dS)N) HTTPStatus)Path)socket) SSLContext) TracebackType) TYPE_CHECKINGOptionalUnion) HTTPServer)bind_unix_socketTF)Update) get_logger)ExtBot)BotUpdater) class_namec@sleZdZdZdZ ddededddeed ee ee e ff d d Z dd ee jd dfddZdddZdS) WebhookServerz6Thin wrapper around ``tornado.httpserver.HTTPServer``.) _http_server _server_lock_shutdown_lock is_runninglistenportunixNrr webhook_appWebhookAppClassssl_ctxrcCsv|rtstdt||d|_||_||_d|_d|_|r&t|t r&||_n |r/t t ||_t |_t |_dS)Nz.This OS does not support binding unix sockets.) ssl_optionsF)UNIX_AVAILABLE RuntimeErrorr rrrrr isinstancerr strasyncioLockrr)selfrrrrrr%U/var/www/html/venv/lib/python3.10/site-packages/telegram/ext/_utils/webhookhandler.py__init__As zWebhookServer.__init__readyreturnc s|j4IdH1|jr|j|jn |jj|j|jdd|_|dur)|t dWdIdHdS1IdHs?wYdS)N)addressTzWebhook Server started.) rrr add_socketrrrset_LOGGERdebug)r$r(r%r%r& serve_foreverWs .zWebhookServer.serve_foreverc s|j4IdH5|jstd WdIdHdSd|_|j|jIdHtdWdIdHdS1IdHsCwYdS)Nz.Webhook Server is already shut down. ReturningFzWebhook Server stopped)rrr-r.rstopclose_all_connectionsr$r%r%r&shutdownds   .zWebhookServer.shutdownNr)N)__name__ __module__ __qualname____doc__ __slots__r!intrrr rrr'r"Eventr/r3r%r%r%r&r4s"  rc @sJeZdZdZ ddedddejdeefdd Zd e j j d dfd d Z dS)rz!Application used in the WebserverN webhook_pathbotr update_queue secret_tokencCs6|||d|_|dt|jfg}tjj||dS)N)r>r?r@z/?)shared_objectsTelegramHandlertornadoweb Applicationr')r$r=r>r?r@handlersr%r%r&r'rs zWebhookAppClass.__init__handlerr)cCsdS)zIOverrides the default implementation since we have our own logging setup.Nr%)r$rGr%r%r& log_requestszWebhookAppClass.log_requestr4) r6r7r8r9r!r"Queuerr'rCrDRequestHandlerrHr%r%r%r&ros rc@s|eZdZdZdZdZdddejdedd fd d Z dd d Z dddZ dddZ de eede ede edd fddZd S)rBz:BaseHandler that processes incoming requests from Telegram)r>r@r?)POSTr>rr?r@r)NcCs(||_||_||_|rtddSdS)zFInitialize for each request - that's the interface provided by tornadozLThe webhook server has a secret token, expecting it in incoming requests nowN)r>r?r@r-r.)r$r>r?r@r%r%r& initializeszTelegramHandler.initializecCs|dddS)zSets default headers Content-Typez!application/json; charset="utf-8"N) set_headerr2r%r%r&set_default_headerssz#TelegramHandler.set_default_headersc std||jj}t|}|t j td|z t ||j }WntyH}ztjd||dtjjt jdd|d}~ww|ritd|jt|j tr^|j ||j|IdHdSdS) zHandle incoming POST requestzWebhook triggeredzWebhook received data: %szySomething went wrong processing the data received from Telegram. Received data was *not* processed! Received data was: %rexc_infozUpdate could not be processedreasonNz%Received Update with ID %d on Webhook)r-r._validate_postrequestbodydecodejsonloads set_statusrOKr de_jsonr> ExceptioncriticalrCrD HTTPError BAD_REQUEST update_idr rinsert_callback_datar?put)r$ json_stringdataupdateexcr%r%r&posts>        zTelegramHandler.postcCs|jjdd}|dkrtjtj|jdurC|jjd}|s/t dtjjtjdd||jkrEt d|tjjtjdddSdS) z+Only accept requests with content type JSONrMNzapplication/jsonzX-Telegram-Bot-Api-Secret-Tokenz(Request did not include the secret tokenrRz&Request had the wrong secret token: %sz"Request had the wrong secret token) rUheadersgetrCrDr_r FORBIDDENr@r-r.)r$ ct_headertokenr%r%r&rTs"    zTelegramHandler._validate_posttypvaluetbcCs0tjd|jjd|r|r|r|||fn|ddS)z@Override the default logging and instead use our custom logging.z%s - %szException in TelegramHandlerrPN)r-r.rU remote_ip)r$rnrorpr%r%r& log_exceptions  zTelegramHandler.log_exceptionr5)r6r7r8r9r:SUPPORTED_METHODSr"rIr!rLrOrhrTrtype BaseExceptionrrrr%r%r%r&rBs"  $ rB)&r"rXhttprpathlibrrsslrtypesrtypingrrr tornado.webrCtornado.httpserverr tornado.netutilr r ImportErrortelegramr telegram._utils.loggingr telegram.ext._extbotrrr6r-rrDrErrJrBr%r%r%r&s2             ;