difftreelog
NFTPAR-183: Storage Refactoring. Reduced collection id and token id from u64 to u32.
in: master
2 files changed
pallets/nft/src/lib.rsdiffbeforeafterboth1#![cfg_attr(not(feature = "std"), no_std)]23#[cfg(feature = "std")]4pub use std::*;56#[cfg(feature = "std")]7pub use serde::*;89use codec::{Decode, Encode};10pub use frame_support::{11 construct_runtime, decl_event, decl_module, decl_storage, decl_error,12 dispatch::DispatchResult,13 ensure, fail, parameter_types,14 traits::{15 Currency, ExistenceRequirement, Get, Imbalance, KeyOwnerProofSystem, OnUnbalanced,16 Randomness, WithdrawReason,17 },18 weights::{19 constants::{BlockExecutionWeight, ExtrinsicBaseWeight, RocksDbWeight, WEIGHT_PER_SECOND},20 DispatchInfo, GetDispatchInfo, IdentityFee, Pays, PostDispatchInfo, Weight,21 WeightToFeePolynomial,22 },23 IsSubType, StorageValue,24};2526use frame_system::{self as system, ensure_signed, ensure_root};27use sp_runtime::sp_std::prelude::Vec;28use sp_runtime::{29 traits::{30 DispatchInfoOf, Dispatchable, PostDispatchInfoOf, Saturating, SignedExtension, Zero,31 },32 transaction_validity::{33 InvalidTransaction, TransactionValidity, TransactionValidityError, ValidTransaction,34 },35 FixedPointOperand, FixedU128,36};37use pallet_contracts::ContractAddressFor;38use sp_runtime::traits::StaticLookup;3940#[cfg(test)]41mod mock;4243#[cfg(test)]44mod tests;4546mod default_weights;4748// Structs49// #region5051#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq)]52#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]53pub enum CollectionMode {54 Invalid,55 NFT,56 // decimal points57 Fungible(u32),58 // decimal points59 ReFungible(u32),60}6162impl Into<u8> for CollectionMode {63 fn into(self) -> u8 {64 match self {65 CollectionMode::Invalid => 0,66 CollectionMode::NFT => 1,67 CollectionMode::Fungible(_) => 2,68 CollectionMode::ReFungible(_) => 3,69 }70 }71}7273#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq)]74#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]75pub enum AccessMode {76 Normal,77 WhiteList,78}79impl Default for AccessMode {80 fn default() -> Self {81 Self::Normal82 }83}8485impl Default for CollectionMode {86 fn default() -> Self {87 Self::Invalid88 }89}9091#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]92#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]93pub struct Ownership<AccountId> {94 pub owner: AccountId,95 pub fraction: u128,96}9798#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]99#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]100pub struct CollectionType<AccountId> {101 pub owner: AccountId,102 pub mode: CollectionMode,103 pub access: AccessMode,104 pub decimal_points: u32,105 pub name: Vec<u16>, // 64 include null escape char106 pub description: Vec<u16>, // 256 include null escape char107 pub token_prefix: Vec<u8>, // 16 include null escape char108 pub mint_mode: bool,109 pub offchain_schema: Vec<u8>,110 pub sponsor: AccountId, // Who pays fees. If set to default address, the fees are applied to the transaction sender111 pub unconfirmed_sponsor: AccountId, // Sponsor address that has not yet confirmed sponsorship112 pub variable_on_chain_schema: Vec<u8>, //113 pub const_on_chain_schema: Vec<u8>, //114}115116#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]117#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]118pub struct NftItemType<AccountId> {119 pub collection: u64,120 pub owner: AccountId,121 pub const_data: Vec<u8>,122 pub variable_data: Vec<u8>,123}124125#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]126#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]127pub struct FungibleItemType<AccountId> {128 pub collection: u64,129 pub owner: AccountId,130 pub value: u128,131}132133#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]134#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]135pub struct ReFungibleItemType<AccountId> {136 pub collection: u64,137 pub owner: Vec<Ownership<AccountId>>,138 pub const_data: Vec<u8>,139 pub variable_data: Vec<u8>,140}141142#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]143#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]144pub struct ApprovePermissions<AccountId> {145 pub approved: AccountId,146 pub amount: u64,147}148149#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]150#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]151pub struct VestingItem<AccountId, Moment> {152 pub sender: AccountId,153 pub recipient: AccountId,154 pub collection_id: u64,155 pub item_id: u64,156 pub amount: u64,157 pub vesting_date: Moment,158}159160#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]161#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]162pub struct BasketItem<AccountId, BlockNumber> {163 pub address: AccountId,164 pub start_block: BlockNumber,165}166167#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]168#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]169pub struct ChainLimits {170 pub collection_numbers_limit: u64,171 pub account_token_ownership_limit: u64,172 pub collections_admins_limit: u64,173 pub custom_data_limit: u32,174175 // Timeouts for item types in passed blocks176 pub nft_sponsor_transfer_timeout: u32,177 pub fungible_sponsor_transfer_timeout: u32,178 pub refungible_sponsor_transfer_timeout: u32,179}180181pub trait WeightInfo {182 fn create_collection() -> Weight;183 fn destroy_collection() -> Weight;184 fn add_to_white_list() -> Weight;185 fn remove_from_white_list() -> Weight;186 fn set_public_access_mode() -> Weight;187 fn set_mint_permission() -> Weight;188 fn change_collection_owner() -> Weight;189 fn add_collection_admin() -> Weight;190 fn remove_collection_admin() -> Weight;191 fn set_collection_sponsor() -> Weight;192 fn confirm_sponsorship() -> Weight;193 fn remove_collection_sponsor() -> Weight;194 fn create_item(s: usize) -> Weight;195 fn burn_item() -> Weight;196 fn transfer() -> Weight;197 fn approve() -> Weight;198 fn transfer_from() -> Weight;199 fn set_offchain_schema() -> Weight;200 fn set_const_on_chain_schema() -> Weight;201 fn set_variable_on_chain_schema() -> Weight;202 fn set_variable_meta_data() -> Weight;203 fn enable_contract_sponsoring() -> Weight;204}205206#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]207#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]208pub struct CreateNftData {209 pub const_data: Vec<u8>,210 pub variable_data: Vec<u8>,211}212213#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]214#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]215pub struct CreateFungibleData {216}217218#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]219#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]220pub struct CreateReFungibleData {221 pub const_data: Vec<u8>,222 pub variable_data: Vec<u8>,223}224225#[derive(Encode, Decode, Debug, Clone, PartialEq)]226#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]227pub enum CreateItemData {228 NFT(CreateNftData),229 Fungible(CreateFungibleData),230 ReFungible(CreateReFungibleData)231}232233impl CreateItemData {234 pub fn len(&self) -> usize {235 let len = match self {236 CreateItemData::NFT(data) => data.variable_data.len() + data.const_data.len(),237 CreateItemData::ReFungible(data) => data.variable_data.len() + data.const_data.len(),238 _ => 0239 };240 241 return len;242 }243}244245impl From<CreateNftData> for CreateItemData {246 fn from(item: CreateNftData) -> Self {247 CreateItemData::NFT(item)248 }249}250251impl From<CreateReFungibleData> for CreateItemData {252 fn from(item: CreateReFungibleData) -> Self {253 CreateItemData::ReFungible(item)254 }255}256257impl From<CreateFungibleData> for CreateItemData {258 fn from(item: CreateFungibleData) -> Self {259 CreateItemData::Fungible(item)260 }261}262263264decl_error! {265 /// Error for non-fungible-token module.266 pub enum Error for Module<T: Trait> {267 /// Total collections bound exceeded.268 TotalCollectionsLimitExceeded,269 /// Decimal_points parameter must be lower than 4.270 CollectionDecimalPointLimitExceeded, 271 /// Collection name can not be longer than 63 char.272 CollectionNameLimitExceeded, 273 /// Collection description can not be longer than 255 char.274 CollectionDescriptionLimitExceeded, 275 /// Token prefix can not be longer than 15 char.276 CollectionTokenPrefixLimitExceeded,277 /// This collection does not exist.278 CollectionNotFound,279 /// Item not exists.280 TokenNotFound,281 /// Arithmetic calculation overflow.282 NumOverflow, 283 /// Account already has admin role.284 AlreadyAdmin, 285 /// You do not own this collection.286 NoPermission,287 /// This address is not set as sponsor, use setCollectionSponsor first.288 ConfirmUnsetSponsorFail,289 /// Collection is not in mint mode.290 PublicMintingNotAllowed,291 /// Sender parameter and item owner must be equal.292 MustBeTokenOwner,293 /// Item balance not enough.294 TokenValueTooLow,295 /// Size of item is too large.296 NftSizeLimitExceeded,297 /// No approve found298 ApproveNotFound,299 /// Requested value more than approved.300 TokenValueNotEnough,301 /// Only approved addresses can call this method.302 ApproveRequired,303 /// Address is not in white list.304 AddresNotInWhiteList,305 /// Number of collection admins bound exceeded.306 CollectionAdminsLimitExceeded,307 /// Owned tokens by a single address bound exceeded.308 AddressOwnershipLimitExceeded,309 /// Length of items properties must be greater than 0.310 EmptyArgument,311 /// const_data exceeded data limit.312 TokenConstDataLimitExceeded,313 /// variable_data exceeded data limit.314 TokenVariableDataLimitExceeded,315 /// Not NFT item data used to mint in NFT collection.316 NotNftDataUsedToMintNftCollectionToken,317 /// Not Fungible item data used to mint in Fungible collection.318 NotFungibleDataUsedToMintFungibleCollectionToken,319 /// Not Re Fungible item data used to mint in Re Fungible collection.320 NotReFungibleDataUsedToMintReFungibleCollectionToken,321 /// Unexpected collection type.322 UnexpectedCollectionType,323 /// Can't store metadata in fungible tokens.324 CantStoreMetadataInFungibleTokens325 }326}327328pub trait Trait: system::Trait + Sized + transaction_payment::Trait + pallet_contracts::Trait {329 type Event: From<Event<Self>> + Into<<Self as system::Trait>::Event>;330331 /// Weight information for extrinsics in this pallet.332 type WeightInfo: WeightInfo;333}334335#[cfg(feature = "runtime-benchmarks")]336mod benchmarking;337338// #endregion339340decl_storage! {341 trait Store for Module<T: Trait> as Nft {342343 // Private members344 NextCollectionID: u64;345 CreatedCollectionCount: u64;346 ChainVersion: u64;347 ItemListIndex: map hasher(blake2_128_concat) u64 => u64;348349 // Chain limits struct350 pub ChainLimit get(fn chain_limit) config(): ChainLimits;351352 // Bound counters353 CollectionCount: u64;354 pub AccountItemCount get(fn account_item_count): map hasher(identity) T::AccountId => u64;355356 // Basic collections357 pub Collection get(fn collection) config(): map hasher(identity) u64 => CollectionType<T::AccountId>;358 pub AdminList get(fn admin_list_collection): map hasher(identity) u64 => Vec<T::AccountId>;359 pub WhiteList get(fn white_list): map hasher(identity) u64 => Vec<T::AccountId>;360361 /// Balance owner per collection map362 pub Balance get(fn balance_count): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) T::AccountId => u64;363364 /// second parameter: item id + owner account id365 pub ApprovedList get(fn approved): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) (u64, T::AccountId) => Vec<ApprovePermissions<T::AccountId>>;366367 /// Item collections368 pub NftItemList get(fn nft_item_id) config(): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => NftItemType<T::AccountId>;369 pub FungibleItemList get(fn fungible_item_id) config(): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => FungibleItemType<T::AccountId>;370 pub ReFungibleItemList get(fn refungible_item_id) config(): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => ReFungibleItemType<T::AccountId>;371372 /// Index list373 pub AddressTokens get(fn address_tokens): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) T::AccountId => Vec<u64>;374375 /// Tokens transfer baskets376 pub NftTransferBasket get(fn nft_transfer_basket): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => T::BlockNumber;377 pub FungibleTransferBasket get(fn fungible_transfer_basket): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => Vec<BasketItem<T::AccountId, T::BlockNumber>>;378 pub ReFungibleTransferBasket get(fn refungible_transfer_basket): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => T::BlockNumber;379380 // Contract Sponsorship and Ownership381 pub ContractOwner get(fn contract_owner): map hasher(twox_64_concat) T::AccountId => T::AccountId;382 pub ContractSelfSponsoring get(fn contract_self_sponsoring): map hasher(twox_64_concat) T::AccountId => bool;383 pub ContractSponsorBasket get(fn contract_sponsor_basket): map hasher(twox_64_concat) T::AccountId => T::BlockNumber;384 pub ContractSponsoringRateLimit get(fn contract_sponsoring_rate_limit): map hasher(twox_64_concat) T::AccountId => T::BlockNumber;385 }386 add_extra_genesis {387 build(|config: &GenesisConfig<T>| {388 // Modification of storage389 for (_num, _c) in &config.collection {390 <Module<T>>::init_collection(_c);391 }392393 for (_num, _q, _i) in &config.nft_item_id {394 <Module<T>>::init_nft_token(_i);395 }396397 for (_num, _q, _i) in &config.fungible_item_id {398 <Module<T>>::init_fungible_token(_i);399 }400401 for (_num, _q, _i) in &config.refungible_item_id {402 <Module<T>>::init_refungible_token(_i);403 }404 })405 }406}407408decl_event!(409 pub enum Event<T>410 where411 AccountId = <T as system::Trait>::AccountId,412 {413 /// New collection was created414 /// 415 /// # Arguments416 /// 417 /// * collection_id: Globally unique identifier of newly created collection.418 /// 419 /// * mode: [CollectionMode] converted into u8.420 /// 421 /// * account_id: Collection owner.422 Created(u64, u8, AccountId),423424 /// New item was created.425 /// 426 /// # Arguments427 /// 428 /// * collection_id: Id of the collection where item was created.429 /// 430 /// * item_id: Id of an item. Unique within the collection.431 ItemCreated(u64, u64),432433 /// Collection item was burned.434 /// 435 /// # Arguments436 /// 437 /// collection_id.438 /// 439 /// item_id: Identifier of burned NFT.440 ItemDestroyed(u64, u64),441 }442);443444decl_module! {445 pub struct Module<T: Trait> for enum Call where origin: T::Origin {446447 fn deposit_event() = default;448 type Error = Error<T>;449450 fn on_initialize(now: T::BlockNumber) -> Weight {451452 if ChainVersion::get() < 2453 {454 let value = NextCollectionID::get();455 CreatedCollectionCount::put(value);456 ChainVersion::put(2);457 }458459 0460 }461462 /// This method creates a Collection of NFTs. Each Token may have multiple properties encoded as an array of bytes of certain length. The initial owner and admin of the collection are set to the address that signed the transaction. Both addresses can be changed later.463 /// 464 /// # Permissions465 /// 466 /// * Anyone.467 /// 468 /// # Arguments469 /// 470 /// * collection_name: UTF-16 string with collection name (limit 64 characters), will be stored as zero-terminated.471 /// 472 /// * collection_description: UTF-16 string with collection description (limit 256 characters), will be stored as zero-terminated.473 /// 474 /// * token_prefix: UTF-8 string with token prefix.475 /// 476 /// * mode: [CollectionMode] collection type and type dependent data.477 // returns collection ID478 #[weight = T::WeightInfo::create_collection()]479 pub fn create_collection(origin,480 collection_name: Vec<u16>,481 collection_description: Vec<u16>,482 token_prefix: Vec<u8>,483 mode: CollectionMode) -> DispatchResult {484485 // Anyone can create a collection486 let who = ensure_signed(origin)?;487488 let decimal_points = match mode {489 CollectionMode::Fungible(points) => points,490 CollectionMode::ReFungible(points) => points,491 _ => 0492 };493494 // bound Total number of collections495 ensure!(CollectionCount::get() < ChainLimit::get().collection_numbers_limit, Error::<T>::TotalCollectionsLimitExceeded);496497 // check params498 ensure!(decimal_points <= 4, Error::<T>::CollectionDecimalPointLimitExceeded);499500 let mut name = collection_name.to_vec();501 name.push(0);502 ensure!(name.len() <= 64, Error::<T>::CollectionNameLimitExceeded);503504 let mut description = collection_description.to_vec();505 description.push(0);506 ensure!(name.len() <= 256, Error::<T>::CollectionDescriptionLimitExceeded);507508 let mut prefix = token_prefix.to_vec();509 prefix.push(0);510 ensure!(prefix.len() <= 16, Error::<T>::CollectionTokenPrefixLimitExceeded);511512 // Generate next collection ID513 let next_id = CreatedCollectionCount::get()514 .checked_add(1)515 .ok_or(Error::<T>::NumOverflow)?;516517 // bound counter518 let total = CollectionCount::get()519 .checked_add(1)520 .ok_or(Error::<T>::NumOverflow)?;521522 CreatedCollectionCount::put(next_id);523 CollectionCount::put(total);524525 // Create new collection526 let new_collection = CollectionType {527 owner: who.clone(),528 name: name,529 mode: mode.clone(),530 mint_mode: false,531 access: AccessMode::Normal,532 description: description,533 decimal_points: decimal_points,534 token_prefix: prefix,535 offchain_schema: Vec::new(),536 sponsor: T::AccountId::default(),537 unconfirmed_sponsor: T::AccountId::default(),538 variable_on_chain_schema: Vec::new(),539 const_on_chain_schema: Vec::new(),540 };541542 // Add new collection to map543 <Collection<T>>::insert(next_id, new_collection);544545 // call event546 Self::deposit_event(RawEvent::Created(next_id, mode.into(), who.clone()));547548 Ok(())549 }550551 /// **DANGEROUS**: Destroys collection and all NFTs within this collection. Users irrecoverably lose their assets and may lose real money.552 /// 553 /// # Permissions554 /// 555 /// * Collection Owner.556 /// 557 /// # Arguments558 /// 559 /// * collection_id: collection to destroy.560 #[weight = T::WeightInfo::destroy_collection()]561 pub fn destroy_collection(origin, collection_id: u64) -> DispatchResult {562563 let sender = ensure_signed(origin)?;564 Self::check_owner_permissions(collection_id, sender)?;565566 <AddressTokens<T>>::remove_prefix(collection_id);567 <ApprovedList<T>>::remove_prefix(collection_id);568 <Balance<T>>::remove_prefix(collection_id);569 <ItemListIndex>::remove(collection_id);570 <AdminList<T>>::remove(collection_id);571 <Collection<T>>::remove(collection_id);572 <WhiteList<T>>::remove(collection_id);573574 <NftItemList<T>>::remove_prefix(collection_id);575 <FungibleItemList<T>>::remove_prefix(collection_id);576 <ReFungibleItemList<T>>::remove_prefix(collection_id);577578 <NftTransferBasket<T>>::remove_prefix(collection_id);579 <FungibleTransferBasket<T>>::remove_prefix(collection_id);580 <ReFungibleTransferBasket<T>>::remove_prefix(collection_id);581582 if CollectionCount::get() > 0583 {584 // bound couter585 let total = CollectionCount::get()586 .checked_sub(1)587 .ok_or(Error::<T>::NumOverflow)?;588589 CollectionCount::put(total);590 }591592 Ok(())593 }594595 /// Add an address to white list.596 /// 597 /// # Permissions598 /// 599 /// * Collection Owner600 /// * Collection Admin601 /// 602 /// # Arguments603 /// 604 /// * collection_id.605 /// 606 /// * address.607 #[weight = T::WeightInfo::add_to_white_list()]608 pub fn add_to_white_list(origin, collection_id: u64, address: T::AccountId) -> DispatchResult{609610 let sender = ensure_signed(origin)?;611 Self::check_owner_or_admin_permissions(collection_id, sender)?;612613 let mut white_list_collection: Vec<T::AccountId>;614 if <WhiteList<T>>::contains_key(collection_id) {615 white_list_collection = <WhiteList<T>>::get(collection_id);616 if !white_list_collection.contains(&address.clone())617 {618 white_list_collection.push(address.clone());619 }620 }621 else {622 white_list_collection = Vec::new();623 white_list_collection.push(address.clone());624 }625626 <WhiteList<T>>::insert(collection_id, white_list_collection);627 Ok(())628 }629630 /// Remove an address from white list.631 /// 632 /// # Permissions633 /// 634 /// * Collection Owner635 /// * Collection Admin636 /// 637 /// # Arguments638 /// 639 /// * collection_id.640 /// 641 /// * address.642 #[weight = T::WeightInfo::remove_from_white_list()]643 pub fn remove_from_white_list(origin, collection_id: u64, address: T::AccountId) -> DispatchResult{644645 let sender = ensure_signed(origin)?;646 Self::check_owner_or_admin_permissions(collection_id, sender)?;647648 if <WhiteList<T>>::contains_key(collection_id) {649 let mut white_list_collection = <WhiteList<T>>::get(collection_id);650 if white_list_collection.contains(&address.clone())651 {652 white_list_collection.retain(|i| *i != address.clone());653 <WhiteList<T>>::insert(collection_id, white_list_collection);654 }655 }656657 Ok(())658 }659660 /// Toggle between normal and white list access for the methods with access for `Anyone`.661 /// 662 /// # Permissions663 /// 664 /// * Collection Owner.665 /// 666 /// # Arguments667 /// 668 /// * collection_id.669 /// 670 /// * mode: [AccessMode]671 #[weight = T::WeightInfo::set_public_access_mode()]672 pub fn set_public_access_mode(origin, collection_id: u64, mode: AccessMode) -> DispatchResult673 {674 let sender = ensure_signed(origin)?;675676 Self::check_owner_permissions(collection_id, sender)?;677 let mut target_collection = <Collection<T>>::get(collection_id);678 target_collection.access = mode;679 <Collection<T>>::insert(collection_id, target_collection);680681 Ok(())682 }683684 /// Allows Anyone to create tokens if:685 /// * White List is enabled, and686 /// * Address is added to white list, and687 /// * This method was called with True parameter688 /// 689 /// # Permissions690 /// * Collection Owner691 ///692 /// # Arguments693 /// 694 /// * collection_id.695 /// 696 /// * mint_permission: Boolean parameter. If True, allows minting to Anyone with conditions above.697 #[weight = T::WeightInfo::set_mint_permission()]698 pub fn set_mint_permission(origin, collection_id: u64, mint_permission: bool) -> DispatchResult699 {700 let sender = ensure_signed(origin)?;701702 Self::check_owner_permissions(collection_id, sender)?;703 let mut target_collection = <Collection<T>>::get(collection_id);704 target_collection.mint_mode = mint_permission;705 <Collection<T>>::insert(collection_id, target_collection);706707 Ok(())708 }709710 /// Change the owner of the collection.711 /// 712 /// # Permissions713 /// 714 /// * Collection Owner.715 /// 716 /// # Arguments717 /// 718 /// * collection_id.719 /// 720 /// * new_owner.721 #[weight = T::WeightInfo::change_collection_owner()]722 pub fn change_collection_owner(origin, collection_id: u64, new_owner: T::AccountId) -> DispatchResult {723724 let sender = ensure_signed(origin)?;725 Self::check_owner_permissions(collection_id, sender)?;726 let mut target_collection = <Collection<T>>::get(collection_id);727 target_collection.owner = new_owner;728 <Collection<T>>::insert(collection_id, target_collection);729730 Ok(())731 }732733 /// Adds an admin of the Collection.734 /// NFT Collection can be controlled by multiple admin addresses (some which can also be servers, for example). Admins can issue and burn NFTs, as well as add and remove other admins, but cannot change NFT or Collection ownership. 735 /// 736 /// # Permissions737 /// 738 /// * Collection Owner.739 /// * Collection Admin.740 /// 741 /// # Arguments742 /// 743 /// * collection_id: ID of the Collection to add admin for.744 /// 745 /// * new_admin_id: Address of new admin to add.746 #[weight = T::WeightInfo::add_collection_admin()]747 pub fn add_collection_admin(origin, collection_id: u64, new_admin_id: T::AccountId) -> DispatchResult {748749 let sender = ensure_signed(origin)?;750 Self::check_owner_or_admin_permissions(collection_id, sender)?;751 let mut admin_arr: Vec<T::AccountId> = Vec::new();752753 if <AdminList<T>>::contains_key(collection_id)754 {755 admin_arr = <AdminList<T>>::get(collection_id);756 ensure!(!admin_arr.contains(&new_admin_id), Error::<T>::AlreadyAdmin);757 }758759 // Number of collection admins760 ensure!((admin_arr.len() as u64) < ChainLimit::get().collections_admins_limit, Error::<T>::CollectionAdminsLimitExceeded);761762 admin_arr.push(new_admin_id);763 <AdminList<T>>::insert(collection_id, admin_arr);764765 Ok(())766 }767768 /// Remove admin address of the Collection. An admin address can remove itself. List of admins may become empty, in which case only Collection Owner will be able to add an Admin.769 ///770 /// # Permissions771 /// 772 /// * Collection Owner.773 /// * Collection Admin.774 /// 775 /// # Arguments776 /// 777 /// * collection_id: ID of the Collection to remove admin for.778 /// 779 /// * account_id: Address of admin to remove.780 #[weight = T::WeightInfo::remove_collection_admin()]781 pub fn remove_collection_admin(origin, collection_id: u64, account_id: T::AccountId) -> DispatchResult {782783 let sender = ensure_signed(origin)?;784 Self::check_owner_or_admin_permissions(collection_id, sender)?;785786 if <AdminList<T>>::contains_key(collection_id)787 {788 let mut admin_arr = <AdminList<T>>::get(collection_id);789 admin_arr.retain(|i| *i != account_id);790 <AdminList<T>>::insert(collection_id, admin_arr);791 }792793 Ok(())794 }795796 /// # Permissions797 /// 798 /// * Collection Owner799 /// 800 /// # Arguments801 /// 802 /// * collection_id.803 /// 804 /// * new_sponsor.805 #[weight = T::WeightInfo::set_collection_sponsor()]806 pub fn set_collection_sponsor(origin, collection_id: u64, new_sponsor: T::AccountId) -> DispatchResult {807808 let sender = ensure_signed(origin)?;809 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);810811 let mut target_collection = <Collection<T>>::get(collection_id);812 ensure!(sender == target_collection.owner, Error::<T>::NoPermission);813814 target_collection.unconfirmed_sponsor = new_sponsor;815 <Collection<T>>::insert(collection_id, target_collection);816817 Ok(())818 }819820 /// # Permissions821 /// 822 /// * Sponsor.823 /// 824 /// # Arguments825 /// 826 /// * collection_id.827 #[weight = T::WeightInfo::confirm_sponsorship()]828 pub fn confirm_sponsorship(origin, collection_id: u64) -> DispatchResult {829830 let sender = ensure_signed(origin)?;831 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);832833 let mut target_collection = <Collection<T>>::get(collection_id);834 ensure!(sender == target_collection.unconfirmed_sponsor, Error::<T>::ConfirmUnsetSponsorFail);835836 target_collection.sponsor = target_collection.unconfirmed_sponsor;837 target_collection.unconfirmed_sponsor = T::AccountId::default();838 <Collection<T>>::insert(collection_id, target_collection);839840 Ok(())841 }842843 /// Switch back to pay-per-own-transaction model.844 ///845 /// # Permissions846 ///847 /// * Collection owner.848 /// 849 /// # Arguments850 /// 851 /// * collection_id.852 #[weight = T::WeightInfo::remove_collection_sponsor()]853 pub fn remove_collection_sponsor(origin, collection_id: u64) -> DispatchResult {854855 let sender = ensure_signed(origin)?;856 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);857858 let mut target_collection = <Collection<T>>::get(collection_id);859 ensure!(sender == target_collection.owner, Error::<T>::NoPermission);860861 target_collection.sponsor = T::AccountId::default();862 <Collection<T>>::insert(collection_id, target_collection);863864 Ok(())865 }866867 /// This method creates a concrete instance of NFT Collection created with CreateCollection method.868 /// 869 /// # Permissions870 /// 871 /// * Collection Owner.872 /// * Collection Admin.873 /// * Anyone if874 /// * White List is enabled, and875 /// * Address is added to white list, and876 /// * MintPermission is enabled (see SetMintPermission method)877 /// 878 /// # Arguments879 /// 880 /// * collection_id: ID of the collection.881 /// 882 /// * owner: Address, initial owner of the NFT.883 ///884 /// * data: Token data to store on chain.885 // #[weight =886 // (130_000_000 as Weight)887 // .saturating_add((2135 as Weight).saturating_mul((properties.len() as u64) as Weight))888 // .saturating_add(RocksDbWeight::get().reads(10 as Weight))889 // .saturating_add(RocksDbWeight::get().writes(8 as Weight))]890891 #[weight = T::WeightInfo::create_item(data.len())]892 pub fn create_item(origin, collection_id: u64, owner: T::AccountId, data: CreateItemData) -> DispatchResult {893894 let sender = ensure_signed(origin)?;895896 Self::collection_exists(collection_id)?;897898 let target_collection = <Collection<T>>::get(collection_id);899900 Self::can_create_items_in_collection(collection_id, &target_collection, &sender, &owner)?;901 Self::validate_create_item_args(&target_collection, &data)?;902 Self::create_item_no_validation(collection_id, &target_collection, owner, data)?;903904 Ok(())905 }906907 /// This method creates multiple instances of NFT Collection created with CreateCollection method.908 /// 909 /// # Permissions910 /// 911 /// * Collection Owner.912 /// * Collection Admin.913 /// * Anyone if914 /// * White List is enabled, and915 /// * Address is added to white list, and916 /// * MintPermission is enabled (see SetMintPermission method)917 /// 918 /// # Arguments919 /// 920 /// * collection_id: ID of the collection.921 /// 922 /// * itemsData: Array items properties. Each property is an array of bytes itself, see [create_item].923 /// 924 /// * owner: Address, initial owner of the NFT.925 #[weight = T::WeightInfo::create_item(items_data.into_iter()926 .map(|data| { data.len() })927 .sum())]928 pub fn create_multiple_items(origin, collection_id: u64, owner: T::AccountId, items_data: Vec<CreateItemData>) -> DispatchResult {929930 ensure!(items_data.len() > 0, Error::<T>::EmptyArgument);931 let sender = ensure_signed(origin)?;932933 Self::collection_exists(collection_id)?;934 let target_collection = <Collection<T>>::get(collection_id);935936 Self::can_create_items_in_collection(collection_id, &target_collection, &sender, &owner)?;937938 for data in &items_data {939 Self::validate_create_item_args(&target_collection, data)?;940 }941 for data in &items_data {942 Self::create_item_no_validation(collection_id, &target_collection, owner.clone(), data.clone())?;943 }944945 Ok(())946 }947948 /// Destroys a concrete instance of NFT.949 /// 950 /// # Permissions951 /// 952 /// * Collection Owner.953 /// * Collection Admin.954 /// * Current NFT Owner.955 /// 956 /// # Arguments957 /// 958 /// * collection_id: ID of the collection.959 /// 960 /// * item_id: ID of NFT to burn.961 #[weight = T::WeightInfo::burn_item()]962 pub fn burn_item(origin, collection_id: u64, item_id: u64) -> DispatchResult {963964 let sender = ensure_signed(origin)?;965 Self::collection_exists(collection_id)?;966967 // Transfer permissions check968 let target_collection = <Collection<T>>::get(collection_id);969 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||970 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),971 Error::<T>::NoPermission);972973 if target_collection.access == AccessMode::WhiteList {974 Self::check_white_list(collection_id, &sender)?;975 }976977 match target_collection.mode978 {979 CollectionMode::NFT => Self::burn_nft_item(collection_id, item_id)?,980 CollectionMode::Fungible(_) => Self::burn_fungible_item(collection_id, item_id)?,981 CollectionMode::ReFungible(_) => Self::burn_refungible_item(collection_id, item_id, sender.clone())?,982 _ => ()983 };984985 // call event986 Self::deposit_event(RawEvent::ItemDestroyed(collection_id, item_id));987988 Ok(())989 }990991 /// Change ownership of the token.992 /// 993 /// # Permissions994 /// 995 /// * Collection Owner996 /// * Collection Admin997 /// * Current NFT owner998 ///999 /// # Arguments1000 /// 1001 /// * recipient: Address of token recipient.1002 /// 1003 /// * collection_id.1004 /// 1005 /// * item_id: ID of the item1006 /// * Non-Fungible Mode: Required.1007 /// * Fungible Mode: Ignored.1008 /// * Re-Fungible Mode: Required.1009 /// 1010 /// * value: Amount to transfer.1011 /// * Non-Fungible Mode: Ignored1012 /// * Fungible Mode: Must specify transferred amount1013 /// * Re-Fungible Mode: Must specify transferred portion (between 0 and 1)1014 #[weight = T::WeightInfo::transfer()]1015 pub fn transfer(origin, recipient: T::AccountId, collection_id: u64, item_id: u64, value: u64) -> DispatchResult {10161017 let sender = ensure_signed(origin)?;10181019 // Transfer permissions check1020 let target_collection = <Collection<T>>::get(collection_id);1021 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1022 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1023 Error::<T>::NoPermission);10241025 if target_collection.access == AccessMode::WhiteList {1026 Self::check_white_list(collection_id, &sender)?;1027 Self::check_white_list(collection_id, &recipient)?;1028 }10291030 match target_collection.mode1031 {1032 CollectionMode::NFT => Self::transfer_nft(collection_id, item_id, sender.clone(), recipient)?,1033 CollectionMode::Fungible(_) => Self::transfer_fungible(collection_id, item_id, value, sender.clone(), recipient)?,1034 CollectionMode::ReFungible(_) => Self::transfer_refungible(collection_id, item_id, value, sender.clone(), recipient)?,1035 _ => ()1036 };10371038 Ok(())1039 }10401041 /// Set, change, or remove approved address to transfer the ownership of the NFT.1042 /// 1043 /// # Permissions1044 /// 1045 /// * Collection Owner1046 /// * Collection Admin1047 /// * Current NFT owner1048 /// 1049 /// # Arguments1050 /// 1051 /// * approved: Address that is approved to transfer this NFT or zero (if needed to remove approval).1052 /// 1053 /// * collection_id.1054 /// 1055 /// * item_id: ID of the item.1056 #[weight = T::WeightInfo::approve()]1057 pub fn approve(origin, approved: T::AccountId, collection_id: u64, item_id: u64) -> DispatchResult {10581059 let sender = ensure_signed(origin)?;10601061 // Transfer permissions check1062 let target_collection = <Collection<T>>::get(collection_id);1063 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1064 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1065 Error::<T>::NoPermission);10661067 if target_collection.access == AccessMode::WhiteList {1068 Self::check_white_list(collection_id, &sender)?;1069 Self::check_white_list(collection_id, &approved)?;1070 }10711072 // amount param stub1073 let amount = 100000000;10741075 let list_exists = <ApprovedList<T>>::contains_key(collection_id, (item_id, sender.clone()));1076 if list_exists {10771078 let mut list = <ApprovedList<T>>::get(collection_id, (item_id, sender.clone()));1079 let item_contains = list.iter().any(|i| i.approved == approved);10801081 if !item_contains {1082 list.push(ApprovePermissions { approved: approved.clone(), amount: amount });1083 <ApprovedList<T>>::insert(collection_id, (item_id, sender.clone()), list);1084 }1085 } else {10861087 let mut list = Vec::new();1088 list.push(ApprovePermissions { approved: approved.clone(), amount: amount });1089 <ApprovedList<T>>::insert(collection_id, (item_id, sender.clone()), list);1090 }10911092 Ok(())1093 }1094 1095 /// Change ownership of a NFT on behalf of the owner. See Approve method for additional information. After this method executes, the approval is removed so that the approved address will not be able to transfer this NFT again from this owner.1096 /// 1097 /// # Permissions1098 /// * Collection Owner1099 /// * Collection Admin1100 /// * Current NFT owner1101 /// * Address approved by current NFT owner1102 /// 1103 /// # Arguments1104 /// 1105 /// * from: Address that owns token.1106 /// 1107 /// * recipient: Address of token recipient.1108 /// 1109 /// * collection_id.1110 /// 1111 /// * item_id: ID of the item.1112 /// 1113 /// * value: Amount to transfer.1114 #[weight = T::WeightInfo::transfer_from()]1115 pub fn transfer_from(origin, from: T::AccountId, recipient: T::AccountId, collection_id: u64, item_id: u64, value: u64 ) -> DispatchResult {11161117 let sender = ensure_signed(origin)?;1118 let mut appoved_transfer = false;11191120 // Check approve1121 if <ApprovedList<T>>::contains_key(collection_id, (item_id, from.clone())) {1122 let list_itm = <ApprovedList<T>>::get(collection_id, (item_id, from.clone()));1123 let opt_item = list_itm.iter().find(|i| i.approved == sender.clone());1124 if opt_item.is_some()1125 {1126 appoved_transfer = true;1127 ensure!(opt_item.unwrap().amount >= value, Error::<T>::TokenValueNotEnough);1128 }1129 }11301131 // Transfer permissions check1132 let target_collection = <Collection<T>>::get(collection_id);1133 ensure!(appoved_transfer || Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1134 Error::<T>::NoPermission);11351136 if target_collection.access == AccessMode::WhiteList {1137 Self::check_white_list(collection_id, &sender)?;1138 Self::check_white_list(collection_id, &recipient)?;1139 }11401141 // remove approve1142 let approve_list: Vec<ApprovePermissions<T::AccountId>> = <ApprovedList<T>>::get(collection_id, (item_id, from.clone()))1143 .into_iter().filter(|i| i.approved != sender.clone()).collect();1144 <ApprovedList<T>>::insert(collection_id, (item_id, from.clone()), approve_list);114511461147 match target_collection.mode1148 {1149 CollectionMode::NFT => Self::transfer_nft(collection_id, item_id, from, recipient)?,1150 CollectionMode::Fungible(_) => Self::transfer_fungible(collection_id, item_id, value, from.clone(), recipient)?,1151 CollectionMode::ReFungible(_) => Self::transfer_refungible(collection_id, item_id, value, from.clone(), recipient)?,1152 _ => ()1153 };11541155 Ok(())1156 }11571158 ///1159 #[weight = 0]1160 pub fn safe_transfer_from(origin, collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {11611162 // let no_perm_mes = "You do not have permissions to modify this collection";1163 // ensure!(<ApprovedList<T>>::contains_key((collection_id, item_id)), no_perm_mes);1164 // let list_itm = <ApprovedList<T>>::get((collection_id, item_id));1165 // ensure!(list_itm.contains(&new_owner.clone()), no_perm_mes);11661167 // // on_nft_received call11681169 // Self::transfer(origin, collection_id, item_id, new_owner)?;11701171 Ok(())1172 }11731174 /// Set off-chain data schema.1175 /// 1176 /// # Permissions1177 /// 1178 /// * Collection Owner1179 /// * Collection Admin1180 /// 1181 /// # Arguments1182 /// 1183 /// * collection_id.1184 /// 1185 /// * schema: String representing the offchain data schema.1186 #[weight = T::WeightInfo::set_variable_meta_data()]1187 pub fn set_variable_meta_data (1188 origin,1189 collection_id: u64,1190 item_id: u64,1191 data: Vec<u8>1192 ) -> DispatchResult {1193 let sender = ensure_signed(origin)?;1194 1195 Self::collection_exists(collection_id)?;1196 1197 ensure!(ChainLimit::get().custom_data_limit >= data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);11981199 // Modify permissions check1200 let target_collection = <Collection<T>>::get(collection_id);1201 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1202 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1203 Error::<T>::NoPermission);12041205 Self::item_exists(collection_id, item_id, &target_collection.mode)?;12061207 match target_collection.mode1208 {1209 CollectionMode::NFT => Self::set_nft_variable_data(collection_id, item_id, data)?,1210 CollectionMode::ReFungible(_) => Self::set_re_fungible_variable_data(collection_id, item_id, data)?,1211 CollectionMode::Fungible(_) => fail!(Error::<T>::CantStoreMetadataInFungibleTokens),1212 _ => fail!(Error::<T>::UnexpectedCollectionType)1213 };12141215 Ok(())1216 }1217 12181219 /// Set off-chain data schema.1220 /// 1221 /// # Permissions1222 /// 1223 /// * Collection Owner1224 /// * Collection Admin1225 /// 1226 /// # Arguments1227 /// 1228 /// * collection_id.1229 /// 1230 /// * schema: String representing the offchain data schema.1231 #[weight = T::WeightInfo::set_offchain_schema()]1232 pub fn set_offchain_schema(1233 origin,1234 collection_id: u64,1235 schema: Vec<u8>1236 ) -> DispatchResult {1237 let sender = ensure_signed(origin)?;1238 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;12391240 let mut target_collection = <Collection<T>>::get(collection_id);1241 target_collection.offchain_schema = schema;1242 <Collection<T>>::insert(collection_id, target_collection);12431244 Ok(())1245 }12461247 /// Set const on-chain data schema.1248 /// 1249 /// # Permissions1250 /// 1251 /// * Collection Owner1252 /// * Collection Admin1253 /// 1254 /// # Arguments1255 /// 1256 /// * collection_id.1257 /// 1258 /// * schema: String representing the const on-chain data schema.1259 #[weight = T::WeightInfo::set_const_on_chain_schema()]1260 pub fn set_const_on_chain_schema (1261 origin,1262 collection_id: u64,1263 schema: Vec<u8>1264 ) -> DispatchResult {1265 let sender = ensure_signed(origin)?;1266 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;12671268 let mut target_collection = <Collection<T>>::get(collection_id);1269 target_collection.const_on_chain_schema = schema;1270 <Collection<T>>::insert(collection_id, target_collection);12711272 Ok(())1273 }12741275 /// Set variable on-chain data schema.1276 /// 1277 /// # Permissions1278 /// 1279 /// * Collection Owner1280 /// * Collection Admin1281 /// 1282 /// # Arguments1283 /// 1284 /// * collection_id.1285 /// 1286 /// * schema: String representing the variable on-chain data schema.1287 #[weight = T::WeightInfo::set_const_on_chain_schema()]1288 pub fn set_variable_on_chain_schema (1289 origin,1290 collection_id: u64,1291 schema: Vec<u8>1292 ) -> DispatchResult {1293 let sender = ensure_signed(origin)?;1294 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;12951296 let mut target_collection = <Collection<T>>::get(collection_id);1297 target_collection.variable_on_chain_schema = schema;1298 <Collection<T>>::insert(collection_id, target_collection);12991300 Ok(())1301 }13021303 // Sudo permissions function1304 #[weight = 0]1305 pub fn set_chain_limits(1306 origin,1307 limits: ChainLimits1308 ) -> DispatchResult {1309 ensure_root(origin)?;1310 <ChainLimit>::put(limits);1311 Ok(())1312 }13131314 /// Enable smart contract self-sponsoring.1315 /// 1316 /// # Permissions1317 /// 1318 /// * Contract Owner1319 /// 1320 /// # Arguments1321 /// 1322 /// * contract address1323 /// * enable flag1324 /// 1325 #[weight = T::WeightInfo::enable_contract_sponsoring()]1326 pub fn enable_contract_sponsoring(1327 origin,1328 contract_address: T::AccountId,1329 enable: bool1330 ) -> DispatchResult {13311332 let sender = ensure_signed(origin)?;13331334 #[cfg(feature = "runtime-benchmarks")]1335 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());13361337 let mut is_owner = false;1338 if <ContractOwner<T>>::contains_key(contract_address.clone()) {1339 let owner = <ContractOwner<T>>::get(&contract_address);1340 is_owner = sender == owner;1341 }1342 ensure!(is_owner, Error::<T>::NoPermission);13431344 <ContractSelfSponsoring<T>>::insert(contract_address, enable);1345 Ok(())1346 }13471348 /// Set the rate limit for contract sponsoring to specified number of blocks.1349 /// 1350 /// If not set (has the default value of 0 blocks), the sponsoring will be disabled. 1351 /// If set to the number B (for blocks), the transactions will be sponsored with a rate 1352 /// limit of B, i.e. fees for every transaction sent to this smart contract will be paid 1353 /// from contract endowment if there are at least B blocks between such transactions. 1354 /// Nonetheless, if transactions are sent more frequently, the fees are paid by the sender.1355 /// 1356 /// # Permissions1357 /// 1358 /// * Contract Owner1359 /// 1360 /// # Arguments1361 /// 1362 /// -`contract_address`: Address of the contract to sponsor1363 /// -`rate_limit`: Number of blocks to wait until the next sponsored transaction is allowed1364 /// 1365 #[weight = 0]1366 pub fn set_contract_sponsoring_rate_limit(1367 origin,1368 contract_address: T::AccountId,1369 rate_limit: T::BlockNumber1370 ) -> DispatchResult {1371 let sender = ensure_signed(origin)?;1372 let mut is_owner = false;1373 if <ContractOwner<T>>::contains_key(contract_address.clone()) {1374 let owner = <ContractOwner<T>>::get(&contract_address);1375 is_owner = sender == owner;1376 }1377 ensure!(is_owner, Error::<T>::NoPermission);13781379 <ContractSponsoringRateLimit<T>>::insert(contract_address, rate_limit);1380 Ok(())1381 }13821383 // #[cfg(feature = "runtime-benchmarks")]1384 // #[weight = 0]1385 // pub fn add_contract_sponsoring_debug(1386 // origin,1387 // contract_address: T::AccountId, 1388 // owner: T::AccountId) -> DispatchResult {1389 // let sender = ensure_signed(origin)?;1390 // <ContractOwner<T>>::insert(contract_address.clone(), owner);1391 // Ok(())1392 // }1393 1394 }1395}13961397impl<T: Trait> Module<T> {13981399 fn can_create_items_in_collection(collection_id: u64, collection: &CollectionType<T::AccountId>, sender: &T::AccountId, owner: &T::AccountId) -> DispatchResult {14001401 if !Self::is_owner_or_admin_permissions(collection_id, sender.clone()) {1402 ensure!(collection.mint_mode == true, Error::<T>::PublicMintingNotAllowed);1403 Self::check_white_list(collection_id, owner)?;1404 Self::check_white_list(collection_id, sender)?;1405 }14061407 Ok(())1408 }14091410 fn validate_create_item_args(target_collection: &CollectionType<T::AccountId>, data: &CreateItemData) -> DispatchResult {1411 match target_collection.mode1412 {1413 CollectionMode::NFT => {1414 if let CreateItemData::NFT(data) = data {1415 // check sizes1416 ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, Error::<T>::TokenConstDataLimitExceeded);1417 ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);1418 } else {1419 fail!(Error::<T>::NotNftDataUsedToMintNftCollectionToken);1420 }1421 },1422 CollectionMode::Fungible(_) => {1423 if let CreateItemData::Fungible(_) = data {1424 } else {1425 fail!(Error::<T>::NotFungibleDataUsedToMintFungibleCollectionToken);1426 }1427 },1428 CollectionMode::ReFungible(_) => {1429 if let CreateItemData::ReFungible(data) = data {14301431 // check sizes1432 ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, Error::<T>::TokenConstDataLimitExceeded);1433 ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);1434 } else {1435 fail!(Error::<T>::NotReFungibleDataUsedToMintReFungibleCollectionToken);1436 }1437 },1438 _ => { fail!(Error::<T>::UnexpectedCollectionType); }1439 };14401441 Ok(())1442 }14431444 fn create_item_no_validation(collection_id: u64, collection: &CollectionType<T::AccountId>, owner: T::AccountId, data: CreateItemData) -> DispatchResult {1445 match data1446 {1447 CreateItemData::NFT(data) => {1448 let item = NftItemType {1449 collection: collection_id,1450 owner,1451 const_data: data.const_data,1452 variable_data: data.variable_data1453 };14541455 Self::add_nft_item(item)?;1456 },1457 CreateItemData::Fungible(_) => {1458 let item = FungibleItemType {1459 collection: collection_id,1460 owner,1461 value: (10 as u128).pow(collection.decimal_points)1462 };14631464 Self::add_fungible_item(item)?;1465 },1466 CreateItemData::ReFungible(data) => {1467 let mut owner_list = Vec::new();1468 let value = (10 as u128).pow(collection.decimal_points);1469 owner_list.push(Ownership {owner: owner.clone(), fraction: value});14701471 let item = ReFungibleItemType {1472 collection: collection_id,1473 owner: owner_list,1474 const_data: data.const_data,1475 variable_data: data.variable_data1476 };14771478 Self::add_refungible_item(item)?;1479 }1480 };148114821483 // call event1484 Self::deposit_event(RawEvent::ItemCreated(collection_id, <ItemListIndex>::get(collection_id)));14851486 Ok(())1487 }14881489 fn add_fungible_item(item: FungibleItemType<T::AccountId>) -> DispatchResult {1490 let current_index = <ItemListIndex>::get(item.collection)1491 .checked_add(1)1492 .ok_or(Error::<T>::NumOverflow)?;1493 let itemcopy = item.clone();1494 let owner = item.owner.clone();1495 let value = item.value as u64;14961497 Self::add_token_index(item.collection, current_index, owner.clone())?;14981499 <ItemListIndex>::insert(item.collection, current_index);1500 <FungibleItemList<T>>::insert(item.collection, current_index, itemcopy);15011502 // Add current block1503 let v: Vec<BasketItem<T::AccountId, T::BlockNumber>> = Vec::new();1504 <FungibleTransferBasket<T>>::insert(item.collection, current_index, v);1505 1506 // Update balance1507 let new_balance = <Balance<T>>::get(item.collection, owner.clone())1508 .checked_add(value)1509 .ok_or(Error::<T>::NumOverflow)?;1510 <Balance<T>>::insert(item.collection, owner.clone(), new_balance);15111512 Ok(())1513 }15141515 fn add_refungible_item(item: ReFungibleItemType<T::AccountId>) -> DispatchResult {1516 let current_index = <ItemListIndex>::get(item.collection)1517 .checked_add(1)1518 .ok_or(Error::<T>::NumOverflow)?;1519 let itemcopy = item.clone();15201521 let value = item.owner.first().unwrap().fraction as u64;1522 let owner = item.owner.first().unwrap().owner.clone();15231524 Self::add_token_index(item.collection, current_index, owner.clone())?;15251526 <ItemListIndex>::insert(item.collection, current_index);1527 <ReFungibleItemList<T>>::insert(item.collection, current_index, itemcopy);15281529 // Add current block1530 let block_number: T::BlockNumber = 0.into();1531 <ReFungibleTransferBasket<T>>::insert(item.collection, current_index, block_number);15321533 // Update balance1534 let new_balance = <Balance<T>>::get(item.collection, owner.clone())1535 .checked_add(value)1536 .ok_or(Error::<T>::NumOverflow)?;1537 <Balance<T>>::insert(item.collection, owner.clone(), new_balance);15381539 Ok(())1540 }15411542 fn add_nft_item(item: NftItemType<T::AccountId>) -> DispatchResult {1543 let current_index = <ItemListIndex>::get(item.collection)1544 .checked_add(1)1545 .ok_or(Error::<T>::NumOverflow)?;15461547 let item_owner = item.owner.clone();1548 let collection_id = item.collection.clone();1549 Self::add_token_index(collection_id, current_index, item.owner.clone())?;15501551 <ItemListIndex>::insert(collection_id, current_index);1552 <NftItemList<T>>::insert(collection_id, current_index, item);15531554 // Add current block1555 let block_number: T::BlockNumber = 0.into();1556 <NftTransferBasket<T>>::insert(collection_id, current_index, block_number);15571558 // Update balance1559 let new_balance = <Balance<T>>::get(collection_id, item_owner.clone())1560 .checked_add(1)1561 .ok_or(Error::<T>::NumOverflow)?;1562 <Balance<T>>::insert(collection_id, item_owner.clone(), new_balance);15631564 Ok(())1565 }15661567 fn burn_refungible_item(1568 collection_id: u64,1569 item_id: u64,1570 owner: T::AccountId,1571 ) -> DispatchResult {1572 ensure!(1573 <ReFungibleItemList<T>>::contains_key(collection_id, item_id),1574 Error::<T>::TokenNotFound1575 );1576 let collection = <ReFungibleItemList<T>>::get(collection_id, item_id);1577 let item = collection1578 .owner1579 .iter()1580 .filter(|&i| i.owner == owner)1581 .next()1582 .unwrap();1583 Self::remove_token_index(collection_id, item_id, owner.clone())?;15841585 // remove approve list1586 <ApprovedList<T>>::remove(collection_id, (item_id, owner.clone()));15871588 // update balance1589 let new_balance = <Balance<T>>::get(collection_id, item.owner.clone())1590 .checked_sub(item.fraction as u64)1591 .ok_or(Error::<T>::NumOverflow)?;1592 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);15931594 <ReFungibleItemList<T>>::remove(collection_id, item_id);15951596 Ok(())1597 }15981599 fn burn_nft_item(collection_id: u64, item_id: u64) -> DispatchResult {1600 ensure!(1601 <NftItemList<T>>::contains_key(collection_id, item_id),1602 Error::<T>::TokenNotFound1603 );1604 let item = <NftItemList<T>>::get(collection_id, item_id);1605 Self::remove_token_index(collection_id, item_id, item.owner.clone())?;16061607 // remove approve list1608 <ApprovedList<T>>::remove(collection_id, (item_id, item.owner.clone()));16091610 // update balance1611 let new_balance = <Balance<T>>::get(collection_id, item.owner.clone())1612 .checked_sub(1)1613 .ok_or(Error::<T>::NumOverflow)?;1614 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);1615 <NftItemList<T>>::remove(collection_id, item_id);16161617 Ok(())1618 }16191620 fn burn_fungible_item(collection_id: u64, item_id: u64) -> DispatchResult {1621 ensure!(1622 <FungibleItemList<T>>::contains_key(collection_id, item_id),1623 Error::<T>::TokenNotFound1624 );1625 let item = <FungibleItemList<T>>::get(collection_id, item_id);1626 Self::remove_token_index(collection_id, item_id, item.owner.clone())?;16271628 // remove approve list1629 <ApprovedList<T>>::remove(collection_id, (item_id, item.owner.clone()));16301631 // update balance1632 let new_balance = <Balance<T>>::get(collection_id, item.owner.clone())1633 .checked_sub(item.value as u64)1634 .ok_or(Error::<T>::NumOverflow)?;1635 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);16361637 <FungibleItemList<T>>::remove(collection_id, item_id);16381639 Ok(())1640 }16411642 fn collection_exists(collection_id: u64) -> DispatchResult {1643 ensure!(1644 <Collection<T>>::contains_key(collection_id),1645 Error::<T>::CollectionNotFound1646 );1647 Ok(())1648 }16491650 fn check_owner_permissions(collection_id: u64, subject: T::AccountId) -> DispatchResult {1651 Self::collection_exists(collection_id)?;16521653 let target_collection = <Collection<T>>::get(collection_id);1654 ensure!(1655 subject == target_collection.owner,1656 Error::<T>::NoPermission1657 );16581659 Ok(())1660 }16611662 fn is_owner_or_admin_permissions(collection_id: u64, subject: T::AccountId) -> bool {1663 let target_collection = <Collection<T>>::get(collection_id);1664 let mut result: bool = subject == target_collection.owner;1665 let exists = <AdminList<T>>::contains_key(collection_id);16661667 if !result & exists {1668 if <AdminList<T>>::get(collection_id).contains(&subject) {1669 result = true1670 }1671 }16721673 result1674 }16751676 fn check_owner_or_admin_permissions(1677 collection_id: u64,1678 subject: T::AccountId,1679 ) -> DispatchResult {1680 Self::collection_exists(collection_id)?;1681 let result = Self::is_owner_or_admin_permissions(collection_id, subject.clone());16821683 ensure!(1684 result,1685 Error::<T>::NoPermission1686 );1687 Ok(())1688 }16891690 fn is_item_owner(subject: T::AccountId, collection_id: u64, item_id: u64) -> bool {1691 let target_collection = <Collection<T>>::get(collection_id);16921693 match target_collection.mode {1694 CollectionMode::NFT => {1695 <NftItemList<T>>::get(collection_id, item_id).owner == subject1696 }1697 CollectionMode::Fungible(_) => {1698 <FungibleItemList<T>>::get(collection_id, item_id).owner == subject1699 }1700 CollectionMode::ReFungible(_) => {1701 <ReFungibleItemList<T>>::get(collection_id, item_id)1702 .owner1703 .iter()1704 .any(|i| i.owner == subject)1705 }1706 CollectionMode::Invalid => false,1707 }1708 }17091710 fn check_white_list(collection_id: u64, address: &T::AccountId) -> DispatchResult {1711 let mes = Error::<T>::AddresNotInWhiteList;1712 ensure!(<WhiteList<T>>::contains_key(collection_id), mes);1713 let wl = <WhiteList<T>>::get(collection_id);1714 ensure!(wl.contains(address), mes);17151716 Ok(())1717 }17181719 fn transfer_fungible(1720 collection_id: u64,1721 item_id: u64,1722 value: u64,1723 owner: T::AccountId,1724 new_owner: T::AccountId,1725 ) -> DispatchResult {1726 ensure!(1727 <FungibleItemList<T>>::contains_key(collection_id, item_id),1728 Error::<T>::TokenNotFound1729 );17301731 let full_item = <FungibleItemList<T>>::get(collection_id, item_id);1732 let amount = full_item.value;17331734 ensure!(amount >= value.into(), Error::<T>::TokenValueTooLow);17351736 // update balance1737 let balance_old_owner = <Balance<T>>::get(collection_id, owner.clone())1738 .checked_sub(value)1739 .ok_or(Error::<T>::NumOverflow)?;1740 <Balance<T>>::insert(collection_id, owner.clone(), balance_old_owner);17411742 let mut new_owner_account_id = 0;1743 let new_owner_items = <AddressTokens<T>>::get(collection_id, new_owner.clone());1744 if new_owner_items.len() > 0 {1745 new_owner_account_id = new_owner_items[0];1746 }17471748 let val64 = value.into();17491750 // transfer1751 if amount == val64 && new_owner_account_id == 0 {1752 // change owner1753 // new owner do not have account1754 let mut new_full_item = full_item.clone();1755 new_full_item.owner = new_owner.clone();1756 <FungibleItemList<T>>::insert(collection_id, item_id, new_full_item);17571758 // update balance1759 let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone())1760 .checked_add(value)1761 .ok_or(Error::<T>::NumOverflow)?;1762 <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);17631764 // update index collection1765 Self::move_token_index(collection_id, item_id, owner.clone(), new_owner.clone())?;1766 } else {1767 let mut new_full_item = full_item.clone();1768 new_full_item.value -= val64;17691770 // separate amount1771 if new_owner_account_id > 0 {1772 // new owner has account1773 let mut item = <FungibleItemList<T>>::get(collection_id, new_owner_account_id);1774 item.value += val64;17751776 // update balance1777 let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone())1778 .checked_add(value)1779 .ok_or(Error::<T>::NumOverflow)?;1780 <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);17811782 <FungibleItemList<T>>::insert(collection_id, new_owner_account_id, item);1783 } else {1784 // new owner do not have account1785 let item = FungibleItemType {1786 collection: collection_id,1787 owner: new_owner.clone(),1788 value: val64,1789 };17901791 Self::add_fungible_item(item)?;1792 }17931794 if amount == val64 {1795 Self::remove_token_index(collection_id, item_id, full_item.owner.clone())?;17961797 // remove approve list1798 <ApprovedList<T>>::remove(collection_id, (item_id, full_item.owner.clone()));1799 <FungibleItemList<T>>::remove(collection_id, item_id);1800 }18011802 <FungibleItemList<T>>::insert(collection_id, item_id, new_full_item);1803 }18041805 Ok(())1806 }18071808 fn transfer_refungible(1809 collection_id: u64,1810 item_id: u64,1811 value: u64,1812 owner: T::AccountId,1813 new_owner: T::AccountId,1814 ) -> DispatchResult {1815 ensure!(1816 <ReFungibleItemList<T>>::contains_key(collection_id, item_id),1817 Error::<T>::TokenNotFound1818 );18191820 let full_item = <ReFungibleItemList<T>>::get(collection_id, item_id);1821 let item = full_item1822 .owner1823 .iter()1824 .filter(|i| i.owner == owner)1825 .next()1826 .ok_or(Error::<T>::NumOverflow)?;1827 let amount = item.fraction;18281829 ensure!(amount >= value.into(), Error::<T>::TokenValueTooLow);18301831 // update balance1832 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())1833 .checked_sub(value)1834 .ok_or(Error::<T>::NumOverflow)?;1835 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);18361837 let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone())1838 .checked_add(value)1839 .ok_or(Error::<T>::NumOverflow)?;1840 <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);18411842 let old_owner = item.owner.clone();1843 let new_owner_has_account = full_item.owner.iter().any(|i| i.owner == new_owner);1844 let val64 = value.into();18451846 // transfer1847 if amount == val64 && !new_owner_has_account {1848 // change owner1849 // new owner do not have account1850 let mut new_full_item = full_item.clone();1851 new_full_item1852 .owner1853 .iter_mut()1854 .find(|i| i.owner == owner)1855 .unwrap()1856 .owner = new_owner.clone();1857 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);18581859 // update index collection1860 Self::move_token_index(collection_id, item_id, old_owner.clone(), new_owner.clone())?;1861 } else {1862 let mut new_full_item = full_item.clone();1863 new_full_item1864 .owner1865 .iter_mut()1866 .find(|i| i.owner == owner)1867 .unwrap()1868 .fraction -= val64;18691870 // separate amount1871 if new_owner_has_account {1872 // new owner has account1873 new_full_item1874 .owner1875 .iter_mut()1876 .find(|i| i.owner == new_owner)1877 .unwrap()1878 .fraction += val64;1879 } else {1880 // new owner do not have account1881 new_full_item.owner.push(Ownership {1882 owner: new_owner.clone(),1883 fraction: val64,1884 });1885 Self::add_token_index(collection_id, item_id, new_owner.clone())?;1886 }18871888 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);1889 }18901891 Ok(())1892 }18931894 fn transfer_nft(1895 collection_id: u64,1896 item_id: u64,1897 sender: T::AccountId,1898 new_owner: T::AccountId,1899 ) -> DispatchResult {1900 ensure!(1901 <NftItemList<T>>::contains_key(collection_id, item_id),1902 Error::<T>::TokenNotFound1903 );19041905 let mut item = <NftItemList<T>>::get(collection_id, item_id);19061907 ensure!(1908 sender == item.owner,1909 Error::<T>::MustBeTokenOwner1910 );19111912 // update balance1913 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())1914 .checked_sub(1)1915 .ok_or(Error::<T>::NumOverflow)?;1916 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);19171918 let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone())1919 .checked_add(1)1920 .ok_or(Error::<T>::NumOverflow)?;1921 <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);19221923 // change owner1924 let old_owner = item.owner.clone();1925 item.owner = new_owner.clone();1926 <NftItemList<T>>::insert(collection_id, item_id, item);19271928 // update index collection1929 Self::move_token_index(collection_id, item_id, old_owner.clone(), new_owner.clone())?;19301931 // reset approved list1932 <ApprovedList<T>>::remove(collection_id, (item_id, old_owner));1933 Ok(())1934 }1935 1936 fn item_exists(1937 collection_id: u64,1938 item_id: u64,1939 mode: &CollectionMode1940 ) -> DispatchResult {1941 match mode {1942 CollectionMode::NFT => ensure!(<NftItemList<T>>::contains_key(collection_id, item_id), Error::<T>::TokenNotFound),1943 CollectionMode::ReFungible(_) => ensure!(<ReFungibleItemList<T>>::contains_key(collection_id, item_id), Error::<T>::TokenNotFound),1944 CollectionMode::Fungible(_) => ensure!(<FungibleItemList<T>>::contains_key(collection_id, item_id), Error::<T>::TokenNotFound),1945 _ => ()1946 };1947 1948 Ok(())1949 }19501951 fn set_re_fungible_variable_data(1952 collection_id: u64,1953 item_id: u64,1954 data: Vec<u8>1955 ) -> DispatchResult {1956 let mut item = <ReFungibleItemList<T>>::get(collection_id, item_id);19571958 item.variable_data = data;19591960 <ReFungibleItemList<T>>::insert(collection_id, item_id, item);19611962 Ok(())1963 }19641965 fn set_nft_variable_data(1966 collection_id: u64,1967 item_id: u64,1968 data: Vec<u8>1969 ) -> DispatchResult {1970 let mut item = <NftItemList<T>>::get(collection_id, item_id);1971 1972 item.variable_data = data;19731974 <NftItemList<T>>::insert(collection_id, item_id, item);1975 1976 Ok(())1977 }19781979 fn init_collection(item: &CollectionType<T::AccountId>) {1980 // check params1981 assert!(1982 item.decimal_points <= 4,1983 "decimal_points parameter must be lower than 4"1984 );1985 assert!(1986 item.name.len() <= 64,1987 "Collection name can not be longer than 63 char"1988 );1989 assert!(1990 item.name.len() <= 256,1991 "Collection description can not be longer than 255 char"1992 );1993 assert!(1994 item.token_prefix.len() <= 16,1995 "Token prefix can not be longer than 15 char"1996 );19971998 // Generate next collection ID1999 let next_id = CreatedCollectionCount::get()2000 .checked_add(1)2001 .unwrap();20022003 CreatedCollectionCount::put(next_id);2004 }20052006 fn init_nft_token(item: &NftItemType<T::AccountId>) {2007 let current_index = <ItemListIndex>::get(item.collection)2008 .checked_add(1)2009 .unwrap();20102011 let item_owner = item.owner.clone();2012 let collection_id = item.collection.clone();2013 Self::add_token_index(collection_id, current_index, item.owner.clone()).unwrap();20142015 <ItemListIndex>::insert(collection_id, current_index);20162017 // Update balance2018 let new_balance = <Balance<T>>::get(collection_id, item_owner.clone())2019 .checked_add(1)2020 .unwrap();2021 <Balance<T>>::insert(collection_id, item_owner.clone(), new_balance);2022 }20232024 fn init_fungible_token(item: &FungibleItemType<T::AccountId>) {2025 let current_index = <ItemListIndex>::get(item.collection)2026 .checked_add(1)2027 .unwrap();2028 let owner = item.owner.clone();2029 let value = item.value as u64;20302031 Self::add_token_index(item.collection, current_index, owner.clone()).unwrap();20322033 <ItemListIndex>::insert(item.collection, current_index);20342035 // Update balance2036 let new_balance = <Balance<T>>::get(item.collection, owner.clone())2037 .checked_add(value)2038 .unwrap();2039 <Balance<T>>::insert(item.collection, owner.clone(), new_balance);2040 }20412042 fn init_refungible_token(item: &ReFungibleItemType<T::AccountId>) {2043 let current_index = <ItemListIndex>::get(item.collection)2044 .checked_add(1)2045 .unwrap();20462047 let value = item.owner.first().unwrap().fraction as u64;2048 let owner = item.owner.first().unwrap().owner.clone();20492050 Self::add_token_index(item.collection, current_index, owner.clone()).unwrap();20512052 <ItemListIndex>::insert(item.collection, current_index);20532054 // Update balance2055 let new_balance = <Balance<T>>::get(item.collection, owner.clone())2056 .checked_add(value)2057 .unwrap();2058 <Balance<T>>::insert(item.collection, owner.clone(), new_balance);2059 }20602061 fn add_token_index(collection_id: u64, item_index: u64, owner: T::AccountId) -> DispatchResult {20622063 // add to account limit2064 if <AccountItemCount<T>>::contains_key(owner.clone()) {20652066 // bound Owned tokens by a single address2067 let count = <AccountItemCount<T>>::get(owner.clone());2068 ensure!(count < ChainLimit::get().account_token_ownership_limit, Error::<T>::AddressOwnershipLimitExceeded);20692070 <AccountItemCount<T>>::insert(owner.clone(), count2071 .checked_add(1)2072 .ok_or(Error::<T>::NumOverflow)?);2073 }2074 else {2075 <AccountItemCount<T>>::insert(owner.clone(), 1);2076 }20772078 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());2079 if list_exists {2080 let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());2081 let item_contains = list.contains(&item_index.clone());20822083 if !item_contains {2084 list.push(item_index.clone());2085 }20862087 <AddressTokens<T>>::insert(collection_id, owner.clone(), list);2088 } else {2089 let mut itm = Vec::new();2090 itm.push(item_index.clone());2091 <AddressTokens<T>>::insert(collection_id, owner, itm);2092 2093 }20942095 Ok(())2096 }20972098 fn remove_token_index(2099 collection_id: u64,2100 item_index: u64,2101 owner: T::AccountId,2102 ) -> DispatchResult {21032104 // update counter2105 <AccountItemCount<T>>::insert(owner.clone(), 2106 <AccountItemCount<T>>::get(owner.clone())2107 .checked_sub(1)2108 .ok_or(Error::<T>::NumOverflow)?);210921102111 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());2112 if list_exists {2113 let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());2114 let item_contains = list.contains(&item_index.clone());21152116 if item_contains {2117 list.retain(|&item| item != item_index);2118 <AddressTokens<T>>::insert(collection_id, owner, list);2119 }2120 }21212122 Ok(())2123 }21242125 fn move_token_index(2126 collection_id: u64,2127 item_index: u64,2128 old_owner: T::AccountId,2129 new_owner: T::AccountId,2130 ) -> DispatchResult {2131 Self::remove_token_index(collection_id, item_index, old_owner)?;2132 Self::add_token_index(collection_id, item_index, new_owner)?;21332134 Ok(())2135 }2136}21372138////////////////////////////////////////////////////////////////////////////////////////////////////2139// Economic models2140// #region21412142/// Fee multiplier.2143pub type Multiplier = FixedU128;21442145type BalanceOf<T> = <<T as transaction_payment::Trait>::Currency as Currency<2146 <T as system::Trait>::AccountId,2147>>::Balance;2148type NegativeImbalanceOf<T> = <<T as transaction_payment::Trait>::Currency as Currency<2149 <T as system::Trait>::AccountId,2150>>::NegativeImbalance;21512152/// Require the transactor pay for themselves and maybe include a tip to gain additional priority2153/// in the queue.2154#[derive(Encode, Decode, Clone, Eq, PartialEq)]2155pub struct ChargeTransactionPayment<T: Trait + Send + Sync>(2156 #[codec(compact)] BalanceOf<T>2157);21582159impl<T: Trait + Send + Sync> sp_std::fmt::Debug2160 for ChargeTransactionPayment<T>2161{2162 #[cfg(feature = "std")]2163 fn fmt(&self, f: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {2164 write!(f, "ChargeTransactionPayment<{:?}>", self.0)2165 }2166 #[cfg(not(feature = "std"))]2167 fn fmt(&self, _: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {2168 Ok(())2169 }2170}21712172impl<T: Trait + Send + Sync> ChargeTransactionPayment<T>2173where2174 T::Call: Dispatchable<Info = DispatchInfo, PostInfo = PostDispatchInfo> + IsSubType<Call<T>> + IsSubType<pallet_contracts::Call<T>>,2175 BalanceOf<T>: Send + Sync + FixedPointOperand,2176{2177 /// utility constructor. Used only in client/factory code.2178 pub fn from(fee: BalanceOf<T>) -> Self {2179 Self(fee)2180 }21812182 pub fn traditional_fee(2183 len: usize,2184 info: &DispatchInfoOf<T::Call>,2185 tip: BalanceOf<T>,2186 ) -> BalanceOf<T>2187 where2188 T::Call: Dispatchable<Info = DispatchInfo>,2189 {2190 <transaction_payment::Module<T>>::compute_fee(len as u32, info, tip)2191 }21922193 fn withdraw_fee(2194 &self,2195 who: &T::AccountId,2196 call: &T::Call,2197 info: &DispatchInfoOf<T::Call>,2198 len: usize,2199 ) -> Result<(BalanceOf<T>, Option<NegativeImbalanceOf<T>>), TransactionValidityError> {2200 let tip = self.0;22012202 // Set fee based on call type. Creating collection costs 1 Unique.2203 // All other transactions have traditional fees so far2204 // let fee = match call.is_sub_type() {2205 // Some(Call::create_collection(..)) => <BalanceOf<T>>::from(1_000_000_000),2206 // _ => Self::traditional_fee(len, info, tip), // Flat fee model, use only for testing purposes2207 // // _ => <BalanceOf<T>>::from(100)2208 // };2209 let fee = Self::traditional_fee(len, info, tip);22102211 // Determine who is paying transaction fee based on ecnomic model2212 // Parse call to extract collection ID and access collection sponsor2213 let mut sponsor: T::AccountId = match IsSubType::<Call<T>>::is_sub_type(call) {2214 Some(Call::create_item(collection_id, _properties, _owner)) => {2215 <Collection<T>>::get(collection_id).sponsor2216 }2217 Some(Call::transfer(_new_owner, collection_id, _item_id, _value)) => {2218 let _collection_mode = <Collection<T>>::get(collection_id).mode;22192220 // sponsor timeout2221 let sponsor_transfer = match _collection_mode {2222 CollectionMode::NFT => {2223 let basket = <NftTransferBasket<T>>::get(collection_id, _item_id);2224 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2225 let limit_time = basket + ChainLimit::get().nft_sponsor_transfer_timeout.into();2226 if block_number >= limit_time {2227 <NftTransferBasket<T>>::insert(collection_id, _item_id, block_number);2228 true2229 }2230 else {2231 false2232 }2233 }2234 CollectionMode::Fungible(_) => {2235 let mut basket = <FungibleTransferBasket<T>>::get(collection_id, _item_id);2236 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2237 if basket.iter().any(|i| i.address == _new_owner.clone())2238 {2239 let item = basket.iter_mut().find(|i| i.address == _new_owner.clone()).unwrap().clone();2240 let limit_time = item.start_block + ChainLimit::get().fungible_sponsor_transfer_timeout.into();2241 if block_number >= limit_time {2242 basket.retain(|x| x.address == item.address);2243 basket.push(BasketItem { start_block: block_number, address: _new_owner.clone() });2244 <FungibleTransferBasket<T>>::insert(collection_id, _item_id, basket);2245 true2246 }2247 else {2248 false2249 }2250 }2251 else {2252 basket.push(BasketItem { start_block: block_number, address: _new_owner.clone()});2253 true2254 }2255 }2256 CollectionMode::ReFungible(_) => {2257 let basket = <ReFungibleTransferBasket<T>>::get(collection_id, _item_id);2258 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2259 let limit_time = basket + ChainLimit::get().nft_sponsor_transfer_timeout.into();2260 if block_number >= limit_time {2261 <ReFungibleTransferBasket<T>>::insert(collection_id, _item_id, block_number);2262 true2263 } else {2264 false2265 }2266 }2267 _ => {2268 false2269 },2270 };22712272 if !sponsor_transfer {2273 T::AccountId::default()2274 } else {2275 <Collection<T>>::get(collection_id).sponsor2276 }2277 }22782279 _ => T::AccountId::default(),2280 };22812282 // Sponsor smart contracts2283 sponsor = match IsSubType::<pallet_contracts::Call<T>>::is_sub_type(call) {22842285 // On instantiation: set the contract owner2286 Some(pallet_contracts::Call::instantiate(_endowment, _gas_limit, code_hash, data)) => {22872288 let new_contract_address = <T as pallet_contracts::Trait>::DetermineContractAddress::contract_address_for(2289 code_hash,2290 &data,2291 &who,2292 );2293 <ContractOwner<T>>::insert(new_contract_address.clone(), who.clone());22942295 T::AccountId::default()2296 },22972298 // When the contract is called, check if the sponsoring is enabled and pay fees from contract endowment if it is2299 Some(pallet_contracts::Call::call(dest, _value, _gas_limit, _data)) => {23002301 let called_contract: T::AccountId = T::Lookup::lookup((*dest).clone()).unwrap_or(T::AccountId::default());23022303 let mut sponsor_transfer = false;2304 if <ContractSponsoringRateLimit<T>>::contains_key(called_contract.clone()) {2305 let last_tx_block = <ContractSponsorBasket<T>>::get(&called_contract);2306 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2307 let rate_limit = <ContractSponsoringRateLimit<T>>::get(&called_contract);2308 let limit_time = last_tx_block + rate_limit;23092310 if block_number >= limit_time {2311 <ContractSponsorBasket<T>>::insert(called_contract.clone(), block_number);2312 sponsor_transfer = true;2313 }2314 } else {2315 sponsor_transfer = false;2316 }2317 2318 2319 let mut sp = T::AccountId::default();2320 if sponsor_transfer {2321 if <ContractSelfSponsoring<T>>::contains_key(called_contract.clone()) {2322 if <ContractSelfSponsoring<T>>::get(called_contract.clone()) {2323 sp = called_contract;2324 }2325 }2326 }23272328 sp2329 },23302331 _ => sponsor,2332 };23332334 let mut who_pays_fee: T::AccountId = sponsor.clone();2335 if sponsor == T::AccountId::default() {2336 who_pays_fee = who.clone();2337 }23382339 // Only mess with balances if fee is not zero.2340 if fee.is_zero() {2341 return Ok((fee, None));2342 }23432344 match <T as transaction_payment::Trait>::Currency::withdraw(2345 &who_pays_fee,2346 fee,2347 if tip.is_zero() {2348 WithdrawReason::TransactionPayment.into()2349 } else {2350 WithdrawReason::TransactionPayment | WithdrawReason::Tip2351 },2352 ExistenceRequirement::KeepAlive,2353 ) {2354 Ok(imbalance) => Ok((fee, Some(imbalance))),2355 Err(_) => Err(InvalidTransaction::Payment.into()),2356 }2357 }2358}235923602361impl<T: Trait + Send + Sync> SignedExtension2362 for ChargeTransactionPayment<T>2363where2364 BalanceOf<T>: Send + Sync + From<u64> + FixedPointOperand,2365 T::Call: Dispatchable<Info = DispatchInfo, PostInfo = PostDispatchInfo> + IsSubType<Call<T>> + IsSubType<pallet_contracts::Call<T>>,2366{2367 const IDENTIFIER: &'static str = "ChargeTransactionPayment";2368 type AccountId = T::AccountId;2369 type Call = T::Call;2370 type AdditionalSigned = ();2371 type Pre = (2372 BalanceOf<T>,2373 Self::AccountId,2374 Option<NegativeImbalanceOf<T>>,2375 BalanceOf<T>,2376 );2377 fn additional_signed(&self) -> sp_std::result::Result<(), TransactionValidityError> {2378 Ok(())2379 }23802381 fn validate(2382 &self,2383 _who: &Self::AccountId,2384 _call: &Self::Call,2385 _info: &DispatchInfoOf<Self::Call>,2386 _len: usize,2387 ) -> TransactionValidity {2388 Ok(ValidTransaction::default())2389 }23902391 fn pre_dispatch(2392 self,2393 who: &Self::AccountId,2394 call: &Self::Call,2395 info: &DispatchInfoOf<Self::Call>,2396 len: usize,2397 ) -> Result<Self::Pre, TransactionValidityError> {2398 let (fee, imbalance) = self.withdraw_fee(who, call, info, len)?;2399 Ok((self.0, who.clone(), imbalance, fee))2400 }24012402 fn post_dispatch(2403 pre: Self::Pre,2404 info: &DispatchInfoOf<Self::Call>,2405 post_info: &PostDispatchInfoOf<Self::Call>,2406 len: usize,2407 _result: &DispatchResult,2408 ) -> Result<(), TransactionValidityError> {2409 let (tip, who, imbalance, fee) = pre;2410 if let Some(payed) = imbalance {2411 let actual_fee = <transaction_payment::Module<T>>::compute_actual_fee(2412 len as u32, info, post_info, tip,2413 );2414 let refund = fee.saturating_sub(actual_fee);2415 let actual_payment =2416 match <T as transaction_payment::Trait>::Currency::deposit_into_existing(2417 &who, refund,2418 ) {2419 Ok(refund_imbalance) => {2420 // The refund cannot be larger than the up front payed max weight.2421 // `PostDispatchInfo::calc_unspent` guards against such a case.2422 match payed.offset(refund_imbalance) {2423 Ok(actual_payment) => actual_payment,2424 Err(_) => return Err(InvalidTransaction::Payment.into()),2425 }2426 }2427 // We do not recreate the account using the refund. The up front payment2428 // is gone in that case.2429 Err(_) => payed,2430 };2431 let imbalances = actual_payment.split(tip);2432 <T as transaction_payment::Trait>::OnTransactionPayment::on_unbalanceds(2433 Some(imbalances.0).into_iter().chain(Some(imbalances.1)),2434 );2435 }2436 Ok(())2437 }2438}24392440// #endregion1#![cfg_attr(not(feature = "std"), no_std)]23#[cfg(feature = "std")]4pub use std::*;56#[cfg(feature = "std")]7pub use serde::*;89use codec::{Decode, Encode};10pub use frame_support::{11 construct_runtime, decl_event, decl_module, decl_storage, decl_error,12 dispatch::DispatchResult,13 ensure, fail, parameter_types,14 traits::{15 Currency, ExistenceRequirement, Get, Imbalance, KeyOwnerProofSystem, OnUnbalanced,16 Randomness, WithdrawReason,17 },18 weights::{19 constants::{BlockExecutionWeight, ExtrinsicBaseWeight, RocksDbWeight, WEIGHT_PER_SECOND},20 DispatchInfo, GetDispatchInfo, IdentityFee, Pays, PostDispatchInfo, Weight,21 WeightToFeePolynomial,22 },23 IsSubType, StorageValue,24};2526use frame_system::{self as system, ensure_signed, ensure_root};27use sp_runtime::sp_std::prelude::Vec;28use sp_runtime::{29 traits::{30 DispatchInfoOf, Dispatchable, PostDispatchInfoOf, Saturating, SignedExtension, Zero,31 },32 transaction_validity::{33 InvalidTransaction, TransactionValidity, TransactionValidityError, ValidTransaction,34 },35 FixedPointOperand, FixedU128,36};37use pallet_contracts::ContractAddressFor;38use sp_runtime::traits::StaticLookup;3940#[cfg(test)]41mod mock;4243#[cfg(test)]44mod tests;4546mod default_weights;4748// Structs49// #region5051pub type CollectionId = u32;52pub type TokenId = u32;5354#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq)]55#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]56pub enum CollectionMode {57 Invalid,58 NFT,59 // decimal points60 Fungible(u32),61 // decimal points62 ReFungible(u32),63}6465impl Into<u8> for CollectionMode {66 fn into(self) -> u8 {67 match self {68 CollectionMode::Invalid => 0,69 CollectionMode::NFT => 1,70 CollectionMode::Fungible(_) => 2,71 CollectionMode::ReFungible(_) => 3,72 }73 }74}7576#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq)]77#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]78pub enum AccessMode {79 Normal,80 WhiteList,81}82impl Default for AccessMode {83 fn default() -> Self {84 Self::Normal85 }86}8788impl Default for CollectionMode {89 fn default() -> Self {90 Self::Invalid91 }92}9394#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]95#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]96pub struct Ownership<AccountId> {97 pub owner: AccountId,98 pub fraction: u128,99}100101#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]102#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]103pub struct CollectionType<AccountId> {104 pub owner: AccountId,105 pub mode: CollectionMode,106 pub access: AccessMode,107 pub decimal_points: u32,108 pub name: Vec<u16>, // 64 include null escape char109 pub description: Vec<u16>, // 256 include null escape char110 pub token_prefix: Vec<u8>, // 16 include null escape char111 pub mint_mode: bool,112 pub offchain_schema: Vec<u8>,113 pub sponsor: AccountId, // Who pays fees. If set to default address, the fees are applied to the transaction sender114 pub unconfirmed_sponsor: AccountId, // Sponsor address that has not yet confirmed sponsorship115 pub variable_on_chain_schema: Vec<u8>, //116 pub const_on_chain_schema: Vec<u8>, //117}118119#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]120#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]121pub struct NftItemType<AccountId> {122 pub collection: CollectionId,123 pub owner: AccountId,124 pub const_data: Vec<u8>,125 pub variable_data: Vec<u8>,126}127128#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]129#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]130pub struct FungibleItemType<AccountId> {131 pub collection: CollectionId,132 pub owner: AccountId,133 pub value: u128,134}135136#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]137#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]138pub struct ReFungibleItemType<AccountId> {139 pub collection: CollectionId,140 pub owner: Vec<Ownership<AccountId>>,141 pub const_data: Vec<u8>,142 pub variable_data: Vec<u8>,143}144145#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]146#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]147pub struct ApprovePermissions<AccountId> {148 pub approved: AccountId,149 pub amount: u64,150}151152#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]153#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]154pub struct VestingItem<AccountId, Moment> {155 pub sender: AccountId,156 pub recipient: AccountId,157 pub collection_id: CollectionId,158 pub item_id: TokenId,159 pub amount: u64,160 pub vesting_date: Moment,161}162163#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]164#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]165pub struct BasketItem<AccountId, BlockNumber> {166 pub address: AccountId,167 pub start_block: BlockNumber,168}169170#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]171#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]172pub struct ChainLimits {173 pub collection_numbers_limit: u32,174 pub account_token_ownership_limit: u32,175 pub collections_admins_limit: u64,176 pub custom_data_limit: u32,177178 // Timeouts for item types in passed blocks179 pub nft_sponsor_transfer_timeout: u32,180 pub fungible_sponsor_transfer_timeout: u32,181 pub refungible_sponsor_transfer_timeout: u32,182}183184pub trait WeightInfo {185 fn create_collection() -> Weight;186 fn destroy_collection() -> Weight;187 fn add_to_white_list() -> Weight;188 fn remove_from_white_list() -> Weight;189 fn set_public_access_mode() -> Weight;190 fn set_mint_permission() -> Weight;191 fn change_collection_owner() -> Weight;192 fn add_collection_admin() -> Weight;193 fn remove_collection_admin() -> Weight;194 fn set_collection_sponsor() -> Weight;195 fn confirm_sponsorship() -> Weight;196 fn remove_collection_sponsor() -> Weight;197 fn create_item(s: usize) -> Weight;198 fn burn_item() -> Weight;199 fn transfer() -> Weight;200 fn approve() -> Weight;201 fn transfer_from() -> Weight;202 fn set_offchain_schema() -> Weight;203 fn set_const_on_chain_schema() -> Weight;204 fn set_variable_on_chain_schema() -> Weight;205 fn set_variable_meta_data() -> Weight;206 fn enable_contract_sponsoring() -> Weight;207}208209#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]210#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]211pub struct CreateNftData {212 pub const_data: Vec<u8>,213 pub variable_data: Vec<u8>,214}215216#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]217#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]218pub struct CreateFungibleData {219}220221#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]222#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]223pub struct CreateReFungibleData {224 pub const_data: Vec<u8>,225 pub variable_data: Vec<u8>,226}227228#[derive(Encode, Decode, Debug, Clone, PartialEq)]229#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]230pub enum CreateItemData {231 NFT(CreateNftData),232 Fungible(CreateFungibleData),233 ReFungible(CreateReFungibleData)234}235236impl CreateItemData {237 pub fn len(&self) -> usize {238 let len = match self {239 CreateItemData::NFT(data) => data.variable_data.len() + data.const_data.len(),240 CreateItemData::ReFungible(data) => data.variable_data.len() + data.const_data.len(),241 _ => 0242 };243 244 return len;245 }246}247248impl From<CreateNftData> for CreateItemData {249 fn from(item: CreateNftData) -> Self {250 CreateItemData::NFT(item)251 }252}253254impl From<CreateReFungibleData> for CreateItemData {255 fn from(item: CreateReFungibleData) -> Self {256 CreateItemData::ReFungible(item)257 }258}259260impl From<CreateFungibleData> for CreateItemData {261 fn from(item: CreateFungibleData) -> Self {262 CreateItemData::Fungible(item)263 }264}265266267decl_error! {268 /// Error for non-fungible-token module.269 pub enum Error for Module<T: Trait> {270 /// Total collections bound exceeded.271 TotalCollectionsLimitExceeded,272 /// Decimal_points parameter must be lower than 4.273 CollectionDecimalPointLimitExceeded, 274 /// Collection name can not be longer than 63 char.275 CollectionNameLimitExceeded, 276 /// Collection description can not be longer than 255 char.277 CollectionDescriptionLimitExceeded, 278 /// Token prefix can not be longer than 15 char.279 CollectionTokenPrefixLimitExceeded,280 /// This collection does not exist.281 CollectionNotFound,282 /// Item not exists.283 TokenNotFound,284 /// Arithmetic calculation overflow.285 NumOverflow, 286 /// Account already has admin role.287 AlreadyAdmin, 288 /// You do not own this collection.289 NoPermission,290 /// This address is not set as sponsor, use setCollectionSponsor first.291 ConfirmUnsetSponsorFail,292 /// Collection is not in mint mode.293 PublicMintingNotAllowed,294 /// Sender parameter and item owner must be equal.295 MustBeTokenOwner,296 /// Item balance not enough.297 TokenValueTooLow,298 /// Size of item is too large.299 NftSizeLimitExceeded,300 /// No approve found301 ApproveNotFound,302 /// Requested value more than approved.303 TokenValueNotEnough,304 /// Only approved addresses can call this method.305 ApproveRequired,306 /// Address is not in white list.307 AddresNotInWhiteList,308 /// Number of collection admins bound exceeded.309 CollectionAdminsLimitExceeded,310 /// Owned tokens by a single address bound exceeded.311 AddressOwnershipLimitExceeded,312 /// Length of items properties must be greater than 0.313 EmptyArgument,314 /// const_data exceeded data limit.315 TokenConstDataLimitExceeded,316 /// variable_data exceeded data limit.317 TokenVariableDataLimitExceeded,318 /// Not NFT item data used to mint in NFT collection.319 NotNftDataUsedToMintNftCollectionToken,320 /// Not Fungible item data used to mint in Fungible collection.321 NotFungibleDataUsedToMintFungibleCollectionToken,322 /// Not Re Fungible item data used to mint in Re Fungible collection.323 NotReFungibleDataUsedToMintReFungibleCollectionToken,324 /// Unexpected collection type.325 UnexpectedCollectionType,326 /// Can't store metadata in fungible tokens.327 CantStoreMetadataInFungibleTokens328 }329}330331pub trait Trait: system::Trait + Sized + transaction_payment::Trait + pallet_contracts::Trait {332 type Event: From<Event<Self>> + Into<<Self as system::Trait>::Event>;333334 /// Weight information for extrinsics in this pallet.335 type WeightInfo: WeightInfo;336}337338#[cfg(feature = "runtime-benchmarks")]339mod benchmarking;340341// #endregion342343decl_storage! {344 trait Store for Module<T: Trait> as Nft {345346 // Private members347 NextCollectionID: CollectionId;348 CreatedCollectionCount: u32;349 ChainVersion: u64;350 ItemListIndex: map hasher(blake2_128_concat) CollectionId => TokenId;351352 // Chain limits struct353 pub ChainLimit get(fn chain_limit) config(): ChainLimits;354355 // Bound counters356 CollectionCount: u32;357 pub AccountItemCount get(fn account_item_count): map hasher(identity) T::AccountId => u32;358359 // Basic collections360 pub Collection get(fn collection) config(): map hasher(identity) CollectionId => CollectionType<T::AccountId>;361 pub AdminList get(fn admin_list_collection): map hasher(identity) CollectionId => Vec<T::AccountId>;362 pub WhiteList get(fn white_list): map hasher(identity) CollectionId => Vec<T::AccountId>;363364 /// Balance owner per collection map365 pub Balance get(fn balance_count): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) T::AccountId => u64;366367 /// second parameter: item id + owner account id368 pub ApprovedList get(fn approved): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) (TokenId, T::AccountId) => Vec<ApprovePermissions<T::AccountId>>;369370 /// Item collections371 pub NftItemList get(fn nft_item_id) config(): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) TokenId => NftItemType<T::AccountId>;372 pub FungibleItemList get(fn fungible_item_id) config(): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) TokenId => FungibleItemType<T::AccountId>;373 pub ReFungibleItemList get(fn refungible_item_id) config(): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) TokenId => ReFungibleItemType<T::AccountId>;374375 /// Index list376 pub AddressTokens get(fn address_tokens): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) T::AccountId => Vec<TokenId>;377378 /// Tokens transfer baskets379 pub NftTransferBasket get(fn nft_transfer_basket): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) TokenId => T::BlockNumber;380 pub FungibleTransferBasket get(fn fungible_transfer_basket): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) TokenId => Vec<BasketItem<T::AccountId, T::BlockNumber>>;381 pub ReFungibleTransferBasket get(fn refungible_transfer_basket): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) TokenId => T::BlockNumber;382383 // Contract Sponsorship and Ownership384 pub ContractOwner get(fn contract_owner): map hasher(twox_64_concat) T::AccountId => T::AccountId;385 pub ContractSelfSponsoring get(fn contract_self_sponsoring): map hasher(twox_64_concat) T::AccountId => bool;386 pub ContractSponsorBasket get(fn contract_sponsor_basket): map hasher(twox_64_concat) T::AccountId => T::BlockNumber;387 pub ContractSponsoringRateLimit get(fn contract_sponsoring_rate_limit): map hasher(twox_64_concat) T::AccountId => T::BlockNumber;388 }389 add_extra_genesis {390 build(|config: &GenesisConfig<T>| {391 // Modification of storage392 for (_num, _c) in &config.collection {393 <Module<T>>::init_collection(_c);394 }395396 for (_num, _q, _i) in &config.nft_item_id {397 <Module<T>>::init_nft_token(_i);398 }399400 for (_num, _q, _i) in &config.fungible_item_id {401 <Module<T>>::init_fungible_token(_i);402 }403404 for (_num, _q, _i) in &config.refungible_item_id {405 <Module<T>>::init_refungible_token(_i);406 }407 })408 }409}410411decl_event!(412 pub enum Event<T>413 where414 AccountId = <T as system::Trait>::AccountId,415 {416 /// New collection was created417 /// 418 /// # Arguments419 /// 420 /// * collection_id: Globally unique identifier of newly created collection.421 /// 422 /// * mode: [CollectionMode] converted into u8.423 /// 424 /// * account_id: Collection owner.425 Created(CollectionId, u8, AccountId),426427 /// New item was created.428 /// 429 /// # Arguments430 /// 431 /// * collection_id: Id of the collection where item was created.432 /// 433 /// * item_id: Id of an item. Unique within the collection.434 ItemCreated(CollectionId, TokenId),435436 /// Collection item was burned.437 /// 438 /// # Arguments439 /// 440 /// collection_id.441 /// 442 /// item_id: Identifier of burned NFT.443 ItemDestroyed(CollectionId, TokenId),444 }445);446447decl_module! {448 pub struct Module<T: Trait> for enum Call where origin: T::Origin {449450 fn deposit_event() = default;451 type Error = Error<T>;452453 fn on_initialize(now: T::BlockNumber) -> Weight {454455 if ChainVersion::get() < 2456 {457 let value = NextCollectionID::get();458 CreatedCollectionCount::put(value);459 ChainVersion::put(2);460 }461462 0463 }464465 /// This method creates a Collection of NFTs. Each Token may have multiple properties encoded as an array of bytes of certain length. The initial owner and admin of the collection are set to the address that signed the transaction. Both addresses can be changed later.466 /// 467 /// # Permissions468 /// 469 /// * Anyone.470 /// 471 /// # Arguments472 /// 473 /// * collection_name: UTF-16 string with collection name (limit 64 characters), will be stored as zero-terminated.474 /// 475 /// * collection_description: UTF-16 string with collection description (limit 256 characters), will be stored as zero-terminated.476 /// 477 /// * token_prefix: UTF-8 string with token prefix.478 /// 479 /// * mode: [CollectionMode] collection type and type dependent data.480 // returns collection ID481 #[weight = T::WeightInfo::create_collection()]482 pub fn create_collection(origin,483 collection_name: Vec<u16>,484 collection_description: Vec<u16>,485 token_prefix: Vec<u8>,486 mode: CollectionMode) -> DispatchResult {487488 // Anyone can create a collection489 let who = ensure_signed(origin)?;490491 let decimal_points = match mode {492 CollectionMode::Fungible(points) => points,493 CollectionMode::ReFungible(points) => points,494 _ => 0495 };496497 // bound Total number of collections498 ensure!(CollectionCount::get() < ChainLimit::get().collection_numbers_limit, Error::<T>::TotalCollectionsLimitExceeded);499500 // check params501 ensure!(decimal_points <= 4, Error::<T>::CollectionDecimalPointLimitExceeded);502503 let mut name = collection_name.to_vec();504 name.push(0);505 ensure!(name.len() <= 64, Error::<T>::CollectionNameLimitExceeded);506507 let mut description = collection_description.to_vec();508 description.push(0);509 ensure!(name.len() <= 256, Error::<T>::CollectionDescriptionLimitExceeded);510511 let mut prefix = token_prefix.to_vec();512 prefix.push(0);513 ensure!(prefix.len() <= 16, Error::<T>::CollectionTokenPrefixLimitExceeded);514515 // Generate next collection ID516 let next_id = CreatedCollectionCount::get()517 .checked_add(1)518 .ok_or(Error::<T>::NumOverflow)?;519520 // bound counter521 let total = CollectionCount::get()522 .checked_add(1)523 .ok_or(Error::<T>::NumOverflow)?;524525 CreatedCollectionCount::put(next_id);526 CollectionCount::put(total);527528 // Create new collection529 let new_collection = CollectionType {530 owner: who.clone(),531 name: name,532 mode: mode.clone(),533 mint_mode: false,534 access: AccessMode::Normal,535 description: description,536 decimal_points: decimal_points,537 token_prefix: prefix,538 offchain_schema: Vec::new(),539 sponsor: T::AccountId::default(),540 unconfirmed_sponsor: T::AccountId::default(),541 variable_on_chain_schema: Vec::new(),542 const_on_chain_schema: Vec::new(),543 };544545 // Add new collection to map546 <Collection<T>>::insert(next_id, new_collection);547548 // call event549 Self::deposit_event(RawEvent::Created(next_id, mode.into(), who.clone()));550551 Ok(())552 }553554 /// **DANGEROUS**: Destroys collection and all NFTs within this collection. Users irrecoverably lose their assets and may lose real money.555 /// 556 /// # Permissions557 /// 558 /// * Collection Owner.559 /// 560 /// # Arguments561 /// 562 /// * collection_id: collection to destroy.563 #[weight = T::WeightInfo::destroy_collection()]564 pub fn destroy_collection(origin, collection_id: CollectionId) -> DispatchResult {565566 let sender = ensure_signed(origin)?;567 Self::check_owner_permissions(collection_id, sender)?;568569 <AddressTokens<T>>::remove_prefix(collection_id);570 <ApprovedList<T>>::remove_prefix(collection_id);571 <Balance<T>>::remove_prefix(collection_id);572 <ItemListIndex>::remove(collection_id);573 <AdminList<T>>::remove(collection_id);574 <Collection<T>>::remove(collection_id);575 <WhiteList<T>>::remove(collection_id);576577 <NftItemList<T>>::remove_prefix(collection_id);578 <FungibleItemList<T>>::remove_prefix(collection_id);579 <ReFungibleItemList<T>>::remove_prefix(collection_id);580581 <NftTransferBasket<T>>::remove_prefix(collection_id);582 <FungibleTransferBasket<T>>::remove_prefix(collection_id);583 <ReFungibleTransferBasket<T>>::remove_prefix(collection_id);584585 if CollectionCount::get() > 0586 {587 // bound couter588 let total = CollectionCount::get()589 .checked_sub(1)590 .ok_or(Error::<T>::NumOverflow)?;591592 CollectionCount::put(total);593 }594595 Ok(())596 }597598 /// Add an address to white list.599 /// 600 /// # Permissions601 /// 602 /// * Collection Owner603 /// * Collection Admin604 /// 605 /// # Arguments606 /// 607 /// * collection_id.608 /// 609 /// * address.610 #[weight = T::WeightInfo::add_to_white_list()]611 pub fn add_to_white_list(origin, collection_id: CollectionId, address: T::AccountId) -> DispatchResult{612613 let sender = ensure_signed(origin)?;614 Self::check_owner_or_admin_permissions(collection_id, sender)?;615616 let mut white_list_collection: Vec<T::AccountId>;617 if <WhiteList<T>>::contains_key(collection_id) {618 white_list_collection = <WhiteList<T>>::get(collection_id);619 if !white_list_collection.contains(&address.clone())620 {621 white_list_collection.push(address.clone());622 }623 }624 else {625 white_list_collection = Vec::new();626 white_list_collection.push(address.clone());627 }628629 <WhiteList<T>>::insert(collection_id, white_list_collection);630 Ok(())631 }632633 /// Remove an address from white list.634 /// 635 /// # Permissions636 /// 637 /// * Collection Owner638 /// * Collection Admin639 /// 640 /// # Arguments641 /// 642 /// * collection_id.643 /// 644 /// * address.645 #[weight = T::WeightInfo::remove_from_white_list()]646 pub fn remove_from_white_list(origin, collection_id: CollectionId, address: T::AccountId) -> DispatchResult{647648 let sender = ensure_signed(origin)?;649 Self::check_owner_or_admin_permissions(collection_id, sender)?;650651 if <WhiteList<T>>::contains_key(collection_id) {652 let mut white_list_collection = <WhiteList<T>>::get(collection_id);653 if white_list_collection.contains(&address.clone())654 {655 white_list_collection.retain(|i| *i != address.clone());656 <WhiteList<T>>::insert(collection_id, white_list_collection);657 }658 }659660 Ok(())661 }662663 /// Toggle between normal and white list access for the methods with access for `Anyone`.664 /// 665 /// # Permissions666 /// 667 /// * Collection Owner.668 /// 669 /// # Arguments670 /// 671 /// * collection_id.672 /// 673 /// * mode: [AccessMode]674 #[weight = T::WeightInfo::set_public_access_mode()]675 pub fn set_public_access_mode(origin, collection_id: CollectionId, mode: AccessMode) -> DispatchResult676 {677 let sender = ensure_signed(origin)?;678679 Self::check_owner_permissions(collection_id, sender)?;680 let mut target_collection = <Collection<T>>::get(collection_id);681 target_collection.access = mode;682 <Collection<T>>::insert(collection_id, target_collection);683684 Ok(())685 }686687 /// Allows Anyone to create tokens if:688 /// * White List is enabled, and689 /// * Address is added to white list, and690 /// * This method was called with True parameter691 /// 692 /// # Permissions693 /// * Collection Owner694 ///695 /// # Arguments696 /// 697 /// * collection_id.698 /// 699 /// * mint_permission: Boolean parameter. If True, allows minting to Anyone with conditions above.700 #[weight = T::WeightInfo::set_mint_permission()]701 pub fn set_mint_permission(origin, collection_id: CollectionId, mint_permission: bool) -> DispatchResult702 {703 let sender = ensure_signed(origin)?;704705 Self::check_owner_permissions(collection_id, sender)?;706 let mut target_collection = <Collection<T>>::get(collection_id);707 target_collection.mint_mode = mint_permission;708 <Collection<T>>::insert(collection_id, target_collection);709710 Ok(())711 }712713 /// Change the owner of the collection.714 /// 715 /// # Permissions716 /// 717 /// * Collection Owner.718 /// 719 /// # Arguments720 /// 721 /// * collection_id.722 /// 723 /// * new_owner.724 #[weight = T::WeightInfo::change_collection_owner()]725 pub fn change_collection_owner(origin, collection_id: CollectionId, new_owner: T::AccountId) -> DispatchResult {726727 let sender = ensure_signed(origin)?;728 Self::check_owner_permissions(collection_id, sender)?;729 let mut target_collection = <Collection<T>>::get(collection_id);730 target_collection.owner = new_owner;731 <Collection<T>>::insert(collection_id, target_collection);732733 Ok(())734 }735736 /// Adds an admin of the Collection.737 /// NFT Collection can be controlled by multiple admin addresses (some which can also be servers, for example). Admins can issue and burn NFTs, as well as add and remove other admins, but cannot change NFT or Collection ownership. 738 /// 739 /// # Permissions740 /// 741 /// * Collection Owner.742 /// * Collection Admin.743 /// 744 /// # Arguments745 /// 746 /// * collection_id: ID of the Collection to add admin for.747 /// 748 /// * new_admin_id: Address of new admin to add.749 #[weight = T::WeightInfo::add_collection_admin()]750 pub fn add_collection_admin(origin, collection_id: CollectionId, new_admin_id: T::AccountId) -> DispatchResult {751752 let sender = ensure_signed(origin)?;753 Self::check_owner_or_admin_permissions(collection_id, sender)?;754 let mut admin_arr: Vec<T::AccountId> = Vec::new();755756 if <AdminList<T>>::contains_key(collection_id)757 {758 admin_arr = <AdminList<T>>::get(collection_id);759 ensure!(!admin_arr.contains(&new_admin_id), Error::<T>::AlreadyAdmin);760 }761762 // Number of collection admins763 ensure!((admin_arr.len() as u64) < ChainLimit::get().collections_admins_limit, Error::<T>::CollectionAdminsLimitExceeded);764765 admin_arr.push(new_admin_id);766 <AdminList<T>>::insert(collection_id, admin_arr);767768 Ok(())769 }770771 /// Remove admin address of the Collection. An admin address can remove itself. List of admins may become empty, in which case only Collection Owner will be able to add an Admin.772 ///773 /// # Permissions774 /// 775 /// * Collection Owner.776 /// * Collection Admin.777 /// 778 /// # Arguments779 /// 780 /// * collection_id: ID of the Collection to remove admin for.781 /// 782 /// * account_id: Address of admin to remove.783 #[weight = T::WeightInfo::remove_collection_admin()]784 pub fn remove_collection_admin(origin, collection_id: CollectionId, account_id: T::AccountId) -> DispatchResult {785786 let sender = ensure_signed(origin)?;787 Self::check_owner_or_admin_permissions(collection_id, sender)?;788789 if <AdminList<T>>::contains_key(collection_id)790 {791 let mut admin_arr = <AdminList<T>>::get(collection_id);792 admin_arr.retain(|i| *i != account_id);793 <AdminList<T>>::insert(collection_id, admin_arr);794 }795796 Ok(())797 }798799 /// # Permissions800 /// 801 /// * Collection Owner802 /// 803 /// # Arguments804 /// 805 /// * collection_id.806 /// 807 /// * new_sponsor.808 #[weight = T::WeightInfo::set_collection_sponsor()]809 pub fn set_collection_sponsor(origin, collection_id: CollectionId, new_sponsor: T::AccountId) -> DispatchResult {810811 let sender = ensure_signed(origin)?;812 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);813814 let mut target_collection = <Collection<T>>::get(collection_id);815 ensure!(sender == target_collection.owner, Error::<T>::NoPermission);816817 target_collection.unconfirmed_sponsor = new_sponsor;818 <Collection<T>>::insert(collection_id, target_collection);819820 Ok(())821 }822823 /// # Permissions824 /// 825 /// * Sponsor.826 /// 827 /// # Arguments828 /// 829 /// * collection_id.830 #[weight = T::WeightInfo::confirm_sponsorship()]831 pub fn confirm_sponsorship(origin, collection_id: CollectionId) -> DispatchResult {832833 let sender = ensure_signed(origin)?;834 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);835836 let mut target_collection = <Collection<T>>::get(collection_id);837 ensure!(sender == target_collection.unconfirmed_sponsor, Error::<T>::ConfirmUnsetSponsorFail);838839 target_collection.sponsor = target_collection.unconfirmed_sponsor;840 target_collection.unconfirmed_sponsor = T::AccountId::default();841 <Collection<T>>::insert(collection_id, target_collection);842843 Ok(())844 }845846 /// Switch back to pay-per-own-transaction model.847 ///848 /// # Permissions849 ///850 /// * Collection owner.851 /// 852 /// # Arguments853 /// 854 /// * collection_id.855 #[weight = T::WeightInfo::remove_collection_sponsor()]856 pub fn remove_collection_sponsor(origin, collection_id: CollectionId) -> DispatchResult {857858 let sender = ensure_signed(origin)?;859 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);860861 let mut target_collection = <Collection<T>>::get(collection_id);862 ensure!(sender == target_collection.owner, Error::<T>::NoPermission);863864 target_collection.sponsor = T::AccountId::default();865 <Collection<T>>::insert(collection_id, target_collection);866867 Ok(())868 }869870 /// This method creates a concrete instance of NFT Collection created with CreateCollection method.871 /// 872 /// # Permissions873 /// 874 /// * Collection Owner.875 /// * Collection Admin.876 /// * Anyone if877 /// * White List is enabled, and878 /// * Address is added to white list, and879 /// * MintPermission is enabled (see SetMintPermission method)880 /// 881 /// # Arguments882 /// 883 /// * collection_id: ID of the collection.884 /// 885 /// * owner: Address, initial owner of the NFT.886 ///887 /// * data: Token data to store on chain.888 // #[weight =889 // (130_000_000 as Weight)890 // .saturating_add((2135 as Weight).saturating_mul((properties.len() as u64) as Weight))891 // .saturating_add(RocksDbWeight::get().reads(10 as Weight))892 // .saturating_add(RocksDbWeight::get().writes(8 as Weight))]893894 #[weight = T::WeightInfo::create_item(data.len())]895 pub fn create_item(origin, collection_id: CollectionId, owner: T::AccountId, data: CreateItemData) -> DispatchResult {896897 let sender = ensure_signed(origin)?;898899 Self::collection_exists(collection_id)?;900901 let target_collection = <Collection<T>>::get(collection_id);902903 Self::can_create_items_in_collection(collection_id, &target_collection, &sender, &owner)?;904 Self::validate_create_item_args(&target_collection, &data)?;905 Self::create_item_no_validation(collection_id, &target_collection, owner, data)?;906907 Ok(())908 }909910 /// This method creates multiple instances of NFT Collection created with CreateCollection method.911 /// 912 /// # Permissions913 /// 914 /// * Collection Owner.915 /// * Collection Admin.916 /// * Anyone if917 /// * White List is enabled, and918 /// * Address is added to white list, and919 /// * MintPermission is enabled (see SetMintPermission method)920 /// 921 /// # Arguments922 /// 923 /// * collection_id: ID of the collection.924 /// 925 /// * itemsData: Array items properties. Each property is an array of bytes itself, see [create_item].926 /// 927 /// * owner: Address, initial owner of the NFT.928 #[weight = T::WeightInfo::create_item(items_data.into_iter()929 .map(|data| { data.len() })930 .sum())]931 pub fn create_multiple_items(origin, collection_id: CollectionId, owner: T::AccountId, items_data: Vec<CreateItemData>) -> DispatchResult {932933 ensure!(items_data.len() > 0, Error::<T>::EmptyArgument);934 let sender = ensure_signed(origin)?;935936 Self::collection_exists(collection_id)?;937 let target_collection = <Collection<T>>::get(collection_id);938939 Self::can_create_items_in_collection(collection_id, &target_collection, &sender, &owner)?;940941 for data in &items_data {942 Self::validate_create_item_args(&target_collection, data)?;943 }944 for data in &items_data {945 Self::create_item_no_validation(collection_id, &target_collection, owner.clone(), data.clone())?;946 }947948 Ok(())949 }950951 /// Destroys a concrete instance of NFT.952 /// 953 /// # Permissions954 /// 955 /// * Collection Owner.956 /// * Collection Admin.957 /// * Current NFT Owner.958 /// 959 /// # Arguments960 /// 961 /// * collection_id: ID of the collection.962 /// 963 /// * item_id: ID of NFT to burn.964 #[weight = T::WeightInfo::burn_item()]965 pub fn burn_item(origin, collection_id: CollectionId, item_id: TokenId) -> DispatchResult {966967 let sender = ensure_signed(origin)?;968 Self::collection_exists(collection_id)?;969970 // Transfer permissions check971 let target_collection = <Collection<T>>::get(collection_id);972 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||973 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),974 Error::<T>::NoPermission);975976 if target_collection.access == AccessMode::WhiteList {977 Self::check_white_list(collection_id, &sender)?;978 }979980 match target_collection.mode981 {982 CollectionMode::NFT => Self::burn_nft_item(collection_id, item_id)?,983 CollectionMode::Fungible(_) => Self::burn_fungible_item(collection_id, item_id)?,984 CollectionMode::ReFungible(_) => Self::burn_refungible_item(collection_id, item_id, sender.clone())?,985 _ => ()986 };987988 // call event989 Self::deposit_event(RawEvent::ItemDestroyed(collection_id, item_id));990991 Ok(())992 }993994 /// Change ownership of the token.995 /// 996 /// # Permissions997 /// 998 /// * Collection Owner999 /// * Collection Admin1000 /// * Current NFT owner1001 ///1002 /// # Arguments1003 /// 1004 /// * recipient: Address of token recipient.1005 /// 1006 /// * collection_id.1007 /// 1008 /// * item_id: ID of the item1009 /// * Non-Fungible Mode: Required.1010 /// * Fungible Mode: Ignored.1011 /// * Re-Fungible Mode: Required.1012 /// 1013 /// * value: Amount to transfer.1014 /// * Non-Fungible Mode: Ignored1015 /// * Fungible Mode: Must specify transferred amount1016 /// * Re-Fungible Mode: Must specify transferred portion (between 0 and 1)1017 #[weight = T::WeightInfo::transfer()]1018 pub fn transfer(origin, recipient: T::AccountId, collection_id: CollectionId, item_id: TokenId, value: u64) -> DispatchResult {10191020 let sender = ensure_signed(origin)?;10211022 // Transfer permissions check1023 let target_collection = <Collection<T>>::get(collection_id);1024 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1025 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1026 Error::<T>::NoPermission);10271028 if target_collection.access == AccessMode::WhiteList {1029 Self::check_white_list(collection_id, &sender)?;1030 Self::check_white_list(collection_id, &recipient)?;1031 }10321033 match target_collection.mode1034 {1035 CollectionMode::NFT => Self::transfer_nft(collection_id, item_id, sender.clone(), recipient)?,1036 CollectionMode::Fungible(_) => Self::transfer_fungible(collection_id, item_id, value, sender.clone(), recipient)?,1037 CollectionMode::ReFungible(_) => Self::transfer_refungible(collection_id, item_id, value, sender.clone(), recipient)?,1038 _ => ()1039 };10401041 Ok(())1042 }10431044 /// Set, change, or remove approved address to transfer the ownership of the NFT.1045 /// 1046 /// # Permissions1047 /// 1048 /// * Collection Owner1049 /// * Collection Admin1050 /// * Current NFT owner1051 /// 1052 /// # Arguments1053 /// 1054 /// * approved: Address that is approved to transfer this NFT or zero (if needed to remove approval).1055 /// 1056 /// * collection_id.1057 /// 1058 /// * item_id: ID of the item.1059 #[weight = T::WeightInfo::approve()]1060 pub fn approve(origin, approved: T::AccountId, collection_id: CollectionId, item_id: TokenId) -> DispatchResult {10611062 let sender = ensure_signed(origin)?;10631064 // Transfer permissions check1065 let target_collection = <Collection<T>>::get(collection_id);1066 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1067 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1068 Error::<T>::NoPermission);10691070 if target_collection.access == AccessMode::WhiteList {1071 Self::check_white_list(collection_id, &sender)?;1072 Self::check_white_list(collection_id, &approved)?;1073 }10741075 // amount param stub1076 let amount = 100000000;10771078 let list_exists = <ApprovedList<T>>::contains_key(collection_id, (item_id, sender.clone()));1079 if list_exists {10801081 let mut list = <ApprovedList<T>>::get(collection_id, (item_id, sender.clone()));1082 let item_contains = list.iter().any(|i| i.approved == approved);10831084 if !item_contains {1085 list.push(ApprovePermissions { approved: approved.clone(), amount: amount });1086 <ApprovedList<T>>::insert(collection_id, (item_id, sender.clone()), list);1087 }1088 } else {10891090 let mut list = Vec::new();1091 list.push(ApprovePermissions { approved: approved.clone(), amount: amount });1092 <ApprovedList<T>>::insert(collection_id, (item_id, sender.clone()), list);1093 }10941095 Ok(())1096 }1097 1098 /// Change ownership of a NFT on behalf of the owner. See Approve method for additional information. After this method executes, the approval is removed so that the approved address will not be able to transfer this NFT again from this owner.1099 /// 1100 /// # Permissions1101 /// * Collection Owner1102 /// * Collection Admin1103 /// * Current NFT owner1104 /// * Address approved by current NFT owner1105 /// 1106 /// # Arguments1107 /// 1108 /// * from: Address that owns token.1109 /// 1110 /// * recipient: Address of token recipient.1111 /// 1112 /// * collection_id.1113 /// 1114 /// * item_id: ID of the item.1115 /// 1116 /// * value: Amount to transfer.1117 #[weight = T::WeightInfo::transfer_from()]1118 pub fn transfer_from(origin, from: T::AccountId, recipient: T::AccountId, collection_id: CollectionId, item_id: TokenId, value: u64 ) -> DispatchResult {11191120 let sender = ensure_signed(origin)?;1121 let mut appoved_transfer = false;11221123 // Check approve1124 if <ApprovedList<T>>::contains_key(collection_id, (item_id, from.clone())) {1125 let list_itm = <ApprovedList<T>>::get(collection_id, (item_id, from.clone()));1126 let opt_item = list_itm.iter().find(|i| i.approved == sender.clone());1127 if opt_item.is_some()1128 {1129 appoved_transfer = true;1130 ensure!(opt_item.unwrap().amount >= value, Error::<T>::TokenValueNotEnough);1131 }1132 }11331134 // Transfer permissions check1135 let target_collection = <Collection<T>>::get(collection_id);1136 ensure!(appoved_transfer || Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1137 Error::<T>::NoPermission);11381139 if target_collection.access == AccessMode::WhiteList {1140 Self::check_white_list(collection_id, &sender)?;1141 Self::check_white_list(collection_id, &recipient)?;1142 }11431144 // remove approve1145 let approve_list: Vec<ApprovePermissions<T::AccountId>> = <ApprovedList<T>>::get(collection_id, (item_id, from.clone()))1146 .into_iter().filter(|i| i.approved != sender.clone()).collect();1147 <ApprovedList<T>>::insert(collection_id, (item_id, from.clone()), approve_list);114811491150 match target_collection.mode1151 {1152 CollectionMode::NFT => Self::transfer_nft(collection_id, item_id, from, recipient)?,1153 CollectionMode::Fungible(_) => Self::transfer_fungible(collection_id, item_id, value, from.clone(), recipient)?,1154 CollectionMode::ReFungible(_) => Self::transfer_refungible(collection_id, item_id, value, from.clone(), recipient)?,1155 _ => ()1156 };11571158 Ok(())1159 }11601161 ///1162 #[weight = 0]1163 pub fn safe_transfer_from(origin, collection_id: CollectionId, item_id: TokenId, new_owner: T::AccountId) -> DispatchResult {11641165 // let no_perm_mes = "You do not have permissions to modify this collection";1166 // ensure!(<ApprovedList<T>>::contains_key((collection_id, item_id)), no_perm_mes);1167 // let list_itm = <ApprovedList<T>>::get((collection_id, item_id));1168 // ensure!(list_itm.contains(&new_owner.clone()), no_perm_mes);11691170 // // on_nft_received call11711172 // Self::transfer(origin, collection_id, item_id, new_owner)?;11731174 Ok(())1175 }11761177 /// Set off-chain data schema.1178 /// 1179 /// # Permissions1180 /// 1181 /// * Collection Owner1182 /// * Collection Admin1183 /// 1184 /// # Arguments1185 /// 1186 /// * collection_id.1187 /// 1188 /// * schema: String representing the offchain data schema.1189 #[weight = T::WeightInfo::set_variable_meta_data()]1190 pub fn set_variable_meta_data (1191 origin,1192 collection_id: CollectionId,1193 item_id: TokenId,1194 data: Vec<u8>1195 ) -> DispatchResult {1196 let sender = ensure_signed(origin)?;1197 1198 Self::collection_exists(collection_id)?;1199 1200 ensure!(ChainLimit::get().custom_data_limit >= data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);12011202 // Modify permissions check1203 let target_collection = <Collection<T>>::get(collection_id);1204 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1205 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1206 Error::<T>::NoPermission);12071208 Self::item_exists(collection_id, item_id, &target_collection.mode)?;12091210 match target_collection.mode1211 {1212 CollectionMode::NFT => Self::set_nft_variable_data(collection_id, item_id, data)?,1213 CollectionMode::ReFungible(_) => Self::set_re_fungible_variable_data(collection_id, item_id, data)?,1214 CollectionMode::Fungible(_) => fail!(Error::<T>::CantStoreMetadataInFungibleTokens),1215 _ => fail!(Error::<T>::UnexpectedCollectionType)1216 };12171218 Ok(())1219 }1220 12211222 /// Set off-chain data schema.1223 /// 1224 /// # Permissions1225 /// 1226 /// * Collection Owner1227 /// * Collection Admin1228 /// 1229 /// # Arguments1230 /// 1231 /// * collection_id.1232 /// 1233 /// * schema: String representing the offchain data schema.1234 #[weight = T::WeightInfo::set_offchain_schema()]1235 pub fn set_offchain_schema(1236 origin,1237 collection_id: CollectionId,1238 schema: Vec<u8>1239 ) -> DispatchResult {1240 let sender = ensure_signed(origin)?;1241 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;12421243 let mut target_collection = <Collection<T>>::get(collection_id);1244 target_collection.offchain_schema = schema;1245 <Collection<T>>::insert(collection_id, target_collection);12461247 Ok(())1248 }12491250 /// Set const on-chain data schema.1251 /// 1252 /// # Permissions1253 /// 1254 /// * Collection Owner1255 /// * Collection Admin1256 /// 1257 /// # Arguments1258 /// 1259 /// * collection_id.1260 /// 1261 /// * schema: String representing the const on-chain data schema.1262 #[weight = T::WeightInfo::set_const_on_chain_schema()]1263 pub fn set_const_on_chain_schema (1264 origin,1265 collection_id: CollectionId,1266 schema: Vec<u8>1267 ) -> DispatchResult {1268 let sender = ensure_signed(origin)?;1269 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;12701271 let mut target_collection = <Collection<T>>::get(collection_id);1272 target_collection.const_on_chain_schema = schema;1273 <Collection<T>>::insert(collection_id, target_collection);12741275 Ok(())1276 }12771278 /// Set variable on-chain data schema.1279 /// 1280 /// # Permissions1281 /// 1282 /// * Collection Owner1283 /// * Collection Admin1284 /// 1285 /// # Arguments1286 /// 1287 /// * collection_id.1288 /// 1289 /// * schema: String representing the variable on-chain data schema.1290 #[weight = T::WeightInfo::set_const_on_chain_schema()]1291 pub fn set_variable_on_chain_schema (1292 origin,1293 collection_id: CollectionId,1294 schema: Vec<u8>1295 ) -> DispatchResult {1296 let sender = ensure_signed(origin)?;1297 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;12981299 let mut target_collection = <Collection<T>>::get(collection_id);1300 target_collection.variable_on_chain_schema = schema;1301 <Collection<T>>::insert(collection_id, target_collection);13021303 Ok(())1304 }13051306 // Sudo permissions function1307 #[weight = 0]1308 pub fn set_chain_limits(1309 origin,1310 limits: ChainLimits1311 ) -> DispatchResult {1312 ensure_root(origin)?;1313 <ChainLimit>::put(limits);1314 Ok(())1315 }13161317 /// Enable smart contract self-sponsoring.1318 /// 1319 /// # Permissions1320 /// 1321 /// * Contract Owner1322 /// 1323 /// # Arguments1324 /// 1325 /// * contract address1326 /// * enable flag1327 /// 1328 #[weight = T::WeightInfo::enable_contract_sponsoring()]1329 pub fn enable_contract_sponsoring(1330 origin,1331 contract_address: T::AccountId,1332 enable: bool1333 ) -> DispatchResult {13341335 let sender = ensure_signed(origin)?;13361337 #[cfg(feature = "runtime-benchmarks")]1338 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());13391340 let mut is_owner = false;1341 if <ContractOwner<T>>::contains_key(contract_address.clone()) {1342 let owner = <ContractOwner<T>>::get(&contract_address);1343 is_owner = sender == owner;1344 }1345 ensure!(is_owner, Error::<T>::NoPermission);13461347 <ContractSelfSponsoring<T>>::insert(contract_address, enable);1348 Ok(())1349 }13501351 /// Set the rate limit for contract sponsoring to specified number of blocks.1352 /// 1353 /// If not set (has the default value of 0 blocks), the sponsoring will be disabled. 1354 /// If set to the number B (for blocks), the transactions will be sponsored with a rate 1355 /// limit of B, i.e. fees for every transaction sent to this smart contract will be paid 1356 /// from contract endowment if there are at least B blocks between such transactions. 1357 /// Nonetheless, if transactions are sent more frequently, the fees are paid by the sender.1358 /// 1359 /// # Permissions1360 /// 1361 /// * Contract Owner1362 /// 1363 /// # Arguments1364 /// 1365 /// -`contract_address`: Address of the contract to sponsor1366 /// -`rate_limit`: Number of blocks to wait until the next sponsored transaction is allowed1367 /// 1368 #[weight = 0]1369 pub fn set_contract_sponsoring_rate_limit(1370 origin,1371 contract_address: T::AccountId,1372 rate_limit: T::BlockNumber1373 ) -> DispatchResult {1374 let sender = ensure_signed(origin)?;1375 let mut is_owner = false;1376 if <ContractOwner<T>>::contains_key(contract_address.clone()) {1377 let owner = <ContractOwner<T>>::get(&contract_address);1378 is_owner = sender == owner;1379 }1380 ensure!(is_owner, Error::<T>::NoPermission);13811382 <ContractSponsoringRateLimit<T>>::insert(contract_address, rate_limit);1383 Ok(())1384 }13851386 // #[cfg(feature = "runtime-benchmarks")]1387 // #[weight = 0]1388 // pub fn add_contract_sponsoring_debug(1389 // origin,1390 // contract_address: T::AccountId, 1391 // owner: T::AccountId) -> DispatchResult {1392 // let sender = ensure_signed(origin)?;1393 // <ContractOwner<T>>::insert(contract_address.clone(), owner);1394 // Ok(())1395 // }1396 1397 }1398}13991400impl<T: Trait> Module<T> {14011402 fn can_create_items_in_collection(collection_id: CollectionId, collection: &CollectionType<T::AccountId>, sender: &T::AccountId, owner: &T::AccountId) -> DispatchResult {14031404 if !Self::is_owner_or_admin_permissions(collection_id, sender.clone()) {1405 ensure!(collection.mint_mode == true, Error::<T>::PublicMintingNotAllowed);1406 Self::check_white_list(collection_id, owner)?;1407 Self::check_white_list(collection_id, sender)?;1408 }14091410 Ok(())1411 }14121413 fn validate_create_item_args(target_collection: &CollectionType<T::AccountId>, data: &CreateItemData) -> DispatchResult {1414 match target_collection.mode1415 {1416 CollectionMode::NFT => {1417 if let CreateItemData::NFT(data) = data {1418 // check sizes1419 ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, Error::<T>::TokenConstDataLimitExceeded);1420 ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);1421 } else {1422 fail!(Error::<T>::NotNftDataUsedToMintNftCollectionToken);1423 }1424 },1425 CollectionMode::Fungible(_) => {1426 if let CreateItemData::Fungible(_) = data {1427 } else {1428 fail!(Error::<T>::NotFungibleDataUsedToMintFungibleCollectionToken);1429 }1430 },1431 CollectionMode::ReFungible(_) => {1432 if let CreateItemData::ReFungible(data) = data {14331434 // check sizes1435 ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, Error::<T>::TokenConstDataLimitExceeded);1436 ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);1437 } else {1438 fail!(Error::<T>::NotReFungibleDataUsedToMintReFungibleCollectionToken);1439 }1440 },1441 _ => { fail!(Error::<T>::UnexpectedCollectionType); }1442 };14431444 Ok(())1445 }14461447 fn create_item_no_validation(collection_id: CollectionId, collection: &CollectionType<T::AccountId>, owner: T::AccountId, data: CreateItemData) -> DispatchResult {1448 match data1449 {1450 CreateItemData::NFT(data) => {1451 let item = NftItemType {1452 collection: collection_id,1453 owner,1454 const_data: data.const_data,1455 variable_data: data.variable_data1456 };14571458 Self::add_nft_item(item)?;1459 },1460 CreateItemData::Fungible(_) => {1461 let item = FungibleItemType {1462 collection: collection_id,1463 owner,1464 value: (10 as u128).pow(collection.decimal_points)1465 };14661467 Self::add_fungible_item(item)?;1468 },1469 CreateItemData::ReFungible(data) => {1470 let mut owner_list = Vec::new();1471 let value = (10 as u128).pow(collection.decimal_points);1472 owner_list.push(Ownership {owner: owner.clone(), fraction: value});14731474 let item = ReFungibleItemType {1475 collection: collection_id,1476 owner: owner_list,1477 const_data: data.const_data,1478 variable_data: data.variable_data1479 };14801481 Self::add_refungible_item(item)?;1482 }1483 };148414851486 // call event1487 Self::deposit_event(RawEvent::ItemCreated(collection_id, <ItemListIndex>::get(collection_id)));14881489 Ok(())1490 }14911492 fn add_fungible_item(item: FungibleItemType<T::AccountId>) -> DispatchResult {1493 let current_index = <ItemListIndex>::get(item.collection)1494 .checked_add(1)1495 .ok_or(Error::<T>::NumOverflow)?;1496 let itemcopy = item.clone();1497 let owner = item.owner.clone();1498 let value = item.value as u64;14991500 Self::add_token_index(item.collection, current_index, owner.clone())?;15011502 <ItemListIndex>::insert(item.collection, current_index);1503 <FungibleItemList<T>>::insert(item.collection, current_index, itemcopy);15041505 // Add current block1506 let v: Vec<BasketItem<T::AccountId, T::BlockNumber>> = Vec::new();1507 <FungibleTransferBasket<T>>::insert(item.collection, current_index, v);1508 1509 // Update balance1510 let new_balance = <Balance<T>>::get(item.collection, owner.clone())1511 .checked_add(value)1512 .ok_or(Error::<T>::NumOverflow)?;1513 <Balance<T>>::insert(item.collection, owner.clone(), new_balance);15141515 Ok(())1516 }15171518 fn add_refungible_item(item: ReFungibleItemType<T::AccountId>) -> DispatchResult {1519 let current_index = <ItemListIndex>::get(item.collection)1520 .checked_add(1)1521 .ok_or(Error::<T>::NumOverflow)?;1522 let itemcopy = item.clone();15231524 let value = item.owner.first().unwrap().fraction as u64;1525 let owner = item.owner.first().unwrap().owner.clone();15261527 Self::add_token_index(item.collection, current_index, owner.clone())?;15281529 <ItemListIndex>::insert(item.collection, current_index);1530 <ReFungibleItemList<T>>::insert(item.collection, current_index, itemcopy);15311532 // Add current block1533 let block_number: T::BlockNumber = 0.into();1534 <ReFungibleTransferBasket<T>>::insert(item.collection, current_index, block_number);15351536 // Update balance1537 let new_balance = <Balance<T>>::get(item.collection, owner.clone())1538 .checked_add(value)1539 .ok_or(Error::<T>::NumOverflow)?;1540 <Balance<T>>::insert(item.collection, owner.clone(), new_balance);15411542 Ok(())1543 }15441545 fn add_nft_item(item: NftItemType<T::AccountId>) -> DispatchResult {1546 let current_index = <ItemListIndex>::get(item.collection)1547 .checked_add(1)1548 .ok_or(Error::<T>::NumOverflow)?;15491550 let item_owner = item.owner.clone();1551 let collection_id = item.collection.clone();1552 Self::add_token_index(collection_id, current_index, item.owner.clone())?;15531554 <ItemListIndex>::insert(collection_id, current_index);1555 <NftItemList<T>>::insert(collection_id, current_index, item);15561557 // Add current block1558 let block_number: T::BlockNumber = 0.into();1559 <NftTransferBasket<T>>::insert(collection_id, current_index, block_number);15601561 // Update balance1562 let new_balance = <Balance<T>>::get(collection_id, item_owner.clone())1563 .checked_add(1)1564 .ok_or(Error::<T>::NumOverflow)?;1565 <Balance<T>>::insert(collection_id, item_owner.clone(), new_balance);15661567 Ok(())1568 }15691570 fn burn_refungible_item(1571 collection_id: CollectionId,1572 item_id: TokenId,1573 owner: T::AccountId,1574 ) -> DispatchResult {1575 ensure!(1576 <ReFungibleItemList<T>>::contains_key(collection_id, item_id),1577 Error::<T>::TokenNotFound1578 );1579 let collection = <ReFungibleItemList<T>>::get(collection_id, item_id);1580 let item = collection1581 .owner1582 .iter()1583 .filter(|&i| i.owner == owner)1584 .next()1585 .unwrap();1586 Self::remove_token_index(collection_id, item_id, owner.clone())?;15871588 // remove approve list1589 <ApprovedList<T>>::remove(collection_id, (item_id, owner.clone()));15901591 // update balance1592 let new_balance = <Balance<T>>::get(collection_id, item.owner.clone())1593 .checked_sub(item.fraction as u64)1594 .ok_or(Error::<T>::NumOverflow)?;1595 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);15961597 <ReFungibleItemList<T>>::remove(collection_id, item_id);15981599 Ok(())1600 }16011602 fn burn_nft_item(collection_id: CollectionId, item_id: TokenId) -> DispatchResult {1603 ensure!(1604 <NftItemList<T>>::contains_key(collection_id, item_id),1605 Error::<T>::TokenNotFound1606 );1607 let item = <NftItemList<T>>::get(collection_id, item_id);1608 Self::remove_token_index(collection_id, item_id, item.owner.clone())?;16091610 // remove approve list1611 <ApprovedList<T>>::remove(collection_id, (item_id, item.owner.clone()));16121613 // update balance1614 let new_balance = <Balance<T>>::get(collection_id, item.owner.clone())1615 .checked_sub(1)1616 .ok_or(Error::<T>::NumOverflow)?;1617 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);1618 <NftItemList<T>>::remove(collection_id, item_id);16191620 Ok(())1621 }16221623 fn burn_fungible_item(collection_id: CollectionId, item_id: TokenId) -> DispatchResult {1624 ensure!(1625 <FungibleItemList<T>>::contains_key(collection_id, item_id),1626 Error::<T>::TokenNotFound1627 );1628 let item = <FungibleItemList<T>>::get(collection_id, item_id);1629 Self::remove_token_index(collection_id, item_id, item.owner.clone())?;16301631 // remove approve list1632 <ApprovedList<T>>::remove(collection_id, (item_id, item.owner.clone()));16331634 // update balance1635 let new_balance = <Balance<T>>::get(collection_id, item.owner.clone())1636 .checked_sub(item.value as u64)1637 .ok_or(Error::<T>::NumOverflow)?;1638 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);16391640 <FungibleItemList<T>>::remove(collection_id, item_id);16411642 Ok(())1643 }16441645 fn collection_exists(collection_id: CollectionId) -> DispatchResult {1646 ensure!(1647 <Collection<T>>::contains_key(collection_id),1648 Error::<T>::CollectionNotFound1649 );1650 Ok(())1651 }16521653 fn check_owner_permissions(collection_id: CollectionId, subject: T::AccountId) -> DispatchResult {1654 Self::collection_exists(collection_id)?;16551656 let target_collection = <Collection<T>>::get(collection_id);1657 ensure!(1658 subject == target_collection.owner,1659 Error::<T>::NoPermission1660 );16611662 Ok(())1663 }16641665 fn is_owner_or_admin_permissions(collection_id: CollectionId, subject: T::AccountId) -> bool {1666 let target_collection = <Collection<T>>::get(collection_id);1667 let mut result: bool = subject == target_collection.owner;1668 let exists = <AdminList<T>>::contains_key(collection_id);16691670 if !result & exists {1671 if <AdminList<T>>::get(collection_id).contains(&subject) {1672 result = true1673 }1674 }16751676 result1677 }16781679 fn check_owner_or_admin_permissions(1680 collection_id: CollectionId,1681 subject: T::AccountId,1682 ) -> DispatchResult {1683 Self::collection_exists(collection_id)?;1684 let result = Self::is_owner_or_admin_permissions(collection_id, subject.clone());16851686 ensure!(1687 result,1688 Error::<T>::NoPermission1689 );1690 Ok(())1691 }16921693 fn is_item_owner(subject: T::AccountId, collection_id: CollectionId, item_id: TokenId) -> bool {1694 let target_collection = <Collection<T>>::get(collection_id);16951696 match target_collection.mode {1697 CollectionMode::NFT => {1698 <NftItemList<T>>::get(collection_id, item_id).owner == subject1699 }1700 CollectionMode::Fungible(_) => {1701 <FungibleItemList<T>>::get(collection_id, item_id).owner == subject1702 }1703 CollectionMode::ReFungible(_) => {1704 <ReFungibleItemList<T>>::get(collection_id, item_id)1705 .owner1706 .iter()1707 .any(|i| i.owner == subject)1708 }1709 CollectionMode::Invalid => false,1710 }1711 }17121713 fn check_white_list(collection_id: CollectionId, address: &T::AccountId) -> DispatchResult {1714 let mes = Error::<T>::AddresNotInWhiteList;1715 ensure!(<WhiteList<T>>::contains_key(collection_id), mes);1716 let wl = <WhiteList<T>>::get(collection_id);1717 ensure!(wl.contains(address), mes);17181719 Ok(())1720 }17211722 fn transfer_fungible(1723 collection_id: CollectionId,1724 item_id: TokenId,1725 value: u64,1726 owner: T::AccountId,1727 new_owner: T::AccountId,1728 ) -> DispatchResult {1729 ensure!(1730 <FungibleItemList<T>>::contains_key(collection_id, item_id),1731 Error::<T>::TokenNotFound1732 );17331734 let full_item = <FungibleItemList<T>>::get(collection_id, item_id);1735 let amount = full_item.value;17361737 ensure!(amount >= value.into(), Error::<T>::TokenValueTooLow);17381739 // update balance1740 let balance_old_owner = <Balance<T>>::get(collection_id, owner.clone())1741 .checked_sub(value)1742 .ok_or(Error::<T>::NumOverflow)?;1743 <Balance<T>>::insert(collection_id, owner.clone(), balance_old_owner);17441745 let mut new_owner_account_id = 0;1746 let new_owner_items = <AddressTokens<T>>::get(collection_id, new_owner.clone());1747 if new_owner_items.len() > 0 {1748 new_owner_account_id = new_owner_items[0];1749 }17501751 let val64 = value.into();17521753 // transfer1754 if amount == val64 && new_owner_account_id == 0 {1755 // change owner1756 // new owner do not have account1757 let mut new_full_item = full_item.clone();1758 new_full_item.owner = new_owner.clone();1759 <FungibleItemList<T>>::insert(collection_id, item_id, new_full_item);17601761 // update balance1762 let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone())1763 .checked_add(value)1764 .ok_or(Error::<T>::NumOverflow)?;1765 <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);17661767 // update index collection1768 Self::move_token_index(collection_id, item_id, owner.clone(), new_owner.clone())?;1769 } else {1770 let mut new_full_item = full_item.clone();1771 new_full_item.value -= val64;17721773 // separate amount1774 if new_owner_account_id > 0 {1775 // new owner has account1776 let mut item = <FungibleItemList<T>>::get(collection_id, new_owner_account_id);1777 item.value += val64;17781779 // update balance1780 let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone())1781 .checked_add(value)1782 .ok_or(Error::<T>::NumOverflow)?;1783 <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);17841785 <FungibleItemList<T>>::insert(collection_id, new_owner_account_id, item);1786 } else {1787 // new owner do not have account1788 let item = FungibleItemType {1789 collection: collection_id,1790 owner: new_owner.clone(),1791 value: val64,1792 };17931794 Self::add_fungible_item(item)?;1795 }17961797 if amount == val64 {1798 Self::remove_token_index(collection_id, item_id, full_item.owner.clone())?;17991800 // remove approve list1801 <ApprovedList<T>>::remove(collection_id, (item_id, full_item.owner.clone()));1802 <FungibleItemList<T>>::remove(collection_id, item_id);1803 }18041805 <FungibleItemList<T>>::insert(collection_id, item_id, new_full_item);1806 }18071808 Ok(())1809 }18101811 fn transfer_refungible(1812 collection_id: CollectionId,1813 item_id: TokenId,1814 value: u64,1815 owner: T::AccountId,1816 new_owner: T::AccountId,1817 ) -> DispatchResult {1818 ensure!(1819 <ReFungibleItemList<T>>::contains_key(collection_id, item_id),1820 Error::<T>::TokenNotFound1821 );18221823 let full_item = <ReFungibleItemList<T>>::get(collection_id, item_id);1824 let item = full_item1825 .owner1826 .iter()1827 .filter(|i| i.owner == owner)1828 .next()1829 .ok_or(Error::<T>::NumOverflow)?;1830 let amount = item.fraction;18311832 ensure!(amount >= value.into(), Error::<T>::TokenValueTooLow);18331834 // update balance1835 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())1836 .checked_sub(value)1837 .ok_or(Error::<T>::NumOverflow)?;1838 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);18391840 let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone())1841 .checked_add(value)1842 .ok_or(Error::<T>::NumOverflow)?;1843 <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);18441845 let old_owner = item.owner.clone();1846 let new_owner_has_account = full_item.owner.iter().any(|i| i.owner == new_owner);1847 let val64 = value.into();18481849 // transfer1850 if amount == val64 && !new_owner_has_account {1851 // change owner1852 // new owner do not have account1853 let mut new_full_item = full_item.clone();1854 new_full_item1855 .owner1856 .iter_mut()1857 .find(|i| i.owner == owner)1858 .unwrap()1859 .owner = new_owner.clone();1860 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);18611862 // update index collection1863 Self::move_token_index(collection_id, item_id, old_owner.clone(), new_owner.clone())?;1864 } else {1865 let mut new_full_item = full_item.clone();1866 new_full_item1867 .owner1868 .iter_mut()1869 .find(|i| i.owner == owner)1870 .unwrap()1871 .fraction -= val64;18721873 // separate amount1874 if new_owner_has_account {1875 // new owner has account1876 new_full_item1877 .owner1878 .iter_mut()1879 .find(|i| i.owner == new_owner)1880 .unwrap()1881 .fraction += val64;1882 } else {1883 // new owner do not have account1884 new_full_item.owner.push(Ownership {1885 owner: new_owner.clone(),1886 fraction: val64,1887 });1888 Self::add_token_index(collection_id, item_id, new_owner.clone())?;1889 }18901891 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);1892 }18931894 Ok(())1895 }18961897 fn transfer_nft(1898 collection_id: CollectionId,1899 item_id: TokenId,1900 sender: T::AccountId,1901 new_owner: T::AccountId,1902 ) -> DispatchResult {1903 ensure!(1904 <NftItemList<T>>::contains_key(collection_id, item_id),1905 Error::<T>::TokenNotFound1906 );19071908 let mut item = <NftItemList<T>>::get(collection_id, item_id);19091910 ensure!(1911 sender == item.owner,1912 Error::<T>::MustBeTokenOwner1913 );19141915 // update balance1916 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())1917 .checked_sub(1)1918 .ok_or(Error::<T>::NumOverflow)?;1919 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);19201921 let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone())1922 .checked_add(1)1923 .ok_or(Error::<T>::NumOverflow)?;1924 <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);19251926 // change owner1927 let old_owner = item.owner.clone();1928 item.owner = new_owner.clone();1929 <NftItemList<T>>::insert(collection_id, item_id, item);19301931 // update index collection1932 Self::move_token_index(collection_id, item_id, old_owner.clone(), new_owner.clone())?;19331934 // reset approved list1935 <ApprovedList<T>>::remove(collection_id, (item_id, old_owner));1936 Ok(())1937 }1938 1939 fn item_exists(1940 collection_id: CollectionId,1941 item_id: TokenId,1942 mode: &CollectionMode1943 ) -> DispatchResult {1944 match mode {1945 CollectionMode::NFT => ensure!(<NftItemList<T>>::contains_key(collection_id, item_id), Error::<T>::TokenNotFound),1946 CollectionMode::ReFungible(_) => ensure!(<ReFungibleItemList<T>>::contains_key(collection_id, item_id), Error::<T>::TokenNotFound),1947 CollectionMode::Fungible(_) => ensure!(<FungibleItemList<T>>::contains_key(collection_id, item_id), Error::<T>::TokenNotFound),1948 _ => ()1949 };1950 1951 Ok(())1952 }19531954 fn set_re_fungible_variable_data(1955 collection_id: CollectionId,1956 item_id: TokenId,1957 data: Vec<u8>1958 ) -> DispatchResult {1959 let mut item = <ReFungibleItemList<T>>::get(collection_id, item_id);19601961 item.variable_data = data;19621963 <ReFungibleItemList<T>>::insert(collection_id, item_id, item);19641965 Ok(())1966 }19671968 fn set_nft_variable_data(1969 collection_id: CollectionId,1970 item_id: TokenId,1971 data: Vec<u8>1972 ) -> DispatchResult {1973 let mut item = <NftItemList<T>>::get(collection_id, item_id);1974 1975 item.variable_data = data;19761977 <NftItemList<T>>::insert(collection_id, item_id, item);1978 1979 Ok(())1980 }19811982 fn init_collection(item: &CollectionType<T::AccountId>) {1983 // check params1984 assert!(1985 item.decimal_points <= 4,1986 "decimal_points parameter must be lower than 4"1987 );1988 assert!(1989 item.name.len() <= 64,1990 "Collection name can not be longer than 63 char"1991 );1992 assert!(1993 item.name.len() <= 256,1994 "Collection description can not be longer than 255 char"1995 );1996 assert!(1997 item.token_prefix.len() <= 16,1998 "Token prefix can not be longer than 15 char"1999 );20002001 // Generate next collection ID2002 let next_id = CreatedCollectionCount::get()2003 .checked_add(1)2004 .unwrap();20052006 CreatedCollectionCount::put(next_id);2007 }20082009 fn init_nft_token(item: &NftItemType<T::AccountId>) {2010 let current_index = <ItemListIndex>::get(item.collection)2011 .checked_add(1)2012 .unwrap();20132014 let item_owner = item.owner.clone();2015 let collection_id = item.collection.clone();2016 Self::add_token_index(collection_id, current_index, item.owner.clone()).unwrap();20172018 <ItemListIndex>::insert(collection_id, current_index);20192020 // Update balance2021 let new_balance = <Balance<T>>::get(collection_id, item_owner.clone())2022 .checked_add(1)2023 .unwrap();2024 <Balance<T>>::insert(collection_id, item_owner.clone(), new_balance);2025 }20262027 fn init_fungible_token(item: &FungibleItemType<T::AccountId>) {2028 let current_index = <ItemListIndex>::get(item.collection)2029 .checked_add(1)2030 .unwrap();2031 let owner = item.owner.clone();2032 let value = item.value as u64;20332034 Self::add_token_index(item.collection, current_index, owner.clone()).unwrap();20352036 <ItemListIndex>::insert(item.collection, current_index);20372038 // Update balance2039 let new_balance = <Balance<T>>::get(item.collection, owner.clone())2040 .checked_add(value)2041 .unwrap();2042 <Balance<T>>::insert(item.collection, owner.clone(), new_balance);2043 }20442045 fn init_refungible_token(item: &ReFungibleItemType<T::AccountId>) {2046 let current_index = <ItemListIndex>::get(item.collection)2047 .checked_add(1)2048 .unwrap();20492050 let value = item.owner.first().unwrap().fraction as u64;2051 let owner = item.owner.first().unwrap().owner.clone();20522053 Self::add_token_index(item.collection, current_index, owner.clone()).unwrap();20542055 <ItemListIndex>::insert(item.collection, current_index);20562057 // Update balance2058 let new_balance = <Balance<T>>::get(item.collection, owner.clone())2059 .checked_add(value)2060 .unwrap();2061 <Balance<T>>::insert(item.collection, owner.clone(), new_balance);2062 }20632064 fn add_token_index(collection_id: CollectionId, item_index: TokenId, owner: T::AccountId) -> DispatchResult {20652066 // add to account limit2067 if <AccountItemCount<T>>::contains_key(owner.clone()) {20682069 // bound Owned tokens by a single address2070 let count = <AccountItemCount<T>>::get(owner.clone());2071 ensure!(count < ChainLimit::get().account_token_ownership_limit, Error::<T>::AddressOwnershipLimitExceeded);20722073 <AccountItemCount<T>>::insert(owner.clone(), count2074 .checked_add(1)2075 .ok_or(Error::<T>::NumOverflow)?);2076 }2077 else {2078 <AccountItemCount<T>>::insert(owner.clone(), 1);2079 }20802081 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());2082 if list_exists {2083 let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());2084 let item_contains = list.contains(&item_index.clone());20852086 if !item_contains {2087 list.push(item_index.clone());2088 }20892090 <AddressTokens<T>>::insert(collection_id, owner.clone(), list);2091 } else {2092 let mut itm = Vec::new();2093 itm.push(item_index.clone());2094 <AddressTokens<T>>::insert(collection_id, owner, itm);2095 2096 }20972098 Ok(())2099 }21002101 fn remove_token_index(2102 collection_id: CollectionId,2103 item_index: TokenId,2104 owner: T::AccountId,2105 ) -> DispatchResult {21062107 // update counter2108 <AccountItemCount<T>>::insert(owner.clone(), 2109 <AccountItemCount<T>>::get(owner.clone())2110 .checked_sub(1)2111 .ok_or(Error::<T>::NumOverflow)?);211221132114 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());2115 if list_exists {2116 let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());2117 let item_contains = list.contains(&item_index.clone());21182119 if item_contains {2120 list.retain(|&item| item != item_index);2121 <AddressTokens<T>>::insert(collection_id, owner, list);2122 }2123 }21242125 Ok(())2126 }21272128 fn move_token_index(2129 collection_id: CollectionId,2130 item_index: TokenId,2131 old_owner: T::AccountId,2132 new_owner: T::AccountId,2133 ) -> DispatchResult {2134 Self::remove_token_index(collection_id, item_index, old_owner)?;2135 Self::add_token_index(collection_id, item_index, new_owner)?;21362137 Ok(())2138 }2139}21402141////////////////////////////////////////////////////////////////////////////////////////////////////2142// Economic models2143// #region21442145/// Fee multiplier.2146pub type Multiplier = FixedU128;21472148type BalanceOf<T> = <<T as transaction_payment::Trait>::Currency as Currency<2149 <T as system::Trait>::AccountId,2150>>::Balance;2151type NegativeImbalanceOf<T> = <<T as transaction_payment::Trait>::Currency as Currency<2152 <T as system::Trait>::AccountId,2153>>::NegativeImbalance;21542155/// Require the transactor pay for themselves and maybe include a tip to gain additional priority2156/// in the queue.2157#[derive(Encode, Decode, Clone, Eq, PartialEq)]2158pub struct ChargeTransactionPayment<T: Trait + Send + Sync>(2159 #[codec(compact)] BalanceOf<T>2160);21612162impl<T: Trait + Send + Sync> sp_std::fmt::Debug2163 for ChargeTransactionPayment<T>2164{2165 #[cfg(feature = "std")]2166 fn fmt(&self, f: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {2167 write!(f, "ChargeTransactionPayment<{:?}>", self.0)2168 }2169 #[cfg(not(feature = "std"))]2170 fn fmt(&self, _: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {2171 Ok(())2172 }2173}21742175impl<T: Trait + Send + Sync> ChargeTransactionPayment<T>2176where2177 T::Call: Dispatchable<Info = DispatchInfo, PostInfo = PostDispatchInfo> + IsSubType<Call<T>> + IsSubType<pallet_contracts::Call<T>>,2178 BalanceOf<T>: Send + Sync + FixedPointOperand,2179{2180 /// utility constructor. Used only in client/factory code.2181 pub fn from(fee: BalanceOf<T>) -> Self {2182 Self(fee)2183 }21842185 pub fn traditional_fee(2186 len: usize,2187 info: &DispatchInfoOf<T::Call>,2188 tip: BalanceOf<T>,2189 ) -> BalanceOf<T>2190 where2191 T::Call: Dispatchable<Info = DispatchInfo>,2192 {2193 <transaction_payment::Module<T>>::compute_fee(len as u32, info, tip)2194 }21952196 fn withdraw_fee(2197 &self,2198 who: &T::AccountId,2199 call: &T::Call,2200 info: &DispatchInfoOf<T::Call>,2201 len: usize,2202 ) -> Result<(BalanceOf<T>, Option<NegativeImbalanceOf<T>>), TransactionValidityError> {2203 let tip = self.0;22042205 // Set fee based on call type. Creating collection costs 1 Unique.2206 // All other transactions have traditional fees so far2207 // let fee = match call.is_sub_type() {2208 // Some(Call::create_collection(..)) => <BalanceOf<T>>::from(1_000_000_000),2209 // _ => Self::traditional_fee(len, info, tip), // Flat fee model, use only for testing purposes2210 // // _ => <BalanceOf<T>>::from(100)2211 // };2212 let fee = Self::traditional_fee(len, info, tip);22132214 // Determine who is paying transaction fee based on ecnomic model2215 // Parse call to extract collection ID and access collection sponsor2216 let mut sponsor: T::AccountId = match IsSubType::<Call<T>>::is_sub_type(call) {2217 Some(Call::create_item(collection_id, _properties, _owner)) => {2218 <Collection<T>>::get(collection_id).sponsor2219 }2220 Some(Call::transfer(_new_owner, collection_id, _item_id, _value)) => {2221 let _collection_mode = <Collection<T>>::get(collection_id).mode;22222223 // sponsor timeout2224 let sponsor_transfer = match _collection_mode {2225 CollectionMode::NFT => {2226 let basket = <NftTransferBasket<T>>::get(collection_id, _item_id);2227 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2228 let limit_time = basket + ChainLimit::get().nft_sponsor_transfer_timeout.into();2229 if block_number >= limit_time {2230 <NftTransferBasket<T>>::insert(collection_id, _item_id, block_number);2231 true2232 }2233 else {2234 false2235 }2236 }2237 CollectionMode::Fungible(_) => {2238 let mut basket = <FungibleTransferBasket<T>>::get(collection_id, _item_id);2239 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2240 if basket.iter().any(|i| i.address == _new_owner.clone())2241 {2242 let item = basket.iter_mut().find(|i| i.address == _new_owner.clone()).unwrap().clone();2243 let limit_time = item.start_block + ChainLimit::get().fungible_sponsor_transfer_timeout.into();2244 if block_number >= limit_time {2245 basket.retain(|x| x.address == item.address);2246 basket.push(BasketItem { start_block: block_number, address: _new_owner.clone() });2247 <FungibleTransferBasket<T>>::insert(collection_id, _item_id, basket);2248 true2249 }2250 else {2251 false2252 }2253 }2254 else {2255 basket.push(BasketItem { start_block: block_number, address: _new_owner.clone()});2256 true2257 }2258 }2259 CollectionMode::ReFungible(_) => {2260 let basket = <ReFungibleTransferBasket<T>>::get(collection_id, _item_id);2261 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2262 let limit_time = basket + ChainLimit::get().nft_sponsor_transfer_timeout.into();2263 if block_number >= limit_time {2264 <ReFungibleTransferBasket<T>>::insert(collection_id, _item_id, block_number);2265 true2266 } else {2267 false2268 }2269 }2270 _ => {2271 false2272 },2273 };22742275 if !sponsor_transfer {2276 T::AccountId::default()2277 } else {2278 <Collection<T>>::get(collection_id).sponsor2279 }2280 }22812282 _ => T::AccountId::default(),2283 };22842285 // Sponsor smart contracts2286 sponsor = match IsSubType::<pallet_contracts::Call<T>>::is_sub_type(call) {22872288 // On instantiation: set the contract owner2289 Some(pallet_contracts::Call::instantiate(_endowment, _gas_limit, code_hash, data)) => {22902291 let new_contract_address = <T as pallet_contracts::Trait>::DetermineContractAddress::contract_address_for(2292 code_hash,2293 &data,2294 &who,2295 );2296 <ContractOwner<T>>::insert(new_contract_address.clone(), who.clone());22972298 T::AccountId::default()2299 },23002301 // When the contract is called, check if the sponsoring is enabled and pay fees from contract endowment if it is2302 Some(pallet_contracts::Call::call(dest, _value, _gas_limit, _data)) => {23032304 let called_contract: T::AccountId = T::Lookup::lookup((*dest).clone()).unwrap_or(T::AccountId::default());23052306 let mut sponsor_transfer = false;2307 if <ContractSponsoringRateLimit<T>>::contains_key(called_contract.clone()) {2308 let last_tx_block = <ContractSponsorBasket<T>>::get(&called_contract);2309 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2310 let rate_limit = <ContractSponsoringRateLimit<T>>::get(&called_contract);2311 let limit_time = last_tx_block + rate_limit;23122313 if block_number >= limit_time {2314 <ContractSponsorBasket<T>>::insert(called_contract.clone(), block_number);2315 sponsor_transfer = true;2316 }2317 } else {2318 sponsor_transfer = false;2319 }2320 2321 2322 let mut sp = T::AccountId::default();2323 if sponsor_transfer {2324 if <ContractSelfSponsoring<T>>::contains_key(called_contract.clone()) {2325 if <ContractSelfSponsoring<T>>::get(called_contract.clone()) {2326 sp = called_contract;2327 }2328 }2329 }23302331 sp2332 },23332334 _ => sponsor,2335 };23362337 let mut who_pays_fee: T::AccountId = sponsor.clone();2338 if sponsor == T::AccountId::default() {2339 who_pays_fee = who.clone();2340 }23412342 // Only mess with balances if fee is not zero.2343 if fee.is_zero() {2344 return Ok((fee, None));2345 }23462347 match <T as transaction_payment::Trait>::Currency::withdraw(2348 &who_pays_fee,2349 fee,2350 if tip.is_zero() {2351 WithdrawReason::TransactionPayment.into()2352 } else {2353 WithdrawReason::TransactionPayment | WithdrawReason::Tip2354 },2355 ExistenceRequirement::KeepAlive,2356 ) {2357 Ok(imbalance) => Ok((fee, Some(imbalance))),2358 Err(_) => Err(InvalidTransaction::Payment.into()),2359 }2360 }2361}236223632364impl<T: Trait + Send + Sync> SignedExtension2365 for ChargeTransactionPayment<T>2366where2367 BalanceOf<T>: Send + Sync + From<u64> + FixedPointOperand,2368 T::Call: Dispatchable<Info = DispatchInfo, PostInfo = PostDispatchInfo> + IsSubType<Call<T>> + IsSubType<pallet_contracts::Call<T>>,2369{2370 const IDENTIFIER: &'static str = "ChargeTransactionPayment";2371 type AccountId = T::AccountId;2372 type Call = T::Call;2373 type AdditionalSigned = ();2374 type Pre = (2375 BalanceOf<T>,2376 Self::AccountId,2377 Option<NegativeImbalanceOf<T>>,2378 BalanceOf<T>,2379 );2380 fn additional_signed(&self) -> sp_std::result::Result<(), TransactionValidityError> {2381 Ok(())2382 }23832384 fn validate(2385 &self,2386 _who: &Self::AccountId,2387 _call: &Self::Call,2388 _info: &DispatchInfoOf<Self::Call>,2389 _len: usize,2390 ) -> TransactionValidity {2391 Ok(ValidTransaction::default())2392 }23932394 fn pre_dispatch(2395 self,2396 who: &Self::AccountId,2397 call: &Self::Call,2398 info: &DispatchInfoOf<Self::Call>,2399 len: usize,2400 ) -> Result<Self::Pre, TransactionValidityError> {2401 let (fee, imbalance) = self.withdraw_fee(who, call, info, len)?;2402 Ok((self.0, who.clone(), imbalance, fee))2403 }24042405 fn post_dispatch(2406 pre: Self::Pre,2407 info: &DispatchInfoOf<Self::Call>,2408 post_info: &PostDispatchInfoOf<Self::Call>,2409 len: usize,2410 _result: &DispatchResult,2411 ) -> Result<(), TransactionValidityError> {2412 let (tip, who, imbalance, fee) = pre;2413 if let Some(payed) = imbalance {2414 let actual_fee = <transaction_payment::Module<T>>::compute_actual_fee(2415 len as u32, info, post_info, tip,2416 );2417 let refund = fee.saturating_sub(actual_fee);2418 let actual_payment =2419 match <T as transaction_payment::Trait>::Currency::deposit_into_existing(2420 &who, refund,2421 ) {2422 Ok(refund_imbalance) => {2423 // The refund cannot be larger than the up front payed max weight.2424 // `PostDispatchInfo::calc_unspent` guards against such a case.2425 match payed.offset(refund_imbalance) {2426 Ok(actual_payment) => actual_payment,2427 Err(_) => return Err(InvalidTransaction::Payment.into()),2428 }2429 }2430 // We do not recreate the account using the refund. The up front payment2431 // is gone in that case.2432 Err(_) => payed,2433 };2434 let imbalances = actual_payment.split(tip);2435 <T as transaction_payment::Trait>::OnTransactionPayment::on_unbalanceds(2436 Some(imbalances.0).into_iter().chain(Some(imbalances.1)),2437 );2438 }2439 Ok(())2440 }2441}24422443// #endregionpallets/nft/src/tests.rsdiffbeforeafterboth--- a/pallets/nft/src/tests.rs
+++ b/pallets/nft/src/tests.rs
@@ -2,11 +2,12 @@
use super::*;
use crate::mock::*;
use crate::{AccessMode, ApprovePermissions, CollectionMode,
- Ownership, ChainLimits, CreateItemData, CreateNftData, CreateFungibleData, CreateReFungibleData}; //Err
+ Ownership, ChainLimits, CreateItemData, CreateNftData, CreateFungibleData, CreateReFungibleData,
+ CollectionId, TokenId}; //Err
use frame_support::{assert_noop, assert_ok};
use frame_system::{ RawOrigin };
-fn default_collection_numbers_limit() -> u64 {
+fn default_collection_numbers_limit() -> u32 {
10
}
@@ -34,7 +35,7 @@
CreateReFungibleData { const_data: vec![1, 2, 3], variable_data: vec![3, 2, 1] }
}
-fn create_test_collection_for_owner(mode: &CollectionMode, owner: u64, id: u64) -> u64 {
+fn create_test_collection_for_owner(mode: &CollectionMode, owner: u64, id: CollectionId) -> CollectionId {
let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
@@ -59,11 +60,11 @@
id
}
-fn create_test_collection(mode: &CollectionMode, id: u64) -> u64 {
+fn create_test_collection(mode: &CollectionMode, id: CollectionId) -> CollectionId {
create_test_collection_for_owner(&mode, 1, id)
}
-fn create_test_item(collection_id: u64, data: &CreateItemData) {
+fn create_test_item(collection_id: CollectionId, data: &CreateItemData) {
let origin1 = Origin::signed(1);
assert_ok!(TemplateModule::create_item(
origin1.clone(),
@@ -109,8 +110,8 @@
items_data.clone().into_iter().map(|d| { d.into() }).collect()
));
for (index, data) in items_data.iter().enumerate() {
- assert_eq!(TemplateModule::nft_item_id(1, (index + 1) as u64).const_data.to_vec(), data.const_data);
- assert_eq!(TemplateModule::nft_item_id(1, (index + 1) as u64).variable_data.to_vec(), data.variable_data);
+ assert_eq!(TemplateModule::nft_item_id(1, (index + 1) as TokenId).const_data.to_vec(), data.const_data);
+ assert_eq!(TemplateModule::nft_item_id(1, (index + 1) as TokenId).variable_data.to_vec(), data.variable_data);
}
});
}
@@ -160,7 +161,7 @@
));
for (index, data) in items_data.iter().enumerate() {
- let item = TemplateModule::refungible_item_id(1, (index + 1) as u64);
+ let item = TemplateModule::refungible_item_id(1, (index + 1) as TokenId);
assert_eq!(item.const_data.to_vec(), data.const_data);
assert_eq!(item.variable_data.to_vec(), data.variable_data);
assert_eq!(
@@ -207,7 +208,7 @@
));
for (index, _) in items_data.iter().enumerate() {
- assert_eq!(TemplateModule::fungible_item_id(1, (index + 1) as u64).owner, 1);
+ assert_eq!(TemplateModule::fungible_item_id(1, (index + 1) as TokenId).owner, 1);
}
assert_eq!(TemplateModule::balance_count(1, 1), 3000);
assert_eq!(TemplateModule::address_tokens(1, 1), [1, 2, 3]);