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

difftreelog

feat create_multiple_items chain extension

Yaroslav Bolyukin2021-05-06parent: #172f80a.patch.diff
in: master

2 files changed

modifiedpallets/nft/src/lib.rsdiffbeforeafterboth
--- a/pallets/nft/src/lib.rs
+++ b/pallets/nft/src/lib.rs
@@ -1792,6 +1792,23 @@
         Ok(())
     }
 
+    pub fn create_multiple_items_internal(
+        sender: T::AccountId,
+        collection: &CollectionHandle<T>,
+        owner: T::AccountId,
+        items_data: Vec<CreateItemData>,
+    ) -> DispatchResult {
+        Self::can_create_items_in_collection(&collection, &sender, &owner, items_data.len() as u32)?;
+
+        for data in &items_data {
+            Self::validate_create_item_args(&collection, data)?;
+        }
+        for data in &items_data {
+            Self::create_item_no_validation(&collection, owner.clone(), data.clone())?;
+        }
+
+        Ok(())
+    }
 
     fn is_correct_transfer(collection: &CollectionHandle<T>, recipient: &T::AccountId) -> DispatchResult {
         let collection_id = collection.id;
modifiedruntime/src/chain_extension.rsdiffbeforeafterboth
before · runtime/src/chain_extension.rs
1//! NFT Chain Extension23//4// This file is subject to the terms and conditions defined in5// file 'LICENSE', which is part of this source code package.6//78use codec::{Decode, Encode};910pub use pallet_contracts::chain_extension::RetVal;11use pallet_contracts::chain_extension::{12    ChainExtension, Environment, Ext, InitState, SysConfig, UncheckedFrom,13};1415pub use frame_support::debug;16use frame_support::dispatch::DispatchError;1718extern crate pallet_nft;19pub use pallet_nft::*;20use crate::Runtime;21use sp_runtime::AccountId32;22use crate::Vec;2324/// Create item parameters25#[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}3132/// Transfer parameters33#[derive(Debug, PartialEq, Encode, Decode)]34pub struct NFTExtTransfer<E: Ext> {35    pub recipient: <E::T as SysConfig>::AccountId,36    pub collection_id: u32,37    pub token_id: u32,38    pub amount: u128,39}4041/// The chain Extension of NFT pallet42pub struct NFTExtension;4344impl<C: Config> ChainExtension<C> for NFTExtension {45    fn call<E: Ext>(func_id: u32, env: Environment<E, InitState>) -> Result<RetVal, DispatchError>46    where47        E: Ext<T = C>,48        <E::T as SysConfig>::AccountId: UncheckedFrom<<E::T as SysConfig>::Hash> + AsRef<[u8]>,49    {50        // The memory of the vm stores buf in scale-codec51        match func_id {52            0 => {53                let mut env = env.buf_in_buf_out();54                let input: NFTExtTransfer<E> = env.read_as()?;5556                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;5758                match pallet_nft::Module::<C>::transfer_internal(59                    env.ext().caller().clone(),60                    input.recipient,61                    &collection,62                    input.token_id,63                    input.amount,64                ) {65                    Ok(_) => Ok(RetVal::Converging(func_id)),66                    _ => Err(DispatchError::Other("Transfer error"))67                }68            },69            1 => {70                // Create Item71                let mut env = env.buf_in_buf_out();72                let input: NFTExtCreateItem<E> = env.read_as()?;7374                match pallet_nft::Module::<C>::create_item_internal(75                    env.ext().address().clone(),76                    input.collection_id,77                    input.owner,78                    input.data,79                ) {80                    Ok(_) => Ok(RetVal::Converging(func_id)),81                    _ => Err(DispatchError::Other("CreateItem error"))82                }83            },84            _ => {85                panic!("Passed unknown func_id to test chain extension: {}", func_id);86            }87        }88    }89}