// Copyright (C) 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/>.

//! A module that is responsible for migration of storage.

use crate::{Config, Pallet, Receiver, LOG_TARGET};

use frame_support::{

	traits::{Get, OnRuntimeUpgrade},

	weights::Weight,

};

use xcm::prelude::{InteriorLocation, Location};

/// This migration does not modify storage but can be used to open a bridge and link it to the

/// specified LaneId. This is useful when we want to open a bridge and use a custom LaneId instead

/// of the pre-calculated one provided by the `fn open_bridge extrinsic`.

/// Or perhaps if you want to ensure that your runtime (e.g., for testing) always has an open

/// bridge.

pub struct OpenBridgeForLane<

	T,

	I,

	Lane,

	CreateLane,

	SourceRelativeLocation,

	BridgedUniversalLocation,

	MaybeNotifyRelativeLocation,

>(

	core::marker::PhantomData<(

		T,

		I,

		Lane,

		CreateLane,

		SourceRelativeLocation,

		BridgedUniversalLocation,

		MaybeNotifyRelativeLocation,

	)>,

);

impl<

		T: Config<I>,

		I: 'static,

		Lane: Get<T::LaneId>,

		CreateLane: Get<bool>,

		SourceRelativeLocation: Get<Location>,

		BridgedUniversalLocation: Get<InteriorLocation>,

		MaybeNotifyRelativeLocation: Get<Option<Receiver>>,

	> OnRuntimeUpgrade

	for OpenBridgeForLane<

		T,

		I,

		Lane,

		CreateLane,

		SourceRelativeLocation,

		BridgedUniversalLocation,

		MaybeNotifyRelativeLocation,

	>

{

		);

				);

			}

		};

		// check if already exists

			);

				);

		{

		} else {

		}

	#[cfg(feature = "try-runtime")]

	fn post_upgrade(_state: sp_std::vec::Vec<u8>) -> Result<(), sp_runtime::DispatchError> {

		let bridge_origin_relative_location = SourceRelativeLocation::get();

		let bridge_destination_universal_location = BridgedUniversalLocation::get();

		let lane_id = Lane::get();

		// check that requested bridge is stored

		let Ok(locations) = Pallet::<T, I>::bridge_locations(

			bridge_origin_relative_location.clone(),

			bridge_destination_universal_location.clone(),

		) else {

			return Err(sp_runtime::DispatchError::Other("Invalid locations!"));

		};

		let Some((bridge_id, _)) = Pallet::<T, I>::bridge_by_lane_id(&lane_id) else {

			return Err(sp_runtime::DispatchError::Other("Missing bridge!"));

		};

		frame_support::ensure!(

			locations.bridge_id() == &bridge_id,

			"Bridge is not stored correctly!"

		);

		log::info!(

			target: LOG_TARGET,

			"OpenBridgeForLane - post_upgrade found opened bridge with lane_id: {lane_id:?} \

			between bridge_origin_relative_location: {bridge_origin_relative_location:?} and \

			bridge_destination_universal_location: {bridge_destination_universal_location:?}",

		);

		Ok(())

	}

}