123456#![recursion_limit = "1024"]78#![cfg_attr(not(feature = "std"), no_std)]910#[cfg(feature = "std")]11pub use std::*;1213#[cfg(feature = "std")]14pub use serde::*;1516use codec::{Decode, Encode};17pub use frame_support::{18 construct_runtime, decl_event, decl_module, decl_storage, decl_error,19 dispatch::DispatchResult,20 ensure, fail, parameter_types,21 traits::{22 Currency, ExistenceRequirement, Get, Imbalance, KeyOwnerProofSystem, OnUnbalanced,23 Randomness, IsSubType,24 },25 weights::{26 constants::{BlockExecutionWeight, ExtrinsicBaseWeight, RocksDbWeight, WEIGHT_PER_SECOND},27 DispatchInfo, GetDispatchInfo, IdentityFee, Pays, PostDispatchInfo, Weight,28 WeightToFeePolynomial, DispatchClass,29 },30 StorageValue,31};3233use frame_system::{self as system, ensure_signed, ensure_root};34use sp_runtime::sp_std::prelude::Vec;35use sp_runtime::{36 traits::{37 DispatchInfoOf, Dispatchable, PostDispatchInfoOf, Saturating, SaturatedConversion, SignedExtension, Zero,38 },39 transaction_validity::{40 TransactionPriority, InvalidTransaction, TransactionValidity, TransactionValidityError, ValidTransaction,41 },42 FixedPointOperand, FixedU128,43};44use sp_runtime::traits::StaticLookup;45use pallet_contracts::chain_extension::UncheckedFrom;46use transaction_payment::OnChargeTransaction;4748#[cfg(test)]49mod mock;5051#[cfg(test)]52mod tests;5354mod default_weights;5556pub const MAX_DECIMAL_POINTS: DecimalPoints = 30;57pub const MAX_REFUNGIBLE_PIECES: u128 = 1_000_000_000_000_000_000_000;58pub const MAX_SPONSOR_TIMEOUT: u32 = 10_368_000;59pub const MAX_TOKEN_OWNERSHIP: u32 = 10_000_000;6061626364pub type CollectionId = u32;65pub type TokenId = u32;66pub type DecimalPoints = u8;6768#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq)]69#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]70pub enum CollectionMode {71 Invalid,72 NFT,73 74 Fungible(DecimalPoints),75 ReFungible,76}7778impl Default for CollectionMode {79 fn default() -> Self {80 Self::Invalid81 }82}8384impl Into<u8> for CollectionMode {85 fn into(self) -> u8 {86 match self {87 CollectionMode::Invalid => 0,88 CollectionMode::NFT => 1,89 CollectionMode::Fungible(_) => 2,90 CollectionMode::ReFungible => 3,91 }92 }93}9495#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq)]96#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]97pub enum AccessMode {98 Normal,99 WhiteList,100}101impl Default for AccessMode {102 fn default() -> Self {103 Self::Normal104 }105}106107#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq)]108#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]109pub enum SchemaVersion {110 ImageURL,111 Unique,112}113impl Default for SchemaVersion {114 fn default() -> Self {115 Self::ImageURL116 }117}118119#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]120#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]121pub struct Ownership<AccountId> {122 pub owner: AccountId,123 pub fraction: u128,124}125126#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]127#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]128pub struct CollectionType<AccountId> {129 pub owner: AccountId,130 pub mode: CollectionMode,131 pub access: AccessMode,132 pub decimal_points: DecimalPoints,133 pub name: Vec<u16>, 134 pub description: Vec<u16>, 135 pub token_prefix: Vec<u8>, 136 pub mint_mode: bool,137 pub offchain_schema: Vec<u8>,138 pub schema_version: SchemaVersion,139 pub sponsor: AccountId, 140 pub sponsor_confirmed: bool, 141 pub limits: CollectionLimits, 142 pub variable_on_chain_schema: Vec<u8>, 143 pub const_on_chain_schema: Vec<u8>, 144}145146#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]147#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]148pub struct NftItemType<AccountId> {149 pub owner: AccountId,150 pub const_data: Vec<u8>,151 pub variable_data: Vec<u8>,152}153154#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]155#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]156pub struct FungibleItemType {157 pub value: u128,158}159160#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]161#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]162pub struct ReFungibleItemType<AccountId> {163 pub owner: Vec<Ownership<AccountId>>,164 pub const_data: Vec<u8>,165 pub variable_data: Vec<u8>,166}167168169170171172173174175176177178179#[derive(Encode, Decode, Debug, Clone, PartialEq)]180#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]181pub struct CollectionLimits {182 pub account_token_ownership_limit: u32,183 pub sponsored_data_size: u32,184 pub token_limit: u32,185186 187 pub sponsor_transfer_timeout: u32,188 pub owner_can_transfer: bool,189 pub owner_can_destroy: bool,190}191192impl Default for CollectionLimits {193 fn default() -> CollectionLimits {194 CollectionLimits { 195 account_token_ownership_limit: 10_000_000, 196 token_limit: u32::max_value(),197 sponsored_data_size: u32::max_value(), 198 sponsor_transfer_timeout: 14400,199 owner_can_transfer: true,200 owner_can_destroy: true201 }202 }203}204205#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]206#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]207pub struct ChainLimits {208 pub collection_numbers_limit: u32,209 pub account_token_ownership_limit: u32,210 pub collections_admins_limit: u64,211 pub custom_data_limit: u32,212213 214 pub nft_sponsor_transfer_timeout: u32,215 pub fungible_sponsor_transfer_timeout: u32,216 pub refungible_sponsor_transfer_timeout: u32,217218 219 pub offchain_schema_limit: u32,220 pub variable_on_chain_schema_limit: u32,221 pub const_on_chain_schema_limit: u32,222}223224pub trait WeightInfo {225 fn create_collection() -> Weight;226 fn destroy_collection() -> Weight;227 fn add_to_white_list() -> Weight;228 fn remove_from_white_list() -> Weight;229 fn set_public_access_mode() -> Weight;230 fn set_mint_permission() -> Weight;231 fn change_collection_owner() -> Weight;232 fn add_collection_admin() -> Weight;233 fn remove_collection_admin() -> Weight;234 fn set_collection_sponsor() -> Weight;235 fn confirm_sponsorship() -> Weight;236 fn remove_collection_sponsor() -> Weight;237 fn create_item(s: usize) -> Weight;238 fn burn_item() -> Weight;239 fn transfer() -> Weight;240 fn approve() -> Weight;241 fn transfer_from() -> Weight;242 fn set_offchain_schema() -> Weight;243 fn set_const_on_chain_schema() -> Weight;244 fn set_variable_on_chain_schema() -> Weight;245 fn set_variable_meta_data() -> Weight;246 fn enable_contract_sponsoring() -> Weight;247 fn set_schema_version() -> Weight;248 fn set_chain_limits() -> Weight;249 fn set_contract_sponsoring_rate_limit() -> Weight;250 fn toggle_contract_white_list() -> Weight;251 fn add_to_contract_white_list() -> Weight;252 fn remove_from_contract_white_list() -> Weight;253 fn set_collection_limits() -> Weight;254}255256#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]257#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]258pub struct CreateNftData {259 pub const_data: Vec<u8>,260 pub variable_data: Vec<u8>,261}262263#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]264#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]265pub struct CreateFungibleData {266 pub value: u128,267}268269#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]270#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]271pub struct CreateReFungibleData {272 pub const_data: Vec<u8>,273 pub variable_data: Vec<u8>,274 pub pieces: u128,275}276277#[derive(Encode, Decode, Debug, Clone, PartialEq)]278#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]279pub enum CreateItemData {280 NFT(CreateNftData),281 Fungible(CreateFungibleData),282 ReFungible(CreateReFungibleData),283}284285impl CreateItemData {286 pub fn len(&self) -> usize {287 let len = match self {288 CreateItemData::NFT(data) => data.variable_data.len() + data.const_data.len(),289 CreateItemData::ReFungible(data) => data.variable_data.len() + data.const_data.len(),290 _ => 0291 };292 293 return len;294 }295}296297impl From<CreateNftData> for CreateItemData {298 fn from(item: CreateNftData) -> Self {299 CreateItemData::NFT(item)300 }301}302303impl From<CreateReFungibleData> for CreateItemData {304 fn from(item: CreateReFungibleData) -> Self {305 CreateItemData::ReFungible(item)306 }307}308309impl From<CreateFungibleData> for CreateItemData {310 fn from(item: CreateFungibleData) -> Self {311 CreateItemData::Fungible(item)312 }313}314315316decl_error! {317 318 pub enum Error for Module<T: Config> {319 320 TotalCollectionsLimitExceeded,321 322 CollectionDecimalPointLimitExceeded, 323 324 CollectionNameLimitExceeded, 325 326 CollectionDescriptionLimitExceeded, 327 328 CollectionTokenPrefixLimitExceeded,329 330 CollectionNotFound,331 332 TokenNotFound,333 334 AdminNotFound,335 336 NumOverflow, 337 338 AlreadyAdmin, 339 340 NoPermission,341 342 ConfirmUnsetSponsorFail,343 344 PublicMintingNotAllowed,345 346 MustBeTokenOwner,347 348 TokenValueTooLow,349 350 NftSizeLimitExceeded,351 352 ApproveNotFound,353 354 TokenValueNotEnough,355 356 ApproveRequired,357 358 AddresNotInWhiteList,359 360 CollectionAdminsLimitExceeded,361 362 AddressOwnershipLimitExceeded,363 364 EmptyArgument,365 366 TokenConstDataLimitExceeded,367 368 TokenVariableDataLimitExceeded,369 370 NotNftDataUsedToMintNftCollectionToken,371 372 NotFungibleDataUsedToMintFungibleCollectionToken,373 374 NotReFungibleDataUsedToMintReFungibleCollectionToken,375 376 UnexpectedCollectionType,377 378 CantStoreMetadataInFungibleTokens,379 380 CollectionTokenLimitExceeded,381 382 AccountTokenLimitExceeded,383 384 CollectionLimitBoundsExceeded,385 386 OwnerPermissionsCantBeReverted,387 388 SchemaDataLimitExceeded,389 390 WrongRefungiblePieces391 }392}393394pub trait Config: system::Config + Sized + transaction_payment::Config + pallet_contracts::Config {395 type Event: From<Event<Self>> + Into<<Self as system::Config>::Event>;396397 398 type WeightInfo: WeightInfo;399}400401#[cfg(feature = "runtime-benchmarks")]402mod benchmarking;403404405406decl_storage! {407 trait Store for Module<T: Config> as Nft {408409 410 NextCollectionID: CollectionId;411 CreatedCollectionCount: u32;412 ChainVersion: u64;413 ItemListIndex: map hasher(identity) CollectionId => TokenId;414415 416 pub ChainLimit get(fn chain_limit) config(): ChainLimits;417418 419 CollectionCount: u32;420 pub AccountItemCount get(fn account_item_count): map hasher(twox_64_concat) T::AccountId => u32;421422 423 pub Collection get(fn collection) config(): map hasher(identity) CollectionId => CollectionType<T::AccountId>;424 pub AdminList get(fn admin_list_collection): map hasher(identity) CollectionId => Vec<T::AccountId>;425 pub WhiteList get(fn white_list): double_map hasher(identity) CollectionId, hasher(twox_64_concat) T::AccountId => bool;426427 428 pub Balance get(fn balance_count): double_map hasher(identity) CollectionId, hasher(twox_64_concat) T::AccountId => u128;429430 431 pub Allowances get(fn approved): double_map hasher(identity) CollectionId, hasher(twox_64_concat) (TokenId, T::AccountId, T::AccountId) => u128;432433 434 pub NftItemList get(fn nft_item_id) config(): double_map hasher(identity) CollectionId, hasher(identity) TokenId => NftItemType<T::AccountId>;435 pub FungibleItemList get(fn fungible_item_id) config(): double_map hasher(identity) CollectionId, hasher(twox_64_concat) T::AccountId => FungibleItemType;436 pub ReFungibleItemList get(fn refungible_item_id) config(): double_map hasher(identity) CollectionId, hasher(identity) TokenId => ReFungibleItemType<T::AccountId>;437438 439 pub AddressTokens get(fn address_tokens): double_map hasher(identity) CollectionId, hasher(twox_64_concat) T::AccountId => Vec<TokenId>;440441 442 pub CreateItemBasket get(fn create_item_basket): map hasher(twox_64_concat) (CollectionId, T::AccountId) => T::BlockNumber;443 pub NftTransferBasket get(fn nft_transfer_basket): double_map hasher(identity) CollectionId, hasher(identity) TokenId => T::BlockNumber;444 pub FungibleTransferBasket get(fn fungible_transfer_basket): double_map hasher(identity) CollectionId, hasher(twox_64_concat) T::AccountId => T::BlockNumber;445 pub ReFungibleTransferBasket get(fn refungible_transfer_basket): double_map hasher(identity) CollectionId, hasher(identity) TokenId => T::BlockNumber;446447 448 pub ContractOwner get(fn contract_owner): map hasher(twox_64_concat) T::AccountId => T::AccountId;449 pub ContractSelfSponsoring get(fn contract_self_sponsoring): map hasher(twox_64_concat) T::AccountId => bool;450 pub ContractSponsorBasket get(fn contract_sponsor_basket): map hasher(twox_64_concat) (T::AccountId, T::AccountId) => T::BlockNumber;451 pub ContractSponsoringRateLimit get(fn contract_sponsoring_rate_limit): map hasher(twox_64_concat) T::AccountId => T::BlockNumber;452 pub ContractWhiteListEnabled get(fn contract_white_list_enabled): map hasher(twox_64_concat) T::AccountId => bool; 453 pub ContractWhiteList get(fn contract_white_list): double_map hasher(twox_64_concat) T::AccountId, hasher(twox_64_concat) T::AccountId => bool; 454 }455 add_extra_genesis {456 build(|config: &GenesisConfig<T>| {457 458 for (_num, _c) in &config.collection {459 <Module<T>>::init_collection(_c);460 }461462 for (_num, _c, _i) in &config.nft_item_id {463 <Module<T>>::init_nft_token(*_c, _i);464 }465466 for (collection_id, account_id, fungible_item) in &config.fungible_item_id {467 <Module<T>>::init_fungible_token(*collection_id, account_id, fungible_item);468 }469470 for (_num, _c, _i) in &config.refungible_item_id {471 <Module<T>>::init_refungible_token(*_c, _i);472 }473 })474 }475}476477decl_event!(478 pub enum Event<T>479 where480 AccountId = <T as system::Config>::AccountId,481 {482 483 484 485 486 487 488 489 490 491 Created(CollectionId, u8, AccountId),492493 494 495 496 497 498 499 500 ItemCreated(CollectionId, TokenId),501502 503 504 505 506 507 508 509 ItemDestroyed(CollectionId, TokenId),510 }511);512513decl_module! {514 pub struct Module<T: Config> for enum Call 515 where 516 origin: T::Origin517 {518 fn deposit_event() = default;519 type Error = Error<T>;520521 fn on_initialize(now: T::BlockNumber) -> Weight {522523 if ChainVersion::get() < 2524 {525 let value = NextCollectionID::get();526 CreatedCollectionCount::put(value);527 ChainVersion::put(2);528 }529530 0531 }532533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 #[weight = <T as Config>::WeightInfo::create_collection()]550 pub fn create_collection(origin,551 collection_name: Vec<u16>,552 collection_description: Vec<u16>,553 token_prefix: Vec<u8>,554 mode: CollectionMode) -> DispatchResult {555556 557 let who = ensure_signed(origin)?;558559 let decimal_points = match mode {560 CollectionMode::Fungible(points) => points,561 _ => 0562 };563564 565 ensure!(CollectionCount::get() < ChainLimit::get().collection_numbers_limit, Error::<T>::TotalCollectionsLimitExceeded);566567 568 ensure!(decimal_points <= MAX_DECIMAL_POINTS, Error::<T>::CollectionDecimalPointLimitExceeded);569 ensure!(collection_name.len() <= 64, Error::<T>::CollectionNameLimitExceeded);570 ensure!(collection_description.len() <= 256, Error::<T>::CollectionDescriptionLimitExceeded);571 ensure!(token_prefix.len() <= 16, Error::<T>::CollectionTokenPrefixLimitExceeded);572573 574 let next_id = CreatedCollectionCount::get()575 .checked_add(1)576 .ok_or(Error::<T>::NumOverflow)?;577578 579 let total = CollectionCount::get()580 .checked_add(1)581 .ok_or(Error::<T>::NumOverflow)?;582583 CreatedCollectionCount::put(next_id);584 CollectionCount::put(total);585586 587 let new_collection = CollectionType {588 owner: who.clone(),589 name: collection_name,590 mode: mode.clone(),591 mint_mode: false,592 access: AccessMode::Normal,593 description: collection_description,594 decimal_points: decimal_points,595 token_prefix: token_prefix,596 offchain_schema: Vec::new(),597 schema_version: SchemaVersion::ImageURL,598 sponsor: T::AccountId::default(),599 sponsor_confirmed: false,600 variable_on_chain_schema: Vec::new(),601 const_on_chain_schema: Vec::new(),602 limits: CollectionLimits::default()603 };604605 606 <Collection<T>>::insert(next_id, new_collection);607608 609 Self::deposit_event(RawEvent::Created(next_id, mode.into(), who.clone()));610611 Ok(())612 }613614 615 616 617 618 619 620 621 622 623 #[weight = <T as Config>::WeightInfo::destroy_collection()]624 pub fn destroy_collection(origin, collection_id: CollectionId) -> DispatchResult {625626 let sender = ensure_signed(origin)?;627 Self::check_owner_permissions(collection_id, sender)?;628629 let target_collection = <Collection<T>>::get(collection_id);630 if !target_collection.limits.owner_can_destroy {631 fail!(Error::<T>::NoPermission);632 }633634 <AddressTokens<T>>::remove_prefix(collection_id);635 <Allowances<T>>::remove_prefix(collection_id);636 <Balance<T>>::remove_prefix(collection_id);637 <ItemListIndex>::remove(collection_id);638 <AdminList<T>>::remove(collection_id);639 <Collection<T>>::remove(collection_id);640 <WhiteList<T>>::remove_prefix(collection_id);641642 <NftItemList<T>>::remove_prefix(collection_id);643 <FungibleItemList<T>>::remove_prefix(collection_id);644 <ReFungibleItemList<T>>::remove_prefix(collection_id);645646 <NftTransferBasket<T>>::remove_prefix(collection_id);647 <FungibleTransferBasket<T>>::remove_prefix(collection_id);648 <ReFungibleTransferBasket<T>>::remove_prefix(collection_id);649650 if CollectionCount::get() > 0651 {652 653 let total = CollectionCount::get()654 .checked_sub(1)655 .ok_or(Error::<T>::NumOverflow)?;656657 CollectionCount::put(total);658 }659660 Ok(())661 }662663 664 665 666 667 668 669 670 671 672 673 674 675 #[weight = <T as Config>::WeightInfo::add_to_white_list()]676 pub fn add_to_white_list(origin, collection_id: CollectionId, address: T::AccountId) -> DispatchResult{677678 let sender = ensure_signed(origin)?;679 Self::check_owner_or_admin_permissions(collection_id, sender)?;680681 <WhiteList<T>>::insert(collection_id, address, true);682 683 Ok(())684 }685686 687 688 689 690 691 692 693 694 695 696 697 698 #[weight = <T as Config>::WeightInfo::remove_from_white_list()]699 pub fn remove_from_white_list(origin, collection_id: CollectionId, address: T::AccountId) -> DispatchResult{700701 let sender = ensure_signed(origin)?;702 Self::check_owner_or_admin_permissions(collection_id, sender)?;703704 <WhiteList<T>>::remove(collection_id, address);705706 Ok(())707 }708709 710 711 712 713 714 715 716 717 718 719 720 #[weight = <T as Config>::WeightInfo::set_public_access_mode()]721 pub fn set_public_access_mode(origin, collection_id: CollectionId, mode: AccessMode) -> DispatchResult722 {723 let sender = ensure_signed(origin)?;724725 Self::check_owner_permissions(collection_id, sender)?;726 let mut target_collection = <Collection<T>>::get(collection_id);727 target_collection.access = mode;728 <Collection<T>>::insert(collection_id, target_collection);729730 Ok(())731 }732733 734 735 736 737 738 739 740 741 742 743 744 745 746 #[weight = <T as Config>::WeightInfo::set_mint_permission()]747 pub fn set_mint_permission(origin, collection_id: CollectionId, mint_permission: bool) -> DispatchResult748 {749 let sender = ensure_signed(origin)?;750751 Self::check_owner_permissions(collection_id, sender)?;752 let mut target_collection = <Collection<T>>::get(collection_id);753 target_collection.mint_mode = mint_permission;754 <Collection<T>>::insert(collection_id, target_collection);755756 Ok(())757 }758759 760 761 762 763 764 765 766 767 768 769 770 #[weight = <T as Config>::WeightInfo::change_collection_owner()]771 pub fn change_collection_owner(origin, collection_id: CollectionId, new_owner: T::AccountId) -> DispatchResult {772773 let sender = ensure_signed(origin)?;774 Self::check_owner_permissions(collection_id, sender)?;775 let mut target_collection = <Collection<T>>::get(collection_id);776 target_collection.owner = new_owner;777 <Collection<T>>::insert(collection_id, target_collection);778779 Ok(())780 }781782 783 784 785 786 787 788 789 790 791 792 793 794 795 #[weight = <T as Config>::WeightInfo::add_collection_admin()]796 pub fn add_collection_admin(origin, collection_id: CollectionId, new_admin_id: T::AccountId) -> DispatchResult {797798 let sender = ensure_signed(origin)?;799 Self::check_owner_or_admin_permissions(collection_id, sender)?;800 let mut admin_arr: Vec<T::AccountId> = Vec::new();801802 if <AdminList<T>>::contains_key(collection_id)803 {804 admin_arr = <AdminList<T>>::get(collection_id);805 ensure!(!admin_arr.contains(&new_admin_id), Error::<T>::AlreadyAdmin);806 }807808 809 ensure!((admin_arr.len() as u64) < ChainLimit::get().collections_admins_limit, Error::<T>::CollectionAdminsLimitExceeded);810811 admin_arr.push(new_admin_id);812 <AdminList<T>>::insert(collection_id, admin_arr);813814 Ok(())815 }816817 818 819 820 821 822 823 824 825 826 827 828 829 #[weight = <T as Config>::WeightInfo::remove_collection_admin()]830 pub fn remove_collection_admin(origin, collection_id: CollectionId, account_id: T::AccountId) -> DispatchResult {831832 let sender = ensure_signed(origin)?;833 Self::check_owner_or_admin_permissions(collection_id, sender)?;834 ensure!(<AdminList<T>>::contains_key(collection_id), Error::<T>::AdminNotFound);835836 let mut admin_arr = <AdminList<T>>::get(collection_id);837 admin_arr.retain(|i| *i != account_id);838 <AdminList<T>>::insert(collection_id, admin_arr);839840 Ok(())841 }842843 844 845 846 847 848 849 850 851 852 #[weight = <T as Config>::WeightInfo::set_collection_sponsor()]853 pub fn set_collection_sponsor(origin, collection_id: CollectionId, new_sponsor: T::AccountId) -> 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 = new_sponsor;862 target_collection.sponsor_confirmed = false;863 <Collection<T>>::insert(collection_id, target_collection);864865 Ok(())866 }867868 869 870 871 872 873 874 875 #[weight = <T as Config>::WeightInfo::confirm_sponsorship()]876 pub fn confirm_sponsorship(origin, collection_id: CollectionId) -> DispatchResult {877878 let sender = ensure_signed(origin)?;879 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);880881 let mut target_collection = <Collection<T>>::get(collection_id);882 ensure!(sender == target_collection.sponsor, Error::<T>::ConfirmUnsetSponsorFail);883884 target_collection.sponsor_confirmed = true;885 <Collection<T>>::insert(collection_id, target_collection);886887 Ok(())888 }889890 891 892 893 894 895 896 897 898 899 #[weight = <T as Config>::WeightInfo::remove_collection_sponsor()]900 pub fn remove_collection_sponsor(origin, collection_id: CollectionId) -> DispatchResult {901902 let sender = ensure_signed(origin)?;903 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);904905 let mut target_collection = <Collection<T>>::get(collection_id);906 ensure!(sender == target_collection.owner, Error::<T>::NoPermission);907908 target_collection.sponsor = T::AccountId::default();909 target_collection.sponsor_confirmed = false;910 <Collection<T>>::insert(collection_id, target_collection);911912 Ok(())913 }914915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938939 #[weight = <T as Config>::WeightInfo::create_item(data.len())]940 pub fn create_item(origin, collection_id: CollectionId, owner: T::AccountId, data: CreateItemData) -> DispatchResult {941942 let sender = ensure_signed(origin)?;943944 Self::collection_exists(collection_id)?;945946 let target_collection = <Collection<T>>::get(collection_id);947948 Self::can_create_items_in_collection(collection_id, &target_collection, &sender, &owner)?;949 Self::validate_create_item_args(&target_collection, &data)?;950 Self::create_item_no_validation(collection_id, owner, data)?;951952 Ok(())953 }954955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 #[weight = <T as Config>::WeightInfo::create_item(items_data.into_iter()974 .map(|data| { data.len() })975 .sum())]976 pub fn create_multiple_items(origin, collection_id: CollectionId, owner: T::AccountId, items_data: Vec<CreateItemData>) -> DispatchResult {977978 ensure!(items_data.len() > 0, Error::<T>::EmptyArgument);979 let sender = ensure_signed(origin)?;980981 Self::collection_exists(collection_id)?;982 let target_collection = <Collection<T>>::get(collection_id);983984 Self::can_create_items_in_collection(collection_id, &target_collection, &sender, &owner)?;985986 for data in &items_data {987 Self::validate_create_item_args(&target_collection, data)?;988 }989 for data in &items_data {990 Self::create_item_no_validation(collection_id, owner.clone(), data.clone())?;991 }992993 Ok(())994 }995996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 #[weight = <T as Config>::WeightInfo::burn_item()]1010 pub fn burn_item(origin, collection_id: CollectionId, item_id: TokenId, value: u128) -> DispatchResult {10111012 let sender = ensure_signed(origin)?;1013 Self::collection_exists(collection_id)?;10141015 1016 let target_collection = <Collection<T>>::get(collection_id);1017 ensure!(1018 Self::is_item_owner(sender.clone(), collection_id, item_id) ||1019 (1020 target_collection.limits.owner_can_transfer &&1021 Self::is_owner_or_admin_permissions(collection_id, sender.clone())1022 ),1023 Error::<T>::NoPermission1024 );10251026 if target_collection.access == AccessMode::WhiteList {1027 Self::check_white_list(collection_id, &sender)?;1028 }10291030 match target_collection.mode1031 {1032 CollectionMode::NFT => Self::burn_nft_item(collection_id, item_id)?,1033 CollectionMode::Fungible(_) => Self::burn_fungible_item(&sender, collection_id, value)?,1034 CollectionMode::ReFungible => Self::burn_refungible_item(collection_id, item_id, &sender)?,1035 _ => ()1036 };10371038 1039 Self::deposit_event(RawEvent::ItemDestroyed(collection_id, item_id));10401041 Ok(())1042 }10431044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 #[weight = <T as Config>::WeightInfo::transfer()]1068 pub fn transfer(origin, recipient: T::AccountId, collection_id: CollectionId, item_id: TokenId, value: u128) -> DispatchResult {1069 let sender = ensure_signed(origin)?;1070 Self::transfer_internal(sender, recipient, collection_id, item_id, value)1071 }10721073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 #[weight = <T as Config>::WeightInfo::approve()]1089 pub fn approve(origin, spender: T::AccountId, collection_id: CollectionId, item_id: TokenId, amount: u128) -> DispatchResult {10901091 let sender = ensure_signed(origin)?;10921093 Self::collection_exists(collection_id)?;1094 Self::token_exists(collection_id, item_id, &sender)?;10951096 1097 let target_collection = <Collection<T>>::get(collection_id);1098 ensure!(1099 Self::is_item_owner(sender.clone(), collection_id, item_id) ||1100 (1101 target_collection.limits.owner_can_transfer &&1102 Self::is_owner_or_admin_permissions(collection_id, sender.clone())1103 ),1104 Error::<T>::NoPermission1105 );11061107 if target_collection.access == AccessMode::WhiteList {1108 Self::check_white_list(collection_id, &sender)?;1109 Self::check_white_list(collection_id, &spender)?;1110 }11111112 let allowance_exists = <Allowances<T>>::contains_key(collection_id, (item_id, &sender, &spender));1113 let mut allowance: u128 = amount;1114 if allowance_exists {1115 allowance += <Allowances<T>>::get(collection_id, (item_id, &sender, &spender));1116 }1117 <Allowances<T>>::insert(collection_id, (item_id, sender.clone(), spender.clone()), allowance);11181119 Ok(())1120 }1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 #[weight = <T as Config>::WeightInfo::transfer_from()]1142 pub fn transfer_from(origin, from: T::AccountId, recipient: T::AccountId, collection_id: CollectionId, item_id: TokenId, value: u128 ) -> DispatchResult {11431144 let sender = ensure_signed(origin)?;1145 let mut appoved_transfer = false;11461147 1148 let mut approval: u128 = 0;1149 if <Allowances<T>>::contains_key(collection_id, (item_id, &from, &sender)) {1150 approval = <Allowances<T>>::get(collection_id, (item_id, &from, &sender));1151 ensure!(approval >= value, Error::<T>::TokenValueNotEnough);1152 appoved_transfer = true;1153 }11541155 let target_collection = <Collection<T>>::get(collection_id);11561157 1158 Self::is_correct_transfer(collection_id, &target_collection, &recipient)?;11591160 1161 ensure!(1162 appoved_transfer || 1163 (1164 target_collection.limits.owner_can_transfer &&1165 Self::is_owner_or_admin_permissions(collection_id, sender.clone())1166 ),1167 Error::<T>::NoPermission1168 );11691170 if target_collection.access == AccessMode::WhiteList {1171 Self::check_white_list(collection_id, &sender)?;1172 Self::check_white_list(collection_id, &recipient)?;1173 }11741175 1176 if approval.checked_sub(value).unwrap_or(0) > 0 {1177 <Allowances<T>>::insert(collection_id, (item_id, &from, &sender), approval - value);1178 }1179 else {1180 <Allowances<T>>::remove(collection_id, (item_id, &from, &sender));1181 }11821183 match target_collection.mode1184 {1185 CollectionMode::NFT => Self::transfer_nft(collection_id, item_id, from, recipient)?,1186 CollectionMode::Fungible(_) => Self::transfer_fungible(collection_id, value, &from, &recipient)?,1187 CollectionMode::ReFungible => Self::transfer_refungible(collection_id, item_id, value, from.clone(), recipient)?,1188 _ => ()1189 };11901191 Ok(())1192 }11931194 1195 11961197 1198 1199 1200 12011202 12031204 12051206 1207 12081209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 #[weight = <T as Config>::WeightInfo::set_variable_meta_data()]1222 pub fn set_variable_meta_data (1223 origin,1224 collection_id: CollectionId,1225 item_id: TokenId,1226 data: Vec<u8>1227 ) -> DispatchResult {1228 let sender = ensure_signed(origin)?;1229 1230 Self::collection_exists(collection_id)?;1231 Self::token_exists(collection_id, item_id, &sender)?;12321233 ensure!(ChainLimit::get().custom_data_limit >= data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);12341235 1236 let target_collection = <Collection<T>>::get(collection_id);1237 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1238 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1239 Error::<T>::NoPermission);12401241 match target_collection.mode1242 {1243 CollectionMode::NFT => Self::set_nft_variable_data(collection_id, item_id, data)?,1244 CollectionMode::ReFungible => Self::set_re_fungible_variable_data(collection_id, item_id, data)?,1245 CollectionMode::Fungible(_) => fail!(Error::<T>::CantStoreMetadataInFungibleTokens),1246 _ => fail!(Error::<T>::UnexpectedCollectionType)1247 };12481249 Ok(())1250 }1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 #[weight = <T as Config>::WeightInfo::set_schema_version()]1267 pub fn set_schema_version(1268 origin,1269 collection_id: CollectionId,1270 version: SchemaVersion1271 ) -> DispatchResult {1272 let sender = ensure_signed(origin)?;1273 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;1274 let mut target_collection = <Collection<T>>::get(collection_id);1275 target_collection.schema_version = version;1276 <Collection<T>>::insert(collection_id, target_collection);12771278 Ok(())1279 }12801281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 #[weight = <T as Config>::WeightInfo::set_offchain_schema()]1294 pub fn set_offchain_schema(1295 origin,1296 collection_id: CollectionId,1297 schema: Vec<u8>1298 ) -> DispatchResult {1299 let sender = ensure_signed(origin)?;1300 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;13011302 1303 ensure!(schema.len() as u32 <= ChainLimit::get().offchain_schema_limit, "");13041305 let mut target_collection = <Collection<T>>::get(collection_id);1306 target_collection.offchain_schema = schema;1307 <Collection<T>>::insert(collection_id, target_collection);13081309 Ok(())1310 }13111312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 #[weight = <T as Config>::WeightInfo::set_const_on_chain_schema()]1325 pub fn set_const_on_chain_schema (1326 origin,1327 collection_id: CollectionId,1328 schema: Vec<u8>1329 ) -> DispatchResult {1330 let sender = ensure_signed(origin)?;1331 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;13321333 1334 ensure!(schema.len() as u32 <= ChainLimit::get().const_on_chain_schema_limit, "");13351336 let mut target_collection = <Collection<T>>::get(collection_id);1337 target_collection.const_on_chain_schema = schema;1338 <Collection<T>>::insert(collection_id, target_collection);13391340 Ok(())1341 }13421343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 #[weight = <T as Config>::WeightInfo::set_const_on_chain_schema()]1356 pub fn set_variable_on_chain_schema (1357 origin,1358 collection_id: CollectionId,1359 schema: Vec<u8>1360 ) -> DispatchResult {1361 let sender = ensure_signed(origin)?;1362 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;13631364 1365 ensure!(schema.len() as u32 <= ChainLimit::get().variable_on_chain_schema_limit, "");13661367 let mut target_collection = <Collection<T>>::get(collection_id);1368 target_collection.variable_on_chain_schema = schema;1369 <Collection<T>>::insert(collection_id, target_collection);13701371 Ok(())1372 }13731374 1375 #[weight = <T as Config>::WeightInfo::set_chain_limits()]1376 pub fn set_chain_limits(1377 origin,1378 limits: ChainLimits1379 ) -> DispatchResult {13801381 #[cfg(not(feature = "runtime-benchmarks"))]1382 ensure_root(origin)?;13831384 <ChainLimit>::put(limits);1385 Ok(())1386 }13871388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 #[weight = <T as Config>::WeightInfo::enable_contract_sponsoring()]1400 pub fn enable_contract_sponsoring(1401 origin,1402 contract_address: T::AccountId,1403 enable: bool1404 ) -> DispatchResult {14051406 let sender = ensure_signed(origin)?;14071408 #[cfg(feature = "runtime-benchmarks")]1409 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());14101411 Self::ensure_contract_owned(sender, &contract_address)?;14121413 <ContractSelfSponsoring<T>>::insert(contract_address, enable);1414 Ok(())1415 }14161417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 #[weight = <T as Config>::WeightInfo::set_contract_sponsoring_rate_limit()]1435 pub fn set_contract_sponsoring_rate_limit(1436 origin,1437 contract_address: T::AccountId,1438 rate_limit: T::BlockNumber1439 ) -> DispatchResult {1440 let sender = ensure_signed(origin)?;14411442 #[cfg(feature = "runtime-benchmarks")]1443 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());14441445 Self::ensure_contract_owned(sender, &contract_address)?;1446 <ContractSponsoringRateLimit<T>>::insert(contract_address, rate_limit);1447 Ok(())1448 }14491450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 #[weight = <T as Config>::WeightInfo::toggle_contract_white_list()]1462 pub fn toggle_contract_white_list(1463 origin,1464 contract_address: T::AccountId,1465 enable: bool1466 ) -> DispatchResult {1467 let sender = ensure_signed(origin)?;14681469 #[cfg(feature = "runtime-benchmarks")]1470 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());14711472 Self::ensure_contract_owned(sender, &contract_address)?;1473 <ContractWhiteListEnabled<T>>::insert(contract_address, enable);1474 Ok(())1475 }1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 #[weight = <T as Config>::WeightInfo::add_to_contract_white_list()]1489 pub fn add_to_contract_white_list(1490 origin,1491 contract_address: T::AccountId,1492 account_address: T::AccountId1493 ) -> DispatchResult {1494 let sender = ensure_signed(origin)?;14951496 #[cfg(feature = "runtime-benchmarks")]1497 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());1498 1499 Self::ensure_contract_owned(sender, &contract_address)?; 1500 <ContractWhiteList<T>>::insert(contract_address, account_address, true);1501 Ok(())1502 }15031504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 #[weight = <T as Config>::WeightInfo::remove_from_contract_white_list()]1516 pub fn remove_from_contract_white_list(1517 origin,1518 contract_address: T::AccountId,1519 account_address: T::AccountId1520 ) -> DispatchResult {1521 let sender = ensure_signed(origin)?;15221523 #[cfg(feature = "runtime-benchmarks")]1524 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());15251526 Self::ensure_contract_owned(sender, &contract_address)?;1527 <ContractWhiteList<T>>::remove(contract_address, account_address);1528 Ok(())1529 }15301531 #[weight = <T as Config>::WeightInfo::set_collection_limits()]1532 pub fn set_collection_limits(1533 origin,1534 collection_id: u32,1535 new_limits: CollectionLimits,1536 ) -> DispatchResult {1537 let sender = ensure_signed(origin)?;1538 Self::check_owner_permissions(collection_id, sender.clone())?;1539 let mut target_collection = <Collection<T>>::get(collection_id);1540 let old_limits = target_collection.limits;1541 let chain_limits = ChainLimit::get();15421543 1544 ensure!(new_limits.sponsor_transfer_timeout <= MAX_SPONSOR_TIMEOUT &&1545 new_limits.account_token_ownership_limit <= MAX_TOKEN_OWNERSHIP && 1546 new_limits.sponsored_data_size <= chain_limits.custom_data_limit &&1547 new_limits.sponsored_mint_size <= chain_limits.custom_data_limit,1548 Error::<T>::CollectionLimitBoundsExceeded);15491550 1551 ensure!(old_limits.token_limit >= new_limits.token_limit, Error::<T>::CollectionTokenLimitExceeded);1552 ensure!(new_limits.token_limit > 0, Error::<T>::CollectionTokenLimitExceeded);15531554 ensure!(1555 (old_limits.owner_can_transfer || !new_limits.owner_can_transfer) &&1556 (old_limits.owner_can_destroy || !new_limits.owner_can_destroy),1557 Error::<T>::OwnerPermissionsCantBeReverted,1558 );15591560 target_collection.limits = new_limits;1561 <Collection<T>>::insert(collection_id, target_collection);15621563 Ok(())1564 } 1565 }1566}15671568impl<T: Config> Module<T> {15691570 pub fn transfer_internal(sender: T::AccountId, recipient: T::AccountId, collection_id: CollectionId, item_id: TokenId, value: u128) -> DispatchResult {15711572 let target_collection = <Collection<T>>::get(collection_id);15731574 1575 Self::is_correct_transfer(collection_id, &target_collection, &recipient)?;15761577 1578 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1579 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1580 Error::<T>::NoPermission);15811582 if target_collection.access == AccessMode::WhiteList {1583 Self::check_white_list(collection_id, &sender)?;1584 Self::check_white_list(collection_id, &recipient)?;1585 }15861587 match target_collection.mode1588 {1589 CollectionMode::NFT => Self::transfer_nft(collection_id, item_id, sender.clone(), recipient)?,1590 CollectionMode::Fungible(_) => Self::transfer_fungible(collection_id, value, &sender, &recipient)?,1591 CollectionMode::ReFungible => Self::transfer_refungible(collection_id, item_id, value, sender.clone(), recipient)?,1592 _ => ()1593 };15941595 Ok(())1596 }159715981599 fn is_correct_transfer(collection_id: CollectionId, collection: &CollectionType<T::AccountId>, recipient: &T::AccountId) -> DispatchResult {16001601 1602 let account_items: u32 = <AddressTokens<T>>::get(collection_id, recipient).len() as u32;1603 ensure!(collection.limits.account_token_ownership_limit > account_items, Error::<T>::AccountTokenLimitExceeded);1604 1605 Ok(())1606 }16071608 fn can_create_items_in_collection(collection_id: CollectionId, collection: &CollectionType<T::AccountId>, sender: &T::AccountId, owner: &T::AccountId) -> DispatchResult {16091610 1611 let total_items: u32 = ItemListIndex::get(collection_id);1612 let account_items: u32 = <AddressTokens<T>>::get(collection_id, owner).len() as u32;1613 ensure!(collection.limits.token_limit > total_items, Error::<T>::CollectionTokenLimitExceeded);1614 ensure!(collection.limits.account_token_ownership_limit > account_items, Error::<T>::AccountTokenLimitExceeded);16151616 if !Self::is_owner_or_admin_permissions(collection_id, sender.clone()) {1617 ensure!(collection.mint_mode == true, Error::<T>::PublicMintingNotAllowed);1618 Self::check_white_list(collection_id, owner)?;1619 Self::check_white_list(collection_id, sender)?;1620 }16211622 Ok(())1623 }16241625 fn validate_create_item_args(target_collection: &CollectionType<T::AccountId>, data: &CreateItemData) -> DispatchResult {1626 match target_collection.mode1627 {1628 CollectionMode::NFT => {1629 if let CreateItemData::NFT(data) = data {1630 1631 ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, Error::<T>::TokenConstDataLimitExceeded);1632 ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);1633 } else {1634 fail!(Error::<T>::NotNftDataUsedToMintNftCollectionToken);1635 }1636 },1637 CollectionMode::Fungible(_) => {1638 if let CreateItemData::Fungible(_) = data {1639 } else {1640 fail!(Error::<T>::NotFungibleDataUsedToMintFungibleCollectionToken);1641 }1642 },1643 CollectionMode::ReFungible => {1644 if let CreateItemData::ReFungible(data) = data {16451646 1647 ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, Error::<T>::TokenConstDataLimitExceeded);1648 ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);16491650 1651 ensure!(data.pieces <= MAX_REFUNGIBLE_PIECES, Error::<T>::WrongRefungiblePieces);1652 ensure!(data.pieces > 0, Error::<T>::WrongRefungiblePieces);1653 } else {1654 fail!(Error::<T>::NotReFungibleDataUsedToMintReFungibleCollectionToken);1655 }1656 },1657 _ => { fail!(Error::<T>::UnexpectedCollectionType); }1658 };16591660 Ok(())1661 }16621663 fn create_item_no_validation(collection_id: CollectionId, owner: T::AccountId, data: CreateItemData) -> DispatchResult {1664 match data1665 {1666 CreateItemData::NFT(data) => {1667 let item = NftItemType {1668 owner,1669 const_data: data.const_data,1670 variable_data: data.variable_data1671 };16721673 Self::add_nft_item(collection_id, item)?;1674 },1675 CreateItemData::Fungible(data) => {1676 Self::add_fungible_item(collection_id, &owner, data.value)?;1677 },1678 CreateItemData::ReFungible(data) => {1679 let mut owner_list = Vec::new();1680 owner_list.push(Ownership {owner: owner.clone(), fraction: data.pieces});16811682 let item = ReFungibleItemType {1683 owner: owner_list,1684 const_data: data.const_data,1685 variable_data: data.variable_data1686 };16871688 Self::add_refungible_item(collection_id, item)?;1689 }1690 };16911692 1693 Self::deposit_event(RawEvent::ItemCreated(collection_id, <ItemListIndex>::get(collection_id)));16941695 Ok(())1696 }16971698 fn add_fungible_item(collection_id: CollectionId, owner: &T::AccountId, value: u128) -> DispatchResult {16991700 1701 let mut balance: u128 = 0;1702 if <FungibleItemList<T>>::contains_key(collection_id, owner) {1703 balance = <FungibleItemList<T>>::get(collection_id, owner).value;1704 } 17051706 1707 let item = FungibleItemType {1708 value: balance + value1709 };1710 <FungibleItemList<T>>::insert(collection_id, (*owner).clone(), item);17111712 1713 let new_balance = <Balance<T>>::get(collection_id, owner)1714 .checked_add(value)1715 .ok_or(Error::<T>::NumOverflow)?;1716 <Balance<T>>::insert(collection_id, (*owner).clone(), new_balance);17171718 Ok(())1719 }17201721 fn add_refungible_item(collection_id: CollectionId, item: ReFungibleItemType<T::AccountId>) -> DispatchResult {1722 let current_index = <ItemListIndex>::get(collection_id)1723 .checked_add(1)1724 .ok_or(Error::<T>::NumOverflow)?;1725 let itemcopy = item.clone();17261727 let value = item.owner.first().unwrap().fraction;1728 let owner = item.owner.first().unwrap().owner.clone();17291730 Self::add_token_index(collection_id, current_index, &owner)?;17311732 <ItemListIndex>::insert(collection_id, current_index);1733 <ReFungibleItemList<T>>::insert(collection_id, current_index, itemcopy);17341735 1736 let new_balance = <Balance<T>>::get(collection_id, &owner)1737 .checked_add(value)1738 .ok_or(Error::<T>::NumOverflow)?;1739 <Balance<T>>::insert(collection_id, owner.clone(), new_balance);17401741 Ok(())1742 }17431744 fn add_nft_item(collection_id: CollectionId, item: NftItemType<T::AccountId>) -> DispatchResult {1745 let current_index = <ItemListIndex>::get(collection_id)1746 .checked_add(1)1747 .ok_or(Error::<T>::NumOverflow)?;17481749 let item_owner = item.owner.clone();1750 Self::add_token_index(collection_id, current_index, &item.owner)?;17511752 <ItemListIndex>::insert(collection_id, current_index);1753 <NftItemList<T>>::insert(collection_id, current_index, item);17541755 1756 let new_balance = <Balance<T>>::get(collection_id, item_owner.clone())1757 .checked_add(1)1758 .ok_or(Error::<T>::NumOverflow)?;1759 <Balance<T>>::insert(collection_id, item_owner.clone(), new_balance);17601761 Ok(())1762 }17631764 fn burn_refungible_item(1765 collection_id: CollectionId,1766 item_id: TokenId,1767 owner: &T::AccountId,1768 ) -> DispatchResult {1769 ensure!(1770 <ReFungibleItemList<T>>::contains_key(collection_id, item_id),1771 Error::<T>::TokenNotFound1772 );1773 let mut token = <ReFungibleItemList<T>>::get(collection_id, item_id);1774 let rft_balance = token1775 .owner1776 .iter()1777 .filter(|&i| i.owner == *owner)1778 .next()1779 .unwrap();1780 Self::remove_token_index(collection_id, item_id, owner)?;17811782 1783 let new_balance = <Balance<T>>::get(collection_id, rft_balance.owner.clone())1784 .checked_sub(rft_balance.fraction)1785 .ok_or(Error::<T>::NumOverflow)?;1786 <Balance<T>>::insert(collection_id, rft_balance.owner.clone(), new_balance);17871788 1789 let index = token1790 .owner1791 .iter()1792 .position(|i| i.owner == *owner)1793 .unwrap();1794 token.owner.remove(index);1795 let owner_count = token.owner.len();17961797 1798 if owner_count == 0 {1799 <ReFungibleItemList<T>>::remove(collection_id, item_id);1800 }1801 else {1802 <ReFungibleItemList<T>>::insert(collection_id, item_id, token);1803 }18041805 Ok(())1806 }18071808 fn burn_nft_item(collection_id: CollectionId, item_id: TokenId) -> DispatchResult {1809 ensure!(1810 <NftItemList<T>>::contains_key(collection_id, item_id),1811 Error::<T>::TokenNotFound1812 );1813 let item = <NftItemList<T>>::get(collection_id, item_id);1814 Self::remove_token_index(collection_id, item_id, &item.owner)?;18151816 1817 let new_balance = <Balance<T>>::get(collection_id, &item.owner)1818 .checked_sub(1)1819 .ok_or(Error::<T>::NumOverflow)?;1820 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);1821 <NftItemList<T>>::remove(collection_id, item_id);18221823 Ok(())1824 }18251826 fn burn_fungible_item(owner: &T::AccountId, collection_id: CollectionId, value: u128) -> DispatchResult {1827 ensure!(1828 <FungibleItemList<T>>::contains_key(collection_id, owner),1829 Error::<T>::TokenNotFound1830 );1831 let mut balance = <FungibleItemList<T>>::get(collection_id, owner);1832 ensure!(balance.value >= value, Error::<T>::TokenValueNotEnough);18331834 1835 let new_balance = <Balance<T>>::get(collection_id, owner)1836 .checked_sub(value)1837 .ok_or(Error::<T>::NumOverflow)?;1838 <Balance<T>>::insert(collection_id, (*owner).clone(), new_balance);18391840 if balance.value - value > 0 {1841 balance.value -= value;1842 <FungibleItemList<T>>::insert(collection_id, (*owner).clone(), balance);1843 }1844 else {1845 <FungibleItemList<T>>::remove(collection_id, owner);1846 }18471848 Ok(())1849 }18501851 fn collection_exists(collection_id: CollectionId) -> DispatchResult {1852 ensure!(1853 <Collection<T>>::contains_key(collection_id),1854 Error::<T>::CollectionNotFound1855 );1856 Ok(())1857 }18581859 fn check_owner_permissions(collection_id: CollectionId, subject: T::AccountId) -> DispatchResult {1860 Self::collection_exists(collection_id)?;18611862 let target_collection = <Collection<T>>::get(collection_id);1863 ensure!(1864 subject == target_collection.owner,1865 Error::<T>::NoPermission1866 );18671868 Ok(())1869 }18701871 fn is_owner_or_admin_permissions(collection_id: CollectionId, subject: T::AccountId) -> bool {1872 let target_collection = <Collection<T>>::get(collection_id);1873 let mut result: bool = subject == target_collection.owner;1874 let exists = <AdminList<T>>::contains_key(collection_id);18751876 if !result & exists {1877 if <AdminList<T>>::get(collection_id).contains(&subject) {1878 result = true1879 }1880 }18811882 result1883 }18841885 fn check_owner_or_admin_permissions(1886 collection_id: CollectionId,1887 subject: T::AccountId,1888 ) -> DispatchResult {1889 Self::collection_exists(collection_id)?;1890 let result = Self::is_owner_or_admin_permissions(collection_id, subject.clone());18911892 ensure!(1893 result,1894 Error::<T>::NoPermission1895 );1896 Ok(())1897 }18981899 fn is_item_owner(subject: T::AccountId, collection_id: CollectionId, item_id: TokenId) -> bool {1900 let target_collection = <Collection<T>>::get(collection_id);19011902 match target_collection.mode {1903 CollectionMode::NFT => {1904 <NftItemList<T>>::get(collection_id, item_id).owner == subject1905 }1906 CollectionMode::Fungible(_) => {1907 <FungibleItemList<T>>::contains_key(collection_id, &subject)1908 }1909 CollectionMode::ReFungible => {1910 <ReFungibleItemList<T>>::get(collection_id, item_id)1911 .owner1912 .iter()1913 .any(|i| i.owner == subject)1914 }1915 CollectionMode::Invalid => false,1916 }1917 }19181919 fn check_white_list(collection_id: CollectionId, address: &T::AccountId) -> DispatchResult {1920 let mes = Error::<T>::AddresNotInWhiteList;1921 ensure!(<WhiteList<T>>::contains_key(collection_id, address), mes);19221923 Ok(())1924 }19251926 1927 1928 fn token_exists(1929 collection_id: CollectionId,1930 item_id: TokenId,1931 owner: &T::AccountId1932 ) -> DispatchResult {1933 let target_collection = <Collection<T>>::get(collection_id);1934 let exists = match target_collection.mode1935 {1936 CollectionMode::NFT => <NftItemList<T>>::contains_key(collection_id, item_id),1937 CollectionMode::Fungible(_) => <FungibleItemList<T>>::contains_key(collection_id, owner),1938 CollectionMode::ReFungible => <ReFungibleItemList<T>>::contains_key(collection_id, item_id),1939 _ => false1940 };19411942 ensure!(exists == true, Error::<T>::TokenNotFound);1943 Ok(())1944 }19451946 fn transfer_fungible(1947 collection_id: CollectionId,1948 value: u128,1949 owner: &T::AccountId,1950 recipient: &T::AccountId,1951 ) -> DispatchResult {1952 Self::token_exists(collection_id, 0, owner)?;19531954 let mut balance = <FungibleItemList<T>>::get(collection_id, owner);1955 ensure!(balance.value >= value, Error::<T>::TokenValueTooLow);19561957 1958 Self::add_fungible_item(collection_id, recipient, value)?;19591960 1961 <Balance<T>>::insert(collection_id, (*owner).clone(), balance.value - value);19621963 1964 if balance.value == value {1965 <FungibleItemList<T>>::remove(collection_id, owner);1966 }1967 else {1968 balance.value -= value;1969 <FungibleItemList<T>>::insert(collection_id, (*owner).clone(), balance);1970 }19711972 Ok(())1973 }19741975 fn transfer_refungible(1976 collection_id: CollectionId,1977 item_id: TokenId,1978 value: u128,1979 owner: T::AccountId,1980 new_owner: T::AccountId,1981 ) -> DispatchResult {1982 Self::token_exists(collection_id, item_id, &owner)?;19831984 let full_item = <ReFungibleItemList<T>>::get(collection_id, item_id);1985 let item = full_item1986 .owner1987 .iter()1988 .filter(|i| i.owner == owner)1989 .next()1990 .ok_or(Error::<T>::NumOverflow)?;1991 let amount = item.fraction;19921993 ensure!(amount >= value, Error::<T>::TokenValueTooLow);19941995 1996 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())1997 .checked_sub(value)1998 .ok_or(Error::<T>::NumOverflow)?;1999 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);20002001 let balancenew_owner = <Balance<T>>::get(collection_id, new_owner.clone())2002 .checked_add(value)2003 .ok_or(Error::<T>::NumOverflow)?;2004 <Balance<T>>::insert(collection_id, new_owner.clone(), balancenew_owner);20052006 let old_owner = item.owner.clone();2007 let new_owner_has_account = full_item.owner.iter().any(|i| i.owner == new_owner);20082009 2010 if amount == value && !new_owner_has_account {2011 2012 2013 let mut new_full_item = full_item.clone();2014 new_full_item2015 .owner2016 .iter_mut()2017 .find(|i| i.owner == owner)2018 .unwrap()2019 .owner = new_owner.clone();2020 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);20212022 2023 Self::move_token_index(collection_id, item_id, &old_owner, &new_owner)?;2024 } else {2025 let mut new_full_item = full_item.clone();2026 new_full_item2027 .owner2028 .iter_mut()2029 .find(|i| i.owner == owner)2030 .unwrap()2031 .fraction -= value;20322033 2034 if new_owner_has_account {2035 2036 new_full_item2037 .owner2038 .iter_mut()2039 .find(|i| i.owner == new_owner)2040 .unwrap()2041 .fraction += value;2042 } else {2043 2044 new_full_item.owner.push(Ownership {2045 owner: new_owner.clone(),2046 fraction: value,2047 });2048 Self::add_token_index(collection_id, item_id, &new_owner)?;2049 }20502051 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);2052 }20532054 Ok(())2055 }20562057 fn transfer_nft(2058 collection_id: CollectionId,2059 item_id: TokenId,2060 sender: T::AccountId,2061 new_owner: T::AccountId,2062 ) -> DispatchResult {2063 Self::token_exists(collection_id, item_id, &sender)?;20642065 let mut item = <NftItemList<T>>::get(collection_id, item_id);20662067 ensure!(2068 sender == item.owner,2069 Error::<T>::MustBeTokenOwner2070 );20712072 2073 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())2074 .checked_sub(1)2075 .ok_or(Error::<T>::NumOverflow)?;2076 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);20772078 let balancenew_owner = <Balance<T>>::get(collection_id, new_owner.clone())2079 .checked_add(1)2080 .ok_or(Error::<T>::NumOverflow)?;2081 <Balance<T>>::insert(collection_id, new_owner.clone(), balancenew_owner);20822083 2084 let old_owner = item.owner.clone();2085 item.owner = new_owner.clone();2086 <NftItemList<T>>::insert(collection_id, item_id, item);20872088 2089 Self::move_token_index(collection_id, item_id, &old_owner, &new_owner)?;20902091 Ok(())2092 }2093 2094 fn set_re_fungible_variable_data(2095 collection_id: CollectionId,2096 item_id: TokenId,2097 data: Vec<u8>2098 ) -> DispatchResult {2099 let mut item = <ReFungibleItemList<T>>::get(collection_id, item_id);21002101 item.variable_data = data;21022103 <ReFungibleItemList<T>>::insert(collection_id, item_id, item);21042105 Ok(())2106 }21072108 fn set_nft_variable_data(2109 collection_id: CollectionId,2110 item_id: TokenId,2111 data: Vec<u8>2112 ) -> DispatchResult {2113 let mut item = <NftItemList<T>>::get(collection_id, item_id);2114 2115 item.variable_data = data;21162117 <NftItemList<T>>::insert(collection_id, item_id, item);2118 2119 Ok(())2120 }21212122 fn init_collection(item: &CollectionType<T::AccountId>) {2123 2124 assert!(2125 item.decimal_points <= MAX_DECIMAL_POINTS,2126 "decimal_points parameter must be lower than MAX_DECIMAL_POINTS"2127 );2128 assert!(2129 item.name.len() <= 64,2130 "Collection name can not be longer than 63 char"2131 );2132 assert!(2133 item.name.len() <= 256,2134 "Collection description can not be longer than 255 char"2135 );2136 assert!(2137 item.token_prefix.len() <= 16,2138 "Token prefix can not be longer than 15 char"2139 );21402141 2142 let next_id = CreatedCollectionCount::get()2143 .checked_add(1)2144 .unwrap();21452146 CreatedCollectionCount::put(next_id);2147 }21482149 fn init_nft_token(collection_id: CollectionId, item: &NftItemType<T::AccountId>) {2150 let current_index = <ItemListIndex>::get(collection_id)2151 .checked_add(1)2152 .unwrap();21532154 let item_owner = item.owner.clone();2155 Self::add_token_index(collection_id, current_index, &item.owner).unwrap();21562157 <ItemListIndex>::insert(collection_id, current_index);21582159 2160 let new_balance = <Balance<T>>::get(collection_id, &item_owner)2161 .checked_add(1)2162 .unwrap();2163 <Balance<T>>::insert(collection_id, item_owner.clone(), new_balance);2164 }21652166 fn init_fungible_token(collection_id: CollectionId, owner: &T::AccountId, item: &FungibleItemType) {2167 let current_index = <ItemListIndex>::get(collection_id)2168 .checked_add(1)2169 .unwrap();21702171 Self::add_token_index(collection_id, current_index, owner).unwrap();21722173 <ItemListIndex>::insert(collection_id, current_index);21742175 2176 let new_balance = <Balance<T>>::get(collection_id, owner)2177 .checked_add(item.value)2178 .unwrap();2179 <Balance<T>>::insert(collection_id, (*owner).clone(), new_balance);2180 }21812182 fn init_refungible_token(collection_id: CollectionId, item: &ReFungibleItemType<T::AccountId>) {2183 let current_index = <ItemListIndex>::get(collection_id)2184 .checked_add(1)2185 .unwrap();21862187 let value = item.owner.first().unwrap().fraction;2188 let owner = item.owner.first().unwrap().owner.clone();21892190 Self::add_token_index(collection_id, current_index, &owner).unwrap();21912192 <ItemListIndex>::insert(collection_id, current_index);21932194 2195 let new_balance = <Balance<T>>::get(collection_id, &owner)2196 .checked_add(value)2197 .unwrap();2198 <Balance<T>>::insert(collection_id, owner.clone(), new_balance);2199 }22002201 fn add_token_index(collection_id: CollectionId, item_index: TokenId, owner: &T::AccountId) -> DispatchResult {22022203 2204 if <AccountItemCount<T>>::contains_key(owner) {22052206 2207 let count = <AccountItemCount<T>>::get(owner);2208 ensure!(count < ChainLimit::get().account_token_ownership_limit, Error::<T>::AddressOwnershipLimitExceeded);22092210 <AccountItemCount<T>>::insert(owner.clone(), count2211 .checked_add(1)2212 .ok_or(Error::<T>::NumOverflow)?);2213 }2214 else {2215 <AccountItemCount<T>>::insert(owner.clone(), 1);2216 }22172218 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner);2219 if list_exists {2220 let mut list = <AddressTokens<T>>::get(collection_id, owner);2221 let item_contains = list.contains(&item_index.clone());22222223 if !item_contains {2224 list.push(item_index.clone());2225 }22262227 <AddressTokens<T>>::insert(collection_id, owner.clone(), list);2228 } else {2229 let mut itm = Vec::new();2230 itm.push(item_index.clone());2231 <AddressTokens<T>>::insert(collection_id, owner.clone(), itm);2232 }22332234 Ok(())2235 }22362237 fn remove_token_index(2238 collection_id: CollectionId,2239 item_index: TokenId,2240 owner: &T::AccountId,2241 ) -> DispatchResult {22422243 2244 <AccountItemCount<T>>::insert(owner.clone(), 2245 <AccountItemCount<T>>::get(owner)2246 .checked_sub(1)2247 .ok_or(Error::<T>::NumOverflow)?);224822492250 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner);2251 if list_exists {2252 let mut list = <AddressTokens<T>>::get(collection_id, owner);2253 let item_contains = list.contains(&item_index.clone());22542255 if item_contains {2256 list.retain(|&item| item != item_index);2257 <AddressTokens<T>>::insert(collection_id, owner.clone(), list);2258 }2259 }22602261 Ok(())2262 }22632264 fn move_token_index(2265 collection_id: CollectionId,2266 item_index: TokenId,2267 old_owner: &T::AccountId,2268 new_owner: &T::AccountId,2269 ) -> DispatchResult {2270 Self::remove_token_index(collection_id, item_index, old_owner)?;2271 Self::add_token_index(collection_id, item_index, new_owner)?;22722273 Ok(())2274 }2275 2276 fn ensure_contract_owned(account: T::AccountId, contract: &T::AccountId) -> DispatchResult {2277 if <ContractOwner<T>>::contains_key(contract.clone()) {2278 let owner = <ContractOwner<T>>::get(contract);2279 ensure!(account == owner, Error::<T>::NoPermission);2280 } else {2281 fail!(Error::<T>::NoPermission);2282 }22832284 Ok(())2285 }2286}2287228822892290229122922293pub type Multiplier = FixedU128;22942295type BalanceOf<T> = <<T as transaction_payment::Config>::OnChargeTransaction as transaction_payment::OnChargeTransaction<T>>::Balance;2296229722982299#[derive(Encode, Decode, Clone, Eq, PartialEq)]2300pub struct ChargeTransactionPayment<T: Config>(#[codec(compact)] BalanceOf<T>);23012302impl<T: Config + Send + Sync> sp_std::fmt::Debug 2303 for ChargeTransactionPayment<T>2304{2305 #[cfg(feature = "std")]2306 fn fmt(&self, f: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {2307 write!(f, "ChargeTransactionPayment<{:?}>", self.0)2308 }2309 #[cfg(not(feature = "std"))]2310 fn fmt(&self, _: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {2311 Ok(())2312 }2313}23142315impl<T: Config> ChargeTransactionPayment<T>2316where2317 T::Call: Dispatchable<Info=DispatchInfo, PostInfo=PostDispatchInfo> + IsSubType<Call<T>> + IsSubType<pallet_contracts::Call<T>>,2318 BalanceOf<T>: Send + Sync + From<u64> + FixedPointOperand,2319 T::AccountId: AsRef<[u8]>,2320 T::AccountId: UncheckedFrom<T::Hash>,2321{2322 fn traditional_fee(2323 len: usize,2324 info: &DispatchInfoOf<T::Call>,2325 tip: BalanceOf<T>,2326 ) -> BalanceOf<T>2327 where2328 T::Call: Dispatchable<Info = DispatchInfo>,2329 {2330 <transaction_payment::Module<T>>::compute_fee(len as u32, info, tip)2331 }23322333 fn get_priority(len: usize, info: &DispatchInfoOf<T::Call>, final_fee: BalanceOf<T>) -> TransactionPriority {2334 let weight_saturation = T::BlockWeights::get().max_block / info.weight.max(1);2335 let max_block_length = *T::BlockLength::get().max.get(DispatchClass::Normal);2336 let len_saturation = max_block_length as u64 / (len as u64).max(1);2337 let coefficient: BalanceOf<T> = weight_saturation2338 .min(len_saturation)2339 .saturated_into::<BalanceOf<T>>();2340 final_fee2341 .saturating_mul(coefficient)2342 .saturated_into::<TransactionPriority>()2343 }23442345 fn withdraw_fee(2346 &self,2347 who: &T::AccountId,2348 call: &T::Call,2349 info: &DispatchInfoOf<T::Call>,2350 len: usize,2351 ) -> Result<2352 (2353 BalanceOf<T>,2354 <<T as transaction_payment::Config>::OnChargeTransaction as transaction_payment::OnChargeTransaction<T>>::LiquidityInfo,2355 ),2356 TransactionValidityError,2357 > {2358 let tip = self.0;23592360 2361 2362 2363 2364 2365 2366 2367 let fee = Self::traditional_fee(len, info, tip);23682369 2370 if fee.is_zero() {2371 return <<T as transaction_payment::Config>::OnChargeTransaction as transaction_payment::OnChargeTransaction<T>>::withdraw_fee(who, call, info, fee, tip)2372 .map(|i| (fee, i));2373 }23742375 2376 2377 let mut sponsor: T::AccountId = match IsSubType::<Call<T>>::is_sub_type(call) {2378 Some(Call::create_item(collection_id, _owner, _properties)) => {23792380 2381 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;23822383 let limit = <Collection<T>>::get(collection_id).limits.sponsor_transfer_timeout;2384 let mut sponsored = true;2385 if <CreateItemBasket<T>>::contains_key((collection_id, &who)) {2386 let last_tx_block = <CreateItemBasket<T>>::get((collection_id, &who));2387 let limit_time = last_tx_block + limit.into();2388 if block_number <= limit_time {2389 sponsored = false;2390 }2391 }2392 if sponsored {2393 <CreateItemBasket<T>>::insert((collection_id, who.clone()), block_number);2394 }23952396 2397 if (<Collection<T>>::get(collection_id).limits.sponsored_data_size >= (_properties.len() as u32)) &&2398 (<Collection<T>>::get(collection_id).sponsor_confirmed) &&2399 (sponsored)2400 {2401 <Collection<T>>::get(collection_id).sponsor2402 } else {2403 T::AccountId::default()2404 }2405 }2406 Some(Call::transfer(_new_owner, collection_id, item_id, _value)) => {2407 2408 let mut sponsor_transfer = false;2409 if <Collection<T>>::get(collection_id).sponsor_confirmed {24102411 let collection_limits = <Collection<T>>::get(collection_id).limits;2412 let collection_mode = <Collection<T>>::get(collection_id).mode;2413 2414 2415 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2416 sponsor_transfer = match collection_mode {2417 CollectionMode::NFT => {2418 2419 2420 let limit: u32 = if collection_limits.sponsor_transfer_timeout > 0 {2421 collection_limits.sponsor_transfer_timeout2422 } else {2423 ChainLimit::get().nft_sponsor_transfer_timeout2424 };2425 2426 let mut sponsored = true;2427 if <NftTransferBasket<T>>::contains_key(collection_id, item_id) {2428 let last_tx_block = <NftTransferBasket<T>>::get(collection_id, item_id);2429 let limit_time = last_tx_block + limit.into();2430 if block_number <= limit_time {2431 sponsored = false;2432 }2433 }2434 if sponsored {2435 <NftTransferBasket<T>>::insert(collection_id, item_id, block_number);2436 }24372438 sponsored2439 }2440 CollectionMode::Fungible(_) => {2441 2442 2443 let limit: u32 = if collection_limits.sponsor_transfer_timeout > 0 {2444 collection_limits.sponsor_transfer_timeout2445 } else {2446 ChainLimit::get().fungible_sponsor_transfer_timeout2447 };2448 2449 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2450 let mut sponsored = true;2451 if <FungibleTransferBasket<T>>::contains_key(collection_id, who) {2452 let last_tx_block = <FungibleTransferBasket<T>>::get(collection_id, who);2453 let limit_time = last_tx_block + limit.into();2454 if block_number <= limit_time {2455 sponsored = false;2456 }2457 }2458 if sponsored {2459 <FungibleTransferBasket<T>>::insert(collection_id, who, block_number);2460 }24612462 sponsored2463 }2464 CollectionMode::ReFungible => {2465 2466 2467 let limit: u32 = if collection_limits.sponsor_transfer_timeout > 0 {2468 collection_limits.sponsor_transfer_timeout2469 } else {2470 ChainLimit::get().refungible_sponsor_transfer_timeout2471 };2472 2473 let mut sponsored = true;2474 if <ReFungibleTransferBasket<T>>::contains_key(collection_id, item_id) {2475 let last_tx_block = <ReFungibleTransferBasket<T>>::get(collection_id, item_id);2476 let limit_time = last_tx_block + limit.into();2477 if block_number <= limit_time {2478 sponsored = false;2479 }2480 }2481 if sponsored {2482 <ReFungibleTransferBasket<T>>::insert(collection_id, item_id, block_number);2483 }24842485 sponsored2486 }2487 _ => {2488 false2489 },2490 };2491 }24922493 if !sponsor_transfer {2494 T::AccountId::default()2495 } else {2496 <Collection<T>>::get(collection_id).sponsor2497 }2498 }24992500 _ => T::AccountId::default(),2501 };25022503 2504 sponsor = match IsSubType::<pallet_contracts::Call<T>>::is_sub_type(call) {25052506 2507 Some(pallet_contracts::Call::instantiate(_endowment, _gas_limit, code_hash, _data, salt)) => {25082509 let new_contract_address = <pallet_contracts::Module<T>>::contract_address(2510 &who,2511 code_hash,2512 salt,2513 );2514 <ContractOwner<T>>::insert(new_contract_address.clone(), who.clone());25152516 T::AccountId::default()2517 },25182519 2520 Some(pallet_contracts::Call::call(dest, _value, _gas_limit, _data)) => {25212522 let called_contract: T::AccountId = T::Lookup::lookup((*dest).clone()).unwrap_or(T::AccountId::default());25232524 let owned_contract = <ContractOwner<T>>::contains_key(called_contract.clone())2525 && <ContractOwner<T>>::get(called_contract.clone()) == *who;2526 let white_list_enabled = <ContractWhiteListEnabled<T>>::contains_key(called_contract.clone()) && <ContractWhiteListEnabled<T>>::get(called_contract.clone());2527 2528 if !owned_contract && white_list_enabled {2529 if !<ContractWhiteList<T>>::contains_key(called_contract.clone(), who) {2530 return Err(InvalidTransaction::Call.into());2531 }2532 }25332534 let mut sponsor_transfer = false;2535 if <ContractSponsoringRateLimit<T>>::contains_key(called_contract.clone()) {2536 let last_tx_block = <ContractSponsorBasket<T>>::get((&called_contract, &who));2537 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2538 let rate_limit = <ContractSponsoringRateLimit<T>>::get(&called_contract);2539 let limit_time = last_tx_block + rate_limit;25402541 if block_number >= limit_time {2542 <ContractSponsorBasket<T>>::insert((called_contract.clone(), who.clone()), block_number);2543 sponsor_transfer = true;2544 }2545 } else {2546 sponsor_transfer = false;2547 }2548 2549 2550 let mut sp = T::AccountId::default();2551 if sponsor_transfer {2552 if <ContractSelfSponsoring<T>>::contains_key(called_contract.clone()) {2553 if <ContractSelfSponsoring<T>>::get(called_contract.clone()) {2554 sp = called_contract;2555 }2556 }2557 }25582559 sp2560 },25612562 _ => sponsor,2563 };25642565 let mut who_pays_fee: T::AccountId = sponsor.clone();2566 if sponsor == T::AccountId::default() {2567 who_pays_fee = who.clone();2568 }25692570 <<T as transaction_payment::Config>::OnChargeTransaction as transaction_payment::OnChargeTransaction<T>>::withdraw_fee(&who_pays_fee, call, info, fee, tip)2571 .map(|i| (fee, i))2572 }2573}257425752576impl<T: Config + Send + Sync> SignedExtension2577 for ChargeTransactionPayment<T>2578where2579 BalanceOf<T>: Send + Sync + From<u64> + FixedPointOperand,2580 T::Call: Dispatchable<Info = DispatchInfo, PostInfo = PostDispatchInfo> + IsSubType<Call<T>> + IsSubType<pallet_contracts::Call<T>>,2581 T::AccountId: AsRef<[u8]>,2582 T::AccountId: UncheckedFrom<T::Hash>,2583{2584 const IDENTIFIER: &'static str = "ChargeTransactionPayment";2585 type AccountId = T::AccountId;2586 type Call = T::Call;2587 type AdditionalSigned = ();2588 type Pre = (2589 2590 BalanceOf<T>,2591 2592 Self::AccountId,2593 2594 <<T as transaction_payment::Config>::OnChargeTransaction as transaction_payment::OnChargeTransaction<T>>::LiquidityInfo,2595 );2596 fn additional_signed(&self) -> sp_std::result::Result<(), TransactionValidityError> {2597 Ok(())2598 }25992600 fn validate(2601 &self,2602 who: &Self::AccountId,2603 call: &Self::Call,2604 info: &DispatchInfoOf<Self::Call>,2605 len: usize,2606 ) -> TransactionValidity {2607 let (fee, _) = self.withdraw_fee(who, call, info, len)?;2608 Ok(ValidTransaction {2609 priority: Self::get_priority(len, info, fee),2610 ..Default::default()2611 })2612 }26132614 fn pre_dispatch(2615 self,2616 who: &Self::AccountId,2617 call: &Self::Call,2618 info: &DispatchInfoOf<Self::Call>,2619 len: usize,2620 ) -> Result<Self::Pre, TransactionValidityError> {2621 let (_fee, imbalance) = self.withdraw_fee(who, call, info, len)?;2622 Ok((self.0, who.clone(), imbalance))2623 }26242625 fn post_dispatch(2626 pre: Self::Pre,2627 info: &DispatchInfoOf<Self::Call>,2628 post_info: &PostDispatchInfoOf<Self::Call>,2629 len: usize,2630 _result: &DispatchResult,2631 ) -> Result<(), TransactionValidityError> {2632 let (tip, who, imbalance) = pre;2633 let actual_fee = transaction_payment::Module::<T>::compute_actual_fee(2634 len as u32,2635 info,2636 post_info,2637 tip,2638 );2639 <T as transaction_payment::Config>::OnChargeTransaction::correct_and_deposit_fee(&who, info, post_info, actual_fee, tip, imbalance)?;2640 Ok(())2641 }2642}26432644