1234567891011121314151617#![cfg_attr(not(feature = "std"), no_std)]1819use core::{20 convert::{TryFrom, TryInto},21 fmt,22};23use frame_support::{24 storage::{bounded_btree_map::BoundedBTreeMap, bounded_btree_set::BoundedBTreeSet},25 traits::Get,26 parameter_types,27};2829#[cfg(feature = "serde")]30use serde::{Serialize, Deserialize};3132use sp_core::U256;33use sp_runtime::{ArithmeticError, sp_std::prelude::Vec, Permill};34use codec::{Decode, Encode, EncodeLike, MaxEncodedLen};35use frame_support::{BoundedVec, traits::ConstU32};36use derivative::Derivative;37use scale_info::TypeInfo;3839pub mod rmrk;404142use rmrk::{43 CollectionInfo, NftInfo, ResourceInfo, PropertyInfo, BaseInfo, PartType, Theme, ThemeProperty,44 ResourceTypes, BasicResource, ComposableResource, SlotResource,45};46pub use rmrk::{47 primitives::{48 CollectionId as RmrkCollectionId, NftId as RmrkNftId, BaseId as RmrkBaseId,49 PartId as RmrkPartId, ResourceId as RmrkResourceId,50 },51 NftChild as RmrkNftChild, AccountIdOrCollectionNftTuple as RmrkAccountIdOrCollectionNftTuple,52 FixedPart as RmrkFixedPart, SlotPart as RmrkSlotPart, EquippableList as RmrkEquippableList,53};5455mod bounded;56pub mod budget;57pub mod mapping;58mod migration;5960pub const MAX_DECIMAL_POINTS: DecimalPoints = 30;61pub const MAX_REFUNGIBLE_PIECES: u128 = 1_000_000_000_000_000_000_000;62pub const MAX_SPONSOR_TIMEOUT: u32 = 10_368_000;6364pub const MAX_TOKEN_OWNERSHIP: u32 = if cfg!(not(feature = "limit-testing")) {65 100_00066} else {67 1068};69pub const COLLECTION_NUMBER_LIMIT: u32 = if cfg!(not(feature = "limit-testing")) {70 100_00071} else {72 1073};74pub const CUSTOM_DATA_LIMIT: u32 = if cfg!(not(feature = "limit-testing")) {75 204876} else {77 1078};79pub const COLLECTION_ADMINS_LIMIT: u32 = 5;80pub const COLLECTION_TOKEN_LIMIT: u32 = u32::MAX;81pub const ACCOUNT_TOKEN_OWNERSHIP_LIMIT: u32 = if cfg!(not(feature = "limit-testing")) {82 1_000_00083} else {84 1085};868788pub const NFT_SPONSOR_TRANSFER_TIMEOUT: u32 = 5;89pub const FUNGIBLE_SPONSOR_TRANSFER_TIMEOUT: u32 = 5;90pub const REFUNGIBLE_SPONSOR_TRANSFER_TIMEOUT: u32 = 5;9192pub const SPONSOR_APPROVE_TIMEOUT: u32 = 5;939495pub const OFFCHAIN_SCHEMA_LIMIT: u32 = 8192;96pub const VARIABLE_ON_CHAIN_SCHEMA_LIMIT: u32 = 8192;97pub const CONST_ON_CHAIN_SCHEMA_LIMIT: u32 = 32768;9899pub const COLLECTION_FIELD_LIMIT: u32 = CONST_ON_CHAIN_SCHEMA_LIMIT;100101pub const MAX_COLLECTION_NAME_LENGTH: u32 = 64;102pub const MAX_COLLECTION_DESCRIPTION_LENGTH: u32 = 256;103pub const MAX_TOKEN_PREFIX_LENGTH: u32 = 16;104105pub const MAX_PROPERTY_KEY_LENGTH: u32 = 256;106pub const MAX_PROPERTY_VALUE_LENGTH: u32 = 32768;107pub const MAX_PROPERTIES_PER_ITEM: u32 = 64;108109pub const MAX_COLLECTION_PROPERTIES_SIZE: u32 = 40960;110pub const MAX_TOKEN_PROPERTIES_SIZE: u32 = 32768;111112113pub const RMRK_STRING_LIMIT: u32 = 128;114pub const RMRK_COLLECTION_SYMBOL_LIMIT: u32 = 100;115pub const RMRK_RESOURCE_SYMBOL_LIMIT: u32 = 10;116pub const RMRK_KEY_LIMIT: u32 = 32;117pub const RMRK_VALUE_LIMIT: u32 = 256;118119120121pub const MAX_ITEMS_PER_BATCH: u32 = 200;122123pub type CustomDataLimit = ConstU32<CUSTOM_DATA_LIMIT>;124125#[derive(126 Encode,127 Decode,128 PartialEq,129 Eq,130 PartialOrd,131 Ord,132 Clone,133 Copy,134 Debug,135 Default,136 TypeInfo,137 MaxEncodedLen,138)]139#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]140pub struct CollectionId(pub u32);141impl EncodeLike<u32> for CollectionId {}142impl EncodeLike<CollectionId> for u32 {}143144#[derive(145 Encode,146 Decode,147 PartialEq,148 Eq,149 PartialOrd,150 Ord,151 Clone,152 Copy,153 Debug,154 Default,155 TypeInfo,156 MaxEncodedLen,157)]158#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]159pub struct TokenId(pub u32);160impl EncodeLike<u32> for TokenId {}161impl EncodeLike<TokenId> for u32 {}162163impl TokenId {164 pub fn try_next(self) -> Result<TokenId, ArithmeticError> {165 self.0166 .checked_add(1)167 .ok_or(ArithmeticError::Overflow)168 .map(Self)169 }170}171172impl From<TokenId> for U256 {173 fn from(t: TokenId) -> Self {174 t.0.into()175 }176}177178impl TryFrom<U256> for TokenId {179 type Error = &'static str;180181 fn try_from(value: U256) -> Result<Self, Self::Error> {182 Ok(TokenId(value.try_into().map_err(|_| "too large token id")?))183 }184}185186#[derive(Encode, Decode, Clone, PartialEq, TypeInfo)]187#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]188pub struct TokenData<CrossAccountId> {189 pub properties: Vec<Property>,190 pub owner: Option<CrossAccountId>,191}192193pub struct OverflowError;194impl From<OverflowError> for &'static str {195 fn from(_: OverflowError) -> Self {196 "overflow occured"197 }198}199200pub type DecimalPoints = u8;201202#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq, TypeInfo, MaxEncodedLen)]203#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]204pub enum CollectionMode {205 NFT,206 207 Fungible(DecimalPoints),208 ReFungible,209}210211impl CollectionMode {212 pub fn id(&self) -> u8 {213 match self {214 CollectionMode::NFT => 1,215 CollectionMode::Fungible(_) => 2,216 CollectionMode::ReFungible => 3,217 }218 }219}220221pub trait SponsoringResolve<AccountId, Call> {222 fn resolve(who: &AccountId, call: &Call) -> Option<AccountId>;223}224225#[derive(Encode, Decode, Eq, Debug, Clone, Copy, PartialEq, TypeInfo, MaxEncodedLen)]226#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]227pub enum AccessMode {228 Normal,229 AllowList,230}231impl Default for AccessMode {232 fn default() -> Self {233 Self::Normal234 }235}236237#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq, TypeInfo, MaxEncodedLen)]238#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]239pub enum SchemaVersion {240 ImageURL,241 Unique,242}243impl Default for SchemaVersion {244 fn default() -> Self {245 Self::ImageURL246 }247}248249#[derive(Encode, Decode, Default, Debug, Clone, PartialEq, TypeInfo)]250#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]251pub struct Ownership<AccountId> {252 pub owner: AccountId,253 pub fraction: u128,254}255256#[derive(Encode, Decode, Debug, Clone, PartialEq, TypeInfo, MaxEncodedLen)]257#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]258pub enum SponsorshipState<AccountId> {259 260 Disabled,261 Unconfirmed(AccountId),262 263 Confirmed(AccountId),264}265266impl<AccountId> SponsorshipState<AccountId> {267 pub fn sponsor(&self) -> Option<&AccountId> {268 match self {269 Self::Confirmed(sponsor) => Some(sponsor),270 _ => None,271 }272 }273274 pub fn pending_sponsor(&self) -> Option<&AccountId> {275 match self {276 Self::Unconfirmed(sponsor) | Self::Confirmed(sponsor) => Some(sponsor),277 _ => None,278 }279 }280281 pub fn confirmed(&self) -> bool {282 matches!(self, Self::Confirmed(_))283 }284}285286impl<T> Default for SponsorshipState<T> {287 fn default() -> Self {288 Self::Disabled289 }290}291292293#[struct_versioning::versioned(version = 2, upper)]294#[derive(Encode, Decode, Clone, PartialEq, TypeInfo, MaxEncodedLen)]295pub struct Collection<AccountId> {296 pub owner: AccountId,297 pub mode: CollectionMode,298 #[version(..2)]299 pub access: AccessMode,300 pub name: BoundedVec<u16, ConstU32<MAX_COLLECTION_NAME_LENGTH>>,301 pub description: BoundedVec<u16, ConstU32<MAX_COLLECTION_DESCRIPTION_LENGTH>>,302 pub token_prefix: BoundedVec<u8, ConstU32<MAX_TOKEN_PREFIX_LENGTH>>,303304 #[version(..2)]305 pub mint_mode: bool,306307 #[version(..2)]308 pub offchain_schema: BoundedVec<u8, ConstU32<OFFCHAIN_SCHEMA_LIMIT>>,309310 #[version(..2)]311 pub schema_version: SchemaVersion,312 pub sponsorship: SponsorshipState<AccountId>,313314 pub limits: CollectionLimits,315316 #[version(2.., upper(Default::default()))]317 pub permissions: CollectionPermissions,318319 #[version(..2)]320 pub variable_on_chain_schema: BoundedVec<u8, ConstU32<VARIABLE_ON_CHAIN_SCHEMA_LIMIT>>,321322 #[version(..2)]323 pub const_on_chain_schema: BoundedVec<u8, ConstU32<CONST_ON_CHAIN_SCHEMA_LIMIT>>,324325 #[version(..2)]326 pub meta_update_permission: MetaUpdatePermission,327}328329330#[derive(Encode, Decode, Clone, PartialEq, TypeInfo)]331#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]332pub struct RpcCollection<AccountId> {333 pub owner: AccountId,334 pub mode: CollectionMode,335 pub name: Vec<u16>,336 pub description: Vec<u16>,337 pub token_prefix: Vec<u8>,338 pub sponsorship: SponsorshipState<AccountId>,339 pub limits: CollectionLimits,340 pub permissions: CollectionPermissions,341 pub token_property_permissions: Vec<PropertyKeyPermission>,342 pub properties: Vec<Property>,343}344345#[derive(Encode, Decode, Clone, PartialEq, TypeInfo, Derivative, MaxEncodedLen)]346#[derivative(Debug, Default(bound = ""))]347pub struct CreateCollectionData<AccountId> {348 #[derivative(Default(value = "CollectionMode::NFT"))]349 pub mode: CollectionMode,350 pub access: Option<AccessMode>,351 pub name: BoundedVec<u16, ConstU32<MAX_COLLECTION_NAME_LENGTH>>,352 pub description: BoundedVec<u16, ConstU32<MAX_COLLECTION_DESCRIPTION_LENGTH>>,353 pub token_prefix: BoundedVec<u8, ConstU32<MAX_TOKEN_PREFIX_LENGTH>>,354 pub pending_sponsor: Option<AccountId>,355 pub limits: Option<CollectionLimits>,356 pub permissions: Option<CollectionPermissions>,357 pub token_property_permissions: CollectionPropertiesPermissionsVec,358 pub properties: CollectionPropertiesVec,359}360361pub type CollectionPropertiesPermissionsVec =362 BoundedVec<PropertyKeyPermission, ConstU32<MAX_PROPERTIES_PER_ITEM>>;363364pub type CollectionPropertiesVec = BoundedVec<Property, ConstU32<MAX_PROPERTIES_PER_ITEM>>;365366367368#[derive(Encode, Decode, Debug, Default, Clone, PartialEq, TypeInfo, MaxEncodedLen)]369#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]370pub struct CollectionLimits {371 pub account_token_ownership_limit: Option<u32>,372 pub sponsored_data_size: Option<u32>,373374 375 376 377 378 pub sponsored_data_rate_limit: Option<SponsoringRateLimit>,379 pub token_limit: Option<u32>,380381 382 pub sponsor_transfer_timeout: Option<u32>,383 pub sponsor_approve_timeout: Option<u32>,384 pub owner_can_transfer: Option<bool>,385 pub owner_can_destroy: Option<bool>,386 pub transfers_enabled: Option<bool>,387}388389impl CollectionLimits {390 pub fn account_token_ownership_limit(&self) -> u32 {391 self.account_token_ownership_limit392 .unwrap_or(ACCOUNT_TOKEN_OWNERSHIP_LIMIT)393 .min(MAX_TOKEN_OWNERSHIP)394 }395 pub fn sponsored_data_size(&self) -> u32 {396 self.sponsored_data_size397 .unwrap_or(CUSTOM_DATA_LIMIT)398 .min(CUSTOM_DATA_LIMIT)399 }400 pub fn token_limit(&self) -> u32 {401 self.token_limit402 .unwrap_or(COLLECTION_TOKEN_LIMIT)403 .min(COLLECTION_TOKEN_LIMIT)404 }405 pub fn sponsor_transfer_timeout(&self, default: u32) -> u32 {406 self.sponsor_transfer_timeout407 .unwrap_or(default)408 .min(MAX_SPONSOR_TIMEOUT)409 }410 pub fn sponsor_approve_timeout(&self) -> u32 {411 self.sponsor_approve_timeout412 .unwrap_or(SPONSOR_APPROVE_TIMEOUT)413 .min(MAX_SPONSOR_TIMEOUT)414 }415 pub fn owner_can_transfer(&self) -> bool {416 self.owner_can_transfer.unwrap_or(true)417 }418 pub fn owner_can_destroy(&self) -> bool {419 self.owner_can_destroy.unwrap_or(true)420 }421 pub fn transfers_enabled(&self) -> bool {422 self.transfers_enabled.unwrap_or(true)423 }424 pub fn sponsored_data_rate_limit(&self) -> Option<u32> {425 match self426 .sponsored_data_rate_limit427 .unwrap_or(SponsoringRateLimit::SponsoringDisabled)428 {429 SponsoringRateLimit::SponsoringDisabled => None,430 SponsoringRateLimit::Blocks(v) => Some(v.min(MAX_SPONSOR_TIMEOUT)),431 }432 }433}434435436#[derive(Encode, Decode, Debug, Default, Clone, PartialEq, TypeInfo, MaxEncodedLen)]437#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]438pub struct CollectionPermissions {439 pub access: Option<AccessMode>,440 pub mint_mode: Option<bool>,441 pub nesting: Option<NestingRule>,442}443444impl CollectionPermissions {445 pub fn access(&self) -> AccessMode {446 self.access.unwrap_or(AccessMode::Normal)447 }448 pub fn mint_mode(&self) -> bool {449 self.mint_mode.unwrap_or(false)450 }451 pub fn nesting(&self) -> &NestingRule {452 static DEFAULT: NestingRule = NestingRule::Disabled;453 self.nesting.as_ref().unwrap_or(&DEFAULT)454 }455}456457#[derive(Encode, Decode, Clone, PartialEq, TypeInfo, MaxEncodedLen, Derivative)]458#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]459#[derivative(Debug)]460pub enum NestingRule {461 462 Disabled,463 464 Owner,465 466 OwnerRestricted(467 #[cfg_attr(feature = "serde1", serde(with = "bounded::set_serde"))]468 #[derivative(Debug(format_with = "bounded::set_debug"))]469 BoundedBTreeSet<CollectionId, ConstU32<16>>,470 ),471 472 Permissive,473}474475#[derive(Encode, Decode, Debug, Clone, Copy, PartialEq, TypeInfo, MaxEncodedLen)]476#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]477pub enum SponsoringRateLimit {478 SponsoringDisabled,479 Blocks(u32),480}481482#[derive(Encode, Decode, MaxEncodedLen, Default, PartialEq, Clone, Derivative, TypeInfo)]483#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]484#[derivative(Debug)]485pub struct CreateNftData {486 #[cfg_attr(feature = "serde1", serde(with = "bounded::vec_serde"))]487 #[derivative(Debug(format_with = "bounded::vec_debug"))]488 pub properties: CollectionPropertiesVec,489}490491#[derive(Encode, Decode, MaxEncodedLen, Default, Debug, Clone, PartialEq, TypeInfo)]492#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]493pub struct CreateFungibleData {494 pub value: u128,495}496497#[derive(Encode, Decode, MaxEncodedLen, Default, PartialEq, Clone, Derivative, TypeInfo)]498#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]499#[derivative(Debug)]500pub struct CreateReFungibleData {501 #[cfg_attr(feature = "serde1", serde(with = "bounded::vec_serde"))]502 #[derivative(Debug(format_with = "bounded::vec_debug"))]503 pub const_data: BoundedVec<u8, CustomDataLimit>,504 pub pieces: u128,505}506507#[derive(Encode, Decode, Debug, Clone, PartialEq, TypeInfo, MaxEncodedLen)]508#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]509pub enum MetaUpdatePermission {510 ItemOwner,511 Admin,512 None,513}514515#[derive(Encode, Decode, MaxEncodedLen, PartialEq, Clone, Debug, TypeInfo)]516#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]517pub enum CreateItemData {518 NFT(CreateNftData),519 Fungible(CreateFungibleData),520 ReFungible(CreateReFungibleData),521}522523#[derive(Encode, Decode, MaxEncodedLen, PartialEq, Clone, TypeInfo, Derivative)]524#[derivative(Debug)]525pub struct CreateNftExData<CrossAccountId> {526 #[derivative(Debug(format_with = "bounded::vec_debug"))]527 pub properties: CollectionPropertiesVec,528 pub owner: CrossAccountId,529}530531#[derive(Encode, Decode, MaxEncodedLen, PartialEq, Clone, TypeInfo, Derivative)]532#[derivative(Debug(bound = "CrossAccountId: fmt::Debug + Ord"))]533pub struct CreateRefungibleExData<CrossAccountId> {534 #[derivative(Debug(format_with = "bounded::vec_debug"))]535 pub const_data: BoundedVec<u8, CustomDataLimit>,536 #[derivative(Debug(format_with = "bounded::map_debug"))]537 pub users: BoundedBTreeMap<CrossAccountId, u128, ConstU32<MAX_ITEMS_PER_BATCH>>,538}539540#[derive(Encode, Decode, MaxEncodedLen, PartialEq, Clone, TypeInfo, Derivative)]541#[derivative(Debug(bound = "CrossAccountId: fmt::Debug + Ord"))]542pub enum CreateItemExData<CrossAccountId> {543 NFT(544 #[derivative(Debug(format_with = "bounded::vec_debug"))]545 BoundedVec<CreateNftExData<CrossAccountId>, ConstU32<MAX_ITEMS_PER_BATCH>>,546 ),547 Fungible(548 #[derivative(Debug(format_with = "bounded::map_debug"))]549 BoundedBTreeMap<CrossAccountId, u128, ConstU32<MAX_ITEMS_PER_BATCH>>,550 ),551 552 RefungibleMultipleItems(553 #[derivative(Debug(format_with = "bounded::vec_debug"))]554 BoundedVec<CreateRefungibleExData<CrossAccountId>, ConstU32<MAX_ITEMS_PER_BATCH>>,555 ),556 557 RefungibleMultipleOwners(CreateRefungibleExData<CrossAccountId>),558}559560impl CreateItemData {561 pub fn data_size(&self) -> usize {562 match self {563 CreateItemData::ReFungible(data) => data.const_data.len(),564 _ => 0,565 }566 }567}568569impl From<CreateNftData> for CreateItemData {570 fn from(item: CreateNftData) -> Self {571 CreateItemData::NFT(item)572 }573}574575impl From<CreateReFungibleData> for CreateItemData {576 fn from(item: CreateReFungibleData) -> Self {577 CreateItemData::ReFungible(item)578 }579}580581impl From<CreateFungibleData> for CreateItemData {582 fn from(item: CreateFungibleData) -> Self {583 CreateItemData::Fungible(item)584 }585}586587#[derive(Encode, Decode, MaxEncodedLen, PartialEq, Clone, Debug, TypeInfo)]588#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]589590pub struct TokenChild {591 pub token: TokenId,592 pub collection: CollectionId,593}594595#[derive(Encode, Decode, MaxEncodedLen, PartialEq, Clone, Debug, TypeInfo)]596#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]597pub struct CollectionStats {598 pub created: u32,599 pub destroyed: u32,600 pub alive: u32,601}602603#[derive(Encode, Decode, Clone, Debug)]604#[cfg_attr(feature = "std", derive(PartialEq))]605pub struct PhantomType<T>(core::marker::PhantomData<T>);606607impl<T: TypeInfo + 'static> TypeInfo for PhantomType<T> {608 type Identity = PhantomType<T>;609610 fn type_info() -> scale_info::Type {611 use scale_info::{612 Type, Path,613 build::{FieldsBuilder, UnnamedFields},614 type_params,615 };616 Type::builder()617 .path(Path::new("up_data_structs", "PhantomType"))618 .type_params(type_params!(T))619 .composite(<FieldsBuilder<UnnamedFields>>::default().field(|b| b.ty::<[T; 0]>()))620 }621}622impl<T> MaxEncodedLen for PhantomType<T> {623 fn max_encoded_len() -> usize {624 0625 }626}627628pub type PropertyKey = BoundedVec<u8, ConstU32<MAX_PROPERTY_KEY_LENGTH>>;629pub type PropertyValue = BoundedVec<u8, ConstU32<MAX_PROPERTY_VALUE_LENGTH>>;630631#[derive(Encode, Decode, TypeInfo, Debug, MaxEncodedLen, PartialEq, Clone)]632#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]633pub struct PropertyPermission {634 pub mutable: bool,635 pub collection_admin: bool,636 pub token_owner: bool,637}638639impl PropertyPermission {640 pub fn none() -> Self {641 Self {642 mutable: true,643 collection_admin: false,644 token_owner: false,645 }646 }647}648649#[derive(Encode, Decode, Debug, TypeInfo, Clone, PartialEq, MaxEncodedLen)]650#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]651pub struct Property {652 #[cfg_attr(feature = "serde1", serde(with = "bounded::vec_serde"))]653 pub key: PropertyKey,654655 #[cfg_attr(feature = "serde1", serde(with = "bounded::vec_serde"))]656 pub value: PropertyValue,657}658659impl Into<(PropertyKey, PropertyValue)> for Property {660 fn into(self) -> (PropertyKey, PropertyValue) {661 (self.key, self.value)662 }663}664665#[derive(Encode, Decode, TypeInfo, Debug, MaxEncodedLen, PartialEq, Clone)]666#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]667pub struct PropertyKeyPermission {668 #[cfg_attr(feature = "serde1", serde(with = "bounded::vec_serde"))]669 pub key: PropertyKey,670671 pub permission: PropertyPermission,672}673674impl Into<(PropertyKey, PropertyPermission)> for PropertyKeyPermission {675 fn into(self) -> (PropertyKey, PropertyPermission) {676 (self.key, self.permission)677 }678}679680#[derive(Debug)]681pub enum PropertiesError {682 NoSpaceForProperty,683 PropertyLimitReached,684 InvalidCharacterInPropertyKey,685 PropertyKeyIsTooLong,686 EmptyPropertyKey,687}688689#[derive(Clone, Copy)]690pub enum PropertyScope {691 None,692 Rmrk,693}694695impl PropertyScope {696 pub fn apply(self, key: PropertyKey) -> Result<PropertyKey, PropertiesError> {697 let scope_str: &[u8] = match self {698 Self::None => return Ok(key),699 Self::Rmrk => b"rmrk",700 };701702 [scope_str, b":", key.as_slice()]703 .concat()704 .try_into()705 .map_err(|_| PropertiesError::PropertyKeyIsTooLong)706 }707}708709pub trait TrySetProperty: Sized {710 type Value;711712 fn try_scoped_set(713 &mut self,714 scope: PropertyScope,715 key: PropertyKey,716 value: Self::Value,717 ) -> Result<(), PropertiesError>;718719 fn try_scoped_set_from_iter<I, KV>(720 &mut self,721 scope: PropertyScope,722 iter: I,723 ) -> Result<(), PropertiesError>724 where725 I: Iterator<Item = KV>,726 KV: Into<(PropertyKey, Self::Value)>,727 {728 for kv in iter {729 let (key, value) = kv.into();730 self.try_scoped_set(scope, key, value)?;731 }732733 Ok(())734 }735736 fn try_set(&mut self, key: PropertyKey, value: Self::Value) -> Result<(), PropertiesError> {737 self.try_scoped_set(PropertyScope::None, key, value)738 }739740 fn try_set_from_iter<I, KV>(&mut self, iter: I) -> Result<(), PropertiesError>741 where742 I: Iterator<Item = KV>,743 KV: Into<(PropertyKey, Self::Value)>,744 {745 self.try_scoped_set_from_iter(PropertyScope::None, iter)746 }747}748749#[derive(Encode, Decode, TypeInfo, Derivative, Clone, PartialEq, MaxEncodedLen)]750#[derivative(Default(bound = ""))]751pub struct PropertiesMap<Value>(752 BoundedBTreeMap<PropertyKey, Value, ConstU32<MAX_PROPERTIES_PER_ITEM>>,753);754755impl<Value> PropertiesMap<Value> {756 pub fn new() -> Self {757 Self(BoundedBTreeMap::new())758 }759760 pub fn remove(&mut self, key: &PropertyKey) -> Result<Option<Value>, PropertiesError> {761 Self::check_property_key(key)?;762763 Ok(self.0.remove(key))764 }765766 pub fn get(&self, key: &PropertyKey) -> Option<&Value> {767 self.0.get(key)768 }769770 pub fn contains_key(&self, key: &PropertyKey) -> bool {771 self.0.contains_key(key)772 }773774 fn check_property_key(key: &PropertyKey) -> Result<(), PropertiesError> {775 if key.is_empty() {776 return Err(PropertiesError::EmptyPropertyKey);777 }778779 for byte in key.as_slice().iter() {780 let byte = *byte;781782 if !byte.is_ascii_alphanumeric() && byte != b'_' && byte != b'-' {783 return Err(PropertiesError::InvalidCharacterInPropertyKey);784 }785 }786787 Ok(())788 }789}790791impl<Value> IntoIterator for PropertiesMap<Value> {792 type Item = (PropertyKey, Value);793 type IntoIter = <794 BoundedBTreeMap<795 PropertyKey,796 Value,797 ConstU32<MAX_PROPERTIES_PER_ITEM>798 > as IntoIterator799 >::IntoIter;800801 fn into_iter(self) -> Self::IntoIter {802 self.0.into_iter()803 }804}805806impl<Value> TrySetProperty for PropertiesMap<Value> {807 type Value = Value;808809 fn try_scoped_set(810 &mut self,811 scope: PropertyScope,812 key: PropertyKey,813 value: Self::Value,814 ) -> Result<(), PropertiesError> {815 Self::check_property_key(&key)?;816817 let key = scope.apply(key)?;818 self.0819 .try_insert(key, value)820 .map_err(|_| PropertiesError::PropertyLimitReached)?;821822 Ok(())823 }824}825826pub type PropertiesPermissionMap = PropertiesMap<PropertyPermission>;827828#[derive(Encode, Decode, TypeInfo, Clone, PartialEq, MaxEncodedLen)]829pub struct Properties {830 map: PropertiesMap<PropertyValue>,831 consumed_space: u32,832 space_limit: u32,833}834835impl Properties {836 pub fn new(space_limit: u32) -> Self {837 Self {838 map: PropertiesMap::new(),839 consumed_space: 0,840 space_limit,841 }842 }843844 pub fn remove(&mut self, key: &PropertyKey) -> Result<Option<PropertyValue>, PropertiesError> {845 let value = self.map.remove(key)?;846847 if let Some(ref value) = value {848 let value_len = value.len() as u32;849 self.consumed_space -= value_len;850 }851852 Ok(value)853 }854855 pub fn get(&self, key: &PropertyKey) -> Option<&PropertyValue> {856 self.map.get(key)857 }858}859860impl IntoIterator for Properties {861 type Item = (PropertyKey, PropertyValue);862 type IntoIter = <PropertiesMap<PropertyValue> as IntoIterator>::IntoIter;863864 fn into_iter(self) -> Self::IntoIter {865 self.map.into_iter()866 }867}868869impl TrySetProperty for Properties {870 type Value = PropertyValue;871872 fn try_scoped_set(873 &mut self,874 scope: PropertyScope,875 key: PropertyKey,876 value: Self::Value,877 ) -> Result<(), PropertiesError> {878 let value_len = value.len();879880 if self.consumed_space as usize + value_len > self.space_limit as usize881 && !cfg!(feature = "runtime-benchmarks")882 {883 return Err(PropertiesError::NoSpaceForProperty);884 }885886 self.map.try_scoped_set(scope, key, value)?;887888 self.consumed_space += value_len as u32;889890 Ok(())891 }892}893894pub struct CollectionProperties;895896impl Get<Properties> for CollectionProperties {897 fn get() -> Properties {898 Properties::new(MAX_COLLECTION_PROPERTIES_SIZE)899 }900}901902pub struct TokenProperties;903904impl Get<Properties> for TokenProperties {905 fn get() -> Properties {906 Properties::new(MAX_TOKEN_PROPERTIES_SIZE)907 }908}909910911912parameter_types! {913 #[derive(PartialEq, TypeInfo)]914 pub const RmrkStringLimit: u32 = 128;915 #[derive(PartialEq)]916 pub const RmrkCollectionSymbolLimit: u32 = 100;917 #[derive(PartialEq)]918 pub const RmrkResourceSymbolLimit: u32 = 10;919 #[derive(PartialEq)]920 pub const RmrkKeyLimit: u32 = 32;921 #[derive(PartialEq)]922 pub const RmrkValueLimit: u32 = 256;923 #[derive(PartialEq)]924 pub const RmrkMaxCollectionsEquippablePerPart: u32 = 100;925 #[derive(PartialEq)]926 pub const RmrkPartsLimit: u32 = 3;927}928929impl From<RmrkCollectionId> for CollectionId {930 fn from(id: RmrkCollectionId) -> Self {931 Self(id)932 }933}934935impl From<RmrkNftId> for TokenId {936 fn from(id: RmrkNftId) -> Self {937 Self(id)938 }939}940941pub type RmrkCollectionInfo<AccountId> =942 CollectionInfo<RmrkString, RmrkCollectionSymbol, AccountId>;943pub type RmrkInstanceInfo<AccountId> = NftInfo<AccountId, Permill, RmrkString>;944pub type RmrkResourceInfo = ResourceInfo<RmrkString, RmrkBoundedParts>;945pub type RmrkPropertyInfo = PropertyInfo<RmrkKeyString, RmrkValueString>;946pub type RmrkBaseInfo<AccountId> = BaseInfo<AccountId, RmrkString>;947pub type RmrkPartType =948 PartType<RmrkString, BoundedVec<RmrkCollectionId, RmrkMaxCollectionsEquippablePerPart>>;949pub type RmrkThemeProperty = ThemeProperty<RmrkString>;950pub type RmrkTheme = Theme<RmrkString, Vec<RmrkThemeProperty>>;951pub type RmrkResourceTypes = ResourceTypes<RmrkString, RmrkBoundedParts>;952953pub type RmrkBasicResource = BasicResource<RmrkString>;954pub type RmrkComposableResource = ComposableResource<RmrkString, RmrkBoundedParts>;955pub type RmrkSlotResource = SlotResource<RmrkString>;956957pub type RmrkString = BoundedVec<u8, RmrkStringLimit>;958pub type RmrkCollectionSymbol = BoundedVec<u8, RmrkCollectionSymbolLimit>;959pub type RmrkKeyString = BoundedVec<u8, RmrkKeyLimit>;960pub type RmrkValueString = BoundedVec<u8, RmrkValueLimit>;961pub type RmrkBoundedResource = BoundedVec<u8, RmrkResourceSymbolLimit>;962pub type RmrkBoundedParts = BoundedVec<RmrkPartId, RmrkPartsLimit>; 963964pub type RmrkRpcString = Vec<u8>;965pub type RmrkThemeName = RmrkRpcString;966pub type RmrkPropertyKey = RmrkRpcString;