Merge pull request #8 from uni-arts-chain/dev

admin transfer
uni-arts-chain · Mar 17, 2022 · 5eee455 · 5eee455
2 parents a03e224 + ded537e
commit 5eee455
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/pallets/nft-multi/src/default_weight.rs b/pallets/nft-multi/src/default_weight.rs
@@ -81,6 +81,11 @@ impl crate::WeightInfo for () {
             .saturating_add(DbWeight::get().reads(5 as Weight))
             .saturating_add(DbWeight::get().writes(3 as Weight))
     }
+	fn admin_transfer() -> Weight {
+		(66_234_000 as Weight)
+			.saturating_add(DbWeight::get().reads(5 as Weight))
+			.saturating_add(DbWeight::get().writes(3 as Weight))
+	}
 	fn transfer_and_lock() -> Weight {
 		(66_234_000 as Weight)
 			.saturating_add(DbWeight::get().reads(5 as Weight))

diff --git a/pallets/nft-multi/src/lib.rs b/pallets/nft-multi/src/lib.rs
@@ -59,6 +59,7 @@ pub trait WeightInfo {
     fn create_item() -> Weight;
     fn burn_item() -> Weight;
     fn transfer() -> Weight;
+	fn admin_transfer() -> Weight;
 	fn transfer_and_lock() -> Weight;
 	fn batch_create_nft_item() -> Weight;
 	fn unlock() -> Weight;
@@ -410,6 +411,7 @@ decl_event!(
         ItemCreated(u64, u64),
         ItemDestroyed(  u64, u64),
         ItemTransfer(u64, u64, u64, AccountId, AccountId),
+		AdminTransfer(u64, u64, u64, AccountId, AccountId),
 		ItemLock(u64, u64, u64, AccountId),
 		ItemUnlock(u64, u64, u64, AccountId),
         ItemOrderCreated(u64, u64, u64, u64, AccountId, u64, CurrencyId),
@@ -933,6 +935,27 @@ decl_module! {
             // call event
             Self::deposit_event(RawEvent::ItemTransfer(collection_id, item_id, value, sender, recipient));
 
+            Ok(())
+        }
+
+		#[weight = T::WeightInfo::admin_transfer()]
+        pub fn admin_transfer(origin, from: T::AccountId, to: T::AccountId, collection_id: u64, item_id: u64, value: u64) -> DispatchResult {
+
+            let sender = ensure_signed(origin)?;
+
+            Self::check_admin_permissions(collection_id, sender.clone())?;
+
+            let target_collection = <Collection<T>>::get(collection_id);
+
+			if let CollectionMode::ReFungible(_, _) = target_collection.mode {
+				Self::transfer_refungible(collection_id, item_id, value, from.clone(), to.clone())?;
+				// call event
+				Self::deposit_event(RawEvent::ItemTransfer(collection_id, item_id, value, from.clone(), to.clone()));
+				Self::deposit_event(RawEvent::AdminTransfer(collection_id, item_id, value, from.clone(), to.clone()));
+			} else {
+				panic!("Collection is not ReFungible mode");
+			}
+
             Ok(())
         }