1234567891011121314151617#![cfg_attr(not(feature = "std"), no_std)]1819use frame_support::{pallet_prelude::*, transactional, BoundedVec, dispatch::DispatchResult};20use frame_system::{pallet_prelude::*, ensure_signed};21use sp_runtime::DispatchError;22use up_data_structs::*;23use pallet_common::{Pallet as PalletCommon, Error as CommonError};24use pallet_rmrk_core::{25 Pallet as PalletCore,26 misc::{self, *},27 property::RmrkProperty::*,28};29use pallet_nonfungible::{Pallet as PalletNft, NonfungibleHandle};30use pallet_evm::account::CrossAccountId;3132pub use pallet::*;3334#[frame_support::pallet]35pub mod pallet {36 use super::*;3738 #[pallet::config]39 pub trait Config: frame_system::Config + pallet_rmrk_core::Config {40 type Event: From<Event<Self>> + IsType<<Self as frame_system::Config>::Event>;41 }4243 #[pallet::storage]44 #[pallet::getter(fn internal_part_id)]45 pub type InernalPartId<T: Config> =46 StorageDoubleMap<_, Twox64Concat, CollectionId, Twox64Concat, RmrkPartId, TokenId>;4748 #[pallet::storage]49 #[pallet::getter(fn base_has_default_theme)]50 pub type BaseHasDefaultTheme<T: Config> =51 StorageMap<_, Twox64Concat, CollectionId, bool, ValueQuery>;5253 #[pallet::pallet]54 #[pallet::generate_store(pub(super) trait Store)]55 pub struct Pallet<T>(_);5657 #[pallet::event]58 #[pallet::generate_deposit(pub(super) fn deposit_event)]59 pub enum Event<T: Config> {60 BaseCreated {61 issuer: T::AccountId,62 base_id: RmrkBaseId,63 },64 }6566 #[pallet::error]67 pub enum Error<T> {68 PermissionError,69 NoAvailableBaseId,70 NoAvailablePartId,71 BaseDoesntExist,72 NeedsDefaultThemeFirst,73 }7475 #[pallet::call]76 impl<T: Config> Pallet<T> {77 #[pallet::weight(10_000 + T::DbWeight::get().reads_writes(1,1))]78 #[transactional]79 pub fn create_base(80 origin: OriginFor<T>,81 base_type: RmrkString,82 symbol: RmrkString,83 parts: BoundedVec<RmrkPartType, RmrkPartsLimit>,84 ) -> DispatchResult {85 let sender = ensure_signed(origin)?;86 let cross_sender = T::CrossAccountId::from_sub(sender.clone());8788 let data = CreateCollectionData {89 limits: None,90 token_prefix: symbol91 .into_inner()92 .try_into()93 .map_err(|_| <CommonError<T>>::CollectionTokenPrefixLimitExceeded)?,94 ..Default::default()95 };9697 let collection_id_res =98 <PalletNft<T>>::init_collection(cross_sender.clone(), data, true);99100 if let Err(DispatchError::Arithmetic(_)) = &collection_id_res {101 return Err(<Error<T>>::NoAvailableBaseId.into());102 }103104 let collection_id = collection_id_res?;105106 <PalletCommon<T>>::set_scoped_collection_properties(107 collection_id,108 PropertyScope::Rmrk,109 [110 <PalletCore<T>>::rmrk_property(CollectionType, &misc::CollectionType::Base)?,111 <PalletCore<T>>::rmrk_property(BaseType, &base_type)?,112 ]113 .into_iter(),114 )?;115116 let collection = <PalletCore<T>>::get_nft_collection(collection_id)?;117118 for part in parts {119 let part_id = part.id();120 let part_token_id = Self::create_part(&cross_sender, &collection, part)?;121122 <InernalPartId<T>>::insert(collection_id, part_id, part_token_id);123124 <PalletNft<T>>::set_scoped_token_property(125 collection_id,126 part_token_id,127 PropertyScope::Rmrk,128 <PalletCore<T>>::rmrk_property(ExternalPartId, &part_id)?,129 )?;130 }131132 Self::deposit_event(Event::BaseCreated {133 issuer: sender,134 base_id: collection_id.0,135 });136137 Ok(())138 }139140 #[pallet::weight(10_000 + T::DbWeight::get().reads_writes(1,1))]141 #[transactional]142 pub fn theme_add(143 origin: OriginFor<T>,144 base_id: RmrkBaseId,145 theme: RmrkTheme,146 ) -> DispatchResult {147 let sender = ensure_signed(origin)?;148149 let sender = T::CrossAccountId::from_sub(sender);150 let owner = &sender;151152 let collection_id: CollectionId = base_id.into();153154 let collection = <PalletCore<T>>::get_typed_nft_collection(155 collection_id,156 misc::CollectionType::Base,157 )158 .map_err(|_| <Error<T>>::BaseDoesntExist)?;159 collection.check_is_external()?;160161 if theme.name.as_slice() == b"default" {162 <BaseHasDefaultTheme<T>>::insert(collection_id, true);163 } else if !Self::base_has_default_theme(collection_id) {164 return Err(<Error<T>>::NeedsDefaultThemeFirst.into());165 }166167 let token_id = <PalletCore<T>>::create_nft(168 &sender,169 owner,170 &collection,171 [172 <PalletCore<T>>::rmrk_property(TokenType, &NftType::Theme)?,173 <PalletCore<T>>::rmrk_property(ThemeName, &theme.name)?,174 <PalletCore<T>>::rmrk_property(ThemeInherit, &theme.inherit)?,175 ]176 .into_iter(),177 )178 .map_err(|_| <Error<T>>::PermissionError)?;179180 for property in theme.properties {181 <PalletNft<T>>::set_scoped_token_property(182 collection_id,183 token_id,184 PropertyScope::Rmrk,185 <PalletCore<T>>::rmrk_property(186 UserProperty(property.key.as_slice()),187 &property.value,188 )?,189 )?;190 }191192 Ok(())193 }194 }195}196197impl<T: Config> Pallet<T> {198 fn create_part(199 sender: &T::CrossAccountId,200 collection: &NonfungibleHandle<T>,201 part: RmrkPartType,202 ) -> Result<TokenId, DispatchError> {203 let owner = sender;204205 let src = part.src();206 let z_index = part.z_index();207208 let nft_type = match part {209 RmrkPartType::FixedPart(_) => NftType::FixedPart,210 RmrkPartType::SlotPart(_) => NftType::SlotPart,211 };212213 let token_id = <PalletCore<T>>::create_nft(214 sender,215 owner,216 collection,217 [218 <PalletCore<T>>::rmrk_property(TokenType, &nft_type)?,219 <PalletCore<T>>::rmrk_property(Src, &src)?,220 <PalletCore<T>>::rmrk_property(ZIndex, &z_index)?,221 ]222 .into_iter(),223 )224 .map_err(|err| match err {225 DispatchError::Arithmetic(_) => <Error<T>>::NoAvailablePartId.into(),226 err => err,227 })?;228229 if let RmrkPartType::SlotPart(part) = part {230 <PalletNft<T>>::set_scoped_token_property(231 collection.id,232 token_id,233 PropertyScope::Rmrk,234 <PalletCore<T>>::rmrk_property(EquippableList, &part.equippable)?,235 )?;236 }237238 Ok(token_id)239 }240}