git.delta.rocks / unique-network / refs/commits / d6ccebb78459

difftreelog

Add chain extension to mint

Greg Zaitsev2021-05-04parent: #b8ab366.patch.diff
in: master

2 files changed

modifiedpallets/nft/src/lib.rsdiffbeforeafterboth
--- a/pallets/nft/src/lib.rs
+++ b/pallets/nft/src/lib.rs
@@ -1114,16 +1114,8 @@
         #[weight = <T as Config>::WeightInfo::create_item(data.len())]
         #[transactional]
         pub fn create_item(origin, collection_id: CollectionId, owner: T::AccountId, data: CreateItemData) -> DispatchResult {
-
             let sender = ensure_signed(origin)?;
-
-            let target_collection = Self::get_collection(collection_id)?;
-
-            Self::can_create_items_in_collection(&target_collection, &sender, &owner, 1)?;
-            Self::validate_create_item_args(&target_collection, &data)?;
-            Self::create_item_no_validation(&target_collection, owner, data)?;
-
-            Ok(())
+            Self::create_item_internal(sender, collection_id, owner, data)
         }
 
         /// This method creates multiple items in a collection created with CreateCollection method.
@@ -1763,6 +1755,15 @@
 }
 
 impl<T: Config> Module<T> {
+    pub fn create_item_internal(sender: T::AccountId, collection_id: CollectionId, owner: T::AccountId, data: CreateItemData) -> DispatchResult {
+        let target_collection = Self::get_collection(collection_id)?;
+
+        Self::can_create_items_in_collection(&target_collection, &sender, &owner, 1)?;
+        Self::validate_create_item_args(&target_collection, &data)?;
+        Self::create_item_no_validation(&target_collection, owner, data)?;
+
+        Ok(())
+    }
 
     pub fn transfer_internal(sender: T::AccountId, recipient: T::AccountId, target_collection: &CollectionHandle<T>, item_id: TokenId, value: u128) -> DispatchResult {
         // Limits check
modifiedruntime/src/chain_extension.rsdiffbeforeafterboth
21use sp_runtime::AccountId32;21use sp_runtime::AccountId32;
22use crate::Vec;22use crate::Vec;
23
24/// Create item parameters
25#[derive(Debug, PartialEq, Encode, Decode)]
26pub struct NFTExtCreateItem<E: Ext> {
27 pub owner: <E::T as SysConfig>::AccountId,
28 pub collection_id: u32,
29 pub data: CreateItemData,
30}
2331
24/// Transfer parameters32/// Transfer parameters
25#[derive(Debug, PartialEq, Encode, Decode)]33#[derive(Debug, PartialEq, Encode, Decode)]
67 _ => Err(DispatchError::Other("Transfer error"))75 _ => Err(DispatchError::Other("Transfer error"))
68 }76 }
69 },77 },
78 1 => {
79 // Create Item
80 let mut env = env.buf_in_buf_out();
81 let input: NFTExtCreateItem<E> = env.read_as()?;
82
83 // Contract address to AccountId32
84 let mut bytes_contract: [u8; 32] = [0; 32];
85 let addr_vec_contract: Vec<u8> = env.ext().address().encode();
86 for i in 0..32 {
87 bytes_contract[i] = addr_vec_contract[i];
88 }
89 let sender = AccountId32::from(bytes_contract);
90
91 // Recipient to AccountId32
92 let mut bytes_rec: [u8; 32] = [0; 32];
93 let addr_vec_rec: Vec<u8> = input.owner.encode();
94 for i in 0..32 {
95 bytes_rec[i] = addr_vec_rec[i];
96 }
97 let recipient = AccountId32::from(bytes_rec);
98
99 // Parse data
100 let cdata = input.data.encode();
101 let vdata: [u8; 0] = [0; 0];
102 let item_data = CreateItemData::NFT(CreateNftData{
103 const_data: cdata,
104 variable_data: vdata.to_vec()
105 });
106
107 match pallet_nft::Module::<Runtime>::create_item_internal(sender, input.collection_id, recipient, item_data) {
108 Ok(_) => Ok(RetVal::Converging(func_id)),
109 _ => Err(DispatchError::Other("CreateItem error"))
110 }
111 },
70 _ => {112 _ => {
71 panic!("Passed unknown func_id to test chain extension: {}", func_id);113 panic!("Passed unknown func_id to test chain extension: {}", func_id);
72 }114 }