difftreelog
NFT-56 + 47 added
in: master
2 files changed
pallets/nft/src/lib.rsdiffbeforeafterboth1#![cfg_attr(not(feature = "std"), no_std)]23use codec::{Decode, Encode};4/// A FRAME pallet template with necessary imports56/// Feel free to remove or edit this file as needed.7/// If you change the name of this file, make sure to update its references in runtime/src/lib.rs8/// If you remove this file, you can remove those references910/// For more guidance on Substrate FRAME, see the example pallet11/// https://github.com/paritytech/substrate/blob/master/frame/example/src/lib.rs12use frame_support::{decl_event, decl_module, decl_storage, dispatch::DispatchResult, ensure};13use frame_system::{self as system, ensure_signed};14use sp_runtime::sp_std::prelude::Vec;1516#[cfg(test)]17mod mock;1819#[cfg(test)]20mod tests;2122#[derive(Encode, Decode, Default, Clone, PartialEq)]23#[cfg_attr(feature = "std", derive(Debug))]24pub struct CollectionType<AccountId> {25 pub owner: AccountId,26 pub next_item_id: u64,27 pub custom_data_size: u32,28}2930#[derive(Encode, Decode, Default, Clone, PartialEq)]31#[cfg_attr(feature = "std", derive(Debug))]32pub struct CollectionAdminsType<AccountId> {33 pub admin: AccountId,34 pub collection_id: u64,35}3637#[derive(Encode, Decode, Default, Clone, PartialEq)]38#[cfg_attr(feature = "std", derive(Debug))]39pub struct NftItemType<AccountId> {40 pub collection: u64,41 pub owner: AccountId,42 pub data: Vec<u8>,43}4445/// The pallet's configuration trait.46pub trait Trait: system::Trait {47 // Add other types and constants required to configure this pallet.4849 /// The overarching event type.50 type Event: From<Event<Self>> + Into<<Self as system::Trait>::Event>;51}5253// This pallet's storage items.54decl_storage! {55 // It is important to update your storage name so that your pallet's56 // storage items are isolated from other pallets.57 trait Store for Module<T: Trait> as Nft {5859 /// Next available collection ID60 pub NextCollectionID get(fn next_collection_id): u64;6162 pub Collection get(collection): map hasher(identity) u64 => CollectionType<T::AccountId>;63 //pub Collection get(collection): map hasher(identity) u64 => CollectionType<T::AccountId>;6465 pub AdminList get(admin_list_collection): map hasher(identity) u64 => Vec<T::AccountId>;6667 /// Balance owner per collection map68 pub Balance get(balance_count): map hasher(blake2_128_concat) (u64, T::AccountId) => u64;69 pub ApprovedList get(approved): map hasher(blake2_128_concat) (u64, u64) => Vec<T::AccountId>;7071 pub ItemList get(item_id): map hasher(blake2_128_concat) (u64, u64) => NftItemType<T::AccountId>;72 // pub ItemList get(item_id): map hasher(blake2_128_concat) (u64, u64) => NftItemType<T::AccountId>;7374 pub ItemListIndex get(item_index): map hasher(blake2_128_concat) u64 => u64;75 // pub ItemListIndex get(item_index): map hasher(blake2_128_concat) u64 => u64;76 }77}7879// The pallet's events80decl_event!(81 pub enum Event<T>82 where83 AccountId = <T as system::Trait>::AccountId,84 {85 Created(u32, AccountId),86 }87);8889// The pallet's dispatchable functions.90decl_module! {91 /// The module declaration.92 pub struct Module<T: Trait> for enum Call where origin: T::Origin {9394 // Initializing events95 // this is needed only if you are using events in your pallet96 fn deposit_event() = default;9798 // Initializing events99 // this is needed only if you are using events in your module100 // fn deposit_event<T>() = default;101102 // Create collection of NFT with given parameters103 //104 // @param customDataSz size of custom data in each collection item105 // returns collection ID106107 // Create collection of NFT with given parameters108 //109 // @param customDataSz size of custom data in each collection item110 // returns collection ID111 #[weight = frame_support::weights::SimpleDispatchInfo::default()]112 pub fn create_collection(origin, custom_data_sz: u32) -> DispatchResult {113 // Anyone can create a collection114 let who = ensure_signed(origin)?;115116 // Generate next collection ID117 let next_id = NextCollectionID::get();118119 NextCollectionID::put(next_id);120121 // Create new collection122 let new_collection = CollectionType {123 owner: who,124 next_item_id: next_id,125 custom_data_size: custom_data_sz,126 };127128 // Add new collection to map129 <Collection<T>>::insert(next_id, new_collection);130131 Ok(())132 }133134 #[weight = frame_support::weights::SimpleDispatchInfo::default()]135 pub fn destroy_collection(origin, collection_id: u64) -> DispatchResult {136137 let sender = ensure_signed(origin)?;138 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");139140 let owner = <Collection<T>>::get(collection_id).owner;141 ensure!(sender == owner, "You do not own this collection");142 <Collection<T>>::remove(collection_id);143144 Ok(())145 }146147 #[weight = frame_support::weights::SimpleDispatchInfo::default()]148 pub fn change_collection_owner(origin, collection_id: u64, new_owner: T::AccountId) -> DispatchResult {149150 let sender = ensure_signed(origin)?;151 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");152153 let mut target_collection = <Collection<T>>::get(collection_id);154 ensure!(sender == target_collection.owner, "You do not own this collection");155156 target_collection.owner = new_owner;157 <Collection<T>>::insert(collection_id, target_collection);158159 Ok(())160 }161162 #[weight = frame_support::weights::SimpleDispatchInfo::default()]163 pub fn add_collection_admin(origin, collection_id: u64, new_admin_id: T::AccountId) -> DispatchResult {164165 let sender = ensure_signed(origin)?;166 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");167168 let target_collection = <Collection<T>>::get(collection_id);169 let is_owner = sender == target_collection.owner;170171 let no_perm_mes = "You do not have permissions to modify this collection";172 let exists = <AdminList<T>>::contains_key(collection_id);173174 if !is_owner175 {176 ensure!(exists, no_perm_mes);177 ensure!(<AdminList<T>>::get(collection_id).contains(&sender), no_perm_mes);178 }179180 let mut admin_arr: Vec<T::AccountId> = Vec::new();181 if exists182 {183 admin_arr = <AdminList<T>>::get(collection_id);184 ensure!(!admin_arr.contains(&new_admin_id), "Account already has admin role");185 }186187 admin_arr.push(new_admin_id);188 <AdminList<T>>::insert(collection_id, admin_arr);189190 Ok(())191 }192193 #[weight = frame_support::weights::SimpleDispatchInfo::default()]194 pub fn remove_collection_admin(origin, collection_id: u64, account_id: T::AccountId) -> DispatchResult {195196 let sender = ensure_signed(origin)?;197 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");198199 let target_collection = <Collection<T>>::get(collection_id);200 let is_owner = sender == target_collection.owner;201202 let no_perm_mes = "You do not have permissions to modify this collection";203 let exists = <AdminList<T>>::contains_key(collection_id);204205 if !is_owner206 {207 ensure!(exists, no_perm_mes);208 ensure!(<AdminList<T>>::get(collection_id).contains(&sender), no_perm_mes);209 }210211 if exists212 {213 let mut admin_arr = <AdminList<T>>::get(collection_id);214 admin_arr.retain(|i| *i != account_id);215 <AdminList<T>>::insert(collection_id, admin_arr);216 }217218 Ok(())219 }220221 #[weight = frame_support::weights::SimpleDispatchInfo::default()]222 pub fn create_item(origin, collection_id: u64, properties: Vec<u8>) -> DispatchResult {223224 let sender = ensure_signed(origin)?;225 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");226227 let target_collection = <Collection<T>>::get(collection_id);228 ensure!(target_collection.custom_data_size >= properties.len() as u32, "Size of item is too large");229 let is_owner = sender == target_collection.owner;230231 let no_perm_mes = "You do not have permissions to modify this collection";232 let exists = <AdminList<T>>::contains_key(collection_id);233234 if !is_owner235 {236 ensure!(exists, no_perm_mes);237 ensure!(<AdminList<T>>::get(collection_id).contains(&sender), no_perm_mes);238 }239240 let new_balance = <Balance<T>>::get((collection_id, sender.clone())) + 1;241 <Balance<T>>::insert((collection_id, sender.clone()), new_balance);242243 // Create new item244 let new_item = NftItemType {245 collection: collection_id,246 owner: sender,247 data: properties,248 };249250 let current_index = <ItemListIndex>::get(collection_id);251 <ItemListIndex>::insert(collection_id, current_index);252 <ItemList<T>>::insert((collection_id, current_index), new_item);253254 Ok(())255 }256257 #[weight = frame_support::weights::SimpleDispatchInfo::default()]258 pub fn burn_item(origin, collection_id: u64, item_id: u64) -> DispatchResult {259260 let sender = ensure_signed(origin)?;261 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");262263 let target_collection = <Collection<T>>::get(collection_id);264 let is_owner = sender == target_collection.owner;265266 ensure!(<ItemList<T>>::contains_key((collection_id, item_id)), "Item does not exists");267 let item = <ItemList<T>>::get((collection_id, item_id));268269 if !is_owner270 {271 // check if item owner272 if item.owner != sender273 {274 let no_perm_mes = "You do not have permissions to modify this collection";275276 ensure!(<AdminList<T>>::contains_key(collection_id), no_perm_mes);277 ensure!(<AdminList<T>>::get(collection_id).contains(&sender), no_perm_mes);278 }279 }280 <ItemList<T>>::remove((collection_id, item_id));281282 // update balance283 let new_balance = <Balance<T>>::get((collection_id, item.owner.clone())) - 1;284 <Balance<T>>::insert((collection_id, item.owner.clone()), new_balance);285286 Ok(())287 }288289 #[weight = frame_support::weights::SimpleDispatchInfo::default()]290 pub fn transfer(origin, collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {291292 let sender = ensure_signed(origin)?;293 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");294295 let target_collection = <Collection<T>>::get(collection_id);296 let is_owner = sender == target_collection.owner;297298 ensure!(<ItemList<T>>::contains_key((collection_id, item_id)), "Item does not exists");299 let mut item = <ItemList<T>>::get((collection_id, item_id));300301 if !is_owner302 {303 // check if item owner304 if item.owner != sender305 {306 let no_perm_mes = "You do not have permissions to modify this collection";307308 ensure!(<AdminList<T>>::contains_key(collection_id), no_perm_mes);309 ensure!(<AdminList<T>>::get(collection_id).contains(&sender), no_perm_mes);310 }311 }312 <ItemList<T>>::remove((collection_id, item_id));313314 // update balance315 let balance_old_owner = <Balance<T>>::get((collection_id, item.owner.clone())) - 1;316 <Balance<T>>::insert((collection_id, item.owner.clone()), balance_old_owner);317318 let balance_new_owner = <Balance<T>>::get((collection_id, new_owner.clone())) + 1;319 <Balance<T>>::insert((collection_id, new_owner.clone()), balance_new_owner);320321 // change owner322 item.owner = new_owner;323 <ItemList<T>>::insert((collection_id, item_id), item);324325 // reset approved list326 let itm: Vec<T::AccountId> = Vec::new();327 <ApprovedList<T>>::insert((collection_id, item_id), itm);328329 Ok(())330 }331332 #[weight = frame_support::weights::SimpleDispatchInfo::default()]333 pub fn approve(origin, approved: T::AccountId, collection_id: u64, item_id: u64) -> DispatchResult {334335 let sender = ensure_signed(origin)?;336 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");337338 let target_collection = <Collection<T>>::get(collection_id);339 let is_owner = sender == target_collection.owner;340341 ensure!(<ItemList<T>>::contains_key((collection_id, item_id)), "Item does not exists");342 let item = <ItemList<T>>::get((collection_id, item_id));343344 if !is_owner345 {346 // check if item owner347 if item.owner != sender348 {349 let no_perm_mes = "You do not have permissions to modify this collection";350351 ensure!(<AdminList<T>>::contains_key(collection_id), no_perm_mes);352 ensure!(<AdminList<T>>::get(collection_id).contains(&sender), no_perm_mes);353 }354 }355356 let list_exists = <ApprovedList<T>>::contains_key((collection_id, item_id));357 if list_exists {358359 let mut list = <ApprovedList<T>>::get((collection_id, item_id));360 let item_contains = list.contains(&approved.clone());361362 if !item_contains {363 list.push(approved.clone());364 }365 } else {366367 let mut itm = Vec::new();368 itm.push(approved.clone());369 <ApprovedList<T>>::insert((collection_id, item_id), itm);370 }371372 Ok(())373 }374375 #[weight = frame_support::weights::SimpleDispatchInfo::default()]376 pub fn transfer_from(origin, collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {377378 let no_perm_mes = "You do not have permissions to modify this collection";379 ensure!(<ApprovedList<T>>::contains_key((collection_id, item_id)), no_perm_mes);380 let list_itm = <ApprovedList<T>>::get((collection_id, item_id));381 ensure!(list_itm.contains(&new_owner.clone()), no_perm_mes);382383 Self::transfer(origin, collection_id, item_id, new_owner)?;384385 Ok(())386 }387388 #[weight = frame_support::weights::SimpleDispatchInfo::default()]389 pub fn safe_transfer_from(origin, collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {390391 let no_perm_mes = "You do not have permissions to modify this collection";392 ensure!(<ApprovedList<T>>::contains_key((collection_id, item_id)), no_perm_mes);393 let list_itm = <ApprovedList<T>>::get((collection_id, item_id));394 ensure!(list_itm.contains(&new_owner.clone()), no_perm_mes);395396 // on_nft_received call397398 Self::transfer(origin, collection_id, item_id, new_owner)?;399400 Ok(())401 }402 }403}1#![cfg_attr(not(feature = "std"), no_std)]23use codec::{Decode, Encode};4/// A FRAME pallet template with necessary imports56/// Feel free to remove or edit this file as needed.7/// If you change the name of this file, make sure to update its references in runtime/src/lib.rs8/// If you remove this file, you can remove those references910/// For more guidance on Substrate FRAME, see the example pallet11/// https://github.com/paritytech/substrate/blob/master/frame/example/src/lib.rs12use frame_support::{decl_event, decl_module, decl_storage, dispatch::DispatchResult, ensure};13use frame_system::{self as system, ensure_signed};14use sp_runtime::sp_std::prelude::Vec;1516#[cfg(test)]17mod mock;1819#[cfg(test)]20mod tests;2122#[derive(Encode, Decode, Default, Clone, PartialEq)]23#[cfg_attr(feature = "std", derive(Debug))]24pub struct CollectionType<AccountId> {25 pub owner: AccountId,26 pub next_item_id: u64,27 pub name: Vec<u16>, // 64 include null escape char28 pub description: Vec<u16>, // 256 include null escape char29 pub token_prefix: Vec<u8>, // 16 include null escape char30 pub custom_data_size: u32,31}3233#[derive(Encode, Decode, Default, Clone, PartialEq)]34#[cfg_attr(feature = "std", derive(Debug))]35pub struct CollectionAdminsType<AccountId> {36 pub admin: AccountId,37 pub collection_id: u64,38}3940#[derive(Encode, Decode, Default, Clone, PartialEq)]41#[cfg_attr(feature = "std", derive(Debug))]42pub struct NftItemType<AccountId> {43 pub collection: u64,44 pub owner: AccountId,45 pub data: Vec<u8>,46}4748/// The pallet's configuration trait.49pub trait Trait: system::Trait {50 // Add other types and constants required to configure this pallet.5152 /// The overarching event type.53 type Event: From<Event<Self>> + Into<<Self as system::Trait>::Event>;54}5556// This pallet's storage items.57decl_storage! {58 // It is important to update your storage name so that your pallet's59 // storage items are isolated from other pallets.60 trait Store for Module<T: Trait> as Nft {6162 /// Next available collection ID63 pub NextCollectionID get(fn next_collection_id): u64;64 pub Collection get(fn collection): map hasher(identity) u64 => CollectionType<T::AccountId>;65 pub AdminList get(fn admin_list_collection): map hasher(identity) u64 => Vec<T::AccountId>;6667 /// Balance owner per collection map68 pub Balance get(fn balance_count): map hasher(blake2_128_concat) (u64, T::AccountId) => u64;69 pub ApprovedList get(fn approved): map hasher(blake2_128_concat) (u64, u64) => Vec<T::AccountId>;7071 pub ItemList get(fn item_id): map hasher(blake2_128_concat) (u64, u64) => NftItemType<T::AccountId>;72 pub ItemListIndex get(fn item_index): map hasher(blake2_128_concat) u64 => u64;7374 pub AddressTokens get(fn address_tokens): map hasher(blake2_128_concat) (u64, T::AccountId) => Vec<u64>;75 }76}7778// The pallet's events79decl_event!(80 pub enum Event<T>81 where82 AccountId = <T as system::Trait>::AccountId,83 {84 Created(u64, AccountId),85 ItemCreated(u64),86 ItemDestroyed(u64, u64),87 }88);8990// The pallet's dispatchable functions.91decl_module! {92 /// The module declaration.93 pub struct Module<T: Trait> for enum Call where origin: T::Origin {9495 // Initializing events96 // this is needed only if you are using events in your pallet97 fn deposit_event() = default;9899 // Initializing events100 // this is needed only if you are using events in your module101 // fn deposit_event<T>() = default;102103 // Create collection of NFT with given parameters104 //105 // @param customDataSz size of custom data in each collection item106 // returns collection ID107108 // Create collection of NFT with given parameters109 //110 // @param customDataSz size of custom data in each collection item111 // returns collection ID112 #[weight = frame_support::weights::SimpleDispatchInfo::default()]113 pub fn create_collection( origin, 114 collection_name: Vec<u16>, 115 collection_description: Vec<u16>, 116 token_prefix: Vec<u8>, 117 custom_data_sz: u32) -> DispatchResult {118119 // Anyone can create a collection120 let who = ensure_signed(origin)?;121122 // check params 123 let mut name = collection_name.to_vec();124 name.push(0);125 ensure!(name.len() <= 64, "Collection name can not be longer than 63 char");126127 let mut description = collection_description.to_vec();128 description.push(0);129 ensure!(name.len() <= 256, "Collection description can not be longer than 255 char");130131 let mut prefix = token_prefix.to_vec();132 prefix.push(0);133 ensure!(prefix.len() <= 16, "Token prefix can not be longer than 15 char");134135 // Generate next collection ID136 let next_id = NextCollectionID::get()137 .checked_add(1)138 .expect("collection id error");139140 NextCollectionID::put(next_id);141142 // Create new collection143 let new_collection = CollectionType {144 owner: who.clone(),145 name: name,146 description: description,147 token_prefix: prefix,148 next_item_id: next_id,149 custom_data_size: custom_data_sz,150 };151152 // Add new collection to map153 <Collection<T>>::insert(next_id, new_collection);154155 // call event156 Self::deposit_event(RawEvent::Created(next_id, who.clone()));157158 Ok(())159 }160161 #[weight = frame_support::weights::SimpleDispatchInfo::default()]162 pub fn destroy_collection(origin, collection_id: u64) -> DispatchResult {163164 let sender = ensure_signed(origin)?;165 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");166167 let owner = <Collection<T>>::get(collection_id).owner;168 ensure!(sender == owner, "You do not own this collection");169 <Collection<T>>::remove(collection_id);170171 Ok(())172 }173174 #[weight = frame_support::weights::SimpleDispatchInfo::default()]175 pub fn change_collection_owner(origin, collection_id: u64, new_owner: T::AccountId) -> DispatchResult {176177 let sender = ensure_signed(origin)?;178 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");179180 let mut target_collection = <Collection<T>>::get(collection_id);181 ensure!(sender == target_collection.owner, "You do not own this collection");182183 target_collection.owner = new_owner;184 <Collection<T>>::insert(collection_id, target_collection);185186 Ok(())187 }188189 #[weight = frame_support::weights::SimpleDispatchInfo::default()]190 pub fn add_collection_admin(origin, collection_id: u64, new_admin_id: T::AccountId) -> DispatchResult {191192 let sender = ensure_signed(origin)?;193 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");194195 let target_collection = <Collection<T>>::get(collection_id);196 let is_owner = sender == target_collection.owner;197198 let no_perm_mes = "You do not have permissions to modify this collection";199 let exists = <AdminList<T>>::contains_key(collection_id);200201 if !is_owner202 {203 ensure!(exists, no_perm_mes);204 ensure!(<AdminList<T>>::get(collection_id).contains(&sender), no_perm_mes);205 }206207 let mut admin_arr: Vec<T::AccountId> = Vec::new();208 if exists209 {210 admin_arr = <AdminList<T>>::get(collection_id);211 ensure!(!admin_arr.contains(&new_admin_id), "Account already has admin role");212 }213214 admin_arr.push(new_admin_id);215 <AdminList<T>>::insert(collection_id, admin_arr);216217 Ok(())218 }219220 #[weight = frame_support::weights::SimpleDispatchInfo::default()]221 pub fn remove_collection_admin(origin, collection_id: u64, account_id: T::AccountId) -> DispatchResult {222223 let sender = ensure_signed(origin)?;224 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");225226 let target_collection = <Collection<T>>::get(collection_id);227 let is_owner = sender == target_collection.owner;228229 let no_perm_mes = "You do not have permissions to modify this collection";230 let exists = <AdminList<T>>::contains_key(collection_id);231232 if !is_owner233 {234 ensure!(exists, no_perm_mes);235 ensure!(<AdminList<T>>::get(collection_id).contains(&sender), no_perm_mes);236 }237238 if exists239 {240 let mut admin_arr = <AdminList<T>>::get(collection_id);241 admin_arr.retain(|i| *i != account_id);242 <AdminList<T>>::insert(collection_id, admin_arr);243 }244245 Ok(())246 }247248 #[weight = frame_support::weights::SimpleDispatchInfo::default()]249 pub fn create_item(origin, collection_id: u64, properties: Vec<u8>) -> DispatchResult {250251 let sender = ensure_signed(origin)?;252 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");253254 let target_collection = <Collection<T>>::get(collection_id);255 ensure!(target_collection.custom_data_size >= properties.len() as u32, "Size of item is too large");256 let is_owner = sender == target_collection.owner;257258 let no_perm_mes = "You do not have permissions to modify this collection";259 let exists = <AdminList<T>>::contains_key(collection_id);260261 if !is_owner262 {263 ensure!(exists, no_perm_mes);264 ensure!(<AdminList<T>>::get(collection_id).contains(&sender), no_perm_mes);265 }266267 let new_balance = <Balance<T>>::get((collection_id, sender.clone())) + 1;268 <Balance<T>>::insert((collection_id, sender.clone()), new_balance);269270 // Create new item271 let new_item = NftItemType {272 collection: collection_id,273 owner: sender,274 data: properties,275 };276277278 let current_index = <ItemListIndex>::get(collection_id)279 .checked_add(1)280 .expect("Item list index id error");281282 Self::add_token_index(collection_id, current_index, new_item.owner.clone())?;283284 <ItemListIndex>::insert(collection_id, current_index);285 <ItemList<T>>::insert((collection_id, current_index), new_item);286287 // call event288 Self::deposit_event(RawEvent::ItemCreated(collection_id));289290 Ok(())291 }292293 #[weight = frame_support::weights::SimpleDispatchInfo::default()]294 pub fn burn_item(origin, collection_id: u64, item_id: u64) -> DispatchResult {295296 let sender = ensure_signed(origin)?;297 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");298299 let target_collection = <Collection<T>>::get(collection_id);300 let is_owner = sender == target_collection.owner;301302 ensure!(<ItemList<T>>::contains_key((collection_id, item_id)), "Item does not exists");303 let item = <ItemList<T>>::get((collection_id, item_id));304305 if !is_owner306 {307 // check if item owner308 if item.owner != sender309 {310 let no_perm_mes = "You do not have permissions to modify this collection";311312 ensure!(<AdminList<T>>::contains_key(collection_id), no_perm_mes);313 ensure!(<AdminList<T>>::get(collection_id).contains(&sender), no_perm_mes);314 }315 }316 <ItemList<T>>::remove((collection_id, item_id));317318 Self::remove_token_index(collection_id, item_id, item.owner.clone())?;319320 // update balance321 let new_balance = <Balance<T>>::get((collection_id, item.owner.clone())) - 1;322 <Balance<T>>::insert((collection_id, item.owner.clone()), new_balance);323324 // call event325 Self::deposit_event(RawEvent::ItemDestroyed(collection_id, item_id));326327 Ok(())328 }329330 #[weight = frame_support::weights::SimpleDispatchInfo::default()]331 pub fn transfer(origin, collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {332333 let sender = ensure_signed(origin)?;334 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");335336 let target_collection = <Collection<T>>::get(collection_id);337 let is_owner = sender == target_collection.owner;338339 ensure!(<ItemList<T>>::contains_key((collection_id, item_id)), "Item does not exists");340 let mut item = <ItemList<T>>::get((collection_id, item_id));341342 if !is_owner343 {344 // check if item owner345 if item.owner != sender346 {347 let no_perm_mes = "You do not have permissions to modify this collection";348349 ensure!(<AdminList<T>>::contains_key(collection_id), no_perm_mes);350 ensure!(<AdminList<T>>::get(collection_id).contains(&sender), no_perm_mes);351 }352 }353 <ItemList<T>>::remove((collection_id, item_id));354355 // update balance356 let balance_old_owner = <Balance<T>>::get((collection_id, item.owner.clone())) - 1;357 <Balance<T>>::insert((collection_id, item.owner.clone()), balance_old_owner);358359 let balance_new_owner = <Balance<T>>::get((collection_id, new_owner.clone())) + 1;360 <Balance<T>>::insert((collection_id, new_owner.clone()), balance_new_owner);361362 // change owner363 let old_owner = item.owner.clone();364 item.owner = new_owner.clone();365 <ItemList<T>>::insert((collection_id, item_id), item);366367 // update index collection368 Self::move_token_index(collection_id, item_id, old_owner, new_owner.clone())?;369370 // reset approved list371 let itm: Vec<T::AccountId> = Vec::new();372 <ApprovedList<T>>::insert((collection_id, item_id), itm);373374 Ok(())375 }376377 #[weight = frame_support::weights::SimpleDispatchInfo::default()]378 pub fn approve(origin, approved: T::AccountId, collection_id: u64, item_id: u64) -> DispatchResult {379380 let sender = ensure_signed(origin)?;381 ensure!(<Collection<T>>::contains_key(collection_id), "This collection does not exist");382383 let target_collection = <Collection<T>>::get(collection_id);384 let is_owner = sender == target_collection.owner;385386 ensure!(<ItemList<T>>::contains_key((collection_id, item_id)), "Item does not exists");387 let item = <ItemList<T>>::get((collection_id, item_id));388389 if !is_owner390 {391 // check if item owner392 if item.owner != sender393 {394 let no_perm_mes = "You do not have permissions to modify this collection";395396 ensure!(<AdminList<T>>::contains_key(collection_id), no_perm_mes);397 ensure!(<AdminList<T>>::get(collection_id).contains(&sender), no_perm_mes);398 }399 }400401 let list_exists = <ApprovedList<T>>::contains_key((collection_id, item_id));402 if list_exists {403404 let mut list = <ApprovedList<T>>::get((collection_id, item_id));405 let item_contains = list.contains(&approved.clone());406407 if !item_contains {408 list.push(approved.clone());409 }410 } else {411412 let mut itm = Vec::new();413 itm.push(approved.clone());414 <ApprovedList<T>>::insert((collection_id, item_id), itm);415 }416417 Ok(())418 }419420 #[weight = frame_support::weights::SimpleDispatchInfo::default()]421 pub fn transfer_from(origin, collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {422423 let no_perm_mes = "You do not have permissions to modify this collection";424 ensure!(<ApprovedList<T>>::contains_key((collection_id, item_id)), no_perm_mes);425 let list_itm = <ApprovedList<T>>::get((collection_id, item_id));426 ensure!(list_itm.contains(&new_owner.clone()), no_perm_mes);427428 Self::transfer(origin, collection_id, item_id, new_owner)?;429430 Ok(())431 }432433 #[weight = frame_support::weights::SimpleDispatchInfo::default()]434 pub fn safe_transfer_from(origin, collection_id: u64, item_id: u64, new_owner: T::AccountId) -> DispatchResult {435436 let no_perm_mes = "You do not have permissions to modify this collection";437 ensure!(<ApprovedList<T>>::contains_key((collection_id, item_id)), no_perm_mes);438 let list_itm = <ApprovedList<T>>::get((collection_id, item_id));439 ensure!(list_itm.contains(&new_owner.clone()), no_perm_mes);440441 // on_nft_received call442443 Self::transfer(origin, collection_id, item_id, new_owner)?;444445 Ok(())446 }447 }448}449450451impl<T: Trait> Module<T> {452 fn add_token_index(collection_id: u64, item_index: u64, owner: T::AccountId) -> DispatchResult {453 454 let list_exists = <AddressTokens<T>>::contains_key((collection_id, owner.clone()));455 if list_exists {456457 let mut list = <AddressTokens<T>>::get((collection_id, owner.clone()));458 let item_contains = list.contains(&item_index.clone());459460 if !item_contains {461 list.push(item_index.clone());462 }463464 <AddressTokens<T>>::insert((collection_id, owner.clone()), list);465466 } else {467468 let mut itm = Vec::new();469 itm.push(item_index.clone());470 <AddressTokens<T>>::insert((collection_id, owner), itm);471 }472473 Ok(())474 }475476 fn remove_token_index(collection_id: u64, item_index: u64, owner: T::AccountId) -> DispatchResult {477 478 let list_exists = <AddressTokens<T>>::contains_key((collection_id, owner.clone()));479 if list_exists {480481 let mut list = <AddressTokens<T>>::get((collection_id, owner.clone()));482 let item_contains = list.contains(&item_index.clone());483484 if item_contains {485 list.retain(|&item| item != item_index);486 <AddressTokens<T>>::insert((collection_id, owner), list);487 }488 }489490 Ok(())491 }492493 fn move_token_index(collection_id: u64, item_index: u64, old_owner: T::AccountId, new_owner: T::AccountId) -> DispatchResult {494 495 Self::remove_token_index(collection_id, item_index, old_owner)?;496 Self::add_token_index(collection_id, item_index, new_owner)?;497 498 Ok(())499 }500}pallets/nft/src/tests.rsdiffbeforeafterboth--- a/pallets/nft/src/tests.rs
+++ b/pallets/nft/src/tests.rs
@@ -5,9 +5,14 @@
#[test]
fn create_collection_test() {
new_test_ext().execute_with(|| {
+
+ let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
+ let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
+ let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
+
let size = 1024;
let origin1 = Origin::signed(1);
- assert_ok!(TemplateModule::create_collection(origin1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
assert_eq!(TemplateModule::collection(1).owner, 1);
});
}
@@ -15,10 +20,15 @@
#[test]
fn change_collection_owner() {
new_test_ext().execute_with(|| {
+
+ let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
+ let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
+ let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
+
let size = 1024;
let origin1 = Origin::signed(1);
- assert_ok!(TemplateModule::create_collection(origin1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
assert_ok!(TemplateModule::change_collection_owner(
origin1.clone(),
1,
@@ -31,10 +41,15 @@
#[test]
fn destroy_collection() {
new_test_ext().execute_with(|| {
+
+ let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
+ let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
+ let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
+
let size = 1024;
let origin1 = Origin::signed(1);
- assert_ok!(TemplateModule::create_collection(origin1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
assert_ok!(TemplateModule::destroy_collection(origin1.clone(), 1));
});
}
@@ -42,11 +57,16 @@
#[test]
fn create_item() {
new_test_ext().execute_with(|| {
+
+ let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
+ let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
+ let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
+
let size = 1024;
let origin1 = Origin::signed(1);
let origin2 = Origin::signed(2);
- assert_ok!(TemplateModule::create_collection(origin1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));
assert_ok!(TemplateModule::create_item(
origin2.clone(),
@@ -62,11 +82,16 @@
#[test]
fn burn_item() {
new_test_ext().execute_with(|| {
+
+ let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
+ let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
+ let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
+
let size = 1024;
let origin1 = Origin::signed(1);
let origin2 = Origin::signed(2);
- assert_ok!(TemplateModule::create_collection(origin1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
assert_ok!(TemplateModule::add_collection_admin(origin1.clone(), 1, 2));
assert_ok!(TemplateModule::create_item(
origin2.clone(),
@@ -91,14 +116,19 @@
#[test]
fn add_collection_admin() {
new_test_ext().execute_with(|| {
+
+ let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
+ let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
+ let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
+
let size = 1024;
let origin1 = Origin::signed(1);
let origin2 = Origin::signed(2);
let origin3 = Origin::signed(3);
- assert_ok!(TemplateModule::create_collection(origin1.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin2.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin3.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
assert_eq!(TemplateModule::collection(1).owner, 1);
assert_eq!(TemplateModule::collection(2).owner, 2);
@@ -116,14 +146,19 @@
#[test]
fn remove_collection_admin() {
new_test_ext().execute_with(|| {
+
+ let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
+ let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
+ let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
+
let size = 1024;
let origin1 = Origin::signed(1);
let origin2 = Origin::signed(2);
let origin3 = Origin::signed(3);
- assert_ok!(TemplateModule::create_collection(origin1.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin2.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin3.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
assert_eq!(TemplateModule::collection(1).owner, 1);
assert_eq!(TemplateModule::collection(2).owner, 2);
@@ -149,14 +184,19 @@
#[test]
fn balance_of() {
new_test_ext().execute_with(|| {
+
+ let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
+ let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
+ let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
+
let size = 1024;
let origin1 = Origin::signed(1);
let origin2 = Origin::signed(2);
let origin3 = Origin::signed(3);
- assert_ok!(TemplateModule::create_collection(origin1.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin2.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin3.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
assert_eq!(TemplateModule::collection(1).owner, 1);
assert_eq!(TemplateModule::collection(2).owner, 2);
@@ -181,14 +221,19 @@
#[test]
fn transfer() {
new_test_ext().execute_with(|| {
+
+ let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
+ let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
+ let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
+
let size = 1024;
let origin1 = Origin::signed(1);
let origin2 = Origin::signed(2);
let origin3 = Origin::signed(3);
- assert_ok!(TemplateModule::create_collection(origin1.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin2.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin3.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
assert_eq!(TemplateModule::collection(1).owner, 1);
assert_eq!(TemplateModule::collection(2).owner, 2);
@@ -214,14 +259,19 @@
#[test]
fn approve() {
new_test_ext().execute_with(|| {
+
+ let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
+ let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
+ let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
+
let size = 1024;
let origin1 = Origin::signed(1);
let origin2 = Origin::signed(2);
let origin3 = Origin::signed(3);
- assert_ok!(TemplateModule::create_collection(origin1.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin2.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin3.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
assert_eq!(TemplateModule::collection(1).owner, 1);
assert_eq!(TemplateModule::collection(2).owner, 2);
@@ -243,14 +293,19 @@
#[test]
fn get_approved() {
new_test_ext().execute_with(|| {
+
+ let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
+ let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
+ let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
+
let size = 1024;
let origin1 = Origin::signed(1);
let origin2 = Origin::signed(2);
let origin3 = Origin::signed(3);
- assert_ok!(TemplateModule::create_collection(origin1.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin2.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin3.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
assert_eq!(TemplateModule::collection(1).owner, 1);
assert_eq!(TemplateModule::collection(2).owner, 2);
@@ -272,14 +327,19 @@
#[test]
fn transfer_from() {
new_test_ext().execute_with(|| {
+
+ let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
+ let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
+ let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
+
let size = 1024;
let origin1 = Origin::signed(1);
let origin2 = Origin::signed(2);
let origin3 = Origin::signed(3);
- assert_ok!(TemplateModule::create_collection(origin1.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin2.clone(), size));
- assert_ok!(TemplateModule::create_collection(origin3.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
assert_eq!(TemplateModule::collection(1).owner, 1);
assert_eq!(TemplateModule::collection(2).owner, 2);
@@ -297,3 +357,50 @@
assert_ok!(TemplateModule::transfer_from(origin1.clone(), 1, 1, 2));
});
}
+
+#[test]
+fn index_list() {
+ new_test_ext().execute_with(|| {
+
+ let col_name1: Vec<u16> = "Test1\0".encode_utf16().collect::<Vec<u16>>();
+ let col_desc1: Vec<u16> = "TestDescription1\0".encode_utf16().collect::<Vec<u16>>();
+ let token_prefix1: Vec<u8> = b"token_prefix1\0".to_vec();
+
+ let size = 1024;
+ let origin1 = Origin::signed(1);
+ let origin2 = Origin::signed(2);
+ let origin3 = Origin::signed(3);
+
+ assert_ok!(TemplateModule::create_collection(origin1.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin2.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+ assert_ok!(TemplateModule::create_collection(origin3.clone(), col_name1.clone(), col_desc1.clone(), token_prefix1.clone(), size));
+
+ assert_eq!(TemplateModule::collection(1).owner, 1);
+ assert_eq!(TemplateModule::collection(2).owner, 2);
+ assert_eq!(TemplateModule::collection(3).owner, 3);
+ // create items
+ assert_ok!(TemplateModule::create_item(
+ origin1.clone(),
+ 1,
+ [1, 1, 1].to_vec()
+ ));
+ assert_ok!(TemplateModule::create_item(
+ origin1.clone(),
+ 1,
+ [1, 1, 2].to_vec()
+ ));
+ assert_ok!(TemplateModule::create_item(
+ origin1.clone(),
+ 1,
+ [1, 2, 3].to_vec()
+ ));
+ assert_eq!(TemplateModule::address_tokens((1, 1)).len(), 3);
+ // burn one
+ assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 2));
+ assert_eq!(TemplateModule::address_tokens((1, 1)).len(), 2);
+ // burn another one
+ assert_ok!(TemplateModule::burn_item(origin1.clone(), 1, 3));
+ assert_eq!(TemplateModule::address_tokens((1, 1))[0], 1);
+ });
+}
+