git.delta.rocks / unique-network / refs/commits / 80bfde33245f

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#![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> {7778		/// Creates a new Base.79		/// Modeled after [base interaction](https://github.com/rmrk-team/rmrk-spec/blob/master/standards/rmrk2.0.0/interactions/base.md)80		///81		/// Parameters:82		/// - origin: Caller, will be assigned as the issuer of the Base83		/// - base_type: media type, e.g. "svg"84		/// - symbol: arbitrary client-chosen symbol85		/// - parts: array of Fixed and Slot parts composing the base, confined in length by86		///   RmrkPartsLimit87		#[pallet::weight(10_000 + T::DbWeight::get().reads_writes(1,1))]88		#[transactional]89		pub fn create_base(90			origin: OriginFor<T>,91			base_type: RmrkString,92			symbol: RmrkString,93			parts: BoundedVec<RmrkPartType, RmrkPartsLimit>,94		) -> DispatchResult {95			let sender = ensure_signed(origin)?;96			let cross_sender = T::CrossAccountId::from_sub(sender.clone());9798			let data = CreateCollectionData {99				limits: None,100				token_prefix: symbol101					.into_inner()102					.try_into()103					.map_err(|_| <CommonError<T>>::CollectionTokenPrefixLimitExceeded)?,104				..Default::default()105			};106107			let collection_id_res =108				<PalletNft<T>>::init_collection(cross_sender.clone(), data, true);109110			if let Err(DispatchError::Arithmetic(_)) = &collection_id_res {111				return Err(<Error<T>>::NoAvailableBaseId.into());112			}113114			let collection_id = collection_id_res?;115116			<PalletCommon<T>>::set_scoped_collection_properties(117				collection_id,118				PropertyScope::Rmrk,119				[120					<PalletCore<T>>::rmrk_property(CollectionType, &misc::CollectionType::Base)?,121					<PalletCore<T>>::rmrk_property(BaseType, &base_type)?,122				]123				.into_iter(),124			)?;125126			let collection = <PalletCore<T>>::get_nft_collection(collection_id)?;127128			for part in parts {129				let part_id = part.id();130				let part_token_id = Self::create_part(&cross_sender, &collection, part)?;131132				<InernalPartId<T>>::insert(collection_id, part_id, part_token_id);133134				<PalletNft<T>>::set_scoped_token_property(135					collection_id,136					part_token_id,137					PropertyScope::Rmrk,138					<PalletCore<T>>::rmrk_property(ExternalPartId, &part_id)?,139				)?;140			}141142			Self::deposit_event(Event::BaseCreated {143				issuer: sender,144				base_id: collection_id.0,145			});146147			Ok(())148		}149150		/// Adds a Theme to a Base.151		/// Modeled after [themeadd interaction](https://github.com/rmrk-team/rmrk-spec/blob/master/standards/rmrk2.0.0/interactions/themeadd.md)152		/// Themes are stored in the Themes storage153		/// A Theme named "default" is required prior to adding other Themes.154		///155		/// Parameters:156		/// - origin: The caller of the function, must be issuer of the base157		/// - base_id: The Base containing the Theme to be updated158		/// - theme: The Theme to add to the Base.  A Theme has a name and properties, which are an159		///   array of [key, value, inherit].160		///   - key: arbitrary BoundedString, defined by client161		///   - value: arbitrary BoundedString, defined by client162		///   - inherit: optional bool163		#[pallet::weight(10_000 + T::DbWeight::get().reads_writes(1,1))]164		#[transactional]165		pub fn theme_add(166			origin: OriginFor<T>,167			base_id: RmrkBaseId,168			theme: RmrkTheme,169		) -> DispatchResult {170			let sender = ensure_signed(origin)?;171172			let sender = T::CrossAccountId::from_sub(sender);173			let owner = &sender;174175			let collection_id: CollectionId = base_id.into();176177			let collection = <PalletCore<T>>::get_typed_nft_collection(178				collection_id,179				misc::CollectionType::Base,180			)181			.map_err(|_| <Error<T>>::BaseDoesntExist)?;182			collection.check_is_external()?;183184			if theme.name.as_slice() == b"default" {185				<BaseHasDefaultTheme<T>>::insert(collection_id, true);186			} else if !Self::base_has_default_theme(collection_id) {187				return Err(<Error<T>>::NeedsDefaultThemeFirst.into());188			}189190			let token_id = <PalletCore<T>>::create_nft(191				&sender,192				owner,193				&collection,194				[195					<PalletCore<T>>::rmrk_property(TokenType, &NftType::Theme)?,196					<PalletCore<T>>::rmrk_property(ThemeName, &theme.name)?,197					<PalletCore<T>>::rmrk_property(ThemeInherit, &theme.inherit)?,198				]199				.into_iter(),200			)201			.map_err(|_| <Error<T>>::PermissionError)?;202203			for property in theme.properties {204				<PalletNft<T>>::set_scoped_token_property(205					collection_id,206					token_id,207					PropertyScope::Rmrk,208					<PalletCore<T>>::rmrk_property(209						UserProperty(property.key.as_slice()),210						&property.value,211					)?,212				)?;213			}214215			Ok(())216		}217	}218}219220impl<T: Config> Pallet<T> {221	fn create_part(222		sender: &T::CrossAccountId,223		collection: &NonfungibleHandle<T>,224		part: RmrkPartType,225	) -> Result<TokenId, DispatchError> {226		let owner = sender;227228		let src = part.src();229		let z_index = part.z_index();230231		let nft_type = match part {232			RmrkPartType::FixedPart(_) => NftType::FixedPart,233			RmrkPartType::SlotPart(_) => NftType::SlotPart,234		};235236		let token_id = <PalletCore<T>>::create_nft(237			sender,238			owner,239			collection,240			[241				<PalletCore<T>>::rmrk_property(TokenType, &nft_type)?,242				<PalletCore<T>>::rmrk_property(Src, &src)?,243				<PalletCore<T>>::rmrk_property(ZIndex, &z_index)?,244			]245			.into_iter(),246		)247		.map_err(|err| match err {248			DispatchError::Arithmetic(_) => <Error<T>>::NoAvailablePartId.into(),249			err => err,250		})?;251252		if let RmrkPartType::SlotPart(part) = part {253			<PalletNft<T>>::set_scoped_token_property(254				collection.id,255				token_id,256				PropertyScope::Rmrk,257				<PalletCore<T>>::rmrk_property(EquippableList, &part.equippable)?,258			)?;259		}260261		Ok(token_id)262	}263}