Merge branch 'dev' of inex.dev:def/MirrorTea into dev
This commit is contained in:
commit
67856a849d
|
@ -1,7 +1,7 @@
|
||||||
import asyncio
|
import sys
|
||||||
|
|
||||||
from matrix import MatrixLoop
|
import asyncio as telegram
|
||||||
from abstract_source_platform.telegram import Telegram
|
import nio as matrix
|
||||||
|
|
||||||
from config_dataclass import Config
|
from config_dataclass import Config
|
||||||
|
|
||||||
|
@ -11,14 +11,61 @@ CONFIG_FILE_NAME = "config.yaml"
|
||||||
def main():
|
def main():
|
||||||
config = Config.from_yaml_config(CONFIG_FILE_NAME)
|
config = Config.from_yaml_config(CONFIG_FILE_NAME)
|
||||||
|
|
||||||
asyncio.run(Application(config).run())
|
telegram.run(Application(config).run())
|
||||||
|
|
||||||
|
|
||||||
|
class TelegramLopp:
|
||||||
|
def __init__(self, app):
|
||||||
|
self.app = app
|
||||||
|
self.bot = telegram.Bot(token=app.config.telegram_bot_token)
|
||||||
|
self.dispatcher = telegram.Dispatcher(bot=self.bot)
|
||||||
|
self.dispatcher.register_message_handler(self.on_message)
|
||||||
|
|
||||||
|
async def run(self) -> None:
|
||||||
|
await self.dispatcher.start_polling()
|
||||||
|
|
||||||
|
async def on_message(self, msg):
|
||||||
|
print(msg, file=sys.stderr)
|
||||||
|
|
||||||
|
|
||||||
|
class MatrixLoop:
|
||||||
|
def __init__(self, app):
|
||||||
|
self.app = app
|
||||||
|
self.client = matrix.AsyncClient(
|
||||||
|
app.config.matrix_homeserver_url,
|
||||||
|
app.config.matrix_full_bot_id,
|
||||||
|
)
|
||||||
|
self.client.add_event_callback(self.on_message, matrix.RoomMessage)
|
||||||
|
|
||||||
|
async def prepare(self):
|
||||||
|
await self.client.login(self.app.config.matrix_bot_password)
|
||||||
|
|
||||||
|
async def finish(self):
|
||||||
|
await self.client.close()
|
||||||
|
|
||||||
|
async def run(self):
|
||||||
|
await self.client.sync_forever(timeout=30000)
|
||||||
|
|
||||||
|
async def on_message(self, room, event):
|
||||||
|
print(room, event, file=sys.stderr)
|
||||||
|
|
||||||
|
def upgrade_room(self, room, telegram_nickname):
|
||||||
|
event_dict = matrix.event_builders.event_builder.EventBuilder(
|
||||||
|
name=telegram_nickname
|
||||||
|
).as_dict()
|
||||||
|
client.room_send(
|
||||||
|
room_id=room,
|
||||||
|
message_type=event_dict["type"],
|
||||||
|
content=event_dict["content"],
|
||||||
|
) # предположу что оно так работает
|
||||||
|
# https://matrix-nio.readthedocs.io/en/latest/nio.html#module-nio.event_builders.state_events
|
||||||
|
|
||||||
|
|
||||||
class Application:
|
class Application:
|
||||||
def __init__(self, config):
|
def __init__(self, config):
|
||||||
self.config = config
|
self.config = config
|
||||||
self.matrix_loop = MatrixLoop(self)
|
self.matrix_loop = MatrixLoop(self)
|
||||||
self.telegram = Telegram(self)
|
self.telegram = TelegramLopp(self)
|
||||||
|
|
||||||
async def run(self):
|
async def run(self):
|
||||||
try:
|
try:
|
||||||
|
|
Reference in a new issue