htcli 1.1.0__py3-none-any.whl

src/htcli/client/rpc/staking.py

@@ -0,0 +1,216 @@
+"""
+RPC client for staking-related queries using scalecodec for decoding.
+"""
+
+from typing import Optional
+
+from substrateinterface import SubstrateInterface
+
+from ...models.responses.staking import StakeInfo
+from ...utils.logging import get_logger
+
+logger = get_logger(__name__)
+
+
+class StakingRpcClient:
+    """RPC client for staking-related queries."""
+
+    def __init__(self, substrate: Optional[SubstrateInterface] = None):
+        """Initialize the staking RPC client."""
+        self.substrate = substrate
+
+    def get_coldkey_stakes(self, coldkey: str) -> list[StakeInfo]:
+        """
+        Get stake information for a specific coldkey using custom RPC method with full SCALE decoding.
+
+        Args:
+            coldkey: The coldkey account ID (0x... format)
+
+        Returns:
+            List of StakeInfo objects with all fields decoded
+
+        Reference:
+            - mesh-template chain_functions.py lines 1690-1702, 2179-2194
+            - hypertensor-evm/pallets/network/src/lib.rs lines 1218-1223 (SubnetNodeStakeInfo)
+        """
+        try:
+            # Call RPC method
+            result = self.substrate.rpc_request(
+                method="network_getColdkeyStakes", params=[coldkey]
+            )
+
+            if not result or not result.get("result"):
+                return []
+
+            result_data = result["result"]
+
+            # Decode Vec<SubnetNodeStakeInfo> using custom type registry
+            from ...utils.blockchain.type_registry import (
+                decode_vec_subnet_node_stake_info,
+            )
+
+            decoded_list = decode_vec_subnet_node_stake_info(result_data)
+
+            if not decoded_list:
+                return []
+
+            # Convert decoded dicts to StakeInfo model objects
+            stakes = []
+            for decoded in decoded_list:
+                stake_info = StakeInfo(
+                    account_id=coldkey,
+                    subnet_id=decoded.get("subnet_id", 0),
+                    direct_stake_balance=decoded.get("balance", 0),
+                    delegate_stake_shares=0,  # Not in SubnetNodeStakeInfo
+                    delegate_stake_balance=0,  # Not in SubnetNodeStakeInfo
+                )
+                stakes.append(stake_info)
+            return stakes
+
+        except Exception as e:
+            logger.error(f"Error getting coldkey stakes: {e}")
+            return []
+
+    def get_validator_stakes(self, validator_id: int) -> list[StakeInfo]:
+        """
+        Get direct node stake information for a validator.
+
+        Upstream validator ownership RPCs return NodeStakeInfo entries keyed by
+        validator_id instead of the legacy coldkey SubnetNodeStakeInfo payload.
+        """
+        try:
+            result = self.substrate.rpc_request(
+                method="network_getValidatorStakes", params=[validator_id]
+            )
+
+            if not result or not result.get("result"):
+                return []
+
+            result_data = result["result"]
+
+            from ...utils.blockchain.type_registry import decode_vec_node_stake_info
+
+            decoded_list = decode_vec_node_stake_info(result_data)
+
+            if not decoded_list:
+                return []
+
+            stakes = []
+            for decoded in decoded_list:
+                stake_info = StakeInfo(
+                    account_id=f"validator:{validator_id}",
+                    subnet_id=decoded.get("subnet_id", 0) or 0,
+                    direct_stake_balance=decoded.get("balance", 0),
+                    delegate_stake_shares=0,
+                    delegate_stake_balance=0,
+                )
+                stakes.append(stake_info)
+            return stakes
+
+        except Exception as e:
+            logger.error(f"Error getting validator stakes: {e}")
+            return []
+
+    def get_delegate_stakes(self, coldkey: str) -> list[StakeInfo]:
+        """
+        Get delegate stake information for a specific coldkey using custom RPC method with full SCALE decoding.
+
+        Args:
+            coldkey: The coldkey account ID (0x... format)
+
+        Returns:
+            List of StakeInfo objects with all fields decoded
+
+        Reference:
+            - mesh-template chain_functions.py lines 1704-1716, 2196-2213
+            - hypertensor-evm/pallets/network/src/lib.rs lines 1238-1242 (DelegateStakeInfo)
+        """
+        try:
+            # Call RPC method
+            result = self.substrate.rpc_request(
+                method="network_getDelegateStakes", params=[coldkey]
+            )
+
+            if not result or not result.get("result"):
+                return []
+
+            result_data = result["result"]
+
+            # Decode Vec<DelegateStakeInfo> using custom type registry
+            from ...utils.blockchain.type_registry import decode_vec_delegate_stake_info
+
+            decoded_list = decode_vec_delegate_stake_info(result_data)
+
+            if not decoded_list:
+                return []
+
+            # Convert decoded dicts to StakeInfo model objects
+            # DelegateStakeInfo has: subnet_id, shares, balance
+            stakes = []
+            for decoded in decoded_list:
+                stake_info = StakeInfo(
+                    account_id=coldkey,
+                    subnet_id=decoded.get("subnet_id", 0),
+                    delegate_stake_balance=decoded.get("balance", 0),
+                    delegate_stake_shares=decoded.get("shares", 0),  # Delegate shares
+                )
+                stakes.append(stake_info)
+            return stakes
+
+        except Exception as e:
+            logger.error(f"Error getting delegate stakes: {e}")
+            return []
+
+    def get_node_delegate_stakes(self, coldkey: str) -> list[StakeInfo]:
+        """
+        Get node delegate stake information for a specific coldkey using custom RPC method with full SCALE decoding.
+
+        Args:
+            coldkey: The coldkey account ID (0x... format)
+
+        Returns:
+            List of StakeInfo objects with all fields decoded
+
+        Reference:
+            - mesh-template chain_functions.py lines 1718-1730, 2215-2234
+            - hypertensor-evm/pallets/network/src/lib.rs lines 1257-1262 (NodeDelegateStakeInfo)
+        """
+        try:
+            # Call RPC method
+            result = self.substrate.rpc_request(
+                method="network_getNodeDelegateStakes", params=[coldkey]
+            )
+
+            if not result or not result.get("result"):
+                return []
+
+            result_data = result["result"]
+
+            # Decode Vec<NodeDelegateStakeInfo> using custom type registry
+            from ...utils.blockchain.type_registry import (
+                decode_vec_node_delegate_stake_info,
+            )
+
+            decoded_list = decode_vec_node_delegate_stake_info(result_data)
+
+            if not decoded_list:
+                return []
+
+            # Convert decoded dicts to StakeInfo model objects
+            # NodeDelegateStakeInfo has: subnet_id, subnet_node_id, shares, balance
+            stakes = []
+            for decoded in decoded_list:
+                stake_info = StakeInfo(
+                    account_id=coldkey,
+                    subnet_id=decoded.get("subnet_id", 0),
+                    delegate_stake_balance=decoded.get("balance", 0),
+                    delegate_stake_shares=decoded.get(
+                        "shares", 0
+                    ),  # Node delegate shares
+                )
+                stakes.append(stake_info)
+            return stakes
+
+        except Exception as e:
+            logger.error(f"Error getting node delegate stakes: {e}")
+            return []