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::ConstU16,26};27use sp_std::collections::{btree_map::BTreeMap, btree_set::BTreeSet};2829#[cfg(feature = "serde")]30use serde::{Serialize, Deserialize};3132use sp_core::U256;33use sp_runtime::{ArithmeticError, sp_std::prelude::Vec};34use codec::{Decode, Encode, EncodeLike, MaxEncodedLen};35use frame_support::{BoundedVec, traits::ConstU32};36use derivative::Derivative;37use scale_info::TypeInfo;3839pub mod mapping;40mod migration;4142pub const MAX_DECIMAL_POINTS: DecimalPoints = 30;43pub const MAX_REFUNGIBLE_PIECES: u128 = 1_000_000_000_000_000_000_000;44pub const MAX_SPONSOR_TIMEOUT: u32 = 10_368_000;4546pub const MAX_TOKEN_OWNERSHIP: u32 = if cfg!(not(feature = "limit-testing")) {47 100_00048} else {49 1050};51pub const COLLECTION_NUMBER_LIMIT: u32 = if cfg!(not(feature = "limit-testing")) {52 100_00053} else {54 1055};56pub const CUSTOM_DATA_LIMIT: u32 = if cfg!(not(feature = "limit-testing")) {57 204858} else {59 1060};61pub const COLLECTION_ADMINS_LIMIT: u32 = 5;62pub const COLLECTION_TOKEN_LIMIT: u32 = u32::MAX;63pub const ACCOUNT_TOKEN_OWNERSHIP_LIMIT: u32 = if cfg!(not(feature = "limit-testing")) {64 1_000_00065} else {66 1067};686970pub const NFT_SPONSOR_TRANSFER_TIMEOUT: u32 = 5;71pub const FUNGIBLE_SPONSOR_TRANSFER_TIMEOUT: u32 = 5;72pub const REFUNGIBLE_SPONSOR_TRANSFER_TIMEOUT: u32 = 5;7374pub const SPONSOR_APPROVE_TIMEOUT: u32 = 5;757677pub const OFFCHAIN_SCHEMA_LIMIT: u32 = 8192;78pub const VARIABLE_ON_CHAIN_SCHEMA_LIMIT: u32 = 8192;79pub const CONST_ON_CHAIN_SCHEMA_LIMIT: u32 = 32768;8081pub const MAX_COLLECTION_NAME_LENGTH: u32 = 64;82pub const MAX_COLLECTION_DESCRIPTION_LENGTH: u32 = 256;83pub const MAX_TOKEN_PREFIX_LENGTH: u32 = 16;84858687pub const MAX_ITEMS_PER_BATCH: u32 = 200;8889pub type CustomDataLimit = ConstU32<CUSTOM_DATA_LIMIT>;9091#[derive(92 Encode,93 Decode,94 PartialEq,95 Eq,96 PartialOrd,97 Ord,98 Clone,99 Copy,100 Debug,101 Default,102 TypeInfo,103 MaxEncodedLen,104)]105#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]106pub struct CollectionId(pub u32);107impl EncodeLike<u32> for CollectionId {}108impl EncodeLike<CollectionId> for u32 {}109110#[derive(111 Encode,112 Decode,113 PartialEq,114 Eq,115 PartialOrd,116 Ord,117 Clone,118 Copy,119 Debug,120 Default,121 TypeInfo,122 MaxEncodedLen,123)]124#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]125pub struct TokenId(pub u32);126impl EncodeLike<u32> for TokenId {}127impl EncodeLike<TokenId> for u32 {}128129impl TokenId {130 pub fn try_next(self) -> Result<TokenId, ArithmeticError> {131 self.0132 .checked_add(1)133 .ok_or(ArithmeticError::Overflow)134 .map(Self)135 }136}137138impl From<TokenId> for U256 {139 fn from(t: TokenId) -> Self {140 t.0.into()141 }142}143144impl TryFrom<U256> for TokenId {145 type Error = &'static str;146147 fn try_from(value: U256) -> Result<Self, Self::Error> {148 Ok(TokenId(value.try_into().map_err(|_| "too large token id")?))149 }150}151152pub struct OverflowError;153impl From<OverflowError> for &'static str {154 fn from(_: OverflowError) -> Self {155 "overflow occured"156 }157}158159pub type DecimalPoints = u8;160161#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq, TypeInfo, MaxEncodedLen)]162#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]163pub enum CollectionMode {164 NFT,165 166 Fungible(DecimalPoints),167 ReFungible,168}169170impl CollectionMode {171 pub fn id(&self) -> u8 {172 match self {173 CollectionMode::NFT => 1,174 CollectionMode::Fungible(_) => 2,175 CollectionMode::ReFungible => 3,176 }177 }178}179180pub trait SponsoringResolve<AccountId, Call> {181 fn resolve(who: &AccountId, call: &Call) -> Option<AccountId>;182}183184#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq, TypeInfo, MaxEncodedLen)]185#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]186pub enum AccessMode {187 Normal,188 AllowList,189}190impl Default for AccessMode {191 fn default() -> Self {192 Self::Normal193 }194}195196#[derive(Encode, Decode, Eq, Debug, Clone, PartialEq, TypeInfo, MaxEncodedLen)]197#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]198pub enum SchemaVersion {199 ImageURL,200 Unique,201}202impl Default for SchemaVersion {203 fn default() -> Self {204 Self::ImageURL205 }206}207208#[derive(Encode, Decode, Default, Debug, Clone, PartialEq, TypeInfo)]209#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]210pub struct Ownership<AccountId> {211 pub owner: AccountId,212 pub fraction: u128,213}214215#[derive(Encode, Decode, Debug, Clone, PartialEq, TypeInfo, MaxEncodedLen)]216#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]217pub enum SponsorshipState<AccountId> {218 219 Disabled,220 Unconfirmed(AccountId),221 222 Confirmed(AccountId),223}224225impl<AccountId> SponsorshipState<AccountId> {226 pub fn sponsor(&self) -> Option<&AccountId> {227 match self {228 Self::Confirmed(sponsor) => Some(sponsor),229 _ => None,230 }231 }232233 pub fn pending_sponsor(&self) -> Option<&AccountId> {234 match self {235 Self::Unconfirmed(sponsor) | Self::Confirmed(sponsor) => Some(sponsor),236 _ => None,237 }238 }239240 pub fn confirmed(&self) -> bool {241 matches!(self, Self::Confirmed(_))242 }243}244245impl<T> Default for SponsorshipState<T> {246 fn default() -> Self {247 Self::Disabled248 }249}250251#[derive(Encode, Decode, Clone, PartialEq, TypeInfo, MaxEncodedLen)]252#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]253pub struct Collection<AccountId> {254 pub owner: AccountId,255 pub mode: CollectionMode,256 pub access: AccessMode,257 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]258 pub name: BoundedVec<u16, ConstU32<MAX_COLLECTION_NAME_LENGTH>>,259 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]260 pub description: BoundedVec<u16, ConstU32<MAX_COLLECTION_DESCRIPTION_LENGTH>>,261 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]262 pub token_prefix: BoundedVec<u8, ConstU32<MAX_TOKEN_PREFIX_LENGTH>>,263 pub mint_mode: bool,264 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]265 pub offchain_schema: BoundedVec<u8, ConstU32<OFFCHAIN_SCHEMA_LIMIT>>,266 pub schema_version: SchemaVersion,267 pub sponsorship: SponsorshipState<AccountId>,268 pub limits: CollectionLimits, 269 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]270 pub variable_on_chain_schema: BoundedVec<u8, ConstU32<VARIABLE_ON_CHAIN_SCHEMA_LIMIT>>,271 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]272 pub const_on_chain_schema: BoundedVec<u8, ConstU32<CONST_ON_CHAIN_SCHEMA_LIMIT>>,273 pub meta_update_permission: MetaUpdatePermission,274}275276#[derive(Encode, Decode, Clone, PartialEq, TypeInfo, Debug, Derivative, MaxEncodedLen)]277#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]278#[derivative(Default(bound = ""))]279pub struct CreateCollectionData<AccountId> {280 #[derivative(Default(value = "CollectionMode::NFT"))]281 pub mode: CollectionMode,282 pub access: Option<AccessMode>,283 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]284 pub name: BoundedVec<u16, ConstU32<MAX_COLLECTION_NAME_LENGTH>>,285 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]286 pub description: BoundedVec<u16, ConstU32<MAX_COLLECTION_DESCRIPTION_LENGTH>>,287 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]288 pub token_prefix: BoundedVec<u8, ConstU32<MAX_TOKEN_PREFIX_LENGTH>>,289 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]290 pub offchain_schema: BoundedVec<u8, ConstU32<OFFCHAIN_SCHEMA_LIMIT>>,291 pub schema_version: Option<SchemaVersion>,292 pub pending_sponsor: Option<AccountId>,293 pub limits: Option<CollectionLimits>,294 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]295 pub variable_on_chain_schema: BoundedVec<u8, ConstU32<VARIABLE_ON_CHAIN_SCHEMA_LIMIT>>,296 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]297 pub const_on_chain_schema: BoundedVec<u8, ConstU32<CONST_ON_CHAIN_SCHEMA_LIMIT>>,298 pub meta_update_permission: Option<MetaUpdatePermission>,299}300301#[derive(Encode, Decode, Debug, Clone, PartialEq, TypeInfo)]302#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]303pub struct NftItemType<AccountId> {304 pub owner: AccountId,305 pub const_data: Vec<u8>,306 pub variable_data: Vec<u8>,307}308309#[derive(Encode, Decode, Default, Debug, Clone, PartialEq, TypeInfo)]310#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]311pub struct FungibleItemType {312 pub value: u128,313}314315#[derive(Encode, Decode, Debug, Clone, PartialEq, TypeInfo)]316#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]317pub struct ReFungibleItemType<AccountId> {318 pub owner: Vec<Ownership<AccountId>>,319 pub const_data: Vec<u8>,320 pub variable_data: Vec<u8>,321}322323324#[derive(Encode, Decode, Debug, Default, Clone, PartialEq, TypeInfo, MaxEncodedLen)]325#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]326pub struct CollectionLimits {327 pub account_token_ownership_limit: Option<u32>,328 pub sponsored_data_size: Option<u32>,329 330 331 332 pub sponsored_data_rate_limit: Option<SponsoringRateLimit>,333 pub token_limit: Option<u32>,334335 336 pub sponsor_transfer_timeout: Option<u32>,337 pub sponsor_approve_timeout: Option<u32>,338 pub owner_can_transfer: Option<bool>,339 pub owner_can_destroy: Option<bool>,340 pub transfers_enabled: Option<bool>,341}342343impl CollectionLimits {344 pub fn account_token_ownership_limit(&self) -> u32 {345 self.account_token_ownership_limit346 .unwrap_or(ACCOUNT_TOKEN_OWNERSHIP_LIMIT)347 .min(MAX_TOKEN_OWNERSHIP)348 }349 pub fn sponsored_data_size(&self) -> u32 {350 self.sponsored_data_size351 .unwrap_or(CUSTOM_DATA_LIMIT)352 .min(CUSTOM_DATA_LIMIT)353 }354 pub fn token_limit(&self) -> u32 {355 self.token_limit356 .unwrap_or(COLLECTION_TOKEN_LIMIT)357 .min(COLLECTION_TOKEN_LIMIT)358 }359 pub fn sponsor_transfer_timeout(&self, default: u32) -> u32 {360 self.sponsor_transfer_timeout361 .unwrap_or(default)362 .min(MAX_SPONSOR_TIMEOUT)363 }364 pub fn sponsor_approve_timeout(&self) -> u32 {365 self.sponsor_approve_timeout366 .unwrap_or(SPONSOR_APPROVE_TIMEOUT)367 .min(MAX_SPONSOR_TIMEOUT)368 }369 pub fn owner_can_transfer(&self) -> bool {370 self.owner_can_transfer.unwrap_or(true)371 }372 pub fn owner_can_destroy(&self) -> bool {373 self.owner_can_destroy.unwrap_or(true)374 }375 pub fn transfers_enabled(&self) -> bool {376 self.transfers_enabled.unwrap_or(true)377 }378 pub fn sponsored_data_rate_limit(&self) -> Option<u32> {379 match self380 .sponsored_data_rate_limit381 .unwrap_or(SponsoringRateLimit::SponsoringDisabled)382 {383 SponsoringRateLimit::SponsoringDisabled => None,384 SponsoringRateLimit::Blocks(v) => Some(v.min(MAX_SPONSOR_TIMEOUT)),385 }386 }387}388389#[derive(Encode, Decode, Debug, Clone, Copy, PartialEq, TypeInfo, MaxEncodedLen)]390#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]391pub enum SponsoringRateLimit {392 SponsoringDisabled,393 Blocks(u32),394}395396397#[cfg(feature = "serde1")]398mod bounded_serde {399 use core::convert::TryFrom;400 use frame_support::{BoundedVec, traits::Get};401 use serde::{402 ser::{self, Serialize},403 de::{self, Deserialize, Error},404 };405 use sp_std::vec::Vec;406407 pub fn serialize<D, V, S>(value: &BoundedVec<V, S>, serializer: D) -> Result<D::Ok, D::Error>408 where409 D: ser::Serializer,410 V: Serialize,411 {412 (value as &Vec<_>).serialize(serializer)413 }414415 pub fn deserialize<'de, D, V, S>(deserializer: D) -> Result<BoundedVec<V, S>, D::Error>416 where417 D: de::Deserializer<'de>,418 V: de::Deserialize<'de>,419 S: Get<u32>,420 {421 422 let vec = <Vec<V>>::deserialize(deserializer)?;423 let len = vec.len();424 TryFrom::try_from(vec).map_err(|_| D::Error::invalid_length(len, &"lesser size"))425 }426}427428fn bounded_debug<V, S>(v: &BoundedVec<V, S>, f: &mut fmt::Formatter) -> Result<(), fmt::Error>429where430 V: fmt::Debug,431{432 use core::fmt::Debug;433 (&v as &Vec<V>).fmt(f)434}435436#[cfg(feature = "serde1")]437#[allow(dead_code)]438mod bounded_map_serde {439 use core::convert::TryFrom;440 use sp_std::collections::btree_map::BTreeMap;441 use frame_support::{traits::Get, storage::bounded_btree_map::BoundedBTreeMap};442 use serde::{443 ser::{self, Serialize},444 de::{self, Deserialize, Error},445 };446 pub fn serialize<D, K, V, S>(447 value: &BoundedBTreeMap<K, V, S>,448 serializer: D,449 ) -> Result<D::Ok, D::Error>450 where451 D: ser::Serializer,452 K: Serialize + Ord,453 V: Serialize,454 {455 (value as &BTreeMap<_, _>).serialize(serializer)456 }457458 pub fn deserialize<'de, D, K, V, S>(459 deserializer: D,460 ) -> Result<BoundedBTreeMap<K, V, S>, D::Error>461 where462 D: de::Deserializer<'de>,463 K: de::Deserialize<'de> + Ord,464 V: de::Deserialize<'de>,465 S: Get<u32>,466 {467 let map = <BTreeMap<K, V>>::deserialize(deserializer)?;468 let len = map.len();469 TryFrom::try_from(map).map_err(|_| D::Error::invalid_length(len, &"lesser size"))470 }471}472473fn bounded_map_debug<K, V, S>(474 v: &BoundedBTreeMap<K, V, S>,475 f: &mut fmt::Formatter,476) -> Result<(), fmt::Error>477where478 K: fmt::Debug + Ord,479 V: fmt::Debug,480{481 use core::fmt::Debug;482 (&v as &BTreeMap<K, V>).fmt(f)483}484485#[derive(Encode, Decode, MaxEncodedLen, Default, PartialEq, Clone, Derivative, TypeInfo)]486#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]487#[derivative(Debug)]488pub struct CreateNftData {489 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]490 #[derivative(Debug(format_with = "bounded_debug"))]491 pub const_data: BoundedVec<u8, CustomDataLimit>,492 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]493 #[derivative(Debug(format_with = "bounded_debug"))]494 pub variable_data: BoundedVec<u8, CustomDataLimit>,495}496497#[derive(Encode, Decode, MaxEncodedLen, Default, Debug, Clone, PartialEq, TypeInfo)]498#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]499pub struct CreateFungibleData {500 pub value: u128,501}502503#[derive(Encode, Decode, MaxEncodedLen, Default, PartialEq, Clone, Derivative, TypeInfo)]504#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]505#[derivative(Debug)]506pub struct CreateReFungibleData {507 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]508 #[derivative(Debug(format_with = "bounded_debug"))]509 pub const_data: BoundedVec<u8, CustomDataLimit>,510 #[cfg_attr(feature = "serde1", serde(with = "bounded_serde"))]511 #[derivative(Debug(format_with = "bounded_debug"))]512 pub variable_data: BoundedVec<u8, CustomDataLimit>,513 pub pieces: u128,514}515516#[derive(Encode, Decode, Debug, Clone, PartialEq, TypeInfo, MaxEncodedLen)]517#[cfg_attr(feature = "std", derive(Serialize, Deserialize))]518pub enum MetaUpdatePermission {519 ItemOwner,520 Admin,521 None,522}523524impl Default for MetaUpdatePermission {525 fn default() -> Self {526 Self::ItemOwner527 }528}529530#[derive(Encode, Decode, MaxEncodedLen, PartialEq, Clone, Debug, TypeInfo)]531#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]532pub enum CreateItemData {533 NFT(CreateNftData),534 Fungible(CreateFungibleData),535 ReFungible(CreateReFungibleData),536}537538#[derive(Encode, Decode, MaxEncodedLen, PartialEq, Clone, TypeInfo, Derivative)]539#[derivative(Debug)]540pub struct CreateNftExData<CrossAccountId> {541 #[derivative(Debug(format_with = "bounded_debug"))]542 pub const_data: BoundedVec<u8, CustomDataLimit>,543 #[derivative(Debug(format_with = "bounded_debug"))]544 pub variable_data: BoundedVec<u8, CustomDataLimit>,545 pub owner: CrossAccountId,546}547548#[derive(Encode, Decode, MaxEncodedLen, PartialEq, Clone, TypeInfo, Derivative)]549#[derivative(Debug(bound = "CrossAccountId: fmt::Debug + Ord"))]550pub struct CreateRefungibleExData<CrossAccountId> {551 #[derivative(Debug(format_with = "bounded_debug"))]552 pub const_data: BoundedVec<u8, CustomDataLimit>,553 #[derivative(Debug(format_with = "bounded_debug"))]554 pub variable_data: BoundedVec<u8, CustomDataLimit>,555 #[derivative(Debug(format_with = "bounded_map_debug"))]556 pub users: BoundedBTreeMap<CrossAccountId, u128, ConstU32<MAX_ITEMS_PER_BATCH>>,557}558559#[derive(Encode, Decode, MaxEncodedLen, PartialEq, Clone, TypeInfo, Derivative)]560#[derivative(Debug(bound = "CrossAccountId: fmt::Debug + Ord"))]561pub enum CreateItemExData<CrossAccountId> {562 NFT(563 #[derivative(Debug(format_with = "bounded_debug"))]564 BoundedVec<CreateNftExData<CrossAccountId>, ConstU32<MAX_ITEMS_PER_BATCH>>,565 ),566 Fungible(567 #[derivative(Debug(format_with = "bounded_map_debug"))]568 BoundedBTreeMap<CrossAccountId, u128, ConstU32<MAX_ITEMS_PER_BATCH>>,569 ),570 571 RefungibleMultipleItems(572 #[derivative(Debug(format_with = "bounded_debug"))]573 BoundedVec<CreateRefungibleExData<CrossAccountId>, ConstU32<MAX_ITEMS_PER_BATCH>>,574 ),575 576 RefungibleMultipleOwners(CreateRefungibleExData<CrossAccountId>),577}578579impl CreateItemData {580 pub fn data_size(&self) -> usize {581 match self {582 CreateItemData::NFT(data) => data.variable_data.len() + data.const_data.len(),583 CreateItemData::ReFungible(data) => data.variable_data.len() + data.const_data.len(),584 _ => 0,585 }586 }587}588589impl From<CreateNftData> for CreateItemData {590 fn from(item: CreateNftData) -> Self {591 CreateItemData::NFT(item)592 }593}594595impl From<CreateReFungibleData> for CreateItemData {596 fn from(item: CreateReFungibleData) -> Self {597 CreateItemData::ReFungible(item)598 }599}600601impl From<CreateFungibleData> for CreateItemData {602 fn from(item: CreateFungibleData) -> Self {603 CreateItemData::Fungible(item)604 }605}606607#[derive(Encode, Decode, MaxEncodedLen, PartialEq, Clone, Debug, TypeInfo)]608#[cfg_attr(feature = "serde1", derive(Serialize, Deserialize))]609pub struct CollectionStats {610 pub created: u32,611 pub destroyed: u32,612 pub alive: u32,613}