pallet-derivatives 0.7.0

// This file is part of Substrate.

// Copyright (C) Parity Technologies (UK) Ltd.
// SPDX-License-Identifier: Apache-2.0

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// 	http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

use super::*;

pub type PredefinedIdNftsInstance = unique_items::Instance3;
pub type NftLocalId = u64;
pub type NftFullId = (CollectionAutoId, NftLocalId);
impl unique_items::Config<PredefinedIdNftsInstance> for Test {
	type ItemId = NftFullId;
}

// Below are the operations implementations:
// * NFT creation (with a predefined ID)
// * NFT transfer from one account to another (checks if the `from` account is the NFT's current
//   owner)

impl Create<WithConfig<ConfigValue<Owner<AccountId>>, PredefinedId<NftFullId>>>
	for PredefinedIdNfts
{
	fn create(
		strategy: WithConfig<ConfigValue<Owner<AccountId>>, PredefinedId<NftFullId>>,
	) -> Result<NftFullId, DispatchError> {
		let WithConfig { config: ConfigValue(owner), extra: id_assignment } = strategy;
		let id = id_assignment.params;

		unique_items::ItemOwner::<Test, PredefinedIdNftsInstance>::try_mutate(
			id,
			|current_owner| {
				if current_owner.is_none() {
					*current_owner = Some(owner);
					Ok(())
				} else {
					Err(unique_items::Error::<Test, PredefinedIdNftsInstance>::AlreadyExists)
				}
			},
		)?;

		Ok(id)
	}
}
impl AssetDefinition for PredefinedIdNfts {
	type Id = (CollectionAutoId, NftLocalId);
}
impl Inspect<CanCreate> for PredefinedIdNfts {
	fn inspect(id: &Self::Id, _: CanCreate) -> Result<bool, DispatchError> {
		let nft_exists =
			unique_items::ItemOwner::<Test, PredefinedIdNftsInstance>::contains_key(id);

		Ok(!nft_exists)
	}
}
impl Update<ChangeOwnerFrom<AccountId>> for PredefinedIdNfts {
	fn update(
		id: &Self::Id,
		strategy: ChangeOwnerFrom<AccountId>,
		new_owner: &AccountId,
	) -> DispatchResult {
		let CheckState(check_owner, _) = strategy;

		unique_items::ItemOwner::<Test, PredefinedIdNftsInstance>::try_mutate(id, |owner| {
			match owner {
				Some(current_owner) => {
					if *current_owner == check_owner {
						*owner = Some(*new_owner);
						Ok(())
					} else {
						Err(unique_items::Error::<Test, PredefinedIdNftsInstance>::NoPermission
							.into())
					}
				},
				None => {
					Err(unique_items::Error::<Test, PredefinedIdNftsInstance>::UnknownItem.into())
				},
			}
		})
	}
}