MirrorTea/mirrortea/__main__.py

import asyncio
import os
import sqlite3
import sys

import aiogram as telegram
import nio as matrix

TELEGRAM_USER_MATRIX_CHATS_SQL = '''
CREATE TABLE IF NOT EXISTS telegram_user_matrix_chats (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    telegram_user_id INTEGER NOT NULL,
    matrix_chat_id INTEGER NOT NULL,
    FOREIGN KEY(telegram_user_id) REFERENCES telegram_users(id),
    FOREIGN KEY(matrix_chat_id) REFERENCES matrix_chats(id)
)
'''

def main():
    config = Config(
        db_path=os.environ['DB_PATH'],
        matrix_bot_id=os.environ['MATRIX_BOT_ID'],
        matrix_homeserver_url=os.environ['MATRIX_HOMESERVER_URL'],
        matrix_owner_id=os.environ['MATRIX_OWNER_ID'],
        matrix_password=os.environ['MATRIX_PASSWORD'],
        telegram_bot_token=os.environ['TELEGRAM_BOT_TOKEN'],
    )

    asyncio.run(Application(config).run())

class Config:
    def __init__(self, **kwargs):
        self.db_path               = kwargs['db_path']
        self.matrix_bot_id         = kwargs['matrix_bot_id']
        self.matrix_homeserver_url = kwargs['matrix_homeserver_url']
        self.matrix_owner_id       = kwargs['matrix_owner_id']
        self.matrix_password       = kwargs['matrix_password']
        self.telegram_bot_token    = kwargs['telegram_bot_token']

class Application:
    def __init__(self, config):
        self.config = config
        self.sqlite_adapter = SqliteAdapter(self, config.db_path)
        self.matrix_loop = MatrixLoop(self)
        self.telegram_loop = TelegramLoop(self)

    async def run(self):
        try:
            await self.matrix_loop.prepare()
            await asyncio.gather(
                self.matrix_loop.run(),
                self.telegram_loop.run(),
            )
        finally:
            if self.matrix_loop:
                await self.matrix_loop.finish()

class MatrixLoop:
    def __init__(self, app):
        self.app = app
        self.client = matrix.AsyncClient(
            app.config.matrix_homeserver_url,
            app.config.matrix_bot_id,
        )
        self.client.add_event_callback(self.on_message, matrix.RoomMessage)

    async def prepare(self):
        await self.client.login(self.app.config.matrix_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)

class TelegramLoop:
    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):
        await self.dispatcher.start_polling()

    async def on_message(self, msg):
        print(msg, file=sys.stderr)

class SqliteAdapter:
    def __init__(self, app, path):
        self.app = app
        self.path = path
        self.conn = sqlite3.connect(path)
        self._create_tables()

    def _create_tables(self):
        self.conn.execute(TELEGRAM_USER_MATRIX_CHATS_SQL)
        self.conn.commit()

if __name__ == '__main__':
    main()