mirror of
https://github.com/Mooling0602/MatrixPyBridge-MCDR.git
synced 2026-03-01 05:29:44 +08:00
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
from typing import Callable, Union, Optional
|
|
from logging import Logger
|
|
from nio import (
|
|
AsyncClient,
|
|
SyncResponse,
|
|
SyncError,
|
|
MatrixRoom,
|
|
RoomMessageText,
|
|
)
|
|
|
|
|
|
class MatrixClient:
|
|
def __init__(
|
|
self,
|
|
homeserver: str,
|
|
user_id: str,
|
|
token: str,
|
|
device_id: Optional[str] = "mcdr",
|
|
logger: Union[Logger, Callable, None] = print,
|
|
):
|
|
self.connected: bool = True
|
|
self.client = AsyncClient(homeserver=homeserver)
|
|
self.client.user_id = user_id
|
|
self.client.access_token = token
|
|
self.client.device_id = device_id
|
|
self.logger = logger
|
|
self.network_status: bool = True
|
|
|
|
async def connect(self) -> None:
|
|
if self.connected:
|
|
if self.logger:
|
|
message = (
|
|
"You have connected the matrix client, "
|
|
"please disconnect it before reconnect!"
|
|
)
|
|
|
|
if hasattr(self.logger, "info"):
|
|
self.logger.info(message)
|
|
else:
|
|
self.logger(message)
|
|
return
|
|
|
|
async def receive_messages() -> None:
|
|
client = self.client
|
|
|
|
async def on_sync_response(response: SyncResponse):
|
|
if hasattr(self.logger, "debug"):
|
|
self.logger.debug(f"MatrixClient response: {response}")
|
|
|
|
async def on_sync_error(response: SyncError):
|
|
self.network_status = False
|
|
_message = f"Sync error with matrix homeserver: {response.status_code}"
|
|
_ansi_color_red = "\033[31m"
|
|
_ansi_reset_color = "\033[0m"
|
|
if not self.logger:
|
|
print(_ansi_color_red + _message + _ansi_reset_color)
|
|
if hasattr(self.logger, "error"):
|
|
self.logger.error(_message)
|
|
else:
|
|
self.logger(_ansi_color_red + _message + _ansi_reset_color)
|
|
|
|
client.add_response_callback(on_sync_response, SyncResponse)
|
|
client.add_response_callback(on_sync_error, SyncError)
|
|
|
|
async def message_callback(
|
|
room: MatrixRoom, event: RoomMessageText
|
|
) -> None: # 处理接受到的消息回调
|
|
raise NotImplementedError() # 咕咕咕……
|