Update clear_identity to allow for identified argument

opentensor · Oct 31, 2023 · 4de469a · 4de469a
1 parent e66888d
commit 4de469a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pallets/registry/src/lib.rs b/pallets/registry/src/lib.rs
@@ -130,16 +130,17 @@ pub mod pallet {
 
 		#[pallet::call_index(1)]
 		#[pallet::weight(T::WeightInfo::clear_identity())]
-		pub fn clear_identity(origin: OriginFor<T>) -> DispatchResultWithPostInfo {
+		pub fn clear_identity(origin: OriginFor<T>, identified: T::AccountId) -> DispatchResultWithPostInfo {
 			let who = ensure_signed(origin)?;
+			ensure!(T::CanRegister::can_register(&who, &identified), Error::<T>::CannotRegister);
 
-			let id = <IdentityOf<T>>::take(&who).ok_or(Error::<T>::NotRegistered)?;
+			let id = <IdentityOf<T>>::take(&identified).ok_or(Error::<T>::NotRegistered)?;
 			let deposit = id.total_deposit();
 
 			let err_amount = T::Currency::unreserve(&who, deposit);
 			debug_assert!(err_amount.is_zero());
 
-			Self::deposit_event(Event::IdentityDissolved { who });
+			Self::deposit_event(Event::IdentityDissolved { who: identified });
 
 			Ok(().into())
 		}