add toggle off for transfers

Author: unconst

pallets/admin-utils/src/lib.rs

@@ -1287,6 +1287,27 @@ pub mod pallet {
             ensure_root(origin)?;
             T::Grandpa::schedule_change(next_authorities, in_blocks, forced)
         }
+
+        /// Enables or disables Liquid Alpha for a given subnet.
+        ///
+        /// # Parameters
+        /// - `origin`: The origin of the call, which must be the root account or subnet owner.
+        /// - `netuid`: The unique identifier for the subnet.
+        /// - `enabled`: A boolean flag to enable or disable Liquid Alpha.
+        ///
+        /// # Weight
+        /// This function has a fixed weight of 0 and is classified as an operational transaction that does not incur any fees.
+        #[pallet::call_index(61)]
+        #[pallet::weight((0, DispatchClass::Operational, Pays::No))]
+        pub fn sudo_set_toggle_transfer(
+            origin: OriginFor<T>,
+            netuid: u16,
+            toggle: bool,
+        ) -> DispatchResult {
+            pallet_subtensor::Pallet::<T>::ensure_subnet_owner_or_root(origin, netuid)?;
+            pallet_subtensor::Pallet::<T>::toggle_transfer(netuid, toggle)
+        }
+
     }
 }
 

pallets/subtensor/src/lib.rs

@@ -1069,6 +1069,9 @@ pub mod pallet {
     /// ============================
     /// ==== Subnet Locks =====
     /// ============================
+    #[pallet::storage] // --- MAP ( netuid ) --> transfer_toggle
+    pub type TransferToggle<T: Config> =
+        StorageMap<_, Identity, u16, bool, ValueQuery, DefaultTrue<T>>;
     #[pallet::storage] // --- MAP ( netuid ) --> total_subnet_locked
     pub type SubnetLocked<T: Config> =
         StorageMap<_, Identity, u16, u64, ValueQuery, DefaultZeroU64<T>>;

pallets/subtensor/src/macros/errors.rs

@@ -189,5 +189,7 @@ mod errors {
         InsufficientLiquidity,
         /// Slippage is too high for the transaction.
         SlippageTooHigh,
+        /// Subnet disallows transfer.
+        TransferDisallowed,
     }
 }

pallets/subtensor/src/macros/events.rs

@@ -265,5 +265,12 @@ mod events {
         /// Parameters:
         /// (coldkey, hotkey, origin_netuid, destination_netuid, amount)
         StakeSwapped(T::AccountId, T::AccountId, u16, u16, u64),
+
+        /// Event called when transfer is toggled on a subnet.
+        ///
+        /// Parameters:
+        /// (netuid, bool)
+        TransferToggle(u16, bool),
+
     }
 }

pallets/subtensor/src/staking/move_stake.rs

@@ -96,6 +96,22 @@ impl<T: Config> Pallet<T> {
     ///
     /// # Events
     /// Emits a `StakeTransferred` event upon successful completion of the transfer.
+    pub fn toggle_transfer( 
+        netuid: u16, 
+        toggle: bool
+    ) -> dispatch::DispatchResult {
+        TransferToggle::<T>::insert( netuid, toggle );
+        log::debug!(
+            "TransferToggle( netuid: {:?}, toggle: {:?} ) ",
+            netuid,
+            toggle
+        );
+        Self::deposit_event(Event::TransferToggle(
+            netuid,
+            toggle
+        ));
+        Ok(())
+    }
     pub fn do_transfer_stake(
         origin: T::RuntimeOrigin,
         destination_coldkey: T::AccountId,
@@ -107,6 +123,16 @@ impl<T: Config> Pallet<T> {
         // Ensure the extrinsic is signed by the origin_coldkey.
         let coldkey = ensure_signed(origin)?;
 
+        // Ensure transfer is toggled.
+        ensure!(
+            TransferToggle::<T>::get( origin_netuid ),
+            Error::<T>::TransferDisallowed
+        );
+        ensure!(
+            TransferToggle::<T>::get( destination_netuid ),
+            Error::<T>::TransferDisallowed
+        );
+
         // Validate input and move stake
         let tao_moved = Self::transition_stake_internal(
             &coldkey,