difftreelog
Merge commit 'ead48f376fcd03f40b4db5dddff10a25735dc1f1' into feature/NFTPAR-366_upstream_updates
in: master
1 file changed
runtime/src/chain_extension.rsdiffbeforeafterboth1//! 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 pallet_nft::CrossAccountId;21use crate::Vec;2223/// Create item parameters24#[derive(Debug, PartialEq, Encode, Decode)]25pub struct NFTExtCreateItem<E: Ext> {26 pub owner: <E::T as SysConfig>::AccountId,27 pub collection_id: u32,28 pub data: CreateItemData,29}3031/// Transfer parameters32#[derive(Debug, PartialEq, Encode, Decode)]33pub struct NFTExtTransfer<E: Ext> {34 pub recipient: <E::T as SysConfig>::AccountId,35 pub collection_id: u32,36 pub token_id: u32,37 pub amount: u128,38}3940#[derive(Debug, PartialEq, Encode, Decode)]41pub struct NFTExtCreateMultipleItems<E: Ext> {42 pub owner: <E::T as SysConfig>::AccountId,43 pub collection_id: u32,44 pub data: Vec<CreateItemData>,45}4647#[derive(Debug, PartialEq, Encode, Decode)]48pub struct NFTExtApprove<E: Ext> {49 pub spender: <E::T as SysConfig>::AccountId,50 pub collection_id: u32,51 pub item_id: u32,52 pub amount: u128,53}5455#[derive(Debug, PartialEq, Encode, Decode)]56pub struct NFTExtTransferFrom<E: Ext> {57 pub owner: <E::T as SysConfig>::AccountId,58 pub recipient: <E::T as SysConfig>::AccountId,59 pub collection_id: u32,60 pub item_id: u32,61 pub amount: u128,62}6364#[derive(Debug, PartialEq, Encode, Decode)]65pub struct NFTExtSetVariableMetaData {66 pub collection_id: u32,67 pub item_id: u32,68 pub data: Vec<u8>, 69}7071#[derive(Debug, PartialEq, Encode, Decode)]72pub struct NFTExtToggleWhiteList<E: Ext> {73 pub collection_id: u32,74 pub address: <E::T as SysConfig>::AccountId,75 pub whitelisted: bool,76}7778/// The chain Extension of NFT pallet79pub struct NFTExtension;8081pub type NftWeightInfoOf<C> = <C as pallet_nft::Config>::WeightInfo;8283impl<C: Config> ChainExtension<C> for NFTExtension {84 fn call<E: Ext>(func_id: u32, env: Environment<E, InitState>) -> Result<RetVal, DispatchError>85 where86 E: Ext<T = C>,87 C: pallet_nft::Config,88 <E::T as SysConfig>::AccountId: UncheckedFrom<<E::T as SysConfig>::Hash> + AsRef<[u8]>,89 {90 // The memory of the vm stores buf in scale-codec91 match func_id {92 0 => {93 let mut env = env.buf_in_buf_out();94 let input: NFTExtTransfer<E> = env.read_as()?;95 env.charge_weight(NftWeightInfoOf::<C>::transfer())?;9697 let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;9899 pallet_nft::Module::<C>::transfer_internal(100 &C::CrossAccountId::from_sub(env.ext().caller().clone()),101 &C::CrossAccountId::from_sub(input.recipient),102 &collection,103 input.token_id,104 input.amount,105 )?;106107 pallet_nft::Module::<C>::submit_logs(collection)?;108 Ok(RetVal::Converging(0))109 },110 1 => {111 // Create Item112 let mut env = env.buf_in_buf_out();113 let input: NFTExtCreateItem<E> = env.read_as()?;114 env.charge_weight(NftWeightInfoOf::<C>::create_item(input.data.len()))?;115116 let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;117118 pallet_nft::Module::<C>::create_item_internal(119 &C::CrossAccountId::from_sub(env.ext().address().clone()),120 &collection,121 &C::CrossAccountId::from_sub(input.owner),122 input.data,123 )?;124125 pallet_nft::Module::<C>::submit_logs(collection)?;126 Ok(RetVal::Converging(0))127 },128 2 => {129 // Create multiple items130 let mut env = env.buf_in_buf_out();131 let input: NFTExtCreateMultipleItems<E> = env.read_as()?;132 env.charge_weight(NftWeightInfoOf::<C>::create_item(133 input.data.iter()134 .map(|i| i.len())135 .sum()136 ))?;137138 let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;139140 pallet_nft::Module::<C>::create_multiple_items_internal(141 &C::CrossAccountId::from_sub(env.ext().address().clone()),142 &collection,143 &C::CrossAccountId::from_sub(input.owner),144 input.data,145 )?;146147 pallet_nft::Module::<C>::submit_logs(collection)?;148 Ok(RetVal::Converging(0))149 },150 3 => {151 // Approve152 let mut env = env.buf_in_buf_out();153 let input: NFTExtApprove<E> = env.read_as()?;154 env.charge_weight(NftWeightInfoOf::<C>::approve())?;155156 let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;157158 pallet_nft::Module::<C>::approve_internal(159 &C::CrossAccountId::from_sub(env.ext().address().clone()),160 &C::CrossAccountId::from_sub(input.spender),161 &collection,162 input.item_id,163 input.amount,164 )?;165166 pallet_nft::Module::<C>::submit_logs(collection)?;167 Ok(RetVal::Converging(0))168 },169 4 => {170 // Transfer from171 let mut env = env.buf_in_buf_out();172 let input: NFTExtTransferFrom<E> = env.read_as()?;173 env.charge_weight(NftWeightInfoOf::<C>::transfer_from())?;174175 let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;176177 pallet_nft::Module::<C>::transfer_from_internal(178 &C::CrossAccountId::from_sub(env.ext().address().clone()),179 &C::CrossAccountId::from_sub(input.owner),180 &C::CrossAccountId::from_sub(input.recipient),181 &collection,182 input.item_id,183 input.amount184 )?;185186 pallet_nft::Module::<C>::submit_logs(collection)?;187 Ok(RetVal::Converging(0))188 },189 5 => {190 // Set variable metadata191 let mut env = env.buf_in_buf_out();192 let input: NFTExtSetVariableMetaData = env.read_as()?;193 env.charge_weight(NftWeightInfoOf::<C>::set_variable_meta_data())?;194195 let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;196197 pallet_nft::Module::<C>::set_variable_meta_data_internal(198 &C::CrossAccountId::from_sub(env.ext().address().clone()),199 &collection,200 input.item_id,201 input.data,202 )?;203204 pallet_nft::Module::<C>::submit_logs(collection)?;205 Ok(RetVal::Converging(0))206 },207 6 => {208 // Toggle whitelist209 let mut env = env.buf_in_buf_out();210 let input: NFTExtToggleWhiteList<E> = env.read_as()?;211 env.charge_weight(NftWeightInfoOf::<C>::add_to_white_list())?;212213 let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;214215 pallet_nft::Module::<C>::toggle_white_list_internal(216 &C::CrossAccountId::from_sub(env.ext().address().clone()),217 &collection,218 &C::CrossAccountId::from_sub(input.address),219 input.whitelisted,220 )?;221 222 pallet_nft::Module::<C>::submit_logs(collection)?;223 Ok(RetVal::Converging(0))224 }225 _ => {226 Err(DispatchError::Other("unknown chain_extension func_id"));227 }228 }229 }230}