referral sales

app/client_bot/routers/content.py

@@ -8,7 +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.core.models.content.user_content import UserAction, UserContent
 
 from app.client_bot.routers.home import router as home_router
 from app.client_bot.routers.tonconnect import router as tonconnect_router
@@ -94,7 +94,17 @@ router.callback_query.register(t_callback_purchase_node_content, F.data.startswi
 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:
-        args = query.query[1:]
+        source_args = query.query[1:]
+        source_args_ext = query.query
+
+        if query.query.startswith('Q'):
+            license_onchain_address = query.query[1:]
+            args = db_session.query(UserContent).filter_by(
+                onchain_address=license_onchain_address,
+            ).first().content.cid.serialize_v2()
+        else:
+            args = query.query[1:]
+    
         cid = ContentId.deserialize(args)
 
         content_list = []
@@ -174,7 +184,7 @@ async def t_inline_query_node_content(query: types.InlineQuery, memory=None, use
 
         content_share_link = {
             'text': user.translated('p_shareLinkContext').format(title=' – '.join(audio_title)),
-            'url': f"https://t.me/{CLIENT_TELEGRAM_BOT_USERNAME}/content?startapp={content.cid.serialize_v2()}"
+            'url': f"https://t.me/{CLIENT_TELEGRAM_BOT_USERNAME}/content?startapp={source_args}"
         }
 
         # Create inline query result using decrypted content's file_id
@@ -192,7 +202,7 @@ async def t_inline_query_node_content(query: types.InlineQuery, memory=None, use
                                     'audio': user.translated('shareTrack_button'),
                                     'video': user.translated('shareVideo_button'),
                                 }[content_type_declared],
-                                'switch_inline_query': f"C{content.cid.serialize_v2()}"
+                                'switch_inline_query': source_args_ext
                             },
                             {
                                 'text': user.translated('shareLink_button'),
@@ -201,7 +211,7 @@ async def t_inline_query_node_content(query: types.InlineQuery, memory=None, use
                         ],
                         [{
                             'text': user.translated('viewTrack_button'),
-                            'url': f"https://t.me/MY_Web3Bot/content?startapp={content.cid.serialize_v2()}"
+                            'url': f"https://t.me/MY_Web3Bot/content?startapp={source_args}"
                         }]
                     ]),
                     **result_kwargs
@@ -229,7 +239,7 @@ async def t_inline_query_node_content(query: types.InlineQuery, memory=None, use
 
                         [{
                             'text': user.translated('viewTrack_button'),
-                            'url': f"https://t.me/MY_Web3Bot/content?startapp={content.cid.serialize_v2()}"
+                            'url': f"https://t.me/MY_Web3Bot/content?startapp={source_args}"
                         }]
                     ]),
                     **result_kwargs
@@ -248,5 +258,5 @@ async def t_chosen_inline_result_node_content(query: types.ChosenInlineResult, m
     # return await query.answer([])
 
 
-router.inline_query.register(t_inline_query_node_content, F.query.startswith('C'))
+router.inline_query.register(t_inline_query_node_content, F.query.startswith('C') | F.query.startswith('Q'))
 router.chosen_inline_result.register(t_chosen_inline_result_node_content, lambda: True)