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 ReFungible,48}4950impl Into<u8> for CollectionMode {51 fn into(self) -> u8{52 match self {53 CollectionMode::Invalid => 0,54 CollectionMode::NFT(_) => 1,55 CollectionMode::Fungible(_) => 2,56 CollectionMode::ReFungible => 3,57 }58 }59}6061#[derive(Encode, Decode, Debug, Clone, PartialEq)]62pub enum AccessMode {63 Normal,64 WhiteList,65}66impl Default for AccessMode { fn default() -> Self { Self::Normal } }6768impl Default for CollectionMode { fn default() -> Self { Self::Invalid } }6970#[derive(Encode, Decode, Default, Clone, PartialEq)]71#[cfg_attr(feature = "std", derive(Debug))]72pub struct Ownership<AccountId> {73 pub owner: AccountId,74 pub fraction: u12875}7677#[derive(Encode, Decode, Default, Clone, PartialEq)]78#[cfg_attr(feature = "std", derive(Debug))]79pub struct CollectionType<AccountId> {80 pub owner: AccountId,81 pub mode: CollectionMode,82 pub access: AccessMode,83 pub next_item_id: u64,84 pub decimal_points: u32,85 pub name: Vec<u16>, 86 pub description: Vec<u16>, 87 pub token_prefix: Vec<u8>, 88 pub custom_data_size: u32,89 pub offchain_schema: Vec<u8>,90 pub sponsor: AccountId, 91 pub unconfirmed_sponsor: AccountId, 92}9394#[derive(Encode, Decode, Default, Clone, PartialEq)]95#[cfg_attr(feature = "std", derive(Debug))]96pub struct CollectionAdminsType<AccountId> {97 pub admin: AccountId,98 pub collection_id: u64,99}100101#[derive(Encode, Decode, Default, Clone, PartialEq)]102#[cfg_attr(feature = "std", derive(Debug))]103pub struct NftItemType<AccountId> {104 pub collection: u64,105 pub owner: AccountId,106 pub data: Vec<u8>,107}108109#[derive(Encode, Decode, Default, Clone, PartialEq)]110#[cfg_attr(feature = "std", derive(Debug))]111pub struct FungibleItemType<AccountId> {112 pub collection: u64,113 pub owner: Vec<AccountId>,114 pub data: Vec<u64>,115}116117#[derive(Encode, Decode, Default, Clone, PartialEq)]118#[cfg_attr(feature = "std", derive(Debug))]119pub struct ReFungibleItemType<AccountId> {120 pub collection: u64,121 pub owner: Vec<Ownership<AccountId>>,122}123124pub trait Trait: system::Trait {125 type Event: From<Event<Self>> + Into<<Self as system::Trait>::Event>;126127}128129decl_storage! {130 trait Store for Module<T: Trait> as Nft {131132 133 NextCollectionID: u64;134 ItemListIndex: map hasher(blake2_128_concat) u64 => u64;135136 pub Collection get(fn collection): map hasher(identity) u64 => CollectionType<T::AccountId>;137 pub AdminList get(fn admin_list_collection): map hasher(identity) u64 => Vec<T::AccountId>;138 pub WhiteList get(fn white_list): map hasher(identity) u64 => Vec<T::AccountId>;139140 141 pub Balance get(fn balance_count): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) T::AccountId => u64;142 pub ApprovedList get(fn approved): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => Vec<T::AccountId>;143144 145 pub NftItemList get(fn nft_item_id): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => NftItemType<T::AccountId>;146 pub FungibleItemList get(fn fungible_item_id): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => FungibleItemType<T::AccountId>;147 pub ReFungibleItemList get(fn refungible_item_id): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) u64 => ReFungibleItemType<T::AccountId>;148149 pub AddressTokens get(fn address_tokens): double_map hasher(blake2_128_concat) u64, hasher(blake2_128_concat) T::AccountId => Vec<u64>;150 }151}152153decl_event!(154 pub enum Event<T>155 where156 AccountId = <T as system::Trait>::AccountId,157 {158 Created(u64, u8, AccountId),159 ItemCreated(u64, u64),160 ItemDestroyed(u64, u64),161 }162);163164decl_module! {165 pub struct Module<T: Trait> for enum Call where origin: T::Origin {166167 fn deposit_event() = default;168169 170 171 172 173 #[weight = 0]174 pub fn create_collection( origin,175 collection_name: Vec<u16>,176 collection_description: Vec<u16>,177 token_prefix: Vec<u8>,178 mode: CollectionMode) -> DispatchResult {179180 181 let who = ensure_signed(origin)?;182 let custom_data_size = match mode {183 CollectionMode::NFT(size) => size,184 _ => 0185 };186187 let decimal_points = match mode {188 CollectionMode::Fungible(points) => points,189 _ => 0190 };191192 193 ensure!(decimal_points < 100, "decimal_points parameter must be lower than 100"); 194195 let mut name = collection_name.to_vec();196 name.push(0);197 ensure!(name.len() <= 64, "Collection name can not be longer than 63 char");198199 let mut description = collection_description.to_vec();200 description.push(0);201 ensure!(name.len() <= 256, "Collection description can not be longer than 255 char");202203 let mut prefix = token_prefix.to_vec();204 prefix.push(0);205 ensure!(prefix.len() <= 16, "Token prefix can not be longer than 15 char");206207 208 let next_id = NextCollectionID::get()209 .checked_add(1)210 .expect("collection id error");211212 NextCollectionID::put(next_id);213214 215 let new_collection = CollectionType {216 owner: who.clone(),217 name: name,218 mode: mode.clone(),219 access: AccessMode::Normal,220 description: description,221 decimal_points: decimal_points,222 token_prefix: prefix,223 next_item_id: next_id,224 offchain_schema: Vec::new(),225 custom_data_size: custom_data_size,226 sponsor: T::AccountId::default(),227 unconfirmed_sponsor: T::AccountId::default(),228 };229230 231 <Collection<T>>::insert(next_id, new_collection);232233 234 Self::deposit_event(RawEvent::Created(next_id, mode.into(), who.clone()));235236 Ok(())237 }238239 #[weight = 0]240 pub fn destroy_collection(origin, collection_id: u64) -> DispatchResult {241242 let sender = ensure_signed(origin)?;243 Self::check_owner_permissions(collection_id, sender)?;244245 <AddressTokens<T>>::remove_prefix(collection_id);246 <ApprovedList<T>>::remove_prefix(collection_id);247 <Balance<T>>::remove_prefix(collection_id);248 <ItemListIndex>::remove(collection_id);249 <AdminList<T>>::remove(collection_id);250 <Collection<T>>::remove(collection_id);251 <WhiteList<T>>::remove(collection_id);252253 Ok(())254 }255256 #[weight = 0]257 pub fn change_collection_owner(origin, collection_id: u64, new_owner: T::AccountId) -> DispatchResult {258259 let sender = ensure_signed(origin)?;260 Self::check_owner_permissions(collection_id, sender)?;261 let mut target_collection = <Collection<T>>::get(collection_id);262 target_collection.owner = new_owner;263 <Collection<T>>::insert(collection_id, target_collection);264265 Ok(())266 }267268 #[weight = 0]269 pub fn add_collection_admin(origin, collection_id: u64, new_admin_id: T::AccountId) -> DispatchResult {270271 let sender = ensure_signed(origin)?;272 Self::check_owner_or_admin_permissions(collection_id, sender)?;273 let mut admin_arr: Vec<T::AccountId> = Vec::new();274275 if <AdminList<T>>::contains_key(collection_id)276 {277 admin_arr = <AdminList<T>>::get(collection_id);278 ensure!(!admin_arr.contains(&new_admin_id), "Account already has admin role");279 }280281 admin_arr.push(new_admin_id);282 <AdminList<T>>::insert(collection_id, admin_arr);283284 Ok(())285 }286287 #[weight = 0]288 pub fn remove_collection_admin(origin, collection_id: u64, account_id: T::AccountId) -> DispatchResult {289290 let sender = ensure_signed(origin)?;291 Self::check_owner_or_admin_permissions(collection_id, sender)?;292293 if <AdminList<T>>::contains_key(collection_id)294 {295 let mut admin_arr = <AdminList<T>>::get(collection_id);296 admin_arr.retain(|i| *i != account_id);297 <AdminList<T>>::insert(collection_id, admin_arr);298 }299300 Ok(())301 }302303 #[weight = 0]304 pub fn set_collection_sponsor(origin, collection_id: u64, new_sponsor: T::AccountId) -> DispatchResult {305306 let sender = ensure_signed(origin)?;307 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");308309 let mut target_collection = <Collection<T>>::get(collection_id);310 ensure!(sender == target_collection.owner, "You do not own this collection");311312 target_collection.unconfirmed_sponsor = new_sponsor;313 <Collection<T>>::insert(collection_id, target_collection);314315 Ok(())316 }317318 #[weight = 0]319 pub fn confirm_sponsorship(origin, collection_id: u64) -> DispatchResult {320321 let sender = ensure_signed(origin)?;322 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");323324 let mut target_collection = <Collection<T>>::get(collection_id);325 ensure!(sender == target_collection.unconfirmed_sponsor, "This address is not set as sponsor, use setCollectionSponsor first");326327 target_collection.sponsor = target_collection.unconfirmed_sponsor;328 target_collection.unconfirmed_sponsor = T::AccountId::default();329 <Collection<T>>::insert(collection_id, target_collection);330331 Ok(())332 }333334 #[weight = 0]335 pub fn remove_collection_sponsor(origin, collection_id: u64) -> DispatchResult {336337 let sender = ensure_signed(origin)?;338 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");339340 let mut target_collection = <Collection<T>>::get(collection_id);341 ensure!(sender == target_collection.owner, "You do not own this collection");342343 target_collection.sponsor = T::AccountId::default();344 <Collection<T>>::insert(collection_id, target_collection);345346 Ok(())347 }348 349 #[weight = 0]350 pub fn create_item(origin, collection_id: u64, properties: Vec<u8>, owner: T::AccountId) -> DispatchResult {351352 let sender = ensure_signed(origin)?;353354 355 let target_collection = <Collection<T>>::get(collection_id);356 ensure!(target_collection.custom_data_size >= properties.len() as u32, "Size of item is too large");357358 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;359360 let new_balance = <Balance<T>>::get(collection_id, owner.clone()) + 1;361 <Balance<T>>::insert(collection_id, owner.clone(), new_balance);362363 364 match target_collection.mode 365 {366 CollectionMode::NFT(_) => {367 368 let item = NftItemType {369 collection: collection_id,370 owner: owner,371 data: properties,372 };373 374 Self::add_nft_item(item)?;375 376 },377 _ => ()378 };379380 381 Self::deposit_event(RawEvent::ItemCreated(collection_id, <ItemListIndex>::get(collection_id)));382383 Ok(())384 }385386 #[weight = 0]387 pub fn burn_item(origin, collection_id: u64, item_id: u64) -> DispatchResult {388389 let sender = ensure_signed(origin)?;390 let item_owner = Self::is_item_owner(sender.clone(), collection_id, item_id);391 if !item_owner392 {393 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;394 }395 396 Self::burn_nft_item(collection_id, item_id)?;397398 399 Self::deposit_event(RawEvent::ItemDestroyed(collection_id, item_id));400401 Ok(())402 }403404 #[weight = 0]405 pub fn transfer(origin, recipient: T::AccountId, collection_id: u64, item_id: u64, value: u64) -> DispatchResult {406407 let sender = ensure_signed(origin)?;408 let item_owner = Self::is_item_owner(sender.clone(), collection_id, item_id);409 if !item_owner410 {411 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;412 }413414 let target_collection = <Collection<T>>::get(collection_id);415416 417 match target_collection.mode 418 {419 CollectionMode::NFT(_) => Self::transfer_nft(collection_id, item_id, recipient)?,420 _ => ()421 };422423 Ok(())424 }425426 #[weight = 0]427 pub fn approve(origin, approved: T::AccountId, collection_id: u64, item_id: u64) -> DispatchResult {428429 let sender = ensure_signed(origin)?;430431 let item_owner = Self::is_item_owner(sender.clone(), collection_id, item_id);432 if !item_owner433 {434 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;435 }436437 let list_exists = <ApprovedList<T>>::contains_key(collection_id, item_id);438 if list_exists {439440 let mut list = <ApprovedList<T>>::get(collection_id, item_id);441 let item_contains = list.contains(&approved.clone());442443 if !item_contains {444 list.push(approved.clone());445 }446 } else {447448 let mut itm = Vec::new();449 itm.push(approved.clone());450 <ApprovedList<T>>::insert(collection_id, item_id, itm);451 }452453 Ok(())454 }455456 #[weight = 0]457 pub fn transfer_from(origin, recipient: T::AccountId, collection_id: u64, item_id: u64, ) -> DispatchResult {458459 let mut approved: bool = false; 460 let sender = ensure_signed(origin)?;461 let approved_list_exists = <ApprovedList<T>>::contains_key(collection_id, item_id);462 if approved_list_exists463 {464 let list_itm = <ApprovedList<T>>::get(collection_id, item_id);465 approved = list_itm.contains(&recipient.clone());466 }467468 if !approved469 {470 Self::check_owner_or_admin_permissions(collection_id, sender)?;471 }472 473 let target_collection = <Collection<T>>::get(collection_id);474475 match target_collection.mode476 {477 CollectionMode::NFT(_) => Self::transfer_nft(collection_id, item_id, recipient)?,478 479 _ => ()480 };481482 Ok(())483 }484485 #[weight = 0]486 pub fn safe_transfer_from(origin, collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {487488 489 490 491 492493 494495 496497 Ok(())498 }499500 #[weight = 0]501 pub fn set_offchain_schema(502 origin,503 collection_id: u64,504 schema: Vec<u8>505 ) -> DispatchResult {506 let sender = ensure_signed(origin)?;507 Self::check_owner_or_admin_permissions(collection_id, sender.clone())?;508 509 let mut target_collection = <Collection<T>>::get(collection_id);510 target_collection.offchain_schema = schema;511 <Collection<T>>::insert(collection_id, target_collection);512513 Ok(()) 514 }515 }516}517518impl<T: Trait> Module<T> {519520 fn collection_exists(collection_id: u64) -> DispatchResult{521 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");522 Ok(())523 }524525 fn check_owner_permissions(collection_id: u64, subject: T::AccountId) -> DispatchResult {526527 Self::collection_exists(collection_id)?;528529 let target_collection = <Collection<T>>::get(collection_id);530 ensure!(subject == target_collection.owner, "You do not own this collection");531532 Ok(())533 }534535 fn check_owner_or_admin_permissions(collection_id: u64, subject: T::AccountId) -> DispatchResult {536537 Self::collection_exists(collection_id)?;538539 let target_collection = <Collection<T>>::get(collection_id);540 let is_owner = subject == target_collection.owner;541542 let no_perm_mes = "You do not have permissions to modify this collection";543 let exists = <AdminList<T>>::contains_key(collection_id);544545 if !is_owner546 {547 ensure!(exists, no_perm_mes);548 ensure!(<AdminList<T>>::get(collection_id).contains(&subject), no_perm_mes);549 }550 Ok(())551 }552553 fn is_item_owner(subject: T::AccountId, collection_id: u64, item_id: u64) -> bool{554555 let target_collection = <Collection<T>>::get(collection_id);556557 match target_collection.mode {558 CollectionMode::NFT(_) => <NftItemList<T>>::get(collection_id, item_id).owner == subject,559 CollectionMode::Fungible(_) => <FungibleItemList<T>>::get(collection_id, item_id).owner.contains(&subject),560 CollectionMode::ReFungible => <ReFungibleItemList<T>>::get(collection_id, item_id).owner.iter().any(|i| i.owner == subject),561 CollectionMode::Invalid => false562 }563 }564565 fn add_nft_item(item: NftItemType<T::AccountId>) -> DispatchResult {566567 let current_index = <ItemListIndex>::get(item.collection)568 .checked_add(1)569 .expect("Item list index id error");570571 Self::add_token_index(item.collection, current_index, item.owner.clone())?;572573 <ItemListIndex>::insert(item.collection, current_index);574 <NftItemList<T>>::insert(item.collection, current_index, item);575576 Ok(())577 }578579 fn burn_nft_item(collection_id: u64, item_id: u64) -> DispatchResult {580 581 let item = <NftItemList<T>>::get(collection_id, item_id);582 Self::remove_token_index(collection_id, item_id, item.owner.clone())?;583584 585 let new_balance = <Balance<T>>::get(collection_id, item.owner.clone()).checked_sub(1).unwrap();586 <Balance<T>>::insert(collection_id, item.owner.clone(), new_balance);587 <NftItemList<T>>::remove(collection_id, item_id);588589 Ok(())590 }591592 fn transfer_nft(collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {593594 let mut item = <NftItemList<T>>::get(collection_id, item_id);595596 597 let balance_old_owner = <Balance<T>>::get(collection_id, item.owner.clone()).checked_sub(1).unwrap();598 <Balance<T>>::insert(collection_id, item.owner.clone(), balance_old_owner);599600 let balance_new_owner = <Balance<T>>::get(collection_id, new_owner.clone()).checked_add(1).unwrap();601 <Balance<T>>::insert(collection_id, new_owner.clone(), balance_new_owner);602603 604 let old_owner = item.owner.clone();605 item.owner = new_owner.clone();606 <NftItemList<T>>::insert(collection_id, item_id, item);607608 609 Self::move_token_index(collection_id, item_id, old_owner, new_owner.clone())?;610611 612 let itm: Vec<T::AccountId> = Vec::new();613 <ApprovedList<T>>::insert(collection_id, item_id, itm);614615 Ok(())616 }617618 fn add_token_index(collection_id: u64, item_index: u64, owner: T::AccountId) -> DispatchResult {619 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());620 if list_exists {621 let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());622 let item_contains = list.contains(&item_index.clone());623624 if !item_contains {625 list.push(item_index.clone());626 }627628 <AddressTokens<T>>::insert(collection_id, owner.clone(), list);629 } else {630 let mut itm = Vec::new();631 itm.push(item_index.clone());632 <AddressTokens<T>>::insert(collection_id, owner, itm);633 }634635 Ok(())636 }637638 fn remove_token_index(639 collection_id: u64,640 item_index: u64,641 owner: T::AccountId,642 ) -> DispatchResult {643 let list_exists = <AddressTokens<T>>::contains_key(collection_id, owner.clone());644 if list_exists {645 let mut list = <AddressTokens<T>>::get(collection_id, owner.clone());646 let item_contains = list.contains(&item_index.clone());647648 if item_contains {649 list.retain(|&item| item != item_index);650 <AddressTokens<T>>::insert(collection_id, owner, list);651 }652 }653654 Ok(())655 }656657 fn move_token_index(658 collection_id: u64,659 item_index: u64,660 old_owner: T::AccountId,661 new_owner: T::AccountId,662 ) -> DispatchResult {663 Self::remove_token_index(collection_id, item_index, old_owner)?;664 Self::add_token_index(collection_id, item_index, new_owner)?;665666 Ok(())667 }668}669670671672673674675pub type Multiplier = FixedU128;676677type BalanceOf<T> =678 <<T as transaction_payment::Trait>::Currency as Currency<<T as system::Trait>::AccountId>>::Balance;679type NegativeImbalanceOf<T> = <<T as transaction_payment::Trait>::Currency as Currency<680 <T as system::Trait>::AccountId,>>::NegativeImbalance;681682683684685686#[derive(Encode, Decode, Clone, Eq, PartialEq)]687pub struct ChargeTransactionPayment<T: transaction_payment::Trait + Send + Sync>(#[codec(compact)] BalanceOf<T>);688689impl<T:Trait + transaction_payment::Trait + Send + Sync> sp_std::fmt::Debug for ChargeTransactionPayment<T> {690 #[cfg(feature = "std")]691 fn fmt(&self, f: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {692 write!(f, "ChargeTransactionPayment<{:?}>", self.0)693 }694 #[cfg(not(feature = "std"))]695 fn fmt(&self, _: &mut sp_std::fmt::Formatter) -> sp_std::fmt::Result {696 Ok(())697 }698}699700impl<T:Trait + transaction_payment::Trait + Send + Sync> ChargeTransactionPayment<T> where701 T::Call: Dispatchable<Info=DispatchInfo, PostInfo=PostDispatchInfo> + IsSubType<Module<T>, T>,702 BalanceOf<T>: Send + Sync + FixedPointOperand,703{704 705 pub fn from(fee: BalanceOf<T>) -> Self {706 Self(fee)707 }708709 pub fn traditional_fee(710 len: usize,711 info: &DispatchInfoOf<T::Call>,712 tip: BalanceOf<T>,713 ) -> BalanceOf<T> where714 T::Call: Dispatchable<Info=DispatchInfo>,715 {716 <transaction_payment::Module<T>>::compute_fee(len as u32, info, tip)717 }718719 fn withdraw_fee(720 &self,721 who: &T::AccountId,722 call: &T::Call,723 info: &DispatchInfoOf<T::Call>,724 len: usize,725 ) -> Result<(BalanceOf<T>, Option<NegativeImbalanceOf<T>>), TransactionValidityError> {726 let tip = self.0;727728 729 730 let fee = match call.is_sub_type() {731 Some(Call::create_collection(..)) => <BalanceOf<T>>::from(1_000_000_000),732 _ => Self::traditional_fee(len, info, tip)733734 735 736 };737738 739 740 let sponsor: T::AccountId = match call.is_sub_type() {741 Some(Call::create_item(collection_id, _properties, _owner)) => {742 <Collection<T>>::get(collection_id).sponsor743 },744 Some(Call::transfer(_new_owner, collection_id, _item_id, _value)) => {745 <Collection<T>>::get(collection_id).sponsor746 },747748 _ => T::AccountId::default()749 };750751 let mut who_pays_fee: T::AccountId = sponsor.clone();752 if sponsor == T::AccountId::default() {753 who_pays_fee = who.clone();754 }755756 757 if fee.is_zero() {758 return Ok((fee, None));759 }760761 match <T as transaction_payment::Trait>::Currency::withdraw(762 &who_pays_fee,763 fee,764 if tip.is_zero() {765 WithdrawReason::TransactionPayment.into()766 } else {767 WithdrawReason::TransactionPayment | WithdrawReason::Tip768 },769 ExistenceRequirement::KeepAlive,770 ) {771 Ok(imbalance) => Ok((fee, Some(imbalance))),772 Err(_) => Err(InvalidTransaction::Payment.into()),773 }774 }775}776777impl<T:Trait + transaction_payment::Trait + Send + Sync> SignedExtension for ChargeTransactionPayment<T> where778 BalanceOf<T>: Send + Sync + From<u64> + FixedPointOperand,779 T::Call: Dispatchable<Info=DispatchInfo, PostInfo=PostDispatchInfo> + IsSubType<Module<T>, T>,780{781 const IDENTIFIER: &'static str = "ChargeTransactionPayment";782 type AccountId = T::AccountId;783 type Call = T::Call;784 type AdditionalSigned = ();785 type Pre = (BalanceOf<T>, Self::AccountId, Option<NegativeImbalanceOf<T>>, BalanceOf<T>);786 fn additional_signed(&self) -> sp_std::result::Result<(), TransactionValidityError> { Ok(()) }787788 fn validate(789 &self,790 who: &Self::AccountId,791 call: &Self::Call,792 info: &DispatchInfoOf<Self::Call>,793 len: usize,794 ) -> TransactionValidity {795 let (fee, _) = self.withdraw_fee(who, call, info, len)?;796797 let mut r = ValidTransaction::default();798 799 800 r.priority = fee.saturated_into::<TransactionPriority>();801 Ok(r)802 }803804 fn pre_dispatch(805 self,806 who: &Self::AccountId,807 call: &Self::Call,808 info: &DispatchInfoOf<Self::Call>,809 len: usize810 ) -> Result<Self::Pre, TransactionValidityError> {811 let (fee, imbalance) = self.withdraw_fee(who, call, info, len)?;812 Ok((self.0, who.clone(), imbalance, fee))813 }814815 fn post_dispatch(816 pre: Self::Pre,817 info: &DispatchInfoOf<Self::Call>,818 post_info: &PostDispatchInfoOf<Self::Call>,819 len: usize,820 _result: &DispatchResult,821 ) -> Result<(), TransactionValidityError> {822 let (tip, who, imbalance, fee) = pre;823 if let Some(payed) = imbalance {824 let actual_fee = <transaction_payment::Module<T>>::compute_actual_fee(825 len as u32,826 info,827 post_info,828 tip,829 );830 let refund = fee.saturating_sub(actual_fee);831 let actual_payment = match <T as transaction_payment::Trait>::Currency::deposit_into_existing(&who, refund) {832 Ok(refund_imbalance) => {833 834 835 match payed.offset(refund_imbalance) {836 Ok(actual_payment) => actual_payment,837 Err(_) => return Err(InvalidTransaction::Payment.into()),838 }839 }840 841 842 Err(_) => payed,843 };844 let imbalances = actual_payment.split(tip);845 <T as transaction_payment::Trait>::OnTransactionPayment::on_unbalanceds(Some(imbalances.0).into_iter()846 .chain(Some(imbalances.1)));847 }848 Ok(())849 }850}