`35f0b49c19be`

source

runtime/src/chain_extension.rs8.1 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 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                )?;221222                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}