add runtimeAPI for new dyn staking fee

Author: camfairchild

pallets/subtensor/runtime-api/src/lib.rs

@@ -46,6 +46,7 @@ sp_api::decl_runtime_apis! {
         fn get_stake_info_for_coldkey( coldkey_account: AccountId32 ) -> Vec<StakeInfo<AccountId32>>;
         fn get_stake_info_for_coldkeys( coldkey_accounts: Vec<AccountId32> ) -> Vec<(AccountId32, Vec<StakeInfo<AccountId32>>)>;
         fn get_stake_info_for_hotkey_coldkey_netuid( hotkey_account: AccountId32, coldkey_account: AccountId32, netuid: u16 ) -> Option<StakeInfo<AccountId32>>;
+        fn get_stake_fee_for_hotkey_coldkey_netuid( hotkey_account: AccountId32, coldkey_account: AccountId32, origin_netuid: u16, destination_netuid: u16, amount: i64 ) -> u64;
     }
 
     pub trait SubnetRegistrationRuntimeApi {

pallets/subtensor/src/rpc_info/stake_info.rs

@@ -1,7 +1,9 @@
 use super::*;
 use frame_support::pallet_prelude::{Decode, Encode};
+use sp_std::ops::Neg;
 extern crate alloc;
 use codec::Compact;
+use substrate_fixed::types::I96F32;
 
 #[freeze_struct("5cfb3c84c3af3116")]
 #[derive(Decode, Encode, PartialEq, Eq, Clone, Debug, TypeInfo)]
@@ -112,4 +114,23 @@ impl<T: Config> Pallet<T> {
             is_registered,
         })
     }
+
+    pub fn get_stake_fee_for_hotkey_coldkey_netuid(
+        hotkey_account: T::AccountId,
+        _coldkey_account: T::AccountId,
+        origin_netuid: u16,
+        _destination_netuid: u16,
+        amount: i64,
+    ) -> u64 {
+        if amount >= 0 {
+            DefaultStakingFee::<T>::get()
+        } else {
+            // Calculate fee for unstake (negative amount)
+            Self::calculate_staking_fee(
+                origin_netuid,
+                &hotkey_account,
+                I96F32::saturating_from_num(amount.neg()),
+            )
+        }
+    }
 }

runtime/src/lib.rs

@@ -2066,6 +2066,10 @@ impl_runtime_apis! {
         fn get_stake_info_for_hotkey_coldkey_netuid( hotkey_account: AccountId32, coldkey_account: AccountId32, netuid: u16 ) -> Option<StakeInfo<AccountId32>> {
             SubtensorModule::get_stake_info_for_hotkey_coldkey_netuid( hotkey_account, coldkey_account, netuid )
         }
+
+        fn get_stake_fee_for_hotkey_coldkey_netuid( hotkey_account: AccountId32, coldkey_account: AccountId32, origin_netuid: u16, destination_netuid: u16, amount: i64 ) -> u64 {
+            SubtensorModule::get_stake_fee_for_hotkey_coldkey_netuid( hotkey_account, coldkey_account, origin_netuid, destination_netuid, amount )
+        }
     }
 
     impl subtensor_custom_rpc_runtime_api::SubnetRegistrationRuntimeApi<Block> for Runtime {