// Copyright (C) Moondance Labs Ltd.

// This file is part of Tanssi.

// Tanssi 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.

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

use {

    crate::rpc::generate_rpc_builder::{GenerateRpcBuilder, GenerateRpcBuilderParams},

    cumulus_client_consensus_common::{

        ParachainBlockImport as TParachainBlockImport, ParachainBlockImportMarker,

    },

    cumulus_client_service::{

        prepare_node_config, start_relay_chain_tasks, DARecoveryProfile, ParachainHostFunctions,

        StartRelayChainTasksParams,

    },

    cumulus_primitives_core::ParaId,

    cumulus_relay_chain_interface::{

        call_remote_runtime_function, OverseerHandle, RelayChainInterface,

    },

    dancebox_runtime::{

        opaque::{Block, Hash},

        RuntimeApi,

    },

    dc_orchestrator_chain_interface::OrchestratorChainInterface,

    dp_slot_duration_runtime_api::TanssiSlotDurationApi,

    nimbus_primitives::{NimbusId, NimbusPair},

    node_common::service::{MinimalCumulusRuntimeApi, NodeBuilder, NodeBuilderConfig},

    sc_basic_authorship::ProposerFactory,

    sc_consensus::{BasicQueue, BlockImport},

    sc_executor::WasmExecutor,

    sc_network::NetworkBlock,

    sc_network_sync::SyncingService,

    sc_service::{

        Configuration, ImportQueue, SpawnTaskHandle, TFullBackend, TFullClient, TaskManager,

    },

    sc_telemetry::TelemetryHandle,

    sc_transaction_pool::FullPool,

    sp_api::ProvideRuntimeApi,

    sp_consensus::EnableProofRecording,

    sp_consensus_aura::SlotDuration,

    sp_keystore::KeystorePtr,

    std::{marker::PhantomData, sync::Arc, time::Duration},

    substrate_prometheus_endpoint::Registry,

    tc_consensus::{

        collators::lookahead::{

            self as lookahead_tanssi_aura, BuyCoreParams, Params as LookaheadTanssiAuraParams,

        },

        OrchestratorAuraWorkerAuxData,

    },

    tokio_util::sync::CancellationToken,

};

#[allow(deprecated)]

use sc_executor::NativeElseWasmExecutor;

type FullBackend = TFullBackend<Block>;

/// Native executor type.

pub struct ParachainNativeExecutor;

impl sc_executor::NativeExecutionDispatch for ParachainNativeExecutor {

    type ExtendHostFunctions = ParachainHostFunctions;

}

#[derive(Default, Copy, Clone)]

pub struct ContainerChainNodeConfig<RuntimeApi>(PhantomData<RuntimeApi>);

impl<RuntimeApi> NodeBuilderConfig for ContainerChainNodeConfig<RuntimeApi> {

    type Block = Block;

    /// RuntimeApi is customizable to allow supporting more features than the common subset of

    /// runtime api features.

    type RuntimeApi = RuntimeApi;

    type ParachainExecutor = ContainerChainExecutor;

}

impl<RuntimeApi> ContainerChainNodeConfig<RuntimeApi> {

}

/// Orchestrator Parachain Block import. We cannot use the one in cumulus as it overrides the best

/// chain selection rule

#[derive(Clone)]

pub struct OrchestratorParachainBlockImport<BI> {

    inner: BI,

}

impl<BI> OrchestratorParachainBlockImport<BI> {

    /// Create a new instance.

}

/// We simply rely on the inner

#[async_trait::async_trait]

impl<BI> BlockImport<Block> for OrchestratorParachainBlockImport<BI>

where

    BI: BlockImport<Block> + Send + Sync,

{

    type Error = BI::Error;

    async fn check_block(

        &self,

        block: sc_consensus::BlockCheckParams<Block>,

    async fn import_block(

        &mut self,

        params: sc_consensus::BlockImportParams<Block>,

}

/// But we need to implement the ParachainBlockImportMarker trait to fullfil

impl<BI> ParachainBlockImportMarker for OrchestratorParachainBlockImport<BI> {}

// Orchestrator chain types

#[allow(deprecated)]

pub type ParachainExecutor = NativeElseWasmExecutor<ParachainNativeExecutor>;

pub type ParachainClient = TFullClient<Block, RuntimeApi, ParachainExecutor>;

pub type ParachainBackend = TFullBackend<Block>;

pub type DevParachainBlockImport = OrchestratorParachainBlockImport<Arc<ParachainClient>>;

pub type ParachainBlockImport =

    TParachainBlockImport<Block, Arc<ParachainClient>, ParachainBackend>;

pub type ParachainProposerFactory =

    ProposerFactory<FullPool<Block, ParachainClient>, ParachainClient, EnableProofRecording>;

// Container chains types

type ContainerChainExecutor = WasmExecutor<ParachainHostFunctions>;

pub type ContainerChainClient<RuntimeApi> = TFullClient<Block, RuntimeApi, ContainerChainExecutor>;

pub type ContainerChainBackend = TFullBackend<Block>;

type ContainerChainBlockImport<RuntimeApi> =

    TParachainBlockImport<Block, Arc<ContainerChainClient<RuntimeApi>>, ContainerChainBackend>;

tp_traits::alias!(

    pub trait MinimalContainerRuntimeApi:

        MinimalCumulusRuntimeApi<Block, ContainerChainClient<Self>>

        + sp_api::ConstructRuntimeApi<

            Block,

            ContainerChainClient<Self>,

            RuntimeApi:

                TanssiSlotDurationApi<Block>

                + async_backing_primitives::UnincludedSegmentApi<Block>,

        >

        + Sized

);

/// Start a node with the given parachain `Configuration` and relay chain `Configuration`.

///

/// This is the actual implementation that is abstract over the executor and the runtime api.

pub async fn start_node_impl_container<

    RuntimeApi: MinimalContainerRuntimeApi,

    TGenerateRpcBuilder: GenerateRpcBuilder<RuntimeApi>,

>(

    parachain_config: Configuration,

    relay_chain_interface: Arc<dyn RelayChainInterface>,

    orchestrator_chain_interface: Arc<dyn OrchestratorChainInterface>,

    keystore: KeystorePtr,

    para_id: ParaId,

    collation_params: Option<crate::spawner::CollationParams>,

    generate_rpc_builder: TGenerateRpcBuilder,

) -> sc_service::error::Result<(

    TaskManager,

    Arc<ContainerChainClient<RuntimeApi>>,

    Arc<ParachainBackend>,

)> {

    let parachain_config = prepare_node_config(parachain_config);

    // Create a `NodeBuilder` which helps setup parachain nodes common systems.

    let node_builder = ContainerChainNodeConfig::new_builder(&parachain_config, None)?;

    let (block_import, import_queue) =

        container_chain_import_queue(&parachain_config, &node_builder);

    let import_queue_service = import_queue.service();

    log::info!("are we collators? {:?}", collation_params.is_some());

    let node_builder = node_builder

        .build_cumulus_network::<_, sc_network::NetworkWorker<_, _>>(

            &parachain_config,

            para_id,

            import_queue,

            relay_chain_interface.clone(),

        )

        .await?;

    let force_authoring = parachain_config.force_authoring;

    let prometheus_registry = parachain_config.prometheus_registry().cloned();

    let rpc_builder = generate_rpc_builder.generate(GenerateRpcBuilderParams {

        task_manager: &node_builder.task_manager,

        container_chain_config: &parachain_config,

        client: node_builder.client.clone(),

        backend: node_builder.backend.clone(),

        sync_service: node_builder.network.sync_service.clone(),

        transaction_pool: node_builder.transaction_pool.clone(),

        prometheus_registry: node_builder.prometheus_registry.clone(),

        command_sink: None,

        xcm_senders: None,

        network: node_builder.network.network.clone(),

    })?;

    let node_builder = node_builder.spawn_common_tasks(parachain_config, rpc_builder)?;

    let announce_block = {

        let sync_service = node_builder.network.sync_service.clone();

    };

    let relay_chain_slot_duration = Duration::from_secs(6);

    let overseer_handle = relay_chain_interface

        .overseer_handle()

    let (mut node_builder, _) = node_builder.extract_import_queue_service();

    start_relay_chain_tasks(StartRelayChainTasksParams {

        client: node_builder.client.clone(),

        announce_block: announce_block.clone(),

        para_id,

        relay_chain_interface: relay_chain_interface.clone(),

        task_manager: &mut node_builder.task_manager,

        da_recovery_profile: if collation_params.is_some() {

            DARecoveryProfile::Collator

        } else {

            DARecoveryProfile::FullNode

        },

        import_queue: import_queue_service,

        relay_chain_slot_duration,

        recovery_handle: Box::new(overseer_handle.clone()),

        sync_service: node_builder.network.sync_service.clone(),

    })?;

    if let Some(collation_params) = collation_params {

        let node_spawn_handle = node_builder.task_manager.spawn_handle().clone();

        let node_client = node_builder.client.clone();

        let node_backend = node_builder.backend.clone();

        start_consensus_container(

            node_client.clone(),

            node_backend.clone(),

            collation_params,

            block_import.clone(),

            prometheus_registry.clone(),

            node_spawn_handle.clone(),

            relay_chain_interface.clone(),

            orchestrator_chain_interface.clone(),

            node_builder.transaction_pool.clone(),

            node_builder.network.sync_service.clone(),

            keystore.clone(),

            force_authoring,

            relay_chain_slot_duration,

            para_id,

            overseer_handle.clone(),

            announce_block.clone(),

        );

    }

    node_builder.network.start_network.start_network();

    Ok((

        node_builder.task_manager,

        node_builder.client,

        node_builder.backend,

    ))

}

fn start_consensus_container<RuntimeApi: MinimalContainerRuntimeApi>(

    client: Arc<ContainerChainClient<RuntimeApi>>,

    backend: Arc<FullBackend>,

    collation_params: crate::spawner::CollationParams,

    block_import: ContainerChainBlockImport<RuntimeApi>,

    prometheus_registry: Option<Registry>,

    telemetry: Option<TelemetryHandle>,

    spawner: SpawnTaskHandle,

    relay_chain_interface: Arc<dyn RelayChainInterface>,

    orchestrator_chain_interface: Arc<dyn OrchestratorChainInterface>,

    transaction_pool: Arc<sc_transaction_pool::FullPool<Block, ContainerChainClient<RuntimeApi>>>,

    sync_oracle: Arc<SyncingService<Block>>,

    keystore: KeystorePtr,

    force_authoring: bool,

    relay_chain_slot_duration: Duration,

    para_id: ParaId,

    overseer_handle: OverseerHandle,

    announce_block: Arc<dyn Fn(Hash, Option<Vec<u8>>) + Send + Sync>,

) {

    let crate::spawner::CollationParams {

        collator_key,

        orchestrator_tx_pool,

        orchestrator_client,

        orchestrator_para_id,

        solochain,

    } = collation_params;

    let slot_duration = if solochain {

        // Solochains use Babe instead of Aura, which has 6s slot duration

        let relay_slot_ms = relay_chain_slot_duration.as_millis();

        SlotDuration::from_millis(

            u64::try_from(relay_slot_ms).expect("relay chain slot duration overflows u64"),

        )

    } else {

        cumulus_client_consensus_aura::slot_duration(

            orchestrator_client

                .as_deref()

                .expect("solochain is false, orchestrator_client must be Some"),

        )

        .expect("start_consensus_container: slot duration should exist")

    };

    let proposer_factory = sc_basic_authorship::ProposerFactory::with_proof_recording(

        spawner.clone(),

        client.clone(),

        transaction_pool,

        prometheus_registry.as_ref(),

        telemetry.clone(),

    );

    let proposer = cumulus_client_consensus_proposer::Proposer::new(proposer_factory);

    let collator_service = cumulus_client_collator::service::CollatorService::new(

        client.clone(),

        Arc::new(spawner.clone()),

        announce_block,

        client.clone(),

    );

    let relay_chain_interace_for_cidp = relay_chain_interface.clone();

    let relay_chain_interace_for_orch = relay_chain_interface.clone();

    let orchestrator_client_for_cidp = orchestrator_client.clone();

    let client_for_cidp = client.clone();

    let client_for_hash_provider = client.clone();

    let client_for_slot_duration = client.clone();

    let buy_core_params = if solochain {

        BuyCoreParams::Solochain {}

    } else {

        BuyCoreParams::Orchestrator {

            orchestrator_tx_pool: orchestrator_tx_pool

                .expect("solochain is false, orchestrator_tx_pool must be Some"),

            orchestrator_client: orchestrator_client

                .expect("solochain is false, orchestrator_client must be Some"),

        }

    };

    let params = LookaheadTanssiAuraParams {

                } else {

                };

                        authorities,

                        relay_parent

                    );

                } else {

                        authorities,

                        latest_header

                    );

                }

        block_import,

        para_client: client,

        relay_client: relay_chain_interface,

        sync_oracle,

        keystore,

        collator_key,

        para_id,

        overseer_handle,

        orchestrator_slot_duration: slot_duration,

        force_authoring,

        relay_chain_slot_duration,

        proposer,

        collator_service,

        authoring_duration: Duration::from_millis(2000),

        para_backend: backend,

        code_hash_provider,

        // This cancellation token is no-op as it is not shared outside.

        cancellation_token: CancellationToken::new(),

        buy_core_params,

    };

    let (fut, _exit_notification_receiver) =

        lookahead_tanssi_aura::run::<_, Block, NimbusPair, _, _, _, _, _, _, _, _, _, _, _, _, _>(

            params,

        );

    spawner.spawn("tanssi-aura-container", None, fut);

}

// Log string that will be shown for the container chain: `[Container-2000]`.

// This needs to be a separate function because the `prefix_logs_with` macro

// has trouble parsing expressions.