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
1117
1118 let sender = ensure_signed(origin)?;1117 let sender = ensure_signed(origin)?;
1119
1120 let target_collection = Self::get_collection(collection_id)?;1118 Self::create_item_internal(sender, collection_id, owner, data)
1121
1122 Self::can_create_items_in_collection(&target_collection, &sender, &owner, 1)?;
1123 Self::validate_create_item_args(&target_collection, &data)?;
1124 Self::create_item_no_validation(&target_collection, owner, data)?;
1125
1126 Ok(())
1127 }1119 }
11281120
1129 /// This method creates multiple items in a collection created with CreateCollection method.1121 /// This method creates multiple items in a collection created with CreateCollection method.
1763}1755}
17641756
1765impl<T: Config> Module<T> {1757impl<T: Config> Module<T> {
1758 pub fn create_item_internal(sender: T::AccountId, collection_id: CollectionId, owner: T::AccountId, data: CreateItemData) -> DispatchResult {
1759 let target_collection = Self::get_collection(collection_id)?;
1760
1761 Self::can_create_items_in_collection(&target_collection, &sender, &owner, 1)?;
1762 Self::validate_create_item_args(&target_collection, &data)?;
1763 Self::create_item_no_validation(&target_collection, owner, data)?;
1764
1765 Ok(())
1766 }
17661767
1767 pub fn transfer_internal(sender: T::AccountId, recipient: T::AccountId, target_collection: &CollectionHandle<T>, item_id: TokenId, value: u128) -> DispatchResult {1768 pub fn transfer_internal(sender: T::AccountId, recipient: T::AccountId, target_collection: &CollectionHandle<T>, item_id: TokenId, value: u128) -> DispatchResult {
1768 // Limits check1769 // Limits check
modifiedruntime/src/chain_extension.rsdiffbeforeafterboth
--- a/runtime/src/chain_extension.rs
+++ b/runtime/src/chain_extension.rs
@@ -21,6 +21,14 @@
 use sp_runtime::AccountId32;
 use crate::Vec;
 
+/// Create item parameters
+#[derive(Debug, PartialEq, Encode, Decode)]
+pub struct NFTExtCreateItem<E: Ext> {
+    pub owner: <E::T as SysConfig>::AccountId,
+    pub collection_id: u32,
+    pub data: CreateItemData,
+}
+
 /// Transfer parameters
 #[derive(Debug, PartialEq, Encode, Decode)]
 pub struct NFTExtTransfer<E: Ext> {
@@ -67,8 +75,42 @@
                     _ => Err(DispatchError::Other("Transfer error"))
                 }
             },
-			_ => {
-				panic!("Passed unknown func_id to test chain extension: {}", func_id);
+            1 => {
+                // Create Item
+                let mut env = env.buf_in_buf_out();
+                let input: NFTExtCreateItem<E> = env.read_as()?;
+
+                // Contract address to AccountId32
+                let mut bytes_contract: [u8; 32] = [0; 32];
+                let addr_vec_contract: Vec<u8> = env.ext().address().encode();
+                for i in 0..32 {
+                    bytes_contract[i] = addr_vec_contract[i];
+                }
+                let sender = AccountId32::from(bytes_contract);
+
+                // Recipient to AccountId32
+                let mut bytes_rec: [u8; 32] = [0; 32];
+                let addr_vec_rec: Vec<u8> = input.owner.encode();
+                for i in 0..32 {
+                    bytes_rec[i] = addr_vec_rec[i];
+                }
+                let recipient = AccountId32::from(bytes_rec);
+
+                // Parse data
+                let cdata = input.data.encode();
+                let vdata: [u8; 0] = [0; 0];
+                let item_data = CreateItemData::NFT(CreateNftData{
+                    const_data: cdata,
+                    variable_data: vdata.to_vec()
+                });
+
+                match pallet_nft::Module::<Runtime>::create_item_internal(sender, input.collection_id, recipient, item_data) {
+                    Ok(_) => Ok(RetVal::Converging(func_id)),
+                    _ => Err(DispatchError::Other("CreateItem error"))
+                }
+            },
+            _ => {
+                panic!("Passed unknown func_id to test chain extension: {}", func_id);
             }
         }
     }