`0a7b60da8a0b`

source

runtime/src/chain_extension.rs7.2 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                pallet_nft::Module::<C>::transfer_internal(99                    env.ext().address().clone(),100                    input.recipient,101                    &collection,102                    input.token_id,103                    input.amount,104                )?;105                106                Ok(RetVal::Converging(0))107            },108            1 => {109                // Create Item110                let mut env = env.buf_in_buf_out();111                let input: NFTExtCreateItem<E> = env.read_as()?;112                env.charge_weight(NftWeightInfoOf::<C>::create_item(input.data.len()))?;113114                pallet_nft::Module::<C>::create_item_internal(115                    env.ext().address().clone(),116                    input.collection_id,117                    input.owner,118                    input.data,119                )?;120                121                Ok(RetVal::Converging(0))122            },123            2 => {124                // Create multiple items125                let mut env = env.buf_in_buf_out();126                let input: NFTExtCreateMultipleItems<E> = env.read_as()?;127                env.charge_weight(NftWeightInfoOf::<C>::create_item(128                    input.data.iter()129                        .map(|i| i.len())130                        .sum()131                ))?;132133                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;134135                pallet_nft::Module::<C>::create_multiple_items_internal(136                    env.ext().address().clone(),137                    &collection,138                    input.owner,139                    input.data,140                )?;141                142                Ok(RetVal::Converging(0))143            },144            3 => {145                // Approve146                let mut env = env.buf_in_buf_out();147                let input: NFTExtApprove<E> = env.read_as()?;148                env.charge_weight(NftWeightInfoOf::<C>::approve())?;149150                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;151152                pallet_nft::Module::<C>::approve_internal(153                    env.ext().address().clone(),154                    input.spender,155                    &collection,156                    input.item_id,157                    input.amount,158                )?;159160                Ok(RetVal::Converging(0))161            },162            4 => {163                // Transfer from164                let mut env = env.buf_in_buf_out();165                let input: NFTExtTransferFrom<E> = env.read_as()?;166                env.charge_weight(NftWeightInfoOf::<C>::transfer_from())?;167168                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;169170                pallet_nft::Module::<C>::transfer_from_internal(171                    env.ext().address().clone(),172                    input.owner,173                    input.recipient,174                    &collection,175                    input.item_id,176                    input.amount177                )?;178179                Ok(RetVal::Converging(0))180            },181            5 => {182                // Set variable metadata183                let mut env = env.buf_in_buf_out();184                let input: NFTExtSetVariableMetaData = env.read_as()?;185                env.charge_weight(NftWeightInfoOf::<C>::set_variable_meta_data())?;186187                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;188189                pallet_nft::Module::<C>::set_variable_meta_data_internal(190                    env.ext().address().clone(),191                    &collection,192                    input.item_id,193                    input.data,194                )?;195196                Ok(RetVal::Converging(0))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                )?;212213                Ok(RetVal::Converging(0))214            }215            _ => {216                Err(DispatchError::Other("unknown chain_extension func_id"))217            }218        }219    }220}