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 pallet_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,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 + pallet_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 501 502 ItemCreated(CollectionId, TokenId, AccountId),503504 505 506 507 508 509 510 511 ItemDestroyed(CollectionId, TokenId),512513 514 515 516 517 518 519 520 521 522 523 524 Transfer(CollectionId, TokenId, AccountId, AccountId, u128),525 }526);527528decl_module! {529 pub struct Module<T: Config> for enum Call 530 where 531 origin: T::Origin532 {533 fn deposit_event() = default;534 type Error = Error<T>;535536 fn on_initialize(now: T::BlockNumber) -> Weight {537538 if ChainVersion::get() < 2539 {540 let value = NextCollectionID::get();541 CreatedCollectionCount::put(value);542 ChainVersion::put(2);543 }544545 0546 }547548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 #[weight = <T as Config>::WeightInfo::create_collection()]565 pub fn create_collection(origin,566 collection_name: Vec<u16>,567 collection_description: Vec<u16>,568 token_prefix: Vec<u8>,569 mode: CollectionMode) -> DispatchResult {570571 572 let who = ensure_signed(origin)?;573574 let decimal_points = match mode {575 CollectionMode::Fungible(points) => points,576 _ => 0577 };578579 let chain_limit = ChainLimit::get();580581 582 ensure!(CollectionCount::get() < chain_limit.collection_numbers_limit, Error::<T>::TotalCollectionsLimitExceeded);583584 585 ensure!(decimal_points <= MAX_DECIMAL_POINTS, Error::<T>::CollectionDecimalPointLimitExceeded);586 ensure!(collection_name.len() <= 64, Error::<T>::CollectionNameLimitExceeded);587 ensure!(collection_description.len() <= 256, Error::<T>::CollectionDescriptionLimitExceeded);588 ensure!(token_prefix.len() <= 16, Error::<T>::CollectionTokenPrefixLimitExceeded);589590 591 let next_id = CreatedCollectionCount::get()592 .checked_add(1)593 .ok_or(Error::<T>::NumOverflow)?;594595 596 let total = CollectionCount::get()597 .checked_add(1)598 .ok_or(Error::<T>::NumOverflow)?;599600 CreatedCollectionCount::put(next_id);601 CollectionCount::put(total);602603 let limits = CollectionLimits {604 sponsored_data_size: chain_limit.custom_data_limit,605 ..Default::default()606 };607608 609 let new_collection = CollectionType {610 owner: who.clone(),611 name: collection_name,612 mode: mode.clone(),613 mint_mode: false,614 access: AccessMode::Normal,615 description: collection_description,616 decimal_points: decimal_points,617 token_prefix: token_prefix,618 offchain_schema: Vec::new(),619 schema_version: SchemaVersion::ImageURL,620 sponsor: T::AccountId::default(),621 sponsor_confirmed: false,622 variable_on_chain_schema: Vec::new(),623 const_on_chain_schema: Vec::new(),624 limits,625 };626627 628 <Collection<T>>::insert(next_id, new_collection);629630 631 Self::deposit_event(RawEvent::Created(next_id, mode.into(), who.clone()));632633 Ok(())634 }635636 637 638 639 640 641 642 643 644 645 #[weight = <T as Config>::WeightInfo::destroy_collection()]646 pub fn destroy_collection(origin, collection_id: CollectionId) -> DispatchResult {647648 let sender = ensure_signed(origin)?;649 Self::check_owner_permissions(collection_id, sender)?;650651 let target_collection = <Collection<T>>::get(collection_id);652 if !target_collection.limits.owner_can_destroy {653 fail!(Error::<T>::NoPermission);654 }655656 <AddressTokens<T>>::remove_prefix(collection_id);657 <Allowances<T>>::remove_prefix(collection_id);658 <Balance<T>>::remove_prefix(collection_id);659 <ItemListIndex>::remove(collection_id);660 <AdminList<T>>::remove(collection_id);661 <Collection<T>>::remove(collection_id);662 <WhiteList<T>>::remove_prefix(collection_id);663664 <NftItemList<T>>::remove_prefix(collection_id);665 <FungibleItemList<T>>::remove_prefix(collection_id);666 <ReFungibleItemList<T>>::remove_prefix(collection_id);667668 <NftTransferBasket<T>>::remove_prefix(collection_id);669 <FungibleTransferBasket<T>>::remove_prefix(collection_id);670 <ReFungibleTransferBasket<T>>::remove_prefix(collection_id);671672 if CollectionCount::get() > 0673 {674 675 let total = CollectionCount::get()676 .checked_sub(1)677 .ok_or(Error::<T>::NumOverflow)?;678679 CollectionCount::put(total);680 }681682 Ok(())683 }684685 686 687 688 689 690 691 692 693 694 695 696 697 #[weight = <T as Config>::WeightInfo::add_to_white_list()]698 pub fn add_to_white_list(origin, collection_id: CollectionId, address: T::AccountId) -> DispatchResult{699700 let sender = ensure_signed(origin)?;701 Self::check_owner_or_admin_permissions(collection_id, sender)?;702703 <WhiteList<T>>::insert(collection_id, address, true);704 705 Ok(())706 }707708 709 710 711 712 713 714 715 716 717 718 719 720 #[weight = <T as Config>::WeightInfo::remove_from_white_list()]721 pub fn remove_from_white_list(origin, collection_id: CollectionId, address: T::AccountId) -> DispatchResult{722723 let sender = ensure_signed(origin)?;724 Self::check_owner_or_admin_permissions(collection_id, sender)?;725726 <WhiteList<T>>::remove(collection_id, address);727728 Ok(())729 }730731 732 733 734 735 736 737 738 739 740 741 742 #[weight = <T as Config>::WeightInfo::set_public_access_mode()]743 pub fn set_public_access_mode(origin, collection_id: CollectionId, mode: AccessMode) -> DispatchResult744 {745 let sender = ensure_signed(origin)?;746747 Self::check_owner_permissions(collection_id, sender)?;748 let mut target_collection = <Collection<T>>::get(collection_id);749 target_collection.access = mode;750 <Collection<T>>::insert(collection_id, target_collection);751752 Ok(())753 }754755 756 757 758 759 760 761 762 763 764 765 766 767 768 #[weight = <T as Config>::WeightInfo::set_mint_permission()]769 pub fn set_mint_permission(origin, collection_id: CollectionId, mint_permission: bool) -> DispatchResult770 {771 let sender = ensure_signed(origin)?;772773 Self::check_owner_permissions(collection_id, sender)?;774 let mut target_collection = <Collection<T>>::get(collection_id);775 target_collection.mint_mode = mint_permission;776 <Collection<T>>::insert(collection_id, target_collection);777778 Ok(())779 }780781 782 783 784 785 786 787 788 789 790 791 792 #[weight = <T as Config>::WeightInfo::change_collection_owner()]793 pub fn change_collection_owner(origin, collection_id: CollectionId, new_owner: T::AccountId) -> DispatchResult {794795 let sender = ensure_signed(origin)?;796 Self::check_owner_permissions(collection_id, sender)?;797 let mut target_collection = <Collection<T>>::get(collection_id);798 target_collection.owner = new_owner;799 <Collection<T>>::insert(collection_id, target_collection);800801 Ok(())802 }803804 805 806 807 808 809 810 811 812 813 814 815 816 817 #[weight = <T as Config>::WeightInfo::add_collection_admin()]818 pub fn add_collection_admin(origin, collection_id: CollectionId, new_admin_id: T::AccountId) -> DispatchResult {819820 let sender = ensure_signed(origin)?;821 Self::check_owner_or_admin_permissions(collection_id, sender)?;822 let mut admin_arr: Vec<T::AccountId> = Vec::new();823824 if <AdminList<T>>::contains_key(collection_id)825 {826 admin_arr = <AdminList<T>>::get(collection_id);827 ensure!(!admin_arr.contains(&new_admin_id), Error::<T>::AlreadyAdmin);828 }829830 831 ensure!((admin_arr.len() as u64) < ChainLimit::get().collections_admins_limit, Error::<T>::CollectionAdminsLimitExceeded);832833 admin_arr.push(new_admin_id);834 <AdminList<T>>::insert(collection_id, admin_arr);835836 Ok(())837 }838839 840 841 842 843 844 845 846 847 848 849 850 851 #[weight = <T as Config>::WeightInfo::remove_collection_admin()]852 pub fn remove_collection_admin(origin, collection_id: CollectionId, account_id: T::AccountId) -> DispatchResult {853854 let sender = ensure_signed(origin)?;855 Self::check_owner_or_admin_permissions(collection_id, sender)?;856 ensure!(<AdminList<T>>::contains_key(collection_id), Error::<T>::AdminNotFound);857858 let mut admin_arr = <AdminList<T>>::get(collection_id);859 admin_arr.retain(|i| *i != account_id);860 <AdminList<T>>::insert(collection_id, admin_arr);861862 Ok(())863 }864865 866 867 868 869 870 871 872 873 874 #[weight = <T as Config>::WeightInfo::set_collection_sponsor()]875 pub fn set_collection_sponsor(origin, collection_id: CollectionId, new_sponsor: T::AccountId) -> DispatchResult {876877 let sender = ensure_signed(origin)?;878 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);879880 let mut target_collection = <Collection<T>>::get(collection_id);881 ensure!(sender == target_collection.owner, Error::<T>::NoPermission);882883 target_collection.sponsor = new_sponsor;884 target_collection.sponsor_confirmed = false;885 <Collection<T>>::insert(collection_id, target_collection);886887 Ok(())888 }889890 891 892 893 894 895 896 897 #[weight = <T as Config>::WeightInfo::confirm_sponsorship()]898 pub fn confirm_sponsorship(origin, collection_id: CollectionId) -> DispatchResult {899900 let sender = ensure_signed(origin)?;901 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);902903 let mut target_collection = <Collection<T>>::get(collection_id);904 ensure!(sender == target_collection.sponsor, Error::<T>::ConfirmUnsetSponsorFail);905906 target_collection.sponsor_confirmed = true;907 <Collection<T>>::insert(collection_id, target_collection);908909 Ok(())910 }911912 913 914 915 916 917 918 919 920 921 #[weight = <T as Config>::WeightInfo::remove_collection_sponsor()]922 pub fn remove_collection_sponsor(origin, collection_id: CollectionId) -> DispatchResult {923924 let sender = ensure_signed(origin)?;925 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);926927 let mut target_collection = <Collection<T>>::get(collection_id);928 ensure!(sender == target_collection.owner, Error::<T>::NoPermission);929930 target_collection.sponsor = T::AccountId::default();931 target_collection.sponsor_confirmed = false;932 <Collection<T>>::insert(collection_id, target_collection);933934 Ok(())935 }936937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960961 #[weight = <T as Config>::WeightInfo::create_item(data.len())]962 pub fn create_item(origin, collection_id: CollectionId, owner: T::AccountId, data: CreateItemData) -> DispatchResult {963964 let sender = ensure_signed(origin)?;965966 Self::collection_exists(collection_id)?;967968 let target_collection = <Collection<T>>::get(collection_id);969970 Self::can_create_items_in_collection(collection_id, &target_collection, &sender, &owner)?;971 Self::validate_create_item_args(&target_collection, &data)?;972 Self::create_item_no_validation(collection_id, owner, data)?;973974 Ok(())975 }976977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 #[weight = <T as Config>::WeightInfo::create_item(items_data.into_iter()996 .map(|data| { data.len() })997 .sum())]998 pub fn create_multiple_items(origin, collection_id: CollectionId, owner: T::AccountId, items_data: Vec<CreateItemData>) -> DispatchResult {9991000 ensure!(items_data.len() > 0, Error::<T>::EmptyArgument);1001 let sender = ensure_signed(origin)?;10021003 Self::collection_exists(collection_id)?;1004 let target_collection = <Collection<T>>::get(collection_id);10051006 Self::can_create_items_in_collection(collection_id, &target_collection, &sender, &owner)?;10071008 for data in &items_data {1009 Self::validate_create_item_args(&target_collection, data)?;1010 }1011 for data in &items_data {1012 Self::create_item_no_validation(collection_id, owner.clone(), data.clone())?;1013 }10141015 Ok(())1016 }10171018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 #[weight = <T as Config>::WeightInfo::burn_item()]1032 pub fn burn_item(origin, collection_id: CollectionId, item_id: TokenId, value: u128) -> DispatchResult {10331034 let sender = ensure_signed(origin)?;1035 Self::collection_exists(collection_id)?;10361037 1038 let target_collection = <Collection<T>>::get(collection_id);1039 ensure!(1040 Self::is_item_owner(sender.clone(), collection_id, item_id) ||1041 (1042 target_collection.limits.owner_can_transfer &&1043 Self::is_owner_or_admin_permissions(collection_id, sender.clone())1044 ),1045 Error::<T>::NoPermission1046 );10471048 if target_collection.access == AccessMode::WhiteList {1049 Self::check_white_list(collection_id, &sender)?;1050 }10511052 match target_collection.mode1053 {1054 CollectionMode::NFT => Self::burn_nft_item(collection_id, item_id)?,1055 CollectionMode::Fungible(_) => Self::burn_fungible_item(&sender, collection_id, value)?,1056 CollectionMode::ReFungible => Self::burn_refungible_item(collection_id, item_id, &sender)?,1057 _ => ()1058 };10591060 1061 Self::deposit_event(RawEvent::ItemDestroyed(collection_id, item_id));10621063 Ok(())1064 }10651066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 #[weight = <T as Config>::WeightInfo::transfer()]1090 pub fn transfer(origin, recipient: T::AccountId, collection_id: CollectionId, item_id: TokenId, value: u128) -> DispatchResult {1091 let sender = ensure_signed(origin)?;1092 Self::transfer_internal(sender, recipient, collection_id, item_id, value)1093 }10941095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 #[weight = <T as Config>::WeightInfo::approve()]1111 pub fn approve(origin, spender: T::AccountId, collection_id: CollectionId, item_id: TokenId, amount: u128) -> DispatchResult {11121113 let sender = ensure_signed(origin)?;11141115 Self::collection_exists(collection_id)?;1116 Self::token_exists(collection_id, item_id, &sender)?;11171118 1119 let target_collection = <Collection<T>>::get(collection_id);1120 let allowance_limit = if (1121 target_collection.limits.owner_can_transfer &&1122 Self::is_owner_or_admin_permissions(1123 collection_id,1124 sender.clone(),1125 )1126 ) {1127 None1128 } else if let Some(amount) = Self::owned_amount(1129 sender.clone(),1130 collection_id,1131 item_id,1132 ) {1133 Some(amount)1134 } else {1135 fail!(Error::<T>::NoPermission);1136 };11371138 if target_collection.access == AccessMode::WhiteList {1139 Self::check_white_list(collection_id, &sender)?;1140 Self::check_white_list(collection_id, &spender)?;1141 }11421143 let allowance_exists = <Allowances<T>>::contains_key(collection_id, (item_id, &sender, &spender));1144 let mut allowance: u128 = amount;1145 if allowance_exists {1146 allowance += <Allowances<T>>::get(collection_id, (item_id, &sender, &spender));1147 }1148 if let Some(limit) = allowance_limit {1149 ensure!(limit >= allowance, Error::<T>::TokenValueTooLow);1150 }1151 <Allowances<T>>::insert(collection_id, (item_id, sender.clone(), spender.clone()), allowance);11521153 Ok(())1154 }1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 #[weight = <T as Config>::WeightInfo::transfer_from()]1176 pub fn transfer_from(origin, from: T::AccountId, recipient: T::AccountId, collection_id: CollectionId, item_id: TokenId, value: u128 ) -> DispatchResult {11771178 let sender = ensure_signed(origin)?;1179 let mut appoved_transfer = false;11801181 1182 let mut approval: u128 = 0;1183 if <Allowances<T>>::contains_key(collection_id, (item_id, &from, &sender)) {1184 approval = <Allowances<T>>::get(collection_id, (item_id, &from, &sender));1185 ensure!(approval >= value, Error::<T>::TokenValueNotEnough);1186 appoved_transfer = true;1187 }11881189 let target_collection = <Collection<T>>::get(collection_id);11901191 1192 Self::is_correct_transfer(collection_id, &target_collection, &recipient)?;11931194 1195 ensure!(1196 appoved_transfer || 1197 (1198 target_collection.limits.owner_can_transfer &&1199 Self::is_owner_or_admin_permissions(collection_id, sender.clone())1200 ),1201 Error::<T>::NoPermission1202 );12031204 if target_collection.access == AccessMode::WhiteList {1205 Self::check_white_list(collection_id, &sender)?;1206 Self::check_white_list(collection_id, &recipient)?;1207 }12081209 1210 if approval.checked_sub(value).unwrap_or(0) > 0 {1211 <Allowances<T>>::insert(collection_id, (item_id, &from, &sender), approval - value);1212 }1213 else {1214 <Allowances<T>>::remove(collection_id, (item_id, &from, &sender));1215 }12161217 match target_collection.mode1218 {1219 CollectionMode::NFT => Self::transfer_nft(collection_id, item_id, from, recipient)?,1220 CollectionMode::Fungible(_) => Self::transfer_fungible(collection_id, value, &from, &recipient)?,1221 CollectionMode::ReFungible => Self::transfer_refungible(collection_id, item_id, value, from.clone(), recipient)?,1222 _ => ()1223 };12241225 Ok(())1226 }12271228 1229 12301231 1232 1233 1234 12351236 12371238 12391240 1241 12421243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 #[weight = <T as Config>::WeightInfo::set_variable_meta_data()]1256 pub fn set_variable_meta_data (1257 origin,1258 collection_id: CollectionId,1259 item_id: TokenId,1260 data: Vec<u8>1261 ) -> DispatchResult {1262 let sender = ensure_signed(origin)?;1263 1264 Self::collection_exists(collection_id)?;1265 Self::token_exists(collection_id, item_id, &sender)?;12661267 ensure!(ChainLimit::get().custom_data_limit >= data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);12681269 1270 let target_collection = <Collection<T>>::get(collection_id);1271 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1272 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1273 Error::<T>::NoPermission);12741275 match target_collection.mode1276 {1277 CollectionMode::NFT => Self::set_nft_variable_data(collection_id, item_id, data)?,1278 CollectionMode::ReFungible => Self::set_re_fungible_variable_data(collection_id, item_id, data)?,1279 CollectionMode::Fungible(_) => fail!(Error::<T>::CantStoreMetadataInFungibleTokens),1280 _ => fail!(Error::<T>::UnexpectedCollectionType)1281 };12821283 Ok(())1284 }1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 #[weight = <T as Config>::WeightInfo::set_schema_version()]1301 pub fn set_schema_version(1302 origin,1303 collection_id: CollectionId,1304 version: SchemaVersion1305 ) -> DispatchResult {1306 let sender = ensure_signed(origin)?;1307 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;1308 let mut target_collection = <Collection<T>>::get(collection_id);1309 target_collection.schema_version = version;1310 <Collection<T>>::insert(collection_id, target_collection);13111312 Ok(())1313 }13141315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 #[weight = <T as Config>::WeightInfo::set_offchain_schema()]1328 pub fn set_offchain_schema(1329 origin,1330 collection_id: CollectionId,1331 schema: Vec<u8>1332 ) -> DispatchResult {1333 let sender = ensure_signed(origin)?;1334 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;13351336 1337 ensure!(schema.len() as u32 <= ChainLimit::get().offchain_schema_limit, "");13381339 let mut target_collection = <Collection<T>>::get(collection_id);1340 target_collection.offchain_schema = schema;1341 <Collection<T>>::insert(collection_id, target_collection);13421343 Ok(())1344 }13451346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 #[weight = <T as Config>::WeightInfo::set_const_on_chain_schema()]1359 pub fn set_const_on_chain_schema (1360 origin,1361 collection_id: CollectionId,1362 schema: Vec<u8>1363 ) -> DispatchResult {1364 let sender = ensure_signed(origin)?;1365 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;13661367 1368 ensure!(schema.len() as u32 <= ChainLimit::get().const_on_chain_schema_limit, "");13691370 let mut target_collection = <Collection<T>>::get(collection_id);1371 target_collection.const_on_chain_schema = schema;1372 <Collection<T>>::insert(collection_id, target_collection);13731374 Ok(())1375 }13761377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 #[weight = <T as Config>::WeightInfo::set_const_on_chain_schema()]1390 pub fn set_variable_on_chain_schema (1391 origin,1392 collection_id: CollectionId,1393 schema: Vec<u8>1394 ) -> DispatchResult {1395 let sender = ensure_signed(origin)?;1396 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;13971398 1399 ensure!(schema.len() as u32 <= ChainLimit::get().variable_on_chain_schema_limit, "");14001401 let mut target_collection = <Collection<T>>::get(collection_id);1402 target_collection.variable_on_chain_schema = schema;1403 <Collection<T>>::insert(collection_id, target_collection);14041405 Ok(())1406 }14071408 1409 #[weight = <T as Config>::WeightInfo::set_chain_limits()]1410 pub fn set_chain_limits(1411 origin,1412 limits: ChainLimits1413 ) -> DispatchResult {14141415 #[cfg(not(feature = "runtime-benchmarks"))]1416 ensure_root(origin)?;14171418 <ChainLimit>::put(limits);1419 Ok(())1420 }14211422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 #[weight = <T as Config>::WeightInfo::enable_contract_sponsoring()]1434 pub fn enable_contract_sponsoring(1435 origin,1436 contract_address: T::AccountId,1437 enable: bool1438 ) -> DispatchResult {14391440 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)?;14461447 <ContractSelfSponsoring<T>>::insert(contract_address, enable);1448 Ok(())1449 }14501451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 #[weight = <T as Config>::WeightInfo::set_contract_sponsoring_rate_limit()]1469 pub fn set_contract_sponsoring_rate_limit(1470 origin,1471 contract_address: T::AccountId,1472 rate_limit: T::BlockNumber1473 ) -> DispatchResult {1474 let sender = ensure_signed(origin)?;14751476 #[cfg(feature = "runtime-benchmarks")]1477 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());14781479 Self::ensure_contract_owned(sender, &contract_address)?;1480 <ContractSponsoringRateLimit<T>>::insert(contract_address, rate_limit);1481 Ok(())1482 }14831484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 #[weight = <T as Config>::WeightInfo::toggle_contract_white_list()]1496 pub fn toggle_contract_white_list(1497 origin,1498 contract_address: T::AccountId,1499 enable: bool1500 ) -> DispatchResult {1501 let sender = ensure_signed(origin)?;15021503 #[cfg(feature = "runtime-benchmarks")]1504 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());15051506 Self::ensure_contract_owned(sender, &contract_address)?;1507 <ContractWhiteListEnabled<T>>::insert(contract_address, enable);1508 Ok(())1509 }1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 #[weight = <T as Config>::WeightInfo::add_to_contract_white_list()]1523 pub fn add_to_contract_white_list(1524 origin,1525 contract_address: T::AccountId,1526 account_address: T::AccountId1527 ) -> DispatchResult {1528 let sender = ensure_signed(origin)?;15291530 #[cfg(feature = "runtime-benchmarks")]1531 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());1532 1533 Self::ensure_contract_owned(sender, &contract_address)?; 1534 <ContractWhiteList<T>>::insert(contract_address, account_address, true);1535 Ok(())1536 }15371538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 #[weight = <T as Config>::WeightInfo::remove_from_contract_white_list()]1550 pub fn remove_from_contract_white_list(1551 origin,1552 contract_address: T::AccountId,1553 account_address: T::AccountId1554 ) -> DispatchResult {1555 let sender = ensure_signed(origin)?;15561557 #[cfg(feature = "runtime-benchmarks")]1558 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());15591560 Self::ensure_contract_owned(sender, &contract_address)?;1561 <ContractWhiteList<T>>::remove(contract_address, account_address);1562 Ok(())1563 }15641565 #[weight = <T as Config>::WeightInfo::set_collection_limits()]1566 pub fn set_collection_limits(1567 origin,1568 collection_id: u32,1569 new_limits: CollectionLimits,1570 ) -> DispatchResult {1571 let sender = ensure_signed(origin)?;1572 Self::check_owner_permissions(collection_id, sender.clone())?;1573 let mut target_collection = <Collection<T>>::get(collection_id);1574 let old_limits = target_collection.limits;1575 let chain_limits = ChainLimit::get();15761577 1578 ensure!(new_limits.sponsor_transfer_timeout <= MAX_SPONSOR_TIMEOUT &&1579 new_limits.account_token_ownership_limit <= MAX_TOKEN_OWNERSHIP && 1580 new_limits.sponsored_data_size <= chain_limits.custom_data_limit,1581 Error::<T>::CollectionLimitBoundsExceeded);15821583 1584 ensure!(old_limits.token_limit >= new_limits.token_limit, Error::<T>::CollectionTokenLimitExceeded);1585 ensure!(new_limits.token_limit > 0, Error::<T>::CollectionTokenLimitExceeded);15861587 ensure!(1588 (old_limits.owner_can_transfer || !new_limits.owner_can_transfer) &&1589 (old_limits.owner_can_destroy || !new_limits.owner_can_destroy),1590 Error::<T>::OwnerPermissionsCantBeReverted,1591 );15921593 target_collection.limits = new_limits;1594 <Collection<T>>::insert(collection_id, target_collection);15951596 Ok(())1597 } 1598 }1599}16001601impl<T: Config> Module<T> {16021603 pub fn transfer_internal(sender: T::AccountId, recipient: T::AccountId, collection_id: CollectionId, item_id: TokenId, value: u128) -> DispatchResult {16041605 let target_collection = <Collection<T>>::get(collection_id);16061607 1608 Self::is_correct_transfer(collection_id, &target_collection, &recipient)?;16091610 1611 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1612 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1613 Error::<T>::NoPermission);16141615 if target_collection.access == AccessMode::WhiteList {1616 Self::check_white_list(collection_id, &sender)?;1617 Self::check_white_list(collection_id, &recipient)?;1618 }16191620 match target_collection.mode1621 {1622 CollectionMode::NFT => Self::transfer_nft(collection_id, item_id, sender.clone(), recipient.clone())?,1623 CollectionMode::Fungible(_) => Self::transfer_fungible(collection_id, value, &sender, &recipient)?,1624 CollectionMode::ReFungible => Self::transfer_refungible(collection_id, item_id, value, sender.clone(), recipient.clone())?,1625 _ => ()1626 };16271628 Self::deposit_event(RawEvent::Transfer(collection_id, item_id, sender, recipient, value));16291630 Ok(())1631 }163216331634 fn is_correct_transfer(collection_id: CollectionId, collection: &CollectionType<T::AccountId>, recipient: &T::AccountId) -> DispatchResult {16351636 1637 let account_items: u32 = <AddressTokens<T>>::get(collection_id, recipient).len() as u32;1638 ensure!(collection.limits.account_token_ownership_limit > account_items, Error::<T>::AccountTokenLimitExceeded);1639 1640 Ok(())1641 }16421643 fn can_create_items_in_collection(collection_id: CollectionId, collection: &CollectionType<T::AccountId>, sender: &T::AccountId, owner: &T::AccountId) -> DispatchResult {16441645 1646 let total_items: u32 = ItemListIndex::get(collection_id);1647 let account_items: u32 = <AddressTokens<T>>::get(collection_id, owner).len() as u32;1648 ensure!(collection.limits.token_limit > total_items, Error::<T>::CollectionTokenLimitExceeded);1649 ensure!(collection.limits.account_token_ownership_limit > account_items, Error::<T>::AccountTokenLimitExceeded);16501651 if !Self::is_owner_or_admin_permissions(collection_id, sender.clone()) {1652 ensure!(collection.mint_mode == true, Error::<T>::PublicMintingNotAllowed);1653 Self::check_white_list(collection_id, owner)?;1654 Self::check_white_list(collection_id, sender)?;1655 }16561657 Ok(())1658 }16591660 fn validate_create_item_args(target_collection: &CollectionType<T::AccountId>, data: &CreateItemData) -> DispatchResult {1661 match target_collection.mode1662 {1663 CollectionMode::NFT => {1664 if let CreateItemData::NFT(data) = data {1665 1666 ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, Error::<T>::TokenConstDataLimitExceeded);1667 ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);1668 } else {1669 fail!(Error::<T>::NotNftDataUsedToMintNftCollectionToken);1670 }1671 },1672 CollectionMode::Fungible(_) => {1673 if let CreateItemData::Fungible(_) = data {1674 } else {1675 fail!(Error::<T>::NotFungibleDataUsedToMintFungibleCollectionToken);1676 }1677 },1678 CollectionMode::ReFungible => {1679 if let CreateItemData::ReFungible(data) = data {16801681 1682 ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, Error::<T>::TokenConstDataLimitExceeded);1683 ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);16841685 1686 ensure!(data.pieces <= MAX_REFUNGIBLE_PIECES, Error::<T>::WrongRefungiblePieces);1687 ensure!(data.pieces > 0, Error::<T>::WrongRefungiblePieces);1688 } else {1689 fail!(Error::<T>::NotReFungibleDataUsedToMintReFungibleCollectionToken);1690 }1691 },1692 _ => { fail!(Error::<T>::UnexpectedCollectionType); }1693 };16941695 Ok(())1696 }16971698 fn create_item_no_validation(collection_id: CollectionId, owner: T::AccountId, data: CreateItemData) -> DispatchResult {1699 match data1700 {1701 CreateItemData::NFT(data) => {1702 let item = NftItemType {1703 owner: owner.clone(),1704 const_data: data.const_data,1705 variable_data: data.variable_data1706 };17071708 Self::add_nft_item(collection_id, item)?;1709 },1710 CreateItemData::Fungible(data) => {1711 Self::add_fungible_item(collection_id, &owner, data.value)?;1712 },1713 CreateItemData::ReFungible(data) => {1714 let mut owner_list = Vec::new();1715 owner_list.push(Ownership {owner: owner.clone(), fraction: data.pieces});17161717 let item = ReFungibleItemType {1718 owner: owner_list,1719 const_data: data.const_data,1720 variable_data: data.variable_data1721 };17221723 Self::add_refungible_item(collection_id, item)?;1724 }1725 };17261727 1728 Self::deposit_event(RawEvent::ItemCreated(collection_id, <ItemListIndex>::get(collection_id), owner));17291730 Ok(())1731 }17321733 fn add_fungible_item(collection_id: CollectionId, owner: &T::AccountId, value: u128) -> DispatchResult {17341735 1736 let mut balance: u128 = 0;1737 if <FungibleItemList<T>>::contains_key(collection_id, owner) {1738 balance = <FungibleItemList<T>>::get(collection_id, owner).value;1739 } 17401741 1742 let item = FungibleItemType {1743 value: balance + value1744 };1745 <FungibleItemList<T>>::insert(collection_id, (*owner).clone(), item);17461747 1748 let new_balance = <Balance<T>>::get(collection_id, owner)1749 .checked_add(value)1750 .ok_or(Error::<T>::NumOverflow)?;1751 <Balance<T>>::insert(collection_id, (*owner).clone(), new_balance);17521753 Ok(())1754 }17551756 fn add_refungible_item(collection_id: CollectionId, item: ReFungibleItemType<T::AccountId>) -> DispatchResult {1757 let current_index = <ItemListIndex>::get(collection_id)1758 .checked_add(1)1759 .ok_or(Error::<T>::NumOverflow)?;1760 let itemcopy = item.clone();17611762 let value = item.owner.first().unwrap().fraction;1763 let owner = item.owner.first().unwrap().owner.clone();17641765 Self::add_token_index(collection_id, current_index, &owner)?;17661767 <ItemListIndex>::insert(collection_id, current_index);1768 <ReFungibleItemList<T>>::insert(collection_id, current_index, itemcopy);17691770 1771 let new_balance = <Balance<T>>::get(collection_id, &owner)1772 .checked_add(value)1773 .ok_or(Error::<T>::NumOverflow)?;1774 <Balance<T>>::insert(collection_id, owner.clone(), new_balance);17751776 Ok(())1777 }17781779 fn add_nft_item(collection_id: CollectionId, item: NftItemType<T::AccountId>) -> DispatchResult {1780 let current_index = <ItemListIndex>::get(collection_id)1781 .checked_add(1)1782 .ok_or(Error::<T>::NumOverflow)?;17831784 let item_owner = item.owner.clone();1785 Self::add_token_index(collection_id, current_index, &item.owner)?;17861787 <ItemListIndex>::insert(collection_id, current_index);1788 <NftItemList<T>>::insert(collection_id, current_index, item);17891790 1791 let new_balance = <Balance<T>>::get(collection_id, item_owner.clone())1792 .checked_add(1)1793 .ok_or(Error::<T>::NumOverflow)?;1794 <Balance<T>>::insert(collection_id, item_owner.clone(), new_balance);17951796 Ok(())1797 }17981799 fn burn_refungible_item(1800 collection_id: CollectionId,1801 item_id: TokenId,1802 owner: &T::AccountId,1803 ) -> DispatchResult {1804 ensure!(1805 <ReFungibleItemList<T>>::contains_key(collection_id, item_id),1806 Error::<T>::TokenNotFound1807 );1808 let mut token = <ReFungibleItemList<T>>::get(collection_id, item_id);1809 let rft_balance = token1810 .owner1811 .iter()1812 .filter(|&i| i.owner == *owner)1813 .next()1814 .unwrap();1815 Self::remove_token_index(collection_id, item_id, owner)?;18161817 1818 let new_balance = <Balance<T>>::get(collection_id, rft_balance.owner.clone())1819 .checked_sub(rft_balance.fraction)1820 .ok_or(Error::<T>::NumOverflow)?;1821 <Balance<T>>::insert(collection_id, rft_balance.owner.clone(), new_balance);18221823 1824 let index = token1825 .owner1826 .iter()1827 .position(|i| i.owner == *owner)1828 .unwrap();1829 token.owner.remove(index);1830 let owner_count = token.owner.len();18311832 1833 if owner_count == 0 {1834 <ReFungibleItemList<T>>::remove(collection_id, item_id);1835 }1836 else {1837 <ReFungibleItemList<T>>::insert(collection_id, item_id, token);1838 }18391840 Ok(())1841 }18421843 fn burn_nft_item(collection_id: CollectionId, item_id: TokenId) -> DispatchResult {1844 ensure!(1845 <NftItemList<T>>::contains_key(collection_id, item_id),1846 Error::<T>::TokenNotFound1847 );1848 let item = <NftItemList<T>>::get(collection_id, item_id);1849 Self::remove_token_index(collection_id, item_id, &item.owner)?;18501851 1852 let new_balance = <Balance<T>>::get(collection_id, &item.owner)1853 .checked_sub(1)1854 .ok_or(Error::<T>::NumOverflow)?;1855 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);1856 <NftItemList<T>>::remove(collection_id, item_id);18571858 Ok(())1859 }18601861 fn burn_fungible_item(owner: &T::AccountId, collection_id: CollectionId, value: u128) -> DispatchResult {1862 ensure!(1863 <FungibleItemList<T>>::contains_key(collection_id, owner),1864 Error::<T>::TokenNotFound1865 );1866 let mut balance = <FungibleItemList<T>>::get(collection_id, owner);1867 ensure!(balance.value >= value, Error::<T>::TokenValueNotEnough);18681869 1870 let new_balance = <Balance<T>>::get(collection_id, owner)1871 .checked_sub(value)1872 .ok_or(Error::<T>::NumOverflow)?;1873 <Balance<T>>::insert(collection_id, (*owner).clone(), new_balance);18741875 if balance.value - value > 0 {1876 balance.value -= value;1877 <FungibleItemList<T>>::insert(collection_id, (*owner).clone(), balance);1878 }1879 else {1880 <FungibleItemList<T>>::remove(collection_id, owner);1881 }18821883 Ok(())1884 }18851886 fn collection_exists(collection_id: CollectionId) -> DispatchResult {1887 ensure!(1888 <Collection<T>>::contains_key(collection_id),1889 Error::<T>::CollectionNotFound1890 );1891 Ok(())1892 }18931894 fn check_owner_permissions(collection_id: CollectionId, subject: T::AccountId) -> DispatchResult {1895 Self::collection_exists(collection_id)?;18961897 let target_collection = <Collection<T>>::get(collection_id);1898 ensure!(1899 subject == target_collection.owner,1900 Error::<T>::NoPermission1901 );19021903 Ok(())1904 }19051906 fn is_owner_or_admin_permissions(collection_id: CollectionId, subject: T::AccountId) -> bool {1907 let target_collection = <Collection<T>>::get(collection_id);1908 let mut result: bool = subject == target_collection.owner;1909 let exists = <AdminList<T>>::contains_key(collection_id);19101911 if !result & exists {1912 if <AdminList<T>>::get(collection_id).contains(&subject) {1913 result = true1914 }1915 }19161917 result1918 }19191920 fn check_owner_or_admin_permissions(1921 collection_id: CollectionId,1922 subject: T::AccountId,1923 ) -> DispatchResult {1924 Self::collection_exists(collection_id)?;1925 let result = Self::is_owner_or_admin_permissions(collection_id, subject.clone());19261927 ensure!(1928 result,1929 Error::<T>::NoPermission1930 );1931 Ok(())1932 }19331934 fn owned_amount(1935 subject: T::AccountId,1936 collection_id: CollectionId,1937 item_id: TokenId,1938 ) -> Option<u128> {1939 let target_collection = <Collection<T>>::get(collection_id);19401941 match target_collection.mode {1942 CollectionMode::NFT => {1943 if <NftItemList<T>>::get(collection_id, item_id).owner == subject {1944 return Some(1)1945 }1946 None1947 },1948 CollectionMode::Fungible(_) => {1949 if <FungibleItemList<T>>::contains_key(collection_id, &subject) {1950 return Some(<FungibleItemList<T>>::get(collection_id, &subject)1951 .value);1952 }1953 None1954 },1955 CollectionMode::ReFungible => <ReFungibleItemList<T>>::get(collection_id, item_id)1956 .owner1957 .iter()1958 .find(|i| i.owner == subject)1959 .map(|i| i.fraction),1960 CollectionMode::Invalid => None,1961 }1962 }19631964 fn is_item_owner(subject: T::AccountId, collection_id: CollectionId, item_id: TokenId) -> bool {1965 let target_collection = <Collection<T>>::get(collection_id);19661967 match target_collection.mode {1968 CollectionMode::NFT => {1969 <NftItemList<T>>::get(collection_id, item_id).owner == subject1970 }1971 CollectionMode::Fungible(_) => {1972 <FungibleItemList<T>>::contains_key(collection_id, &subject)1973 }1974 CollectionMode::ReFungible => {1975 <ReFungibleItemList<T>>::get(collection_id, item_id)1976 .owner1977 .iter()1978 .any(|i| i.owner == subject)1979 }1980 CollectionMode::Invalid => false,1981 }1982 }19831984 fn check_white_list(collection_id: CollectionId, address: &T::AccountId) -> DispatchResult {1985 let mes = Error::<T>::AddresNotInWhiteList;1986 ensure!(<WhiteList<T>>::contains_key(collection_id, address), mes);19871988 Ok(())1989 }19901991 1992 1993 fn token_exists(1994 collection_id: CollectionId,1995 item_id: TokenId,1996 owner: &T::AccountId1997 ) -> DispatchResult {1998 let target_collection = <Collection<T>>::get(collection_id);1999 let exists = match target_collection.mode2000 {2001 CollectionMode::NFT => <NftItemList<T>>::contains_key(collection_id, item_id),2002 CollectionMode::Fungible(_) => <FungibleItemList<T>>::contains_key(collection_id, owner),2003 CollectionMode::ReFungible => <ReFungibleItemList<T>>::contains_key(collection_id, item_id),2004 _ => false2005 };20062007 ensure!(exists == true, Error::<T>::TokenNotFound);2008 Ok(())2009 }20102011 fn transfer_fungible(2012 collection_id: CollectionId,2013 value: u128,2014 owner: &T::AccountId,2015 recipient: &T::AccountId,2016 ) -> DispatchResult {2017 Self::token_exists(collection_id, 0, owner)?;20182019 let mut balance = <FungibleItemList<T>>::get(collection_id, owner);2020 ensure!(balance.value >= value, Error::<T>::TokenValueTooLow);20212022 2023 Self::add_fungible_item(collection_id, recipient, value)?;20242025 2026 <Balance<T>>::insert(collection_id, (*owner).clone(), balance.value - value);20272028 2029 if balance.value == value {2030 <FungibleItemList<T>>::remove(collection_id, owner);2031 }2032 else {2033 balance.value -= value;2034 <FungibleItemList<T>>::insert(collection_id, (*owner).clone(), balance);2035 }20362037 Ok(())2038 }20392040 fn transfer_refungible(2041 collection_id: CollectionId,2042 item_id: TokenId,2043 value: u128,2044 owner: T::AccountId,2045 new_owner: T::AccountId,2046 ) -> DispatchResult {2047 Self::token_exists(collection_id, item_id, &owner)?;20482049 let full_item = <ReFungibleItemList<T>>::get(collection_id, item_id);2050 let item = full_item2051 .owner2052 .iter()2053 .filter(|i| i.owner == owner)2054 .next()2055 .ok_or(Error::<T>::NumOverflow)?;2056 let amount = item.fraction;20572058 ensure!(amount >= value, Error::<T>::TokenValueTooLow);20592060 2061 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())2062 .checked_sub(value)2063 .ok_or(Error::<T>::NumOverflow)?;2064 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);20652066 let balancenew_owner = <Balance<T>>::get(collection_id, new_owner.clone())2067 .checked_add(value)2068 .ok_or(Error::<T>::NumOverflow)?;2069 <Balance<T>>::insert(collection_id, new_owner.clone(), balancenew_owner);20702071 let old_owner = item.owner.clone();2072 let new_owner_has_account = full_item.owner.iter().any(|i| i.owner == new_owner);20732074 2075 if amount == value && !new_owner_has_account {2076 2077 2078 let mut new_full_item = full_item.clone();2079 new_full_item2080 .owner2081 .iter_mut()2082 .find(|i| i.owner == owner)2083 .unwrap()2084 .owner = new_owner.clone();2085 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);20862087 2088 Self::move_token_index(collection_id, item_id, &old_owner, &new_owner)?;2089 } else {2090 let mut new_full_item = full_item.clone();2091 new_full_item2092 .owner2093 .iter_mut()2094 .find(|i| i.owner == owner)2095 .unwrap()2096 .fraction -= value;20972098 2099 if new_owner_has_account {2100 2101 new_full_item2102 .owner2103 .iter_mut()2104 .find(|i| i.owner == new_owner)2105 .unwrap()2106 .fraction += value;2107 } else {2108 2109 new_full_item.owner.push(Ownership {2110 owner: new_owner.clone(),2111 fraction: value,2112 });2113 Self::add_token_index(collection_id, item_id, &new_owner)?;2114 }21152116 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);2117 }21182119 Ok(())2120 }21212122 fn transfer_nft(2123 collection_id: CollectionId,2124 item_id: TokenId,2125 sender: T::AccountId,2126 new_owner: T::AccountId,2127 ) -> DispatchResult {2128 Self::token_exists(collection_id, item_id, &sender)?;21292130 let mut item = <NftItemList<T>>::get(collection_id, item_id);21312132 ensure!(2133 sender == item.owner,2134 Error::<T>::MustBeTokenOwner2135 );21362137 2138 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())2139 .checked_sub(1)2140 .ok_or(Error::<T>::NumOverflow)?;2141 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);21422143 let balancenew_owner = <Balance<T>>::get(collection_id, new_owner.clone())2144 .checked_add(1)2145 .ok_or(Error::<T>::NumOverflow)?;2146 <Balance<T>>::insert(collection_id, new_owner.clone(), balancenew_owner);21472148 2149 let old_owner = item.owner.clone();2150 item.owner = new_owner.clone();2151 <NftItemList<T>>::insert(collection_id, item_id, item);21522153 2154 Self::move_token_index(collection_id, item_id, &old_owner, &new_owner)?;21552156 Ok(())2157 }2158 2159 fn set_re_fungible_variable_data(2160 collection_id: CollectionId,2161 item_id: TokenId,2162 data: Vec<u8>2163 ) -> DispatchResult {2164 let mut item = <ReFungibleItemList<T>>::get(collection_id, item_id);21652166 item.variable_data = data;21672168 <ReFungibleItemList<T>>::insert(collection_id, item_id, item);21692170 Ok(())2171 }21722173 fn set_nft_variable_data(2174 collection_id: CollectionId,2175 item_id: TokenId,2176 data: Vec<u8>2177 ) -> DispatchResult {2178 let mut item = <NftItemList<T>>::get(collection_id, item_id);2179 2180 item.variable_data = data;21812182 <NftItemList<T>>::insert(collection_id, item_id, item);2183 2184 Ok(())2185 }21862187 fn init_collection(item: &CollectionType<T::AccountId>) {2188 2189 assert!(2190 item.decimal_points <= MAX_DECIMAL_POINTS,2191 "decimal_points parameter must be lower than MAX_DECIMAL_POINTS"2192 );2193 assert!(2194 item.name.len() <= 64,2195 "Collection name can not be longer than 63 char"2196 );2197 assert!(2198 item.name.len() <= 256,2199 "Collection description can not be longer than 255 char"2200 );2201 assert!(2202 item.token_prefix.len() <= 16,2203 "Token prefix can not be longer than 15 char"2204 );22052206 2207 let next_id = CreatedCollectionCount::get()2208 .checked_add(1)2209 .unwrap();22102211 CreatedCollectionCount::put(next_id);2212 }22132214 fn init_nft_token(collection_id: CollectionId, item: &NftItemType<T::AccountId>) {2215 let current_index = <ItemListIndex>::get(collection_id)2216 .checked_add(1)2217 .unwrap();22182219 let item_owner = item.owner.clone();2220 Self::add_token_index(collection_id, current_index, &item.owner).unwrap();22212222 <ItemListIndex>::insert(collection_id, current_index);22232224 2225 let new_balance = <Balance<T>>::get(collection_id, &item_owner)2226 .checked_add(1)2227 .unwrap();2228 <Balance<T>>::insert(collection_id, item_owner.clone(), new_balance);2229 }22302231 fn init_fungible_token(collection_id: CollectionId, owner: &T::AccountId, item: &FungibleItemType) {2232 let current_index = <ItemListIndex>::get(collection_id)2233 .checked_add(1)2234 .unwrap();22352236 Self::add_token_index(collection_id, current_index, owner).unwrap();22372238 <ItemListIndex>::insert(collection_id, current_index);22392240 2241 let new_balance = <Balance<T>>::get(collection_id, owner)2242 .checked_add(item.value)2243 .unwrap();2244 <Balance<T>>::insert(collection_id, (*owner).clone(), new_balance);2245 }22462247 fn init_refungible_token(collection_id: CollectionId, item: &ReFungibleItemType<T::AccountId>) {2248 let current_index = <ItemListIndex>::get(collection_id)2249 .checked_add(1)2250 .unwrap();22512252 let value = item.owner.first().unwrap().fraction;2253 let owner = item.owner.first().unwrap().owner.clone();22542255 Self::add_token_index(collection_id, current_index, &owner).unwrap();22562257 <ItemListIndex>::insert(collection_id, current_index);22582259 2260 let new_balance = <Balance<T>>::get(collection_id, &owner)2261 .checked_add(value)2262 .unwrap();2263 <Balance<T>>::insert(collection_id, owner.clone(), new_balance);2264 }22652266 fn add_token_index(collection_id: CollectionId, item_index: TokenId, owner: &T::AccountId) -> DispatchResult {22672268 2269 if <AccountItemCount<T>>::contains_key(owner) {22702271 2272 let count = <AccountItemCount<T>>::get(owner);2273 ensure!(count < ChainLimit::get().account_token_ownership_limit, Error::<T>::AddressOwnershipLimitExceeded);22742275 <AccountItemCount<T>>::insert(owner.clone(), count2276 .checked_add(1)2277 .ok_or(Error::<T>::NumOverflow)?);2278 }2279 else {2280 <AccountItemCount<T>>::insert(owner.clone(), 1);2281 }22822283 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner);2284 if list_exists {2285 let mut list = <AddressTokens<T>>::get(collection_id, owner);2286 let item_contains = list.contains(&item_index.clone());22872288 if !item_contains {2289 list.push(item_index.clone());2290 }22912292 <AddressTokens<T>>::insert(collection_id, owner.clone(), list);2293 } else {2294 let mut itm = Vec::new();2295 itm.push(item_index.clone());2296 <AddressTokens<T>>::insert(collection_id, owner.clone(), itm);2297 }22982299 Ok(())2300 }23012302 fn remove_token_index(2303 collection_id: CollectionId,2304 item_index: TokenId,2305 owner: &T::AccountId,2306 ) -> DispatchResult {23072308 2309 <AccountItemCount<T>>::insert(owner.clone(), 2310 <AccountItemCount<T>>::get(owner)2311 .checked_sub(1)2312 .ok_or(Error::<T>::NumOverflow)?);231323142315 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner);2316 if list_exists {2317 let mut list = <AddressTokens<T>>::get(collection_id, owner);2318 let item_contains = list.contains(&item_index.clone());23192320 if item_contains {2321 list.retain(|&item| item != item_index);2322 <AddressTokens<T>>::insert(collection_id, owner.clone(), list);2323 }2324 }23252326 Ok(())2327 }23282329 fn move_token_index(2330 collection_id: CollectionId,2331 item_index: TokenId,2332 old_owner: &T::AccountId,2333 new_owner: &T::AccountId,2334 ) -> DispatchResult {2335 Self::remove_token_index(collection_id, item_index, old_owner)?;2336 Self::add_token_index(collection_id, item_index, new_owner)?;23372338 Ok(())2339 }2340 2341 fn ensure_contract_owned(account: T::AccountId, contract: &T::AccountId) -> DispatchResult {2342 if <ContractOwner<T>>::contains_key(contract.clone()) {2343 let owner = <ContractOwner<T>>::get(contract);2344 ensure!(account == owner, Error::<T>::NoPermission);2345 } else {2346 fail!(Error::<T>::NoPermission);2347 }23482349 Ok(())2350 }2351}2352235323542355235623572358pub type Multiplier = FixedU128;23592360type BalanceOf<T> = <<T as pallet_transaction_payment::Config>::OnChargeTransaction as pallet_transaction_payment::OnChargeTransaction<T>>::Balance;2361236223632364#[derive(Encode, Decode, Clone, Eq, PartialEq)]2365pub struct ChargeTransactionPayment<T: Config>(#[codec(compact)] BalanceOf<T>);23662367impl<T: Config + Send + Sync> sp_std::fmt::Debug 2368 for ChargeTransactionPayment<T>2369{2370 #[cfg(feature = "std")]2371 fn fmt(&self, f: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {2372 write!(f, "ChargeTransactionPayment<{:?}>", self.0)2373 }2374 #[cfg(not(feature = "std"))]2375 fn fmt(&self, _: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {2376 Ok(())2377 }2378}23792380impl<T: Config> ChargeTransactionPayment<T>2381where2382 T::Call: Dispatchable<Info=DispatchInfo, PostInfo=PostDispatchInfo> + IsSubType<Call<T>> + IsSubType<pallet_contracts::Call<T>>,2383 BalanceOf<T>: Send + Sync + From<u64> + FixedPointOperand,2384 T::AccountId: AsRef<[u8]>,2385 T::AccountId: UncheckedFrom<T::Hash>,2386{2387 fn traditional_fee(2388 len: usize,2389 info: &DispatchInfoOf<T::Call>,2390 tip: BalanceOf<T>,2391 ) -> BalanceOf<T>2392 where2393 T::Call: Dispatchable<Info = DispatchInfo>,2394 {2395 <pallet_transaction_payment::Module<T>>::compute_fee(len as u32, info, tip)2396 }23972398 fn get_priority(len: usize, info: &DispatchInfoOf<T::Call>, final_fee: BalanceOf<T>) -> TransactionPriority {2399 let weight_saturation = T::BlockWeights::get().max_block / info.weight.max(1);2400 let max_block_length = *T::BlockLength::get().max.get(DispatchClass::Normal);2401 let len_saturation = max_block_length as u64 / (len as u64).max(1);2402 let coefficient: BalanceOf<T> = weight_saturation2403 .min(len_saturation)2404 .saturated_into::<BalanceOf<T>>();2405 final_fee2406 .saturating_mul(coefficient)2407 .saturated_into::<TransactionPriority>()2408 }24092410 fn withdraw_fee(2411 &self,2412 who: &T::AccountId,2413 call: &T::Call,2414 info: &DispatchInfoOf<T::Call>,2415 len: usize,2416 ) -> Result<2417 (2418 BalanceOf<T>,2419 <<T as pallet_transaction_payment::Config>::OnChargeTransaction as pallet_transaction_payment::OnChargeTransaction<T>>::LiquidityInfo,2420 ),2421 TransactionValidityError,2422 > {2423 let tip = self.0;24242425 2426 2427 2428 2429 2430 2431 2432 let fee = Self::traditional_fee(len, info, tip);24332434 2435 if fee.is_zero() {2436 return <<T as pallet_transaction_payment::Config>::OnChargeTransaction as pallet_transaction_payment::OnChargeTransaction<T>>::withdraw_fee(who, call, info, fee, tip)2437 .map(|i| (fee, i));2438 }24392440 2441 2442 let mut sponsor: T::AccountId = match IsSubType::<Call<T>>::is_sub_type(call) {2443 Some(Call::create_item(collection_id, _owner, _properties)) => {24442445 2446 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;24472448 let limit = <Collection<T>>::get(collection_id).limits.sponsor_transfer_timeout;2449 let mut sponsored = true;2450 if <CreateItemBasket<T>>::contains_key((collection_id, &who)) {2451 let last_tx_block = <CreateItemBasket<T>>::get((collection_id, &who));2452 let limit_time = last_tx_block + limit.into();2453 if block_number <= limit_time {2454 sponsored = false;2455 }2456 }2457 if sponsored {2458 <CreateItemBasket<T>>::insert((collection_id, who.clone()), block_number);2459 }24602461 2462 if (<Collection<T>>::get(collection_id).limits.sponsored_data_size >= (_properties.len() as u32)) &&2463 (<Collection<T>>::get(collection_id).sponsor_confirmed) &&2464 (sponsored)2465 {2466 <Collection<T>>::get(collection_id).sponsor2467 } else {2468 T::AccountId::default()2469 }2470 }2471 Some(Call::transfer(_new_owner, collection_id, item_id, _value)) => {2472 2473 let mut sponsor_transfer = false;2474 if <Collection<T>>::get(collection_id).sponsor_confirmed {24752476 let collection_limits = <Collection<T>>::get(collection_id).limits;2477 let collection_mode = <Collection<T>>::get(collection_id).mode;2478 2479 2480 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2481 sponsor_transfer = match collection_mode {2482 CollectionMode::NFT => {2483 2484 2485 let limit: u32 = if collection_limits.sponsor_transfer_timeout > 0 {2486 collection_limits.sponsor_transfer_timeout2487 } else {2488 ChainLimit::get().nft_sponsor_transfer_timeout2489 };2490 2491 let mut sponsored = true;2492 if <NftTransferBasket<T>>::contains_key(collection_id, item_id) {2493 let last_tx_block = <NftTransferBasket<T>>::get(collection_id, item_id);2494 let limit_time = last_tx_block + limit.into();2495 if block_number <= limit_time {2496 sponsored = false;2497 }2498 }2499 if sponsored {2500 <NftTransferBasket<T>>::insert(collection_id, item_id, block_number);2501 }25022503 sponsored2504 }2505 CollectionMode::Fungible(_) => {2506 2507 2508 let limit: u32 = if collection_limits.sponsor_transfer_timeout > 0 {2509 collection_limits.sponsor_transfer_timeout2510 } else {2511 ChainLimit::get().fungible_sponsor_transfer_timeout2512 };2513 2514 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2515 let mut sponsored = true;2516 if <FungibleTransferBasket<T>>::contains_key(collection_id, who) {2517 let last_tx_block = <FungibleTransferBasket<T>>::get(collection_id, who);2518 let limit_time = last_tx_block + limit.into();2519 if block_number <= limit_time {2520 sponsored = false;2521 }2522 }2523 if sponsored {2524 <FungibleTransferBasket<T>>::insert(collection_id, who, block_number);2525 }25262527 sponsored2528 }2529 CollectionMode::ReFungible => {2530 2531 2532 let limit: u32 = if collection_limits.sponsor_transfer_timeout > 0 {2533 collection_limits.sponsor_transfer_timeout2534 } else {2535 ChainLimit::get().refungible_sponsor_transfer_timeout2536 };2537 2538 let mut sponsored = true;2539 if <ReFungibleTransferBasket<T>>::contains_key(collection_id, item_id) {2540 let last_tx_block = <ReFungibleTransferBasket<T>>::get(collection_id, item_id);2541 let limit_time = last_tx_block + limit.into();2542 if block_number <= limit_time {2543 sponsored = false;2544 }2545 }2546 if sponsored {2547 <ReFungibleTransferBasket<T>>::insert(collection_id, item_id, block_number);2548 }25492550 sponsored2551 }2552 _ => {2553 false2554 },2555 };2556 }25572558 if !sponsor_transfer {2559 T::AccountId::default()2560 } else {2561 <Collection<T>>::get(collection_id).sponsor2562 }2563 }25642565 _ => T::AccountId::default(),2566 };25672568 2569 sponsor = match IsSubType::<pallet_contracts::Call<T>>::is_sub_type(call) {25702571 2572 Some(pallet_contracts::Call::instantiate(_endowment, _gas_limit, code_hash, _data, salt)) => {25732574 let new_contract_address = <pallet_contracts::Module<T>>::contract_address(2575 &who,2576 code_hash,2577 salt,2578 );2579 <ContractOwner<T>>::insert(new_contract_address.clone(), who.clone());25802581 T::AccountId::default()2582 },25832584 2585 Some(pallet_contracts::Call::call(dest, _value, _gas_limit, _data)) => {25862587 let called_contract: T::AccountId = T::Lookup::lookup((*dest).clone()).unwrap_or(T::AccountId::default());25882589 let owned_contract = <ContractOwner<T>>::contains_key(called_contract.clone())2590 && <ContractOwner<T>>::get(called_contract.clone()) == *who;2591 let white_list_enabled = <ContractWhiteListEnabled<T>>::contains_key(called_contract.clone()) && <ContractWhiteListEnabled<T>>::get(called_contract.clone());2592 2593 if !owned_contract && white_list_enabled {2594 if !<ContractWhiteList<T>>::contains_key(called_contract.clone(), who) {2595 return Err(InvalidTransaction::Call.into());2596 }2597 }25982599 let mut sponsor_transfer = false;2600 if <ContractSponsoringRateLimit<T>>::contains_key(called_contract.clone()) {2601 let last_tx_block = <ContractSponsorBasket<T>>::get((&called_contract, &who));2602 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2603 let rate_limit = <ContractSponsoringRateLimit<T>>::get(&called_contract);2604 let limit_time = last_tx_block + rate_limit;26052606 if block_number >= limit_time {2607 <ContractSponsorBasket<T>>::insert((called_contract.clone(), who.clone()), block_number);2608 sponsor_transfer = true;2609 }2610 } else {2611 sponsor_transfer = false;2612 }2613 2614 2615 let mut sp = T::AccountId::default();2616 if sponsor_transfer {2617 if <ContractSelfSponsoring<T>>::contains_key(called_contract.clone()) {2618 if <ContractSelfSponsoring<T>>::get(called_contract.clone()) {2619 sp = called_contract;2620 }2621 }2622 }26232624 sp2625 },26262627 _ => sponsor,2628 };26292630 let mut who_pays_fee: T::AccountId = sponsor.clone();2631 if sponsor == T::AccountId::default() {2632 who_pays_fee = who.clone();2633 }26342635 <<T as pallet_transaction_payment::Config>::OnChargeTransaction as pallet_transaction_payment::OnChargeTransaction<T>>::withdraw_fee(&who_pays_fee, call, info, fee, tip)2636 .map(|i| (fee, i))2637 }2638}263926402641impl<T: Config + Send + Sync> SignedExtension2642 for ChargeTransactionPayment<T>2643where2644 BalanceOf<T>: Send + Sync + From<u64> + FixedPointOperand,2645 T::Call: Dispatchable<Info = DispatchInfo, PostInfo = PostDispatchInfo> + IsSubType<Call<T>> + IsSubType<pallet_contracts::Call<T>>,2646 T::AccountId: AsRef<[u8]>,2647 T::AccountId: UncheckedFrom<T::Hash>,2648{2649 const IDENTIFIER: &'static str = "ChargeTransactionPayment";2650 type AccountId = T::AccountId;2651 type Call = T::Call;2652 type AdditionalSigned = ();2653 type Pre = (2654 2655 BalanceOf<T>,2656 2657 Self::AccountId,2658 2659 <<T as pallet_transaction_payment::Config>::OnChargeTransaction as pallet_transaction_payment::OnChargeTransaction<T>>::LiquidityInfo,2660 );2661 fn additional_signed(&self) -> sp_std::result::Result<(), TransactionValidityError> {2662 Ok(())2663 }26642665 fn validate(2666 &self,2667 who: &Self::AccountId,2668 call: &Self::Call,2669 info: &DispatchInfoOf<Self::Call>,2670 len: usize,2671 ) -> TransactionValidity {2672 let (fee, _) = self.withdraw_fee(who, call, info, len)?;2673 Ok(ValidTransaction {2674 priority: Self::get_priority(len, info, fee),2675 ..Default::default()2676 })2677 }26782679 fn pre_dispatch(2680 self,2681 who: &Self::AccountId,2682 call: &Self::Call,2683 info: &DispatchInfoOf<Self::Call>,2684 len: usize,2685 ) -> Result<Self::Pre, TransactionValidityError> {2686 let (_fee, imbalance) = self.withdraw_fee(who, call, info, len)?;2687 Ok((self.0, who.clone(), imbalance))2688 }26892690 fn post_dispatch(2691 pre: Self::Pre,2692 info: &DispatchInfoOf<Self::Call>,2693 post_info: &PostDispatchInfoOf<Self::Call>,2694 len: usize,2695 _result: &DispatchResult,2696 ) -> Result<(), TransactionValidityError> {2697 let (tip, who, imbalance) = pre;2698 let actual_fee = pallet_transaction_payment::Module::<T>::compute_actual_fee(2699 len as u32,2700 info,2701 post_info,2702 tip,2703 );2704 <T as pallet_transaction_payment::Config>::OnChargeTransaction::correct_and_deposit_fee(&who, info, post_info, actual_fee, tip, imbalance)?;2705 Ok(())2706 }2707}27082709