`c68888f211f1`

source

runtime/src/chain_extension.rs7.5 KiBsource history
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::Vec;2122/// Create item parameters23#[derive(Debug, PartialEq, Encode, Decode)]24pub struct NFTExtCreateItem<E: Ext> {25    pub owner: <E::T as SysConfig>::AccountId,26    pub collection_id: u32,27    pub data: CreateItemData,28}2930/// Transfer parameters31#[derive(Debug, PartialEq, Encode, Decode)]32pub struct NFTExtTransfer<E: Ext> {33    pub recipient: <E::T as SysConfig>::AccountId,34    pub collection_id: u32,35    pub token_id: u32,36    pub amount: u128,37}3839#[derive(Debug, PartialEq, Encode, Decode)]40pub struct NFTExtCreateMultipleItems<E: Ext> {41    pub owner: <E::T as SysConfig>::AccountId,42    pub collection_id: u32,43    pub data: Vec<CreateItemData>,44}4546#[derive(Debug, PartialEq, Encode, Decode)]47pub struct NFTExtApprove<E: Ext> {48    pub spender: <E::T as SysConfig>::AccountId,49    pub collection_id: u32,50    pub item_id: u32,51    pub amount: u128,52}5354#[derive(Debug, PartialEq, Encode, Decode)]55pub struct NFTExtTransferFrom<E: Ext> {56    pub owner: <E::T as SysConfig>::AccountId,57    pub recipient: <E::T as SysConfig>::AccountId,58    pub collection_id: u32,59    pub item_id: u32,60    pub amount: u128,61}6263#[derive(Debug, PartialEq, Encode, Decode)]64pub struct NFTExtSetVariableMetaData {65    pub collection_id: u32,66    pub item_id: u32,67    pub data: Vec<u8>,   68}6970#[derive(Debug, PartialEq, Encode, Decode)]71pub struct NFTExtToggleWhiteList<E: Ext> {72    pub collection_id: u32,73    pub address: <E::T as SysConfig>::AccountId,74    pub whitelisted: bool,75}7677/// The chain Extension of NFT pallet78pub struct NFTExtension;7980pub type NftWeightInfoOf<C> = <C as pallet_nft::Config>::WeightInfo;8182impl<C: Config> ChainExtension<C> for NFTExtension {83    fn call<E: Ext>(func_id: u32, env: Environment<E, InitState>) -> Result<RetVal, DispatchError>84    where85        E: Ext<T = C>,86        C: pallet_nft::Config,87        <E::T as SysConfig>::AccountId: UncheckedFrom<<E::T as SysConfig>::Hash> + AsRef<[u8]>,88    {89        // The memory of the vm stores buf in scale-codec90        match func_id {91            0 => {92                let mut env = env.buf_in_buf_out();93                let input: NFTExtTransfer<E> = env.read_as()?;94                env.charge_weight(NftWeightInfoOf::<C>::transfer())?;9596                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;9798                match pallet_nft::Module::<C>::transfer_internal(99                    env.ext().caller().clone(),100                    input.recipient,101                    &collection,102                    input.token_id,103                    input.amount,104                ) {105                    Ok(_) => Ok(RetVal::Converging(func_id)),106                    _ => Err(DispatchError::Other("Transfer error"))107                }108            },109            1 => {110                // Create Item111                let mut env = env.buf_in_buf_out();112                let input: NFTExtCreateItem<E> = env.read_as()?;113                env.charge_weight(NftWeightInfoOf::<C>::create_item(input.data.len()))?;114115                match pallet_nft::Module::<C>::create_item_internal(116                    env.ext().address().clone(),117                    input.collection_id,118                    input.owner,119                    input.data,120                ) {121                    Ok(_) => Ok(RetVal::Converging(func_id)),122                    _ => Err(DispatchError::Other("CreateItem error"))123                }124            },125            2 => {126                // Create multiple items127                let mut env = env.buf_in_buf_out();128                let input: NFTExtCreateMultipleItems<E> = env.read_as()?;129                env.charge_weight(NftWeightInfoOf::<C>::create_item(130                    input.data.iter()131                        .map(|i| i.len())132                        .sum()133                ))?;134135                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;136137                match pallet_nft::Module::<C>::create_multiple_items_internal(138                    env.ext().address().clone(),139                    &collection,140                    input.owner,141                    input.data,142                ) {143                    Ok(_) => Ok(RetVal::Converging(func_id)),144                    _ => Err(DispatchError::Other("CreateMultipleItems error"))145                }146            },147            3 => {148                // Approve149                let mut env = env.buf_in_buf_out();150                let input: NFTExtApprove<E> = env.read_as()?;151                env.charge_weight(NftWeightInfoOf::<C>::approve())?;152153                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;154155                pallet_nft::Module::<C>::approve_internal(156                    env.ext().address().clone(),157                    input.spender,158                    &collection,159                    input.item_id,160                    input.amount,161                )?;162                Ok(RetVal::Converging(func_id))163            },164            4 => {165                // Transfer from166                let mut env = env.buf_in_buf_out();167                let input: NFTExtTransferFrom<E> = env.read_as()?;168                env.charge_weight(NftWeightInfoOf::<C>::transfer_from())?;169170                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;171172                pallet_nft::Module::<C>::transfer_from_internal(173                    env.ext().address().clone(),174                    input.owner,175                    input.recipient,176                    &collection,177                    input.item_id,178                    input.amount179                )?;180                Ok(RetVal::Converging(func_id))181            },182            5 => {183                // Set variable metadata184                let mut env = env.buf_in_buf_out();185                let input: NFTExtSetVariableMetaData = env.read_as()?;186                env.charge_weight(NftWeightInfoOf::<C>::set_variable_meta_data())?;187188                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;189190                pallet_nft::Module::<C>::set_variable_meta_data_internal(191                    env.ext().address().clone(),192                    &collection,193                    input.item_id,194                    input.data,195                )?;196                Ok(RetVal::Converging(func_id))197            },198            6 => {199                // Toggle whitelist200                let mut env = env.buf_in_buf_out();201                let input: NFTExtToggleWhiteList<E> = env.read_as()?;202                env.charge_weight(NftWeightInfoOf::<C>::add_to_white_list())?;203204                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;205206                pallet_nft::Module::<C>::toggle_white_list_internal(207                    &env.ext().address().clone(),208                    &collection,209                    &input.address,210                    input.whitelisted,211                )?;212                Ok(RetVal::Converging(func_id))213            }214            _ => {215                panic!("Passed unknown func_id to test chain extension: {}", func_id);216            }217        }218    }219}