toggle_user_liquidity

Author: JohnReedV

pallets/subtensor/src/subnets/subnet.rs

@@ -4,17 +4,6 @@ use subtensor_runtime_common::{NetUid, TaoCurrency};
 use subtensor_swap_interface::SwapHandler;
 
 impl<T: Config> Pallet<T> {
-    /// Fetches the total count of subnets.
-    ///
-    /// This function retrieves the total number of subnets present on the chain.
-    ///
-    /// # Returns:
-    /// * 'u16': The total number of subnets.
-    ///
-    pub fn get_num_subnets() -> u16 {
-        TotalNetworks::<T>::get()
-    }
-
     /// Returns true if the subnetwork exists.
     ///
     /// This function checks if a subnetwork with the given UID exists.
@@ -256,8 +245,7 @@ impl<T: Config> Pallet<T> {
             Self::deposit_event(Event::SubnetIdentitySet(netuid_to_register));
         }
 
-        T::SwapInterface::try_initialize_v3(netuid_to_register)?;
-
+        T::SwapInterface::toggle_user_liquidity(netuid_to_register, true);
         // --- 18. Emit the NetworkAdded event.
         log::info!("NetworkAdded( netuid:{netuid_to_register:?}, mechanism:{mechid:?} )");
         Self::deposit_event(Event::NetworkAdded(netuid_to_register, mechid));

pallets/swap-interface/src/lib.rs

@@ -35,7 +35,7 @@ pub trait SwapHandler<AccountId> {
     );
     fn is_user_liquidity_enabled(netuid: NetUid) -> bool;
     fn dissolve_all_liquidity_providers(netuid: NetUid) -> DispatchResult;
-    fn try_initialize_v3(netuid: NetUid) -> DispatchResult;
+    fn toggle_user_liquidity(netuid: NetUid, enabled: bool);
 }
 
 #[derive(Debug, PartialEq)]

pallets/swap/src/pallet/impls.rs

@@ -1360,14 +1360,10 @@ impl<T: Config> Pallet<T> {
     /// - **V2 / non‑V3 path**:
     ///   * No per‑position records exist; still defensively clear the same V3 storages (safe no‑ops).
     pub fn do_dissolve_all_liquidity_providers(netuid: NetUid) -> DispatchResult {
-        let mechid = T::SubnetInfo::mechanism(netuid.into());
-        let v3_initialized = SwapV3Initialized::<T>::get(netuid);
         let user_lp_enabled =
         <Self as subtensor_swap_interface::SwapHandler<T::AccountId>>::is_user_liquidity_enabled(netuid);
 
-        let is_v3_mode = mechid == 1 && v3_initialized;
-
-        if is_v3_mode {
+        if SwapV3Initialized::<T>::get(netuid) {
             // -------- V3: close every position, aggregate refunds, clear state --------
 
             // 1) Snapshot all (owner, position_id).
@@ -1575,8 +1571,8 @@ impl<T: Config> SwapHandler<T::AccountId> for Pallet<T> {
     fn dissolve_all_liquidity_providers(netuid: NetUid) -> DispatchResult {
         Self::do_dissolve_all_liquidity_providers(netuid)
     }
-    fn try_initialize_v3(netuid: NetUid) -> DispatchResult {
-        Self::maybe_initialize_v3(netuid).map_err(|e| e.into())
+    fn toggle_user_liquidity(netuid: NetUid, enabled: bool) {
+        EnabledUserLiquidity::<T>::insert(netuid, enabled)
     }
 }