1234567891011121314151617use super::*;18use up_data_structs::PropertyScope;19use core::convert::AsRef;2021pub const RESOURCE_ID_PREFIX: &str = "rsid-";22pub const USER_PROPERTY_PREFIX: &str = "userprop-";2324pub enum RmrkProperty<'r> {25 Metadata,26 CollectionType,27 RmrkInternalCollectionId,28 TokenType,29 Transferable,30 RoyaltyInfo,31 Equipped,32 ResourcePriorities,33 NextResourceId,34 ResourceId(RmrkResourceId),35 PendingNftAccept,36 PendingChildren,37 Parts,38 Base,39 Src,40 EquippedNft,41 BaseType,42 ExternalPartId,43 EquippableList,44 ZIndex,45 ThemeName,46 ThemeInherit,47 UserProperty(&'r [u8]),48}4950impl<'r> RmrkProperty<'r> {51 pub fn to_key<T: Config>(self) -> Result<PropertyKey, Error<T>> {52 fn get_bytes<T: AsRef<[u8]>>(container: &T) -> &[u8] {53 container.as_ref()54 }5556 macro_rules! key {57 ($($component:expr),+) => {58 PropertyKey::try_from([$(key!(@ &$component)),+].concat())59 .map_err(|_| <Error<T>>::RmrkPropertyKeyIsTooLong)60 };6162 (@ $key:expr) => {63 get_bytes($key)64 };65 }6667 match self {68 Self::Metadata => key!("metadata"),69 Self::CollectionType => key!("collection-type"),70 Self::RmrkInternalCollectionId => key!("internal-id"),71 Self::TokenType => key!("token-type"),72 Self::Transferable => key!("transferable"),73 Self::RoyaltyInfo => key!("royalty-info"),74 Self::Equipped => key!("equipped"),75 Self::ResourcePriorities => key!("resource-priorities"),76 Self::NextResourceId => key!("next-resource-id"),77 Self::ResourceId(id) => key!(RESOURCE_ID_PREFIX, id.to_le_bytes()),78 Self::PendingNftAccept => key!("pending-nft-accept"),79 Self::PendingChildren => key!("pending-children"),80 Self::Parts => key!("parts"),81 Self::Base => key!("base"),82 Self::Src => key!("src"),83 Self::EquippedNft => key!("equipped-nft"),84 Self::BaseType => key!("base-type"),85 Self::ExternalPartId => key!("ext-part-id"),86 Self::EquippableList => key!("equippable-list"),87 Self::ZIndex => key!("z-index"),88 Self::ThemeName => key!("theme-name"),89 Self::ThemeInherit => key!("theme-inherit"),90 Self::UserProperty(name) => key!(USER_PROPERTY_PREFIX, name),91 }92 }93}9495pub fn strip_key_prefix(key: &PropertyKey, prefix: &str) -> Option<PropertyKey> {96 let key_prefix = PropertyKey::try_from(prefix.as_bytes().to_vec()).ok()?;97 let key_prefix = PropertyScope::Rmrk.apply(key_prefix).ok()?;9899 key.as_slice().strip_prefix(key_prefix.as_slice())?100 .to_vec().try_into().ok()101}102103pub fn is_valid_key_prefix(key: &PropertyKey, prefix: &str) -> bool {104 strip_key_prefix(key, prefix).is_some()105}