use super::*;
use alloc::vec::Vec;
use pezframe_support::{
pezpallet_prelude::{DispatchResult, *},
traits::nonfungible::{Inspect, Mutate, Transfer},
};
impl<T: Config> Inspect<T::AccountId> for Pezpallet<T> {
type ItemId = u128;
fn owner(item: &Self::ItemId) -> Option<T::AccountId> {
let record = Regions::<T>::get(RegionId::from(*item))?;
record.owner
}
fn attribute(item: &Self::ItemId, key: &[u8]) -> Option<Vec<u8>> {
let id = RegionId::from(*item);
let item = Regions::<T>::get(id)?;
match key {
b"begin" => Some(id.begin.encode()),
b"end" => Some(item.end.encode()),
b"length" => Some(item.end.saturating_sub(id.begin).encode()),
b"core" => Some(id.core.encode()),
b"part" => Some(id.mask.encode()),
b"owner" => Some(item.owner.encode()),
b"paid" => Some(item.paid.encode()),
_ => None,
}
}
}
impl<T: Config> Transfer<T::AccountId> for Pezpallet<T> {
fn transfer(item: &Self::ItemId, dest: &T::AccountId) -> DispatchResult {
Self::do_transfer((*item).into(), None, dest.clone()).map_err(Into::into)
}
}
impl<T: Config> Mutate<T::AccountId> for Pezpallet<T> {
fn mint_into(item: &Self::ItemId, who: &T::AccountId) -> DispatchResult {
let region_id: RegionId = (*item).into();
let record = Regions::<T>::get(®ion_id).ok_or(Error::<T>::UnknownRegion)?;
ensure!(record.owner.is_none(), Error::<T>::NotAllowed);
Self::issue(
region_id.core,
region_id.begin,
region_id.mask,
record.end,
Some(who.clone()),
record.paid,
);
Ok(())
}
fn burn(item: &Self::ItemId, maybe_check_owner: Option<&T::AccountId>) -> DispatchResult {
let region_id: RegionId = (*item).into();
let mut record = Regions::<T>::get(®ion_id).ok_or(Error::<T>::UnknownRegion)?;
if let Some(owner) = maybe_check_owner {
ensure!(Some(owner.clone()) == record.owner, Error::<T>::NotOwner);
}
record.owner = None;
Regions::<T>::insert(region_id, record);
Ok(())
}
}