Add move_stake and transfer_stake precompiles.

Author: shamil-gadelshin

pallets/subtensor/src/macros/dispatches.rs

@@ -2435,5 +2435,85 @@ mod dispatches {
                 allow_partial,
             )
         }
+
+        /// move_stake with EVM origin
+        #[pallet::call_index(127)]
+        #[pallet::weight((Weight::from_parts(157_100_000, 0)
+        .saturating_add(T::DbWeight::get().reads(15_u64))
+        .saturating_add(T::DbWeight::get().writes(7_u64)), DispatchClass::Operational, Pays::Yes))]
+        pub fn move_stake_evm(
+            origin: OriginFor<T>,
+            origin_hotkey: T::AccountId,
+            destination_hotkey: T::AccountId,
+            origin_netuid: NetUid,
+            destination_netuid: NetUid,
+            alpha_amount: AlphaCurrency,
+        ) -> DispatchResult {
+            let account_id = crate::ensure_evm_origin(<T as Config>::RuntimeOrigin::from(origin))?;
+            let verified_evm_origin = RawOrigin::Signed(account_id);
+            
+            Self::do_move_stake(
+                verified_evm_origin.into(),
+                origin_hotkey,
+                destination_hotkey,
+                origin_netuid,
+                destination_netuid,
+                alpha_amount,
+            )
+        }
+
+        /// transfer_stake with EVM origin
+        #[pallet::call_index(128)]
+        #[pallet::weight((Weight::from_parts(154_800_000, 0)
+        .saturating_add(T::DbWeight::get().reads(13_u64))
+        .saturating_add(T::DbWeight::get().writes(6_u64)), DispatchClass::Operational, Pays::Yes))]
+        pub fn transfer_stake_evm(
+            origin: OriginFor<T>,
+            destination_coldkey: T::AccountId,
+            hotkey: T::AccountId,
+            origin_netuid: NetUid,
+            destination_netuid: NetUid,
+            alpha_amount: AlphaCurrency,
+        ) -> DispatchResult {
+            let account_id = crate::ensure_evm_origin(<T as Config>::RuntimeOrigin::from(origin))?;
+            let verified_evm_origin = RawOrigin::Signed(account_id);
+            
+            Self::do_transfer_stake(
+                verified_evm_origin.into(),
+                destination_coldkey,
+                hotkey,
+                origin_netuid,
+                destination_netuid,
+                alpha_amount,
+            )
+        }
+
+        /// swap_stake with EVM origin
+        #[pallet::call_index(129)]
+        #[pallet::weight((
+            Weight::from_parts(351_300_000, 0)
+            .saturating_add(T::DbWeight::get().reads(32))
+            .saturating_add(T::DbWeight::get().writes(17)),
+            DispatchClass::Operational,
+            Pays::Yes
+        ))]
+        pub fn swap_stake(
+            origin: OriginFor<T>,
+            hotkey: T::AccountId,
+            origin_netuid: NetUid,
+            destination_netuid: NetUid,
+            alpha_amount: AlphaCurrency,
+        ) -> DispatchResult {
+            let account_id = crate::ensure_evm_origin(<T as Config>::RuntimeOrigin::from(origin))?;
+            let verified_evm_origin = RawOrigin::Signed(account_id);
+            
+            Self::do_swap_stake(
+                verified_evm_origin.into(),
+                hotkey,
+                origin_netuid,
+                destination_netuid,
+                alpha_amount,
+            )
+        }
     }
 }

precompiles/src/staking.rs

@@ -179,15 +179,17 @@ where
         let origin_netuid = try_u16_from_u256(origin_netuid)?;
         let destination_netuid = try_u16_from_u256(destination_netuid)?;
         let alpha_amount: u64 = amount_alpha.unique_saturated_into();
-        let call = pallet_subtensor::Call::<R>::move_stake {
+        let call = pallet_subtensor::Call::<R>::move_stake_evm {
             origin_hotkey,
             destination_hotkey,
             origin_netuid: origin_netuid.into(),
             destination_netuid: destination_netuid.into(),
             alpha_amount: alpha_amount.into(),
         };
 
-        handle.try_dispatch_runtime_call::<R, _>(call, RawOrigin::Signed(account_id))
+        let evm_origin = R::EvmOriginHelper::make_evm_origin(account_id);
+
+        handle.try_dispatch_runtime_call_with_custom_origin::<R, _>(call, evm_origin)
     }
 
     #[precompile::public("transferStake(bytes32,bytes32,uint256,uint256,uint256)")]
@@ -205,15 +207,17 @@ where
         let origin_netuid = try_u16_from_u256(origin_netuid)?;
         let destination_netuid = try_u16_from_u256(destination_netuid)?;
         let alpha_amount: u64 = amount_alpha.unique_saturated_into();
-        let call = pallet_subtensor::Call::<R>::transfer_stake {
+        let call = pallet_subtensor::Call::<R>::transfer_stake_evm {
             destination_coldkey,
             hotkey,
             origin_netuid: origin_netuid.into(),
             destination_netuid: destination_netuid.into(),
             alpha_amount: alpha_amount.into(),
         };
 
-        handle.try_dispatch_runtime_call::<R, _>(call, RawOrigin::Signed(account_id))
+        let evm_origin = R::EvmOriginHelper::make_evm_origin(account_id);
+
+        handle.try_dispatch_runtime_call_with_custom_origin::<R, _>(call, evm_origin)
     }
 
     #[precompile::public("getTotalColdkeyStake(bytes32)")]