feat: add curator agent freezing toggling (#109)

Closes CHAIN-100

Author: devwckd

pallets/emission0/src/weights.rs

@@ -2,9 +2,9 @@
 //! Autogenerated weights for `pallet_emission0`
 //!
 //! THIS FILE WAS AUTO-GENERATED USING THE SUBSTRATE BENCHMARK CLI VERSION 43.0.0
-//! DATE: 2025-06-10, STEPS: `50`, REPEAT: `20`, LOW RANGE: `[]`, HIGH RANGE: `[]`
+//! DATE: 2025-06-18, STEPS: `50`, REPEAT: `20`, LOW RANGE: `[]`, HIGH RANGE: `[]`
 //! WORST CASE MAP SIZE: `1000000`
-//! HOSTNAME: `Luizs-MacBook-Pro.local`, CPU: `<UNKNOWN>`
+//! HOSTNAME: `MacBook-Pro-de-Joao.local`, CPU: `<UNKNOWN>`
 //! WASM-EXECUTION: `Compiled`, CHAIN: `Some("dev")`, DB CACHE: `1024`
 
 // Executed Command:
@@ -94,7 +94,7 @@ impl<T: frame_system::Config> WeightInfo for SubstrateWeight<T> {
 		// Proof Size summary in bytes:
 		//  Measured:  `333`
 		//  Estimated: `3529`
-		// Minimum execution time: 12_000_000 picoseconds.
+		// Minimum execution time: 11_000_000 picoseconds.
 		Weight::from_parts(12_000_000, 3529)
 			.saturating_add(T::DbWeight::get().reads(2_u64))
 			.saturating_add(T::DbWeight::get().writes(1_u64))
@@ -151,7 +151,7 @@ impl WeightInfo for () {
 		// Proof Size summary in bytes:
 		//  Measured:  `333`
 		//  Estimated: `3529`
-		// Minimum execution time: 12_000_000 picoseconds.
+		// Minimum execution time: 11_000_000 picoseconds.
 		Weight::from_parts(12_000_000, 3529)
 			.saturating_add(RocksDbWeight::get().reads(2_u64))
 			.saturating_add(RocksDbWeight::get().writes(1_u64))

pallets/governance/api/src/lib.rs

@@ -15,6 +15,10 @@ pub trait GovernanceApi<AccountId> {
 
     fn set_allocator(key: &AccountId);
 
+    fn can_create_namespace(key: &AccountId) -> bool;
+
+    fn can_register_agent(key: &AccountId) -> bool;
+
     #[doc(hidden)]
     #[cfg(feature = "runtime-benchmarks")]
     fn force_set_whitelisted(key: &AccountId);

pallets/governance/src/benchmarking.rs

@@ -266,4 +266,16 @@ mod benchmarks {
             data,
         )
     }
+
+    #[benchmark]
+    fn toggle_agent_freezing() {
+        #[extrinsic_call]
+        toggle_agent_freezing(RawOrigin::Signed(curator::<T>()))
+    }
+
+    #[benchmark]
+    fn toggle_namespace_freezing() {
+        #[extrinsic_call]
+        toggle_namespace_freezing(RawOrigin::Signed(curator::<T>()))
+    }
 }

pallets/governance/src/lib.rs

@@ -44,6 +44,7 @@ pub mod pallet {
     const STORAGE_VERSION: StorageVersion = StorageVersion::new(5);
 
     use pallet_permission0_api::{CuratorPermissions, Permission0Api, Permission0CuratorApi};
+    use polkadot_sdk::sp_core::ConstBool;
     use proposal::GlobalParamsData;
     use weights::WeightInfo;
 
@@ -101,6 +102,14 @@ pub mod pallet {
     pub type TreasuryEmissionFee<T: Config> =
         StorageValue<_, Percent, ValueQuery, T::DefaultTreasuryEmissionFee>;
 
+    /// Determines if new agents can be registered on the chain.
+    #[pallet::storage]
+    pub type AgentsFrozen<T: Config> = StorageValue<_, bool, ValueQuery, ConstBool<false>>;
+
+    /// Determines if new namespaces can be created on the chain.
+    #[pallet::storage]
+    pub type NamespacesFrozen<T: Config> = StorageValue<_, bool, ValueQuery, ConstBool<false>>;
+
     #[pallet::config]
     pub trait Config:
         polkadot_sdk::frame_system::Config + pallet_torus0::Config + pallet_emission0::Config
@@ -378,6 +387,44 @@ pub mod pallet {
 
             Ok(())
         }
+
+        #[pallet::call_index(19)]
+        #[pallet::weight((<T as Config>::WeightInfo::toggle_agent_freezing(), DispatchClass::Normal, Pays::No))]
+        pub fn toggle_agent_freezing(origin: OriginFor<T>) -> DispatchResult {
+            let curator = <T as pallet::Config>::Permission0::ensure_curator_permission(
+                origin,
+                CuratorPermissions::AGENT_FREEZING_TOGGLING,
+            )?;
+
+            let new_state = !crate::AgentsFrozen::<T>::get();
+            AgentsFrozen::<T>::set(new_state);
+
+            crate::Pallet::<T>::deposit_event(crate::Event::AgentFreezingToggled {
+                curator,
+                new_state,
+            });
+
+            Ok(())
+        }
+
+        #[pallet::call_index(20)]
+        #[pallet::weight((<T as Config>::WeightInfo::toggle_namespace_freezing(), DispatchClass::Normal, Pays::No))]
+        pub fn toggle_namespace_freezing(origin: OriginFor<T>) -> DispatchResult {
+            let curator = <T as pallet::Config>::Permission0::ensure_curator_permission(
+                origin,
+                CuratorPermissions::NAMESPACE_FREEZING_TOGGLING,
+            )?;
+
+            let new_state = !crate::NamespacesFrozen::<T>::get();
+            NamespacesFrozen::<T>::set(new_state);
+
+            crate::Pallet::<T>::deposit_event(crate::Event::NamespaceFreezingToggled {
+                curator,
+                new_state,
+            });
+
+            Ok(())
+        }
     }
 
     #[pallet::event]
@@ -413,6 +460,16 @@ pub mod pallet {
             agent: T::AccountId,
             penalty: Percent,
         },
+        /// The agent freezing feature was toggled by a curator.
+        AgentFreezingToggled {
+            curator: T::AccountId,
+            new_state: bool,
+        },
+        /// The namespace freezing feature was toggled by a curator.
+        NamespaceFreezingToggled {
+            curator: T::AccountId,
+            new_state: bool,
+        },
     }
 
     #[pallet::error]
@@ -529,6 +586,14 @@ impl<T: Config> pallet_governance_api::GovernanceApi<T::AccountId> for Pallet<T>
         Allocators::<T>::insert(key, ());
     }
 
+    fn can_create_namespace(key: &T::AccountId) -> bool {
+        !NamespacesFrozen::<T>::get() || Self::is_whitelisted(key)
+    }
+
+    fn can_register_agent(key: &T::AccountId) -> bool {
+        !AgentsFrozen::<T>::get() || Self::is_whitelisted(key)
+    }
+
     #[cfg(feature = "runtime-benchmarks")]
     fn force_set_whitelisted(key: &T::AccountId) {
         Whitelist::<T>::insert(key, ());