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

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;7980impl<C: Config> ChainExtension<C> for NFTExtension {81    fn call<E: Ext>(func_id: u32, env: Environment<E, InitState>) -> Result<RetVal, DispatchError>82    where83        E: Ext<T = C>,84        <E::T as SysConfig>::AccountId: UncheckedFrom<<E::T as SysConfig>::Hash> + AsRef<[u8]>,85    {86        // The memory of the vm stores buf in scale-codec87        match func_id {88            0 => {89                let mut env = env.buf_in_buf_out();90                let input: NFTExtTransfer<E> = env.read_as()?;9192                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;9394                match pallet_nft::Module::<C>::transfer_internal(95                    env.ext().caller().clone(),96                    input.recipient,97                    &collection,98                    input.token_id,99                    input.amount,100                ) {101                    Ok(_) => Ok(RetVal::Converging(func_id)),102                    _ => Err(DispatchError::Other("Transfer error"))103                }104            },105            1 => {106                // Create Item107                let mut env = env.buf_in_buf_out();108                let input: NFTExtCreateItem<E> = env.read_as()?;109110                match pallet_nft::Module::<C>::create_item_internal(111                    env.ext().address().clone(),112                    input.collection_id,113                    input.owner,114                    input.data,115                ) {116                    Ok(_) => Ok(RetVal::Converging(func_id)),117                    _ => Err(DispatchError::Other("CreateItem error"))118                }119            },120            2 => {121                // Create multiple items122                let mut env = env.buf_in_buf_out();123                let input: NFTExtCreateMultipleItems<E> = env.read_as()?;124125                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;126127                match pallet_nft::Module::<C>::create_multiple_items_internal(128                    env.ext().address().clone(),129                    &collection,130                    input.owner,131                    input.data,132                ) {133                    Ok(_) => Ok(RetVal::Converging(func_id)),134                    _ => Err(DispatchError::Other("CreateMultipleItems error"))135                }136            },137            3 => {138                // Approve139                let mut env = env.buf_in_buf_out();140                let input: NFTExtApprove<E> = env.read_as()?;141142                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;143144                pallet_nft::Module::<C>::approve_internal(145                    env.ext().address().clone(),146                    input.spender,147                    &collection,148                    input.item_id,149                    input.amount,150                )?;151                Ok(RetVal::Converging(func_id))152            },153            4 => {154                // Transfer from155                let mut env = env.buf_in_buf_out();156                let input: NFTExtTransferFrom<E> = env.read_as()?;157158                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;159160                pallet_nft::Module::<C>::transfer_from_internal(161                    env.ext().address().clone(),162                    input.owner,163                    input.recipient,164                    &collection,165                    input.item_id,166                    input.amount167                )?;168                Ok(RetVal::Converging(func_id))169            },170            5 => {171                // Set variable metadata172                let mut env = env.buf_in_buf_out();173                let input: NFTExtSetVariableMetaData = env.read_as()?;174175                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;176177                pallet_nft::Module::<C>::set_variable_meta_data_internal(178                    env.ext().address().clone(),179                    &collection,180                    input.item_id,181                    input.data,182                )?;183                Ok(RetVal::Converging(func_id))184            },185            6 => {186                // Toggle whitelist187                let mut env = env.buf_in_buf_out();188                let input: NFTExtToggleWhiteList<E> = env.read_as()?;189190                let collection = pallet_nft::Module::<C>::get_collection(input.collection_id)?;191192                pallet_nft::Module::<C>::toggle_white_list_internal(193                    &env.ext().address().clone(),194                    &collection,195                    &input.address,196                    input.whitelisted,197                )?;198                Ok(RetVal::Converging(func_id))199            }200            _ => {201                panic!("Passed unknown func_id to test chain extension: {}", func_id);202            }203        }204    }205}