// Copyright 2019-2021 Parity Technologies (UK) Ltd.
// This file is part of Parity Bridges Common.

// Parity Bridges Common 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.

// Parity Bridges Common 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 Parity Bridges Common.  If not, see <http://www.gnu.org/licenses/>.

//! The library of substrate relay. contains some public codes to provide to substrate relay.

#![warn(missing_docs)]

use relay_substrate_client::Error as SubstrateError;
use std::marker::PhantomData;

pub mod error;
pub mod finality;
pub mod messages_lane;
pub mod messages_metrics;
pub mod messages_source;
pub mod messages_target;
pub mod on_demand;
pub mod parachains;

/// Transaction creation parameters.
#[derive(Clone, Debug)]
pub struct TransactionParams<TS> {
	/// Transactions author.
	pub signer: TS,
	/// Transactions mortality.
	pub mortality: Option<u32>,
}

/// Tagged relay account, which balance may be exposed as metrics by the relay.
#[derive(Clone, Debug)]
pub enum TaggedAccount<AccountId> {
	/// Account, used to sign headers relay transactions from given bridged chain.
	Headers {
		/// Account id.
		id: AccountId,
		/// Name of the bridged chain, which headers are relayed.
		bridged_chain: String,
	},
	/// Account, used to sign parachains relay transactions from given bridged relay chain.
	Parachains {
		/// Account id.
		id: AccountId,
		/// Name of the bridged relay chain with parachain heads.
		bridged_chain: String,
	},
	/// Account, used to sign message relay transactions from given bridged chain.
	Messages {
		/// Account id.
		id: AccountId,
		/// Name of the bridged chain, which sends us messages or delivery confirmations.
		bridged_chain: String,
	},
}

impl<AccountId> TaggedAccount<AccountId> {
	/// Returns reference to the account id.
	pub fn id(&self) -> &AccountId {
		match *self {
			TaggedAccount::Headers { ref id, .. } => id,
			TaggedAccount::Parachains { ref id, .. } => id,
			TaggedAccount::Messages { ref id, .. } => id,
		}
	}

	/// Returns stringified account tag.
	pub fn tag(&self) -> String {
		match *self {
			TaggedAccount::Headers { ref bridged_chain, .. } => format!("{bridged_chain}Headers"),
			TaggedAccount::Parachains { ref bridged_chain, .. } => {
				format!("{bridged_chain}Parachains")
			},
			TaggedAccount::Messages { ref bridged_chain, .. } => {
				format!("{bridged_chain}Messages")
			},
		}
	}
}

/// Batch call builder.
pub trait BatchCallBuilder<Call> {
	/// Associated error type.
	type Error;
	/// If `true`, then batch calls are supported at the chain.
	const BATCH_CALL_SUPPORTED: bool;

	/// Create batch call from given calls vector.
	fn build_batch_call(_calls: Vec<Call>) -> Result<Call, Self::Error>;
}

impl<Call> BatchCallBuilder<Call> for () {
	type Error = SubstrateError;
	const BATCH_CALL_SUPPORTED: bool = false;

	fn build_batch_call(_calls: Vec<Call>) -> Result<Call, SubstrateError> {
		debug_assert!(
			false,
			"only called if `BATCH_CALL_SUPPORTED` is true;\
			`BATCH_CALL_SUPPORTED` is false;\
			qed"
		);

		Err(SubstrateError::Custom("<() as BatchCallBuilder>::build_batch_call() is called".into()))
	}
}

/// Batch call builder for bundled runtimes.
pub struct BundledBatchCallBuilder<R>(PhantomData<R>);

impl<R> BatchCallBuilder<<R as frame_system::Config>::RuntimeCall> for BundledBatchCallBuilder<R>
where
	R: pallet_utility::Config<RuntimeCall = <R as frame_system::Config>::RuntimeCall>,
	<R as frame_system::Config>::RuntimeCall: From<pallet_utility::Call<R>>,
{
	type Error = SubstrateError;
	const BATCH_CALL_SUPPORTED: bool = true;

	fn build_batch_call(
		mut calls: Vec<<R as frame_system::Config>::RuntimeCall>,
	) -> Result<<R as frame_system::Config>::RuntimeCall, SubstrateError> {
		Ok(if calls.len() == 1 {
			calls.remove(0)
		} else {
			pallet_utility::Call::batch_all { calls }.into()
		})
	}
}