// Copyright 2019 Parity Technologies (UK) Ltd.
// This file is part of Substrate.

// Substrate is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// Substrate is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with Substrate.  If not, see <http://www.gnu.org/licenses/>.

#![recursion_limit="128"]

use sr_primitives::{generic, BuildStorage, traits::{BlakeTwo256, Block as _, Verify}};
use support::{
	Parameter, traits::Get, parameter_types,
	metadata::{
		DecodeDifferent, StorageMetadata, StorageEntryModifier, StorageEntryType, DefaultByteGetter,
		StorageEntryMetadata, StorageHasher,
	},
	StorageValue, StorageMap, StorageLinkedMap, StorageDoubleMap,
};
use inherents::{
	ProvideInherent, InherentData, InherentIdentifier, RuntimeString, MakeFatalError
};
use primitives::{H256, sr25519};

mod system;

pub trait Currency {}

// Test for:
// * No default instance
// * Custom InstantiableTrait
// * Origin, Inherent, Event
mod module1 {
	use super::*;

	pub trait Trait<I>: system::Trait where <Self as system::Trait>::BlockNumber: From<u32> {
		type Event: From<Event<Self, I>> + Into<<Self as system::Trait>::Event>;
		type Origin: From<Origin<Self, I>>;
		type SomeParameter: Get<u32>;
		type GenericType: Default + Clone + codec::Codec + codec::EncodeLike;
	}

	support::decl_module! {
		pub struct Module<T: Trait<I>, I: InstantiableThing> for enum Call where
			origin: <T as system::Trait>::Origin,
			T::BlockNumber: From<u32>
		{
			fn offchain_worker() {}

			fn deposit_event() = default;

			fn one(origin) {
				system::ensure_root(origin)?;
				Self::deposit_event(RawEvent::AnotherVariant(3));
			}
		}
	}

	support::decl_storage! {
		trait Store for Module<T: Trait<I>, I: InstantiableThing> as Module1 where
			T::BlockNumber: From<u32> + std::fmt::Display
		{
			pub Value config(value): T::GenericType;
			pub Map: map u32 => u64;
			pub LinkedMap: linked_map u32 => u64;
		}

		add_extra_genesis {
			config(test) : T::BlockNumber;
			build(|config: &Self| {
				println!("{}", config.test);
			});
		}
	}

	support::decl_event! {
		pub enum Event<T, I> where Phantom = std::marker::PhantomData<T> {
			_Phantom(Phantom),
			AnotherVariant(u32),
		}
	}

	#[derive(PartialEq, Eq, Clone, sr_primitives::RuntimeDebug)]
	pub enum Origin<T: Trait<I>, I> where T::BlockNumber: From<u32> {
		Members(u32),
		_Phantom(std::marker::PhantomData<(T, I)>),
	}

	pub const INHERENT_IDENTIFIER: InherentIdentifier = *b"12345678";

	impl<T: Trait<I>, I: InstantiableThing> ProvideInherent for Module<T, I> where
		T::BlockNumber: From<u32>
	{
		type Call = Call<T, I>;
		type Error = MakeFatalError<RuntimeString>;
		const INHERENT_IDENTIFIER: InherentIdentifier = INHERENT_IDENTIFIER;

		fn create_inherent(_data: &InherentData) -> Option<Self::Call> {
			unimplemented!();
		}

		fn check_inherent(_: &Self::Call, _: &InherentData) -> std::result::Result<(), Self::Error> {
			unimplemented!();
		}
	}
}

// Test for:
// * default instance
// * use of no_genesis_config_phantom_data
mod module2 {
	use super::*;

	pub trait Trait<I=DefaultInstance>: system::Trait {
		type Amount: Parameter + Default;
		type Event: From<Event<Self, I>> + Into<<Self as system::Trait>::Event>;
		type Origin: From<Origin<Self, I>>;
	}

	impl<T: Trait<I>, I: Instance> Currency for Module<T, I> {}

	support::decl_module! {
		pub struct Module<T: Trait<I>, I: Instance=DefaultInstance> for enum Call where
			origin: <T as system::Trait>::Origin
		{
			fn deposit_event() = default;
		}
	}

	support::decl_storage! {
		trait Store for Module<T: Trait<I>, I: Instance=DefaultInstance> as Module2 {
			pub Value config(value): T::Amount;
			pub Map config(map): map u64 => u64;
			pub LinkedMap config(linked_map): linked_map u64 => Vec<u8>;
			pub DoubleMap config(double_map): double_map u64, blake2_256(u64) => u64;
		}
	}

	support::decl_event! {
		pub enum Event<T, I=DefaultInstance> where Amount = <T as Trait<I>>::Amount {
			Variant(Amount),
		}
	}

	#[derive(PartialEq, Eq, Clone, sr_primitives::RuntimeDebug)]
	pub enum Origin<T: Trait<I>, I=DefaultInstance> {
		Members(u32),
		_Phantom(std::marker::PhantomData<(T, I)>),
	}

	pub const INHERENT_IDENTIFIER: InherentIdentifier = *b"12345678";

	impl<T: Trait<I>, I: Instance> ProvideInherent for Module<T, I> {
		type Call = Call<T, I>;
		type Error = MakeFatalError<RuntimeString>;
		const INHERENT_IDENTIFIER: InherentIdentifier = INHERENT_IDENTIFIER;

		fn create_inherent(_data: &InherentData) -> Option<Self::Call> {
			unimplemented!();
		}

		fn check_inherent(_call: &Self::Call, _data: &InherentData) -> std::result::Result<(), Self::Error> {
			unimplemented!();
		}
	}
}

// Test for:
// * Depends on multiple instances of a module with instances
mod module3 {
	use super::*;

	pub trait Trait: module2::Trait + module2::Trait<module2::Instance1> + system::Trait {
		type Currency: Currency;
		type Currency2: Currency;
	}

	support::decl_module! {
		pub struct Module<T: Trait> for enum Call where origin: <T as system::Trait>::Origin {}
	}
}

parameter_types! {
	pub const SomeValue: u32 = 100;
}

impl module1::Trait<module1::Instance1> for Runtime {
	type Event = Event;
	type Origin = Origin;
	type SomeParameter = SomeValue;
	type GenericType = u32;
}
impl module1::Trait<module1::Instance2> for Runtime {
	type Event = Event;
	type Origin = Origin;
	type SomeParameter = SomeValue;
	type GenericType = u32;
}
impl module2::Trait for Runtime {
	type Amount = u16;
	type Event = Event;
	type Origin = Origin;
}
impl module2::Trait<module2::Instance1> for Runtime {
	type Amount = u32;
	type Event = Event;
	type Origin = Origin;
}
impl module2::Trait<module2::Instance2> for Runtime {
	type Amount = u32;
	type Event = Event;
	type Origin = Origin;
}
impl module2::Trait<module2::Instance3> for Runtime {
	type Amount = u64;
	type Event = Event;
	type Origin = Origin;
}
impl module3::Trait for Runtime {
	type Currency = Module2_2;
	type Currency2 = Module2_3;
}

pub type Signature = sr25519::Signature;
pub type AccountId = <Signature as Verify>::Signer;
pub type BlockNumber = u64;
pub type Index = u64;

impl system::Trait for Runtime {
	type Hash = H256;
	type Origin = Origin;
	type BlockNumber = BlockNumber;
	type AccountId = AccountId;
	type Event = Event;
}

support::construct_runtime!(
	pub enum Runtime where
		Block = Block,
		NodeBlock = Block,
		UncheckedExtrinsic = UncheckedExtrinsic
	{
		System: system::{Module, Call, Event},
		Module1_1: module1::<Instance1>::{
			Module, Call, Storage, Event<T>, Config<T>, Origin<T>, Inherent
		},
		Module1_2: module1::<Instance2>::{
			Module, Call, Storage, Event<T>, Config<T>, Origin<T>, Inherent
		},
		Module2: module2::{Module, Call, Storage, Event<T>, Config<T>, Origin<T>, Inherent},
		Module2_1: module2::<Instance1>::{
			Module, Call, Storage, Event<T>, Config<T>, Origin<T>, Inherent
		},
		Module2_2: module2::<Instance2>::{
			Module, Call, Storage, Event<T>, Config<T>, Origin<T>, Inherent
		},
		Module2_3: module2::<Instance3>::{
			Module, Call, Storage, Event<T>, Config<T>, Origin<T>, Inherent
		},
		Module3: module3::{Module, Call},
	}
);

pub type Header = generic::Header<BlockNumber, BlakeTwo256>;
pub type Block = generic::Block<Header, UncheckedExtrinsic>;
pub type UncheckedExtrinsic = generic::UncheckedExtrinsic<u32, Call, Signature, ()>;

fn new_test_ext() -> runtime_io::TestExternalities {
	GenesisConfig{
		module1_Instance1: Some(module1::GenesisConfig {
			value: 3,
			test: 2,
		}),
		module1_Instance2: Some(module1::GenesisConfig {
			value: 4,
			test: 5,
		}),
		module2: Some(module2::GenesisConfig {
			value: 4,
			map: vec![(0, 0)],
			linked_map: vec![(0, vec![0])],
			double_map: vec![(0, 0, 0)],
		}),
		module2_Instance1: Some(module2::GenesisConfig {
			value: 4,
			map: vec![(0, 0)],
			linked_map: vec![(0, vec![0])],
			double_map: vec![(0, 0, 0)],
		}),
		module2_Instance2: None,
		module2_Instance3: None,
	}.build_storage().unwrap().into()
}

#[test]
fn storage_instance_independance() {
	let mut storage = (std::collections::HashMap::new(), std::collections::HashMap::new());
	runtime_io::with_storage(&mut storage, || {
		module2::Value::<Runtime>::put(0);
		module2::Value::<Runtime, module2::Instance1>::put(0);
		module2::Value::<Runtime, module2::Instance2>::put(0);
		module2::Value::<Runtime, module2::Instance3>::put(0);
		module2::Map::<module2::DefaultInstance>::insert(0, 0);
		module2::Map::<module2::Instance1>::insert(0, 0);
		module2::Map::<module2::Instance2>::insert(0, 0);
		module2::Map::<module2::Instance3>::insert(0, 0);
		module2::LinkedMap::<module2::DefaultInstance>::insert::<_, Vec<u8>>(0, vec![]);
		module2::LinkedMap::<module2::Instance1>::insert::<_, Vec<u8>>(0, vec![]);
		module2::LinkedMap::<module2::Instance2>::insert::<_, Vec<u8>>(0, vec![]);
		module2::LinkedMap::<module2::Instance3>::insert::<_, Vec<u8>>(0, vec![]);
		module2::DoubleMap::<module2::DefaultInstance>::insert(&0, &0, &0);
		module2::DoubleMap::<module2::Instance1>::insert(&0, &0, &0);
		module2::DoubleMap::<module2::Instance2>::insert(&0, &0, &0);
		module2::DoubleMap::<module2::Instance3>::insert(&0, &0, &0);
	});
	// 16 storage values + 4 linked_map head.
	assert_eq!(storage.0.len(), 16 + 4);
}

#[test]
fn storage_with_instance_basic_operation() {
	new_test_ext().execute_with(|| {
		type Value = module2::Value<Runtime, module2::Instance1>;
		type Map = module2::Map<module2::Instance1>;
		type LinkedMap = module2::LinkedMap<module2::Instance1>;
		type DoubleMap = module2::DoubleMap<module2::Instance1>;

		assert_eq!(Value::exists(), true);
		assert_eq!(Value::get(), 4);
		Value::put(1);
		assert_eq!(Value::get(), 1);
		assert_eq!(Value::take(), 1);
		assert_eq!(Value::get(), 0);
		Value::mutate(|a| *a=2);
		assert_eq!(Value::get(), 2);
		Value::kill();
		assert_eq!(Value::exists(), false);
		assert_eq!(Value::get(), 0);

		let key = 1;
		assert_eq!(Map::exists(0), true);
		assert_eq!(Map::exists(key), false);
		Map::insert(key, 1);
		assert_eq!(Map::get(key), 1);
		assert_eq!(Map::take(key), 1);
		assert_eq!(Map::get(key), 0);
		Map::mutate(key, |a| *a=2);
		assert_eq!(Map::get(key), 2);
		Map::remove(key);
		assert_eq!(Map::exists(key), false);
		assert_eq!(Map::get(key), 0);

		assert_eq!(LinkedMap::exists(0), true);
		assert_eq!(LinkedMap::exists(key), false);
		LinkedMap::insert(key, vec![1]);
		assert_eq!(LinkedMap::enumerate().count(), 2);
		assert_eq!(LinkedMap::get(key), vec![1]);
		assert_eq!(LinkedMap::take(key), vec![1]);
		assert_eq!(LinkedMap::enumerate().count(), 1);
		assert_eq!(LinkedMap::get(key), vec![]);
		LinkedMap::mutate(key, |a| *a=vec![2]);
		assert_eq!(LinkedMap::enumerate().count(), 2);
		assert_eq!(LinkedMap::get(key), vec![2]);
		LinkedMap::remove(key);
		assert_eq!(LinkedMap::enumerate().count(), 1);
		assert_eq!(LinkedMap::exists(key), false);
		assert_eq!(LinkedMap::get(key), vec![]);
		assert_eq!(LinkedMap::exists(key), false);
		assert_eq!(LinkedMap::enumerate().count(), 1);
		LinkedMap::insert(key, &vec![1]);
		assert_eq!(LinkedMap::enumerate().count(), 2);

		let key1 = 1;
		let key2 = 1;
		assert_eq!(DoubleMap::exists(&0, &0), true);
		assert_eq!(DoubleMap::exists(&key1, &key2), false);
		DoubleMap::insert(&key1, &key2, &1);
		assert_eq!(DoubleMap::get(&key1, &key2), 1);
		assert_eq!(DoubleMap::take(&key1, &key2), 1);
		assert_eq!(DoubleMap::get(&key1, &key2), 0);
		DoubleMap::mutate(&key1, &key2, |a| *a=2);
		assert_eq!(DoubleMap::get(&key1, &key2), 2);
		DoubleMap::remove(&key1, &key2);
		assert_eq!(DoubleMap::get(&key1, &key2), 0);
	});
}

const EXPECTED_METADATA: StorageMetadata = StorageMetadata {
	prefix: DecodeDifferent::Encode("Instance2Module2"),
	entries: DecodeDifferent::Encode(
		&[
			StorageEntryMetadata {
				name: DecodeDifferent::Encode("Value"),
				modifier: StorageEntryModifier::Default,
				ty: StorageEntryType::Plain(DecodeDifferent::Encode("T::Amount")),
				default: DecodeDifferent::Encode(
					DefaultByteGetter(
						&module2::__GetByteStructValue(
							std::marker::PhantomData::<(Runtime, module2::Instance2)>
						)
					)
				),
				documentation: DecodeDifferent::Encode(&[]),
			},
			StorageEntryMetadata {
				name: DecodeDifferent::Encode("Map"),
				modifier: StorageEntryModifier::Default,
				ty: StorageEntryType::Map {
					hasher: StorageHasher::Blake2_256,
					key: DecodeDifferent::Encode("u64"),
					value: DecodeDifferent::Encode("u64"),
					is_linked: false,
				},
				default: DecodeDifferent::Encode(
					DefaultByteGetter(
						&module2::__GetByteStructMap(
							std::marker::PhantomData::<(Runtime, module2::Instance2)>
						)
					)
				),
				documentation: DecodeDifferent::Encode(&[]),
			},
			StorageEntryMetadata {
				name: DecodeDifferent::Encode("LinkedMap"),
				modifier: StorageEntryModifier::Default,
				ty: StorageEntryType::Map {
					hasher: StorageHasher::Blake2_256,
					key: DecodeDifferent::Encode("u64"),
					value: DecodeDifferent::Encode("Vec<u8>"),
					is_linked: true,
				},
				default: DecodeDifferent::Encode(
					DefaultByteGetter(
						&module2::__GetByteStructLinkedMap(
							std::marker::PhantomData::<(Runtime, module2::Instance2)>
						)
					)
				),
				documentation: DecodeDifferent::Encode(&[]),
			},
			StorageEntryMetadata {
				name: DecodeDifferent::Encode("DoubleMap"),
				modifier: StorageEntryModifier::Default,
				ty: StorageEntryType::DoubleMap {
					hasher: StorageHasher::Blake2_256,
					key2_hasher: StorageHasher::Blake2_256,
					key1: DecodeDifferent::Encode("u64"),
					key2: DecodeDifferent::Encode("u64"),
					value: DecodeDifferent::Encode("u64"),
				},
				default: DecodeDifferent::Encode(
					DefaultByteGetter(
						&module2::__GetByteStructDoubleMap(
							std::marker::PhantomData::<(Runtime, module2::Instance2)>
						)
					)
				),
				documentation: DecodeDifferent::Encode(&[]),
			}
		]
	)
};

#[test]
fn test_instance_storage_metadata() {
	let metadata = Module2_2::storage_metadata();
	pretty_assertions::assert_eq!(EXPECTED_METADATA, metadata);
}

#[test]
fn instance_prefix_is_prefix_of_entries() {
	use module2::Instance;

	let prefix = module2::Instance2::PREFIX;
	assert!(module2::Instance2::PREFIX_FOR_Value.starts_with(prefix));
	assert!(module2::Instance2::PREFIX_FOR_Map.starts_with(prefix));
	assert!(module2::Instance2::PREFIX_FOR_LinkedMap.starts_with(prefix));
	assert!(module2::Instance2::PREFIX_FOR_DoubleMap.starts_with(prefix));
}