feat: some func signature changes + hotkey stake

Author: Shr1ftyy

runtime/src/precompiles/solidity/staking.abi

@@ -103,7 +103,26 @@
                 "type": "bytes32"
             }
         ],
-        "name": "getStakeColdkey",
+        "name": "getTotalColdkeyStake",
+        "outputs": [
+            {
+                "internalType": "uint256",
+                "name": "",
+                "type": "uint256"
+            }
+        ],
+        "stateMutability": "view",
+        "type": "function"
+    },
+    {
+        "inputs": [
+            {
+                "internalType": "bytes32",
+                "name": "hotkey",
+                "type": "bytes32"
+            }
+        ],
+        "name": "getTotalHotkeyStake",
         "outputs": [
             {
                 "internalType": "uint256",

runtime/src/precompiles/solidity/staking.sol

@@ -46,13 +46,25 @@ interface IStaking {
    * @dev Returns the amount of RAO staked by the coldkey.
    *
    * This function allows external accounts and contracts to query the amount of RAO staked by the coldkey
-   * which effectively calls `get_coldkey_stake` on the subtensor pallet with
+   * which effectively calls `get_total_coldkey_stake` on the subtensor pallet with
    * specified coldkey as a parameter.
    *
    * @param coldkey The coldkey public key (32 bytes).
    * @return The amount of RAO staked by the coldkey.
    */
-  function getStakeColdkey(bytes32 coldkey) external view returns (uint256);
+  function getTotalColdkeyStake(bytes32 coldkey) external view returns (uint256);
+
+  /**
+   * @dev Returns the total amount of stake under a hotkey (delegative or otherwise)
+   * 
+   * This function allows external accounts and contracts to query the total amount of RAO staked under a hotkey
+   * which effectively calls `get_total_hotkey_stake` on the subtensor pallet with
+   * specified hotkey as a parameter.
+   * 
+   * @param hotkey The hotkey public key (32 bytes).
+   * @return The total amount of RAO staked under the hotkey.
+   */
+  function getTotalHotkeyStake(bytes32 hotkey) external view returns (uint256);
 
   /**
    * @dev Delegates staking to a proxy account.

runtime/src/precompiles/staking.rs

@@ -61,8 +61,10 @@ impl StakingPrecompile {
             Self::remove_stake(handle, &method_input)
         } else if method_id == get_method_id("getStake(bytes32,bytes32,uint256)") {
             Self::get_stake(&method_input)
-        } else if method_id == get_method_id("getStakeColdkey(bytes32)") {
-            Self::get_stake_coldkey(&method_input)
+        } else if method_id == get_method_id("getTotalColdkeyStake(bytes32)") {
+            Self::get_total_coldkey_stake(&method_input)
+        } else if method_id == get_method_id("getTotalHotkeyStake(bytes32)") {
+            Self::get_total_hotkey_stake(&method_input)
         } else if method_id == get_method_id("addProxy(bytes32)") {
             Self::add_proxy(handle, &method_input)
         } else if method_id == get_method_id("removeProxy(bytes32)") {
@@ -116,7 +118,7 @@ impl StakingPrecompile {
         Self::dispatch(handle, call)
     }
 
-    fn get_stake_coldkey(data: &[u8]) -> PrecompileResult { 
+    fn get_total_coldkey_stake(data: &[u8]) -> PrecompileResult { 
         let coldkey: AccountId32 = Self::parse_pub_key(data)?.into();
 
         // get total stake of coldkey
@@ -131,6 +133,21 @@ impl StakingPrecompile {
         })
     }
 
+    fn get_total_hotkey_stake(data: &[u8]) -> PrecompileResult {
+        let hotkey: AccountId32 = Self::parse_pub_key(data)?.into();
+
+        // get total stake of hotkey
+        let total_stake = pallet_subtensor::Pallet::<Runtime>::get_total_stake_for_hotkey(&hotkey);
+        let result_u256 = U256::from(total_stake);
+        let mut result = [0_u8; 32];
+        U256::to_big_endian(&result_u256, &mut result);
+
+        Ok(PrecompileOutput {
+            exit_status: ExitSucceed::Returned,
+            output: result.into(),
+        })
+    }
+
     fn add_proxy(handle: &mut impl PrecompileHandle, data: &[u8]) -> PrecompileResult {
         let delegate = AccountId32::from(Self::parse_pub_key(data)?);
         let delegate = <Runtime as frame_system::Config>::Lookup::unlookup(delegate);