fuel-core 0.48.0

use super::scalars::{
    U32,
    U64,
};
use crate::{
    database::database_description::IndexationKind,
    fuel_core_graphql_api::{
        Config as GraphQLConfig,
        query_costs,
    },
    graphql_api::{
        api_service::TxPool,
        database::{
            IndexationFlags,
            ReadDatabase,
        },
    },
};
use async_graphql::{
    Context,
    Object,
};
use std::time::UNIX_EPOCH;

pub struct NodeInfo {
    utxo_validation: bool,
    vm_backtrace: bool,
    max_tx: U64,
    max_gas: U64,
    max_size: U64,
    max_depth: U64,
    node_version: String,
    indexation: IndexationFlags,
}

#[Object]
impl NodeInfo {
    async fn utxo_validation(&self) -> bool {
        self.utxo_validation
    }

    async fn vm_backtrace(&self) -> bool {
        self.vm_backtrace
    }

    async fn max_tx(&self) -> U64 {
        self.max_tx
    }

    async fn max_gas(&self) -> U64 {
        self.max_gas
    }

    async fn max_size(&self) -> U64 {
        self.max_size
    }

    async fn max_depth(&self) -> U64 {
        self.max_depth
    }

    async fn node_version(&self) -> String {
        self.node_version.to_owned()
    }

    async fn indexation(&self) -> &IndexationFlags {
        &self.indexation
    }

    #[graphql(complexity = "query_costs().storage_read + child_complexity")]
    async fn tx_pool_stats(
        &self,
        ctx: &Context<'_>,
    ) -> async_graphql::Result<TxPoolStats> {
        let tx_pool = ctx.data_unchecked::<TxPool>();
        Ok(TxPoolStats(tx_pool.latest_pool_stats()))
    }

    #[graphql(complexity = "query_costs().get_peers + child_complexity")]
    async fn peers(&self, _ctx: &Context<'_>) -> async_graphql::Result<Vec<PeerInfo>> {
        #[cfg(feature = "p2p")]
        {
            let p2p: &crate::fuel_core_graphql_api::api_service::P2pService =
                _ctx.data_unchecked();
            let peer_info = p2p.all_peer_info().await?;
            let peers = peer_info.into_iter().map(PeerInfo).collect();
            Ok(peers)
        }
        #[cfg(not(feature = "p2p"))]
        {
            Err(async_graphql::Error::new(
                "Peering is disabled in this build, try using the `p2p` feature flag.",
            ))
        }
    }
}

#[derive(Default)]
pub struct NodeQuery {}

#[Object]
impl NodeQuery {
    #[graphql(complexity = "query_costs().storage_read + child_complexity")]
    async fn node_info(&self, ctx: &Context<'_>) -> async_graphql::Result<NodeInfo> {
        let config = ctx.data_unchecked::<GraphQLConfig>();

        const VERSION: &str = env!("CARGO_PKG_VERSION");

        let db = ctx.data_unchecked::<ReadDatabase>();
        let read_view = db.view()?;
        Ok(NodeInfo {
            utxo_validation: config.utxo_validation,
            vm_backtrace: false,
            max_tx: (config.max_tx as u64).into(),
            max_gas: config.max_gas.into(),
            max_size: (config.max_size as u64).into(),
            max_depth: (config.max_txpool_dependency_chain_length as u64).into(),
            node_version: VERSION.to_owned(),
            indexation: read_view.indexation_flags,
        })
    }
}

struct PeerInfo(fuel_core_types::services::p2p::PeerInfo);

#[Object]
impl PeerInfo {
    /// The libp2p peer id
    async fn id(&self) -> String {
        self.0.id.to_string()
    }

    /// The advertised multi-addrs that can be used to connect to this peer
    async fn addresses(&self) -> Vec<String> {
        self.0.peer_addresses.iter().cloned().collect()
    }

    /// The self-reported version of the client the peer is using
    async fn client_version(&self) -> Option<String> {
        self.0.client_version.clone()
    }

    /// The last reported height of the peer
    async fn block_height(&self) -> Option<U32> {
        self.0
            .heartbeat_data
            .block_height
            .map(|height| (*height).into())
    }

    /// The last heartbeat from this peer in unix epoch time ms
    async fn last_heartbeat_ms(&self) -> U64 {
        let time = self.0.heartbeat_data.last_heartbeat;
        let time = time
            .duration_since(UNIX_EPOCH)
            .unwrap_or_default()
            .as_millis();
        U64(time.try_into().unwrap_or_default())
    }

    /// The internal fuel p2p reputation of this peer
    async fn app_score(&self) -> f64 {
        self.0.app_score
    }
}

struct TxPoolStats(fuel_core_txpool::TxPoolStats);

#[Object]
impl TxPoolStats {
    /// The number of transactions in the pool
    async fn tx_count(&self) -> U64 {
        self.0.tx_count.into()
    }

    /// The total size of the transactions in the pool
    async fn total_size(&self) -> U64 {
        self.0.total_size.into()
    }

    /// The total gas of the transactions in the pool
    async fn total_gas(&self) -> U64 {
        self.0.total_gas.into()
    }
}

#[Object]
impl IndexationFlags {
    /// Is balances indexation enabled
    async fn balances(&self) -> bool {
        self.contains(&IndexationKind::Balances)
    }

    /// Is coins to spend indexation enabled
    async fn coins_to_spend(&self) -> bool {
        self.contains(&IndexationKind::CoinsToSpend)
    }

    /// Is asset metadata indexation enabled
    async fn asset_metadata(&self) -> bool {
        self.contains(&IndexationKind::AssetMetadata)
    }
}