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, WithdrawReason,24 },25 weights::{26 constants::{BlockExecutionWeight, ExtrinsicBaseWeight, RocksDbWeight, WEIGHT_PER_SECOND},27 DispatchInfo, GetDispatchInfo, IdentityFee, Pays, PostDispatchInfo, Weight,28 WeightToFeePolynomial,29 },30 IsSubType, 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 pallet_contracts::ContractAddressFor;45use sp_runtime::traits::StaticLookup;4647#[cfg(test)]48mod mock;4950#[cfg(test)]51mod tests;5253mod default_weights;5455pub const MAX_DECIMAL_POINTS: DecimalPoints = 30;56pub const MAX_SPONSOR_TIMEOUT: u32 = 10_368_000;57pub const MAX_TOKEN_OWNERSHIP: u32 = 10_000_000;5859606162pub type CollectionId = u32;63pub type TokenId = u32;64pub type DecimalPoints = u8;6566#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq)]67#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]68pub enum CollectionMode {69 Invalid,70 NFT,71 72 Fungible(DecimalPoints),73 74 ReFungible(DecimalPoints),75}7677impl Default for CollectionMode {78 fn default() -> Self {79 Self::Invalid80 }81}8283impl Into<u8> for CollectionMode {84 fn into(self) -> u8 {85 match self {86 CollectionMode::Invalid => 0,87 CollectionMode::NFT => 1,88 CollectionMode::Fungible(_) => 2,89 CollectionMode::ReFungible(_) => 3,90 }91 }92}9394#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq)]95#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]96pub enum AccessMode {97 Normal,98 WhiteList,99}100impl Default for AccessMode {101 fn default() -> Self {102 Self::Normal103 }104}105106#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq)]107#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]108pub enum SchemaVersion {109 ImageURL,110 Unique,111}112impl Default for SchemaVersion {113 fn default() -> Self {114 Self::ImageURL115 }116}117118#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]119#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]120pub struct Ownership<AccountId> {121 pub owner: AccountId,122 pub fraction: u128,123}124125#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]126#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]127pub struct CollectionType<AccountId> {128 pub owner: AccountId,129 pub mode: CollectionMode,130 pub access: AccessMode,131 pub decimal_points: DecimalPoints,132 pub name: Vec<u16>, 133 pub description: Vec<u16>, 134 pub token_prefix: Vec<u8>, 135 pub mint_mode: bool,136 pub offchain_schema: Vec<u8>,137 pub schema_version: SchemaVersion,138 pub sponsor: AccountId, 139 pub sponsor_confirmed: bool, 140 pub limits: CollectionLimits, 141 pub variable_on_chain_schema: Vec<u8>, 142 pub const_on_chain_schema: Vec<u8>, 143}144145#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]146#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]147pub struct NftItemType<AccountId> {148 pub owner: AccountId,149 pub const_data: Vec<u8>,150 pub variable_data: Vec<u8>,151}152153#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]154#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]155pub struct FungibleItemType {156 pub value: u128,157}158159#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]160#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]161pub struct ReFungibleItemType<AccountId> {162 pub owner: Vec<Ownership<AccountId>>,163 pub const_data: Vec<u8>,164 pub variable_data: Vec<u8>,165}166167168169170171172173174175176177178#[derive(Encode, Decode, Debug, Clone, PartialEq)]179#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]180pub struct CollectionLimits {181 pub account_token_ownership_limit: u32,182 pub sponsored_data_size: u32,183 pub token_limit: u32,184185 186 pub sponsor_transfer_timeout: u32,187}188189impl Default for CollectionLimits {190 fn default() -> CollectionLimits {191 CollectionLimits { 192 account_token_ownership_limit: 10_000_000, 193 token_limit: u32::max_value(),194 sponsored_data_size: u32::max_value(), 195 sponsor_transfer_timeout: 14400 }196 }197}198199#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]200#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]201pub struct ChainLimits {202 pub collection_numbers_limit: u32,203 pub account_token_ownership_limit: u32,204 pub collections_admins_limit: u64,205 pub custom_data_limit: u32,206207 208 pub nft_sponsor_transfer_timeout: u32,209 pub fungible_sponsor_transfer_timeout: u32,210 pub refungible_sponsor_transfer_timeout: u32,211212 213 pub offchain_schema_limit: u32,214 pub variable_on_chain_schema_limit: u32,215 pub const_on_chain_schema_limit: u32,216}217218pub trait WeightInfo {219 fn create_collection() -> Weight;220 fn destroy_collection() -> Weight;221 fn add_to_white_list() -> Weight;222 fn remove_from_white_list() -> Weight;223 fn set_public_access_mode() -> Weight;224 fn set_mint_permission() -> Weight;225 fn change_collection_owner() -> Weight;226 fn add_collection_admin() -> Weight;227 fn remove_collection_admin() -> Weight;228 fn set_collection_sponsor() -> Weight;229 fn confirm_sponsorship() -> Weight;230 fn remove_collection_sponsor() -> Weight;231 fn create_item(s: usize) -> Weight;232 fn burn_item() -> Weight;233 fn transfer() -> Weight;234 fn approve() -> Weight;235 fn transfer_from() -> Weight;236 fn set_offchain_schema() -> Weight;237 fn set_const_on_chain_schema() -> Weight;238 fn set_variable_on_chain_schema() -> Weight;239 fn set_variable_meta_data() -> Weight;240 fn enable_contract_sponsoring() -> Weight;241 fn set_schema_version() -> Weight;242 fn set_chain_limits() -> Weight;243 fn set_contract_sponsoring_rate_limit() -> Weight;244 fn toggle_contract_white_list() -> Weight;245 fn add_to_contract_white_list() -> Weight;246 fn remove_from_contract_white_list() -> Weight;247 fn set_collection_limits() -> Weight;248}249250#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]251#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]252pub struct CreateNftData {253 pub const_data: Vec<u8>,254 pub variable_data: Vec<u8>,255}256257#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]258#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]259pub struct CreateFungibleData {260 pub value: u128,261}262263#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]264#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]265pub struct CreateReFungibleData {266 pub const_data: Vec<u8>,267 pub variable_data: Vec<u8>,268}269270#[derive(Encode, Decode, Debug, Clone, PartialEq)]271#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]272pub enum CreateItemData {273 NFT(CreateNftData),274 Fungible(CreateFungibleData),275 ReFungible(CreateReFungibleData),276}277278impl CreateItemData {279 pub fn len(&self) -> usize {280 let len = match self {281 CreateItemData::NFT(data) => data.variable_data.len() + data.const_data.len(),282 CreateItemData::ReFungible(data) => data.variable_data.len() + data.const_data.len(),283 _ => 0284 };285 286 return len;287 }288}289290impl From<CreateNftData> for CreateItemData {291 fn from(item: CreateNftData) -> Self {292 CreateItemData::NFT(item)293 }294}295296impl From<CreateReFungibleData> for CreateItemData {297 fn from(item: CreateReFungibleData) -> Self {298 CreateItemData::ReFungible(item)299 }300}301302impl From<CreateFungibleData> for CreateItemData {303 fn from(item: CreateFungibleData) -> Self {304 CreateItemData::Fungible(item)305 }306}307308309decl_error! {310 311 pub enum Error for Module<T: Trait> {312 313 TotalCollectionsLimitExceeded,314 315 CollectionDecimalPointLimitExceeded, 316 317 CollectionNameLimitExceeded, 318 319 CollectionDescriptionLimitExceeded, 320 321 CollectionTokenPrefixLimitExceeded,322 323 CollectionNotFound,324 325 TokenNotFound,326 327 AdminNotFound,328 329 NumOverflow, 330 331 AlreadyAdmin, 332 333 NoPermission,334 335 ConfirmUnsetSponsorFail,336 337 PublicMintingNotAllowed,338 339 MustBeTokenOwner,340 341 TokenValueTooLow,342 343 NftSizeLimitExceeded,344 345 ApproveNotFound,346 347 TokenValueNotEnough,348 349 ApproveRequired,350 351 AddresNotInWhiteList,352 353 CollectionAdminsLimitExceeded,354 355 AddressOwnershipLimitExceeded,356 357 EmptyArgument,358 359 TokenConstDataLimitExceeded,360 361 TokenVariableDataLimitExceeded,362 363 NotNftDataUsedToMintNftCollectionToken,364 365 NotFungibleDataUsedToMintFungibleCollectionToken,366 367 NotReFungibleDataUsedToMintReFungibleCollectionToken,368 369 UnexpectedCollectionType,370 371 CantStoreMetadataInFungibleTokens,372 373 CollectionTokenLimitExceeded,374 375 AccountTokenLimitExceeded,376 377 CollectionLimitBoundsExceeded,378 379 SchemaDataLimitExceeded380 }381}382383pub trait Trait: system::Trait + Sized + transaction_payment::Trait + pallet_contracts::Trait {384 type Event: From<Event<Self>> + Into<<Self as system::Trait>::Event>;385386 387 type WeightInfo: WeightInfo;388}389390#[cfg(feature = "runtime-benchmarks")]391mod benchmarking;392393394395decl_storage! {396 trait Store for Module<T: Trait> as Nft {397398 399 NextCollectionID: CollectionId;400 CreatedCollectionCount: u32;401 ChainVersion: u64;402 ItemListIndex: map hasher(identity) CollectionId => TokenId;403404 405 pub ChainLimit get(fn chain_limit) config(): ChainLimits;406407 408 CollectionCount: u32;409 pub AccountItemCount get(fn account_item_count): map hasher(twox_64_concat) T::AccountId => u32;410411 412 pub Collection get(fn collection) config(): map hasher(identity) CollectionId => CollectionType<T::AccountId>;413 pub AdminList get(fn admin_list_collection): map hasher(identity) CollectionId => Vec<T::AccountId>;414 pub WhiteList get(fn white_list): double_map hasher(identity) CollectionId, hasher(twox_64_concat) T::AccountId => bool;415416 417 pub Balance get(fn balance_count): double_map hasher(identity) CollectionId, hasher(twox_64_concat) T::AccountId => u128;418419 420 pub Allowances get(fn approved): double_map hasher(identity) CollectionId, hasher(twox_64_concat) (TokenId, T::AccountId, T::AccountId) => u128;421422 423 pub NftItemList get(fn nft_item_id) config(): double_map hasher(identity) CollectionId, hasher(identity) TokenId => NftItemType<T::AccountId>;424 pub FungibleItemList get(fn fungible_item_id) config(): double_map hasher(identity) CollectionId, hasher(twox_64_concat) T::AccountId => FungibleItemType;425 pub ReFungibleItemList get(fn refungible_item_id) config(): double_map hasher(identity) CollectionId, hasher(identity) TokenId => ReFungibleItemType<T::AccountId>;426427 428 pub AddressTokens get(fn address_tokens): double_map hasher(identity) CollectionId, hasher(twox_64_concat) T::AccountId => Vec<TokenId>;429430 431 pub CreateItemBasket get(fn create_item_basket): map hasher(twox_64_concat) (CollectionId, T::AccountId) => T::BlockNumber;432 pub NftTransferBasket get(fn nft_transfer_basket): double_map hasher(identity) CollectionId, hasher(identity) TokenId => T::BlockNumber;433 pub FungibleTransferBasket get(fn fungible_transfer_basket): double_map hasher(identity) CollectionId, hasher(twox_64_concat) T::AccountId => T::BlockNumber;434 pub ReFungibleTransferBasket get(fn refungible_transfer_basket): double_map hasher(identity) CollectionId, hasher(identity) TokenId => T::BlockNumber;435436 437 pub ContractOwner get(fn contract_owner): map hasher(twox_64_concat) T::AccountId => T::AccountId;438 pub ContractSelfSponsoring get(fn contract_self_sponsoring): map hasher(twox_64_concat) T::AccountId => bool;439 pub ContractSponsorBasket get(fn contract_sponsor_basket): map hasher(twox_64_concat) (T::AccountId, T::AccountId) => T::BlockNumber;440 pub ContractSponsoringRateLimit get(fn contract_sponsoring_rate_limit): map hasher(twox_64_concat) T::AccountId => T::BlockNumber;441 pub ContractWhiteListEnabled get(fn contract_white_list_enabled): map hasher(twox_64_concat) T::AccountId => bool; 442 pub ContractWhiteList get(fn contract_white_list): double_map hasher(twox_64_concat) T::AccountId, hasher(twox_64_concat) T::AccountId => bool; 443 }444 add_extra_genesis {445 build(|config: &GenesisConfig<T>| {446 447 for (_num, _c) in &config.collection {448 <Module<T>>::init_collection(_c);449 }450451 for (_num, _c, _i) in &config.nft_item_id {452 <Module<T>>::init_nft_token(*_c, _i);453 }454455 for (collection_id, account_id, fungible_item) in &config.fungible_item_id {456 <Module<T>>::init_fungible_token(*collection_id, account_id, fungible_item);457 }458459 for (_num, _c, _i) in &config.refungible_item_id {460 <Module<T>>::init_refungible_token(*_c, _i);461 }462 })463 }464}465466decl_event!(467 pub enum Event<T>468 where469 AccountId = <T as system::Trait>::AccountId,470 {471 472 473 474 475 476 477 478 479 480 Created(CollectionId, u8, AccountId),481482 483 484 485 486 487 488 489 ItemCreated(CollectionId, TokenId),490491 492 493 494 495 496 497 498 ItemDestroyed(CollectionId, TokenId),499 }500);501502decl_module! {503 pub struct Module<T: Trait> for enum Call where origin: T::Origin {504505 fn deposit_event() = default;506 type Error = Error<T>;507508 fn on_initialize(now: T::BlockNumber) -> Weight {509510 if ChainVersion::get() < 2511 {512 let value = NextCollectionID::get();513 CreatedCollectionCount::put(value);514 ChainVersion::put(2);515 }516517 0518 }519520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 #[weight = T::WeightInfo::create_collection()]537 pub fn create_collection(origin,538 collection_name: Vec<u16>,539 collection_description: Vec<u16>,540 token_prefix: Vec<u8>,541 mode: CollectionMode) -> DispatchResult {542543 544 let who = ensure_signed(origin)?;545546 let decimal_points = match mode {547 CollectionMode::Fungible(points) => points,548 CollectionMode::ReFungible(points) => points,549 _ => 0550 };551552 553 ensure!(CollectionCount::get() < ChainLimit::get().collection_numbers_limit, Error::<T>::TotalCollectionsLimitExceeded);554555 556 ensure!(decimal_points <= MAX_DECIMAL_POINTS, Error::<T>::CollectionDecimalPointLimitExceeded);557 ensure!(collection_name.len() <= 64, Error::<T>::CollectionNameLimitExceeded);558 ensure!(collection_description.len() <= 256, Error::<T>::CollectionDescriptionLimitExceeded);559 ensure!(token_prefix.len() <= 16, Error::<T>::CollectionTokenPrefixLimitExceeded);560561 562 let next_id = CreatedCollectionCount::get()563 .checked_add(1)564 .ok_or(Error::<T>::NumOverflow)?;565566 567 let total = CollectionCount::get()568 .checked_add(1)569 .ok_or(Error::<T>::NumOverflow)?;570571 CreatedCollectionCount::put(next_id);572 CollectionCount::put(total);573574 575 let new_collection = CollectionType {576 owner: who.clone(),577 name: collection_name,578 mode: mode.clone(),579 mint_mode: false,580 access: AccessMode::Normal,581 description: collection_description,582 decimal_points: decimal_points,583 token_prefix: token_prefix,584 offchain_schema: Vec::new(),585 schema_version: SchemaVersion::ImageURL,586 sponsor: T::AccountId::default(),587 sponsor_confirmed: false,588 variable_on_chain_schema: Vec::new(),589 const_on_chain_schema: Vec::new(),590 limits: CollectionLimits::default(),591 };592593 594 <Collection<T>>::insert(next_id, new_collection);595596 597 Self::deposit_event(RawEvent::Created(next_id, mode.into(), who.clone()));598599 Ok(())600 }601602 603 604 605 606 607 608 609 610 611 #[weight = T::WeightInfo::destroy_collection()]612 pub fn destroy_collection(origin, collection_id: CollectionId) -> DispatchResult {613614 let sender = ensure_signed(origin)?;615 Self::check_owner_permissions(collection_id, sender)?;616617 <AddressTokens<T>>::remove_prefix(collection_id);618 <Allowances<T>>::remove_prefix(collection_id);619 <Balance<T>>::remove_prefix(collection_id);620 <ItemListIndex>::remove(collection_id);621 <AdminList<T>>::remove(collection_id);622 <Collection<T>>::remove(collection_id);623 <WhiteList<T>>::remove_prefix(collection_id);624625 <NftItemList<T>>::remove_prefix(collection_id);626 <FungibleItemList<T>>::remove_prefix(collection_id);627 <ReFungibleItemList<T>>::remove_prefix(collection_id);628629 <NftTransferBasket<T>>::remove_prefix(collection_id);630 <FungibleTransferBasket<T>>::remove_prefix(collection_id);631 <ReFungibleTransferBasket<T>>::remove_prefix(collection_id);632633 if CollectionCount::get() > 0634 {635 636 let total = CollectionCount::get()637 .checked_sub(1)638 .ok_or(Error::<T>::NumOverflow)?;639640 CollectionCount::put(total);641 }642643 Ok(())644 }645646 647 648 649 650 651 652 653 654 655 656 657 658 #[weight = T::WeightInfo::add_to_white_list()]659 pub fn add_to_white_list(origin, collection_id: CollectionId, address: T::AccountId) -> DispatchResult{660661 let sender = ensure_signed(origin)?;662 Self::check_owner_or_admin_permissions(collection_id, sender)?;663664 <WhiteList<T>>::insert(collection_id, address, true);665 666 Ok(())667 }668669 670 671 672 673 674 675 676 677 678 679 680 681 #[weight = T::WeightInfo::remove_from_white_list()]682 pub fn remove_from_white_list(origin, collection_id: CollectionId, address: T::AccountId) -> DispatchResult{683684 let sender = ensure_signed(origin)?;685 Self::check_owner_or_admin_permissions(collection_id, sender)?;686687 <WhiteList<T>>::remove(collection_id, address);688689 Ok(())690 }691692 693 694 695 696 697 698 699 700 701 702 703 #[weight = T::WeightInfo::set_public_access_mode()]704 pub fn set_public_access_mode(origin, collection_id: CollectionId, mode: AccessMode) -> DispatchResult705 {706 let sender = ensure_signed(origin)?;707708 Self::check_owner_permissions(collection_id, sender)?;709 let mut target_collection = <Collection<T>>::get(collection_id);710 target_collection.access = mode;711 <Collection<T>>::insert(collection_id, target_collection);712713 Ok(())714 }715716 717 718 719 720 721 722 723 724 725 726 727 728 729 #[weight = T::WeightInfo::set_mint_permission()]730 pub fn set_mint_permission(origin, collection_id: CollectionId, mint_permission: bool) -> DispatchResult731 {732 let sender = ensure_signed(origin)?;733734 Self::check_owner_permissions(collection_id, sender)?;735 let mut target_collection = <Collection<T>>::get(collection_id);736 target_collection.mint_mode = mint_permission;737 <Collection<T>>::insert(collection_id, target_collection);738739 Ok(())740 }741742 743 744 745 746 747 748 749 750 751 752 753 #[weight = T::WeightInfo::change_collection_owner()]754 pub fn change_collection_owner(origin, collection_id: CollectionId, new_owner: T::AccountId) -> DispatchResult {755756 let sender = ensure_signed(origin)?;757 Self::check_owner_permissions(collection_id, sender)?;758 let mut target_collection = <Collection<T>>::get(collection_id);759 target_collection.owner = new_owner;760 <Collection<T>>::insert(collection_id, target_collection);761762 Ok(())763 }764765 766 767 768 769 770 771 772 773 774 775 776 777 778 #[weight = T::WeightInfo::add_collection_admin()]779 pub fn add_collection_admin(origin, collection_id: CollectionId, new_admin_id: T::AccountId) -> DispatchResult {780781 let sender = ensure_signed(origin)?;782 Self::check_owner_or_admin_permissions(collection_id, sender)?;783 let mut admin_arr: Vec<T::AccountId> = Vec::new();784785 if <AdminList<T>>::contains_key(collection_id)786 {787 admin_arr = <AdminList<T>>::get(collection_id);788 ensure!(!admin_arr.contains(&new_admin_id), Error::<T>::AlreadyAdmin);789 }790791 792 ensure!((admin_arr.len() as u64) < ChainLimit::get().collections_admins_limit, Error::<T>::CollectionAdminsLimitExceeded);793794 admin_arr.push(new_admin_id);795 <AdminList<T>>::insert(collection_id, admin_arr);796797 Ok(())798 }799800 801 802 803 804 805 806 807 808 809 810 811 812 #[weight = T::WeightInfo::remove_collection_admin()]813 pub fn remove_collection_admin(origin, collection_id: CollectionId, account_id: T::AccountId) -> DispatchResult {814815 let sender = ensure_signed(origin)?;816 Self::check_owner_or_admin_permissions(collection_id, sender)?;817 ensure!(<AdminList<T>>::contains_key(collection_id), Error::<T>::AdminNotFound);818819 let mut admin_arr = <AdminList<T>>::get(collection_id);820 admin_arr.retain(|i| *i != account_id);821 <AdminList<T>>::insert(collection_id, admin_arr);822823 Ok(())824 }825826 827 828 829 830 831 832 833 834 835 #[weight = T::WeightInfo::set_collection_sponsor()]836 pub fn set_collection_sponsor(origin, collection_id: CollectionId, new_sponsor: T::AccountId) -> DispatchResult {837838 let sender = ensure_signed(origin)?;839 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);840841 let mut target_collection = <Collection<T>>::get(collection_id);842 ensure!(sender == target_collection.owner, Error::<T>::NoPermission);843844 target_collection.sponsor = new_sponsor;845 target_collection.sponsor_confirmed = false;846 <Collection<T>>::insert(collection_id, target_collection);847848 Ok(())849 }850851 852 853 854 855 856 857 858 #[weight = T::WeightInfo::confirm_sponsorship()]859 pub fn confirm_sponsorship(origin, collection_id: CollectionId) -> DispatchResult {860861 let sender = ensure_signed(origin)?;862 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);863864 let mut target_collection = <Collection<T>>::get(collection_id);865 ensure!(sender == target_collection.sponsor, Error::<T>::ConfirmUnsetSponsorFail);866867 target_collection.sponsor_confirmed = true;868 <Collection<T>>::insert(collection_id, target_collection);869870 Ok(())871 }872873 874 875 876 877 878 879 880 881 882 #[weight = T::WeightInfo::remove_collection_sponsor()]883 pub fn remove_collection_sponsor(origin, collection_id: CollectionId) -> DispatchResult {884885 let sender = ensure_signed(origin)?;886 ensure!(<Collection<T>>::contains_key(collection_id), Error::<T>::CollectionNotFound);887888 let mut target_collection = <Collection<T>>::get(collection_id);889 ensure!(sender == target_collection.owner, Error::<T>::NoPermission);890891 target_collection.sponsor = T::AccountId::default();892 target_collection.sponsor_confirmed = false;893 <Collection<T>>::insert(collection_id, target_collection);894895 Ok(())896 }897898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921922 #[weight = T::WeightInfo::create_item(data.len())]923 pub fn create_item(origin, collection_id: CollectionId, owner: T::AccountId, data: CreateItemData) -> DispatchResult {924925 let sender = ensure_signed(origin)?;926927 Self::collection_exists(collection_id)?;928929 let target_collection = <Collection<T>>::get(collection_id);930931 Self::can_create_items_in_collection(collection_id, &target_collection, &sender, &owner)?;932 Self::validate_create_item_args(&target_collection, &data)?;933 Self::create_item_no_validation(collection_id, &target_collection, owner, data)?;934935 Ok(())936 }937938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 #[weight = T::WeightInfo::create_item(items_data.into_iter()957 .map(|data| { data.len() })958 .sum())]959 pub fn create_multiple_items(origin, collection_id: CollectionId, owner: T::AccountId, items_data: Vec<CreateItemData>) -> DispatchResult {960961 ensure!(items_data.len() > 0, Error::<T>::EmptyArgument);962 let sender = ensure_signed(origin)?;963964 Self::collection_exists(collection_id)?;965 let target_collection = <Collection<T>>::get(collection_id);966967 Self::can_create_items_in_collection(collection_id, &target_collection, &sender, &owner)?;968969 for data in &items_data {970 Self::validate_create_item_args(&target_collection, data)?;971 }972 for data in &items_data {973 Self::create_item_no_validation(collection_id, &target_collection, owner.clone(), data.clone())?;974 }975976 Ok(())977 }978979 980 981 982 983 984 985 986 987 988 989 990 991 992 #[weight = T::WeightInfo::burn_item()]993 pub fn burn_item(origin, collection_id: CollectionId, item_id: TokenId, value: u128) -> DispatchResult {994995 let sender = ensure_signed(origin)?;996 Self::collection_exists(collection_id)?;997998 999 let target_collection = <Collection<T>>::get(collection_id);1000 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1001 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1002 Error::<T>::NoPermission);10031004 if target_collection.access == AccessMode::WhiteList {1005 Self::check_white_list(collection_id, &sender)?;1006 }10071008 match target_collection.mode1009 {1010 CollectionMode::NFT => Self::burn_nft_item(collection_id, item_id)?,1011 CollectionMode::Fungible(_) => Self::burn_fungible_item(&sender, collection_id, value)?,1012 CollectionMode::ReFungible(_) => Self::burn_refungible_item(collection_id, item_id, sender.clone())?,1013 _ => ()1014 };10151016 1017 Self::deposit_event(RawEvent::ItemDestroyed(collection_id, item_id));10181019 Ok(())1020 }10211022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 #[weight = T::WeightInfo::transfer()]1046 pub fn transfer(origin, recipient: T::AccountId, collection_id: CollectionId, item_id: TokenId, value: u128) -> DispatchResult {10471048 let sender = ensure_signed(origin)?;1049 let target_collection = <Collection<T>>::get(collection_id);10501051 1052 Self::is_correct_transfer(collection_id, &target_collection, &recipient)?;10531054 1055 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1056 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1057 Error::<T>::NoPermission);10581059 if target_collection.access == AccessMode::WhiteList {1060 Self::check_white_list(collection_id, &sender)?;1061 Self::check_white_list(collection_id, &recipient)?;1062 }10631064 match target_collection.mode1065 {1066 CollectionMode::NFT => Self::transfer_nft(collection_id, item_id, sender.clone(), recipient)?,1067 CollectionMode::Fungible(_) => Self::transfer_fungible(collection_id, value, &sender, &recipient)?,1068 CollectionMode::ReFungible(_) => Self::transfer_refungible(collection_id, item_id, value, sender.clone(), recipient)?,1069 _ => ()1070 };10711072 Ok(())1073 }10741075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 #[weight = T::WeightInfo::approve()]1091 pub fn approve(origin, spender: T::AccountId, collection_id: CollectionId, item_id: TokenId, amount: u128) -> DispatchResult {10921093 let sender = ensure_signed(origin)?;10941095 1096 let target_collection = <Collection<T>>::get(collection_id);1097 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1098 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1099 Error::<T>::NoPermission);11001101 if target_collection.access == AccessMode::WhiteList {1102 Self::check_white_list(collection_id, &sender)?;1103 Self::check_white_list(collection_id, &spender)?;1104 }11051106 let allowance_exists = <Allowances<T>>::contains_key(collection_id, (item_id, &sender, &spender));1107 let mut allowance: u128 = amount;1108 if allowance_exists {1109 allowance += <Allowances<T>>::get(collection_id, (item_id, &sender, &spender));1110 }1111 <Allowances<T>>::insert(collection_id, (item_id, sender.clone(), spender.clone()), allowance);11121113 Ok(())1114 }1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 #[weight = T::WeightInfo::transfer_from()]1136 pub fn transfer_from(origin, from: T::AccountId, recipient: T::AccountId, collection_id: CollectionId, item_id: TokenId, value: u128 ) -> DispatchResult {11371138 let sender = ensure_signed(origin)?;1139 let mut appoved_transfer = false;11401141 1142 let mut approval: u128 = 0;1143 if <Allowances<T>>::contains_key(collection_id, (item_id, &from, &recipient)) {1144 approval = <Allowances<T>>::get(collection_id, (item_id, &from, &recipient));1145 ensure!(approval >= value, Error::<T>::TokenValueNotEnough);1146 appoved_transfer = true;1147 }11481149 let target_collection = <Collection<T>>::get(collection_id);11501151 1152 Self::is_correct_transfer(collection_id, &target_collection, &recipient)?;11531154 1155 ensure!(appoved_transfer || Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1156 Error::<T>::NoPermission);11571158 if target_collection.access == AccessMode::WhiteList {1159 Self::check_white_list(collection_id, &sender)?;1160 Self::check_white_list(collection_id, &recipient)?;1161 }11621163 1164 if approval.checked_sub(value).unwrap_or(0) > 0 {1165 <Allowances<T>>::insert(collection_id, (item_id, &from, &recipient), approval - value);1166 }1167 else {1168 <Allowances<T>>::remove(collection_id, (item_id, &from, &recipient));1169 }11701171 match target_collection.mode1172 {1173 CollectionMode::NFT => Self::transfer_nft(collection_id, item_id, from, recipient)?,1174 CollectionMode::Fungible(_) => Self::transfer_fungible(collection_id, value, &from, &recipient)?,1175 CollectionMode::ReFungible(_) => Self::transfer_refungible(collection_id, item_id, value, from.clone(), recipient)?,1176 _ => ()1177 };11781179 Ok(())1180 }11811182 #[weight = 0]1183 pub fn safe_transfer_from(origin, collection_id: CollectionId, item_id: TokenId, new_owner: T::AccountId) -> DispatchResult {11841185 1186 1187 1188 11891190 11911192 11931194 Ok(())1195 }11961197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 #[weight = T::WeightInfo::set_variable_meta_data()]1210 pub fn set_variable_meta_data (1211 origin,1212 collection_id: CollectionId,1213 item_id: TokenId,1214 data: Vec<u8>1215 ) -> DispatchResult {1216 let sender = ensure_signed(origin)?;1217 1218 Self::collection_exists(collection_id)?;1219 1220 ensure!(ChainLimit::get().custom_data_limit >= data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);12211222 1223 let target_collection = <Collection<T>>::get(collection_id);1224 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id) ||1225 Self::is_owner_or_admin_permissions(collection_id, sender.clone()),1226 Error::<T>::NoPermission);12271228 Self::item_exists(collection_id, item_id, &target_collection.mode)?;12291230 match target_collection.mode1231 {1232 CollectionMode::NFT => Self::set_nft_variable_data(collection_id, item_id, data)?,1233 CollectionMode::ReFungible(_) => Self::set_re_fungible_variable_data(collection_id, item_id, data)?,1234 CollectionMode::Fungible(_) => fail!(Error::<T>::CantStoreMetadataInFungibleTokens),1235 _ => fail!(Error::<T>::UnexpectedCollectionType)1236 };12371238 Ok(())1239 }1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 #[weight = T::WeightInfo::set_schema_version()]1256 pub fn set_schema_version(1257 origin,1258 collection_id: CollectionId,1259 version: SchemaVersion1260 ) -> DispatchResult {1261 let sender = ensure_signed(origin)?;1262 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;1263 let mut target_collection = <Collection<T>>::get(collection_id);1264 target_collection.schema_version = version;1265 <Collection<T>>::insert(collection_id, target_collection);12661267 Ok(())1268 }12691270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 #[weight = T::WeightInfo::set_offchain_schema()]1283 pub fn set_offchain_schema(1284 origin,1285 collection_id: CollectionId,1286 schema: Vec<u8>1287 ) -> DispatchResult {1288 let sender = ensure_signed(origin)?;1289 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;12901291 1292 ensure!(schema.len() as u32 > ChainLimit::get().offchain_schema_limit, "");12931294 let mut target_collection = <Collection<T>>::get(collection_id);1295 target_collection.offchain_schema = schema;1296 <Collection<T>>::insert(collection_id, target_collection);12971298 Ok(())1299 }13001301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 #[weight = T::WeightInfo::set_const_on_chain_schema()]1314 pub fn set_const_on_chain_schema (1315 origin,1316 collection_id: CollectionId,1317 schema: Vec<u8>1318 ) -> DispatchResult {1319 let sender = ensure_signed(origin)?;1320 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;13211322 1323 ensure!(schema.len() as u32 > ChainLimit::get().const_on_chain_schema_limit, "");13241325 let mut target_collection = <Collection<T>>::get(collection_id);1326 target_collection.const_on_chain_schema = schema;1327 <Collection<T>>::insert(collection_id, target_collection);13281329 Ok(())1330 }13311332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 #[weight = T::WeightInfo::set_const_on_chain_schema()]1345 pub fn set_variable_on_chain_schema (1346 origin,1347 collection_id: CollectionId,1348 schema: Vec<u8>1349 ) -> DispatchResult {1350 let sender = ensure_signed(origin)?;1351 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;13521353 1354 ensure!(schema.len() as u32 > ChainLimit::get().variable_on_chain_schema_limit, "");13551356 let mut target_collection = <Collection<T>>::get(collection_id);1357 target_collection.variable_on_chain_schema = schema;1358 <Collection<T>>::insert(collection_id, target_collection);13591360 Ok(())1361 }13621363 1364 #[weight = T::WeightInfo::set_chain_limits()]1365 pub fn set_chain_limits(1366 origin,1367 limits: ChainLimits1368 ) -> DispatchResult {13691370 #[cfg(not(feature = "runtime-benchmarks"))]1371 ensure_root(origin)?;13721373 <ChainLimit>::put(limits);1374 Ok(())1375 }13761377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 #[weight = T::WeightInfo::enable_contract_sponsoring()]1389 pub fn enable_contract_sponsoring(1390 origin,1391 contract_address: T::AccountId,1392 enable: bool1393 ) -> DispatchResult {13941395 let sender = ensure_signed(origin)?;13961397 #[cfg(feature = "runtime-benchmarks")]1398 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());13991400 Self::ensure_contract_owned(sender, &contract_address)?;14011402 <ContractSelfSponsoring<T>>::insert(contract_address, enable);1403 Ok(())1404 }14051406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 #[weight = T::WeightInfo::set_contract_sponsoring_rate_limit()]1424 pub fn set_contract_sponsoring_rate_limit(1425 origin,1426 contract_address: T::AccountId,1427 rate_limit: T::BlockNumber1428 ) -> DispatchResult {1429 let sender = ensure_signed(origin)?;14301431 #[cfg(feature = "runtime-benchmarks")]1432 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());14331434 Self::ensure_contract_owned(sender, &contract_address)?;1435 <ContractSponsoringRateLimit<T>>::insert(contract_address, rate_limit);1436 Ok(())1437 }14381439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 #[weight = T::WeightInfo::toggle_contract_white_list()]1451 pub fn toggle_contract_white_list(1452 origin,1453 contract_address: T::AccountId,1454 enable: bool1455 ) -> DispatchResult {1456 let sender = ensure_signed(origin)?;14571458 #[cfg(feature = "runtime-benchmarks")]1459 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());14601461 Self::ensure_contract_owned(sender, &contract_address)?;1462 <ContractWhiteListEnabled<T>>::insert(contract_address, enable);1463 Ok(())1464 }1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 #[weight = T::WeightInfo::add_to_contract_white_list()]1478 pub fn add_to_contract_white_list(1479 origin,1480 contract_address: T::AccountId,1481 account_address: T::AccountId1482 ) -> DispatchResult {1483 let sender = ensure_signed(origin)?;14841485 #[cfg(feature = "runtime-benchmarks")]1486 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());1487 1488 Self::ensure_contract_owned(sender, &contract_address)?; 1489 <ContractWhiteList<T>>::insert(contract_address, account_address, true);1490 Ok(())1491 }14921493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 #[weight = T::WeightInfo::remove_from_contract_white_list()]1505 pub fn remove_from_contract_white_list(1506 origin,1507 contract_address: T::AccountId,1508 account_address: T::AccountId1509 ) -> DispatchResult {1510 let sender = ensure_signed(origin)?;15111512 #[cfg(feature = "runtime-benchmarks")]1513 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());15141515 Self::ensure_contract_owned(sender, &contract_address)?;1516 <ContractWhiteList<T>>::remove(contract_address, account_address);1517 Ok(())1518 }15191520 #[weight = T::WeightInfo::set_collection_limits()]1521 pub fn set_collection_limits(1522 origin,1523 collection_id: u32,1524 limits: CollectionLimits,1525 ) -> DispatchResult {1526 let sender = ensure_signed(origin)?;1527 Self::check_owner_permissions(collection_id, sender.clone())?;1528 let mut target_collection = <Collection<T>>::get(collection_id);1529 let chain_limits = ChainLimit::get();1530 let climits = target_collection.limits;15311532 1533 ensure!(limits.sponsor_transfer_timeout <= MAX_SPONSOR_TIMEOUT &&1534 limits.account_token_ownership_limit <= MAX_TOKEN_OWNERSHIP, 1535 Error::<T>::CollectionLimitBoundsExceeded);15361537 1538 ensure!(climits.token_limit > limits.token_limit && 1539 limits.token_limit <= chain_limits.account_token_ownership_limit, 1540 Error::<T>::AccountTokenLimitExceeded);15411542 target_collection.limits = limits;1543 <Collection<T>>::insert(collection_id, target_collection);15441545 Ok(())1546 } 1547 }1548}15491550impl<T: Trait> Module<T> {15511552 fn is_correct_transfer(collection_id: CollectionId, collection: &CollectionType<T::AccountId>, recipient: &T::AccountId) -> DispatchResult {15531554 1555 let account_items: u32 = <AddressTokens<T>>::get(collection_id, recipient).len() as u32;1556 ensure!(collection.limits.account_token_ownership_limit > account_items, Error::<T>::AccountTokenLimitExceeded);1557 1558 Ok(())1559 }15601561 fn can_create_items_in_collection(collection_id: CollectionId, collection: &CollectionType<T::AccountId>, sender: &T::AccountId, owner: &T::AccountId) -> DispatchResult {15621563 1564 let total_items: u32 = ItemListIndex::get(collection_id);1565 let account_items: u32 = <AddressTokens<T>>::get(collection_id, owner).len() as u32;1566 ensure!(collection.limits.token_limit > total_items, Error::<T>::CollectionTokenLimitExceeded);1567 ensure!(collection.limits.account_token_ownership_limit > account_items, Error::<T>::AccountTokenLimitExceeded);15681569 if !Self::is_owner_or_admin_permissions(collection_id, sender.clone()) {1570 ensure!(collection.mint_mode == true, Error::<T>::PublicMintingNotAllowed);1571 Self::check_white_list(collection_id, owner)?;1572 Self::check_white_list(collection_id, sender)?;1573 }15741575 Ok(())1576 }15771578 fn validate_create_item_args(target_collection: &CollectionType<T::AccountId>, data: &CreateItemData) -> DispatchResult {1579 match target_collection.mode1580 {1581 CollectionMode::NFT => {1582 if let CreateItemData::NFT(data) = data {1583 1584 ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, Error::<T>::TokenConstDataLimitExceeded);1585 ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);1586 } else {1587 fail!(Error::<T>::NotNftDataUsedToMintNftCollectionToken);1588 }1589 },1590 CollectionMode::Fungible(_) => {1591 if let CreateItemData::Fungible(_) = data {1592 } else {1593 fail!(Error::<T>::NotFungibleDataUsedToMintFungibleCollectionToken);1594 }1595 },1596 CollectionMode::ReFungible(_) => {1597 if let CreateItemData::ReFungible(data) = data {15981599 1600 ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, Error::<T>::TokenConstDataLimitExceeded);1601 ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);1602 } else {1603 fail!(Error::<T>::NotReFungibleDataUsedToMintReFungibleCollectionToken);1604 }1605 },1606 _ => { fail!(Error::<T>::UnexpectedCollectionType); }1607 };16081609 Ok(())1610 }16111612 fn create_item_no_validation(collection_id: CollectionId, collection: &CollectionType<T::AccountId>, owner: T::AccountId, data: CreateItemData) -> DispatchResult {1613 match data1614 {1615 CreateItemData::NFT(data) => {1616 let item = NftItemType {1617 owner,1618 const_data: data.const_data,1619 variable_data: data.variable_data1620 };16211622 Self::add_nft_item(collection_id, item)?;1623 },1624 CreateItemData::Fungible(data) => {1625 Self::add_fungible_item(collection_id, &owner, data.value)?;1626 },1627 CreateItemData::ReFungible(data) => {1628 let mut owner_list = Vec::new();1629 let value = (10 as u128).pow(collection.decimal_points as u32);1630 owner_list.push(Ownership {owner: owner.clone(), fraction: value});16311632 let item = ReFungibleItemType {1633 owner: owner_list,1634 const_data: data.const_data,1635 variable_data: data.variable_data1636 };16371638 Self::add_refungible_item(collection_id, item)?;1639 }1640 };16411642 1643 Self::deposit_event(RawEvent::ItemCreated(collection_id, <ItemListIndex>::get(collection_id)));16441645 Ok(())1646 }16471648 fn add_fungible_item(collection_id: CollectionId, owner: &T::AccountId, value: u128) -> DispatchResult {16491650 1651 let mut balance: u128 = 0;1652 if <FungibleItemList<T>>::contains_key(collection_id, owner) {1653 balance = <FungibleItemList<T>>::get(collection_id, owner).value;1654 } 16551656 1657 let item = FungibleItemType {1658 value: balance + value1659 };1660 <FungibleItemList<T>>::insert(collection_id, (*owner).clone(), item);16611662 1663 let new_balance = <Balance<T>>::get(collection_id, owner)1664 .checked_add(value)1665 .ok_or(Error::<T>::NumOverflow)?;1666 <Balance<T>>::insert(collection_id, (*owner).clone(), new_balance);16671668 Ok(())1669 }16701671 fn add_refungible_item(collection_id: CollectionId, item: ReFungibleItemType<T::AccountId>) -> DispatchResult {1672 let current_index = <ItemListIndex>::get(collection_id)1673 .checked_add(1)1674 .ok_or(Error::<T>::NumOverflow)?;1675 let itemcopy = item.clone();16761677 let value = item.owner.first().unwrap().fraction;1678 let owner = item.owner.first().unwrap().owner.clone();16791680 Self::add_token_index(collection_id, current_index, owner.clone())?;16811682 <ItemListIndex>::insert(collection_id, current_index);1683 <ReFungibleItemList<T>>::insert(collection_id, current_index, itemcopy);16841685 1686 let new_balance = <Balance<T>>::get(collection_id, owner.clone())1687 .checked_add(value)1688 .ok_or(Error::<T>::NumOverflow)?;1689 <Balance<T>>::insert(collection_id, owner.clone(), new_balance);16901691 Ok(())1692 }16931694 fn add_nft_item(collection_id: CollectionId, item: NftItemType<T::AccountId>) -> DispatchResult {1695 let current_index = <ItemListIndex>::get(collection_id)1696 .checked_add(1)1697 .ok_or(Error::<T>::NumOverflow)?;16981699 let item_owner = item.owner.clone();1700 Self::add_token_index(collection_id, current_index, item.owner.clone())?;17011702 <ItemListIndex>::insert(collection_id, current_index);1703 <NftItemList<T>>::insert(collection_id, current_index, item);17041705 1706 let new_balance = <Balance<T>>::get(collection_id, item_owner.clone())1707 .checked_add(1)1708 .ok_or(Error::<T>::NumOverflow)?;1709 <Balance<T>>::insert(collection_id, item_owner.clone(), new_balance);17101711 Ok(())1712 }17131714 fn burn_refungible_item(1715 collection_id: CollectionId,1716 item_id: TokenId,1717 owner: T::AccountId,1718 ) -> DispatchResult {1719 ensure!(1720 <ReFungibleItemList<T>>::contains_key(collection_id, item_id),1721 Error::<T>::TokenNotFound1722 );1723 let collection = <ReFungibleItemList<T>>::get(collection_id, item_id);1724 let item = collection1725 .owner1726 .iter()1727 .filter(|&i| i.owner == owner)1728 .next()1729 .unwrap();1730 Self::remove_token_index(collection_id, item_id, owner.clone())?;17311732 1733 let new_balance = <Balance<T>>::get(collection_id, item.owner.clone())1734 .checked_sub(item.fraction)1735 .ok_or(Error::<T>::NumOverflow)?;1736 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);17371738 <ReFungibleItemList<T>>::remove(collection_id, item_id);17391740 Ok(())1741 }17421743 fn burn_nft_item(collection_id: CollectionId, item_id: TokenId) -> DispatchResult {1744 ensure!(1745 <NftItemList<T>>::contains_key(collection_id, item_id),1746 Error::<T>::TokenNotFound1747 );1748 let item = <NftItemList<T>>::get(collection_id, item_id);1749 Self::remove_token_index(collection_id, item_id, item.owner.clone())?;17501751 1752 let new_balance = <Balance<T>>::get(collection_id, item.owner.clone())1753 .checked_sub(1)1754 .ok_or(Error::<T>::NumOverflow)?;1755 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);1756 <NftItemList<T>>::remove(collection_id, item_id);17571758 Ok(())1759 }17601761 fn burn_fungible_item(owner: &T::AccountId, collection_id: CollectionId, value: u128) -> DispatchResult {1762 ensure!(1763 <FungibleItemList<T>>::contains_key(collection_id, owner),1764 Error::<T>::TokenNotFound1765 );1766 let mut balance = <FungibleItemList<T>>::get(collection_id, owner);1767 ensure!(balance.value >= value, Error::<T>::TokenValueNotEnough);17681769 1770 let new_balance = <Balance<T>>::get(collection_id, owner)1771 .checked_sub(value)1772 .ok_or(Error::<T>::NumOverflow)?;1773 <Balance<T>>::insert(collection_id, (*owner).clone(), new_balance);17741775 if balance.value - value > 0 {1776 balance.value -= value;1777 <FungibleItemList<T>>::insert(collection_id, (*owner).clone(), balance);1778 }1779 else {1780 <FungibleItemList<T>>::remove(collection_id, owner);1781 }17821783 Ok(())1784 }17851786 fn collection_exists(collection_id: CollectionId) -> DispatchResult {1787 ensure!(1788 <Collection<T>>::contains_key(collection_id),1789 Error::<T>::CollectionNotFound1790 );1791 Ok(())1792 }17931794 fn check_owner_permissions(collection_id: CollectionId, subject: T::AccountId) -> DispatchResult {1795 Self::collection_exists(collection_id)?;17961797 let target_collection = <Collection<T>>::get(collection_id);1798 ensure!(1799 subject == target_collection.owner,1800 Error::<T>::NoPermission1801 );18021803 Ok(())1804 }18051806 fn is_owner_or_admin_permissions(collection_id: CollectionId, subject: T::AccountId) -> bool {1807 let target_collection = <Collection<T>>::get(collection_id);1808 let mut result: bool = subject == target_collection.owner;1809 let exists = <AdminList<T>>::contains_key(collection_id);18101811 if !result & exists {1812 if <AdminList<T>>::get(collection_id).contains(&subject) {1813 result = true1814 }1815 }18161817 result1818 }18191820 fn check_owner_or_admin_permissions(1821 collection_id: CollectionId,1822 subject: T::AccountId,1823 ) -> DispatchResult {1824 Self::collection_exists(collection_id)?;1825 let result = Self::is_owner_or_admin_permissions(collection_id, subject.clone());18261827 ensure!(1828 result,1829 Error::<T>::NoPermission1830 );1831 Ok(())1832 }18331834 fn is_item_owner(subject: T::AccountId, collection_id: CollectionId, item_id: TokenId) -> bool {1835 let target_collection = <Collection<T>>::get(collection_id);18361837 match target_collection.mode {1838 CollectionMode::NFT => {1839 <NftItemList<T>>::get(collection_id, item_id).owner == subject1840 }1841 CollectionMode::Fungible(_) => {1842 <FungibleItemList<T>>::contains_key(collection_id, &subject)1843 }1844 CollectionMode::ReFungible(_) => {1845 <ReFungibleItemList<T>>::get(collection_id, item_id)1846 .owner1847 .iter()1848 .any(|i| i.owner == subject)1849 }1850 CollectionMode::Invalid => false,1851 }1852 }18531854 fn check_white_list(collection_id: CollectionId, address: &T::AccountId) -> DispatchResult {1855 let mes = Error::<T>::AddresNotInWhiteList;1856 ensure!(<WhiteList<T>>::contains_key(collection_id, address), mes);18571858 Ok(())1859 }18601861 fn transfer_fungible(1862 collection_id: CollectionId,1863 value: u128,1864 owner: &T::AccountId,1865 recipient: &T::AccountId,1866 ) -> DispatchResult {1867 ensure!(1868 <FungibleItemList<T>>::contains_key(collection_id, owner),1869 Error::<T>::TokenNotFound1870 );18711872 let mut balance = <FungibleItemList<T>>::get(collection_id, owner);1873 ensure!(balance.value >= value, Error::<T>::TokenValueTooLow);18741875 1876 Self::add_fungible_item(collection_id, recipient, value)?;18771878 1879 <Balance<T>>::insert(collection_id, (*owner).clone(), balance.value - value);18801881 1882 if balance.value == value {1883 <FungibleItemList<T>>::remove(collection_id, owner);1884 }1885 else {1886 balance.value -= value;1887 <FungibleItemList<T>>::insert(collection_id, (*owner).clone(), balance);1888 }18891890 Ok(())1891 }18921893 fn transfer_refungible(1894 collection_id: CollectionId,1895 item_id: TokenId,1896 value: u128,1897 owner: T::AccountId,1898 new_owner: T::AccountId,1899 ) -> DispatchResult {1900 ensure!(1901 <ReFungibleItemList<T>>::contains_key(collection_id, item_id),1902 Error::<T>::TokenNotFound1903 );19041905 let full_item = <ReFungibleItemList<T>>::get(collection_id, item_id);1906 let item = full_item1907 .owner1908 .iter()1909 .filter(|i| i.owner == owner)1910 .next()1911 .ok_or(Error::<T>::NumOverflow)?;1912 let amount = item.fraction;19131914 ensure!(amount >= value, Error::<T>::TokenValueTooLow);19151916 1917 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())1918 .checked_sub(value)1919 .ok_or(Error::<T>::NumOverflow)?;1920 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);19211922 let balancenew_owner = <Balance<T>>::get(collection_id, new_owner.clone())1923 .checked_add(value)1924 .ok_or(Error::<T>::NumOverflow)?;1925 <Balance<T>>::insert(collection_id, new_owner.clone(), balancenew_owner);19261927 let old_owner = item.owner.clone();1928 let new_owner_has_account = full_item.owner.iter().any(|i| i.owner == new_owner);19291930 1931 if amount == value && !new_owner_has_account {1932 1933 1934 let mut new_full_item = full_item.clone();1935 new_full_item1936 .owner1937 .iter_mut()1938 .find(|i| i.owner == owner)1939 .unwrap()1940 .owner = new_owner.clone();1941 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);19421943 1944 Self::move_token_index(collection_id, item_id, old_owner.clone(), new_owner.clone())?;1945 } else {1946 let mut new_full_item = full_item.clone();1947 new_full_item1948 .owner1949 .iter_mut()1950 .find(|i| i.owner == owner)1951 .unwrap()1952 .fraction -= value;19531954 1955 if new_owner_has_account {1956 1957 new_full_item1958 .owner1959 .iter_mut()1960 .find(|i| i.owner == new_owner)1961 .unwrap()1962 .fraction += value;1963 } else {1964 1965 new_full_item.owner.push(Ownership {1966 owner: new_owner.clone(),1967 fraction: value,1968 });1969 Self::add_token_index(collection_id, item_id, new_owner.clone())?;1970 }19711972 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);1973 }19741975 Ok(())1976 }19771978 fn transfer_nft(1979 collection_id: CollectionId,1980 item_id: TokenId,1981 sender: T::AccountId,1982 new_owner: T::AccountId,1983 ) -> DispatchResult {1984 ensure!(1985 <NftItemList<T>>::contains_key(collection_id, item_id),1986 Error::<T>::TokenNotFound1987 );19881989 let mut item = <NftItemList<T>>::get(collection_id, item_id);19901991 ensure!(1992 sender == item.owner,1993 Error::<T>::MustBeTokenOwner1994 );19951996 1997 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())1998 .checked_sub(1)1999 .ok_or(Error::<T>::NumOverflow)?;2000 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);20012002 let balancenew_owner = <Balance<T>>::get(collection_id, new_owner.clone())2003 .checked_add(1)2004 .ok_or(Error::<T>::NumOverflow)?;2005 <Balance<T>>::insert(collection_id, new_owner.clone(), balancenew_owner);20062007 2008 let old_owner = item.owner.clone();2009 item.owner = new_owner.clone();2010 <NftItemList<T>>::insert(collection_id, item_id, item);20112012 2013 Self::move_token_index(collection_id, item_id, old_owner.clone(), new_owner.clone())?;20142015 Ok(())2016 }2017 2018 fn item_exists(2019 collection_id: CollectionId,2020 item_id: TokenId,2021 mode: &CollectionMode2022 ) -> DispatchResult {2023 match mode {2024 CollectionMode::NFT => ensure!(<NftItemList<T>>::contains_key(collection_id, item_id), Error::<T>::TokenNotFound),2025 CollectionMode::ReFungible(_) => ensure!(<ReFungibleItemList<T>>::contains_key(collection_id, item_id), Error::<T>::TokenNotFound),2026 _ => ()2027 };2028 2029 Ok(())2030 }20312032 fn set_re_fungible_variable_data(2033 collection_id: CollectionId,2034 item_id: TokenId,2035 data: Vec<u8>2036 ) -> DispatchResult {2037 let mut item = <ReFungibleItemList<T>>::get(collection_id, item_id);20382039 item.variable_data = data;20402041 <ReFungibleItemList<T>>::insert(collection_id, item_id, item);20422043 Ok(())2044 }20452046 fn set_nft_variable_data(2047 collection_id: CollectionId,2048 item_id: TokenId,2049 data: Vec<u8>2050 ) -> DispatchResult {2051 let mut item = <NftItemList<T>>::get(collection_id, item_id);2052 2053 item.variable_data = data;20542055 <NftItemList<T>>::insert(collection_id, item_id, item);2056 2057 Ok(())2058 }20592060 fn init_collection(item: &CollectionType<T::AccountId>) {2061 2062 assert!(2063 item.decimal_points <= MAX_DECIMAL_POINTS,2064 "decimal_points parameter must be lower than MAX_DECIMAL_POINTS"2065 );2066 assert!(2067 item.name.len() <= 64,2068 "Collection name can not be longer than 63 char"2069 );2070 assert!(2071 item.name.len() <= 256,2072 "Collection description can not be longer than 255 char"2073 );2074 assert!(2075 item.token_prefix.len() <= 16,2076 "Token prefix can not be longer than 15 char"2077 );20782079 2080 let next_id = CreatedCollectionCount::get()2081 .checked_add(1)2082 .unwrap();20832084 CreatedCollectionCount::put(next_id);2085 }20862087 fn init_nft_token(collection_id: CollectionId, item: &NftItemType<T::AccountId>) {2088 let current_index = <ItemListIndex>::get(collection_id)2089 .checked_add(1)2090 .unwrap();20912092 let item_owner = item.owner.clone();2093 Self::add_token_index(collection_id, current_index, item.owner.clone()).unwrap();20942095 <ItemListIndex>::insert(collection_id, current_index);20962097 2098 let new_balance = <Balance<T>>::get(collection_id, item_owner.clone())2099 .checked_add(1)2100 .unwrap();2101 <Balance<T>>::insert(collection_id, item_owner.clone(), new_balance);2102 }21032104 fn init_fungible_token(collection_id: CollectionId, owner: &T::AccountId, item: &FungibleItemType) {2105 let current_index = <ItemListIndex>::get(collection_id)2106 .checked_add(1)2107 .unwrap();21082109 Self::add_token_index(collection_id, current_index, (*owner).clone()).unwrap();21102111 <ItemListIndex>::insert(collection_id, current_index);21122113 2114 let new_balance = <Balance<T>>::get(collection_id, owner)2115 .checked_add(item.value)2116 .unwrap();2117 <Balance<T>>::insert(collection_id, (*owner).clone(), new_balance);2118 }21192120 fn init_refungible_token(collection_id: CollectionId, item: &ReFungibleItemType<T::AccountId>) {2121 let current_index = <ItemListIndex>::get(collection_id)2122 .checked_add(1)2123 .unwrap();21242125 let value = item.owner.first().unwrap().fraction;2126 let owner = item.owner.first().unwrap().owner.clone();21272128 Self::add_token_index(collection_id, current_index, owner.clone()).unwrap();21292130 <ItemListIndex>::insert(collection_id, current_index);21312132 2133 let new_balance = <Balance<T>>::get(collection_id, owner.clone())2134 .checked_add(value)2135 .unwrap();2136 <Balance<T>>::insert(collection_id, owner.clone(), new_balance);2137 }21382139 fn add_token_index(collection_id: CollectionId, item_index: TokenId, owner: T::AccountId) -> DispatchResult {21402141 2142 if <AccountItemCount<T>>::contains_key(owner.clone()) {21432144 2145 let count = <AccountItemCount<T>>::get(owner.clone());2146 ensure!(count < ChainLimit::get().account_token_ownership_limit, Error::<T>::AddressOwnershipLimitExceeded);21472148 <AccountItemCount<T>>::insert(owner.clone(), count2149 .checked_add(1)2150 .ok_or(Error::<T>::NumOverflow)?);2151 }2152 else {2153 <AccountItemCount<T>>::insert(owner.clone(), 1);2154 }21552156 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());2157 if list_exists {2158 let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());2159 let item_contains = list.contains(&item_index.clone());21602161 if !item_contains {2162 list.push(item_index.clone());2163 }21642165 <AddressTokens<T>>::insert(collection_id, owner.clone(), list);2166 } else {2167 let mut itm = Vec::new();2168 itm.push(item_index.clone());2169 <AddressTokens<T>>::insert(collection_id, owner, itm);2170 2171 }21722173 Ok(())2174 }21752176 fn remove_token_index(2177 collection_id: CollectionId,2178 item_index: TokenId,2179 owner: T::AccountId,2180 ) -> DispatchResult {21812182 2183 <AccountItemCount<T>>::insert(owner.clone(), 2184 <AccountItemCount<T>>::get(owner.clone())2185 .checked_sub(1)2186 .ok_or(Error::<T>::NumOverflow)?);218721882189 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());2190 if list_exists {2191 let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());2192 let item_contains = list.contains(&item_index.clone());21932194 if item_contains {2195 list.retain(|&item| item != item_index);2196 <AddressTokens<T>>::insert(collection_id, owner, list);2197 }2198 }21992200 Ok(())2201 }22022203 fn move_token_index(2204 collection_id: CollectionId,2205 item_index: TokenId,2206 old_owner: T::AccountId,2207 new_owner: T::AccountId,2208 ) -> DispatchResult {2209 Self::remove_token_index(collection_id, item_index, old_owner)?;2210 Self::add_token_index(collection_id, item_index, new_owner)?;22112212 Ok(())2213 }2214 2215 fn ensure_contract_owned(account: T::AccountId, contract: &T::AccountId) -> DispatchResult {2216 if <ContractOwner<T>>::contains_key(contract.clone()) {2217 let owner = <ContractOwner<T>>::get(contract);2218 ensure!(account == owner, Error::<T>::NoPermission);2219 } else {2220 fail!(Error::<T>::NoPermission);2221 }22222223 Ok(())2224 }2225}2226222722282229223022312232pub type Multiplier = FixedU128;22332234type BalanceOf<T> = <<T as transaction_payment::Trait>::Currency as Currency<2235 <T as system::Trait>::AccountId,2236>>::Balance;2237type NegativeImbalanceOf<T> = <<T as transaction_payment::Trait>::Currency as Currency<2238 <T as system::Trait>::AccountId,2239>>::NegativeImbalance;2240224122422243#[derive(Encode, Decode, Clone, Eq, PartialEq)]2244pub struct ChargeTransactionPayment<T: Trait + Send + Sync>(2245 #[codec(compact)] BalanceOf<T>2246);22472248impl<T: Trait + Send + Sync> sp_std::fmt::Debug2249 for ChargeTransactionPayment<T>2250{2251 #[cfg(feature = "std")]2252 fn fmt(&self, f: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {2253 write!(f, "ChargeTransactionPayment<{:?}>", self.0)2254 }2255 #[cfg(not(feature = "std"))]2256 fn fmt(&self, _: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {2257 Ok(())2258 }2259}22602261impl<T: Trait + Send + Sync> ChargeTransactionPayment<T>2262where2263 T::Call: Dispatchable<Info = DispatchInfo, PostInfo = PostDispatchInfo> + IsSubType<Call<T>> + IsSubType<pallet_contracts::Call<T>>,2264 BalanceOf<T>: Send + Sync + FixedPointOperand,2265{2266 2267 pub fn from(fee: BalanceOf<T>) -> Self {2268 Self(fee)2269 }22702271 pub fn traditional_fee(2272 len: usize,2273 info: &DispatchInfoOf<T::Call>,2274 tip: BalanceOf<T>,2275 ) -> BalanceOf<T>2276 where2277 T::Call: Dispatchable<Info = DispatchInfo>,2278 {2279 <transaction_payment::Module<T>>::compute_fee(len as u32, info, tip)2280 }22812282 fn get_priority(len: usize, info: &DispatchInfoOf<T::Call>, final_fee: BalanceOf<T>) -> TransactionPriority {2283 let weight_saturation = T::MaximumBlockWeight::get() / info.weight.max(1);2284 let len_saturation = T::MaximumBlockLength::get() as u64 / (len as u64).max(1);2285 let coefficient: BalanceOf<T> = weight_saturation.min(len_saturation).saturated_into::<BalanceOf<T>>();2286 final_fee.saturating_mul(coefficient).saturated_into::<TransactionPriority>()2287 }22882289 fn withdraw_fee(2290 &self,2291 who: &T::AccountId,2292 call: &T::Call,2293 info: &DispatchInfoOf<T::Call>,2294 len: usize,2295 ) -> Result<(BalanceOf<T>, Option<NegativeImbalanceOf<T>>), TransactionValidityError> {2296 let tip = self.0;22972298 2299 2300 2301 2302 2303 2304 2305 let fee = Self::traditional_fee(len, info, tip);23062307 2308 if fee.is_zero() {2309 return Ok((fee, None));2310 }23112312 2313 2314 let mut sponsor: T::AccountId = match IsSubType::<Call<T>>::is_sub_type(call) {2315 Some(Call::create_item(collection_id, _owner, _properties)) => {23162317 2318 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;23192320 let limit = <Collection<T>>::get(collection_id).limits.sponsor_transfer_timeout;2321 let mut sponsored = true;2322 if <CreateItemBasket<T>>::contains_key((collection_id, &who)) {2323 let last_tx_block = <CreateItemBasket<T>>::get((collection_id, &who));2324 let limit_time = last_tx_block + limit.into();2325 if block_number <= limit_time {2326 sponsored = false;2327 }2328 }2329 if sponsored {2330 <CreateItemBasket<T>>::insert((collection_id, who.clone()), block_number);2331 }23322333 2334 if (<Collection<T>>::get(collection_id).limits.sponsored_data_size >= (_properties.len() as u32)) &&2335 (<Collection<T>>::get(collection_id).sponsor_confirmed) &&2336 (sponsored)2337 {2338 <Collection<T>>::get(collection_id).sponsor2339 } else {2340 T::AccountId::default()2341 }2342 }2343 Some(Call::transfer(_new_owner, collection_id, item_id, _value)) => {2344 2345 let mut sponsor_transfer = false;2346 if <Collection<T>>::get(collection_id).sponsor_confirmed {23472348 let collection_limits = <Collection<T>>::get(collection_id).limits;2349 let collection_mode = <Collection<T>>::get(collection_id).mode;2350 2351 2352 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2353 sponsor_transfer = match collection_mode {2354 CollectionMode::NFT => {2355 2356 2357 let limit: u32 = if collection_limits.sponsor_transfer_timeout > 0 {2358 collection_limits.sponsor_transfer_timeout2359 } else {2360 ChainLimit::get().nft_sponsor_transfer_timeout2361 };2362 2363 let mut sponsored = true;2364 if <NftTransferBasket<T>>::contains_key(collection_id, item_id) {2365 let last_tx_block = <NftTransferBasket<T>>::get(collection_id, item_id);2366 let limit_time = last_tx_block + limit.into();2367 if block_number <= limit_time {2368 sponsored = false;2369 }2370 }2371 if sponsored {2372 <NftTransferBasket<T>>::insert(collection_id, item_id, block_number);2373 }23742375 sponsored2376 }2377 CollectionMode::Fungible(_) => {2378 2379 2380 let limit: u32 = if collection_limits.sponsor_transfer_timeout > 0 {2381 collection_limits.sponsor_transfer_timeout2382 } else {2383 ChainLimit::get().fungible_sponsor_transfer_timeout2384 };2385 2386 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2387 let mut sponsored = true;2388 if <FungibleTransferBasket<T>>::contains_key(collection_id, who) {2389 let last_tx_block = <FungibleTransferBasket<T>>::get(collection_id, who);2390 let limit_time = last_tx_block + limit.into();2391 if block_number <= limit_time {2392 sponsored = false;2393 }2394 }2395 if sponsored {2396 <FungibleTransferBasket<T>>::insert(collection_id, who, block_number);2397 }23982399 sponsored2400 }2401 CollectionMode::ReFungible(_) => {2402 2403 2404 let limit: u32 = if collection_limits.sponsor_transfer_timeout > 0 {2405 collection_limits.sponsor_transfer_timeout2406 } else {2407 ChainLimit::get().refungible_sponsor_transfer_timeout2408 };2409 2410 let mut sponsored = true;2411 if <ReFungibleTransferBasket<T>>::contains_key(collection_id, item_id) {2412 let last_tx_block = <ReFungibleTransferBasket<T>>::get(collection_id, item_id);2413 let limit_time = last_tx_block + limit.into();2414 if block_number <= limit_time {2415 sponsored = false;2416 }2417 }2418 if sponsored {2419 <ReFungibleTransferBasket<T>>::insert(collection_id, item_id, block_number);2420 }24212422 sponsored2423 }2424 _ => {2425 false2426 },2427 };2428 }24292430 if !sponsor_transfer {2431 T::AccountId::default()2432 } else {2433 <Collection<T>>::get(collection_id).sponsor2434 }2435 }24362437 _ => T::AccountId::default(),2438 };24392440 2441 sponsor = match IsSubType::<pallet_contracts::Call<T>>::is_sub_type(call) {24422443 2444 Some(pallet_contracts::Call::instantiate(_endowment, _gas_limit, code_hash, data)) => {24452446 let new_contract_address = <T as pallet_contracts::Trait>::DetermineContractAddress::contract_address_for(2447 code_hash,2448 &data,2449 &who,2450 );2451 <ContractOwner<T>>::insert(new_contract_address.clone(), who.clone());24522453 T::AccountId::default()2454 },24552456 2457 Some(pallet_contracts::Call::call(dest, _value, _gas_limit, _data)) => {24582459 let called_contract: T::AccountId = T::Lookup::lookup((*dest).clone()).unwrap_or(T::AccountId::default());24602461 let owned_contract = <ContractOwner<T>>::contains_key(called_contract.clone())2462 && <ContractOwner<T>>::get(called_contract.clone()) == *who;2463 let white_list_enabled = <ContractWhiteListEnabled<T>>::contains_key(called_contract.clone()) && <ContractWhiteListEnabled<T>>::get(called_contract.clone());2464 2465 if !owned_contract && white_list_enabled {2466 if !<ContractWhiteList<T>>::contains_key(called_contract.clone(), who) {2467 return Err(InvalidTransaction::Call.into());2468 }2469 }24702471 let mut sponsor_transfer = false;2472 if <ContractSponsoringRateLimit<T>>::contains_key(called_contract.clone()) {2473 let last_tx_block = <ContractSponsorBasket<T>>::get((&called_contract, &who));2474 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2475 let rate_limit = <ContractSponsoringRateLimit<T>>::get(&called_contract);2476 let limit_time = last_tx_block + rate_limit;24772478 if block_number >= limit_time {2479 <ContractSponsorBasket<T>>::insert((called_contract.clone(), who.clone()), block_number);2480 sponsor_transfer = true;2481 }2482 } else {2483 sponsor_transfer = false;2484 }2485 2486 2487 let mut sp = T::AccountId::default();2488 if sponsor_transfer {2489 if <ContractSelfSponsoring<T>>::contains_key(called_contract.clone()) {2490 if <ContractSelfSponsoring<T>>::get(called_contract.clone()) {2491 sp = called_contract;2492 }2493 }2494 }24952496 sp2497 },24982499 _ => sponsor,2500 };25012502 let mut who_pays_fee: T::AccountId = sponsor.clone();2503 if sponsor == T::AccountId::default() {2504 who_pays_fee = who.clone();2505 }25062507 match <T as transaction_payment::Trait>::Currency::withdraw(2508 &who_pays_fee,2509 fee,2510 if tip.is_zero() {2511 WithdrawReason::TransactionPayment.into()2512 } else {2513 WithdrawReason::TransactionPayment | WithdrawReason::Tip2514 },2515 ExistenceRequirement::KeepAlive,2516 ) {2517 Ok(imbalance) => Ok((fee, Some(imbalance))),2518 Err(_) => Err(InvalidTransaction::Payment.into()),2519 }2520 }2521}252225232524impl<T: Trait + Send + Sync> SignedExtension2525 for ChargeTransactionPayment<T>2526where2527 BalanceOf<T>: Send + Sync + From<u64> + FixedPointOperand,2528 T::Call: Dispatchable<Info = DispatchInfo, PostInfo = PostDispatchInfo> + IsSubType<Call<T>> + IsSubType<pallet_contracts::Call<T>>,2529{2530 const IDENTIFIER: &'static str = "ChargeTransactionPayment";2531 type AccountId = T::AccountId;2532 type Call = T::Call;2533 type AdditionalSigned = ();2534 type Pre = (2535 BalanceOf<T>,2536 Self::AccountId,2537 Option<NegativeImbalanceOf<T>>,2538 BalanceOf<T>,2539 );2540 fn additional_signed(&self) -> sp_std::result::Result<(), TransactionValidityError> {2541 Ok(())2542 }25432544 fn validate(2545 &self,2546 who: &Self::AccountId,2547 call: &Self::Call,2548 info: &DispatchInfoOf<Self::Call>,2549 len: usize,2550 ) -> TransactionValidity {2551 let (fee, _) = self.withdraw_fee(who, call, info, len)?;2552 Ok(ValidTransaction {2553 priority: Self::get_priority(len, info, fee),2554 ..Default::default()2555 })2556 }25572558 fn pre_dispatch(2559 self,2560 who: &Self::AccountId,2561 call: &Self::Call,2562 info: &DispatchInfoOf<Self::Call>,2563 len: usize,2564 ) -> Result<Self::Pre, TransactionValidityError> {2565 let (fee, imbalance) = self.withdraw_fee(who, call, info, len)?;2566 Ok((self.0, who.clone(), imbalance, fee))2567 }25682569 fn post_dispatch(2570 pre: Self::Pre,2571 info: &DispatchInfoOf<Self::Call>,2572 post_info: &PostDispatchInfoOf<Self::Call>,2573 len: usize,2574 _result: &DispatchResult,2575 ) -> Result<(), TransactionValidityError> {2576 let (tip, who, imbalance, fee) = pre;2577 if let Some(payed) = imbalance {2578 let actual_fee = <transaction_payment::Module<T>>::compute_actual_fee(2579 len as u32, info, post_info, tip,2580 );2581 let refund = fee.saturating_sub(actual_fee);2582 let actual_payment =2583 match <T as transaction_payment::Trait>::Currency::deposit_into_existing(2584 &who, refund,2585 ) {2586 Ok(refund_imbalance) => {2587 2588 2589 match payed.offset(refund_imbalance) {2590 Ok(actual_payment) => actual_payment,2591 Err(_) => return Err(InvalidTransaction::Payment.into()),2592 }2593 }2594 2595 2596 Err(_) => payed,2597 };2598 let imbalances = actual_payment.split(tip);2599 <T as transaction_payment::Trait>::OnTransactionPayment::on_unbalanceds(2600 Some(imbalances.0).into_iter().chain(Some(imbalances.1)),2601 );2602 }2603 Ok(())2604 }2605}26062607