161 lines
4.2 KiB
Cython
161 lines
4.2 KiB
Cython
import cython
|
|
|
|
from ._frame_helper.base cimport APIFrameHelper
|
|
|
|
|
|
cdef dict MESSAGE_TYPE_TO_PROTO
|
|
cdef dict PROTO_TO_MESSAGE_TYPE
|
|
|
|
cdef set OPEN_STATES
|
|
|
|
cdef float KEEP_ALIVE_TIMEOUT_RATIO
|
|
cdef object HANDSHAKE_TIMEOUT
|
|
|
|
cdef bint TYPE_CHECKING
|
|
|
|
cdef object WRITE_EXCEPTIONS
|
|
|
|
cdef object DISCONNECT_REQUEST_MESSAGE
|
|
cdef tuple DISCONNECT_RESPONSE_MESSAGES
|
|
cdef tuple PING_REQUEST_MESSAGES
|
|
cdef tuple PING_RESPONSE_MESSAGES
|
|
cdef object NO_PASSWORD_CONNECT_REQUEST
|
|
|
|
cdef object asyncio_timeout
|
|
cdef object CancelledError
|
|
cdef object asyncio_TimeoutError
|
|
|
|
cdef object ConnectRequest, ConnectResponse
|
|
cdef object DisconnectRequest
|
|
cdef object PingRequest
|
|
cdef object GetTimeRequest, GetTimeResponse
|
|
cdef object HelloRequest, HelloResponse
|
|
|
|
cdef object APIVersion
|
|
|
|
cdef object partial
|
|
|
|
cdef object hr
|
|
|
|
cdef object RESOLVE_TIMEOUT
|
|
cdef object CONNECT_AND_SETUP_TIMEOUT, CONNECT_REQUEST_TIMEOUT
|
|
|
|
cdef object APIConnectionError
|
|
cdef object BadNameAPIError
|
|
cdef object HandshakeAPIError
|
|
cdef object PingFailedAPIError
|
|
cdef object ReadFailedAPIError
|
|
cdef object TimeoutAPIError
|
|
cdef object SocketAPIError
|
|
cdef object InvalidAuthAPIError
|
|
cdef object SocketClosedAPIError
|
|
|
|
cdef object astuple
|
|
|
|
cdef object CONNECTION_STATE_INITIALIZED
|
|
cdef object CONNECTION_STATE_SOCKET_OPENED
|
|
cdef object CONNECTION_STATE_HANDSHAKE_COMPLETE
|
|
cdef object CONNECTION_STATE_CONNECTED
|
|
cdef object CONNECTION_STATE_CLOSED
|
|
|
|
cdef object make_hello_request
|
|
|
|
cpdef void handle_timeout(object fut)
|
|
cpdef void handle_complex_message(
|
|
object fut,
|
|
list responses,
|
|
object do_append,
|
|
object do_stop,
|
|
object resp,
|
|
)
|
|
|
|
cdef object _handle_timeout
|
|
cdef object _handle_complex_message
|
|
|
|
@cython.dataclasses.dataclass
|
|
cdef class ConnectionParams:
|
|
cdef public list addresses
|
|
cdef public object port
|
|
cdef public object password
|
|
cdef public object client_info
|
|
cdef public object keepalive
|
|
cdef public object zeroconf_manager
|
|
cdef public object noise_psk
|
|
cdef public object expected_name
|
|
|
|
cdef class APIConnection:
|
|
|
|
cdef ConnectionParams _params
|
|
cdef public object on_stop
|
|
cdef public object _socket
|
|
cdef public APIFrameHelper _frame_helper
|
|
cdef public object api_version
|
|
cdef public object connection_state
|
|
cdef public dict _message_handlers
|
|
cdef public str log_name
|
|
cdef set _read_exception_futures
|
|
cdef object _ping_timer
|
|
cdef object _pong_timer
|
|
cdef float _keep_alive_interval
|
|
cdef float _keep_alive_timeout
|
|
cdef object _start_connect_future
|
|
cdef object _finish_connect_future
|
|
cdef public Exception _fatal_exception
|
|
cdef bint _expected_disconnect
|
|
cdef object _loop
|
|
cdef bint _send_pending_ping
|
|
cdef public bint is_connected
|
|
cdef bint _handshake_complete
|
|
cdef bint _debug_enabled
|
|
cdef public str received_name
|
|
cdef public str connected_address
|
|
|
|
cpdef void send_message(self, object msg)
|
|
|
|
cdef void send_messages(self, tuple messages)
|
|
|
|
@cython.locals(handlers=set, handlers_copy=set)
|
|
cpdef void process_packet(self, object msg_type_proto, object data)
|
|
|
|
cdef void _async_cancel_pong_timer(self)
|
|
|
|
cdef void _async_schedule_keep_alive(self, object now)
|
|
|
|
cdef void _cleanup(self)
|
|
|
|
cpdef set_log_name(self, str name)
|
|
|
|
cdef _make_connect_request(self)
|
|
|
|
cdef void _process_hello_resp(self, object resp)
|
|
|
|
cdef void _process_login_response(self, object hello_response)
|
|
|
|
cdef void _set_connection_state(self, object state)
|
|
|
|
cpdef void report_fatal_error(self, Exception err)
|
|
|
|
@cython.locals(handlers=set)
|
|
cdef void _add_message_callback_without_remove(self, object on_message, tuple msg_types)
|
|
|
|
cpdef add_message_callback(self, object on_message, tuple msg_types)
|
|
|
|
@cython.locals(handlers=set)
|
|
cpdef void _remove_message_callback(self, object on_message, tuple msg_types)
|
|
|
|
cpdef void _handle_disconnect_request_internal(self, object msg)
|
|
|
|
cpdef void _handle_ping_request_internal(self, object msg)
|
|
|
|
cpdef void _handle_get_time_request_internal(self, object msg)
|
|
|
|
cdef void _set_fatal_exception_if_unset(self, Exception err)
|
|
|
|
cdef void _register_internal_message_handlers(self)
|
|
|
|
cdef void _increase_recv_buffer_size(self)
|
|
|
|
cdef void _set_start_connect_future(self)
|
|
|
|
cdef void _set_finish_connect_future(self)
|