git.delta.rocks / unique-network / refs/commits / 87253dd2aec5

difftreelog

refunfible transfer

str-mv2020-08-04parent: #49a2388.patch.diff
in: master

1 file changed

modifiedpallets/nft/src/lib.rsdiffbeforeafterboth
before · pallets/nft/src/lib.rs
1#![cfg_attr(not(feature = "std"), no_std)]23/// For more guidance on Substrate FRAME, see the example pallet4/// https://github.com/paritytech/substrate/blob/master/frame/example/src/lib.rs56use codec::{Decode, Encode};7pub use frame_support::{8    decl_event, decl_module, decl_storage,9    construct_runtime, parameter_types,10    traits::{Currency, Get, ExistenceRequirement, KeyOwnerProofSystem, OnUnbalanced, Randomness, WithdrawReason, Imbalance},11    weights::{12        DispatchInfo, PostDispatchInfo, constants::{BlockExecutionWeight, ExtrinsicBaseWeight, RocksDbWeight, WEIGHT_PER_SECOND},13        IdentityFee, Weight, WeightToFeePolynomial, GetDispatchInfo, Pays,14    },15    StorageValue,16    dispatch::DispatchResult, 17    IsSubType,18    ensure19};2021use frame_system::{self as system, ensure_signed};22use sp_runtime::sp_std::prelude::Vec;23use sp_std::prelude::*;24use sp_runtime::{25	FixedU128, FixedPointOperand, 26	transaction_validity::{27		TransactionPriority, ValidTransaction, InvalidTransaction, TransactionValidityError, TransactionValidity28	},29	traits::{30        Saturating, Dispatchable, DispatchInfoOf, PostDispatchInfoOf, SignedExtension, Zero, SaturatedConversion,31	},32};3334#[cfg(test)]35mod mock;3637#[cfg(test)]38mod tests;3940#[derive(Encode, Decode, Debug, Eq, Clone, PartialEq)]41pub enum CollectionMode {42    Invalid,43    // custom data size44    NFT(u32),45    // amount46	Fungible(u32),47	ReFungible,48}4950impl Into<u8> for CollectionMode {51    fn into(self) -> u8{52        match self {53            CollectionMode::Invalid => 0,54            CollectionMode::NFT(_) => 1,55            CollectionMode::Fungible(_) => 2,56            CollectionMode::ReFungible => 3,57        }58    }59}6061#[derive(Encode, Decode, Debug, Clone, PartialEq)]62pub enum AccessMode {63    Normal,64	WhiteList,65}66impl Default for AccessMode { fn default() -> Self { Self::Normal } }6768impl Default for CollectionMode { fn default() -> Self { Self::Invalid } }6970#[derive(Encode, Decode, Default, Clone, PartialEq)]71#[cfg_attr(feature = "std", derive(Debug))]72pub struct Ownership<AccountId> {73    pub owner: AccountId,74    pub fraction: u12875}7677#[derive(Encode, Decode, Default, Clone, PartialEq)]78#[cfg_attr(feature = "std", derive(Debug))]79pub struct CollectionType<AccountId> {80    pub owner: AccountId,81    pub mode: CollectionMode,82    pub access: AccessMode,83    pub next_item_id: u64,84    pub decimal_points: u32,85    pub name: Vec<u16>,        // 64 include null escape char86    pub description: Vec<u16>, // 256 include null escape char87    pub token_prefix: Vec<u8>, // 16 include null escape char88    pub custom_data_size: u32,89    pub offchain_schema: Vec<u8>,90    pub sponsor: AccountId,    // Who pays fees. If set to default address, the fees are applied to the transaction sender91    pub unconfirmed_sponsor: AccountId, // Sponsor address that has not yet confirmed sponsorship92}9394#[derive(Encode, Decode, Default, Clone, PartialEq)]95#[cfg_attr(feature = "std", derive(Debug))]96pub struct CollectionAdminsType<AccountId> {97    pub admin: AccountId,98    pub collection_id: u64,99}100101#[derive(Encode, Decode, Default, Clone, PartialEq)]102#[cfg_attr(feature = "std", derive(Debug))]103pub struct NftItemType<AccountId> {104    pub collection: u64,105    pub owner: AccountId,106    pub data: Vec<u8>,107}108109#[derive(Encode, Decode, Default, Clone, PartialEq)]110#[cfg_attr(feature = "std", derive(Debug))]111pub struct FungibleItemType<AccountId> {112    pub collection: u64,113    pub owner: Vec<AccountId>,114    pub data: Vec<u64>,115}116117#[derive(Encode, Decode, Default, Clone, PartialEq)]118#[cfg_attr(feature = "std", derive(Debug))]119pub struct ReFungibleItemType<AccountId> {120    pub collection: u64,121    pub owner: Vec<Ownership<AccountId>>,122}123124pub trait Trait: system::Trait {125    type Event: From<Event<Self>> + Into<<Self as system::Trait>::Event>;126127}128129decl_storage! {130    trait Store for Module<T: Trait> as Nft {131132        // Private members133        NextCollectionID: u64;134        ItemListIndex: map hasher(blake2_128_concat) u64 => u64;135136        pub Collection get(fn collection): map hasher(identity) u64 => CollectionType<T::AccountId>;137        pub AdminList get(fn admin_list_collection): map hasher(identity) u64 => Vec<T::AccountId>;138        pub WhiteList get(fn white_list): map hasher(identity) u64 => Vec<T::AccountId>;139140        // Balance owner per collection map141        pub Balance get(fn balance_count): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) T::AccountId => u64;142        pub ApprovedList get(fn approved): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => Vec<T::AccountId>;143144        // Item collections145        pub NftItemList get(fn nft_item_id): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => NftItemType<T::AccountId>;146        pub FungibleItemList get(fn fungible_item_id): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => FungibleItemType<T::AccountId>;147        pub ReFungibleItemList get(fn refungible_item_id): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => ReFungibleItemType<T::AccountId>;148149        pub AddressTokens get(fn address_tokens): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) T::AccountId => Vec<u64>;150    }151}152153decl_event!(154    pub enum Event<T>155    where156        AccountId = <T as system::Trait>::AccountId,157    {158        Created(u64, u8, AccountId),159        ItemCreated(u64, u64),160        ItemDestroyed(u64, u64),161    }162);163164decl_module! {165    pub struct Module<T: Trait> for enum Call where origin: T::Origin {166167        fn deposit_event() = default;168169        // Create collection of NFT with given parameters170        //171        // @param customDataSz size of custom data in each collection item172        // returns collection ID173        #[weight = 0]174        pub fn create_collection(   origin,175                                    collection_name: Vec<u16>,176                                    collection_description: Vec<u16>,177                                    token_prefix: Vec<u8>,178                                    mode: CollectionMode) -> DispatchResult {179180            // Anyone can create a collection181            let who = ensure_signed(origin)?;182            let custom_data_size = match mode {183                CollectionMode::NFT(size) => size,184                _ => 0185            };186187            let decimal_points = match mode {188                CollectionMode::Fungible(points) => points,189                _ => 0190            };191192            // check params193            ensure!(decimal_points < 100, "decimal_points parameter must be lower than 100"); 194195            let mut name = collection_name.to_vec();196            name.push(0);197            ensure!(name.len() <= 64, "Collection name can not be longer than 63 char");198199            let mut description = collection_description.to_vec();200            description.push(0);201            ensure!(name.len() <= 256, "Collection description can not be longer than 255 char");202203            let mut prefix = token_prefix.to_vec();204            prefix.push(0);205            ensure!(prefix.len() <= 16, "Token prefix can not be longer than 15 char");206207            // Generate next collection ID208            let next_id = NextCollectionID::get()209                .checked_add(1)210                .expect("collection id error");211212            NextCollectionID::put(next_id);213214            // Create new collection215            let new_collection = CollectionType {216                owner: who.clone(),217                name: name,218                mode: mode.clone(),219                access: AccessMode::Normal,220                description: description,221                decimal_points: decimal_points,222                token_prefix: prefix,223                next_item_id: next_id,224                offchain_schema: Vec::new(),225                custom_data_size: custom_data_size,226                sponsor: T::AccountId::default(),227                unconfirmed_sponsor: T::AccountId::default(),228            };229230            // Add new collection to map231            <Collection<T>>::insert(next_id, new_collection);232233            // call event234            Self::deposit_event(RawEvent::Created(next_id, mode.into(), who.clone()));235236            Ok(())237        }238239        #[weight = 0]240        pub fn destroy_collection(origin, collection_id: u64) -> DispatchResult {241242            let sender = ensure_signed(origin)?;243            Self::check_owner_permissions(collection_id, sender)?;244245            <AddressTokens<T>>::remove_prefix(collection_id);246            <ApprovedList<T>>::remove_prefix(collection_id);247            <Balance<T>>::remove_prefix(collection_id);248            <ItemListIndex>::remove(collection_id);249            <AdminList<T>>::remove(collection_id);250            <Collection<T>>::remove(collection_id);251            <WhiteList<T>>::remove(collection_id);252253            Ok(())254        }255256        #[weight = 0]257        pub fn change_collection_owner(origin, collection_id: u64, new_owner: T::AccountId) -> DispatchResult {258259            let sender = ensure_signed(origin)?;260            Self::check_owner_permissions(collection_id, sender)?;261            let mut target_collection = <Collection<T>>::get(collection_id);262            target_collection.owner = new_owner;263            <Collection<T>>::insert(collection_id, target_collection);264265            Ok(())266        }267268        #[weight = 0]269        pub fn add_collection_admin(origin, collection_id: u64, new_admin_id: T::AccountId) -> DispatchResult {270271            let sender = ensure_signed(origin)?;272            Self::check_owner_or_admin_permissions(collection_id, sender)?;273            let mut admin_arr: Vec<T::AccountId> = Vec::new();274275            if <AdminList<T>>::contains_key(collection_id)276            {277                admin_arr = <AdminList<T>>::get(collection_id);278                ensure!(!admin_arr.contains(&new_admin_id), "Account already has admin role");279            }280281            admin_arr.push(new_admin_id);282            <AdminList<T>>::insert(collection_id, admin_arr);283284            Ok(())285        }286287        #[weight = 0]288        pub fn remove_collection_admin(origin, collection_id: u64, account_id: T::AccountId) -> DispatchResult {289290            let sender = ensure_signed(origin)?;291            Self::check_owner_or_admin_permissions(collection_id, sender)?;292293            if <AdminList<T>>::contains_key(collection_id)294            {295                let mut admin_arr = <AdminList<T>>::get(collection_id);296                admin_arr.retain(|i| *i != account_id);297                <AdminList<T>>::insert(collection_id, admin_arr);298            }299300            Ok(())301        }302303        #[weight = 0]304        pub fn set_collection_sponsor(origin, collection_id: u64, new_sponsor: T::AccountId) -> DispatchResult {305306            let sender = ensure_signed(origin)?;307            ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");308309            let mut target_collection = <Collection<T>>::get(collection_id);310            ensure!(sender == target_collection.owner, "You do not own this collection");311312            target_collection.unconfirmed_sponsor = new_sponsor;313            <Collection<T>>::insert(collection_id, target_collection);314315            Ok(())316        }317318        #[weight = 0]319        pub fn confirm_sponsorship(origin, collection_id: u64) -> DispatchResult {320321            let sender = ensure_signed(origin)?;322            ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");323324            let mut target_collection = <Collection<T>>::get(collection_id);325            ensure!(sender == target_collection.unconfirmed_sponsor, "This address is not set as sponsor, use setCollectionSponsor first");326327            target_collection.sponsor = target_collection.unconfirmed_sponsor;328            target_collection.unconfirmed_sponsor = T::AccountId::default();329            <Collection<T>>::insert(collection_id, target_collection);330331            Ok(())332        }333334        #[weight = 0]335        pub fn remove_collection_sponsor(origin, collection_id: u64) -> DispatchResult {336337            let sender = ensure_signed(origin)?;338            ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");339340            let mut target_collection = <Collection<T>>::get(collection_id);341            ensure!(sender == target_collection.owner, "You do not own this collection");342343            target_collection.sponsor = T::AccountId::default();344            <Collection<T>>::insert(collection_id, target_collection);345346            Ok(())347        }348        349        #[weight = 0]350        pub fn create_item(origin, collection_id: u64, properties: Vec<u8>, owner: T::AccountId) -> DispatchResult {351352            let sender = ensure_signed(origin)?;353354            // check size355            let target_collection = <Collection<T>>::get(collection_id);356            ensure!(target_collection.custom_data_size >= properties.len() as u32, "Size of item is too large");357358            Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;359360            let new_balance = <Balance<T>>::get(collection_id, owner.clone()) + 1;361            <Balance<T>>::insert(collection_id, owner.clone(), new_balance);362363            // TODO: implement other modes364            match target_collection.mode 365            {366                CollectionMode::NFT(_) => {367                // Create nft item368                    let item = NftItemType {369                        collection: collection_id,370                        owner: owner,371                        data: properties,372                    };373    374                    Self::add_nft_item(item)?;375    376                },377                _ => ()378            };379380            // call event381            Self::deposit_event(RawEvent::ItemCreated(collection_id, <ItemListIndex>::get(collection_id)));382383            Ok(())384        }385386        #[weight = 0]387        pub fn burn_item(origin, collection_id: u64, item_id: u64) -> DispatchResult {388389            let sender = ensure_signed(origin)?;390            let item_owner = Self::is_item_owner(sender.clone(), collection_id, item_id);391            if !item_owner392            {393                Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;394            }395            396            Self::burn_nft_item(collection_id, item_id)?;397398            // call event399            Self::deposit_event(RawEvent::ItemDestroyed(collection_id, item_id));400401            Ok(())402        }403404        #[weight = 0]405        pub fn transfer(origin, recipient: T::AccountId, collection_id: u64, item_id: u64, value: u64) -> DispatchResult {406407            let sender = ensure_signed(origin)?;408            let item_owner = Self::is_item_owner(sender.clone(), collection_id, item_id);409            if !item_owner410            {411                Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;412            }413414            let target_collection = <Collection<T>>::get(collection_id);415416            // TODO: implement other modes417            match target_collection.mode 418            {419                CollectionMode::NFT(_) => Self::transfer_nft(collection_id, item_id, recipient)?,420                _ => ()421            };422423            Ok(())424        }425426        #[weight = 0]427        pub fn approve(origin, approved: T::AccountId, collection_id: u64, item_id: u64) -> DispatchResult {428429            let sender = ensure_signed(origin)?;430431            let item_owner = Self::is_item_owner(sender.clone(), collection_id, item_id);432            if !item_owner433            {434                Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;435            }436437            let list_exists = <ApprovedList<T>>::contains_key(collection_id, item_id);438            if list_exists {439440                let mut list = <ApprovedList<T>>::get(collection_id, item_id);441                let item_contains = list.contains(&approved.clone());442443                if !item_contains {444                    list.push(approved.clone());445                }446            } else {447448                let mut itm = Vec::new();449                itm.push(approved.clone());450                <ApprovedList<T>>::insert(collection_id, item_id, itm);451            }452453            Ok(())454        }455456        #[weight = 0]457        pub fn transfer_from(origin, recipient: T::AccountId, collection_id: u64, item_id: u64, ) -> DispatchResult {458459            let mut approved: bool = false; 460            let sender = ensure_signed(origin)?;461            let approved_list_exists = <ApprovedList<T>>::contains_key(collection_id, item_id);462            if approved_list_exists463            {464                let list_itm = <ApprovedList<T>>::get(collection_id, item_id);465                approved = list_itm.contains(&recipient.clone());466            }467468            if !approved469            {470                Self::check_owner_or_admin_permissions(collection_id, sender)?;471            }472            473            let target_collection = <Collection<T>>::get(collection_id);474475            match target_collection.mode476            {477                CollectionMode::NFT(_) => Self::transfer_nft(collection_id, item_id, recipient)?,478                // TODO: implement other modes479                _ => ()480            };481482            Ok(())483        }484485        #[weight = 0]486        pub fn safe_transfer_from(origin, collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {487488            // let no_perm_mes = "You do not have permissions to modify this collection";489            // ensure!(<ApprovedList<T>>::contains_key((collection_id, item_id)), no_perm_mes);490            // let list_itm = <ApprovedList<T>>::get((collection_id, item_id));491            // ensure!(list_itm.contains(&new_owner.clone()), no_perm_mes);492493            // // on_nft_received  call494495            // Self::transfer(origin, collection_id, item_id, new_owner)?;496497            Ok(())498        }499500        #[weight = 0]501        pub fn set_offchain_schema(502            origin,503            collection_id: u64,504            schema: Vec<u8>505        ) -> DispatchResult {506            let sender = ensure_signed(origin)?;507            Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;508            509            let mut target_collection = <Collection<T>>::get(collection_id);510            target_collection.offchain_schema = schema;511            <Collection<T>>::insert(collection_id, target_collection);512513            Ok(())        514        }515    }516}517518impl<T: Trait> Module<T> {519520    fn collection_exists(collection_id: u64) -> DispatchResult{521        ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");522        Ok(())523    }524525    fn check_owner_permissions(collection_id: u64, subject: T::AccountId) -> DispatchResult {526527        Self::collection_exists(collection_id)?;528529        let target_collection = <Collection<T>>::get(collection_id);530        ensure!(subject == target_collection.owner, "You do not own this collection");531532        Ok(())533    }534535    fn check_owner_or_admin_permissions(collection_id: u64, subject: T::AccountId) -> DispatchResult {536537        Self::collection_exists(collection_id)?;538539        let target_collection = <Collection<T>>::get(collection_id);540        let is_owner = subject == target_collection.owner;541542        let no_perm_mes = "You do not have permissions to modify this collection";543        let exists = <AdminList<T>>::contains_key(collection_id);544545        if !is_owner546        {547            ensure!(exists, no_perm_mes);548            ensure!(<AdminList<T>>::get(collection_id).contains(&subject), no_perm_mes);549        }550        Ok(())551    }552553    fn is_item_owner(subject: T::AccountId, collection_id: u64, item_id: u64) -> bool{554555        let target_collection = <Collection<T>>::get(collection_id);556557        match target_collection.mode {558            CollectionMode::NFT(_) => <NftItemList<T>>::get(collection_id, item_id).owner == subject,559            CollectionMode::Fungible(_) => <FungibleItemList<T>>::get(collection_id, item_id).owner.contains(&subject),560            CollectionMode::ReFungible => <ReFungibleItemList<T>>::get(collection_id, item_id).owner.iter().any(|i| i.owner == subject),561            CollectionMode::Invalid => false562        }563    }564565    fn add_nft_item(item: NftItemType<T::AccountId>) -> DispatchResult {566567        let current_index = <ItemListIndex>::get(item.collection)568        .checked_add(1)569        .expect("Item list index id error");570571        Self::add_token_index(item.collection, current_index, item.owner.clone())?;572573        <ItemListIndex>::insert(item.collection, current_index);574        <NftItemList<T>>::insert(item.collection, current_index, item);575576        Ok(())577    }578579    fn burn_nft_item(collection_id: u64, item_id: u64) -> DispatchResult {580  581        let item = <NftItemList<T>>::get(collection_id, item_id);582        Self::remove_token_index(collection_id, item_id, item.owner.clone())?;583584        // update balance585        let new_balance = <Balance<T>>::get(collection_id, item.owner.clone()).checked_sub(1).unwrap();586        <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);587        <NftItemList<T>>::remove(collection_id, item_id);588589        Ok(())590    }591592    fn transfer_nft(collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {593594        let mut item = <NftItemList<T>>::get(collection_id, item_id);595596        // update balance597        let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone()).checked_sub(1).unwrap();598        <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);599600        let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone()).checked_add(1).unwrap();601        <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);602603        // change owner604        let old_owner = item.owner.clone();605        item.owner = new_owner.clone();606        <NftItemList<T>>::insert(collection_id, item_id, item);607608        // update index collection609        Self::move_token_index(collection_id, item_id, old_owner, new_owner.clone())?;610611        // reset approved list612        let itm: Vec<T::AccountId> = Vec::new();613        <ApprovedList<T>>::insert(collection_id, item_id, itm);614615        Ok(())616    }617618    fn add_token_index(collection_id: u64, item_index: u64, owner: T::AccountId) -> DispatchResult {619        let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());620        if list_exists {621            let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());622            let item_contains = list.contains(&item_index.clone());623624            if !item_contains {625                list.push(item_index.clone());626            }627628            <AddressTokens<T>>::insert(collection_id, owner.clone(), list);629        } else {630            let mut itm = Vec::new();631            itm.push(item_index.clone());632            <AddressTokens<T>>::insert(collection_id, owner, itm);633        }634635        Ok(())636    }637638    fn remove_token_index(639        collection_id: u64,640        item_index: u64,641        owner: T::AccountId,642    ) -> DispatchResult {643        let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());644        if list_exists {645            let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());646            let item_contains = list.contains(&item_index.clone());647648            if item_contains {649                list.retain(|&item| item != item_index);650                <AddressTokens<T>>::insert(collection_id, owner, list);651            }652        }653654        Ok(())655    }656657    fn move_token_index(658        collection_id: u64,659        item_index: u64,660        old_owner: T::AccountId,661        new_owner: T::AccountId,662    ) -> DispatchResult {663        Self::remove_token_index(collection_id, item_index, old_owner)?;664        Self::add_token_index(collection_id, item_index, new_owner)?;665666        Ok(())667    }668}669670671////////////////////////////////////////////////////////////////////////////////////////////////////672// Economic models673674/// Fee multiplier.675pub type Multiplier = FixedU128;676677type BalanceOf<T> =678	<<T as transaction_payment::Trait>::Currency as Currency<<T as system::Trait>::AccountId>>::Balance;679type NegativeImbalanceOf<T> = <<T as transaction_payment::Trait>::Currency as Currency<680	<T as system::Trait>::AccountId,>>::NegativeImbalance;681682683684/// Require the transactor pay for themselves and maybe include a tip to gain additional priority685/// in the queue.686#[derive(Encode, Decode, Clone, Eq, PartialEq)]687pub struct ChargeTransactionPayment<T: transaction_payment::Trait + Send + Sync>(#[codec(compact)] BalanceOf<T>);688689impl<T:Trait + transaction_payment::Trait + Send + Sync> sp_std::fmt::Debug for ChargeTransactionPayment<T> {690	#[cfg(feature = "std")]691	fn fmt(&self, f: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {692		write!(f, "ChargeTransactionPayment<{:?}>", self.0)693	}694	#[cfg(not(feature = "std"))]695	fn fmt(&self, _: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {696		Ok(())697	}698}699700impl<T:Trait + transaction_payment::Trait + Send + Sync> ChargeTransactionPayment<T> where701	T::Call: Dispatchable<Info=DispatchInfo, PostInfo=PostDispatchInfo> + IsSubType<Module<T>, T>,702	BalanceOf<T>: Send + Sync + FixedPointOperand,703{704	/// utility constructor. Used only in client/factory code.705	pub fn from(fee: BalanceOf<T>) -> Self {706		Self(fee)707	}708709    pub fn traditional_fee(710        len: usize,711        info: &DispatchInfoOf<T::Call>,712        tip: BalanceOf<T>,713    ) -> BalanceOf<T> where714        T::Call: Dispatchable<Info=DispatchInfo>,715    {716        <transaction_payment::Module<T>>::compute_fee(len as u32, info, tip)717    }718719	fn withdraw_fee(720		&self,721        who: &T::AccountId,722        call: &T::Call,723		info: &DispatchInfoOf<T::Call>,724		len: usize,725	) -> Result<(BalanceOf<T>, Option<NegativeImbalanceOf<T>>), TransactionValidityError> {726        let tip = self.0;727728        // Set fee based on call type. Creating collection costs 1 Unique.729        // All other transactions have traditional fees so far730        let fee = match call.is_sub_type() {731            Some(Call::create_collection(..)) => <BalanceOf<T>>::from(1_000_000_000),732            _ => Self::traditional_fee(len, info, tip)733734            // Flat fee model, use only for testing purposes735            // _ => <BalanceOf<T>>::from(100)736        };737738        // Determine who is paying transaction fee based on ecnomic model739        // Parse call to extract collection ID and access collection sponsor740        let sponsor: T::AccountId = match call.is_sub_type() {741            Some(Call::create_item(collection_id, _properties, _owner)) => {742                <Collection<T>>::get(collection_id).sponsor743            },744            Some(Call::transfer(_new_owner, collection_id, _item_id, _value)) => {745                <Collection<T>>::get(collection_id).sponsor746            },747748            _ => T::AccountId::default()749        };750751        let mut who_pays_fee: T::AccountId = sponsor.clone();752        if sponsor == T::AccountId::default() {753            who_pays_fee = who.clone();754        }755756		// Only mess with balances if fee is not zero.757		if fee.is_zero() {758			return Ok((fee, None));759		}760761		match <T as transaction_payment::Trait>::Currency::withdraw(762			&who_pays_fee,763			fee,764			if tip.is_zero() {765				WithdrawReason::TransactionPayment.into()766			} else {767				WithdrawReason::TransactionPayment | WithdrawReason::Tip768			},769			ExistenceRequirement::KeepAlive,770		) {771			Ok(imbalance) => Ok((fee, Some(imbalance))),772			Err(_) => Err(InvalidTransaction::Payment.into()),773		}774	}775}776777impl<T:Trait + transaction_payment::Trait + Send + Sync> SignedExtension for ChargeTransactionPayment<T> where778    BalanceOf<T>: Send + Sync + From<u64> + FixedPointOperand,779    T::Call: Dispatchable<Info=DispatchInfo, PostInfo=PostDispatchInfo> + IsSubType<Module<T>, T>,780{781	const IDENTIFIER: &'static str = "ChargeTransactionPayment";782	type AccountId = T::AccountId;783	type Call = T::Call;784	type AdditionalSigned = ();785	type Pre = (BalanceOf<T>, Self::AccountId, Option<NegativeImbalanceOf<T>>, BalanceOf<T>);786	fn additional_signed(&self) -> sp_std::result::Result<(), TransactionValidityError> { Ok(()) }787788	fn validate(789		&self,790		who: &Self::AccountId,791		call: &Self::Call,792		info: &DispatchInfoOf<Self::Call>,793		len: usize,794	) -> TransactionValidity {795		let (fee, _) = self.withdraw_fee(who, call, info, len)?;796797		let mut r = ValidTransaction::default();798		// NOTE: we probably want to maximize the _fee (of any type) per weight unit_ here, which799		// will be a bit more than setting the priority to tip. For now, this is enough.800		r.priority = fee.saturated_into::<TransactionPriority>();801		Ok(r)802	}803804	fn pre_dispatch(805		self,806		who: &Self::AccountId,807		call: &Self::Call,808		info: &DispatchInfoOf<Self::Call>,809		len: usize810	) -> Result<Self::Pre, TransactionValidityError> {811		let (fee, imbalance) = self.withdraw_fee(who, call, info, len)?;812		Ok((self.0, who.clone(), imbalance, fee))813	}814815	fn post_dispatch(816		pre: Self::Pre,817		info: &DispatchInfoOf<Self::Call>,818		post_info: &PostDispatchInfoOf<Self::Call>,819		len: usize,820		_result: &DispatchResult,821	) -> Result<(), TransactionValidityError> {822		let (tip, who, imbalance, fee) = pre;823		if let Some(payed) = imbalance {824			let actual_fee = <transaction_payment::Module<T>>::compute_actual_fee(825				len as u32,826				info,827				post_info,828				tip,829			);830			let refund = fee.saturating_sub(actual_fee);831			let actual_payment = match <T as transaction_payment::Trait>::Currency::deposit_into_existing(&who, refund) {832				Ok(refund_imbalance) => {833					// The refund cannot be larger than the up front payed max weight.834					// `PostDispatchInfo::calc_unspent` guards against such a case.835					match payed.offset(refund_imbalance) {836						Ok(actual_payment) => actual_payment,837						Err(_) => return Err(InvalidTransaction::Payment.into()),838					}839				}840				// We do not recreate the account using the refund. The up front payment841				// is gone in that case.842				Err(_) => payed,843			};844			let imbalances = actual_payment.split(tip);845			<T as transaction_payment::Trait>::OnTransactionPayment::on_unbalanceds(Some(imbalances.0).into_iter()846				.chain(Some(imbalances.1)));847		}848		Ok(())849	}850}
after · pallets/nft/src/lib.rs
1#![cfg_attr(not(feature = "std"), no_std)]23/// For more guidance on Substrate FRAME, see the example pallet4/// https://github.com/paritytech/substrate/blob/master/frame/example/src/lib.rs56use codec::{Decode, Encode};7pub use frame_support::{8    decl_event, decl_module, decl_storage,9    construct_runtime, parameter_types,10    traits::{Currency, Get, ExistenceRequirement, KeyOwnerProofSystem, OnUnbalanced, Randomness, WithdrawReason, Imbalance},11    weights::{12        DispatchInfo, PostDispatchInfo, constants::{BlockExecutionWeight, ExtrinsicBaseWeight, RocksDbWeight, WEIGHT_PER_SECOND},13        IdentityFee, Weight, WeightToFeePolynomial, GetDispatchInfo, Pays,14    },15    StorageValue,16    dispatch::DispatchResult, 17    IsSubType,18    ensure19};2021use frame_system::{self as system, ensure_signed};22use sp_runtime::sp_std::prelude::Vec;23use sp_std::prelude::*;24use sp_runtime::{25	FixedU128, FixedPointOperand, 26	transaction_validity::{27		TransactionPriority, ValidTransaction, InvalidTransaction, TransactionValidityError, TransactionValidity28	},29	traits::{30        Saturating, Dispatchable, DispatchInfoOf, PostDispatchInfoOf, SignedExtension, Zero, SaturatedConversion,31	},32};3334#[cfg(test)]35mod mock;3637#[cfg(test)]38mod tests;3940#[derive(Encode, Decode, Debug, Eq, Clone, PartialEq)]41pub enum CollectionMode {42    Invalid,43    // custom data size44    NFT(u32),45    // decimal points46    Fungible(u32),47    // custom data size48	ReFungible(u32),49}5051impl Into<u8> for CollectionMode {52    fn into(self) -> u8{53        match self {54            CollectionMode::Invalid => 0,55            CollectionMode::NFT(_) => 1,56            CollectionMode::Fungible(_) => 2,57            CollectionMode::ReFungible(_) => 3,58        }59    }60}6162#[derive(Encode, Decode, Debug, Clone, PartialEq)]63pub enum AccessMode {64    Normal,65	WhiteList,66}67impl Default for AccessMode { fn default() -> Self { Self::Normal } }6869impl Default for CollectionMode { fn default() -> Self { Self::Invalid } }7071#[derive(Encode, Decode, Default, Clone, PartialEq)]72#[cfg_attr(feature = "std", derive(Debug))]73pub struct Ownership<AccountId> {74    pub owner: AccountId,75    pub fraction: u12876}7778#[derive(Encode, Decode, Default, Clone, PartialEq)]79#[cfg_attr(feature = "std", derive(Debug))]80pub struct CollectionType<AccountId> {81    pub owner: AccountId,82    pub mode: CollectionMode,83    pub access: AccessMode,84    pub next_item_id: u64,85    pub decimal_points: u32,86    pub name: Vec<u16>,        // 64 include null escape char87    pub description: Vec<u16>, // 256 include null escape char88    pub token_prefix: Vec<u8>, // 16 include null escape char89    pub custom_data_size: u32,90    pub offchain_schema: Vec<u8>,91    pub sponsor: AccountId,    // Who pays fees. If set to default address, the fees are applied to the transaction sender92    pub unconfirmed_sponsor: AccountId, // Sponsor address that has not yet confirmed sponsorship93}9495#[derive(Encode, Decode, Default, Clone, PartialEq)]96#[cfg_attr(feature = "std", derive(Debug))]97pub struct CollectionAdminsType<AccountId> {98    pub admin: AccountId,99    pub collection_id: u64,100}101102#[derive(Encode, Decode, Default, Clone, PartialEq)]103#[cfg_attr(feature = "std", derive(Debug))]104pub struct NftItemType<AccountId> {105    pub collection: u64,106    pub owner: AccountId,107    pub data: Vec<u8>,108}109110#[derive(Encode, Decode, Default, Clone, PartialEq)]111#[cfg_attr(feature = "std", derive(Debug))]112pub struct FungibleItemType<AccountId> {113    pub collection: u64,114    pub owner: AccountId,115    pub value: u128,116}117118#[derive(Encode, Decode, Default, Clone, PartialEq)]119#[cfg_attr(feature = "std", derive(Debug))]120pub struct ReFungibleItemType<AccountId> {121    pub collection: u64,122    pub owner: Vec<Ownership<AccountId>>,123    pub data: Vec<u8>,124}125126pub trait Trait: system::Trait {127    type Event: From<Event<Self>> + Into<<Self as system::Trait>::Event>;128129}130131decl_storage! {132    trait Store for Module<T: Trait> as Nft {133134        // Private members135        NextCollectionID: u64;136        ItemListIndex: map hasher(blake2_128_concat) u64 => u64;137138        pub Collection get(fn collection): map hasher(identity) u64 => CollectionType<T::AccountId>;139        pub AdminList get(fn admin_list_collection): map hasher(identity) u64 => Vec<T::AccountId>;140        pub WhiteList get(fn white_list): map hasher(identity) u64 => Vec<T::AccountId>;141142        // Balance owner per collection map143        pub Balance get(fn balance_count): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) T::AccountId => u64;144        pub ApprovedList get(fn approved): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => Vec<T::AccountId>;145146        // Item collections147        pub NftItemList get(fn nft_item_id): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => NftItemType<T::AccountId>;148        pub FungibleItemList get(fn fungible_item_id): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => FungibleItemType<T::AccountId>;149        pub ReFungibleItemList get(fn refungible_item_id): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => ReFungibleItemType<T::AccountId>;150151        pub AddressTokens get(fn address_tokens): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) T::AccountId => Vec<u64>;152    }153}154155decl_event!(156    pub enum Event<T>157    where158        AccountId = <T as system::Trait>::AccountId,159    {160        Created(u64, u8, AccountId),161        ItemCreated(u64, u64),162        ItemDestroyed(u64, u64),163    }164);165166decl_module! {167    pub struct Module<T: Trait> for enum Call where origin: T::Origin {168169        fn deposit_event() = default;170171        // Create collection of NFT with given parameters172        //173        // @param customDataSz size of custom data in each collection item174        // returns collection ID175        #[weight = 0]176        pub fn create_collection(   origin,177                                    collection_name: Vec<u16>,178                                    collection_description: Vec<u16>,179                                    token_prefix: Vec<u8>,180                                    mode: CollectionMode) -> DispatchResult {181182            // Anyone can create a collection183            let who = ensure_signed(origin)?;184            let custom_data_size = match mode {185                CollectionMode::NFT(size) => size,186                CollectionMode::ReFungible(size) => size,187                _ => 0188            };189190            let decimal_points = match mode {191                CollectionMode::Fungible(points) => points,192                _ => 0193            };194195            // check params196            ensure!(decimal_points <= 4, "decimal_points parameter must be lower than 4"); 197198            let mut name = collection_name.to_vec();199            name.push(0);200            ensure!(name.len() <= 64, "Collection name can not be longer than 63 char");201202            let mut description = collection_description.to_vec();203            description.push(0);204            ensure!(name.len() <= 256, "Collection description can not be longer than 255 char");205206            let mut prefix = token_prefix.to_vec();207            prefix.push(0);208            ensure!(prefix.len() <= 16, "Token prefix can not be longer than 15 char");209210            // Generate next collection ID211            let next_id = NextCollectionID::get()212                .checked_add(1)213                .expect("collection id error");214215            NextCollectionID::put(next_id);216217            // Create new collection218            let new_collection = CollectionType {219                owner: who.clone(),220                name: name,221                mode: mode.clone(),222                access: AccessMode::Normal,223                description: description,224                decimal_points: decimal_points,225                token_prefix: prefix,226                next_item_id: next_id,227                offchain_schema: Vec::new(),228                custom_data_size: custom_data_size,229                sponsor: T::AccountId::default(),230                unconfirmed_sponsor: T::AccountId::default(),231            };232233            // Add new collection to map234            <Collection<T>>::insert(next_id, new_collection);235236            // call event237            Self::deposit_event(RawEvent::Created(next_id, mode.into(), who.clone()));238239            Ok(())240        }241242        #[weight = 0]243        pub fn destroy_collection(origin, collection_id: u64) -> DispatchResult {244245            let sender = ensure_signed(origin)?;246            Self::check_owner_permissions(collection_id, sender)?;247248            <AddressTokens<T>>::remove_prefix(collection_id);249            <ApprovedList<T>>::remove_prefix(collection_id);250            <Balance<T>>::remove_prefix(collection_id);251            <ItemListIndex>::remove(collection_id);252            <AdminList<T>>::remove(collection_id);253            <Collection<T>>::remove(collection_id);254            <WhiteList<T>>::remove(collection_id);255256            Ok(())257        }258259        #[weight = 0]260        pub fn change_collection_owner(origin, collection_id: u64, new_owner: T::AccountId) -> DispatchResult {261262            let sender = ensure_signed(origin)?;263            Self::check_owner_permissions(collection_id, sender)?;264            let mut target_collection = <Collection<T>>::get(collection_id);265            target_collection.owner = new_owner;266            <Collection<T>>::insert(collection_id, target_collection);267268            Ok(())269        }270271        #[weight = 0]272        pub fn add_collection_admin(origin, collection_id: u64, new_admin_id: T::AccountId) -> DispatchResult {273274            let sender = ensure_signed(origin)?;275            Self::check_owner_or_admin_permissions(collection_id, sender)?;276            let mut admin_arr: Vec<T::AccountId> = Vec::new();277278            if <AdminList<T>>::contains_key(collection_id)279            {280                admin_arr = <AdminList<T>>::get(collection_id);281                ensure!(!admin_arr.contains(&new_admin_id), "Account already has admin role");282            }283284            admin_arr.push(new_admin_id);285            <AdminList<T>>::insert(collection_id, admin_arr);286287            Ok(())288        }289290        #[weight = 0]291        pub fn remove_collection_admin(origin, collection_id: u64, account_id: T::AccountId) -> DispatchResult {292293            let sender = ensure_signed(origin)?;294            Self::check_owner_or_admin_permissions(collection_id, sender)?;295296            if <AdminList<T>>::contains_key(collection_id)297            {298                let mut admin_arr = <AdminList<T>>::get(collection_id);299                admin_arr.retain(|i| *i != account_id);300                <AdminList<T>>::insert(collection_id, admin_arr);301            }302303            Ok(())304        }305306        #[weight = 0]307        pub fn set_collection_sponsor(origin, collection_id: u64, new_sponsor: T::AccountId) -> DispatchResult {308309            let sender = ensure_signed(origin)?;310            ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");311312            let mut target_collection = <Collection<T>>::get(collection_id);313            ensure!(sender == target_collection.owner, "You do not own this collection");314315            target_collection.unconfirmed_sponsor = new_sponsor;316            <Collection<T>>::insert(collection_id, target_collection);317318            Ok(())319        }320321        #[weight = 0]322        pub fn confirm_sponsorship(origin, collection_id: u64) -> DispatchResult {323324            let sender = ensure_signed(origin)?;325            ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");326327            let mut target_collection = <Collection<T>>::get(collection_id);328            ensure!(sender == target_collection.unconfirmed_sponsor, "This address is not set as sponsor, use setCollectionSponsor first");329330            target_collection.sponsor = target_collection.unconfirmed_sponsor;331            target_collection.unconfirmed_sponsor = T::AccountId::default();332            <Collection<T>>::insert(collection_id, target_collection);333334            Ok(())335        }336337        #[weight = 0]338        pub fn remove_collection_sponsor(origin, collection_id: u64) -> DispatchResult {339340            let sender = ensure_signed(origin)?;341            ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");342343            let mut target_collection = <Collection<T>>::get(collection_id);344            ensure!(sender == target_collection.owner, "You do not own this collection");345346            target_collection.sponsor = T::AccountId::default();347            <Collection<T>>::insert(collection_id, target_collection);348349            Ok(())350        }351        352        #[weight = 0]353        pub fn create_item(origin, collection_id: u64, properties: Vec<u8>, owner: T::AccountId) -> DispatchResult {354355            let sender = ensure_signed(origin)?;356357            // check size358            let target_collection = <Collection<T>>::get(collection_id);359            ensure!(target_collection.custom_data_size >= properties.len() as u32, "Size of item is too large");360361            Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;362363            let new_balance = <Balance<T>>::get(collection_id, owner.clone()).checked_add(1).unwrap();364            <Balance<T>>::insert(collection_id, owner.clone(), new_balance);365366            // TODO: implement other modes367            match target_collection.mode 368            {369                CollectionMode::NFT(_) => {370                // Create nft item371                    let item = NftItemType {372                        collection: collection_id,373                        owner: owner,374                        data: properties,375                    };376    377                    Self::add_nft_item(item)?;378    379                },380                CollectionMode::ReFungible(_) => {381                    let mut owner_list = Vec::new();382                    let value = (10 as u128).pow(target_collection.decimal_points);383                    owner_list.push(Ownership {owner: owner, fraction: value});384385                    let item = ReFungibleItemType {386                        collection: collection_id,387                        owner: owner_list,388                        data: properties389                    };390    391                    Self::add_refungible_item(item)?;392                },393                _ => ()394            };395396            // call event397            Self::deposit_event(RawEvent::ItemCreated(collection_id, <ItemListIndex>::get(collection_id)));398399            Ok(())400        }401402        #[weight = 0]403        pub fn burn_item(origin, collection_id: u64, item_id: u64) -> DispatchResult {404405            let sender = ensure_signed(origin)?;406            let item_owner = Self::is_item_owner(sender.clone(), collection_id, item_id);407            if !item_owner408            {409                Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;410            }411            let target_collection = <Collection<T>>::get(collection_id);412413            match target_collection.mode 414            {415                CollectionMode::NFT(_) => Self::burn_nft_item(collection_id, item_id)?,416                CollectionMode::ReFungible(_) => Self::burn_refungible_item(collection_id, item_id, sender.clone())?,417                _ => ()418            };419420            // call event421            Self::deposit_event(RawEvent::ItemDestroyed(collection_id, item_id));422423            Ok(())424        }425426        #[weight = 0]427        pub fn transfer(origin, recipient: T::AccountId, collection_id: u64, item_id: u64, value: u64) -> DispatchResult {428429            let sender = ensure_signed(origin)?;430            let item_owner = Self::is_item_owner(sender.clone(), collection_id, item_id);431            if !item_owner432            {433                Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;434            }435436            let target_collection = <Collection<T>>::get(collection_id);437438            // TODO: implement other modes439            match target_collection.mode 440            {441                CollectionMode::NFT(_) => Self::transfer_nft(collection_id, item_id, recipient)?,442                CollectionMode::ReFungible(_) => Self::transfer_refungible(collection_id, item_id, value, sender.clone(), recipient)?,443                _ => ()444            };445446            Ok(())447        }448449        #[weight = 0]450        pub fn approve(origin, approved: T::AccountId, collection_id: u64, item_id: u64) -> DispatchResult {451452            let sender = ensure_signed(origin)?;453454            let item_owner = Self::is_item_owner(sender.clone(), collection_id, item_id);455            if !item_owner456            {457                Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;458            }459460            let list_exists = <ApprovedList<T>>::contains_key(collection_id, item_id);461            if list_exists {462463                let mut list = <ApprovedList<T>>::get(collection_id, item_id);464                let item_contains = list.contains(&approved.clone());465466                if !item_contains {467                    list.push(approved.clone());468                }469            } else {470471                let mut itm = Vec::new();472                itm.push(approved.clone());473                <ApprovedList<T>>::insert(collection_id, item_id, itm);474            }475476            Ok(())477        }478479        #[weight = 0]480        pub fn transfer_from(origin, recipient: T::AccountId, collection_id: u64, item_id: u64, ) -> DispatchResult {481482            let mut approved: bool = false; 483            let sender = ensure_signed(origin)?;484            let approved_list_exists = <ApprovedList<T>>::contains_key(collection_id, item_id);485            if approved_list_exists486            {487                let list_itm = <ApprovedList<T>>::get(collection_id, item_id);488                approved = list_itm.contains(&recipient.clone());489            }490491            if !approved492            {493                Self::check_owner_or_admin_permissions(collection_id, sender)?;494            }495            496            let target_collection = <Collection<T>>::get(collection_id);497498            match target_collection.mode499            {500                CollectionMode::NFT(_) => Self::transfer_nft(collection_id, item_id, recipient)?,501                // TODO: implement other modes502                _ => ()503            };504505            Ok(())506        }507508        #[weight = 0]509        pub fn safe_transfer_from(origin, collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {510511            // let no_perm_mes = "You do not have permissions to modify this collection";512            // ensure!(<ApprovedList<T>>::contains_key((collection_id, item_id)), no_perm_mes);513            // let list_itm = <ApprovedList<T>>::get((collection_id, item_id));514            // ensure!(list_itm.contains(&new_owner.clone()), no_perm_mes);515516            // // on_nft_received  call517518            // Self::transfer(origin, collection_id, item_id, new_owner)?;519520            Ok(())521        }522523        #[weight = 0]524        pub fn set_offchain_schema(525            origin,526            collection_id: u64,527            schema: Vec<u8>528        ) -> DispatchResult {529            let sender = ensure_signed(origin)?;530            Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;531            532            let mut target_collection = <Collection<T>>::get(collection_id);533            target_collection.offchain_schema = schema;534            <Collection<T>>::insert(collection_id, target_collection);535536            Ok(())        537        }538    }539}540541impl<T: Trait> Module<T> {542543    fn collection_exists(collection_id: u64) -> DispatchResult{544        ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");545        Ok(())546    }547548    fn check_owner_permissions(collection_id: u64, subject: T::AccountId) -> DispatchResult {549550        Self::collection_exists(collection_id)?;551552        let target_collection = <Collection<T>>::get(collection_id);553        ensure!(subject == target_collection.owner, "You do not own this collection");554555        Ok(())556    }557558    fn check_owner_or_admin_permissions(collection_id: u64, subject: T::AccountId) -> DispatchResult {559560        Self::collection_exists(collection_id)?;561562        let target_collection = <Collection<T>>::get(collection_id);563        let is_owner = subject == target_collection.owner;564565        let no_perm_mes = "You do not have permissions to modify this collection";566        let exists = <AdminList<T>>::contains_key(collection_id);567568        if !is_owner569        {570            ensure!(exists, no_perm_mes);571            ensure!(<AdminList<T>>::get(collection_id).contains(&subject), no_perm_mes);572        }573        Ok(())574    }575576    fn is_item_owner(subject: T::AccountId, collection_id: u64, item_id: u64) -> bool{577578        let target_collection = <Collection<T>>::get(collection_id);579580        match target_collection.mode {581            CollectionMode::NFT(_) => <NftItemList<T>>::get(collection_id, item_id).owner == subject,582            CollectionMode::Fungible(_) => <FungibleItemList<T>>::get(collection_id, item_id).owner == subject,583            CollectionMode::ReFungible(_) => <ReFungibleItemList<T>>::get(collection_id, item_id).owner.iter().any(|i| i.owner == subject),584            CollectionMode::Invalid => false585        }586    }587588    fn add_refungible_item(item: ReFungibleItemType<T::AccountId>) -> DispatchResult {589590        let current_index = <ItemListIndex>::get(item.collection)591        .checked_add(1)592        .expect("Item list index id error");593594        Self::add_token_index(item.collection, current_index, item.owner.first().unwrap().owner.clone())?;595596        <ItemListIndex>::insert(item.collection, current_index);597        <ReFungibleItemList<T>>::insert(item.collection, current_index, item);        598599        Ok(())600    }601602    fn burn_refungible_item(collection_id: u64, item_id: u64, owner: T::AccountId) -> DispatchResult {603  604        let collection = <ReFungibleItemList<T>>::get(collection_id, item_id);605        let item = collection.owner.iter().filter(|&i| i.owner == owner).next().unwrap();606        Self::remove_token_index(collection_id, item_id, owner)?;607608        // update balance609        let new_balance = <Balance<T>>::get(collection_id, item.owner.clone()).checked_sub(item.fraction as u64).unwrap();610        <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);611612        // TODO613        <ReFungibleItemList<T>>::remove(collection_id, item_id);614615        Ok(())616    }617618    fn transfer_refungible(collection_id: u64, item_id: u64, value: u64, owner: T::AccountId, new_owner: T::AccountId) -> DispatchResult {619620        let full_item = <ReFungibleItemList<T>>::get(collection_id, item_id);621        let item = full_item.owner.iter().filter(|i| i.owner == owner).next().unwrap();622        let amount = item.fraction;623624        ensure!(amount < value.into(),"Item balance not enouth");625626        // update balance627        let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone()).checked_sub(value).unwrap();628        <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);629630        let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone()).checked_add(value).unwrap();631        <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);632633        let old_owner = item.owner.clone();634        let new_owner_has_account = full_item.owner.iter().any(|i| i.owner == new_owner);635636        // transfer637        if amount == value.into() && !new_owner_has_account638        {639            // change owner640            // new owner do not have account641            let mut new_full_item = full_item.clone();642            new_full_item.owner.iter_mut().find(|i| i.owner == owner).unwrap().owner = new_owner.clone();643            <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);644645            // update index collection646            Self::move_token_index(collection_id, item_id, old_owner.clone(), new_owner.clone())?;647        }648        else649        {650            let mut new_full_item = full_item.clone();651            new_full_item.owner.iter_mut().find(|i| i.owner == owner).unwrap().fraction -= amount;652653            // separate amount654            if new_owner_has_account {655                // new owner has account656                new_full_item.owner.iter_mut().find(|i| i.owner == new_owner).unwrap().fraction += amount;657            }658            else659            {660                // new owner do not have account661                new_full_item.owner.push(Ownership { owner: new_owner.clone(), fraction: amount});662                Self::add_token_index(collection_id, item_id, new_owner.clone())?;663            }664665            <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);666        }667668        Ok(())669    }670    671    fn add_nft_item(item: NftItemType<T::AccountId>) -> DispatchResult {672673        let current_index = <ItemListIndex>::get(item.collection)674        .checked_add(1)675        .expect("Item list index id error");676677        Self::add_token_index(item.collection, current_index, item.owner.clone())?;678679        <ItemListIndex>::insert(item.collection, current_index);680        <NftItemList<T>>::insert(item.collection, current_index, item);681682        Ok(())683    }684685    fn burn_nft_item(collection_id: u64, item_id: u64) -> DispatchResult {686  687        let item = <NftItemList<T>>::get(collection_id, item_id);688        Self::remove_token_index(collection_id, item_id, item.owner.clone())?;689690        // update balance691        let new_balance = <Balance<T>>::get(collection_id, item.owner.clone()).checked_sub(1).unwrap();692        <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);693        <NftItemList<T>>::remove(collection_id, item_id);694695        Ok(())696    }697698    fn transfer_nft(collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {699700        let mut item = <NftItemList<T>>::get(collection_id, item_id);701702        // update balance703        let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone()).checked_sub(1).unwrap();704        <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);705706        let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone()).checked_add(1).unwrap();707        <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);708709        // change owner710        let old_owner = item.owner.clone();711        item.owner = new_owner.clone();712        <NftItemList<T>>::insert(collection_id, item_id, item);713714        // update index collection715        Self::move_token_index(collection_id, item_id, old_owner, new_owner.clone())?;716717        // reset approved list718        let itm: Vec<T::AccountId> = Vec::new();719        <ApprovedList<T>>::insert(collection_id, item_id, itm);720721        Ok(())722    }723724    fn add_token_index(collection_id: u64, item_index: u64, owner: T::AccountId) -> DispatchResult {725        let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());726        if list_exists {727            let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());728            let item_contains = list.contains(&item_index.clone());729730            if !item_contains {731                list.push(item_index.clone());732            }733734            <AddressTokens<T>>::insert(collection_id, owner.clone(), list);735        } else {736            let mut itm = Vec::new();737            itm.push(item_index.clone());738            <AddressTokens<T>>::insert(collection_id, owner, itm);739        }740741        Ok(())742    }743744    fn remove_token_index(745        collection_id: u64,746        item_index: u64,747        owner: T::AccountId,748    ) -> DispatchResult {749        let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());750        if list_exists {751            let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());752            let item_contains = list.contains(&item_index.clone());753754            if item_contains {755                list.retain(|&item| item != item_index);756                <AddressTokens<T>>::insert(collection_id, owner, list);757            }758        }759760        Ok(())761    }762763    fn move_token_index(764        collection_id: u64,765        item_index: u64,766        old_owner: T::AccountId,767        new_owner: T::AccountId,768    ) -> DispatchResult {769        Self::remove_token_index(collection_id, item_index, old_owner)?;770        Self::add_token_index(collection_id, item_index, new_owner)?;771772        Ok(())773    }774}775776777////////////////////////////////////////////////////////////////////////////////////////////////////778// Economic models779780/// Fee multiplier.781pub type Multiplier = FixedU128;782783type BalanceOf<T> =784	<<T as transaction_payment::Trait>::Currency as Currency<<T as system::Trait>::AccountId>>::Balance;785type NegativeImbalanceOf<T> = <<T as transaction_payment::Trait>::Currency as Currency<786	<T as system::Trait>::AccountId,>>::NegativeImbalance;787788789790/// Require the transactor pay for themselves and maybe include a tip to gain additional priority791/// in the queue.792#[derive(Encode, Decode, Clone, Eq, PartialEq)]793pub struct ChargeTransactionPayment<T: transaction_payment::Trait + Send + Sync>(#[codec(compact)] BalanceOf<T>);794795impl<T:Trait + transaction_payment::Trait + Send + Sync> sp_std::fmt::Debug for ChargeTransactionPayment<T> {796	#[cfg(feature = "std")]797	fn fmt(&self, f: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {798		write!(f, "ChargeTransactionPayment<{:?}>", self.0)799	}800	#[cfg(not(feature = "std"))]801	fn fmt(&self, _: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {802		Ok(())803	}804}805806impl<T:Trait + transaction_payment::Trait + Send + Sync> ChargeTransactionPayment<T> where807	T::Call: Dispatchable<Info=DispatchInfo, PostInfo=PostDispatchInfo> + IsSubType<Module<T>, T>,808	BalanceOf<T>: Send + Sync + FixedPointOperand,809{810	/// utility constructor. Used only in client/factory code.811	pub fn from(fee: BalanceOf<T>) -> Self {812		Self(fee)813	}814815    pub fn traditional_fee(816        len: usize,817        info: &DispatchInfoOf<T::Call>,818        tip: BalanceOf<T>,819    ) -> BalanceOf<T> where820        T::Call: Dispatchable<Info=DispatchInfo>,821    {822        <transaction_payment::Module<T>>::compute_fee(len as u32, info, tip)823    }824825	fn withdraw_fee(826		&self,827        who: &T::AccountId,828        call: &T::Call,829		info: &DispatchInfoOf<T::Call>,830		len: usize,831	) -> Result<(BalanceOf<T>, Option<NegativeImbalanceOf<T>>), TransactionValidityError> {832        let tip = self.0;833834        // Set fee based on call type. Creating collection costs 1 Unique.835        // All other transactions have traditional fees so far836        let fee = match call.is_sub_type() {837            Some(Call::create_collection(..)) => <BalanceOf<T>>::from(1_000_000_000),838            _ => Self::traditional_fee(len, info, tip)839840            // Flat fee model, use only for testing purposes841            // _ => <BalanceOf<T>>::from(100)842        };843844        // Determine who is paying transaction fee based on ecnomic model845        // Parse call to extract collection ID and access collection sponsor846        let sponsor: T::AccountId = match call.is_sub_type() {847            Some(Call::create_item(collection_id, _properties, _owner)) => {848                <Collection<T>>::get(collection_id).sponsor849            },850            Some(Call::transfer(_new_owner, collection_id, _item_id, _value)) => {851                <Collection<T>>::get(collection_id).sponsor852            },853854            _ => T::AccountId::default()855        };856857        let mut who_pays_fee: T::AccountId = sponsor.clone();858        if sponsor == T::AccountId::default() {859            who_pays_fee = who.clone();860        }861862		// Only mess with balances if fee is not zero.863		if fee.is_zero() {864			return Ok((fee, None));865		}866867		match <T as transaction_payment::Trait>::Currency::withdraw(868			&who_pays_fee,869			fee,870			if tip.is_zero() {871				WithdrawReason::TransactionPayment.into()872			} else {873				WithdrawReason::TransactionPayment | WithdrawReason::Tip874			},875			ExistenceRequirement::KeepAlive,876		) {877			Ok(imbalance) => Ok((fee, Some(imbalance))),878			Err(_) => Err(InvalidTransaction::Payment.into()),879		}880	}881}882883impl<T:Trait + transaction_payment::Trait + Send + Sync> SignedExtension for ChargeTransactionPayment<T> where884    BalanceOf<T>: Send + Sync + From<u64> + FixedPointOperand,885    T::Call: Dispatchable<Info=DispatchInfo, PostInfo=PostDispatchInfo> + IsSubType<Module<T>, T>,886{887	const IDENTIFIER: &'static str = "ChargeTransactionPayment";888	type AccountId = T::AccountId;889	type Call = T::Call;890	type AdditionalSigned = ();891	type Pre = (BalanceOf<T>, Self::AccountId, Option<NegativeImbalanceOf<T>>, BalanceOf<T>);892	fn additional_signed(&self) -> sp_std::result::Result<(), TransactionValidityError> { Ok(()) }893894	fn validate(895		&self,896		who: &Self::AccountId,897		call: &Self::Call,898		info: &DispatchInfoOf<Self::Call>,899		len: usize,900	) -> TransactionValidity {901		let (fee, _) = self.withdraw_fee(who, call, info, len)?;902903		let mut r = ValidTransaction::default();904		// NOTE: we probably want to maximize the _fee (of any type) per weight unit_ here, which905		// will be a bit more than setting the priority to tip. For now, this is enough.906		r.priority = fee.saturated_into::<TransactionPriority>();907		Ok(r)908	}909910	fn pre_dispatch(911		self,912		who: &Self::AccountId,913		call: &Self::Call,914		info: &DispatchInfoOf<Self::Call>,915		len: usize916	) -> Result<Self::Pre, TransactionValidityError> {917		let (fee, imbalance) = self.withdraw_fee(who, call, info, len)?;918		Ok((self.0, who.clone(), imbalance, fee))919	}920921	fn post_dispatch(922		pre: Self::Pre,923		info: &DispatchInfoOf<Self::Call>,924		post_info: &PostDispatchInfoOf<Self::Call>,925		len: usize,926		_result: &DispatchResult,927	) -> Result<(), TransactionValidityError> {928		let (tip, who, imbalance, fee) = pre;929		if let Some(payed) = imbalance {930			let actual_fee = <transaction_payment::Module<T>>::compute_actual_fee(931				len as u32,932				info,933				post_info,934				tip,935			);936			let refund = fee.saturating_sub(actual_fee);937			let actual_payment = match <T as transaction_payment::Trait>::Currency::deposit_into_existing(&who, refund) {938				Ok(refund_imbalance) => {939					// The refund cannot be larger than the up front payed max weight.940					// `PostDispatchInfo::calc_unspent` guards against such a case.941					match payed.offset(refund_imbalance) {942						Ok(actual_payment) => actual_payment,943						Err(_) => return Err(InvalidTransaction::Payment.into()),944					}945				}946				// We do not recreate the account using the refund. The up front payment947				// is gone in that case.948				Err(_) => payed,949			};950			let imbalances = actual_payment.split(tip);951			<T as transaction_payment::Trait>::OnTransactionPayment::on_unbalanceds(Some(imbalances.0).into_iter()952				.chain(Some(imbalances.1)));953		}954		Ok(())955	}956}