linera-service 0.15.21

// Copyright (c) Zefchain Labs, Inc.
// SPDX-License-Identifier: Apache-2.0

use std::{
    borrow::Cow,
    future::IntoFuture,
    iter,
    net::SocketAddr,
    num::NonZeroU16,
    sync::{Arc, Mutex as StdMutex},
};

use async_graphql::{
    futures_util::Stream,
    registry::{MetaType, MetaTypeId, Registry},
    resolver_utils::ContainerType,
    EmptyMutation, Error, MergedObject, OutputType, Positioned, Request, Response, ScalarType,
    Schema, SimpleObject, Subscription,
};
use async_graphql_axum::{GraphQLRequest, GraphQLResponse, GraphQLSubscription};
use axum::{extract::Path, http::StatusCode, response, response::IntoResponse, Extension, Router};
use futures::{lock::Mutex, Future, FutureExt as _, StreamExt as _, TryStreamExt as _};
use linera_base::{
    crypto::{CryptoError, CryptoHash},
    data_types::{
        Amount, ApplicationDescription, ApplicationPermissions, BlockHeight, Bytecode, Epoch,
        TimeDelta,
    },
    identifiers::{
        Account, AccountOwner, ApplicationId, ChainId, IndexAndEvent, ModuleId, StreamId,
    },
    ownership::{ChainOwnership, TimeoutConfig},
    vm::VmRuntime,
    BcsHexParseError,
};
use linera_chain::{
    types::{ConfirmedBlock, GenericCertificate},
    ChainStateView,
};
use linera_client::chain_listener::{
    ChainListener, ChainListenerConfig, ClientContext, ListenerCommand,
};
use linera_core::{
    client::{chain_client, ChainClient},
    data_types::ClientOutcome,
    wallet::Wallet as _,
    worker::{ChainStateViewReadGuard, Notification, Reason},
};
use linera_execution::{
    committee::Committee, system::AdminOperation, Operation, Query, QueryOutcome, QueryResponse,
    SystemOperation,
};
#[cfg(with_metrics)]
use linera_metrics::monitoring_server;
use linera_sdk::linera_base_types::BlobContent;
use linera_storage::Storage;
use lru::LruCache;
use serde::{Deserialize, Serialize};
use serde_json::json;
use tokio::sync::mpsc::UnboundedReceiver;
use tokio_util::sync::CancellationToken;
use tower_http::cors::CorsLayer;
use tracing::{debug, error, info, instrument, trace};

use crate::util;

/// A pre-serialized JSON string that implements [`OutputType`] as the `JSON` scalar.
///
/// When the `raw_value` feature of `async-graphql` is enabled, the string is
/// emitted directly into the GraphQL response without any parsing or
/// intermediate tree construction.
#[derive(Clone)]
struct RawJson(String);

impl OutputType for RawJson {
    fn type_name() -> Cow<'static, str> {
        Cow::Borrowed("JSON")
    }

    fn create_type_info(registry: &mut Registry) -> String {
        registry.create_output_type::<Self, _>(MetaTypeId::Scalar, |_| MetaType::Scalar {
            name: "JSON".to_string(),
            description: Some("A scalar that can represent any JSON value.".to_string()),
            is_valid: None,
            visible: None,
            inaccessible: false,
            tags: Default::default(),
            specified_by_url: None,
            directive_invocations: Default::default(),
            requires_scopes: Default::default(),
        })
    }

    async fn resolve(
        &self,
        _ctx: &async_graphql::ContextSelectionSet<'_>,
        _field: &Positioned<async_graphql::parser::types::Field>,
    ) -> async_graphql::ServerResult<async_graphql::Value> {
        // Wrap the raw JSON string with the magic token that async-graphql's
        // ConstValue serializer recognises (with feature `raw_value`).
        // When the response is serialised to JSON the raw string is emitted
        // verbatim, avoiding any parsing or tree conversion.
        //
        Ok(async_graphql::Value::Object(
            std::iter::once((
                async_graphql::Name::new(async_graphql_value::RAW_VALUE_TOKEN),
                async_graphql::Value::String(self.0.clone()),
            ))
            .collect(),
        ))
    }
}

/// The set of chains tracked by the wallet.
#[derive(SimpleObject, Serialize, Deserialize, Clone)]
pub struct Chains {
    /// The IDs of the tracked chains.
    pub list: Vec<ChainId>,
    /// The default chain of the wallet, if one is set.
    pub default: Option<ChainId>,
}

/// Our root GraphQL query type.
pub struct QueryRoot<C> {
    context: Arc<Mutex<C>>,
    port: NonZeroU16,
    default_chain: Option<ChainId>,
}

/// Our root GraphQL subscription type.
pub struct SubscriptionRoot<C> {
    context: Arc<Mutex<C>>,
    query_subscriptions: Option<Arc<crate::query_subscription::QuerySubscriptionManager>>,
    cancellation_token: CancellationToken,
}

/// Our root GraphQL mutation type.
pub struct MutationRoot<C> {
    context: Arc<Mutex<C>>,
}

#[derive(Debug, thiserror::Error)]
enum NodeServiceError {
    #[error(transparent)]
    ChainClient(#[from] chain_client::Error),
    #[error(transparent)]
    BcsHex(#[from] BcsHexParseError),
    #[error(transparent)]
    Json(#[from] serde_json::Error),
    #[error("malformed chain ID: {0}")]
    InvalidChainId(CryptoError),
    #[error(transparent)]
    Client(#[from] linera_client::Error),
    #[error("scheduling operations from queries is disabled in read-only mode")]
    ReadOnlyModeOperationsNotAllowed,
}

impl IntoResponse for NodeServiceError {
    fn into_response(self) -> response::Response {
        let status = match self {
            NodeServiceError::InvalidChainId(_) | NodeServiceError::BcsHex(_) => {
                StatusCode::BAD_REQUEST
            }
            NodeServiceError::ReadOnlyModeOperationsNotAllowed => StatusCode::FORBIDDEN,
            _ => StatusCode::INTERNAL_SERVER_ERROR,
        };
        let body = json!({"error": self.to_string()}).to_string();
        (status, body).into_response()
    }
}

#[Subscription]
impl<C> SubscriptionRoot<C>
where
    C: ClientContext + 'static,
{
    /// Subscribes to notifications from the specified chain.
    async fn notifications(
        &self,
        chain_id: ChainId,
    ) -> Result<impl Stream<Item = Notification>, Error> {
        let client = self
            .context
            .lock()
            .await
            .make_chain_client(chain_id)
            .await?;
        Ok(client.subscribe()?)
    }

    /// Subscribes to the result of a pre-registered GraphQL query.
    /// Re-executes the query on every new block and pushes changed results.
    async fn query_result(
        &self,
        #[graphql(desc = "Name of the registered subscription query.")] name: String,
        #[graphql(desc = "The chain to watch.")] chain_id: ChainId,
        #[graphql(desc = "The application to query.")] application_id: ApplicationId,
    ) -> Result<impl Stream<Item = RawJson>, Error> {
        let manager = self
            .query_subscriptions
            .as_ref()
            .ok_or_else(|| Error::new("no subscription queries registered"))?;

        let key = crate::query_subscription::SubscriptionKey {
            name,
            chain_id,
            application_id,
        };

        let receiver = manager
            .subscribe(
                &key,
                Arc::clone(&self.context),
                self.cancellation_token.clone(),
            )
            .map_err(|e| Error::new(e.to_string()))?;

        // `sender.subscribe()` marks the current value as "already seen", so
        // `WatchStream` would skip it and wait for the next change.  Grab the
        // current snapshot first and prepend it to the stream so that every new
        // subscriber gets the latest cached result immediately.
        let current = receiver.borrow().clone();
        let changes = tokio_stream::wrappers::WatchStream::from_changes(receiver)
            .filter_map(|value| async move { value });
        Ok(futures::stream::iter(current).chain(changes).map(RawJson))
    }
}

impl<C> MutationRoot<C>
where
    C: ClientContext,
{
    async fn execute_system_operation(
        &self,
        system_operation: SystemOperation,
        chain_id: ChainId,
    ) -> Result<CryptoHash, Error> {
        let certificate = self
            .apply_client_command(&chain_id, move |client| {
                let operation = Operation::system(system_operation.clone());
                async move {
                    let result = client
                        .execute_operation(operation)
                        .await
                        .map_err(Error::from);
                    (result, client)
                }
            })
            .await?;
        Ok(certificate.hash())
    }

    /// Applies the given function to the chain client.
    /// Updates the wallet regardless of the outcome. As long as the function returns a round
    /// timeout, it will wait and retry.
    async fn apply_client_command<F, Fut, T>(
        &self,
        chain_id: &ChainId,
        mut f: F,
    ) -> Result<T, Error>
    where
        F: FnMut(ChainClient<C::Environment>) -> Fut,
        Fut: Future<Output = (Result<ClientOutcome<T>, Error>, ChainClient<C::Environment>)>,
    {
        loop {
            let client = self
                .context
                .lock()
                .await
                .make_chain_client(*chain_id)
                .await?;
            let mut stream = client.subscribe()?;
            let (result, client) = f(client).await;
            self.context.lock().await.update_wallet(&client).await?;
            let timeout = match result? {
                ClientOutcome::Committed(t) => return Ok(t),
                ClientOutcome::Conflict(certificate) => {
                    return Err(chain_client::Error::Conflict(certificate.hash()).into());
                }
                ClientOutcome::WaitForTimeout(timeout) => timeout,
            };
            drop(client);
            util::wait_for_next_round(&mut stream, timeout).await;
        }
    }
}

#[async_graphql::Object(cache_control(no_cache))]
impl<C> MutationRoot<C>
where
    C: ClientContext + 'static,
{
    /// Processes the inbox and returns the lists of certificate hashes that were created, if any.
    async fn process_inbox(&self, chain_id: ChainId) -> Result<Vec<CryptoHash>, Error> {
        let mut hashes = Vec::new();
        loop {
            let client = self
                .context
                .lock()
                .await
                .make_chain_client(chain_id)
                .await?;
            let result = client.process_inbox().await;
            self.context.lock().await.update_wallet(&client).await?;
            let (certificates, maybe_timeout) = result?;
            hashes.extend(certificates.into_iter().map(|cert| cert.hash()));
            match maybe_timeout {
                None => return Ok(hashes),
                Some(timestamp) => {
                    let mut stream = client.subscribe()?;
                    drop(client);
                    util::wait_for_next_round(&mut stream, timestamp).await;
                }
            }
        }
    }

    /// Synchronizes the chain with the validators. Returns the chain's length.
    ///
    /// This is only used for testing, to make sure that a client is up to date.
    // TODO(#4718): Remove this mutation.
    async fn sync(
        &self,
        #[graphql(desc = "The chain being synchronized.")] chain_id: ChainId,
    ) -> Result<u64, Error> {
        let client = self
            .context
            .lock()
            .await
            .make_chain_client(chain_id)
            .await?;
        let info = client.synchronize_from_validators().await?;
        self.context.lock().await.update_wallet(&client).await?;
        Ok(info.next_block_height.0)
    }

    /// Retries the pending block that was unsuccessfully proposed earlier.
    async fn retry_pending_block(
        &self,
        #[graphql(desc = "The chain on whose block is being retried.")] chain_id: ChainId,
    ) -> Result<Option<CryptoHash>, Error> {
        let client = self
            .context
            .lock()
            .await
            .make_chain_client(chain_id)
            .await?;
        let outcome = client.process_pending_block().await?;
        self.context.lock().await.update_wallet(&client).await?;
        match outcome {
            ClientOutcome::Committed(Some(certificate)) => Ok(Some(certificate.hash())),
            ClientOutcome::Committed(None) => Ok(None),
            ClientOutcome::WaitForTimeout(timeout) => Err(Error::from(format!(
                "Please try again at {}",
                timeout.timestamp
            ))),
            ClientOutcome::Conflict(certificate) => Err(Error::from(format!(
                "A different block was committed: {}",
                certificate.hash()
            ))),
        }
    }

    /// Transfers `amount` units of value from the given owner's account to the recipient.
    /// If no owner is given, try to take the units out of the chain account.
    async fn transfer(
        &self,
        chain_id: ChainId,
        owner: AccountOwner,
        recipient: Account,
        amount: Amount,
    ) -> Result<CryptoHash, Error> {
        self.apply_client_command(&chain_id, move |client| async move {
            let result = client
                .transfer(owner, amount, recipient)
                .await
                .map_err(Error::from)
                .map(|outcome| outcome.map(|certificate| certificate.hash()));
            (result, client)
        })
        .await
    }

    /// Claims `amount` units of value from the given owner's account in the remote
    /// `target` chain. Depending on its configuration, the `target` chain may refuse to
    /// process the message.
    async fn claim(
        &self,
        chain_id: ChainId,
        owner: AccountOwner,
        target_id: ChainId,
        recipient: Account,
        amount: Amount,
    ) -> Result<CryptoHash, Error> {
        self.apply_client_command(&chain_id, move |client| async move {
            let result = client
                .claim(owner, target_id, recipient, amount)
                .await
                .map_err(Error::from)
                .map(|outcome| outcome.map(|certificate| certificate.hash()));
            (result, client)
        })
        .await
    }

    /// Test if a data blob is readable from a transaction in the current chain.
    // TODO(#2490): Consider removing or renaming this.
    async fn read_data_blob(
        &self,
        chain_id: ChainId,
        hash: CryptoHash,
    ) -> Result<CryptoHash, Error> {
        self.apply_client_command(&chain_id, move |client| async move {
            let result = client
                .read_data_blob(hash)
                .await
                .map_err(Error::from)
                .map(|outcome| outcome.map(|certificate| certificate.hash()));
            (result, client)
        })
        .await
    }

    /// Creates (or activates) a new chain with the given owner.
    /// This will automatically subscribe to the future committees created by `admin_chain_id`.
    async fn open_chain(
        &self,
        chain_id: ChainId,
        owner: AccountOwner,
        balance: Option<Amount>,
    ) -> Result<ChainId, Error> {
        let ownership = ChainOwnership::single(owner);
        let balance = balance.unwrap_or(Amount::ZERO);
        let description = self
            .apply_client_command(&chain_id, move |client| {
                let ownership = ownership.clone();
                async move {
                    let result = client
                        .open_chain(ownership, ApplicationPermissions::default(), balance)
                        .await
                        .map_err(Error::from)
                        .map(|outcome| outcome.map(|(chain_id, _)| chain_id));
                    (result, client)
                }
            })
            .await?;
        Ok(description.id())
    }

    /// Creates (or activates) a new chain by installing the given authentication keys.
    /// This will automatically subscribe to the future committees created by `admin_chain_id`.
    #[expect(clippy::too_many_arguments)]
    async fn open_multi_owner_chain(
        &self,
        chain_id: ChainId,
        application_permissions: Option<ApplicationPermissions>,
        owners: Vec<AccountOwner>,
        weights: Option<Vec<u64>>,
        multi_leader_rounds: Option<u32>,
        balance: Option<Amount>,
        #[graphql(desc = "The duration of the fast round, in milliseconds; default: no timeout")]
        fast_round_ms: Option<u64>,
        #[graphql(
            desc = "The duration of the first single-leader and all multi-leader rounds",
            default = 10_000
        )]
        base_timeout_ms: u64,
        #[graphql(
            desc = "The number of milliseconds by which the timeout increases after each \
                    single-leader round",
            default = 1_000
        )]
        timeout_increment_ms: u64,
        #[graphql(
            desc = "The age of an incoming tracked or protected message after which the \
                    validators start transitioning the chain to fallback mode, in milliseconds.",
            default = 86_400_000
        )]
        fallback_duration_ms: u64,
    ) -> Result<ChainId, Error> {
        let owners = if let Some(weights) = weights {
            if weights.len() != owners.len() {
                return Err(Error::new(format!(
                    "There are {} owners but {} weights.",
                    owners.len(),
                    weights.len()
                )));
            }
            owners.into_iter().zip(weights).collect::<Vec<_>>()
        } else {
            owners
                .into_iter()
                .zip(iter::repeat(100))
                .collect::<Vec<_>>()
        };
        let multi_leader_rounds = multi_leader_rounds.unwrap_or(u32::MAX);
        let timeout_config = TimeoutConfig {
            fast_round_duration: fast_round_ms.map(TimeDelta::from_millis),
            base_timeout: TimeDelta::from_millis(base_timeout_ms),
            timeout_increment: TimeDelta::from_millis(timeout_increment_ms),
            fallback_duration: TimeDelta::from_millis(fallback_duration_ms),
        };
        let ownership = ChainOwnership::multiple(owners, multi_leader_rounds, timeout_config);
        let balance = balance.unwrap_or(Amount::ZERO);
        let description = self
            .apply_client_command(&chain_id, move |client| {
                let ownership = ownership.clone();
                let application_permissions = application_permissions.clone().unwrap_or_default();
                async move {
                    let result = client
                        .open_chain(ownership, application_permissions, balance)
                        .await
                        .map_err(Error::from)
                        .map(|outcome| outcome.map(|(chain_id, _)| chain_id));
                    (result, client)
                }
            })
            .await?;
        Ok(description.id())
    }

    /// Closes the chain. Returns `None` if it was already closed.
    async fn close_chain(&self, chain_id: ChainId) -> Result<Option<CryptoHash>, Error> {
        let maybe_cert = self
            .apply_client_command(&chain_id, |client| async move {
                let result = client.close_chain().await.map_err(Error::from);
                (result, client)
            })
            .await?;
        Ok(maybe_cert.as_ref().map(GenericCertificate::hash))
    }

    /// Changes the authentication key of the chain.
    async fn change_owner(
        &self,
        chain_id: ChainId,
        new_owner: AccountOwner,
    ) -> Result<CryptoHash, Error> {
        let operation = SystemOperation::ChangeOwnership {
            super_owners: vec![new_owner],
            owners: Vec::new(),
            multi_leader_rounds: 5,
            open_multi_leader_rounds: false,
            timeout_config: TimeoutConfig::default(),
        };
        self.execute_system_operation(operation, chain_id).await
    }

    /// Changes the authentication key of the chain.
    #[expect(clippy::too_many_arguments)]
    async fn change_multiple_owners(
        &self,
        chain_id: ChainId,
        new_owners: Vec<AccountOwner>,
        new_weights: Vec<u64>,
        multi_leader_rounds: u32,
        open_multi_leader_rounds: bool,
        #[graphql(desc = "The duration of the fast round, in milliseconds; default: no timeout")]
        fast_round_ms: Option<u64>,
        #[graphql(
            desc = "The duration of the first single-leader and all multi-leader rounds",
            default = 10_000
        )]
        base_timeout_ms: u64,
        #[graphql(
            desc = "The number of milliseconds by which the timeout increases after each \
                    single-leader round",
            default = 1_000
        )]
        timeout_increment_ms: u64,
        #[graphql(
            desc = "The age of an incoming tracked or protected message after which the \
                    validators start transitioning the chain to fallback mode, in milliseconds.",
            default = 86_400_000
        )]
        fallback_duration_ms: u64,
    ) -> Result<CryptoHash, Error> {
        let operation = SystemOperation::ChangeOwnership {
            super_owners: Vec::new(),
            owners: new_owners.into_iter().zip(new_weights).collect(),
            multi_leader_rounds,
            open_multi_leader_rounds,
            timeout_config: TimeoutConfig {
                fast_round_duration: fast_round_ms.map(TimeDelta::from_millis),
                base_timeout: TimeDelta::from_millis(base_timeout_ms),
                timeout_increment: TimeDelta::from_millis(timeout_increment_ms),
                fallback_duration: TimeDelta::from_millis(fallback_duration_ms),
            },
        };
        self.execute_system_operation(operation, chain_id).await
    }

    /// Changes the application permissions configuration on this chain.
    #[expect(clippy::too_many_arguments)]
    async fn change_application_permissions(
        &self,
        chain_id: ChainId,
        close_chain: Vec<ApplicationId>,
        execute_operations: Option<Vec<ApplicationId>>,
        mandatory_applications: Vec<ApplicationId>,
        change_application_permissions: Vec<ApplicationId>,
        call_service_as_oracle: Option<Vec<ApplicationId>>,
        make_http_requests: Option<Vec<ApplicationId>>,
    ) -> Result<CryptoHash, Error> {
        let operation = SystemOperation::ChangeApplicationPermissions(ApplicationPermissions {
            execute_operations,
            mandatory_applications,
            close_chain,
            change_application_permissions,
            call_service_as_oracle,
            make_http_requests,
        });
        self.execute_system_operation(operation, chain_id).await
    }

    /// (admin chain only) Registers a new committee. This will notify the subscribers of
    /// the admin chain so that they can migrate to the new epoch (by accepting the
    /// notification as an "incoming message" in a next block).
    async fn create_committee(
        &self,
        chain_id: ChainId,
        committee: Committee,
    ) -> Result<CryptoHash, Error> {
        Ok(self
            .apply_client_command(&chain_id, move |client| {
                let committee = committee.clone();
                async move {
                    let result = client
                        .stage_new_committee(committee)
                        .await
                        .map_err(Error::from);
                    (result, client)
                }
            })
            .await?
            .hash())
    }

    /// (admin chain only) Removes a committee. Once this message is accepted by a chain,
    /// blocks from the retired epoch will not be accepted until they are followed (hence
    /// re-certified) by a block certified by a recent committee.
    async fn remove_committee(&self, chain_id: ChainId, epoch: Epoch) -> Result<CryptoHash, Error> {
        let operation = SystemOperation::Admin(AdminOperation::RemoveCommittee { epoch });
        self.execute_system_operation(operation, chain_id).await
    }

    /// Publishes a new application module.
    async fn publish_module(
        &self,
        chain_id: ChainId,
        contract: Bytecode,
        service: Bytecode,
        vm_runtime: VmRuntime,
    ) -> Result<ModuleId, Error> {
        self.apply_client_command(&chain_id, move |client| {
            let contract = contract.clone();
            let service = service.clone();
            async move {
                let result = client
                    .publish_module(contract, service, vm_runtime)
                    .await
                    .map_err(Error::from)
                    .map(|outcome| outcome.map(|(module_id, _)| module_id));
                (result, client)
            }
        })
        .await
    }

    /// Publishes a new data blob.
    async fn publish_data_blob(
        &self,
        chain_id: ChainId,
        bytes: Vec<u8>,
    ) -> Result<CryptoHash, Error> {
        self.apply_client_command(&chain_id, |client| {
            let bytes = bytes.clone();
            async move {
                let result = client.publish_data_blob(bytes).await.map_err(Error::from);
                (result, client)
            }
        })
        .await
        .map(|_| CryptoHash::new(&BlobContent::new_data(bytes)))
    }

    /// Creates a new application.
    async fn create_application(
        &self,
        chain_id: ChainId,
        module_id: ModuleId,
        parameters: String,
        instantiation_argument: String,
        required_application_ids: Vec<ApplicationId>,
    ) -> Result<ApplicationId, Error> {
        self.apply_client_command(&chain_id, move |client| {
            let parameters = parameters.as_bytes().to_vec();
            let instantiation_argument = instantiation_argument.as_bytes().to_vec();
            let required_application_ids = required_application_ids.clone();
            async move {
                let result = client
                    .create_application_untyped(
                        module_id,
                        parameters,
                        instantiation_argument,
                        required_application_ids,
                    )
                    .await
                    .map_err(Error::from)
                    .map(|outcome| outcome.map(|(application_id, _)| application_id));
                (result, client)
            }
        })
        .await
    }
}

#[async_graphql::Object(cache_control(no_cache))]
impl<C> QueryRoot<C>
where
    C: ClientContext + 'static,
{
    async fn chain(
        &self,
        chain_id: ChainId,
    ) -> Result<ChainStateExtendedView<<C::Environment as linera_core::Environment>::Storage>, Error>
    {
        let client = self
            .context
            .lock()
            .await
            .make_chain_client(chain_id)
            .await?;
        let view = client.chain_state_view().await?;
        Ok(ChainStateExtendedView::new(view))
    }

    async fn applications(&self, chain_id: ChainId) -> Result<Vec<ApplicationOverview>, Error> {
        let client = self
            .context
            .lock()
            .await
            .make_chain_client(chain_id)
            .await?;
        let applications = client
            .chain_state_view()
            .await?
            .execution_state
            .list_applications()
            .await?;

        let overviews = applications
            .into_iter()
            .map(|(id, description)| ApplicationOverview::new(id, description, self.port, chain_id))
            .collect();

        Ok(overviews)
    }

    async fn chains(&self) -> Result<Chains, Error> {
        Ok(Chains {
            list: self
                .context
                .lock()
                .await
                .wallet()
                .chain_ids()
                .try_collect()
                .await?,
            default: self.default_chain,
        })
    }

    async fn block(
        &self,
        hash: Option<CryptoHash>,
        chain_id: ChainId,
    ) -> Result<Option<Arc<ConfirmedBlock>>, Error> {
        let client = self
            .context
            .lock()
            .await
            .make_chain_client(chain_id)
            .await?;
        let hash = match hash {
            Some(hash) => Some(hash),
            None => client.chain_info().await?.block_hash,
        };
        if let Some(hash) = hash {
            Ok(Some(client.read_confirmed_block(hash).await?))
        } else {
            Ok(None)
        }
    }

    async fn events_from_index(
        &self,
        chain_id: ChainId,
        stream_id: StreamId,
        start_index: u32,
    ) -> Result<Vec<IndexAndEvent>, Error> {
        Ok(self
            .context
            .lock()
            .await
            .make_chain_client(chain_id)
            .await?
            .events_from_index(stream_id, start_index)
            .await?)
    }

    async fn blocks(
        &self,
        from: Option<CryptoHash>,
        chain_id: ChainId,
        limit: Option<u32>,
    ) -> Result<Vec<Arc<ConfirmedBlock>>, Error> {
        let client = self
            .context
            .lock()
            .await
            .make_chain_client(chain_id)
            .await?;
        let limit = limit.unwrap_or(10);
        let from = match from {
            Some(from) => Some(from),
            None => client.chain_info().await?.block_hash,
        };
        let Some(from) = from else {
            return Ok(vec![]);
        };
        let mut hash = Some(from);
        let mut values = Vec::new();
        for _ in 0..limit {
            let Some(next_hash) = hash else {
                break;
            };
            let value = client.read_confirmed_block(next_hash).await?;
            hash = value.block().header.previous_block_hash;
            values.push(value);
        }
        Ok(values)
    }

    /// Returns the version information on this node service.
    async fn version(&self) -> linera_version::VersionInfo {
        linera_version::VersionInfo::default()
    }
}

// What follows is a hack to add a chain_id field to `ChainStateView` based on
// https://async-graphql.github.io/async-graphql/en/merging_objects.html

struct ChainStateViewExtension(ChainId);

#[async_graphql::Object(cache_control(no_cache))]
impl ChainStateViewExtension {
    async fn chain_id(&self) -> ChainId {
        self.0
    }
}

#[derive(MergedObject)]
struct ChainStateExtendedView<S: Storage>(ChainStateViewExtension, ReadOnlyChainStateView<S>)
where
    ChainStateView<S::Context>: ContainerType + OutputType;

/// A wrapper type that allows proxying GraphQL queries to a [`ChainStateView`] that's behind a
/// [`ChainStateViewReadGuard`].
pub struct ReadOnlyChainStateView<S: Storage>(ChainStateViewReadGuard<S>)
where
    ChainStateView<S::Context>: ContainerType + OutputType;

impl<S: Storage> ContainerType for ReadOnlyChainStateView<S>
where
    ChainStateView<S::Context>: ContainerType + OutputType,
{
    async fn resolve_field(
        &self,
        context: &async_graphql::Context<'_>,
    ) -> async_graphql::ServerResult<Option<async_graphql::Value>> {
        self.0.resolve_field(context).await
    }
}

impl<S: Storage> OutputType for ReadOnlyChainStateView<S>
where
    ChainStateView<S::Context>: ContainerType + OutputType,
{
    fn type_name() -> Cow<'static, str> {
        ChainStateView::<S::Context>::type_name()
    }

    fn create_type_info(registry: &mut async_graphql::registry::Registry) -> String {
        ChainStateView::<S::Context>::create_type_info(registry)
    }

    async fn resolve(
        &self,
        context: &async_graphql::ContextSelectionSet<'_>,
        field: &async_graphql::Positioned<async_graphql::parser::types::Field>,
    ) -> async_graphql::ServerResult<async_graphql::Value> {
        self.0.resolve(context, field).await
    }
}

impl<S: Storage> ChainStateExtendedView<S>
where
    ChainStateView<S::Context>: ContainerType + OutputType,
{
    fn new(view: ChainStateViewReadGuard<S>) -> Self {
        Self(
            ChainStateViewExtension(view.chain_id()),
            ReadOnlyChainStateView(view),
        )
    }
}

/// A summary of an application registered on a chain.
#[derive(SimpleObject)]
pub struct ApplicationOverview {
    id: ApplicationId,
    description: ApplicationDescription,
    link: String,
}

impl ApplicationOverview {
    fn new(
        id: ApplicationId,
        description: ApplicationDescription,
        port: NonZeroU16,
        chain_id: ChainId,
    ) -> Self {
        Self {
            id,
            description,
            link: format!(
                "http://localhost:{}/chains/{}/applications/{}",
                port.get(),
                chain_id,
                id
            ),
        }
    }
}

/// Schema type that can be either full (with mutations) or read-only.
pub enum NodeServiceSchema<C>
where
    C: ClientContext + 'static,
{
    /// Full schema with mutations enabled.
    Full(Schema<QueryRoot<C>, MutationRoot<C>, SubscriptionRoot<C>>),
    /// Read-only schema with mutations disabled.
    ReadOnly(Schema<QueryRoot<C>, EmptyMutation, SubscriptionRoot<C>>),
}

impl<C> NodeServiceSchema<C>
where
    C: ClientContext,
{
    /// Executes a GraphQL request.
    pub async fn execute(&self, request: impl Into<Request>) -> Response {
        match self {
            Self::Full(schema) => schema.execute(request).await,
            Self::ReadOnly(schema) => schema.execute(request).await,
        }
    }

    /// Returns the SDL (Schema Definition Language) representation.
    pub fn sdl(&self) -> String {
        match self {
            Self::Full(schema) => schema.sdl(),
            Self::ReadOnly(schema) => schema.sdl(),
        }
    }
}

impl<C> Clone for NodeServiceSchema<C>
where
    C: ClientContext,
{
    fn clone(&self) -> Self {
        match self {
            Self::Full(schema) => Self::Full(schema.clone()),
            Self::ReadOnly(schema) => Self::ReadOnly(schema.clone()),
        }
    }
}

#[cfg(with_metrics)]
mod query_cache_metrics {
    use std::sync::LazyLock;

    use linera_base::prometheus_util::{register_int_counter_vec, register_int_gauge};
    use prometheus::{IntCounterVec, IntGauge};

    pub static QUERY_CACHE_HIT: LazyLock<IntCounterVec> = LazyLock::new(|| {
        register_int_counter_vec("query_response_cache_hit", "Query response cache hits", &[])
    });

    pub static QUERY_CACHE_MISS: LazyLock<IntCounterVec> = LazyLock::new(|| {
        register_int_counter_vec(
            "query_response_cache_miss",
            "Query response cache misses",
            &[],
        )
    });

    pub static QUERY_CACHE_INVALIDATION: LazyLock<IntCounterVec> = LazyLock::new(|| {
        register_int_counter_vec(
            "query_response_cache_invalidation",
            "Query response cache invalidations (per chain)",
            &[],
        )
    });

    pub static QUERY_CACHE_ENTRIES: LazyLock<IntGauge> = LazyLock::new(|| {
        register_int_gauge(
            "query_response_cache_entries",
            "Current number of cached query responses across all chains",
        )
    });
}

/// Per-chain cache state: an LRU map plus the `next_block_height` at the time the
/// cache was last invalidated. Both are behind the same mutex.
struct PerChainCache {
    lru: LruCache<(ApplicationId, Vec<u8>), Vec<u8>>,
    next_block_height: BlockHeight,
}

/// An LRU cache for application query responses, keyed per chain.
///
/// Caches serialized response bytes keyed on `(chain_id, application_id, request_bytes)`.
/// The entire per-chain cache is invalidated when a `NewBlock` notification arrives.
///
/// To prevent a race where a slow query inserts stale data *after* an invalidation,
/// each insert carries the chain's `next_block_height` at query time.
/// If a newer block has since been processed, the insert is silently dropped.
struct QueryResponseCache {
    chains: papaya::HashMap<ChainId, StdMutex<PerChainCache>>,
    /// Chains for which we have registered a notification subscription.
    subscribed: papaya::HashSet<ChainId>,
    /// Sender half of the notification channel, used to subscribe new chains lazily.
    notification_sender: StdMutex<Option<tokio::sync::mpsc::UnboundedSender<Notification>>>,
    capacity_per_chain: std::num::NonZeroUsize,
}

impl QueryResponseCache {
    fn new(capacity_per_chain: usize) -> Self {
        Self {
            chains: papaya::HashMap::new(),
            subscribed: papaya::HashSet::new(),
            notification_sender: StdMutex::new(None),
            capacity_per_chain: std::num::NonZeroUsize::new(capacity_per_chain)
                .expect("capacity must be > 0"),
        }
    }

    /// Stores the notification sender (called once during startup).
    fn set_notification_sender(&self, sender: tokio::sync::mpsc::UnboundedSender<Notification>) {
        *self
            .notification_sender
            .lock()
            .expect("sender mutex poisoned") = Some(sender);
    }

    /// Returns the notification sender, if set.
    fn notification_sender(&self) -> Option<tokio::sync::mpsc::UnboundedSender<Notification>> {
        self.notification_sender
            .lock()
            .expect("sender mutex poisoned")
            .clone()
    }

    /// Marks a chain as subscribed to notifications.
    fn mark_subscribed(&self, chain_id: ChainId) {
        self.subscribed.pin().insert(chain_id);
    }

    /// Returns `true` if the chain is not yet subscribed to notifications.
    fn needs_subscription(&self, chain_id: &ChainId) -> bool {
        !self.subscribed.pin().contains(chain_id)
    }

    /// Marks initial chains as subscribed (called during startup).
    fn mark_all_subscribed(&self, chain_ids: &[ChainId]) {
        let pinned = self.subscribed.pin();
        for &chain_id in chain_ids {
            pinned.insert(chain_id);
        }
    }

    /// Looks up a cached response. Returns `Some(bytes)` on hit, `None` on miss
    /// (including when the chain has no cache entry yet).
    fn get(&self, chain_id: ChainId, app_id: &ApplicationId, request: &[u8]) -> Option<Vec<u8>> {
        let pinned = self.chains.pin();
        let result = pinned.get(&chain_id).and_then(|mutex| {
            mutex
                .lock()
                .expect("LRU mutex poisoned")
                .lru
                .get(&(*app_id, request.to_vec()))
                .cloned()
        });
        #[cfg(with_metrics)]
        {
            let metric = if result.is_some() {
                &query_cache_metrics::QUERY_CACHE_HIT
            } else {
                &query_cache_metrics::QUERY_CACHE_MISS
            };
            metric.with_label_values(&[]).inc();
        }
        result
    }

    /// Inserts a response into the cache, unless the chain's `next_block_height` has
    /// advanced past the caller's snapshot (which would mean a new block arrived and
    /// this response is potentially stale).
    fn insert(
        &self,
        chain_id: ChainId,
        app_id: ApplicationId,
        request: Vec<u8>,
        response: Vec<u8>,
        next_block_height: BlockHeight,
    ) {
        let pinned = self.chains.pin();
        let capacity = self.capacity_per_chain;
        let mutex = pinned.get_or_insert_with(chain_id, || {
            StdMutex::new(PerChainCache {
                lru: LruCache::new(capacity),
                next_block_height,
            })
        });
        let mut cache = mutex.lock().expect("LRU mutex poisoned");
        if next_block_height < cache.next_block_height {
            return; // A new block arrived since this query started; discard stale response.
        }
        // If the chain has advanced since the last cache update, also clear stale entries.
        // Note: This should not happen if notifications are timely. Also, this only
        // works when we have a cache miss.
        if next_block_height > cache.next_block_height {
            debug!(
                "Unexpected query cache invalidation for chain {chain_id}:\
                 {next_block_height} > {}",
                cache.next_block_height
            );
            #[cfg(with_metrics)]
            {
                query_cache_metrics::QUERY_CACHE_ENTRIES.sub(cache.lru.len() as i64);
                query_cache_metrics::QUERY_CACHE_INVALIDATION
                    .with_label_values(&[])
                    .inc();
            }
            cache.lru.clear();
            cache.next_block_height = next_block_height;
        }
        #[cfg(with_metrics)]
        let prev_len = cache.lru.len();
        cache.lru.put((app_id, request), response);
        #[cfg(with_metrics)]
        if cache.lru.len() != prev_len {
            query_cache_metrics::QUERY_CACHE_ENTRIES.inc();
        }
    }

    /// Called when a `NewBlock` notification arrives. Records the new
    /// `next_block_height` and clears all cached responses for the chain.
    fn invalidate_chain(&self, chain_id: &ChainId, next_block_height: BlockHeight) {
        let pinned = self.chains.pin();
        let capacity = self.capacity_per_chain;
        let mutex = pinned.get_or_insert_with(*chain_id, || {
            StdMutex::new(PerChainCache {
                lru: LruCache::new(capacity),
                next_block_height,
            })
        });
        let mut cache = mutex.lock().expect("LRU mutex poisoned");
        if next_block_height > cache.next_block_height {
            #[cfg(with_metrics)]
            {
                query_cache_metrics::QUERY_CACHE_ENTRIES.sub(cache.lru.len() as i64);
                query_cache_metrics::QUERY_CACHE_INVALIDATION
                    .with_label_values(&[])
                    .inc();
            }
            cache.lru.clear();
            cache.next_block_height = next_block_height;
        } else {
            debug!(
                "Query cache for chain {chain_id} was already invalidated:\
                 {next_block_height} <= {}",
                cache.next_block_height
            );
        }
    }
}

/// The `NodeService` is a server that exposes a web-server to the client.
/// The node service is primarily used to explore the state of a chain in GraphQL.
pub struct NodeService<C>
where
    C: ClientContext + 'static,
{
    config: ChainListenerConfig,
    port: NonZeroU16,
    #[cfg(with_metrics)]
    metrics_port: NonZeroU16,
    default_chain: Option<ChainId>,
    context: Arc<Mutex<C>>,
    /// If true, disallow mutations and prevent queries from scheduling operations.
    read_only: bool,
    /// Optional LRU cache for application query responses. `None` when caching is disabled.
    query_cache: Option<Arc<QueryResponseCache>>,
    query_subscriptions: Option<Arc<crate::query_subscription::QuerySubscriptionManager>>,
    cancellation_token: CancellationToken,
    enable_memory_profiling: bool,
    /// If true, do not start the chain listener; serve queries from local state only.
    pause: bool,
}

impl<C> Clone for NodeService<C>
where
    C: ClientContext + 'static,
{
    fn clone(&self) -> Self {
        Self {
            config: self.config.clone(),
            port: self.port,
            #[cfg(with_metrics)]
            metrics_port: self.metrics_port,
            default_chain: self.default_chain,
            context: Arc::clone(&self.context),
            read_only: self.read_only,
            query_cache: self.query_cache.clone(),
            query_subscriptions: self.query_subscriptions.clone(),
            cancellation_token: self.cancellation_token.clone(),
            enable_memory_profiling: self.enable_memory_profiling,
            pause: self.pause,
        }
    }
}

impl<C> NodeService<C>
where
    C: ClientContext,
{
    /// Creates a new instance of the node service given a client chain and a port.
    ///
    /// `query_cache_size` controls the per-chain LRU cache capacity for application query
    /// responses. Pass `None` to disable the cache (the default). Enable with
    /// `--query-cache-size <N>`. Incompatible with `--long-lived-services`.
    #[expect(clippy::too_many_arguments)]
    pub fn new(
        config: ChainListenerConfig,
        port: NonZeroU16,
        #[cfg(with_metrics)] metrics_port: NonZeroU16,
        default_chain: Option<ChainId>,
        context: Arc<Mutex<C>>,
        read_only: bool,
        query_cache_size: Option<usize>,
        query_subscriptions: Option<Arc<crate::query_subscription::QuerySubscriptionManager>>,
        cancellation_token: CancellationToken,
        enable_memory_profiling: bool,
        pause: bool,
    ) -> Self {
        let query_cache = query_cache_size.map(|size| Arc::new(QueryResponseCache::new(size)));
        Self {
            config,
            port,
            #[cfg(with_metrics)]
            metrics_port,
            default_chain,
            context,
            read_only,
            query_cache,
            query_subscriptions,
            cancellation_token,
            enable_memory_profiling,
            pause,
        }
    }

    /// Returns the socket address on which the metrics endpoint is served.
    #[cfg(with_metrics)]
    pub fn metrics_address(&self) -> SocketAddr {
        SocketAddr::from(([0, 0, 0, 0], self.metrics_port.get()))
    }

    /// Builds the GraphQL schema served by the node service.
    pub fn schema(&self) -> NodeServiceSchema<C> {
        let query = QueryRoot {
            context: Arc::clone(&self.context),
            port: self.port,
            default_chain: self.default_chain,
        };
        let subscription = SubscriptionRoot {
            context: Arc::clone(&self.context),
            query_subscriptions: self.query_subscriptions.clone(),
            cancellation_token: self.cancellation_token.clone(),
        };

        if self.read_only {
            NodeServiceSchema::ReadOnly(Schema::build(query, EmptyMutation, subscription).finish())
        } else {
            NodeServiceSchema::Full(
                Schema::build(
                    query,
                    MutationRoot {
                        context: Arc::clone(&self.context),
                    },
                    subscription,
                )
                .finish(),
            )
        }
    }

    /// Runs the node service.
    #[instrument(name = "node_service", level = "info", skip_all, fields(port = ?self.port))]
    pub async fn run(
        self,
        cancellation_token: CancellationToken,
        command_receiver: UnboundedReceiver<ListenerCommand>,
    ) -> Result<(), anyhow::Error> {
        let port = self.port.get();
        let index_handler = axum::routing::get(util::graphiql).post(Self::index_handler);
        let application_handler =
            axum::routing::get(util::graphiql).post(Self::application_handler);

        #[cfg(with_metrics)]
        monitoring_server::start_metrics_with_profiling(
            self.metrics_address(),
            cancellation_token.clone(),
            self.enable_memory_profiling,
        )
        .await;

        let base_router = Router::new()
            .route("/", index_handler)
            .route(
                "/chains/{chain_id}/applications/{application_id}",
                application_handler,
            )
            .route("/ready", axum::routing::get(|| async { "ready!" }));

        // Create router with appropriate schema for WebSocket subscriptions.
        let app = match self.schema() {
            NodeServiceSchema::Full(schema) => {
                base_router.route_service("/ws", GraphQLSubscription::new(schema))
            }
            NodeServiceSchema::ReadOnly(schema) => {
                base_router.route_service("/ws", GraphQLSubscription::new(schema))
            }
        }
        .layer(Extension(self.clone()))
        // TODO(#551): Provide application authentication.
        .layer(CorsLayer::permissive());

        info!("GraphiQL IDE: http://localhost:{}", port);

        // Spawn the cache invalidation listener if caching is enabled.
        if let Some(cache) = &self.query_cache {
            let guard = self.context.lock().await;
            let chain_ids: Vec<ChainId> = guard.wallet().chain_ids().try_collect().await?;
            let (tx, mut receiver) = tokio::sync::mpsc::unbounded_channel();
            guard.client().subscribe_extra(chain_ids.clone(), &tx);
            cache.mark_all_subscribed(&chain_ids);
            cache.set_notification_sender(tx);
            drop(guard);
            let cache = Arc::clone(cache);
            tokio::spawn(async move {
                while let Some(notification) = receiver.recv().await {
                    if let Reason::NewBlock { height, .. } = notification.reason {
                        let next_block_height = height
                            .try_add_one()
                            .expect("block height should not overflow");
                        cache.invalidate_chain(&notification.chain_id, next_block_height);
                    }
                }
            });
        }

        let tcp_listener =
            tokio::net::TcpListener::bind(SocketAddr::from(([0, 0, 0, 0], port))).await?;
        let server = axum::serve(tcp_listener, app)
            .with_graceful_shutdown(cancellation_token.clone().cancelled_owned())
            .into_future();

        if self.pause {
            info!("Running in paused mode: chain synchronization is disabled");
            server.await?;
        } else {
            let storage = self.context.lock().await.storage().clone();
            let chain_listener = ChainListener::new(
                self.config,
                self.context,
                storage,
                cancellation_token.clone(),
                command_receiver,
                true,
            )
            .run()
            .await?;
            let mut chain_listener = Box::pin(chain_listener).fuse();
            futures::select! {
                result = chain_listener => result?,
                result = Box::pin(server).fuse() => result?,
            };
        }

        Ok(())
    }

    /// Handles service queries for user applications (including mutations).
    async fn handle_service_request(
        &self,
        application_id: ApplicationId,
        request: Vec<u8>,
        chain_id: ChainId,
        block_hash: Option<CryptoHash>,
    ) -> Result<Vec<u8>, NodeServiceError> {
        // Only cache read-only queries against the latest state (block_hash == None).
        let cache = block_hash
            .is_none()
            .then_some(self.query_cache.as_ref())
            .flatten();

        // Return immediately on cache hit.
        if let Some(cache) = cache {
            if let Some(cached) = cache.get(chain_id, &application_id, &request) {
                return Ok(cached);
            }
        }

        let (
            QueryOutcome {
                response,
                operations,
            },
            block_height,
        ) = self
            .query_user_application(application_id, request.clone(), chain_id, block_hash)
            .await?;
        if operations.is_empty() {
            if let Some(cache) = cache {
                // Lazily subscribe to notifications for chains discovered after startup.
                if cache.needs_subscription(&chain_id) {
                    if let Some(sender) = cache.notification_sender() {
                        self.context
                            .lock()
                            .await
                            .client()
                            .subscribe_extra(vec![chain_id], &sender);
                        cache.mark_subscribed(chain_id);
                    }
                }
                cache.insert(
                    chain_id,
                    application_id,
                    request,
                    response.clone(),
                    block_height,
                );
            }
            return Ok(response);
        }

        if self.read_only {
            return Err(NodeServiceError::ReadOnlyModeOperationsNotAllowed);
        }

        trace!("Query requested a new block with operations: {operations:?}");
        let client = self
            .context
            .lock()
            .await
            .make_chain_client(chain_id)
            .await?;
        let hash = loop {
            let timeout = match client
                .execute_operations(operations.clone(), vec![])
                .await?
            {
                ClientOutcome::Committed(certificate) => break certificate.hash(),
                ClientOutcome::Conflict(certificate) => {
                    return Err(chain_client::Error::Conflict(certificate.hash()).into());
                }
                ClientOutcome::WaitForTimeout(timeout) => timeout,
            };
            let mut stream = client.subscribe().map_err(|_| {
                chain_client::Error::InternalError("Could not subscribe to the local node.")
            })?;
            util::wait_for_next_round(&mut stream, timeout).await;
        };
        let response = async_graphql::Response::new(hash.to_value());
        Ok(serde_json::to_vec(&response)?)
    }

    /// Queries a user application, returning the raw [`QueryOutcome`] and the height of the
    /// chain's latest block at the time of the query (used for cache staleness detection).
    async fn query_user_application(
        &self,
        application_id: ApplicationId,
        bytes: Vec<u8>,
        chain_id: ChainId,
        block_hash: Option<CryptoHash>,
    ) -> Result<(QueryOutcome<Vec<u8>>, BlockHeight), NodeServiceError> {
        let query = Query::User {
            application_id,
            bytes,
        };
        let client = self
            .context
            .lock()
            .await
            .make_chain_client(chain_id)
            .await?;
        let (
            QueryOutcome {
                response,
                operations,
            },
            next_block_height,
        ) = client.query_application(query, block_hash).await?;
        match response {
            QueryResponse::System(_) => {
                unreachable!("cannot get a system response for a user query")
            }
            QueryResponse::User(user_response_bytes) => Ok((
                QueryOutcome {
                    response: user_response_bytes,
                    operations,
                },
                next_block_height,
            )),
        }
    }

    /// Executes a GraphQL query and generates a response for our `Schema`.
    async fn index_handler(service: Extension<Self>, request: GraphQLRequest) -> GraphQLResponse {
        service
            .0
            .schema()
            .execute(request.into_inner())
            .await
            .into()
    }

    /// Executes a GraphQL query against an application.
    /// Pattern matches on the `OperationType` of the query and routes the query
    /// accordingly.
    async fn application_handler(
        Path((chain_id, application_id)): Path<(String, String)>,
        service: Extension<Self>,
        request: String,
    ) -> Result<Vec<u8>, NodeServiceError> {
        let chain_id: ChainId = chain_id.parse().map_err(NodeServiceError::InvalidChainId)?;
        let application_id: ApplicationId = application_id.parse()?;

        debug!(
            %chain_id,
            %application_id,
            "processing request for application:\n{:?}",
            &request
        );
        let response = service
            .0
            .handle_service_request(application_id, request.into_bytes(), chain_id, None)
            .await?;

        Ok(response)
    }
}

#[cfg(test)]
mod tests {
    use linera_base::{
        crypto::CryptoHash,
        data_types::BlockHeight,
        identifiers::{ApplicationId, ChainId},
    };

    use super::QueryResponseCache;

    fn test_chain(n: u64) -> ChainId {
        ChainId(CryptoHash::test_hash(format!("chain-{n}")))
    }

    fn test_app(n: u64) -> ApplicationId {
        ApplicationId::new(CryptoHash::test_hash(format!("app-{n}")))
    }

    #[test]
    fn cache_hit_and_miss() {
        let cache = QueryResponseCache::new(100);
        let chain = test_chain(0);
        let app = test_app(0);
        let request = b"query { balance }".to_vec();
        let response = b"{ \"balance\": 42 }".to_vec();

        // Unknown chain — get returns None.
        assert!(cache.get(chain, &app, &request).is_none());

        // Insert creates the per-chain entry.
        cache.insert(
            chain,
            app,
            request.clone(),
            response.clone(),
            BlockHeight(1),
        );

        // Hit after insert.
        assert_eq!(cache.get(chain, &app, &request), Some(response));
    }

    #[test]
    fn per_chain_isolation() {
        let cache = QueryResponseCache::new(100);
        let chain_a = test_chain(0);
        let chain_b = test_chain(1);
        let app = test_app(0);
        let request = b"q".to_vec();
        let response = b"r".to_vec();

        cache.insert(
            chain_a,
            app,
            request.clone(),
            response.clone(),
            BlockHeight(1),
        );

        // Invalidating chain B must not affect chain A.
        cache.invalidate_chain(&chain_b, BlockHeight(1));
        assert_eq!(cache.get(chain_a, &app, &request), Some(response));
    }

    #[test]
    fn invalidation_clears_all_entries() {
        let cache = QueryResponseCache::new(100);
        let chain = test_chain(0);
        let app = test_app(0);

        cache.insert(chain, app, b"q1".to_vec(), b"r1".to_vec(), BlockHeight(1));
        cache.insert(chain, app, b"q2".to_vec(), b"r2".to_vec(), BlockHeight(1));

        cache.invalidate_chain(&chain, BlockHeight(2));
        assert!(cache.get(chain, &app, b"q1").is_none());
        assert!(cache.get(chain, &app, b"q2").is_none());
    }

    #[test]
    fn lru_eviction() {
        let cache = QueryResponseCache::new(2);
        let chain = test_chain(0);
        let app = test_app(0);

        cache.insert(chain, app, b"q1".to_vec(), b"r1".to_vec(), BlockHeight(1));
        cache.insert(chain, app, b"q2".to_vec(), b"r2".to_vec(), BlockHeight(1));
        // Third insert evicts q1 (least recently used).
        cache.insert(chain, app, b"q3".to_vec(), b"r3".to_vec(), BlockHeight(1));

        assert!(cache.get(chain, &app, b"q1").is_none());
        assert!(cache.get(chain, &app, b"q2").is_some());
        assert!(cache.get(chain, &app, b"q3").is_some());
    }

    #[test]
    fn stale_insert_rejected_after_invalidation() {
        let cache = QueryResponseCache::new(100);
        let chain = test_chain(0);
        let app = test_app(0);

        // Chain is at block 3. A query starts and snapshots this height.
        cache.insert(chain, app, b"q0".to_vec(), b"r0".to_vec(), BlockHeight(3));
        let stale_height = BlockHeight(3);

        // Block 4 arrives while the query is in flight.
        cache.invalidate_chain(&chain, BlockHeight(4));

        // Slow query finishes and tries to insert with the stale height.
        cache.insert(chain, app, b"q".to_vec(), b"stale".to_vec(), stale_height);

        // The stale insert should have been rejected.
        assert!(cache.get(chain, &app, b"q").is_none());
    }
}