license_service

app/client_bot/routers/content.py

@@ -8,6 +8,7 @@ from app.core._keyboards import get_inline_keyboard
 from app.core.models.node_storage import StoredContent
 import json
 from app.core.logger import make_log
+from app.core.models.content.user_content import UserAction
 
 from app.client_bot.routers.home import router as home_router
 from app.client_bot.routers.tonconnect import router as tonconnect_router
@@ -15,10 +16,63 @@ from app.core._config import CLIENT_TELEGRAM_BOT_USERNAME
 from app.core.logger import logger
 from app.core.content.content_id import ContentId
 import base58
+from datetime import datetime, timedelta
+import asyncio
+from app.core._blockchain.ton.connect import TonConnect, wallet_obj_by_name
+
 
 router = Router()
 
 
+async def t_callback_purchase_node_content(query: types.CallbackQuery, memory=None, user=None, db_session=None, chat_wrap=None, **extra):
+    content_oid = int(query.data.split('_')[1])
+    make_log("OwnedContent", f"{user} Try to purchase content: {content_oid}", level='info')
+    content = db_session.query(StoredContent).filter_by(id=content_oid).first()
+    if not content:
+        return await query.answer(user.translated('error_contentNotFound'), show_alert=True)
+
+    license_price = content.meta.get('license', {}).get('listen', {}).get('price')
+    license_price_num = int(license_price)
+    if license_price_num < 1:
+        return await query.answer(user.translated('error_contentPrice'), show_alert=True)
+
+    ton_connect, ton_connection = TonConnect.by_user(db_session, user, callback_fn=())
+    await ton_connect.restore_connection()
+    assert ton_connect.connected, "No connected wallet"
+
+    user_wallet_address = user.wallet_address(db_session)
+
+    memory._app.add_task(ton_connect._sdk_client.send_transaction({
+        'valid_until': int(datetime.now().timestamp() + 300),
+        'messages': [
+            {
+                'address': content.meta['item_address'],
+                'amount': license_price
+            }
+        ]
+    }))
+
+    new_action = UserAction(
+        type='purchase',
+        user_id=user.id,
+        content_id=content.id,
+        telegram_message_id=query.message.message_id,
+        from_address=user_wallet_address,
+        to_address=content.meta['item_address'],
+        status='requested',
+        meta={
+            'confirmation_url': wallet_obj_by_name(ton_connection.wallet_key.split('==')[0])['universal_url']
+        },
+        created_at=datetime.now()
+    )
+    db_session.add(new_action)
+    db_session.commit()
+    await user.send_content(db_session, content, message_id=query.message.message_id)
+
+
+router.callback_query.register(t_callback_purchase_node_content, F.data.startswith('PC_'))
+
+
 async def t_inline_query_node_content(query: types.InlineQuery, memory=None, user=None, db_session=None, chat_wrap=None, **extra):
     make_log("OwnedContent", f"Inline query: {query.query}", level='info')
     try: