Initial commit

app/bot/__init__.py

@@ -0,0 +1,78 @@
+from datetime import datetime
+
+from aiogram import BaseMiddleware, Dispatcher
+from aiogram.fsm.storage.memory import MemoryStorage
+
+from app.bot.routers.index import main_router
+from app.core.logger import logger
+from app.core.models._telegram import Wrapped_CBotChat
+from app.core.models.user import User
+from app.core.storage import db_session
+
+dp = Dispatcher(storage=MemoryStorage())
+
+
+class UserDataMiddleware(BaseMiddleware):
+    async def __call__(self, handler, event, data):
+        update_body = event.message or event.callback_query
+        if not update_body:
+            return
+
+        if update_body.from_user.is_bot is True:
+            return
+
+        user_id = update_body.from_user.id
+        assert user_id >= 1
+        # TODO: maybe make users cache
+
+        with db_session(auto_commit=False) as session:
+            try:
+                user = session.query(User).filter_by(telegram_id=user_id).first()
+            except BaseException as e:
+                logger.error(f"Error when middleware getting user: {e}")
+                user = None
+
+            if user is None:
+                logger.debug(f"User {user_id} not found. Creating new user")
+                user = User(
+                    telegram_id=user_id,
+                    username=update_body.from_user.username,
+                    lang_code='en',
+                    last_use=datetime.now(),
+                    meta=dict(first_name=update_body.from_user.first_name,
+                                  last_name=update_body.from_user.last_name, username=update_body.from_user.username,
+                                  language_code=update_body.from_user.language_code,
+                                  is_premium=update_body.from_user.is_premium),
+                    created=datetime.now()
+                )
+                session.add(user)
+                session.commit()
+            else:
+                if user.username != update_body.from_user.username:
+                    user.username = update_body.from_user.username
+
+                updated_meta_fields = {}
+                if user.meta.get('first_name') != update_body.from_user.first_name:
+                    updated_meta_fields['first_name'] = update_body.from_user.first_name
+
+                if user.meta.get('last_name') != update_body.from_user.last_name:
+                    updated_meta_fields['last_name'] = update_body.from_user.last_name
+
+                user.meta = {
+                    **user.meta,
+                    **updated_meta_fields
+                }
+
+                user.last_use = datetime.now()
+                session.commit()
+
+            data['user'] = user
+            data['db_session'] = session
+            data['chat_wrap'] = Wrapped_CBotChat(data['bot'], chat_id=user_id)
+            data['memory'] = dp._s_memory
+            result = await handler(event, data)
+            return result
+
+
+dp.update.outer_middleware(UserDataMiddleware())
+dp.include_router(main_router)

app/bot/routers/index.py

@@ -0,0 +1,54 @@
+import os
+import sys
+import traceback
+
+from aiogram import types, Router, F
+from aiogram.filters import Command
+
+from app.bot.routers.tonconnect import router as tonconnect_router
+from app.core._utils.tg_process_template import tg_process_template
+from app.core.logger import logger
+
+main_router = Router()
+
+
+async def t_home_menu(__msg, **extra):
+    memory, user, db_session, chat_wrap = extra['memory'], extra['user'], extra['db_session'], extra['chat_wrap']
+    if extra.get('state'):
+        await extra['state'].clear()
+
+    return await tg_process_template(
+        chat_wrap, user.translated('home_menu'), message_id=__msg.message.message_id if isinstance(__msg, types.CallbackQuery) else None
+    )
+
+
+main_router.message.register(t_home_menu, Command('start'))
+main_router.callback_query.register(t_home_menu, F.data == 'home')
+
+main_router.include_routers(tonconnect_router)
+
+closing_router = Router()
+
+
+@closing_router.message()
+async def t_index(message: types.Message, **extra):
+    return await message.answer(extra['user'].translated('error_unknownCommand'), parse_mode='html')
+
+main_router.include_routers(closing_router)
+
+
+@main_router.error()
+async def t_index_error(err_event: types.ErrorEvent, **extra):
+    try:
+        raise err_event.exception
+    except BaseException as e:
+        exc_type, exc_obj, exc_tb = sys.exc_info()
+
+        try:
+            filename = os.path.split(exc_tb.tb_frame.f_code.co_filename)[1]
+        except:
+            filename = None
+
+        logger.error(f"""Error: {e}
+-/ {exc_type} {exc_obj} {filename} {exc_tb.tb_frame.f_lineno}""")
+        traceback.print_tb(exc_tb)

app/bot/routers/tonconnect.py

@@ -0,0 +1,66 @@
+import json
+
+from aiogram import types, Router
+from aiogram.filters import Command
+
+from app.core._keyboards import get_inline_keyboard
+from app.core._utils.tg_process_template import tg_process_template
+from app.core.logger import make_log
+from app.core.models._blockchain.ton.connect import TonConnect, unpack_wallet_info
+
+router = Router()
+
+
+async def pause_ton_connection(ton_connect: TonConnect):
+    if ton_connect.connected:
+        ton_connect._sdk_client.pause_connection()
+
+
+async def t_tonconnect_dev_menu(message: types.Message, memory=None, user=None, db_session=None, chat_wrap=None, **extra):
+    try:
+        command_args = message.text.split(" ")[1:]
+    except BaseException as e:
+        command_args = []
+
+    make_log("TonConnect_DevMenu", f"Command args: {command_args}", level='info')
+    wallet_app_name = 'tonkeeper'
+    if len(command_args) > 0:
+        wallet_app_name = command_args[0].lower()
+
+    keyboard = []
+
+    ton_connect, ton_connection = TonConnect.by_user(db_session, user, callback_fn=())
+    await ton_connect.restore_connection()
+    make_log("TonConnect_DevMenu", f"SDK connected?: {ton_connect.connected}", level='info')
+    if not ton_connect.connected:
+        if ton_connection:
+            make_log("TonConnect_DevMenu", f"Invalidating old connection", level='debug')
+            ton_connection.invalidated = True
+            db_session.commit()
+
+        message_text = f"""<b>Wallet is not connected</b>
+        
+Use /dev_tonconnect <code>{wallet_app_name}</code> for connect to wallet."""
+        connection_link = await ton_connect.new_connection(wallet_app_name)
+        ton_connect.connected
+        make_log("TonConnect_DevMenu", f"New connection link for {wallet_app_name}: {connection_link}", level='debug')
+        keyboard.append([
+            {
+                'text': 'Connect',
+                'url': connection_link
+            }
+        ])
+    else:
+        wallet_info_text = json.dumps(unpack_wallet_info(ton_connect._sdk_client._wallet), indent=4, ensure_ascii=False)
+        message_text = f"""<b>Wallet is connected</b>
+        
+<pre>{wallet_info_text}</pre>"""
+
+    memory.add_task(pause_ton_connection, ton_connect, delay_s=60 * 3)
+
+    return await tg_process_template(
+        chat_wrap, message_text,
+        keyboard=get_inline_keyboard(keyboard) if keyboard else None
+    )
+
+router.message.register(t_tonconnect_dev_menu, Command('dev_tonconnect'))