Feat (ZkSync-lite): Notify decoding progress

Signed-off-by: OjusWiZard <ojuswimail@gmail.com>
rotki · May 20, 2024 · 1542ff2 · 1542ff2
1 parent eaf6193
commit 1542ff2
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 3 deletions.
diff --git a/rotkehlchen/api/rest.py b/rotkehlchen/api/rest.py
@@ -2913,7 +2913,10 @@ def decode_pending_evmlike_transactions(
         """
         decoded_num = 0
         # For now it's only zksync lite
-        decoded_num = self.rotkehlchen.chains_aggregator.zksync_lite.decode_undecoded_transactions(force_redecode=ignore_cache)   # noqa: E501
+        decoded_num = self.rotkehlchen.chains_aggregator.zksync_lite.decode_undecoded_transactions(
+            force_redecode=ignore_cache,
+            send_ws_notifications=True,
+        )
         return {
             'result': {'decoded_tx_number': decoded_num},
             'message': '',

diff --git a/rotkehlchen/api/websockets/typedefs.py b/rotkehlchen/api/websockets/typedefs.py
@@ -25,6 +25,7 @@ class WSMessageType(Enum):
     DATABASE_UPLOAD_RESULT = auto()
     ACCOUNTING_RULE_CONFLICT = auto()
     EVM_UNDECODED_TRANSACTIONS = auto()
+    EVMLIKE_UNDECODED_TRANSACTIONS = auto()
     CALENDAR_REMINDER = auto()
 
     def __str__(self) -> str:

diff --git a/rotkehlchen/chain/zksync_lite/manager.py b/rotkehlchen/chain/zksync_lite/manager.py
@@ -11,6 +11,7 @@
 from pysqlcipher3.dbapi2 import IntegrityError
 
 from rotkehlchen.accounting.structures.balance import Balance
+from rotkehlchen.api.websockets.typedefs import WSMessageType
 from rotkehlchen.assets.asset import Asset, CryptoAsset
 from rotkehlchen.assets.utils import TokenEncounterInfo, get_or_create_evm_token
 from rotkehlchen.chain.ethereum.utils import asset_normalized_value
@@ -61,6 +62,7 @@ def __init__(
             database: 'DBHandler',
     ) -> None:
         self.database = database
+        self.msg_aggregator = database.msg_aggregator
         self.session = requests.session()
         set_user_agent(self.session)
         self.id_to_token: dict[int, CryptoAsset] = {}
@@ -838,7 +840,11 @@ def decode_transaction(
                 (1, transaction.tx_hash),
             )
 
-    def decode_undecoded_transactions(self, force_redecode: bool) -> int:
+    def decode_undecoded_transactions(
+            self,
+            force_redecode: bool,
+            send_ws_notifications: bool = False,
+    ) -> int:
         """Decodes undecoded zksync lite transactions. If force redecode is True
         then all transactions are redecoded.
         Returns the number of decoded transactions (not events in transactions)
@@ -851,7 +857,7 @@ def decode_undecoded_transactions(self, force_redecode: bool) -> int:
         with self.database.conn.read_ctx() as cursor:
             tracked_addresses = self.database.get_blockchain_accounts(cursor).zksync_lite
 
-        for transaction in transactions:
+        for tx_index, transaction in enumerate(transactions):
             with self.database.user_write() as write_cursor:  # delete old tx events
                 write_cursor.execute(
                     'DELETE FROM history_events WHERE event_identifier=?',
@@ -860,4 +866,24 @@ def decode_undecoded_transactions(self, force_redecode: bool) -> int:
 
             self.decode_transaction(transaction, tracked_addresses)
 
+            if send_ws_notifications and tx_index % 10 == 0:
+                self.msg_aggregator.add_message(
+                    message_type=WSMessageType.EVMLIKE_UNDECODED_TRANSACTIONS,
+                    data={
+                        'evm_chain': self.ethereum_inquirer.chain_name,
+                        'total': len(transactions),
+                        'processed': tx_index,
+                    },
+                )
+
+        if send_ws_notifications:
+            self.msg_aggregator.add_message(
+                message_type=WSMessageType.EVMLIKE_UNDECODED_TRANSACTIONS,
+                data={
+                    'evm_chain': self.ethereum_inquirer.chain_name,
+                    'total': len(transactions),
+                    'processed': len(transactions),
+                },
+            )
+
         return len(transactions)