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 core::ops::{Deref, DerefMut};17use codec::{Decode, Encode};18pub use frame_support::{19 construct_runtime, decl_event, decl_module, decl_storage, decl_error,20 dispatch::DispatchResult,21 ensure, fail, parameter_types,22 traits::{23 Currency, ExistenceRequirement, Get, Imbalance, KeyOwnerProofSystem, OnUnbalanced,24 Randomness, IsSubType, WithdrawReasons,25 },26 weights::{27 constants::{BlockExecutionWeight, ExtrinsicBaseWeight, RocksDbWeight, WEIGHT_PER_SECOND},28 DispatchInfo, GetDispatchInfo, IdentityFee, Pays, PostDispatchInfo, Weight,29 WeightToFeePolynomial, DispatchClass,30 },31 StorageValue,32 transactional,33};3435use frame_system::{self as system, ensure_signed, ensure_root};36use sp_runtime::sp_std::prelude::Vec;37use sp_runtime::{38 traits::{39 Hash, DispatchInfoOf, Dispatchable, PostDispatchInfoOf, Saturating, SaturatedConversion, SignedExtension, Zero,40 },41 transaction_validity::{42 TransactionPriority, InvalidTransaction, TransactionValidity, TransactionValidityError, ValidTransaction,43 },44 FixedPointOperand, FixedU128,45};46use sp_runtime::traits::StaticLookup;47use pallet_contracts::chain_extension::UncheckedFrom;48use pallet_transaction_payment::OnChargeTransaction;4950#[cfg(test)]51mod mock;5253#[cfg(test)]54mod tests;5556mod default_weights;5758pub const MAX_DECIMAL_POINTS: DecimalPoints = 30;59pub const MAX_REFUNGIBLE_PIECES: u128 = 1_000_000_000_000_000_000_000;60pub const MAX_SPONSOR_TIMEOUT: u32 = 10_368_000;61pub const MAX_TOKEN_OWNERSHIP: u32 = 10_000_000;6263646566pub type CollectionId = u32;67pub type TokenId = u32;68pub type DecimalPoints = u8;6970#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq)]71#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]72pub enum CollectionMode {73 Invalid,74 NFT,75 76 Fungible(DecimalPoints),77 ReFungible,78}7980impl Default for CollectionMode {81 fn default() -> Self {82 Self::Invalid83 }84}8586impl Into<u8> for CollectionMode {87 fn into(self) -> u8 {88 match self {89 CollectionMode::Invalid => 0,90 CollectionMode::NFT => 1,91 CollectionMode::Fungible(_) => 2,92 CollectionMode::ReFungible => 3,93 }94 }95}9697#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq)]98#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]99pub enum AccessMode {100 Normal,101 WhiteList,102}103impl Default for AccessMode {104 fn default() -> Self {105 Self::Normal106 }107}108109#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq)]110#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]111pub enum SchemaVersion {112 ImageURL,113 Unique,114}115impl Default for SchemaVersion {116 fn default() -> Self {117 Self::ImageURL118 }119}120121#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]122#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]123pub struct Ownership<AccountId> {124 pub owner: AccountId,125 pub fraction: u128,126}127128#[derive(Encode, Decode, Debug, Clone, PartialEq)]129#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]130pub enum SponsorshipState<AccountId> {131 132 Disabled,133 Unconfirmed(AccountId),134 135 Confirmed(AccountId),136}137138impl<AccountId> SponsorshipState<AccountId> {139 fn sponsor(&self) -> Option<&AccountId> {140 match self {141 Self::Confirmed(sponsor) => Some(sponsor),142 _ => None,143 }144 }145146 fn pending_sponsor(&self) -> Option<&AccountId> {147 match self {148 Self::Unconfirmed(sponsor) | Self::Confirmed(sponsor) => Some(sponsor),149 _ => None,150 }151 }152153 fn confirmed(&self) -> bool {154 matches!(self, Self::Confirmed(_))155 }156}157158impl<T> Default for SponsorshipState<T> {159 fn default() -> Self {160 Self::Disabled161 }162}163164#[derive(Encode, Decode, Clone, PartialEq)]165#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]166pub struct Collection<T: Config> {167 pub owner: T::AccountId,168 pub mode: CollectionMode,169 pub access: AccessMode,170 pub decimal_points: DecimalPoints,171 pub name: Vec<u16>, 172 pub description: Vec<u16>, 173 pub token_prefix: Vec<u8>, 174 pub mint_mode: bool,175 pub offchain_schema: Vec<u8>,176 pub schema_version: SchemaVersion,177 pub sponsorship: SponsorshipState<T::AccountId>,178 pub limits: CollectionLimits<T::BlockNumber>, 179 pub variable_on_chain_schema: Vec<u8>, 180 pub const_on_chain_schema: Vec<u8>, 181}182183pub struct CollectionHandle<T: Config> {184 pub id: CollectionId,185 collection: Collection<T>,186}187188impl<T: Config> Deref for CollectionHandle<T> {189 type Target = Collection<T>;190191 fn deref(&self) -> &Self::Target {192 &self.collection193 }194}195196impl<T: Config> DerefMut for CollectionHandle<T> {197 fn deref_mut(&mut self) -> &mut Self::Target {198 &mut self.collection199 }200}201202#[derive(Encode, Decode, Debug, Clone, PartialEq)]203#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]204pub struct NftItemType<AccountId> {205 pub owner: AccountId,206 pub const_data: Vec<u8>,207 pub variable_data: Vec<u8>,208}209210#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]211#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]212pub struct FungibleItemType {213 pub value: u128,214}215216#[derive(Encode, Decode, Debug, Clone, PartialEq)]217#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]218pub struct ReFungibleItemType<AccountId> {219 pub owner: Vec<Ownership<AccountId>>,220 pub const_data: Vec<u8>,221 pub variable_data: Vec<u8>,222}223224225226227228229230231232233234235#[derive(Encode, Decode, Debug, Clone, PartialEq)]236#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]237pub struct CollectionLimits<BlockNumber: Encode + Decode> {238 pub account_token_ownership_limit: u32,239 pub sponsored_data_size: u32,240 241 242 243 pub sponsored_data_rate_limit: Option<BlockNumber>,244 pub token_limit: u32,245246 247 pub sponsor_transfer_timeout: u32,248 pub owner_can_transfer: bool,249 pub owner_can_destroy: bool,250}251252impl<BlockNumber: Encode + Decode> Default for CollectionLimits<BlockNumber> {253 fn default() -> Self {254 Self { 255 account_token_ownership_limit: 10_000_000, 256 token_limit: u32::max_value(),257 sponsored_data_size: u32::MAX, 258 sponsored_data_rate_limit: None,259 sponsor_transfer_timeout: 14400,260 owner_can_transfer: true,261 owner_can_destroy: true262 }263 }264}265266#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]267#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]268pub struct ChainLimits {269 pub collection_numbers_limit: u32,270 pub account_token_ownership_limit: u32,271 pub collections_admins_limit: u64,272 pub custom_data_limit: u32,273274 275 pub nft_sponsor_transfer_timeout: u32,276 pub fungible_sponsor_transfer_timeout: u32,277 pub refungible_sponsor_transfer_timeout: u32,278279 280 pub offchain_schema_limit: u32,281 pub variable_on_chain_schema_limit: u32,282 pub const_on_chain_schema_limit: u32,283}284285pub trait WeightInfo {286 fn create_collection() -> Weight;287 fn destroy_collection() -> Weight;288 fn add_to_white_list() -> Weight;289 fn remove_from_white_list() -> Weight;290 fn set_public_access_mode() -> Weight;291 fn set_mint_permission() -> Weight;292 fn change_collection_owner() -> Weight;293 fn add_collection_admin() -> Weight;294 fn remove_collection_admin() -> Weight;295 fn set_collection_sponsor() -> Weight;296 fn confirm_sponsorship() -> Weight;297 fn remove_collection_sponsor() -> Weight;298 fn create_item(s: usize) -> Weight;299 fn burn_item() -> Weight;300 fn transfer() -> Weight;301 fn approve() -> Weight;302 fn transfer_from() -> Weight;303 fn set_offchain_schema() -> Weight;304 fn set_const_on_chain_schema() -> Weight;305 fn set_variable_on_chain_schema() -> Weight;306 fn set_variable_meta_data() -> Weight;307 fn enable_contract_sponsoring() -> Weight;308 fn set_schema_version() -> Weight;309 fn set_chain_limits() -> Weight;310 fn set_contract_sponsoring_rate_limit() -> Weight;311 fn set_variable_meta_data_sponsoring_rate_limit() -> Weight;312 fn toggle_contract_white_list() -> Weight;313 fn add_to_contract_white_list() -> Weight;314 fn remove_from_contract_white_list() -> Weight;315 fn set_collection_limits() -> Weight;316}317318#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]319#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]320pub struct CreateNftData {321 pub const_data: Vec<u8>,322 pub variable_data: Vec<u8>,323}324325#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]326#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]327pub struct CreateFungibleData {328 pub value: u128,329}330331#[derive(Encode, Decode, Default, Debug, Clone, PartialEq)]332#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]333pub struct CreateReFungibleData {334 pub const_data: Vec<u8>,335 pub variable_data: Vec<u8>,336 pub pieces: u128,337}338339#[derive(Encode, Decode, Debug, Clone, PartialEq)]340#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]341pub enum CreateItemData {342 NFT(CreateNftData),343 Fungible(CreateFungibleData),344 ReFungible(CreateReFungibleData),345}346347impl CreateItemData {348 pub fn len(&self) -> usize {349 let len = match self {350 CreateItemData::NFT(data) => data.variable_data.len() + data.const_data.len(),351 CreateItemData::ReFungible(data) => data.variable_data.len() + data.const_data.len(),352 _ => 0353 };354 355 return len;356 }357}358359impl From<CreateNftData> for CreateItemData {360 fn from(item: CreateNftData) -> Self {361 CreateItemData::NFT(item)362 }363}364365impl From<CreateReFungibleData> for CreateItemData {366 fn from(item: CreateReFungibleData) -> Self {367 CreateItemData::ReFungible(item)368 }369}370371impl From<CreateFungibleData> for CreateItemData {372 fn from(item: CreateFungibleData) -> Self {373 CreateItemData::Fungible(item)374 }375}376377378decl_error! {379 380 pub enum Error for Module<T: Config> {381 382 TotalCollectionsLimitExceeded,383 384 CollectionDecimalPointLimitExceeded, 385 386 CollectionNameLimitExceeded, 387 388 CollectionDescriptionLimitExceeded, 389 390 CollectionTokenPrefixLimitExceeded,391 392 CollectionNotFound,393 394 TokenNotFound,395 396 AdminNotFound,397 398 NumOverflow, 399 400 AlreadyAdmin, 401 402 NoPermission,403 404 ConfirmUnsetSponsorFail,405 406 PublicMintingNotAllowed,407 408 MustBeTokenOwner,409 410 TokenValueTooLow,411 412 NftSizeLimitExceeded,413 414 ApproveNotFound,415 416 TokenValueNotEnough,417 418 ApproveRequired,419 420 AddresNotInWhiteList,421 422 CollectionAdminsLimitExceeded,423 424 AddressOwnershipLimitExceeded,425 426 EmptyArgument,427 428 TokenConstDataLimitExceeded,429 430 TokenVariableDataLimitExceeded,431 432 NotNftDataUsedToMintNftCollectionToken,433 434 NotFungibleDataUsedToMintFungibleCollectionToken,435 436 NotReFungibleDataUsedToMintReFungibleCollectionToken,437 438 UnexpectedCollectionType,439 440 CantStoreMetadataInFungibleTokens,441 442 CollectionTokenLimitExceeded,443 444 AccountTokenLimitExceeded,445 446 CollectionLimitBoundsExceeded,447 448 OwnerPermissionsCantBeReverted,449 450 SchemaDataLimitExceeded,451 452 WrongRefungiblePieces,453 454 BadCreateRefungibleCall,455 }456}457458pub trait Config: system::Config + Sized + pallet_transaction_payment::Config + pallet_contracts::Config {459 type Event: From<Event<Self>> + Into<<Self as system::Config>::Event>;460461 462 type WeightInfo: WeightInfo;463464 type Currency: Currency<Self::AccountId>;465 type CollectionCreationPrice: Get<<<Self as Config>::Currency as Currency<Self::AccountId>>::Balance>;466 type TreasuryAccountId: Get<Self::AccountId>;467}468469#[cfg(feature = "runtime-benchmarks")]470mod benchmarking;471472473474475476477478479480481482483484485486487488489490491492493494495496decl_storage! {497 trait Store for Module<T: Config> as Nft {498499 500 501 CreatedCollectionCount: u32;502 503 ChainVersion: u64;504 505 506 ItemListIndex: map hasher(blake2_128_concat) CollectionId => TokenId;507 508509 510 pub ChainLimit get(fn chain_limit) config(): ChainLimits;511 512513 514 515 516 DestroyedCollectionCount: u32;517 518 519 pub AccountItemCount get(fn account_item_count): map hasher(twox_64_concat) T::AccountId => u32;520 521522 523 524 525 pub CollectionById get(fn collection_id) config(): map hasher(blake2_128_concat) CollectionId => Option<Collection<T>> = None;526 527 528 pub AdminList get(fn admin_list_collection): map hasher(blake2_128_concat) CollectionId => Vec<T::AccountId>;529 530 531 pub WhiteList get(fn white_list): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) T::AccountId => bool;532 533534 535 536 pub Balance get(fn balance_count): double_map hasher(blake2_128_concat) CollectionId, hasher(twox_64_concat) T::AccountId => u128;537538 539 540 541 pub Allowances get(fn approved): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) (TokenId, T::AccountId, T::AccountId) => u128;542543 544 545 pub NftItemList get(fn nft_item_id) config(): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) TokenId => Option<NftItemType<T::AccountId>>;546 547 pub FungibleItemList get(fn fungible_item_id) config(): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) T::AccountId => FungibleItemType;548 549 pub ReFungibleItemList get(fn refungible_item_id) config(): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) TokenId => Option<ReFungibleItemType<T::AccountId>>;550 551552 553 554 pub AddressTokens get(fn address_tokens): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) T::AccountId => Vec<TokenId>;555 556557 558 559 560 pub CreateItemBasket get(fn create_item_basket): map hasher(blake2_128_concat) (CollectionId, T::AccountId) => T::BlockNumber;561 562 pub NftTransferBasket get(fn nft_transfer_basket): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) TokenId => T::BlockNumber;563 564 pub FungibleTransferBasket get(fn fungible_transfer_basket): double_map hasher(blake2_128_concat) CollectionId, hasher(twox_64_concat) T::AccountId => T::BlockNumber;565 566 pub ReFungibleTransferBasket get(fn refungible_transfer_basket): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) TokenId => T::BlockNumber;567 568569 570 571 pub VariableMetaDataBasket get(fn variable_meta_data_basket): double_map hasher(blake2_128_concat) CollectionId, hasher(blake2_128_concat) TokenId => Option<T::BlockNumber> = None;572 573 574 575 pub ContractOwner get(fn contract_owner): map hasher(twox_64_concat) T::AccountId => Option<T::AccountId>;576 577 pub ContractSelfSponsoring get(fn contract_self_sponsoring): map hasher(twox_64_concat) T::AccountId => bool;578 579 pub ContractSponsorBasket get(fn contract_sponsor_basket): map hasher(twox_64_concat) (T::AccountId, T::AccountId) => T::BlockNumber;580 581 pub ContractSponsoringRateLimit get(fn contract_sponsoring_rate_limit): map hasher(twox_64_concat) T::AccountId => T::BlockNumber;582 583 pub ContractWhiteListEnabled get(fn contract_white_list_enabled): map hasher(twox_64_concat) T::AccountId => bool; 584 585 pub ContractWhiteList get(fn contract_white_list): double_map hasher(twox_64_concat) T::AccountId, hasher(blake2_128_concat) T::AccountId => bool; 586 587 }588 add_extra_genesis {589 build(|config: &GenesisConfig<T>| {590 591 for (_num, _c) in &config.collection_id {592 <Module<T>>::init_collection(_c);593 }594595 for (_num, _c, _i) in &config.nft_item_id {596 <Module<T>>::init_nft_token(*_c, _i);597 }598599 for (collection_id, account_id, fungible_item) in &config.fungible_item_id {600 <Module<T>>::init_fungible_token(*collection_id, account_id, fungible_item);601 }602603 for (_num, _c, _i) in &config.refungible_item_id {604 <Module<T>>::init_refungible_token(*_c, _i);605 }606 })607 }608}609610decl_event!(611 pub enum Event<T>612 where613 AccountId = <T as system::Config>::AccountId,614 {615 616 617 618 619 620 621 622 623 624 CollectionCreated(CollectionId, u8, AccountId),625626 627 628 629 630 631 632 633 634 635 ItemCreated(CollectionId, TokenId, AccountId),636637 638 639 640 641 642 643 644 ItemDestroyed(CollectionId, TokenId),645646 647 648 649 650 651 652 653 654 655 656 657 Transfer(CollectionId, TokenId, AccountId, AccountId, u128),658659 660 661 662 663 664 665 666 667 668 Approved(CollectionId, TokenId, AccountId, AccountId, u128),669 }670);671672decl_module! {673 pub struct Module<T: Config> for enum Call 674 where 675 origin: T::Origin676 {677 fn deposit_event() = default;678 type Error = Error<T>;679680 fn on_initialize(now: T::BlockNumber) -> Weight {681 0682 }683684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 #[weight = <T as Config>::WeightInfo::create_collection()]701 #[transactional]702 pub fn create_collection(origin,703 collection_name: Vec<u16>,704 collection_description: Vec<u16>,705 token_prefix: Vec<u8>,706 mode: CollectionMode) -> DispatchResult {707708 709 let who = ensure_signed(origin)?;710711 712 let mut imbalance = <<<T as Config>::Currency as Currency<T::AccountId>>::PositiveImbalance>::zero();713 imbalance.subsume(<<T as Config>::Currency as Currency<T::AccountId>>::deposit_creating(714 &T::TreasuryAccountId::get(),715 T::CollectionCreationPrice::get(),716 ));717 <T as Config>::Currency::settle(718 &who,719 imbalance,720 WithdrawReasons::TRANSFER,721 ExistenceRequirement::KeepAlive,722 ).map_err(|_| Error::<T>::NoPermission)?;723724 let decimal_points = match mode {725 CollectionMode::Fungible(points) => points,726 _ => 0727 };728729 let chain_limit = ChainLimit::get();730731 let created_count = CreatedCollectionCount::get();732 let destroyed_count = DestroyedCollectionCount::get();733734 735 ensure!(created_count - destroyed_count < chain_limit.collection_numbers_limit, Error::<T>::TotalCollectionsLimitExceeded);736737 738 ensure!(decimal_points <= MAX_DECIMAL_POINTS, Error::<T>::CollectionDecimalPointLimitExceeded);739 ensure!(collection_name.len() <= 64, Error::<T>::CollectionNameLimitExceeded);740 ensure!(collection_description.len() <= 256, Error::<T>::CollectionDescriptionLimitExceeded);741 ensure!(token_prefix.len() <= 16, Error::<T>::CollectionTokenPrefixLimitExceeded);742743 744 let next_id = created_count745 .checked_add(1)746 .ok_or(Error::<T>::NumOverflow)?;747748 CreatedCollectionCount::put(next_id);749750 let limits = CollectionLimits {751 sponsored_data_size: chain_limit.custom_data_limit,752 ..Default::default()753 };754755 756 let new_collection = Collection {757 owner: who.clone(),758 name: collection_name,759 mode: mode.clone(),760 mint_mode: false,761 access: AccessMode::Normal,762 description: collection_description,763 decimal_points: decimal_points,764 token_prefix: token_prefix,765 offchain_schema: Vec::new(),766 schema_version: SchemaVersion::ImageURL,767 sponsorship: SponsorshipState::Disabled,768 variable_on_chain_schema: Vec::new(),769 const_on_chain_schema: Vec::new(),770 limits,771 };772773 774 <CollectionById<T>>::insert(next_id, new_collection);775776 777 Self::deposit_event(RawEvent::CollectionCreated(next_id, mode.into(), who.clone()));778779 Ok(())780 }781782 783 784 785 786 787 788 789 790 791 #[weight = <T as Config>::WeightInfo::destroy_collection()]792 #[transactional]793 pub fn destroy_collection(origin, collection_id: CollectionId) -> DispatchResult {794795 let sender = ensure_signed(origin)?;796 let collection = Self::get_collection(collection_id)?;797 Self::check_owner_permissions(&collection, sender)?;798 if !collection.limits.owner_can_destroy {799 fail!(Error::<T>::NoPermission);800 }801802 <AddressTokens<T>>::remove_prefix(collection_id);803 <Allowances<T>>::remove_prefix(collection_id);804 <Balance<T>>::remove_prefix(collection_id);805 <ItemListIndex>::remove(collection_id);806 <AdminList<T>>::remove(collection_id);807 <CollectionById<T>>::remove(collection_id);808 <WhiteList<T>>::remove_prefix(collection_id);809810 <NftItemList<T>>::remove_prefix(collection_id);811 <FungibleItemList<T>>::remove_prefix(collection_id);812 <ReFungibleItemList<T>>::remove_prefix(collection_id);813814 <NftTransferBasket<T>>::remove_prefix(collection_id);815 <FungibleTransferBasket<T>>::remove_prefix(collection_id);816 <ReFungibleTransferBasket<T>>::remove_prefix(collection_id);817818 <VariableMetaDataBasket<T>>::remove_prefix(collection_id);819820 DestroyedCollectionCount::put(DestroyedCollectionCount::get()821 .checked_add(1)822 .ok_or(Error::<T>::NumOverflow)?);823824 Ok(())825 }826827 828 829 830 831 832 833 834 835 836 837 838 839 #[weight = <T as Config>::WeightInfo::add_to_white_list()]840 #[transactional]841 pub fn add_to_white_list(origin, collection_id: CollectionId, address: T::AccountId) -> DispatchResult{842843 let sender = ensure_signed(origin)?;844 let collection = Self::get_collection(collection_id)?;845 Self::check_owner_or_admin_permissions(&collection, sender)?;846847 <WhiteList<T>>::insert(collection_id, address, true);848 849 Ok(())850 }851852 853 854 855 856 857 858 859 860 861 862 863 864 #[weight = <T as Config>::WeightInfo::remove_from_white_list()]865 #[transactional]866 pub fn remove_from_white_list(origin, collection_id: CollectionId, address: T::AccountId) -> DispatchResult{867868 let sender = ensure_signed(origin)?;869 let collection = Self::get_collection(collection_id)?;870 Self::check_owner_or_admin_permissions(&collection, sender)?;871872 <WhiteList<T>>::remove(collection_id, address);873874 Ok(())875 }876877 878 879 880 881 882 883 884 885 886 887 888 #[weight = <T as Config>::WeightInfo::set_public_access_mode()]889 #[transactional]890 pub fn set_public_access_mode(origin, collection_id: CollectionId, mode: AccessMode) -> DispatchResult891 {892 let sender = ensure_signed(origin)?;893894 let mut target_collection = Self::get_collection(collection_id)?;895 Self::check_owner_permissions(&target_collection, sender)?;896 target_collection.access = mode;897 Self::save_collection(target_collection);898899 Ok(())900 }901902 903 904 905 906 907 908 909 910 911 912 913 914 915 #[weight = <T as Config>::WeightInfo::set_mint_permission()]916 #[transactional]917 pub fn set_mint_permission(origin, collection_id: CollectionId, mint_permission: bool) -> DispatchResult918 {919 let sender = ensure_signed(origin)?;920921 let mut target_collection = Self::get_collection(collection_id)?;922 Self::check_owner_permissions(&target_collection, sender)?;923 target_collection.mint_mode = mint_permission;924 Self::save_collection(target_collection);925926 Ok(())927 }928929 930 931 932 933 934 935 936 937 938 939 940 #[weight = <T as Config>::WeightInfo::change_collection_owner()]941 #[transactional]942 pub fn change_collection_owner(origin, collection_id: CollectionId, new_owner: T::AccountId) -> DispatchResult {943944 let sender = ensure_signed(origin)?;945 let mut target_collection = Self::get_collection(collection_id)?;946 Self::check_owner_permissions(&target_collection, sender)?;947 target_collection.owner = new_owner;948 Self::save_collection(target_collection);949950 Ok(())951 }952953 954 955 956 957 958 959 960 961 962 963 964 965 966 #[weight = <T as Config>::WeightInfo::add_collection_admin()]967 #[transactional]968 pub fn add_collection_admin(origin, collection_id: CollectionId, new_admin_id: T::AccountId) -> DispatchResult {969970 let sender = ensure_signed(origin)?;971 let collection = Self::get_collection(collection_id)?;972 Self::check_owner_or_admin_permissions(&collection, sender)?;973 let mut admin_arr = <AdminList<T>>::get(collection_id);974975 match admin_arr.binary_search(&new_admin_id) {976 Ok(_) => {},977 Err(idx) => {978 let limits = ChainLimit::get();979 ensure!(admin_arr.len() < limits.collections_admins_limit as usize, Error::<T>::CollectionAdminsLimitExceeded);980 admin_arr.insert(idx, new_admin_id);981 <AdminList<T>>::insert(collection_id, admin_arr);982 }983 }984 Ok(())985 }986987 988 989 990 991 992 993 994 995 996 997 998 999 #[weight = <T as Config>::WeightInfo::remove_collection_admin()]1000 #[transactional]1001 pub fn remove_collection_admin(origin, collection_id: CollectionId, account_id: T::AccountId) -> DispatchResult {10021003 let sender = ensure_signed(origin)?;1004 let collection = Self::get_collection(collection_id)?;1005 Self::check_owner_or_admin_permissions(&collection, sender)?;1006 let mut admin_arr = <AdminList<T>>::get(collection_id);10071008 match admin_arr.binary_search(&account_id) {1009 Ok(idx) => {1010 admin_arr.remove(idx);1011 <AdminList<T>>::insert(collection_id, admin_arr);1012 },1013 Err(_) => {}1014 }1015 Ok(())1016 }10171018 1019 1020 1021 1022 1023 1024 1025 1026 1027 #[weight = <T as Config>::WeightInfo::set_collection_sponsor()]1028 #[transactional]1029 pub fn set_collection_sponsor(origin, collection_id: CollectionId, new_sponsor: T::AccountId) -> DispatchResult {10301031 let sender = ensure_signed(origin)?;1032 let mut target_collection = Self::get_collection(collection_id)?;1033 Self::check_owner_permissions(&target_collection, sender)?;10341035 target_collection.sponsorship = SponsorshipState::Unconfirmed(new_sponsor);1036 Self::save_collection(target_collection);10371038 Ok(())1039 }10401041 1042 1043 1044 1045 1046 1047 1048 #[weight = <T as Config>::WeightInfo::confirm_sponsorship()]1049 #[transactional]1050 pub fn confirm_sponsorship(origin, collection_id: CollectionId) -> DispatchResult {10511052 let sender = ensure_signed(origin)?;10531054 let mut target_collection = Self::get_collection(collection_id)?;1055 ensure!(1056 target_collection.sponsorship.pending_sponsor() == Some(&sender),1057 Error::<T>::ConfirmUnsetSponsorFail1058 );10591060 target_collection.sponsorship = SponsorshipState::Confirmed(sender);1061 Self::save_collection(target_collection);10621063 Ok(())1064 }10651066 1067 1068 1069 1070 1071 1072 1073 1074 1075 #[weight = <T as Config>::WeightInfo::remove_collection_sponsor()]1076 #[transactional]1077 pub fn remove_collection_sponsor(origin, collection_id: CollectionId) -> DispatchResult {10781079 let sender = ensure_signed(origin)?;10801081 let mut target_collection = Self::get_collection(collection_id)?;1082 Self::check_owner_permissions(&target_collection, sender)?;10831084 target_collection.sponsorship = SponsorshipState::Disabled;1085 Self::save_collection(target_collection);10861087 Ok(())1088 }10891090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 11131114 #[weight = <T as Config>::WeightInfo::create_item(data.len())]1115 #[transactional]1116 pub fn create_item(origin, collection_id: CollectionId, owner: T::AccountId, data: CreateItemData) -> DispatchResult {1117 let sender = ensure_signed(origin)?;1118 Self::create_item_internal(sender, collection_id, owner, data)1119 }11201121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 #[weight = <T as Config>::WeightInfo::create_item(items_data.into_iter()1140 .map(|data| { data.len() })1141 .sum())]1142 #[transactional]1143 pub fn create_multiple_items(origin, collection_id: CollectionId, owner: T::AccountId, items_data: Vec<CreateItemData>) -> DispatchResult {11441145 ensure!(items_data.len() > 0, Error::<T>::EmptyArgument);1146 let sender = ensure_signed(origin)?;11471148 let target_collection = Self::get_collection(collection_id)?;11491150 Self::can_create_items_in_collection(&target_collection, &sender, &owner, items_data.len() as u32)?;11511152 for data in &items_data {1153 Self::validate_create_item_args(&target_collection, data)?;1154 }1155 for data in &items_data {1156 Self::create_item_no_validation(&target_collection, owner.clone(), data.clone())?;1157 }11581159 Ok(())1160 }11611162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 #[weight = <T as Config>::WeightInfo::burn_item()]1176 #[transactional]1177 pub fn burn_item(origin, collection_id: CollectionId, item_id: TokenId, value: u128) -> DispatchResult {11781179 let sender = ensure_signed(origin)?;11801181 1182 let target_collection = Self::get_collection(collection_id)?;1183 ensure!(1184 Self::is_item_owner(sender.clone(), &target_collection, item_id) ||1185 (1186 target_collection.limits.owner_can_transfer &&1187 Self::is_owner_or_admin_permissions(&target_collection, sender.clone())1188 ),1189 Error::<T>::NoPermission1190 );11911192 if target_collection.access == AccessMode::WhiteList {1193 Self::check_white_list(&target_collection, &sender)?;1194 }11951196 match target_collection.mode1197 {1198 CollectionMode::NFT => Self::burn_nft_item(&target_collection, item_id)?,1199 CollectionMode::Fungible(_) => Self::burn_fungible_item(&sender, &target_collection, value)?,1200 CollectionMode::ReFungible => Self::burn_refungible_item(&target_collection, item_id, &sender)?,1201 _ => ()1202 };12031204 1205 Self::deposit_event(RawEvent::ItemDestroyed(target_collection.id, item_id));12061207 Ok(())1208 }12091210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 #[weight = <T as Config>::WeightInfo::transfer()]1234 #[transactional]1235 pub fn transfer(origin, recipient: T::AccountId, collection_id: CollectionId, item_id: TokenId, value: u128) -> DispatchResult {1236 let sender = ensure_signed(origin)?;1237 let collection = Self::get_collection(collection_id)?;12381239 Self::transfer_internal(sender, recipient, &collection, item_id, value)1240 }12411242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 #[weight = <T as Config>::WeightInfo::approve()]1258 #[transactional]1259 pub fn approve(origin, spender: T::AccountId, collection_id: CollectionId, item_id: TokenId, amount: u128) -> DispatchResult {12601261 let sender = ensure_signed(origin)?;1262 let collection = Self::get_collection(collection_id)?;12631264 Self::approve_internal(sender, spender, &collection, item_id, amount)?;12651266 Ok(())1267 }1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 #[weight = <T as Config>::WeightInfo::transfer_from()]1289 #[transactional]1290 pub fn transfer_from(origin, from: T::AccountId, recipient: T::AccountId, collection_id: CollectionId, item_id: TokenId, value: u128 ) -> DispatchResult {12911292 let sender = ensure_signed(origin)?;1293 let collection = Self::get_collection(collection_id)?;12941295 Self::transfer_from_internal(sender, from, recipient, &collection, item_id, value)?;12961297 Ok(())1298 }12991300 1301 13021303 1304 1305 1306 13071308 13091310 13111312 1313 13141315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 #[weight = <T as Config>::WeightInfo::set_variable_meta_data()]1328 #[transactional]1329 pub fn set_variable_meta_data (1330 origin,1331 collection_id: CollectionId,1332 item_id: TokenId,1333 data: Vec<u8>1334 ) -> DispatchResult {1335 let sender = ensure_signed(origin)?;1336 1337 let target_collection = Self::get_collection(collection_id)?;1338 Self::token_exists(&target_collection, item_id)?;13391340 ensure!(ChainLimit::get().custom_data_limit >= data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);13411342 1343 ensure!(Self::is_item_owner(sender.clone(), &target_collection, item_id) ||1344 Self::is_owner_or_admin_permissions(&target_collection, sender.clone()),1345 Error::<T>::NoPermission);13461347 match target_collection.mode1348 {1349 CollectionMode::NFT => Self::set_nft_variable_data(&target_collection, item_id, data)?,1350 CollectionMode::ReFungible => Self::set_re_fungible_variable_data(&target_collection, item_id, data)?,1351 CollectionMode::Fungible(_) => fail!(Error::<T>::CantStoreMetadataInFungibleTokens),1352 _ => fail!(Error::<T>::UnexpectedCollectionType)1353 };13541355 Ok(())1356 }1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 #[weight = <T as Config>::WeightInfo::set_schema_version()]1373 #[transactional]1374 pub fn set_schema_version(1375 origin,1376 collection_id: CollectionId,1377 version: SchemaVersion1378 ) -> DispatchResult {1379 let sender = ensure_signed(origin)?;1380 let mut target_collection = Self::get_collection(collection_id)?;1381 Self::check_owner_or_admin_permissions(&target_collection, sender.clone())?;1382 target_collection.schema_version = version;1383 Self::save_collection(target_collection);13841385 Ok(())1386 }13871388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 #[weight = <T as Config>::WeightInfo::set_offchain_schema()]1401 #[transactional]1402 pub fn set_offchain_schema(1403 origin,1404 collection_id: CollectionId,1405 schema: Vec<u8>1406 ) -> DispatchResult {1407 let sender = ensure_signed(origin)?;1408 let mut target_collection = Self::get_collection(collection_id)?;1409 Self::check_owner_or_admin_permissions(&target_collection, sender.clone())?;14101411 1412 ensure!(schema.len() as u32 <= ChainLimit::get().offchain_schema_limit, "");14131414 target_collection.offchain_schema = schema;1415 Self::save_collection(target_collection);14161417 Ok(())1418 }14191420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 #[weight = <T as Config>::WeightInfo::set_const_on_chain_schema()]1433 #[transactional]1434 pub fn set_const_on_chain_schema (1435 origin,1436 collection_id: CollectionId,1437 schema: Vec<u8>1438 ) -> DispatchResult {1439 let sender = ensure_signed(origin)?;1440 let mut target_collection = Self::get_collection(collection_id)?;1441 Self::check_owner_or_admin_permissions(&target_collection, sender.clone())?;14421443 1444 ensure!(schema.len() as u32 <= ChainLimit::get().const_on_chain_schema_limit, "");14451446 target_collection.const_on_chain_schema = schema;1447 Self::save_collection(target_collection);14481449 Ok(())1450 }14511452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 #[weight = <T as Config>::WeightInfo::set_const_on_chain_schema()]1465 #[transactional]1466 pub fn set_variable_on_chain_schema (1467 origin,1468 collection_id: CollectionId,1469 schema: Vec<u8>1470 ) -> DispatchResult {1471 let sender = ensure_signed(origin)?;1472 let mut target_collection = Self::get_collection(collection_id)?;1473 Self::check_owner_or_admin_permissions(&target_collection, sender.clone())?;14741475 1476 ensure!(schema.len() as u32 <= ChainLimit::get().variable_on_chain_schema_limit, "");14771478 target_collection.variable_on_chain_schema = schema;1479 Self::save_collection(target_collection);14801481 Ok(())1482 }14831484 1485 #[weight = <T as Config>::WeightInfo::set_chain_limits()]1486 #[transactional]1487 pub fn set_chain_limits(1488 origin,1489 limits: ChainLimits1490 ) -> DispatchResult {14911492 #[cfg(not(feature = "runtime-benchmarks"))]1493 ensure_root(origin)?;14941495 <ChainLimit>::put(limits);1496 Ok(())1497 }14981499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 #[weight = <T as Config>::WeightInfo::enable_contract_sponsoring()]1511 #[transactional]1512 pub fn enable_contract_sponsoring(1513 origin,1514 contract_address: T::AccountId,1515 enable: bool1516 ) -> DispatchResult {15171518 let sender = ensure_signed(origin)?;15191520 #[cfg(feature = "runtime-benchmarks")]1521 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());15221523 Self::ensure_contract_owned(sender, &contract_address)?;15241525 <ContractSelfSponsoring<T>>::insert(contract_address, enable);1526 Ok(())1527 }15281529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 #[weight = <T as Config>::WeightInfo::set_contract_sponsoring_rate_limit()]1547 #[transactional]1548 pub fn set_contract_sponsoring_rate_limit(1549 origin,1550 contract_address: T::AccountId,1551 rate_limit: T::BlockNumber1552 ) -> DispatchResult {1553 let sender = ensure_signed(origin)?;15541555 #[cfg(feature = "runtime-benchmarks")]1556 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());15571558 Self::ensure_contract_owned(sender, &contract_address)?;1559 <ContractSponsoringRateLimit<T>>::insert(contract_address, rate_limit);1560 Ok(())1561 }15621563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 #[weight = <T as Config>::WeightInfo::toggle_contract_white_list()]1575 #[transactional]1576 pub fn toggle_contract_white_list(1577 origin,1578 contract_address: T::AccountId,1579 enable: bool1580 ) -> DispatchResult {1581 let sender = ensure_signed(origin)?;15821583 #[cfg(feature = "runtime-benchmarks")]1584 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());15851586 Self::ensure_contract_owned(sender, &contract_address)?;1587 if enable {1588 <ContractWhiteListEnabled<T>>::insert(contract_address, true);1589 } else {1590 <ContractWhiteListEnabled<T>>::remove(contract_address);1591 }1592 Ok(())1593 }1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 #[weight = <T as Config>::WeightInfo::add_to_contract_white_list()]1607 #[transactional]1608 pub fn add_to_contract_white_list(1609 origin,1610 contract_address: T::AccountId,1611 account_address: T::AccountId1612 ) -> DispatchResult {1613 let sender = ensure_signed(origin)?;16141615 #[cfg(feature = "runtime-benchmarks")]1616 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());1617 1618 Self::ensure_contract_owned(sender, &contract_address)?; 1619 <ContractWhiteList<T>>::insert(contract_address, account_address, true);1620 Ok(())1621 }16221623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 #[weight = <T as Config>::WeightInfo::remove_from_contract_white_list()]1635 #[transactional]1636 pub fn remove_from_contract_white_list(1637 origin,1638 contract_address: T::AccountId,1639 account_address: T::AccountId1640 ) -> DispatchResult {1641 let sender = ensure_signed(origin)?;16421643 #[cfg(feature = "runtime-benchmarks")]1644 <ContractOwner<T>>::insert(contract_address.clone(), sender.clone());16451646 Self::ensure_contract_owned(sender, &contract_address)?;1647 <ContractWhiteList<T>>::remove(contract_address, account_address);1648 Ok(())1649 }16501651 #[weight = <T as Config>::WeightInfo::set_collection_limits()]1652 #[transactional]1653 pub fn set_collection_limits(1654 origin,1655 collection_id: u32,1656 new_limits: CollectionLimits<T::BlockNumber>,1657 ) -> DispatchResult {1658 let sender = ensure_signed(origin)?;1659 let mut target_collection = Self::get_collection(collection_id)?;1660 Self::check_owner_permissions(&target_collection, sender.clone())?;1661 let old_limits = &target_collection.limits;1662 let chain_limits = ChainLimit::get();16631664 1665 ensure!(new_limits.sponsor_transfer_timeout <= MAX_SPONSOR_TIMEOUT &&1666 new_limits.account_token_ownership_limit <= MAX_TOKEN_OWNERSHIP && 1667 new_limits.sponsored_data_size <= chain_limits.custom_data_limit,1668 Error::<T>::CollectionLimitBoundsExceeded);16691670 1671 ensure!(old_limits.token_limit >= new_limits.token_limit, Error::<T>::CollectionTokenLimitExceeded);1672 ensure!(new_limits.token_limit > 0, Error::<T>::CollectionTokenLimitExceeded);16731674 ensure!(1675 (old_limits.owner_can_transfer || !new_limits.owner_can_transfer) &&1676 (old_limits.owner_can_destroy || !new_limits.owner_can_destroy),1677 Error::<T>::OwnerPermissionsCantBeReverted,1678 );16791680 target_collection.limits = new_limits;1681 Self::save_collection(target_collection);16821683 Ok(())1684 } 1685 }1686}16871688impl<T: Config> Module<T> {1689 pub fn create_item_internal(sender: T::AccountId, collection_id: CollectionId, owner: T::AccountId, data: CreateItemData) -> DispatchResult {1690 let target_collection = Self::get_collection(collection_id)?;16911692 Self::can_create_items_in_collection(&target_collection, &sender, &owner, 1)?;1693 Self::validate_create_item_args(&target_collection, &data)?;1694 Self::create_item_no_validation(&target_collection, owner, data)?;16951696 Ok(())1697 }16981699 pub fn transfer_internal(sender: T::AccountId, recipient: T::AccountId, target_collection: &CollectionHandle<T>, item_id: TokenId, value: u128) -> DispatchResult {1700 1701 Self::is_correct_transfer(target_collection, &recipient)?;17021703 1704 ensure!(Self::is_item_owner(sender.clone(), target_collection, item_id) ||1705 Self::is_owner_or_admin_permissions(target_collection, sender.clone()),1706 Error::<T>::NoPermission);17071708 if target_collection.access == AccessMode::WhiteList {1709 Self::check_white_list(target_collection, &sender)?;1710 Self::check_white_list(target_collection, &recipient)?;1711 }17121713 match target_collection.mode1714 {1715 CollectionMode::NFT => Self::transfer_nft(target_collection, item_id, sender.clone(), recipient.clone())?,1716 CollectionMode::Fungible(_) => Self::transfer_fungible(target_collection, value, &sender, &recipient)?,1717 CollectionMode::ReFungible => Self::transfer_refungible(target_collection, item_id, value, sender.clone(), recipient.clone())?,1718 _ => ()1719 };17201721 Self::deposit_event(RawEvent::Transfer(target_collection.id, item_id, sender, recipient, value));17221723 Ok(())1724 }17251726 pub fn approve_internal(1727 sender: T::AccountId,1728 spender: T::AccountId,1729 collection: &CollectionHandle<T>,1730 item_id: TokenId,1731 amount: u128,1732 ) -> DispatchResult {1733 Self::token_exists(&collection, item_id)?;17341735 1736 let bypasses_limits = collection.limits.owner_can_transfer &&1737 Self::is_owner_or_admin_permissions(1738 &collection,1739 sender.clone(),1740 );17411742 let allowance_limit = if bypasses_limits {1743 None1744 } else if let Some(amount) = Self::owned_amount(1745 sender.clone(),1746 &collection,1747 item_id,1748 ) {1749 Some(amount)1750 } else {1751 fail!(Error::<T>::NoPermission);1752 };17531754 if collection.access == AccessMode::WhiteList {1755 Self::check_white_list(&collection, &sender)?;1756 Self::check_white_list(&collection, &spender)?;1757 }17581759 let allowance: u128 = amount1760 .checked_add(<Allowances<T>>::get(collection.id, (item_id, &sender, &spender)))1761 .ok_or(Error::<T>::NumOverflow)?;1762 if let Some(limit) = allowance_limit {1763 ensure!(limit >= allowance, Error::<T>::TokenValueTooLow);1764 }1765 <Allowances<T>>::insert(collection.id, (item_id, &sender, &spender), allowance);17661767 Self::deposit_event(RawEvent::Approved(collection.id, item_id, sender, spender, allowance));1768 Ok(())1769 }17701771 pub fn transfer_from_internal(1772 sender: T::AccountId,1773 from: T::AccountId,1774 recipient: T::AccountId,1775 collection: &CollectionHandle<T>,1776 item_id: TokenId,1777 amount: u128,1778 ) -> DispatchResult {1779 1780 let approval: u128 = <Allowances<T>>::get(collection.id, (item_id, &from, &sender));17811782 1783 Self::is_correct_transfer(&collection, &recipient)?;17841785 1786 ensure!(1787 approval >= amount || 1788 (1789 collection.limits.owner_can_transfer &&1790 Self::is_owner_or_admin_permissions(&collection, sender.clone())1791 ),1792 Error::<T>::NoPermission1793 );17941795 if collection.access == AccessMode::WhiteList {1796 Self::check_white_list(&collection, &sender)?;1797 Self::check_white_list(&collection, &recipient)?;1798 }17991800 1801 let allowance = approval.saturating_sub(amount);1802 if allowance > 0 {1803 <Allowances<T>>::insert(collection.id, (item_id, &from, &sender), allowance);1804 } else {1805 <Allowances<T>>::remove(collection.id, (item_id, &from, &sender));1806 }18071808 match collection.mode {1809 CollectionMode::NFT => {1810 Self::transfer_nft(&collection, item_id, from.clone(), recipient.clone())?1811 }1812 CollectionMode::Fungible(_) => {1813 Self::transfer_fungible(&collection, amount, &from, &recipient)?1814 }1815 CollectionMode::ReFungible => {1816 Self::transfer_refungible(&collection, item_id, amount, from.clone(), recipient.clone())?1817 }1818 _ => ()1819 };18201821 Ok(())1822 }18231824 pub fn create_multiple_items_internal(1825 sender: T::AccountId,1826 collection: &CollectionHandle<T>,1827 owner: T::AccountId,1828 items_data: Vec<CreateItemData>,1829 ) -> DispatchResult {1830 Self::can_create_items_in_collection(&collection, &sender, &owner, items_data.len() as u32)?;18311832 for data in &items_data {1833 Self::validate_create_item_args(&collection, data)?;1834 }1835 for data in &items_data {1836 Self::create_item_no_validation(&collection, owner.clone(), data.clone())?;1837 }18381839 Ok(())1840 }18411842 fn is_correct_transfer(collection: &CollectionHandle<T>, recipient: &T::AccountId) -> DispatchResult {1843 let collection_id = collection.id;18441845 1846 let account_items: u32 = <AddressTokens<T>>::get(collection_id, recipient).len() as u32;1847 ensure!(collection.limits.account_token_ownership_limit > account_items, Error::<T>::AccountTokenLimitExceeded);1848 1849 Ok(())1850 }18511852 fn can_create_items_in_collection(collection: &CollectionHandle<T>, sender: &T::AccountId, owner: &T::AccountId, amount: u32) -> DispatchResult {1853 let collection_id = collection.id;18541855 1856 let total_items: u32 = ItemListIndex::get(collection_id)1857 .checked_add(amount)1858 .ok_or(Error::<T>::CollectionTokenLimitExceeded)?;1859 let account_items: u32 = (<AddressTokens<T>>::get(collection_id, owner).len() as u32)1860 .checked_add(amount)1861 .ok_or(Error::<T>::AccountTokenLimitExceeded)?;1862 ensure!(collection.limits.token_limit >= total_items, Error::<T>::CollectionTokenLimitExceeded);1863 ensure!(collection.limits.account_token_ownership_limit >= account_items, Error::<T>::AccountTokenLimitExceeded);18641865 if !Self::is_owner_or_admin_permissions(collection, sender.clone()) {1866 ensure!(collection.mint_mode == true, Error::<T>::PublicMintingNotAllowed);1867 Self::check_white_list(collection, owner)?;1868 Self::check_white_list(collection, sender)?;1869 }18701871 Ok(())1872 }18731874 fn validate_create_item_args(target_collection: &CollectionHandle<T>, data: &CreateItemData) -> DispatchResult {1875 match target_collection.mode1876 {1877 CollectionMode::NFT => {1878 if let CreateItemData::NFT(data) = data {1879 1880 ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, Error::<T>::TokenConstDataLimitExceeded);1881 ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);1882 } else {1883 fail!(Error::<T>::NotNftDataUsedToMintNftCollectionToken);1884 }1885 },1886 CollectionMode::Fungible(_) => {1887 if let CreateItemData::Fungible(_) = data {1888 } else {1889 fail!(Error::<T>::NotFungibleDataUsedToMintFungibleCollectionToken);1890 }1891 },1892 CollectionMode::ReFungible => {1893 if let CreateItemData::ReFungible(data) = data {18941895 1896 ensure!(ChainLimit::get().custom_data_limit >= data.const_data.len() as u32, Error::<T>::TokenConstDataLimitExceeded);1897 ensure!(ChainLimit::get().custom_data_limit >= data.variable_data.len() as u32, Error::<T>::TokenVariableDataLimitExceeded);18981899 1900 ensure!(data.pieces <= MAX_REFUNGIBLE_PIECES, Error::<T>::WrongRefungiblePieces);1901 ensure!(data.pieces > 0, Error::<T>::WrongRefungiblePieces);1902 } else {1903 fail!(Error::<T>::NotReFungibleDataUsedToMintReFungibleCollectionToken);1904 }1905 },1906 _ => { fail!(Error::<T>::UnexpectedCollectionType); }1907 };19081909 Ok(())1910 }19111912 fn create_item_no_validation(collection: &CollectionHandle<T>, owner: T::AccountId, data: CreateItemData) -> DispatchResult {1913 match data1914 {1915 CreateItemData::NFT(data) => {1916 let item = NftItemType {1917 owner: owner.clone(),1918 const_data: data.const_data,1919 variable_data: data.variable_data1920 };19211922 Self::add_nft_item(collection, item)?;1923 },1924 CreateItemData::Fungible(data) => {1925 Self::add_fungible_item(collection, &owner, data.value)?;1926 },1927 CreateItemData::ReFungible(data) => {1928 let mut owner_list = Vec::new();1929 owner_list.push(Ownership {owner: owner.clone(), fraction: data.pieces});19301931 let item = ReFungibleItemType {1932 owner: owner_list,1933 const_data: data.const_data,1934 variable_data: data.variable_data1935 };19361937 Self::add_refungible_item(collection, item)?;1938 }1939 };19401941 Ok(())1942 }19431944 fn add_fungible_item(collection: &CollectionHandle<T>, owner: &T::AccountId, value: u128) -> DispatchResult {1945 let collection_id = collection.id;19461947 1948 let balance: u128 = <FungibleItemList<T>>::get(collection_id, owner).value;19491950 1951 let item = FungibleItemType {1952 value: balance.checked_add(value).ok_or(Error::<T>::NumOverflow)?,1953 };1954 <FungibleItemList<T>>::insert(collection_id, (*owner).clone(), item);19551956 1957 let new_balance = <Balance<T>>::get(collection_id, owner)1958 .checked_add(value)1959 .ok_or(Error::<T>::NumOverflow)?;1960 <Balance<T>>::insert(collection_id, (*owner).clone(), new_balance);19611962 Self::deposit_event(RawEvent::ItemCreated(collection_id, 0, owner.clone()));1963 Ok(())1964 }19651966 fn add_refungible_item(collection: &CollectionHandle<T>, item: ReFungibleItemType<T::AccountId>) -> DispatchResult {1967 let collection_id = collection.id;19681969 let current_index = <ItemListIndex>::get(collection_id)1970 .checked_add(1)1971 .ok_or(Error::<T>::NumOverflow)?;1972 let itemcopy = item.clone();19731974 ensure!(1975 item.owner.len() == 1,1976 Error::<T>::BadCreateRefungibleCall,1977 );1978 let item_owner = item.owner.first().expect("only one owner is defined");19791980 let value = item_owner.fraction;1981 let owner = item_owner.owner.clone();19821983 Self::add_token_index(collection_id, current_index, &owner)?;19841985 <ItemListIndex>::insert(collection_id, current_index);1986 <ReFungibleItemList<T>>::insert(collection_id, current_index, itemcopy);19871988 1989 let new_balance = <Balance<T>>::get(collection_id, &owner)1990 .checked_add(value)1991 .ok_or(Error::<T>::NumOverflow)?;1992 <Balance<T>>::insert(collection_id, owner.clone(), new_balance);19931994 Self::deposit_event(RawEvent::ItemCreated(collection_id, current_index, owner));1995 Ok(())1996 }19971998 fn add_nft_item(collection: &CollectionHandle<T>, item: NftItemType<T::AccountId>) -> DispatchResult {1999 let collection_id = collection.id;20002001 let current_index = <ItemListIndex>::get(collection_id)2002 .checked_add(1)2003 .ok_or(Error::<T>::NumOverflow)?;20042005 let item_owner = item.owner.clone();2006 Self::add_token_index(collection_id, current_index, &item.owner)?;20072008 <ItemListIndex>::insert(collection_id, current_index);2009 <NftItemList<T>>::insert(collection_id, current_index, item);20102011 2012 let new_balance = <Balance<T>>::get(collection_id, item_owner.clone())2013 .checked_add(1)2014 .ok_or(Error::<T>::NumOverflow)?;2015 <Balance<T>>::insert(collection_id, item_owner.clone(), new_balance);20162017 Self::deposit_event(RawEvent::ItemCreated(collection_id, current_index, item_owner));2018 Ok(())2019 }20202021 fn burn_refungible_item(2022 collection: &CollectionHandle<T>,2023 item_id: TokenId,2024 owner: &T::AccountId,2025 ) -> DispatchResult {2026 let collection_id = collection.id;20272028 let mut token = <ReFungibleItemList<T>>::get(collection_id, item_id)2029 .ok_or(Error::<T>::TokenNotFound)?;2030 let rft_balance = token2031 .owner2032 .iter()2033 .find(|&i| i.owner == *owner)2034 .ok_or(Error::<T>::TokenNotFound)?;2035 Self::remove_token_index(collection_id, item_id, owner)?;20362037 2038 let new_balance = <Balance<T>>::get(collection_id, rft_balance.owner.clone())2039 .checked_sub(rft_balance.fraction)2040 .ok_or(Error::<T>::NumOverflow)?;2041 <Balance<T>>::insert(collection_id, rft_balance.owner.clone(), new_balance);20422043 2044 let index = token2045 .owner2046 .iter()2047 .position(|i| i.owner == *owner)2048 .expect("owned item is exists");2049 token.owner.remove(index);2050 let owner_count = token.owner.len();20512052 2053 if owner_count == 0 {2054 <ReFungibleItemList<T>>::remove(collection_id, item_id);2055 <VariableMetaDataBasket<T>>::remove(collection_id, item_id);2056 }2057 else {2058 <ReFungibleItemList<T>>::insert(collection_id, item_id, token);2059 }20602061 Ok(())2062 }20632064 fn burn_nft_item(collection: &CollectionHandle<T>, item_id: TokenId) -> DispatchResult {2065 let collection_id = collection.id;20662067 let item = <NftItemList<T>>::get(collection_id, item_id)2068 .ok_or(Error::<T>::TokenNotFound)?;2069 Self::remove_token_index(collection_id, item_id, &item.owner)?;20702071 2072 let new_balance = <Balance<T>>::get(collection_id, &item.owner)2073 .checked_sub(1)2074 .ok_or(Error::<T>::NumOverflow)?;2075 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);2076 <NftItemList<T>>::remove(collection_id, item_id);2077 <VariableMetaDataBasket<T>>::remove(collection_id, item_id);20782079 Ok(())2080 }20812082 fn burn_fungible_item(owner: &T::AccountId, collection: &CollectionHandle<T>, value: u128) -> DispatchResult {2083 let collection_id = collection.id;20842085 let mut balance = <FungibleItemList<T>>::get(collection_id, owner);2086 ensure!(balance.value >= value, Error::<T>::TokenValueNotEnough);20872088 2089 let new_balance = <Balance<T>>::get(collection_id, owner)2090 .checked_sub(value)2091 .ok_or(Error::<T>::NumOverflow)?;2092 <Balance<T>>::insert(collection_id, (*owner).clone(), new_balance);20932094 if balance.value - value > 0 {2095 balance.value -= value;2096 <FungibleItemList<T>>::insert(collection_id, (*owner).clone(), balance);2097 }2098 else {2099 <FungibleItemList<T>>::remove(collection_id, owner);2100 }21012102 Ok(())2103 }21042105 pub fn get_collection(collection_id: CollectionId) -> Result<CollectionHandle<T>, sp_runtime::DispatchError> {2106 Ok(<CollectionById<T>>::get(collection_id)2107 .map(|collection| CollectionHandle {2108 id: collection_id,2109 collection2110 })2111 .ok_or(Error::<T>::CollectionNotFound)?)2112 }21132114 fn save_collection(collection: CollectionHandle<T>) {2115 <CollectionById<T>>::insert(collection.id, collection.collection);2116 }21172118 fn check_owner_permissions(target_collection: &CollectionHandle<T>, subject: T::AccountId) -> DispatchResult {2119 ensure!(2120 subject == target_collection.owner,2121 Error::<T>::NoPermission2122 );21232124 Ok(())2125 }21262127 fn is_owner_or_admin_permissions(collection: &CollectionHandle<T>, subject: T::AccountId) -> bool {2128 subject == collection.owner || <AdminList<T>>::get(collection.id).contains(&subject)2129 }21302131 fn check_owner_or_admin_permissions(2132 collection: &CollectionHandle<T>,2133 subject: T::AccountId,2134 ) -> DispatchResult {2135 ensure!(Self::is_owner_or_admin_permissions(collection, subject), Error::<T>::NoPermission);21362137 Ok(())2138 }21392140 fn owned_amount(2141 subject: T::AccountId,2142 target_collection: &CollectionHandle<T>,2143 item_id: TokenId,2144 ) -> Option<u128> {2145 let collection_id = target_collection.id;21462147 match target_collection.mode {2148 CollectionMode::NFT => (<NftItemList<T>>::get(collection_id, item_id)?.owner == subject)2149 .then(|| 1),2150 CollectionMode::Fungible(_) => Some(<FungibleItemList<T>>::get(collection_id, &subject)2151 .value),2152 CollectionMode::ReFungible => <ReFungibleItemList<T>>::get(collection_id, item_id)?2153 .owner2154 .iter()2155 .find(|i| i.owner == subject)2156 .map(|i| i.fraction),2157 CollectionMode::Invalid => None,2158 }2159 }21602161 fn is_item_owner(subject: T::AccountId, target_collection: &CollectionHandle<T>, item_id: TokenId) -> bool {2162 match target_collection.mode {2163 CollectionMode::Fungible(_) => true,2164 _ => Self::owned_amount(subject, target_collection, item_id).is_some(),2165 }2166 }21672168 fn check_white_list(collection: &CollectionHandle<T>, address: &T::AccountId) -> DispatchResult {2169 let collection_id = collection.id;21702171 let mes = Error::<T>::AddresNotInWhiteList;2172 ensure!(<WhiteList<T>>::contains_key(collection_id, address), mes);21732174 Ok(())2175 }21762177 2178 2179 fn token_exists(2180 target_collection: &CollectionHandle<T>,2181 item_id: TokenId,2182 ) -> DispatchResult {2183 let collection_id = target_collection.id;2184 let exists = match target_collection.mode2185 {2186 CollectionMode::NFT => <NftItemList<T>>::contains_key(collection_id, item_id),2187 CollectionMode::Fungible(_) => true,2188 CollectionMode::ReFungible => <ReFungibleItemList<T>>::contains_key(collection_id, item_id),2189 _ => false2190 };21912192 ensure!(exists == true, Error::<T>::TokenNotFound);2193 Ok(())2194 }21952196 fn transfer_fungible(2197 collection: &CollectionHandle<T>,2198 value: u128,2199 owner: &T::AccountId,2200 recipient: &T::AccountId,2201 ) -> DispatchResult {2202 let collection_id = collection.id;22032204 let mut balance = <FungibleItemList<T>>::get(collection_id, owner);2205 ensure!(balance.value >= value, Error::<T>::TokenValueTooLow);22062207 2208 Self::add_fungible_item(collection, recipient, value)?;22092210 2211 <Balance<T>>::insert(collection_id, (*owner).clone(), balance.value - value);22122213 2214 if balance.value == value {2215 <FungibleItemList<T>>::remove(collection_id, owner);2216 }2217 else {2218 balance.value -= value;2219 <FungibleItemList<T>>::insert(collection_id, (*owner).clone(), balance);2220 }22212222 Ok(())2223 }22242225 fn transfer_refungible(2226 collection: &CollectionHandle<T>,2227 item_id: TokenId,2228 value: u128,2229 owner: T::AccountId,2230 new_owner: T::AccountId,2231 ) -> DispatchResult {2232 let collection_id = collection.id;2233 let full_item = <ReFungibleItemList<T>>::get(collection_id, item_id)2234 .ok_or(Error::<T>::TokenNotFound)?;22352236 let item = full_item2237 .owner2238 .iter()2239 .filter(|i| i.owner == owner)2240 .next()2241 .ok_or(Error::<T>::TokenNotFound)?;2242 let amount = item.fraction;22432244 ensure!(amount >= value, Error::<T>::TokenValueTooLow);22452246 2247 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())2248 .checked_sub(value)2249 .ok_or(Error::<T>::NumOverflow)?;2250 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);22512252 let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone())2253 .checked_add(value)2254 .ok_or(Error::<T>::NumOverflow)?;2255 <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);22562257 let old_owner = item.owner.clone();2258 let new_owner_has_account = full_item.owner.iter().any(|i| i.owner == new_owner);22592260 2261 if amount == value && !new_owner_has_account {2262 2263 2264 let mut new_full_item = full_item.clone();2265 new_full_item2266 .owner2267 .iter_mut()2268 .find(|i| i.owner == owner)2269 .expect("old owner does present in refungible")2270 .owner = new_owner.clone();2271 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);22722273 2274 Self::move_token_index(collection_id, item_id, &old_owner, &new_owner)?;2275 } else {2276 let mut new_full_item = full_item.clone();2277 new_full_item2278 .owner2279 .iter_mut()2280 .find(|i| i.owner == owner)2281 .expect("old owner does present in refungible")2282 .fraction -= value;22832284 2285 if new_owner_has_account {2286 2287 new_full_item2288 .owner2289 .iter_mut()2290 .find(|i| i.owner == new_owner)2291 .expect("new owner has account")2292 .fraction += value;2293 } else {2294 2295 new_full_item.owner.push(Ownership {2296 owner: new_owner.clone(),2297 fraction: value,2298 });2299 Self::add_token_index(collection_id, item_id, &new_owner)?;2300 }23012302 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);2303 }23042305 Ok(())2306 }23072308 fn transfer_nft(2309 collection: &CollectionHandle<T>,2310 item_id: TokenId,2311 sender: T::AccountId,2312 new_owner: T::AccountId,2313 ) -> DispatchResult {2314 let collection_id = collection.id;2315 let mut item = <NftItemList<T>>::get(collection_id, item_id)2316 .ok_or(Error::<T>::TokenNotFound)?;23172318 ensure!(2319 sender == item.owner,2320 Error::<T>::MustBeTokenOwner2321 );23222323 2324 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone())2325 .checked_sub(1)2326 .ok_or(Error::<T>::NumOverflow)?;2327 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);23282329 let balancenew_owner = <Balance<T>>::get(collection_id, new_owner.clone())2330 .checked_add(1)2331 .ok_or(Error::<T>::NumOverflow)?;2332 <Balance<T>>::insert(collection_id, new_owner.clone(), balancenew_owner);23332334 2335 let old_owner = item.owner.clone();2336 item.owner = new_owner.clone();2337 <NftItemList<T>>::insert(collection_id, item_id, item);23382339 2340 Self::move_token_index(collection_id, item_id, &old_owner, &new_owner)?;23412342 Ok(())2343 }2344 2345 fn set_re_fungible_variable_data(2346 collection: &CollectionHandle<T>,2347 item_id: TokenId,2348 data: Vec<u8>2349 ) -> DispatchResult {2350 let collection_id = collection.id;2351 let mut item = <ReFungibleItemList<T>>::get(collection_id, item_id)2352 .ok_or(Error::<T>::TokenNotFound)?;23532354 item.variable_data = data;23552356 <ReFungibleItemList<T>>::insert(collection_id, item_id, item);23572358 Ok(())2359 }23602361 fn set_nft_variable_data(2362 collection: &CollectionHandle<T>,2363 item_id: TokenId,2364 data: Vec<u8>2365 ) -> DispatchResult {2366 let collection_id = collection.id;2367 let mut item = <NftItemList<T>>::get(collection_id, item_id)2368 .ok_or(Error::<T>::TokenNotFound)?;2369 2370 item.variable_data = data;23712372 <NftItemList<T>>::insert(collection_id, item_id, item);2373 2374 Ok(())2375 }23762377 fn init_collection(item: &Collection<T>) {2378 2379 assert!(2380 item.decimal_points <= MAX_DECIMAL_POINTS,2381 "decimal_points parameter must be lower than MAX_DECIMAL_POINTS"2382 );2383 assert!(2384 item.name.len() <= 64,2385 "Collection name can not be longer than 63 char"2386 );2387 assert!(2388 item.name.len() <= 256,2389 "Collection description can not be longer than 255 char"2390 );2391 assert!(2392 item.token_prefix.len() <= 16,2393 "Token prefix can not be longer than 15 char"2394 );23952396 2397 let next_id = CreatedCollectionCount::get()2398 .checked_add(1)2399 .unwrap();24002401 CreatedCollectionCount::put(next_id);2402 }24032404 fn init_nft_token(collection_id: CollectionId, item: &NftItemType<T::AccountId>) {2405 let current_index = <ItemListIndex>::get(collection_id)2406 .checked_add(1)2407 .unwrap();24082409 let item_owner = item.owner.clone();2410 Self::add_token_index(collection_id, current_index, &item.owner).unwrap();24112412 <ItemListIndex>::insert(collection_id, current_index);24132414 2415 let new_balance = <Balance<T>>::get(collection_id, &item_owner)2416 .checked_add(1)2417 .unwrap();2418 <Balance<T>>::insert(collection_id, item_owner.clone(), new_balance);2419 }24202421 fn init_fungible_token(collection_id: CollectionId, owner: &T::AccountId, item: &FungibleItemType) {2422 let current_index = <ItemListIndex>::get(collection_id)2423 .checked_add(1)2424 .unwrap();24252426 Self::add_token_index(collection_id, current_index, owner).unwrap();24272428 <ItemListIndex>::insert(collection_id, current_index);24292430 2431 let new_balance = <Balance<T>>::get(collection_id, owner)2432 .checked_add(item.value)2433 .unwrap();2434 <Balance<T>>::insert(collection_id, (*owner).clone(), new_balance);2435 }24362437 fn init_refungible_token(collection_id: CollectionId, item: &ReFungibleItemType<T::AccountId>) {2438 let current_index = <ItemListIndex>::get(collection_id)2439 .checked_add(1)2440 .unwrap();24412442 let value = item.owner.first().unwrap().fraction;2443 let owner = item.owner.first().unwrap().owner.clone();24442445 Self::add_token_index(collection_id, current_index, &owner).unwrap();24462447 <ItemListIndex>::insert(collection_id, current_index);24482449 2450 let new_balance = <Balance<T>>::get(collection_id, &owner)2451 .checked_add(value)2452 .unwrap();2453 <Balance<T>>::insert(collection_id, owner.clone(), new_balance);2454 }24552456 fn add_token_index(collection_id: CollectionId, item_index: TokenId, owner: &T::AccountId) -> DispatchResult {2457 2458 if <AccountItemCount<T>>::contains_key(owner) {24592460 2461 let count = <AccountItemCount<T>>::get(owner);2462 ensure!(count < ChainLimit::get().account_token_ownership_limit, Error::<T>::AddressOwnershipLimitExceeded);24632464 <AccountItemCount<T>>::insert(owner.clone(), count2465 .checked_add(1)2466 .ok_or(Error::<T>::NumOverflow)?);2467 }2468 else {2469 <AccountItemCount<T>>::insert(owner.clone(), 1);2470 }24712472 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner);2473 if list_exists {2474 let mut list = <AddressTokens<T>>::get(collection_id, owner);2475 let item_contains = list.contains(&item_index.clone());24762477 if !item_contains {2478 list.push(item_index.clone());2479 }24802481 <AddressTokens<T>>::insert(collection_id, owner.clone(), list);2482 } else {2483 let mut itm = Vec::new();2484 itm.push(item_index.clone());2485 <AddressTokens<T>>::insert(collection_id, owner.clone(), itm);2486 }24872488 Ok(())2489 }24902491 fn remove_token_index(2492 collection_id: CollectionId,2493 item_index: TokenId,2494 owner: &T::AccountId,2495 ) -> DispatchResult {24962497 2498 <AccountItemCount<T>>::insert(owner.clone(), 2499 <AccountItemCount<T>>::get(owner)2500 .checked_sub(1)2501 .ok_or(Error::<T>::NumOverflow)?);250225032504 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner);2505 if list_exists {2506 let mut list = <AddressTokens<T>>::get(collection_id, owner);2507 let item_contains = list.contains(&item_index.clone());25082509 if item_contains {2510 list.retain(|&item| item != item_index);2511 <AddressTokens<T>>::insert(collection_id, owner.clone(), list);2512 }2513 }25142515 Ok(())2516 }25172518 fn move_token_index(2519 collection_id: CollectionId,2520 item_index: TokenId,2521 old_owner: &T::AccountId,2522 new_owner: &T::AccountId,2523 ) -> DispatchResult {2524 Self::remove_token_index(collection_id, item_index, old_owner)?;2525 Self::add_token_index(collection_id, item_index, new_owner)?;25262527 Ok(())2528 }2529 2530 fn ensure_contract_owned(account: T::AccountId, contract: &T::AccountId) -> DispatchResult {2531 ensure!(<ContractOwner<T>>::get(contract) == Some(account), Error::<T>::NoPermission);25322533 Ok(())2534 }2535}2536253725382539254025412542pub type Multiplier = FixedU128;25432544type BalanceOf<T> = <<T as pallet_transaction_payment::Config>::OnChargeTransaction as pallet_transaction_payment::OnChargeTransaction<T>>::Balance;2545254625472548#[derive(Encode, Decode, Clone, Eq, PartialEq)]2549pub struct ChargeTransactionPayment<T: Config>(#[codec(compact)] BalanceOf<T>);25502551impl<T: Config + Send + Sync> sp_std::fmt::Debug 2552 for ChargeTransactionPayment<T>2553{2554 #[cfg(feature = "std")]2555 fn fmt(&self, f: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {2556 write!(f, "ChargeTransactionPayment<{:?}>", self.0)2557 }2558 #[cfg(not(feature = "std"))]2559 fn fmt(&self, _: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {2560 Ok(())2561 }2562}25632564impl<T: Config> ChargeTransactionPayment<T>2565where2566 T::Call: Dispatchable<Info=DispatchInfo, PostInfo=PostDispatchInfo> + IsSubType<Call<T>> + IsSubType<pallet_contracts::Call<T>>,2567 BalanceOf<T>: Send + Sync + From<u64> + FixedPointOperand,2568 T::AccountId: AsRef<[u8]>,2569 T::AccountId: UncheckedFrom<T::Hash>,2570{2571 fn traditional_fee(2572 len: usize,2573 info: &DispatchInfoOf<T::Call>,2574 tip: BalanceOf<T>,2575 ) -> BalanceOf<T>2576 where2577 T::Call: Dispatchable<Info = DispatchInfo>,2578 {2579 <pallet_transaction_payment::Module<T>>::compute_fee(len as u32, info, tip)2580 }25812582 fn get_priority(len: usize, info: &DispatchInfoOf<T::Call>, final_fee: BalanceOf<T>) -> TransactionPriority {2583 let weight_saturation = T::BlockWeights::get().max_block / info.weight.max(1);2584 let max_block_length = *T::BlockLength::get().max.get(DispatchClass::Normal);2585 let len_saturation = max_block_length as u64 / (len as u64).max(1);2586 let coefficient: BalanceOf<T> = weight_saturation2587 .min(len_saturation)2588 .saturated_into::<BalanceOf<T>>();2589 final_fee2590 .saturating_mul(coefficient)2591 .saturated_into::<TransactionPriority>()2592 }25932594 fn withdraw_fee(2595 &self,2596 who: &T::AccountId,2597 call: &T::Call,2598 info: &DispatchInfoOf<T::Call>,2599 len: usize,2600 ) -> Result<2601 (2602 BalanceOf<T>,2603 <<T as pallet_transaction_payment::Config>::OnChargeTransaction as pallet_transaction_payment::OnChargeTransaction<T>>::LiquidityInfo,2604 ),2605 TransactionValidityError,2606 > {2607 let tip = self.0;26082609 let fee = Self::traditional_fee(len, info, tip);26102611 2612 if fee.is_zero() {2613 return <<T as pallet_transaction_payment::Config>::OnChargeTransaction as pallet_transaction_payment::OnChargeTransaction<T>>::withdraw_fee(who, call, info, fee, tip)2614 .map(|i| (fee, i));2615 }26162617 2618 2619 let mut sponsor: Option<T::AccountId> = (|| match IsSubType::<Call<T>>::is_sub_type(call) {2620 Some(Call::create_item(collection_id, _owner, _properties)) => {2621 let collection = <CollectionById<T>>::get(collection_id)?;26222623 2624 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;26252626 let limit = collection.limits.sponsor_transfer_timeout;2627 if <CreateItemBasket<T>>::contains_key((collection_id, &who)) {2628 let last_tx_block = <CreateItemBasket<T>>::get((collection_id, &who));2629 let limit_time = last_tx_block + limit.into();2630 if block_number <= limit_time {2631 return None;2632 }2633 }2634 <CreateItemBasket<T>>::insert((collection_id, who.clone()), block_number);26352636 2637 if collection.limits.sponsored_data_size >= (_properties.len() as u32) {2638 collection.sponsorship.sponsor()2639 .cloned()2640 } else {2641 None2642 }2643 }2644 Some(Call::transfer(_new_owner, collection_id, item_id, _value)) => {2645 let collection = <CollectionById<T>>::get(collection_id)?;2646 2647 let mut sponsor_transfer = false;2648 if collection.sponsorship.confirmed() {26492650 let collection_limits = collection.limits;2651 let collection_mode = collection.mode;2652 2653 2654 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2655 sponsor_transfer = match collection_mode {2656 CollectionMode::NFT => {2657 2658 2659 let limit: u32 = if collection_limits.sponsor_transfer_timeout > 0 {2660 collection_limits.sponsor_transfer_timeout2661 } else {2662 ChainLimit::get().nft_sponsor_transfer_timeout2663 };2664 2665 let mut sponsored = true;2666 if <NftTransferBasket<T>>::contains_key(collection_id, item_id) {2667 let last_tx_block = <NftTransferBasket<T>>::get(collection_id, item_id);2668 let limit_time = last_tx_block + limit.into();2669 if block_number <= limit_time {2670 sponsored = false;2671 }2672 }2673 if sponsored {2674 <NftTransferBasket<T>>::insert(collection_id, item_id, block_number);2675 }26762677 sponsored2678 }2679 CollectionMode::Fungible(_) => {2680 2681 2682 let limit: u32 = if collection_limits.sponsor_transfer_timeout > 0 {2683 collection_limits.sponsor_transfer_timeout2684 } else {2685 ChainLimit::get().fungible_sponsor_transfer_timeout2686 };2687 2688 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2689 let mut sponsored = true;2690 if <FungibleTransferBasket<T>>::contains_key(collection_id, who) {2691 let last_tx_block = <FungibleTransferBasket<T>>::get(collection_id, who);2692 let limit_time = last_tx_block + limit.into();2693 if block_number <= limit_time {2694 sponsored = false;2695 }2696 }2697 if sponsored {2698 <FungibleTransferBasket<T>>::insert(collection_id, who, block_number);2699 }27002701 sponsored2702 }2703 CollectionMode::ReFungible => {2704 2705 2706 let limit: u32 = if collection_limits.sponsor_transfer_timeout > 0 {2707 collection_limits.sponsor_transfer_timeout2708 } else {2709 ChainLimit::get().refungible_sponsor_transfer_timeout2710 };2711 2712 let mut sponsored = true;2713 if <ReFungibleTransferBasket<T>>::contains_key(collection_id, item_id) {2714 let last_tx_block = <ReFungibleTransferBasket<T>>::get(collection_id, item_id);2715 let limit_time = last_tx_block + limit.into();2716 if block_number <= limit_time {2717 sponsored = false;2718 }2719 }2720 if sponsored {2721 <ReFungibleTransferBasket<T>>::insert(collection_id, item_id, block_number);2722 }27232724 sponsored2725 }2726 _ => {2727 false2728 },2729 };2730 }27312732 if !sponsor_transfer {2733 None2734 } else {2735 collection.sponsorship.sponsor()2736 .cloned()2737 }2738 }27392740 Some(Call::set_variable_meta_data(collection_id, item_id, data)) => {2741 let mut sponsor_metadata_changes = false;27422743 let collection = <CollectionById<T>>::get(collection_id)?;27442745 if2746 collection.sponsorship.confirmed() &&2747 2748 2749 !matches!(collection.mode, CollectionMode::Fungible(_)) &&2750 data.len() <= collection.limits.sponsored_data_size as usize2751 {2752 if let Some(rate_limit) = collection.limits.sponsored_data_rate_limit {2753 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;27542755 if <VariableMetaDataBasket<T>>::get(collection_id, item_id)2756 .map(|last_block| block_number - last_block > rate_limit)2757 .unwrap_or(true) 2758 {2759 sponsor_metadata_changes = true;2760 <VariableMetaDataBasket<T>>::insert(collection_id, item_id, block_number);2761 }2762 }2763 }27642765 if !sponsor_metadata_changes {2766 None2767 } else {2768 collection.sponsorship.sponsor().cloned()2769 }2770 }27712772 _ => None,2773 })();27742775 match IsSubType::<pallet_contracts::Call<T>>::is_sub_type(call) {2776 Some(pallet_contracts::Call::call(dest, _value, _gas_limit, _data)) => {27772778 let called_contract: T::AccountId = T::Lookup::lookup((*dest).clone()).unwrap_or(T::AccountId::default());27792780 let owned_contract = <ContractOwner<T>>::get(called_contract.clone()).as_ref() == Some(who);2781 let white_list_enabled = <ContractWhiteListEnabled<T>>::contains_key(called_contract.clone());2782 2783 if !owned_contract && white_list_enabled {2784 if !<ContractWhiteList<T>>::contains_key(called_contract.clone(), who) {2785 return Err(InvalidTransaction::Call.into());2786 }2787 }2788 },2789 _ => {},2790 }27912792 2793 sponsor = sponsor.or_else(|| match IsSubType::<pallet_contracts::Call<T>>::is_sub_type(call) {27942795 2796 Some(pallet_contracts::Call::instantiate(_endowment, _gas_limit, code_hash, _data, salt)) => {27972798 let new_contract_address = <pallet_contracts::Module<T>>::contract_address(2799 &who,2800 code_hash,2801 salt,2802 );2803 <ContractOwner<T>>::insert(new_contract_address.clone(), who.clone());28042805 None2806 },28072808 2809 Some(pallet_contracts::Call::instantiate_with_code(_endowment, _gas_limit, _code, _data, _salt)) => {28102811 let new_contract_address = <pallet_contracts::Module<T>>::contract_address(2812 &who,2813 &T::Hashing::hash(&_code),2814 _salt,2815 );28162817 <ContractOwner<T>>::insert(new_contract_address.clone(), who.clone());28182819 None2820 }28212822 2823 Some(pallet_contracts::Call::call(dest, _value, _gas_limit, _data)) => {28242825 let called_contract: T::AccountId = T::Lookup::lookup((*dest).clone()).unwrap_or(T::AccountId::default());28262827 let mut sponsor_transfer = false;2828 if <ContractSponsoringRateLimit<T>>::contains_key(called_contract.clone()) {2829 let last_tx_block = <ContractSponsorBasket<T>>::get((&called_contract, &who));2830 let block_number = <system::Module<T>>::block_number() as T::BlockNumber;2831 let rate_limit = <ContractSponsoringRateLimit<T>>::get(&called_contract);2832 let limit_time = last_tx_block + rate_limit;28332834 if block_number >= limit_time {2835 <ContractSponsorBasket<T>>::insert((called_contract.clone(), who.clone()), block_number);2836 sponsor_transfer = true;2837 }2838 } else {2839 sponsor_transfer = false;2840 }2841 2842 if sponsor_transfer {2843 if <ContractSelfSponsoring<T>>::contains_key(called_contract.clone()) {2844 if <ContractSelfSponsoring<T>>::get(called_contract.clone()) {2845 return Some(called_contract);2846 }2847 }2848 }28492850 None2851 },28522853 _ => None,2854 });28552856 let who_pays_fee = sponsor.unwrap_or_else(|| who.clone());28572858 <<T as pallet_transaction_payment::Config>::OnChargeTransaction as pallet_transaction_payment::OnChargeTransaction<T>>::withdraw_fee(&who_pays_fee, call, info, fee, tip)2859 .map(|i| (fee, i))2860 }2861}286228632864impl<T: Config + Send + Sync> SignedExtension2865 for ChargeTransactionPayment<T>2866where2867 BalanceOf<T>: Send + Sync + From<u64> + FixedPointOperand,2868 T::Call: Dispatchable<Info = DispatchInfo, PostInfo = PostDispatchInfo> + IsSubType<Call<T>> + IsSubType<pallet_contracts::Call<T>>,2869 T::AccountId: AsRef<[u8]>,2870 T::AccountId: UncheckedFrom<T::Hash>,2871{2872 const IDENTIFIER: &'static str = "ChargeTransactionPayment";2873 type AccountId = T::AccountId;2874 type Call = T::Call;2875 type AdditionalSigned = ();2876 type Pre = (2877 2878 BalanceOf<T>,2879 2880 Self::AccountId,2881 2882 <<T as pallet_transaction_payment::Config>::OnChargeTransaction as pallet_transaction_payment::OnChargeTransaction<T>>::LiquidityInfo,2883 );2884 fn additional_signed(&self) -> sp_std::result::Result<(), TransactionValidityError> {2885 Ok(())2886 }28872888 fn validate(2889 &self,2890 who: &Self::AccountId,2891 call: &Self::Call,2892 info: &DispatchInfoOf<Self::Call>,2893 len: usize,2894 ) -> TransactionValidity {2895 let (fee, _) = self.withdraw_fee(who, call, info, len)?;2896 Ok(ValidTransaction {2897 priority: Self::get_priority(len, info, fee),2898 ..Default::default()2899 })2900 }29012902 fn pre_dispatch(2903 self,2904 who: &Self::AccountId,2905 call: &Self::Call,2906 info: &DispatchInfoOf<Self::Call>,2907 len: usize,2908 ) -> Result<Self::Pre, TransactionValidityError> {2909 let (_fee, imbalance) = self.withdraw_fee(who, call, info, len)?;2910 Ok((self.0, who.clone(), imbalance))2911 }29122913 fn post_dispatch(2914 pre: Self::Pre,2915 info: &DispatchInfoOf<Self::Call>,2916 post_info: &PostDispatchInfoOf<Self::Call>,2917 len: usize,2918 _result: &DispatchResult,2919 ) -> Result<(), TransactionValidityError> {2920 let (tip, who, imbalance) = pre;2921 let actual_fee = pallet_transaction_payment::Module::<T>::compute_actual_fee(2922 len as u32,2923 info,2924 post_info,2925 tip,2926 );2927 <T as pallet_transaction_payment::Config>::OnChargeTransaction::correct_and_deposit_fee(&who, info, post_info, actual_fee, tip, imbalance)?;2928 Ok(())2929 }2930}29312932