1#![cfg_attr(not(feature = "std"), no_std)]23456use codec::{Decode, Encode};7pub use frame_support::{8 decl_event, decl_module, decl_storage,9 construct_runtime, parameter_types,10 traits::{Currency, Get, ExistenceRequirement, KeyOwnerProofSystem, OnUnbalanced, Randomness, WithdrawReason, Imbalance},11 weights::{12 DispatchInfo, PostDispatchInfo, constants::{BlockExecutionWeight, ExtrinsicBaseWeight, RocksDbWeight, WEIGHT_PER_SECOND},13 IdentityFee, Weight, WeightToFeePolynomial, GetDispatchInfo, Pays,14 },15 StorageValue,16 dispatch::DispatchResult, 17 IsSubType,18 ensure19};2021use frame_system::{self as system, ensure_signed};22use sp_runtime::sp_std::prelude::Vec;23use sp_std::prelude::*;24use sp_runtime::{25 FixedU128, FixedPointOperand, 26 transaction_validity::{27 TransactionPriority, ValidTransaction, InvalidTransaction, TransactionValidityError, TransactionValidity28 },29 traits::{30 Saturating, Dispatchable, DispatchInfoOf, PostDispatchInfoOf, SignedExtension, Zero, SaturatedConversion,31 },32};3334#[cfg(test)]35mod mock;3637#[cfg(test)]38mod tests;3940#[derive(Encode, Decode, Debug, Eq, Clone, PartialEq)]41pub enum CollectionMode {42 Invalid,43 44 NFT(u32),45 46 Fungible(u32),47 48 ReFungible(u32, u32),49}5051impl Into<u8> for CollectionMode {52 fn into(self) -> u8{53 match self {54 CollectionMode::Invalid => 0,55 CollectionMode::NFT(_) => 1,56 CollectionMode::Fungible(_) => 2,57 CollectionMode::ReFungible(_, _) => 3,58 }59 }60}6162#[derive(Encode, Decode, Debug, Clone, PartialEq)]63pub enum AccessMode {64 Normal,65 WhiteList,66}67impl Default for AccessMode { fn default() -> Self { Self::Normal } }6869impl Default for CollectionMode { fn default() -> Self { Self::Invalid } }7071#[derive(Encode, Decode, Default, Clone, PartialEq)]72#[cfg_attr(feature = "std", derive(Debug))]73pub struct Ownership<AccountId> {74 pub owner: AccountId,75 pub fraction: u12876}7778#[derive(Encode, Decode, Default, Clone, PartialEq)]79#[cfg_attr(feature = "std", derive(Debug))]80pub struct CollectionType<AccountId> {81 pub owner: AccountId,82 pub mode: CollectionMode,83 pub access: AccessMode,84 pub next_item_id: u64,85 pub decimal_points: u32,86 pub name: Vec<u16>, 87 pub description: Vec<u16>, 88 pub token_prefix: Vec<u8>, 89 pub custom_data_size: u32,90 pub offchain_schema: Vec<u8>,91 pub sponsor: AccountId, 92 pub unconfirmed_sponsor: AccountId, 93}9495#[derive(Encode, Decode, Default, Clone, PartialEq)]96#[cfg_attr(feature = "std", derive(Debug))]97pub struct CollectionAdminsType<AccountId> {98 pub admin: AccountId,99 pub collection_id: u64,100}101102#[derive(Encode, Decode, Default, Clone, PartialEq)]103#[cfg_attr(feature = "std", derive(Debug))]104pub struct NftItemType<AccountId> {105 pub collection: u64,106 pub owner: AccountId,107 pub data: Vec<u8>,108}109110#[derive(Encode, Decode, Default, Clone, PartialEq)]111#[cfg_attr(feature = "std", derive(Debug))]112pub struct FungibleItemType<AccountId> {113 pub collection: u64,114 pub owner: AccountId,115 pub value: u128,116}117118#[derive(Encode, Decode, Default, Clone, PartialEq)]119#[cfg_attr(feature = "std", derive(Debug))]120pub struct ReFungibleItemType<AccountId> {121 pub collection: u64,122 pub owner: Vec<Ownership<AccountId>>,123 pub data: Vec<u8>,124}125126pub trait Trait: system::Trait {127 type Event: From<Event<Self>> + Into<<Self as system::Trait>::Event>;128129}130131decl_storage! {132 trait Store for Module<T: Trait> as Nft {133134 135 NextCollectionID: u64;136 ItemListIndex: map hasher(blake2_128_concat) u64 => u64;137138 pub Collection get(fn collection): map hasher(identity) u64 => CollectionType<T::AccountId>;139 pub AdminList get(fn admin_list_collection): map hasher(identity) u64 => Vec<T::AccountId>;140 pub WhiteList get(fn white_list): map hasher(identity) u64 => Vec<T::AccountId>;141142 143 pub Balance get(fn balance_count): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) T::AccountId => u64;144 pub ApprovedList get(fn approved): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => Vec<T::AccountId>;145146 147 pub NftItemList get(fn nft_item_id): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => NftItemType<T::AccountId>;148 pub FungibleItemList get(fn fungible_item_id): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => FungibleItemType<T::AccountId>;149 pub ReFungibleItemList get(fn refungible_item_id): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => ReFungibleItemType<T::AccountId>;150151 pub AddressTokens get(fn address_tokens): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) T::AccountId => Vec<u64>;152153 154 pub ContractSponsor get(fn contract_sponsor): map hasher(identity) T::AccountId => T::AccountId;155 pub UnconfirmedContractSponsor get(fn unconfirmed_contract_sponsor): map hasher(identity) T::AccountId => T::AccountId;156 }157}158159decl_event!(160 pub enum Event<T>161 where162 AccountId = <T as system::Trait>::AccountId,163 {164 Created(u64, u8, AccountId),165 ItemCreated(u64, u64),166 ItemDestroyed(u64, u64),167 }168);169170decl_module! {171 pub struct Module<T: Trait> for enum Call where origin: T::Origin {172173 fn deposit_event() = default;174175 176 177 178 179 #[weight = 0]180 pub fn create_collection( origin,181 collection_name: Vec<u16>,182 collection_description: Vec<u16>,183 token_prefix: Vec<u8>,184 mode: CollectionMode) -> DispatchResult {185186 187 let who = ensure_signed(origin)?;188 let custom_data_size = match mode {189 CollectionMode::NFT(size) => size,190 CollectionMode::ReFungible(size, _) => size,191 _ => 0192 };193194 let decimal_points = match mode {195 CollectionMode::Fungible(points) => points,196 CollectionMode::ReFungible(_, points) => points,197 _ => 0198 };199200 201 ensure!(decimal_points <= 4, "decimal_points parameter must be lower than 4"); 202203 let mut name = collection_name.to_vec();204 name.push(0);205 ensure!(name.len() <= 64, "Collection name can not be longer than 63 char");206207 let mut description = collection_description.to_vec();208 description.push(0);209 ensure!(name.len() <= 256, "Collection description can not be longer than 255 char");210211 let mut prefix = token_prefix.to_vec();212 prefix.push(0);213 ensure!(prefix.len() <= 16, "Token prefix can not be longer than 15 char");214215 216 let next_id = NextCollectionID::get()217 .checked_add(1)218 .expect("collection id error");219220 NextCollectionID::put(next_id);221222 223 let new_collection = CollectionType {224 owner: who.clone(),225 name: name,226 mode: mode.clone(),227 access: AccessMode::Normal,228 description: description,229 decimal_points: decimal_points,230 token_prefix: prefix,231 next_item_id: next_id,232 offchain_schema: Vec::new(),233 custom_data_size: custom_data_size,234 sponsor: T::AccountId::default(),235 unconfirmed_sponsor: T::AccountId::default(),236 };237238 239 <Collection<T>>::insert(next_id, new_collection);240241 242 Self::deposit_event(RawEvent::Created(next_id, mode.into(), who.clone()));243244 Ok(())245 }246247 #[weight = 0]248 pub fn destroy_collection(origin, collection_id: u64) -> DispatchResult {249250 let sender = ensure_signed(origin)?;251 Self::check_owner_permissions(collection_id, sender)?;252253 <AddressTokens<T>>::remove_prefix(collection_id);254 <ApprovedList<T>>::remove_prefix(collection_id);255 <Balance<T>>::remove_prefix(collection_id);256 <ItemListIndex>::remove(collection_id);257 <AdminList<T>>::remove(collection_id);258 <Collection<T>>::remove(collection_id);259 <WhiteList<T>>::remove(collection_id);260261 Ok(())262 }263264 #[weight = 0]265 pub fn change_collection_owner(origin, collection_id: u64, new_owner: T::AccountId) -> DispatchResult {266267 let sender = ensure_signed(origin)?;268 Self::check_owner_permissions(collection_id, sender)?;269 let mut target_collection = <Collection<T>>::get(collection_id);270 target_collection.owner = new_owner;271 <Collection<T>>::insert(collection_id, target_collection);272273 Ok(())274 }275276 #[weight = 0]277 pub fn add_collection_admin(origin, collection_id: u64, new_admin_id: T::AccountId) -> DispatchResult {278279 let sender = ensure_signed(origin)?;280 Self::check_owner_or_admin_permissions(collection_id, sender)?;281 let mut admin_arr: Vec<T::AccountId> = Vec::new();282283 if <AdminList<T>>::contains_key(collection_id)284 {285 admin_arr = <AdminList<T>>::get(collection_id);286 ensure!(!admin_arr.contains(&new_admin_id), "Account already has admin role");287 }288289 admin_arr.push(new_admin_id);290 <AdminList<T>>::insert(collection_id, admin_arr);291292 Ok(())293 }294295 #[weight = 0]296 pub fn remove_collection_admin(origin, collection_id: u64, account_id: T::AccountId) -> DispatchResult {297298 let sender = ensure_signed(origin)?;299 Self::check_owner_or_admin_permissions(collection_id, sender)?;300301 if <AdminList<T>>::contains_key(collection_id)302 {303 let mut admin_arr = <AdminList<T>>::get(collection_id);304 admin_arr.retain(|i| *i != account_id);305 <AdminList<T>>::insert(collection_id, admin_arr);306 }307308 Ok(())309 }310311 #[weight = 0]312 pub fn set_collection_sponsor(origin, collection_id: u64, new_sponsor: T::AccountId) -> DispatchResult {313314 let sender = ensure_signed(origin)?;315 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");316317 let mut target_collection = <Collection<T>>::get(collection_id);318 ensure!(sender == target_collection.owner, "You do not own this collection");319320 target_collection.unconfirmed_sponsor = new_sponsor;321 <Collection<T>>::insert(collection_id, target_collection);322323 Ok(())324 }325326 #[weight = 0]327 pub fn confirm_sponsorship(origin, collection_id: u64) -> DispatchResult {328329 let sender = ensure_signed(origin)?;330 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");331332 let mut target_collection = <Collection<T>>::get(collection_id);333 ensure!(sender == target_collection.unconfirmed_sponsor, "This address is not set as sponsor, use setCollectionSponsor first");334335 target_collection.sponsor = target_collection.unconfirmed_sponsor;336 target_collection.unconfirmed_sponsor = T::AccountId::default();337 <Collection<T>>::insert(collection_id, target_collection);338339 Ok(())340 }341342 #[weight = 0]343 pub fn remove_collection_sponsor(origin, collection_id: u64) -> DispatchResult {344345 let sender = ensure_signed(origin)?;346 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");347348 let mut target_collection = <Collection<T>>::get(collection_id);349 ensure!(sender == target_collection.owner, "You do not own this collection");350351 target_collection.sponsor = T::AccountId::default();352 <Collection<T>>::insert(collection_id, target_collection);353354 Ok(())355 }356 357 #[weight = 0]358 pub fn create_item(origin, collection_id: u64, properties: Vec<u8>, owner: T::AccountId) -> DispatchResult {359360 let sender = ensure_signed(origin)?;361362 363 let target_collection = <Collection<T>>::get(collection_id);364 ensure!(target_collection.custom_data_size >= properties.len() as u32, "Size of item is too large");365366 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;367368 let new_balance = <Balance<T>>::get(collection_id, owner.clone()).checked_add(1).unwrap();369 <Balance<T>>::insert(collection_id, owner.clone(), new_balance);370371 372 match target_collection.mode 373 {374 CollectionMode::NFT(_) => {375 376 let item = NftItemType {377 collection: collection_id,378 owner: owner,379 data: properties,380 };381 382 Self::add_nft_item(item)?;383 384 },385 CollectionMode::ReFungible(_, _) => {386 let mut owner_list = Vec::new();387 let value = (10 as u128).pow(target_collection.decimal_points);388 owner_list.push(Ownership {owner: owner, fraction: value});389390 let item = ReFungibleItemType {391 collection: collection_id,392 owner: owner_list,393 data: properties394 };395 396 Self::add_refungible_item(item)?;397 },398 _ => ()399 };400401 402 Self::deposit_event(RawEvent::ItemCreated(collection_id, <ItemListIndex>::get(collection_id)));403404 Ok(())405 }406407 #[weight = 0]408 pub fn burn_item(origin, collection_id: u64, item_id: u64) -> DispatchResult {409410 let sender = ensure_signed(origin)?;411 let item_owner = Self::is_item_owner(sender.clone(), collection_id, item_id);412 if !item_owner413 {414 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;415 }416 let target_collection = <Collection<T>>::get(collection_id);417418 match target_collection.mode 419 {420 CollectionMode::NFT(_) => Self::burn_nft_item(collection_id, item_id)?,421 CollectionMode::ReFungible(_, _) => Self::burn_refungible_item(collection_id, item_id, sender.clone())?,422 _ => ()423 };424425 426 Self::deposit_event(RawEvent::ItemDestroyed(collection_id, item_id));427428 Ok(())429 }430431 #[weight = 0]432 pub fn transfer(origin, recipient: T::AccountId, collection_id: u64, item_id: u64, value: u64) -> DispatchResult {433434 let sender = ensure_signed(origin)?;435 ensure!(Self::is_item_owner(sender.clone(), collection_id, item_id), "Only item owner can call transfer method");436437 let target_collection = <Collection<T>>::get(collection_id);438439 440 match target_collection.mode 441 {442 CollectionMode::NFT(_) => Self::transfer_nft(collection_id, item_id, sender.clone(), recipient)?,443 CollectionMode::ReFungible(_, _) => Self::transfer_refungible(collection_id, item_id, value, sender.clone(), recipient)?,444 _ => ()445 };446447 Ok(())448 }449450 #[weight = 0]451 pub fn approve(origin, approved: T::AccountId, collection_id: u64, item_id: u64) -> DispatchResult {452453 let sender = ensure_signed(origin)?;454455 let item_owner = Self::is_item_owner(sender.clone(), collection_id, item_id);456 if !item_owner457 {458 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;459 }460461 let list_exists = <ApprovedList<T>>::contains_key(collection_id, item_id);462 if list_exists {463464 let mut list = <ApprovedList<T>>::get(collection_id, item_id);465 let item_contains = list.contains(&approved.clone());466467 if !item_contains {468 list.push(approved.clone());469 }470 } else {471472 let mut itm = Vec::new();473 itm.push(approved.clone());474 <ApprovedList<T>>::insert(collection_id, item_id, itm);475 }476477 Ok(())478 }479480 #[weight = 0]481 pub fn transfer_from(origin, from: T::AccountId, recipient: T::AccountId, collection_id: u64, item_id: u64, value: u64 ) -> DispatchResult {482483 let mut approved: bool = false; 484 let sender = ensure_signed(origin)?;485 let approved_list_exists = <ApprovedList<T>>::contains_key(collection_id, item_id);486 if approved_list_exists487 {488 let list_itm = <ApprovedList<T>>::get(collection_id, item_id);489 approved = list_itm.contains(&recipient.clone());490 }491492 if !approved493 {494 Self::check_owner_or_admin_permissions(collection_id, sender)?;495 }496 497 let target_collection = <Collection<T>>::get(collection_id);498499 match target_collection.mode500 {501 CollectionMode::NFT(_) => Self::transfer_nft(collection_id, item_id, from, recipient)?,502 CollectionMode::ReFungible(_, _) => Self::transfer_refungible(collection_id, item_id, value, from, recipient)?,503 504 _ => ()505 };506507 Ok(())508 }509510 #[weight = 0]511 pub fn safe_transfer_from(origin, collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {512513 514 515 516 517518 519520 521522 Ok(())523 }524525 #[weight = 0]526 pub fn set_offchain_schema(527 origin,528 collection_id: u64,529 schema: Vec<u8>530 ) -> DispatchResult {531 let sender = ensure_signed(origin)?;532 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;533 534 let mut target_collection = <Collection<T>>::get(collection_id);535 target_collection.offchain_schema = schema;536 <Collection<T>>::insert(collection_id, target_collection);537538 Ok(()) 539 }540 }541}542543impl<T: Trait> Module<T> {544545 fn collection_exists(collection_id: u64) -> DispatchResult{546 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");547 Ok(())548 }549550 fn check_owner_permissions(collection_id: u64, subject: T::AccountId) -> DispatchResult {551552 Self::collection_exists(collection_id)?;553554 let target_collection = <Collection<T>>::get(collection_id);555 ensure!(subject == target_collection.owner, "You do not own this collection");556557 Ok(())558 }559560 fn check_owner_or_admin_permissions(collection_id: u64, subject: T::AccountId) -> DispatchResult {561562 Self::collection_exists(collection_id)?;563564 let target_collection = <Collection<T>>::get(collection_id);565 let is_owner = subject == target_collection.owner;566567 let no_perm_mes = "You do not have permissions to modify this collection";568 let exists = <AdminList<T>>::contains_key(collection_id);569570 if !is_owner571 {572 ensure!(exists, no_perm_mes);573 ensure!(<AdminList<T>>::get(collection_id).contains(&subject), no_perm_mes);574 }575 Ok(())576 }577578 fn is_item_owner(subject: T::AccountId, collection_id: u64, item_id: u64) -> bool{579580 let target_collection = <Collection<T>>::get(collection_id);581582 match target_collection.mode {583 CollectionMode::NFT(_) => <NftItemList<T>>::get(collection_id, item_id).owner == subject,584 CollectionMode::Fungible(_) => <FungibleItemList<T>>::get(collection_id, item_id).owner == subject,585 CollectionMode::ReFungible(_, _) => <ReFungibleItemList<T>>::get(collection_id, item_id).owner.iter().any(|i| i.owner == subject),586 CollectionMode::Invalid => false587 }588 }589590 fn add_refungible_item(item: ReFungibleItemType<T::AccountId>) -> DispatchResult {591592 let current_index = <ItemListIndex>::get(item.collection)593 .checked_add(1)594 .expect("Item list index id error");595596 Self::add_token_index(item.collection, current_index, item.owner.first().unwrap().owner.clone())?;597598 <ItemListIndex>::insert(item.collection, current_index);599 <ReFungibleItemList<T>>::insert(item.collection, current_index, item); 600601 Ok(())602 }603604 fn burn_refungible_item(collection_id: u64, item_id: u64, owner: T::AccountId) -> DispatchResult {605 606 let collection = <ReFungibleItemList<T>>::get(collection_id, item_id);607 let item = collection.owner.iter().filter(|&i| i.owner == owner).next().unwrap();608 Self::remove_token_index(collection_id, item_id, owner)?;609610 611 let new_balance = <Balance<T>>::get(collection_id, item.owner.clone()).checked_sub(item.fraction as u64).unwrap();612 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);613614 615 <ReFungibleItemList<T>>::remove(collection_id, item_id);616617 Ok(())618 }619620 fn transfer_refungible(collection_id: u64, item_id: u64, value: u64, owner: T::AccountId, new_owner: T::AccountId) -> DispatchResult {621622 let full_item = <ReFungibleItemList<T>>::get(collection_id, item_id);623 let item = full_item.owner.iter().filter(|i| i.owner == owner).next().unwrap();624 let amount = item.fraction;625626 ensure!(amount < value.into(),"Item balance not enouth");627628 629 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone()).checked_sub(value).unwrap();630 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);631632 let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone()).checked_add(value).unwrap();633 <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);634635 let old_owner = item.owner.clone();636 let new_owner_has_account = full_item.owner.iter().any(|i| i.owner == new_owner);637638 639 if amount == value.into() && !new_owner_has_account640 {641 642 643 let mut new_full_item = full_item.clone();644 new_full_item.owner.iter_mut().find(|i| i.owner == owner).unwrap().owner = new_owner.clone();645 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);646647 648 Self::move_token_index(collection_id, item_id, old_owner.clone(), new_owner.clone())?;649 }650 else651 {652 let mut new_full_item = full_item.clone();653 new_full_item.owner.iter_mut().find(|i| i.owner == owner).unwrap().fraction -= amount;654655 656 if new_owner_has_account {657 658 new_full_item.owner.iter_mut().find(|i| i.owner == new_owner).unwrap().fraction += amount;659 }660 else661 {662 663 new_full_item.owner.push(Ownership { owner: new_owner.clone(), fraction: amount});664 Self::add_token_index(collection_id, item_id, new_owner.clone())?;665 }666667 <ReFungibleItemList<T>>::insert(collection_id, item_id, new_full_item);668 }669670 Ok(())671 }672 673 fn add_nft_item(item: NftItemType<T::AccountId>) -> DispatchResult {674675 let current_index = <ItemListIndex>::get(item.collection)676 .checked_add(1)677 .expect("Item list index id error");678679 Self::add_token_index(item.collection, current_index, item.owner.clone())?;680681 <ItemListIndex>::insert(item.collection, current_index);682 <NftItemList<T>>::insert(item.collection, current_index, item);683684 Ok(())685 }686687 fn burn_nft_item(collection_id: u64, item_id: u64) -> DispatchResult {688 689 let item = <NftItemList<T>>::get(collection_id, item_id);690 Self::remove_token_index(collection_id, item_id, item.owner.clone())?;691692 693 let new_balance = <Balance<T>>::get(collection_id, item.owner.clone()).checked_sub(1).unwrap();694 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);695 <NftItemList<T>>::remove(collection_id, item_id);696697 Ok(())698 }699700 fn transfer_nft(collection_id: u64, item_id: u64, sender: T::AccountId, new_owner: T::AccountId) -> DispatchResult {701702 let mut item = <NftItemList<T>>::get(collection_id, item_id);703704 ensure!(sender == item.owner,"sender parameter and item owner must be equal");705706 707 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone()).checked_sub(1).unwrap();708 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);709710 let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone()).checked_add(1).unwrap();711 <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);712713 714 let old_owner = item.owner.clone();715 item.owner = new_owner.clone();716 <NftItemList<T>>::insert(collection_id, item_id, item);717718 719 Self::move_token_index(collection_id, item_id, old_owner, new_owner.clone())?;720721 722 let itm: Vec<T::AccountId> = Vec::new();723 <ApprovedList<T>>::insert(collection_id, item_id, itm);724725 Ok(())726 }727728 fn add_token_index(collection_id: u64, item_index: u64, owner: T::AccountId) -> DispatchResult {729 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());730 if list_exists {731 let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());732 let item_contains = list.contains(&item_index.clone());733734 if !item_contains {735 list.push(item_index.clone());736 }737738 <AddressTokens<T>>::insert(collection_id, owner.clone(), list);739 } else {740 let mut itm = Vec::new();741 itm.push(item_index.clone());742 <AddressTokens<T>>::insert(collection_id, owner, itm);743 }744745 Ok(())746 }747748 fn remove_token_index(749 collection_id: u64,750 item_index: u64,751 owner: T::AccountId,752 ) -> DispatchResult {753 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());754 if list_exists {755 let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());756 let item_contains = list.contains(&item_index.clone());757758 if item_contains {759 list.retain(|&item| item != item_index);760 <AddressTokens<T>>::insert(collection_id, owner, list);761 }762 }763764 Ok(())765 }766767 fn move_token_index(768 collection_id: u64,769 item_index: u64,770 old_owner: T::AccountId,771 new_owner: T::AccountId,772 ) -> DispatchResult {773 Self::remove_token_index(collection_id, item_index, old_owner)?;774 Self::add_token_index(collection_id, item_index, new_owner)?;775776 Ok(())777 }778}779780781782783784785pub type Multiplier = FixedU128;786787type BalanceOf<T> =788 <<T as transaction_payment::Trait>::Currency as Currency<<T as system::Trait>::AccountId>>::Balance;789type NegativeImbalanceOf<T> = <<T as transaction_payment::Trait>::Currency as Currency<790 <T as system::Trait>::AccountId,>>::NegativeImbalance;791792793794795796#[derive(Encode, Decode, Clone, Eq, PartialEq)]797pub struct ChargeTransactionPayment<T: transaction_payment::Trait + Send + Sync>(#[codec(compact)] BalanceOf<T>);798799impl<T:Trait + transaction_payment::Trait + Send + Sync> sp_std::fmt::Debug for ChargeTransactionPayment<T> {800 #[cfg(feature = "std")]801 fn fmt(&self, f: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {802 write!(f, "ChargeTransactionPayment<{:?}>", self.0)803 }804 #[cfg(not(feature = "std"))]805 fn fmt(&self, _: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {806 Ok(())807 }808}809810impl<T:Trait + transaction_payment::Trait + Send + Sync> ChargeTransactionPayment<T> where811 T::Call: Dispatchable<Info=DispatchInfo, PostInfo=PostDispatchInfo> + IsSubType<Module<T>, T>,812 BalanceOf<T>: Send + Sync + FixedPointOperand,813{814 815 pub fn from(fee: BalanceOf<T>) -> Self {816 Self(fee)817 }818819 pub fn traditional_fee(820 len: usize,821 info: &DispatchInfoOf<T::Call>,822 tip: BalanceOf<T>,823 ) -> BalanceOf<T> where824 T::Call: Dispatchable<Info=DispatchInfo>,825 {826 <transaction_payment::Module<T>>::compute_fee(len as u32, info, tip)827 }828829 fn withdraw_fee(830 &self,831 who: &T::AccountId,832 call: &T::Call,833 info: &DispatchInfoOf<T::Call>,834 len: usize,835 ) -> Result<(BalanceOf<T>, Option<NegativeImbalanceOf<T>>), TransactionValidityError> {836 let tip = self.0;837838 839 840 let fee = match call.is_sub_type() {841 Some(Call::create_collection(..)) => <BalanceOf<T>>::from(1_000_000_000),842 _ => Self::traditional_fee(len, info, tip)843844 845 846 };847848 849 850 let sponsor: T::AccountId = match call.is_sub_type() {851 Some(Call::create_item(collection_id, _properties, _owner)) => {852 <Collection<T>>::get(collection_id).sponsor853 },854 Some(Call::transfer(_new_owner, collection_id, _item_id, _value)) => {855 <Collection<T>>::get(collection_id).sponsor856 },857858 _ => T::AccountId::default()859 };860861 let mut who_pays_fee: T::AccountId = sponsor.clone();862 if sponsor == T::AccountId::default() {863 who_pays_fee = who.clone();864 }865866 867 if fee.is_zero() {868 return Ok((fee, None));869 }870871 match <T as transaction_payment::Trait>::Currency::withdraw(872 &who_pays_fee,873 fee,874 if tip.is_zero() {875 WithdrawReason::TransactionPayment.into()876 } else {877 WithdrawReason::TransactionPayment | WithdrawReason::Tip878 },879 ExistenceRequirement::KeepAlive,880 ) {881 Ok(imbalance) => Ok((fee, Some(imbalance))),882 Err(_) => Err(InvalidTransaction::Payment.into()),883 }884 }885}886887impl<T:Trait + transaction_payment::Trait + Send + Sync> SignedExtension for ChargeTransactionPayment<T> where888 BalanceOf<T>: Send + Sync + From<u64> + FixedPointOperand,889 T::Call: Dispatchable<Info=DispatchInfo, PostInfo=PostDispatchInfo> + IsSubType<Module<T>, T>,890{891 const IDENTIFIER: &'static str = "ChargeTransactionPayment";892 type AccountId = T::AccountId;893 type Call = T::Call;894 type AdditionalSigned = ();895 type Pre = (BalanceOf<T>, Self::AccountId, Option<NegativeImbalanceOf<T>>, BalanceOf<T>);896 fn additional_signed(&self) -> sp_std::result::Result<(), TransactionValidityError> { Ok(()) }897898 fn validate(899 &self,900 who: &Self::AccountId,901 call: &Self::Call,902 info: &DispatchInfoOf<Self::Call>,903 len: usize,904 ) -> TransactionValidity {905 let (fee, _) = self.withdraw_fee(who, call, info, len)?;906907 let mut r = ValidTransaction::default();908 909 910 r.priority = fee.saturated_into::<TransactionPriority>();911 Ok(r)912 }913914 fn pre_dispatch(915 self,916 who: &Self::AccountId,917 call: &Self::Call,918 info: &DispatchInfoOf<Self::Call>,919 len: usize920 ) -> Result<Self::Pre, TransactionValidityError> {921 let (fee, imbalance) = self.withdraw_fee(who, call, info, len)?;922 Ok((self.0, who.clone(), imbalance, fee))923 }924925 fn post_dispatch(926 pre: Self::Pre,927 info: &DispatchInfoOf<Self::Call>,928 post_info: &PostDispatchInfoOf<Self::Call>,929 len: usize,930 _result: &DispatchResult,931 ) -> Result<(), TransactionValidityError> {932 let (tip, who, imbalance, fee) = pre;933 if let Some(payed) = imbalance {934 let actual_fee = <transaction_payment::Module<T>>::compute_actual_fee(935 len as u32,936 info,937 post_info,938 tip,939 );940 let refund = fee.saturating_sub(actual_fee);941 let actual_payment = match <T as transaction_payment::Trait>::Currency::deposit_into_existing(&who, refund) {942 Ok(refund_imbalance) => {943 944 945 match payed.offset(refund_imbalance) {946 Ok(actual_payment) => actual_payment,947 Err(_) => return Err(InvalidTransaction::Payment.into()),948 }949 }950 951 952 Err(_) => payed,953 };954 let imbalances = actual_payment.split(tip);955 <T as transaction_payment::Trait>::OnTransactionPayment::on_unbalanceds(Some(imbalances.0).into_iter()956 .chain(Some(imbalances.1)));957 }958 Ok(())959 }960}