difftreelog
Add properties to RFT pallet
in: master
1 file changed
pallets/refungible/src/lib.rsdiffbeforeafterboth1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617//! # Refungible Pallet18//!19//! The Refungible pallet provides functionality for handling refungible collections and tokens.20//!21//! - [`Config`]22//! - [`RefungibleHandle`]23//! - [`Pallet`]24//! - [`CommonWeights`]25//!26//! ## Overview27//!28//! The Refungible pallet provides functions for:29//!30//! - RFT collection creation and removal31//! - Minting and burning of RFT tokens32//! - Partition and repartition of RFT tokens33//! - Retrieving number of pieces of RFT token34//! - Retrieving account balances35//! - Transfering RFT token pieces36//! - Burning RFT token pieces37//! - Setting and checking allowance for RFT tokens38//!39//! ### Terminology40//!41//! - **RFT token:** Non fungible token that was partitioned to pieces. If an account owns all42//! of the RFT token pieces than it owns the RFT token and can repartition it.43//!44//! - **RFT Collection:** A collection of RFT tokens. All RFT tokens are part of a collection.45//! Each collection has its own settings and set of permissions.46//!47//! - **RFT token piece:** A fungible part of an RFT token.48//!49//! - **Balance:** RFT token pieces owned by an account50//!51//! - **Allowance:** Maximum number of RFT token pieces that one account is allowed to52//! transfer from the balance of another account53//!54//! - **Burning:** The process of “deleting” a token from a collection or removing token pieces from55//! an account balance.56//!57//! ### Implementations58//!59//! The Refungible pallet provides implementations for the following traits. If these traits provide60//! the functionality that you need, then you can avoid coupling with the Refungible pallet.61//!62//! - [`CommonWeightInfo`](pallet_common::CommonWeightInfo): Functions for retrieval of transaction weight63//! - [`CommonCollectionOperations`](pallet_common::CommonCollectionOperations): Functions for dealing64//! with collections65//! - [`RefungibleExtensions`](pallet_common::RefungibleExtensions): Functions specific for refungible66//! collection67//!68//! ## Interface69//!70//! ### Dispatchable Functions71//!72//! - `init_collection` - Create RFT collection. RFT collection can be configured to allow or deny access for73//! some accounts.74//! - `destroy_collection` - Destroy exising RFT collection. There should be no tokens in the collection.75//! - `burn` - Burn some amount of RFT token pieces owned by account. Burns the RFT token if no pieces left.76//! - `transfer` - Transfer some amount of RFT token pieces. Transfers should be enabled for RFT collection.77//! Nests the RFT token if RFT token pieces are sent to another token.78//! - `create_item` - Mint RFT token in collection. Sender should have permission to mint tokens.79//! - `set_allowance` - Set allowance for another account to transfer balance from sender's account.80//! - `repartition` - Repartition token to selected number of pieces. Sender should own all existing pieces.81//!82//! ## Assumptions83//!84//! * Total number of pieces for one token shouldn't exceed `up_data_structs::MAX_REFUNGIBLE_PIECES`.85//! * Total number of tokens of all types shouldn't be greater than `up_data_structs::MAX_TOKEN_PREFIX_LENGTH`.86//! * Sender should be in collection's allow list to perform operations on tokens.8788#![cfg_attr(not(feature = "std"), no_std)]8990use frame_support::{ensure, BoundedVec};91use up_data_structs::{92 AccessMode, CollectionId, CustomDataLimit, MAX_REFUNGIBLE_PIECES, TokenId,93 CreateCollectionData, CreateRefungibleExData, mapping::TokenAddressMapping, budget::Budget,94};95use pallet_evm::account::CrossAccountId;96use pallet_common::{Error as CommonError, Event as CommonEvent, Pallet as PalletCommon};97use pallet_structure::Pallet as PalletStructure;98use sp_runtime::{ArithmeticError, DispatchError, DispatchResult};99use sp_std::{vec::Vec, vec, collections::btree_map::BTreeMap};100use core::ops::Deref;101use codec::{Encode, Decode, MaxEncodedLen};102use scale_info::TypeInfo;103104pub use pallet::*;105#[cfg(feature = "runtime-benchmarks")]106pub mod benchmarking;107pub mod common;108pub mod erc;109pub mod weights;110pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;111112#[struct_versioning::versioned(version = 2, upper)]113#[derive(Encode, Decode, Default, TypeInfo, MaxEncodedLen)]114pub struct ItemData {115 pub const_data: BoundedVec<u8, CustomDataLimit>,116117 #[version(..2)]118 pub variable_data: BoundedVec<u8, CustomDataLimit>,119}120121#[frame_support::pallet]122pub mod pallet {123 use super::*;124 use frame_support::{125 Blake2_128, Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key,126 traits::StorageVersion,127 };128 use frame_system::pallet_prelude::*;129 use up_data_structs::{CollectionId, TokenId};130 use super::weights::WeightInfo;131132 #[pallet::error]133 pub enum Error<T> {134 /// Not Refungible item data used to mint in Refungible collection.135 NotRefungibleDataUsedToMintFungibleCollectionToken,136 /// Maximum refungibility exceeded137 WrongRefungiblePieces,138 /// Refungible token can't be repartitioned by user who isn't owns all pieces139 RepartitionWhileNotOwningAllPieces,140 /// Refungible token can't nest other tokens141 RefungibleDisallowsNesting,142 /// Setting item properties is not allowed143 SettingPropertiesNotAllowed,144 }145146 #[pallet::config]147 pub trait Config:148 frame_system::Config + pallet_common::Config + pallet_structure::Config149 {150 type WeightInfo: WeightInfo;151 }152153 const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);154155 #[pallet::pallet]156 #[pallet::storage_version(STORAGE_VERSION)]157 #[pallet::generate_store(pub(super) trait Store)]158 pub struct Pallet<T>(_);159160 /// Amount of tokens minted for collection161 #[pallet::storage]162 pub type TokensMinted<T: Config> =163 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;164165 /// Amount of burnt tokens for collection166 #[pallet::storage]167 pub type TokensBurnt<T: Config> =168 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;169170 /// Custom data serialized to bytes for token171 #[pallet::storage]172 pub type TokenData<T: Config> = StorageNMap<173 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),174 Value = ItemData,175 QueryKind = ValueQuery,176 >;177178 /// Total amount of pieces for token179 #[pallet::storage]180 pub type TotalSupply<T: Config> = StorageNMap<181 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),182 Value = u128,183 QueryKind = ValueQuery,184 >;185186 /// Used to enumerate tokens owned by account187 #[pallet::storage]188 pub type Owned<T: Config> = StorageNMap<189 Key = (190 Key<Twox64Concat, CollectionId>,191 Key<Blake2_128Concat, T::CrossAccountId>,192 Key<Twox64Concat, TokenId>,193 ),194 Value = bool,195 QueryKind = ValueQuery,196 >;197198 /// Amount of tokens owned by account199 #[pallet::storage]200 pub type AccountBalance<T: Config> = StorageNMap<201 Key = (202 Key<Twox64Concat, CollectionId>,203 // Owner204 Key<Blake2_128Concat, T::CrossAccountId>,205 ),206 Value = u32,207 QueryKind = ValueQuery,208 >;209210 /// Amount of token pieces owned by account211 #[pallet::storage]212 pub type Balance<T: Config> = StorageNMap<213 Key = (214 Key<Twox64Concat, CollectionId>,215 Key<Twox64Concat, TokenId>,216 // Owner217 Key<Blake2_128Concat, T::CrossAccountId>,218 ),219 Value = u128,220 QueryKind = ValueQuery,221 >;222223 /// Allowance set by an owner for a spender for a token224 #[pallet::storage]225 pub type Allowance<T: Config> = StorageNMap<226 Key = (227 Key<Twox64Concat, CollectionId>,228 Key<Twox64Concat, TokenId>,229 // Owner230 Key<Blake2_128, T::CrossAccountId>,231 // Spender232 Key<Blake2_128Concat, T::CrossAccountId>,233 ),234 Value = u128,235 QueryKind = ValueQuery,236 >;237238 #[pallet::hooks]239 impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {240 fn on_runtime_upgrade() -> Weight {241 if StorageVersion::get::<Pallet<T>>() < StorageVersion::new(1) {242 <TokenData<T>>::translate_values::<ItemDataVersion1, _>(|v| {243 Some(<ItemDataVersion2>::from(v))244 })245 }246247 0248 }249 }250}251252pub struct RefungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);253impl<T: Config> RefungibleHandle<T> {254 pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {255 Self(inner)256 }257 pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {258 self.0259 }260}261impl<T: Config> Deref for RefungibleHandle<T> {262 type Target = pallet_common::CollectionHandle<T>;263264 fn deref(&self) -> &Self::Target {265 &self.0266 }267}268269impl<T: Config> Pallet<T> {270 /// Get number of RFT tokens in collection271 pub fn total_supply(collection: &RefungibleHandle<T>) -> u32 {272 <TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)273 }274275 /// Check that RFT token exists276 ///277 /// - `token`: Token ID.278 pub fn token_exists(collection: &RefungibleHandle<T>, token: TokenId) -> bool {279 <TotalSupply<T>>::contains_key((collection.id, token))280 }281}282283// unchecked calls skips any permission checks284impl<T: Config> Pallet<T> {285 /// Create RFT collection286 ///287 /// `init_collection` will take non-refundable deposit for collection creation.288 ///289 /// - `data`: Contains settings for collection limits and permissions.290 pub fn init_collection(291 owner: T::CrossAccountId,292 data: CreateCollectionData<T::AccountId>,293 ) -> Result<CollectionId, DispatchError> {294 <PalletCommon<T>>::init_collection(owner, data, false)295 }296297 /// Destroy RFT collection298 ///299 /// `destroy_collection` will throw error if collection contains any tokens.300 /// Only owner can destroy collection.301 pub fn destroy_collection(302 collection: RefungibleHandle<T>,303 sender: &T::CrossAccountId,304 ) -> DispatchResult {305 let id = collection.id;306307 if Self::collection_has_tokens(id) {308 return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());309 }310311 // =========312313 PalletCommon::destroy_collection(collection.0, sender)?;314315 <TokensMinted<T>>::remove(id);316 <TokensBurnt<T>>::remove(id);317 <TokenData<T>>::remove_prefix((id,), None);318 <TotalSupply<T>>::remove_prefix((id,), None);319 <Balance<T>>::remove_prefix((id,), None);320 <Allowance<T>>::remove_prefix((id,), None);321 <Owned<T>>::remove_prefix((id,), None);322 <AccountBalance<T>>::remove_prefix((id,), None);323 Ok(())324 }325326 fn collection_has_tokens(collection_id: CollectionId) -> bool {327 <TokenData<T>>::iter_prefix((collection_id,))328 .next()329 .is_some()330 }331332 pub fn burn_token_unchecked(333 collection: &RefungibleHandle<T>,334 token_id: TokenId,335 ) -> DispatchResult {336 let burnt = <TokensBurnt<T>>::get(collection.id)337 .checked_add(1)338 .ok_or(ArithmeticError::Overflow)?;339340 <TokensBurnt<T>>::insert(collection.id, burnt);341 <TokenData<T>>::remove((collection.id, token_id));342 <TotalSupply<T>>::remove((collection.id, token_id));343 <Balance<T>>::remove_prefix((collection.id, token_id), None);344 <Allowance<T>>::remove_prefix((collection.id, token_id), None);345 // TODO: ERC721 transfer event346 Ok(())347 }348349 /// Burn RFT token pieces350 ///351 /// `burn` will decrease total amount of token pieces and amount owned by sender.352 /// `burn` can be called even if there are multiple owners of the RFT token.353 /// If sender wouldn't have any pieces left after `burn` than she will stop being354 /// one of the owners of the token. If there is no account that owns any pieces of355 /// the token than token will be burned too.356 ///357 /// - `amount`: Amount of token pieces to burn.358 /// - `token`: Token who's pieces should be burned359 /// - `collection`: Collection that contains the token360 pub fn burn(361 collection: &RefungibleHandle<T>,362 owner: &T::CrossAccountId,363 token: TokenId,364 amount: u128,365 ) -> DispatchResult {366 let total_supply = <TotalSupply<T>>::get((collection.id, token))367 .checked_sub(amount)368 .ok_or(<CommonError<T>>::TokenValueTooLow)?;369370 // This was probally last owner of this token?371 if total_supply == 0 {372 // Ensure user actually owns this amount373 ensure!(374 <Balance<T>>::get((collection.id, token, owner)) == amount,375 <CommonError<T>>::TokenValueTooLow376 );377 let account_balance = <AccountBalance<T>>::get((collection.id, owner))378 .checked_sub(1)379 // Should not occur380 .ok_or(ArithmeticError::Underflow)?;381382 // =========383384 <Owned<T>>::remove((collection.id, owner, token));385 <PalletStructure<T>>::unnest_if_nested(owner, collection.id, token);386 <AccountBalance<T>>::insert((collection.id, owner), account_balance);387 Self::burn_token_unchecked(collection, token)?;388 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(389 collection.id,390 token,391 owner.clone(),392 amount,393 ));394 return Ok(());395 }396397 let balance = <Balance<T>>::get((collection.id, token, owner))398 .checked_sub(amount)399 .ok_or(<CommonError<T>>::TokenValueTooLow)?;400 let account_balance = if balance == 0 {401 <AccountBalance<T>>::get((collection.id, owner))402 .checked_sub(1)403 // Should not occur404 .ok_or(ArithmeticError::Underflow)?405 } else {406 0407 };408409 // =========410411 if balance == 0 {412 <Owned<T>>::remove((collection.id, owner, token));413 <PalletStructure<T>>::unnest_if_nested(owner, collection.id, token);414 <Balance<T>>::remove((collection.id, token, owner));415 <AccountBalance<T>>::insert((collection.id, owner), account_balance);416 } else {417 <Balance<T>>::insert((collection.id, token, owner), balance);418 }419 <TotalSupply<T>>::insert((collection.id, token), total_supply);420 // TODO: ERC20 transfer event421 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(422 collection.id,423 token,424 owner.clone(),425 amount,426 ));427 Ok(())428 }429430 /// Transfer RFT token pieces from one account to another.431 ///432 /// If the sender is no longer owns any pieces after the `transfer` than she stops being an owner of the token.433 ///434 /// - `from`: Owner of token pieces to transfer.435 /// - `to`: Recepient of transfered token pieces.436 /// - `amount`: Amount of token pieces to transfer.437 /// - `token`: Token whos pieces should be transfered438 /// - `collection`: Collection that contains the token439 pub fn transfer(440 collection: &RefungibleHandle<T>,441 from: &T::CrossAccountId,442 to: &T::CrossAccountId,443 token: TokenId,444 amount: u128,445 nesting_budget: &dyn Budget,446 ) -> DispatchResult {447 ensure!(448 collection.limits.transfers_enabled(),449 <CommonError<T>>::TransferNotAllowed450 );451452 if collection.permissions.access() == AccessMode::AllowList {453 collection.check_allowlist(from)?;454 collection.check_allowlist(to)?;455 }456 <PalletCommon<T>>::ensure_correct_receiver(to)?;457458 let balance_from = <Balance<T>>::get((collection.id, token, from))459 .checked_sub(amount)460 .ok_or(<CommonError<T>>::TokenValueTooLow)?;461 let mut create_target = false;462 let from_to_differ = from != to;463 let balance_to = if from != to {464 let old_balance = <Balance<T>>::get((collection.id, token, to));465 if old_balance == 0 {466 create_target = true;467 }468 Some(469 old_balance470 .checked_add(amount)471 .ok_or(ArithmeticError::Overflow)?,472 )473 } else {474 None475 };476477 let account_balance_from = if balance_from == 0 {478 Some(479 <AccountBalance<T>>::get((collection.id, from))480 .checked_sub(1)481 // Should not occur482 .ok_or(ArithmeticError::Underflow)?,483 )484 } else {485 None486 };487 // Account data is created in token, AccountBalance should be increased488 // But only if from != to as we shouldn't check overflow in this case489 let account_balance_to = if create_target && from_to_differ {490 let account_balance_to = <AccountBalance<T>>::get((collection.id, to))491 .checked_add(1)492 .ok_or(ArithmeticError::Overflow)?;493 ensure!(494 account_balance_to < collection.limits.account_token_ownership_limit(),495 <CommonError<T>>::AccountTokenLimitExceeded,496 );497498 Some(account_balance_to)499 } else {500 None501 };502503 // =========504505 <PalletStructure<T>>::nest_if_sent_to_token(506 from.clone(),507 to,508 collection.id,509 token,510 nesting_budget,511 )?;512513 if let Some(balance_to) = balance_to {514 // from != to515 if balance_from == 0 {516 <Balance<T>>::remove((collection.id, token, from));517 <PalletStructure<T>>::unnest_if_nested(from, collection.id, token);518 } else {519 <Balance<T>>::insert((collection.id, token, from), balance_from);520 }521 <Balance<T>>::insert((collection.id, token, to), balance_to);522 if let Some(account_balance_from) = account_balance_from {523 <AccountBalance<T>>::insert((collection.id, from), account_balance_from);524 <Owned<T>>::remove((collection.id, from, token));525 }526 if let Some(account_balance_to) = account_balance_to {527 <AccountBalance<T>>::insert((collection.id, to), account_balance_to);528 <Owned<T>>::insert((collection.id, to, token), true);529 }530 }531532 // TODO: ERC20 transfer event533 <PalletCommon<T>>::deposit_event(CommonEvent::Transfer(534 collection.id,535 token,536 from.clone(),537 to.clone(),538 amount,539 ));540 Ok(())541 }542543 /// Batched operation to create multiple RFT tokens.544 ///545 /// Same as `create_item` but creates multiple tokens.546 ///547 /// - `data`: Same as 'data` in `create_item` but contains data for multiple tokens.548 pub fn create_multiple_items(549 collection: &RefungibleHandle<T>,550 sender: &T::CrossAccountId,551 data: Vec<CreateRefungibleExData<T::CrossAccountId>>,552 nesting_budget: &dyn Budget,553 ) -> DispatchResult {554 if !collection.is_owner_or_admin(sender) {555 ensure!(556 collection.permissions.mint_mode(),557 <CommonError<T>>::PublicMintingNotAllowed558 );559 collection.check_allowlist(sender)?;560561 for item in data.iter() {562 for user in item.users.keys() {563 collection.check_allowlist(user)?;564 }565 }566 }567568 for item in data.iter() {569 for (owner, _) in item.users.iter() {570 <PalletCommon<T>>::ensure_correct_receiver(owner)?;571 }572 }573574 // Total pieces per tokens575 let totals = data576 .iter()577 .map(|data| {578 Ok(data579 .users580 .iter()581 .map(|u| u.1)582 .try_fold(0u128, |acc, v| acc.checked_add(*v))583 .ok_or(ArithmeticError::Overflow)?)584 })585 .collect::<Result<Vec<_>, DispatchError>>()?;586 for total in &totals {587 ensure!(588 *total <= MAX_REFUNGIBLE_PIECES,589 <Error<T>>::WrongRefungiblePieces590 );591 }592593 let first_token_id = <TokensMinted<T>>::get(collection.id);594 let tokens_minted = first_token_id595 .checked_add(data.len() as u32)596 .ok_or(ArithmeticError::Overflow)?;597 ensure!(598 tokens_minted < collection.limits.token_limit(),599 <CommonError<T>>::CollectionTokenLimitExceeded600 );601602 let mut balances = BTreeMap::new();603 for data in &data {604 for owner in data.users.keys() {605 let balance = balances606 .entry(owner)607 .or_insert_with(|| <AccountBalance<T>>::get((collection.id, owner)));608 *balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;609610 ensure!(611 *balance <= collection.limits.account_token_ownership_limit(),612 <CommonError<T>>::AccountTokenLimitExceeded,613 );614 }615 }616617 for (i, token) in data.iter().enumerate() {618 let token_id = TokenId(first_token_id + i as u32 + 1);619 for (to, _) in token.users.iter() {620 <PalletStructure<T>>::check_nesting(621 sender.clone(),622 to,623 collection.id,624 token_id,625 nesting_budget,626 )?;627 }628 }629630 // =========631632 <TokensMinted<T>>::insert(collection.id, tokens_minted);633 for (account, balance) in balances {634 <AccountBalance<T>>::insert((collection.id, account), balance);635 }636 for (i, token) in data.into_iter().enumerate() {637 let token_id = first_token_id + i as u32 + 1;638 <TotalSupply<T>>::insert((collection.id, token_id), totals[i]);639640 <TokenData<T>>::insert(641 (collection.id, token_id),642 ItemData {643 const_data: token.const_data,644 },645 );646647 for (user, amount) in token.users.into_iter() {648 if amount == 0 {649 continue;650 }651 <Balance<T>>::insert((collection.id, token_id, &user), amount);652 <Owned<T>>::insert((collection.id, &user, TokenId(token_id)), true);653 <PalletStructure<T>>::nest_if_sent_to_token_unchecked(654 &user,655 collection.id,656 TokenId(token_id),657 );658659 // TODO: ERC20 transfer event660 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(661 collection.id,662 TokenId(token_id),663 user,664 amount,665 ));666 }667 }668 Ok(())669 }670671 pub fn set_allowance_unchecked(672 collection: &RefungibleHandle<T>,673 sender: &T::CrossAccountId,674 spender: &T::CrossAccountId,675 token: TokenId,676 amount: u128,677 ) {678 if amount == 0 {679 <Allowance<T>>::remove((collection.id, token, sender, spender));680 } else {681 <Allowance<T>>::insert((collection.id, token, sender, spender), amount);682 }683 // TODO: ERC20 approval event684 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(685 collection.id,686 token,687 sender.clone(),688 spender.clone(),689 amount,690 ))691 }692693 /// Set allowance for the spender to `transfer` or `burn` sender's token pieces.694 ///695 /// - `amount`: Amount of token pieces the spender is allowed to `transfer` or `burn.696 pub fn set_allowance(697 collection: &RefungibleHandle<T>,698 sender: &T::CrossAccountId,699 spender: &T::CrossAccountId,700 token: TokenId,701 amount: u128,702 ) -> DispatchResult {703 if collection.permissions.access() == AccessMode::AllowList {704 collection.check_allowlist(sender)?;705 collection.check_allowlist(spender)?;706 }707708 <PalletCommon<T>>::ensure_correct_receiver(spender)?;709710 if <Balance<T>>::get((collection.id, token, sender)) < amount {711 ensure!(712 collection.ignores_owned_amount(sender) && Self::token_exists(collection, token),713 <CommonError<T>>::CantApproveMoreThanOwned714 );715 }716717 // =========718719 Self::set_allowance_unchecked(collection, sender, spender, token, amount);720 Ok(())721 }722723 /// Returns allowance, which should be set after transaction724 fn check_allowed(725 collection: &RefungibleHandle<T>,726 spender: &T::CrossAccountId,727 from: &T::CrossAccountId,728 token: TokenId,729 amount: u128,730 nesting_budget: &dyn Budget,731 ) -> Result<Option<u128>, DispatchError> {732 if spender.conv_eq(from) {733 return Ok(None);734 }735 if collection.permissions.access() == AccessMode::AllowList {736 // `from`, `to` checked in [`transfer`]737 collection.check_allowlist(spender)?;738 }739 if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {740 // TODO: should collection owner be allowed to perform this transfer?741 ensure!(742 <PalletStructure<T>>::check_indirectly_owned(743 spender.clone(),744 source.0,745 source.1,746 None,747 nesting_budget748 )?,749 <CommonError<T>>::ApprovedValueTooLow,750 );751 return Ok(None);752 }753 let allowance =754 <Allowance<T>>::get((collection.id, token, from, &spender)).checked_sub(amount);755 if allowance.is_none() {756 ensure!(757 collection.ignores_allowance(spender),758 <CommonError<T>>::ApprovedValueTooLow759 );760 }761 Ok(allowance)762 }763764 /// Transfer RFT token pieces from one account to another.765 ///766 /// Same as the [`transfer`] but spender doesn't needs to be an owner of the token pieces.767 /// The owner should set allowance for the spender to transfer pieces.768 ///769 /// [`transfer`]: struct.Pallet.html#method.transfer770 pub fn transfer_from(771 collection: &RefungibleHandle<T>,772 spender: &T::CrossAccountId,773 from: &T::CrossAccountId,774 to: &T::CrossAccountId,775 token: TokenId,776 amount: u128,777 nesting_budget: &dyn Budget,778 ) -> DispatchResult {779 let allowance =780 Self::check_allowed(collection, spender, from, token, amount, nesting_budget)?;781782 // =========783784 Self::transfer(collection, from, to, token, amount, nesting_budget)?;785 if let Some(allowance) = allowance {786 Self::set_allowance_unchecked(collection, from, spender, token, allowance);787 }788 Ok(())789 }790791 /// Burn RFT token pieces from the account.792 ///793 /// Same as the [`burn`] but spender doesn't need to be an owner of the token pieces. The owner should794 /// set allowance for the spender to burn pieces795 ///796 /// [`burn`]: struct.Pallet.html#method.burn797 pub fn burn_from(798 collection: &RefungibleHandle<T>,799 spender: &T::CrossAccountId,800 from: &T::CrossAccountId,801 token: TokenId,802 amount: u128,803 nesting_budget: &dyn Budget,804 ) -> DispatchResult {805 let allowance =806 Self::check_allowed(collection, spender, from, token, amount, nesting_budget)?;807808 // =========809810 Self::burn(collection, from, token, amount)?;811 if let Some(allowance) = allowance {812 Self::set_allowance_unchecked(collection, from, spender, token, allowance);813 }814 Ok(())815 }816817 /// Create RFT token.818 ///819 /// The sender should be the owner/admin of the collection or collection should be configured820 /// to allow public minting.821 ///822 /// - `data`: Contains list of users who will become the owners of the token pieces and amount823 /// of token pieces they will receive.824 pub fn create_item(825 collection: &RefungibleHandle<T>,826 sender: &T::CrossAccountId,827 data: CreateRefungibleExData<T::CrossAccountId>,828 nesting_budget: &dyn Budget,829 ) -> DispatchResult {830 Self::create_multiple_items(collection, sender, vec![data], nesting_budget)831 }832833 /// Repartition RFT token.834 ///835 /// `repartition` will set token balance of the sender and total amount of token pieces.836 /// Sender should own all of the token pieces. `repartition' could be done even if some837 /// token pieces were burned before.838 ///839 /// - `amount`: Total amount of token pieces that the token will have after `repartition`.840 pub fn repartition(841 collection: &RefungibleHandle<T>,842 owner: &T::CrossAccountId,843 token: TokenId,844 amount: u128,845 ) -> DispatchResult {846 ensure!(847 amount <= MAX_REFUNGIBLE_PIECES,848 <Error<T>>::WrongRefungiblePieces849 );850 ensure!(amount > 0, <CommonError<T>>::TokenValueTooLow);851 // Ensure user owns all pieces852 let total_supply = <TotalSupply<T>>::get((collection.id, token));853 let balance = <Balance<T>>::get((collection.id, token, owner));854 ensure!(855 total_supply == balance,856 <Error<T>>::RepartitionWhileNotOwningAllPieces857 );858859 <Balance<T>>::insert((collection.id, token, owner), amount);860 <TotalSupply<T>>::insert((collection.id, token), amount);861 Ok(())862 }863864 fn total_pieces(collection_id: CollectionId, token_id: TokenId) -> Option<u128> {865 <TotalSupply<T>>::try_get((collection_id, token_id)).ok()866 }867}1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617//! # Refungible Pallet18//!19//! The Refungible pallet provides functionality for handling refungible collections and tokens.20//!21//! - [`Config`]22//! - [`RefungibleHandle`]23//! - [`Pallet`]24//! - [`CommonWeights`]25//!26//! ## Overview27//!28//! The Refungible pallet provides functions for:29//!30//! - RFT collection creation and removal31//! - Minting and burning of RFT tokens32//! - Partition and repartition of RFT tokens33//! - Retrieving number of pieces of RFT token34//! - Retrieving account balances35//! - Transfering RFT token pieces36//! - Burning RFT token pieces37//! - Setting and checking allowance for RFT tokens38//!39//! ### Terminology40//!41//! - **RFT token:** Non fungible token that was partitioned to pieces. If an account owns all42//! of the RFT token pieces than it owns the RFT token and can repartition it.43//!44//! - **RFT Collection:** A collection of RFT tokens. All RFT tokens are part of a collection.45//! Each collection has its own settings and set of permissions.46//!47//! - **RFT token piece:** A fungible part of an RFT token.48//!49//! - **Balance:** RFT token pieces owned by an account50//!51//! - **Allowance:** Maximum number of RFT token pieces that one account is allowed to52//! transfer from the balance of another account53//!54//! - **Burning:** The process of “deleting” a token from a collection or removing token pieces from55//! an account balance.56//!57//! ### Implementations58//!59//! The Refungible pallet provides implementations for the following traits. If these traits provide60//! the functionality that you need, then you can avoid coupling with the Refungible pallet.61//!62//! - [`CommonWeightInfo`](pallet_common::CommonWeightInfo): Functions for retrieval of transaction weight63//! - [`CommonCollectionOperations`](pallet_common::CommonCollectionOperations): Functions for dealing64//! with collections65//! - [`RefungibleExtensions`](pallet_common::RefungibleExtensions): Functions specific for refungible66//! collection67//!68//! ## Interface69//!70//! ### Dispatchable Functions71//!72//! - `init_collection` - Create RFT collection. RFT collection can be configured to allow or deny access for73//! some accounts.74//! - `destroy_collection` - Destroy exising RFT collection. There should be no tokens in the collection.75//! - `burn` - Burn some amount of RFT token pieces owned by account. Burns the RFT token if no pieces left.76//! - `transfer` - Transfer some amount of RFT token pieces. Transfers should be enabled for RFT collection.77//! Nests the RFT token if RFT token pieces are sent to another token.78//! - `create_item` - Mint RFT token in collection. Sender should have permission to mint tokens.79//! - `set_allowance` - Set allowance for another account to transfer balance from sender's account.80//! - `repartition` - Repartition token to selected number of pieces. Sender should own all existing pieces.81//!82//! ## Assumptions83//!84//! * Total number of pieces for one token shouldn't exceed `up_data_structs::MAX_REFUNGIBLE_PIECES`.85//! * Total number of tokens of all types shouldn't be greater than `up_data_structs::MAX_TOKEN_PREFIX_LENGTH`.86//! * Sender should be in collection's allow list to perform operations on tokens.8788#![cfg_attr(not(feature = "std"), no_std)]8990use frame_support::{ensure, BoundedVec, transactional};91use up_data_structs::{92 AccessMode, CollectionId, CustomDataLimit, MAX_REFUNGIBLE_PIECES, TokenId,93 CreateCollectionData, CreateRefungibleExData, mapping::TokenAddressMapping, budget::Budget,94 Property, PropertyScope, TrySetProperty, PropertyKey, PropertyPermission95};96use pallet_evm::account::CrossAccountId;97use pallet_common::{Error as CommonError, Event as CommonEvent, Pallet as PalletCommon, CommonCollectionOperations as _};98use pallet_structure::Pallet as PalletStructure;99use sp_runtime::{ArithmeticError, DispatchError, DispatchResult};100use sp_std::{vec::Vec, vec, collections::btree_map::BTreeMap};101use core::ops::Deref;102use codec::{Encode, Decode, MaxEncodedLen};103use scale_info::TypeInfo;104105pub use pallet::*;106#[cfg(feature = "runtime-benchmarks")]107pub mod benchmarking;108pub mod common;109pub mod erc;110pub mod weights;111pub(crate) type SelfWeightOf<T> = <T as Config>::WeightInfo;112113#[struct_versioning::versioned(version = 2, upper)]114#[derive(Encode, Decode, Default, TypeInfo, MaxEncodedLen)]115pub struct ItemData {116 pub const_data: BoundedVec<u8, CustomDataLimit>,117118 #[version(..2)]119 pub variable_data: BoundedVec<u8, CustomDataLimit>,120}121122#[frame_support::pallet]123pub mod pallet {124 use super::*;125 use frame_support::{126 Blake2_128, Blake2_128Concat, Twox64Concat, pallet_prelude::*, storage::Key,127 traits::StorageVersion,128 };129 use frame_system::pallet_prelude::*;130 use up_data_structs::{CollectionId, TokenId};131 use super::weights::WeightInfo;132133 #[pallet::error]134 pub enum Error<T> {135 /// Not Refungible item data used to mint in Refungible collection.136 NotRefungibleDataUsedToMintFungibleCollectionToken,137 /// Maximum refungibility exceeded138 WrongRefungiblePieces,139 /// Refungible token can't be repartitioned by user who isn't owns all pieces140 RepartitionWhileNotOwningAllPieces,141 /// Refungible token can't nest other tokens142 RefungibleDisallowsNesting,143 /// Setting item properties is not allowed144 SettingPropertiesNotAllowed,145 }146147 #[pallet::config]148 pub trait Config:149 frame_system::Config + pallet_common::Config + pallet_structure::Config150 {151 type WeightInfo: WeightInfo;152 }153154 const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);155156 #[pallet::pallet]157 #[pallet::storage_version(STORAGE_VERSION)]158 #[pallet::generate_store(pub(super) trait Store)]159 pub struct Pallet<T>(_);160161 /// Amount of tokens minted for collection162 #[pallet::storage]163 pub type TokensMinted<T: Config> =164 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;165166 /// Amount of burnt tokens for collection167 #[pallet::storage]168 pub type TokensBurnt<T: Config> =169 StorageMap<Hasher = Twox64Concat, Key = CollectionId, Value = u32, QueryKind = ValueQuery>;170171 /// Custom data serialized to bytes for token172 #[pallet::storage]173 pub type TokenData<T: Config> = StorageNMap<174 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),175 Value = ItemData,176 QueryKind = ValueQuery,177 >;178179 #[pallet::storage]180 #[pallet::getter(fn token_properties)]181 pub type TokenProperties<T: Config> = StorageNMap<182 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),183 Value = up_data_structs::Properties,184 QueryKind = ValueQuery,185 OnEmpty = up_data_structs::TokenProperties,186 >;187188 /// Total amount of pieces for token189 #[pallet::storage]190 pub type TotalSupply<T: Config> = StorageNMap<191 Key = (Key<Twox64Concat, CollectionId>, Key<Twox64Concat, TokenId>),192 Value = u128,193 QueryKind = ValueQuery,194 >;195196 /// Used to enumerate tokens owned by account197 #[pallet::storage]198 pub type Owned<T: Config> = StorageNMap<199 Key = (200 Key<Twox64Concat, CollectionId>,201 Key<Blake2_128Concat, T::CrossAccountId>,202 Key<Twox64Concat, TokenId>,203 ),204 Value = bool,205 QueryKind = ValueQuery,206 >;207208 /// Amount of tokens owned by account209 #[pallet::storage]210 pub type AccountBalance<T: Config> = StorageNMap<211 Key = (212 Key<Twox64Concat, CollectionId>,213 // Owner214 Key<Blake2_128Concat, T::CrossAccountId>,215 ),216 Value = u32,217 QueryKind = ValueQuery,218 >;219220 /// Amount of token pieces owned by account221 #[pallet::storage]222 pub type Balance<T: Config> = StorageNMap<223 Key = (224 Key<Twox64Concat, CollectionId>,225 Key<Twox64Concat, TokenId>,226 // Owner227 Key<Blake2_128Concat, T::CrossAccountId>,228 ),229 Value = u128,230 QueryKind = ValueQuery,231 >;232233 /// Allowance set by an owner for a spender for a token234 #[pallet::storage]235 pub type Allowance<T: Config> = StorageNMap<236 Key = (237 Key<Twox64Concat, CollectionId>,238 Key<Twox64Concat, TokenId>,239 // Owner240 Key<Blake2_128, T::CrossAccountId>,241 // Spender242 Key<Blake2_128Concat, T::CrossAccountId>,243 ),244 Value = u128,245 QueryKind = ValueQuery,246 >;247248 #[pallet::hooks]249 impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {250 fn on_runtime_upgrade() -> Weight {251 if StorageVersion::get::<Pallet<T>>() < StorageVersion::new(1) {252 <TokenData<T>>::translate_values::<ItemDataVersion1, _>(|v| {253 Some(<ItemDataVersion2>::from(v))254 })255 }256257 0258 }259 }260}261262pub struct RefungibleHandle<T: Config>(pallet_common::CollectionHandle<T>);263impl<T: Config> RefungibleHandle<T> {264 pub fn cast(inner: pallet_common::CollectionHandle<T>) -> Self {265 Self(inner)266 }267 pub fn into_inner(self) -> pallet_common::CollectionHandle<T> {268 self.0269 }270}271impl<T: Config> Deref for RefungibleHandle<T> {272 type Target = pallet_common::CollectionHandle<T>;273274 fn deref(&self) -> &Self::Target {275 &self.0276 }277}278279impl<T: Config> Pallet<T> {280 /// Get number of RFT tokens in collection281 pub fn total_supply(collection: &RefungibleHandle<T>) -> u32 {282 <TokensMinted<T>>::get(collection.id) - <TokensBurnt<T>>::get(collection.id)283 }284285 /// Check that RFT token exists286 ///287 /// - `token`: Token ID.288 pub fn token_exists(collection: &RefungibleHandle<T>, token: TokenId) -> bool {289 <TotalSupply<T>>::contains_key((collection.id, token))290 }291292 pub fn set_scoped_token_property(293 collection_id: CollectionId,294 token_id: TokenId,295 scope: PropertyScope,296 property: Property,297 ) -> DispatchResult {298 TokenProperties::<T>::try_mutate((collection_id, token_id), |properties| {299 properties.try_scoped_set(scope, property.key, property.value)300 })301 .map_err(<CommonError<T>>::from)?;302303 Ok(())304 }305306 pub fn set_scoped_token_properties(307 collection_id: CollectionId,308 token_id: TokenId,309 scope: PropertyScope,310 properties: impl Iterator<Item = Property>,311 ) -> DispatchResult {312 TokenProperties::<T>::try_mutate((collection_id, token_id), |stored_properties| {313 stored_properties.try_scoped_set_from_iter(scope, properties)314 })315 .map_err(<CommonError<T>>::from)?;316317 Ok(())318 }319}320321// unchecked calls skips any permission checks322impl<T: Config> Pallet<T> {323 /// Create RFT collection324 ///325 /// `init_collection` will take non-refundable deposit for collection creation.326 ///327 /// - `data`: Contains settings for collection limits and permissions.328 pub fn init_collection(329 owner: T::CrossAccountId,330 data: CreateCollectionData<T::AccountId>,331 ) -> Result<CollectionId, DispatchError> {332 <PalletCommon<T>>::init_collection(owner, data, false)333 }334335 /// Destroy RFT collection336 ///337 /// `destroy_collection` will throw error if collection contains any tokens.338 /// Only owner can destroy collection.339 pub fn destroy_collection(340 collection: RefungibleHandle<T>,341 sender: &T::CrossAccountId,342 ) -> DispatchResult {343 let id = collection.id;344345 if Self::collection_has_tokens(id) {346 return Err(<CommonError<T>>::CantDestroyNotEmptyCollection.into());347 }348349 // =========350351 PalletCommon::destroy_collection(collection.0, sender)?;352353 <TokensMinted<T>>::remove(id);354 <TokensBurnt<T>>::remove(id);355 <TokenData<T>>::remove_prefix((id,), None);356 <TotalSupply<T>>::remove_prefix((id,), None);357 <Balance<T>>::remove_prefix((id,), None);358 <Allowance<T>>::remove_prefix((id,), None);359 <Owned<T>>::remove_prefix((id,), None);360 <AccountBalance<T>>::remove_prefix((id,), None);361 Ok(())362 }363364 fn collection_has_tokens(collection_id: CollectionId) -> bool {365 <TokenData<T>>::iter_prefix((collection_id,))366 .next()367 .is_some()368 }369370 pub fn burn_token_unchecked(371 collection: &RefungibleHandle<T>,372 token_id: TokenId,373 ) -> DispatchResult {374 let burnt = <TokensBurnt<T>>::get(collection.id)375 .checked_add(1)376 .ok_or(ArithmeticError::Overflow)?;377378 <TokensBurnt<T>>::insert(collection.id, burnt);379 <TokenData<T>>::remove((collection.id, token_id));380 <TokenProperties<T>>::remove((collection.id, token_id));381 <TotalSupply<T>>::remove((collection.id, token_id));382 <Balance<T>>::remove_prefix((collection.id, token_id), None);383 <Allowance<T>>::remove_prefix((collection.id, token_id), None);384 // TODO: ERC721 transfer event385 Ok(())386 }387388 /// Burn RFT token pieces389 ///390 /// `burn` will decrease total amount of token pieces and amount owned by sender.391 /// `burn` can be called even if there are multiple owners of the RFT token.392 /// If sender wouldn't have any pieces left after `burn` than she will stop being393 /// one of the owners of the token. If there is no account that owns any pieces of394 /// the token than token will be burned too.395 ///396 /// - `amount`: Amount of token pieces to burn.397 /// - `token`: Token who's pieces should be burned398 /// - `collection`: Collection that contains the token399 pub fn burn(400 collection: &RefungibleHandle<T>,401 owner: &T::CrossAccountId,402 token: TokenId,403 amount: u128,404 ) -> DispatchResult {405 let total_supply = <TotalSupply<T>>::get((collection.id, token))406 .checked_sub(amount)407 .ok_or(<CommonError<T>>::TokenValueTooLow)?;408409 // This was probally last owner of this token?410 if total_supply == 0 {411 // Ensure user actually owns this amount412 ensure!(413 <Balance<T>>::get((collection.id, token, owner)) == amount,414 <CommonError<T>>::TokenValueTooLow415 );416 let account_balance = <AccountBalance<T>>::get((collection.id, owner))417 .checked_sub(1)418 // Should not occur419 .ok_or(ArithmeticError::Underflow)?;420421 // =========422423 <Owned<T>>::remove((collection.id, owner, token));424 <PalletStructure<T>>::unnest_if_nested(owner, collection.id, token);425 <AccountBalance<T>>::insert((collection.id, owner), account_balance);426 Self::burn_token_unchecked(collection, token)?;427 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(428 collection.id,429 token,430 owner.clone(),431 amount,432 ));433 return Ok(());434 }435436 let balance = <Balance<T>>::get((collection.id, token, owner))437 .checked_sub(amount)438 .ok_or(<CommonError<T>>::TokenValueTooLow)?;439 let account_balance = if balance == 0 {440 <AccountBalance<T>>::get((collection.id, owner))441 .checked_sub(1)442 // Should not occur443 .ok_or(ArithmeticError::Underflow)?444 } else {445 0446 };447448 // =========449450 if balance == 0 {451 <Owned<T>>::remove((collection.id, owner, token));452 <PalletStructure<T>>::unnest_if_nested(owner, collection.id, token);453 <Balance<T>>::remove((collection.id, token, owner));454 <AccountBalance<T>>::insert((collection.id, owner), account_balance);455 } else {456 <Balance<T>>::insert((collection.id, token, owner), balance);457 }458 <TotalSupply<T>>::insert((collection.id, token), total_supply);459 // TODO: ERC20 transfer event460 <PalletCommon<T>>::deposit_event(CommonEvent::ItemDestroyed(461 collection.id,462 token,463 owner.clone(),464 amount,465 ));466 Ok(())467 }468469 pub fn set_token_property(470 collection: &RefungibleHandle<T>,471 sender: &T::CrossAccountId,472 token_id: TokenId,473 property: Property,474 is_token_create: bool,475 ) -> DispatchResult {476 Self::check_token_change_permission(477 collection,478 sender,479 token_id,480 &property.key,481 is_token_create,482 )?;483484 <TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {485 let property = property.clone();486 properties.try_set(property.key, property.value)487 })488 .map_err(<CommonError<T>>::from)?;489490 <PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertySet(491 collection.id,492 token_id,493 property.key,494 ));495496 Ok(())497 }498499 #[transactional]500 pub fn set_token_properties(501 collection: &RefungibleHandle<T>,502 sender: &T::CrossAccountId,503 token_id: TokenId,504 properties: Vec<Property>,505 is_token_create: bool,506 ) -> DispatchResult {507 for property in properties {508 Self::set_token_property(collection, sender, token_id, property, is_token_create)?;509 }510511 Ok(())512 }513514 pub fn delete_token_property(515 collection: &RefungibleHandle<T>,516 sender: &T::CrossAccountId,517 token_id: TokenId,518 property_key: PropertyKey,519 ) -> DispatchResult {520 Self::check_token_change_permission(collection, sender, token_id, &property_key, false)?;521522 <TokenProperties<T>>::try_mutate((collection.id, token_id), |properties| {523 properties.remove(&property_key)524 })525 .map_err(<CommonError<T>>::from)?;526527 <PalletCommon<T>>::deposit_event(CommonEvent::TokenPropertyDeleted(528 collection.id,529 token_id,530 property_key,531 ));532533 Ok(())534 }535536 fn check_token_change_permission(537 collection: &RefungibleHandle<T>,538 sender: &T::CrossAccountId,539 token_id: TokenId,540 property_key: &PropertyKey,541 is_token_create: bool,542 ) -> DispatchResult {543 let permission = <PalletCommon<T>>::property_permissions(collection.id)544 .get(property_key)545 .cloned()546 .unwrap_or_else(PropertyPermission::none);547548 // Not "try_fold" because total count of pieces is limited by 'MAX_REFUNGIBLE_PIECES'.549 let total_pieces: u128 = <Balance<T>>::iter_prefix((collection.id, token_id,)).fold(0, |total, piece| total + piece.1);550 let balance = collection.balance(sender.clone(), token_id);551552 let check_token_owner = || -> DispatchResult {553 ensure!(balance == total_pieces, <CommonError<T>>::NoPermission);554 Ok(())555 };556557 let is_property_exists = TokenProperties::<T>::get((collection.id, token_id))558 .get(property_key)559 .is_some();560561 match permission {562 PropertyPermission { mutable: false, .. } if is_property_exists => {563 Err(<CommonError<T>>::NoPermission.into())564 }565566 PropertyPermission {567 collection_admin,568 token_owner,569 ..570 } => {571 //TODO: investigate threats during public minting.572 if is_token_create && (collection_admin || token_owner) {573 return Ok(());574 }575576 let mut check_result = Err(<CommonError<T>>::NoPermission.into());577578 if collection_admin {579 check_result = collection.check_is_owner_or_admin(sender);580 }581582 if token_owner {583 check_result.or_else(|_| check_token_owner())584 } else {585 check_result586 }587 }588 }589 }590591 #[transactional]592 pub fn delete_token_properties(593 collection: &RefungibleHandle<T>,594 sender: &T::CrossAccountId,595 token_id: TokenId,596 property_keys: Vec<PropertyKey>,597 ) -> DispatchResult {598 for key in property_keys {599 Self::delete_token_property(collection, sender, token_id, key)?;600 }601602 Ok(())603 }604605 /// Transfer RFT token pieces from one account to another.606 ///607 /// If the sender is no longer owns any pieces after the `transfer` than she stops being an owner of the token.608 ///609 /// - `from`: Owner of token pieces to transfer.610 /// - `to`: Recepient of transfered token pieces.611 /// - `amount`: Amount of token pieces to transfer.612 /// - `token`: Token whos pieces should be transfered613 /// - `collection`: Collection that contains the token614 pub fn transfer(615 collection: &RefungibleHandle<T>,616 from: &T::CrossAccountId,617 to: &T::CrossAccountId,618 token: TokenId,619 amount: u128,620 nesting_budget: &dyn Budget,621 ) -> DispatchResult {622 ensure!(623 collection.limits.transfers_enabled(),624 <CommonError<T>>::TransferNotAllowed625 );626627 if collection.permissions.access() == AccessMode::AllowList {628 collection.check_allowlist(from)?;629 collection.check_allowlist(to)?;630 }631 <PalletCommon<T>>::ensure_correct_receiver(to)?;632633 let balance_from = <Balance<T>>::get((collection.id, token, from))634 .checked_sub(amount)635 .ok_or(<CommonError<T>>::TokenValueTooLow)?;636 let mut create_target = false;637 let from_to_differ = from != to;638 let balance_to = if from != to {639 let old_balance = <Balance<T>>::get((collection.id, token, to));640 if old_balance == 0 {641 create_target = true;642 }643 Some(644 old_balance645 .checked_add(amount)646 .ok_or(ArithmeticError::Overflow)?,647 )648 } else {649 None650 };651652 let account_balance_from = if balance_from == 0 {653 Some(654 <AccountBalance<T>>::get((collection.id, from))655 .checked_sub(1)656 // Should not occur657 .ok_or(ArithmeticError::Underflow)?,658 )659 } else {660 None661 };662 // Account data is created in token, AccountBalance should be increased663 // But only if from != to as we shouldn't check overflow in this case664 let account_balance_to = if create_target && from_to_differ {665 let account_balance_to = <AccountBalance<T>>::get((collection.id, to))666 .checked_add(1)667 .ok_or(ArithmeticError::Overflow)?;668 ensure!(669 account_balance_to < collection.limits.account_token_ownership_limit(),670 <CommonError<T>>::AccountTokenLimitExceeded,671 );672673 Some(account_balance_to)674 } else {675 None676 };677678 // =========679680 <PalletStructure<T>>::nest_if_sent_to_token(681 from.clone(),682 to,683 collection.id,684 token,685 nesting_budget,686 )?;687688 if let Some(balance_to) = balance_to {689 // from != to690 if balance_from == 0 {691 <Balance<T>>::remove((collection.id, token, from));692 <PalletStructure<T>>::unnest_if_nested(from, collection.id, token);693 } else {694 <Balance<T>>::insert((collection.id, token, from), balance_from);695 }696 <Balance<T>>::insert((collection.id, token, to), balance_to);697 if let Some(account_balance_from) = account_balance_from {698 <AccountBalance<T>>::insert((collection.id, from), account_balance_from);699 <Owned<T>>::remove((collection.id, from, token));700 }701 if let Some(account_balance_to) = account_balance_to {702 <AccountBalance<T>>::insert((collection.id, to), account_balance_to);703 <Owned<T>>::insert((collection.id, to, token), true);704 }705 }706707 // TODO: ERC20 transfer event708 <PalletCommon<T>>::deposit_event(CommonEvent::Transfer(709 collection.id,710 token,711 from.clone(),712 to.clone(),713 amount,714 ));715 Ok(())716 }717718 /// Batched operation to create multiple RFT tokens.719 ///720 /// Same as `create_item` but creates multiple tokens.721 ///722 /// - `data`: Same as 'data` in `create_item` but contains data for multiple tokens.723 pub fn create_multiple_items(724 collection: &RefungibleHandle<T>,725 sender: &T::CrossAccountId,726 data: Vec<CreateRefungibleExData<T::CrossAccountId>>,727 nesting_budget: &dyn Budget,728 ) -> DispatchResult {729 if !collection.is_owner_or_admin(sender) {730 ensure!(731 collection.permissions.mint_mode(),732 <CommonError<T>>::PublicMintingNotAllowed733 );734 collection.check_allowlist(sender)?;735736 for item in data.iter() {737 for user in item.users.keys() {738 collection.check_allowlist(user)?;739 }740 }741 }742743 for item in data.iter() {744 for (owner, _) in item.users.iter() {745 <PalletCommon<T>>::ensure_correct_receiver(owner)?;746 }747 }748749 // Total pieces per tokens750 let totals = data751 .iter()752 .map(|data| {753 Ok(data754 .users755 .iter()756 .map(|u| u.1)757 .try_fold(0u128, |acc, v| acc.checked_add(*v))758 .ok_or(ArithmeticError::Overflow)?)759 })760 .collect::<Result<Vec<_>, DispatchError>>()?;761 for total in &totals {762 ensure!(763 *total <= MAX_REFUNGIBLE_PIECES,764 <Error<T>>::WrongRefungiblePieces765 );766 }767768 let first_token_id = <TokensMinted<T>>::get(collection.id);769 let tokens_minted = first_token_id770 .checked_add(data.len() as u32)771 .ok_or(ArithmeticError::Overflow)?;772 ensure!(773 tokens_minted < collection.limits.token_limit(),774 <CommonError<T>>::CollectionTokenLimitExceeded775 );776777 let mut balances = BTreeMap::new();778 for data in &data {779 for owner in data.users.keys() {780 let balance = balances781 .entry(owner)782 .or_insert_with(|| <AccountBalance<T>>::get((collection.id, owner)));783 *balance = balance.checked_add(1).ok_or(ArithmeticError::Overflow)?;784785 ensure!(786 *balance <= collection.limits.account_token_ownership_limit(),787 <CommonError<T>>::AccountTokenLimitExceeded,788 );789 }790 }791792 for (i, token) in data.iter().enumerate() {793 let token_id = TokenId(first_token_id + i as u32 + 1);794 for (to, _) in token.users.iter() {795 <PalletStructure<T>>::check_nesting(796 sender.clone(),797 to,798 collection.id,799 token_id,800 nesting_budget,801 )?;802 }803 }804805 // =========806807 <TokensMinted<T>>::insert(collection.id, tokens_minted);808 for (account, balance) in balances {809 <AccountBalance<T>>::insert((collection.id, account), balance);810 }811 for (i, token) in data.into_iter().enumerate() {812 let token_id = first_token_id + i as u32 + 1;813 <TotalSupply<T>>::insert((collection.id, token_id), totals[i]);814815 <TokenData<T>>::insert(816 (collection.id, token_id),817 ItemData {818 const_data: token.const_data,819 },820 );821822 for (user, amount) in token.users.into_iter() {823 if amount == 0 {824 continue;825 }826 <Balance<T>>::insert((collection.id, token_id, &user), amount);827 <Owned<T>>::insert((collection.id, &user, TokenId(token_id)), true);828 <PalletStructure<T>>::nest_if_sent_to_token_unchecked(829 &user,830 collection.id,831 TokenId(token_id),832 );833834 // TODO: ERC20 transfer event835 <PalletCommon<T>>::deposit_event(CommonEvent::ItemCreated(836 collection.id,837 TokenId(token_id),838 user,839 amount,840 ));841 }842 }843 Ok(())844 }845846 pub fn set_allowance_unchecked(847 collection: &RefungibleHandle<T>,848 sender: &T::CrossAccountId,849 spender: &T::CrossAccountId,850 token: TokenId,851 amount: u128,852 ) {853 if amount == 0 {854 <Allowance<T>>::remove((collection.id, token, sender, spender));855 } else {856 <Allowance<T>>::insert((collection.id, token, sender, spender), amount);857 }858 // TODO: ERC20 approval event859 <PalletCommon<T>>::deposit_event(CommonEvent::Approved(860 collection.id,861 token,862 sender.clone(),863 spender.clone(),864 amount,865 ))866 }867868 /// Set allowance for the spender to `transfer` or `burn` sender's token pieces.869 ///870 /// - `amount`: Amount of token pieces the spender is allowed to `transfer` or `burn.871 pub fn set_allowance(872 collection: &RefungibleHandle<T>,873 sender: &T::CrossAccountId,874 spender: &T::CrossAccountId,875 token: TokenId,876 amount: u128,877 ) -> DispatchResult {878 if collection.permissions.access() == AccessMode::AllowList {879 collection.check_allowlist(sender)?;880 collection.check_allowlist(spender)?;881 }882883 <PalletCommon<T>>::ensure_correct_receiver(spender)?;884885 if <Balance<T>>::get((collection.id, token, sender)) < amount {886 ensure!(887 collection.ignores_owned_amount(sender) && Self::token_exists(collection, token),888 <CommonError<T>>::CantApproveMoreThanOwned889 );890 }891892 // =========893894 Self::set_allowance_unchecked(collection, sender, spender, token, amount);895 Ok(())896 }897898 /// Returns allowance, which should be set after transaction899 fn check_allowed(900 collection: &RefungibleHandle<T>,901 spender: &T::CrossAccountId,902 from: &T::CrossAccountId,903 token: TokenId,904 amount: u128,905 nesting_budget: &dyn Budget,906 ) -> Result<Option<u128>, DispatchError> {907 if spender.conv_eq(from) {908 return Ok(None);909 }910 if collection.permissions.access() == AccessMode::AllowList {911 // `from`, `to` checked in [`transfer`]912 collection.check_allowlist(spender)?;913 }914 if let Some(source) = T::CrossTokenAddressMapping::address_to_token(from) {915 // TODO: should collection owner be allowed to perform this transfer?916 ensure!(917 <PalletStructure<T>>::check_indirectly_owned(918 spender.clone(),919 source.0,920 source.1,921 None,922 nesting_budget923 )?,924 <CommonError<T>>::ApprovedValueTooLow,925 );926 return Ok(None);927 }928 let allowance =929 <Allowance<T>>::get((collection.id, token, from, &spender)).checked_sub(amount);930 if allowance.is_none() {931 ensure!(932 collection.ignores_allowance(spender),933 <CommonError<T>>::ApprovedValueTooLow934 );935 }936 Ok(allowance)937 }938939 /// Transfer RFT token pieces from one account to another.940 ///941 /// Same as the [`transfer`] but spender doesn't needs to be an owner of the token pieces.942 /// The owner should set allowance for the spender to transfer pieces.943 ///944 /// [`transfer`]: struct.Pallet.html#method.transfer945 pub fn transfer_from(946 collection: &RefungibleHandle<T>,947 spender: &T::CrossAccountId,948 from: &T::CrossAccountId,949 to: &T::CrossAccountId,950 token: TokenId,951 amount: u128,952 nesting_budget: &dyn Budget,953 ) -> DispatchResult {954 let allowance =955 Self::check_allowed(collection, spender, from, token, amount, nesting_budget)?;956957 // =========958959 Self::transfer(collection, from, to, token, amount, nesting_budget)?;960 if let Some(allowance) = allowance {961 Self::set_allowance_unchecked(collection, from, spender, token, allowance);962 }963 Ok(())964 }965966 /// Burn RFT token pieces from the account.967 ///968 /// Same as the [`burn`] but spender doesn't need to be an owner of the token pieces. The owner should969 /// set allowance for the spender to burn pieces970 ///971 /// [`burn`]: struct.Pallet.html#method.burn972 pub fn burn_from(973 collection: &RefungibleHandle<T>,974 spender: &T::CrossAccountId,975 from: &T::CrossAccountId,976 token: TokenId,977 amount: u128,978 nesting_budget: &dyn Budget,979 ) -> DispatchResult {980 let allowance =981 Self::check_allowed(collection, spender, from, token, amount, nesting_budget)?;982983 // =========984985 Self::burn(collection, from, token, amount)?;986 if let Some(allowance) = allowance {987 Self::set_allowance_unchecked(collection, from, spender, token, allowance);988 }989 Ok(())990 }991992 /// Create RFT token.993 ///994 /// The sender should be the owner/admin of the collection or collection should be configured995 /// to allow public minting.996 ///997 /// - `data`: Contains list of users who will become the owners of the token pieces and amount998 /// of token pieces they will receive.999 pub fn create_item(1000 collection: &RefungibleHandle<T>,1001 sender: &T::CrossAccountId,1002 data: CreateRefungibleExData<T::CrossAccountId>,1003 nesting_budget: &dyn Budget,1004 ) -> DispatchResult {1005 Self::create_multiple_items(collection, sender, vec![data], nesting_budget)1006 }10071008 /// Repartition RFT token.1009 ///1010 /// `repartition` will set token balance of the sender and total amount of token pieces.1011 /// Sender should own all of the token pieces. `repartition' could be done even if some1012 /// token pieces were burned before.1013 ///1014 /// - `amount`: Total amount of token pieces that the token will have after `repartition`.1015 pub fn repartition(1016 collection: &RefungibleHandle<T>,1017 owner: &T::CrossAccountId,1018 token: TokenId,1019 amount: u128,1020 ) -> DispatchResult {1021 ensure!(1022 amount <= MAX_REFUNGIBLE_PIECES,1023 <Error<T>>::WrongRefungiblePieces1024 );1025 ensure!(amount > 0, <CommonError<T>>::TokenValueTooLow);1026 // Ensure user owns all pieces1027 let total_supply = <TotalSupply<T>>::get((collection.id, token));1028 let balance = <Balance<T>>::get((collection.id, token, owner));1029 ensure!(1030 total_supply == balance,1031 <Error<T>>::RepartitionWhileNotOwningAllPieces1032 );10331034 <Balance<T>>::insert((collection.id, token, owner), amount);1035 <TotalSupply<T>>::insert((collection.id, token), amount);1036 Ok(())1037 }10381039 fn total_pieces(collection_id: CollectionId, token_id: TokenId) -> Option<u128> {1040 <TotalSupply<T>>::try_get((collection_id, token_id)).ok()1041 }1042}