license_service

app/core/models/content/indexation_mixins.py

@@ -1,3 +1,10 @@
+import traceback
+
+import base58
+from sqlalchemy import and_
+
+from app.core.logger import make_log
+from app.core.models import StoredContent
 from app.core.models.wallet_connection import WalletConnection
 from app.core._blockchain.ton.toncenter import toncenter
 from tonsdk.boc import Cell
@@ -35,10 +42,44 @@ class NodeStorageIndexationMixin:
     pass # async def fetch_onchain_metadata(self):
 
 
-
 class UserContentIndexationMixin:
-    pass
+    async def sync_with_chain(self, db_session):
+        errored = False
+        cc_indexator_result = await toncenter.run_get_method(self.onchain_address, 'indexator_data')
+        if cc_indexator_result.get('exit_code', -1) != 0:
+            errored = True
 
+        if not errored:
+            try:
+                cc_indexator_data = unpack_item_indexator_data(cc_indexator_result)
+                assert cc_indexator_data['type'] == 1, "Type is not a content"
+                assert cc_indexator_data['address'] == self.onchain_address, "Address is not equal"
+                values_slice = cc_indexator_data['values'].begin_parse()
+                content_hash_b58 = base58.b58encode(bytes.fromhex(values_slice.read_uint(256).hex()[2:])).decode()
+                stored_content = db_session.query(StoredContent).filter(
+                    and_(
+                        StoredContent.type == 'onchain/content',
+                        StoredContent.hash == content_hash_b58,
+
+                    )
+                ).first()
+                trusted_cop_address_result = await toncenter.run_get_method(stored_content.meta['item_address'], 'get_nft_address_by_index', [['num', cc_indexator_data['index']]])
+                assert trusted_cop_address_result.get('exit_code', -1) == 0, "Trusted cop address error"
+                trusted_cop_address = Cell.one_from_boc(b64decode(trusted_cop_address_result['stack'][0][1]['bytes'])).begin_parse().read_msg_addr().to_string(1, 1, 1)
+                make_log("UserContent", f"Trusted cop address: {trusted_cop_address} / Contract address: {trusted_cop_address}", level="info")
+                assert trusted_cop_address == cc_indexator_data['address'], "Trusted cop address is not equal"
+                self.owner_address = cc_indexator_data['owner_address']
+                self.type = 'nft/listen'
+                self.content_id = stored_content.id
+                db_session.commit()
+            except BaseException as e:
+                errored = True
+                make_log("UserContent", f"Error: {e}" + '\n' + traceback.format_exc(), level="error")
+
+        if errored is True:
+            self.type = 'nft/unknown'
+            self.content_id = None
+            db_session.commit()