signet_cold_sql/

backend.rs

//! Unified SQL backend for cold storage.
//!
//! Supports both PostgreSQL and SQLite via [`sqlx::Any`]. The backend
//! auto-detects the database type at construction time and runs the
//! appropriate migration.

use crate::SqlColdError;
use crate::columns::{
    COL_ACCESS_LIST, COL_ADDRESS, COL_AMOUNT, COL_AUTHORIZATION_LIST, COL_BASE_FEE_PER_GAS,
    COL_BENEFICIARY, COL_BLOB_GAS_USED, COL_BLOB_VERSIONED_HASHES, COL_BLOCK_DATA_HASH,
    COL_BLOCK_HASH, COL_BLOCK_LOG_INDEX, COL_BLOCK_NUMBER, COL_BLOCK_TIMESTAMP, COL_CHAIN_ID,
    COL_CNT, COL_CUMULATIVE_GAS_USED, COL_DATA, COL_DIFFICULTY, COL_EVENT_TYPE,
    COL_EXCESS_BLOB_GAS, COL_EXTRA_DATA, COL_FIRST_LOG_INDEX, COL_FROM_ADDRESS, COL_GAS,
    COL_GAS_LIMIT, COL_GAS_PRICE, COL_GAS_USED, COL_HOST_BLOCK_NUMBER, COL_INPUT, COL_LOGS_BLOOM,
    COL_MAX_BN, COL_MAX_FEE_PER_BLOB_GAS, COL_MAX_FEE_PER_GAS, COL_MAX_PRIORITY_FEE_PER_GAS,
    COL_MIX_HASH, COL_NONCE, COL_OMMERS_HASH, COL_ORDER_INDEX, COL_PARENT_BEACON_BLOCK_ROOT,
    COL_PARENT_HASH, COL_PRIOR_GAS, COL_R_CUMULATIVE_GAS_USED, COL_R_FIRST_LOG_INDEX,
    COL_R_FROM_ADDRESS, COL_R_SUCCESS, COL_R_TX_HASH, COL_R_TX_TYPE, COL_RECEIPTS_ROOT,
    COL_REQUESTS_HASH, COL_REWARD_ADDRESS, COL_ROLLUP_CHAIN_ID, COL_ROLLUP_RECIPIENT, COL_SENDER,
    COL_SIG_R, COL_SIG_S, COL_SIG_Y_PARITY, COL_STATE_ROOT, COL_SUCCESS, COL_TIMESTAMP,
    COL_TO_ADDRESS, COL_TOKEN, COL_TOPIC0, COL_TOPIC1, COL_TOPIC2, COL_TOPIC3,
    COL_TRANSACTIONS_ROOT, COL_TX_HASH, COL_TX_INDEX, COL_TX_TYPE, COL_VALUE, COL_WITHDRAWALS_ROOT,
};
use crate::convert::{
    EVENT_ENTER, EVENT_ENTER_TOKEN, EVENT_TRANSACT, build_receipt, decode_access_list_or_empty,
    decode_authorization_list, decode_b256_vec, decode_u128_required, encode_access_list,
    encode_authorization_list, encode_b256_vec, encode_u128, from_i64, to_i64,
};
use alloy::{
    consensus::{
        Header, Signed, TxEip1559, TxEip2930, TxEip4844, TxEip7702, TxLegacy, TxType,
        transaction::Recovered,
    },
    primitives::{
        Address, B256, BlockNumber, Bloom, Bytes, Log, LogData, Sealable, Signature, TxKind, U256,
    },
};
use signet_cold::{
    BlockData, ColdReceipt, ColdResult, ColdStorage, ColdStorageError, ColdStorageRead,
    ColdStorageWrite, Confirmed, Filter, HeaderSpecifier, ReceiptSpecifier, RpcLog,
    SignetEventsSpecifier, TransactionSpecifier, ZenithHeaderSpecifier,
};
use signet_storage_types::{
    ConfirmationMeta, DbSignetEvent, DbZenithHeader, IndexedReceipt, RecoveredTx, SealedHeader,
    TransactionSigned,
};
use signet_zenith::{
    Passage::{Enter, EnterToken},
    Transactor::Transact,
    Zenith,
};
use sqlx::{AnyPool, Row};
use std::collections::BTreeMap;

/// SQL-based cold storage backend.
///
/// Uses [`sqlx::Any`] for database-agnostic access, supporting both
/// PostgreSQL and SQLite through a single implementation. The backend
/// is determined by the connection URL at construction time.
///
/// # Example
///
/// ```no_run
/// # async fn example() {
/// use signet_cold_sql::SqlColdBackend;
///
/// // SQLite (in-memory)
/// let backend = SqlColdBackend::connect("sqlite::memory:").await.unwrap();
///
/// // PostgreSQL
/// let backend = SqlColdBackend::connect("postgres://localhost/signet").await.unwrap();
/// # }
/// ```
#[derive(Debug, Clone)]
pub struct SqlColdBackend {
    pool: AnyPool,
    is_postgres: bool,
}

/// Returns `true` if the URL refers to an in-memory SQLite database.
///
/// In-memory SQLite databases are per-connection, so the pool must be
/// limited to a single connection to ensure all queries see the same
/// state.
fn is_in_memory_sqlite(url: &str) -> bool {
    url == "sqlite::memory:" || url.contains("mode=memory")
}

/// Delete all rows with `block_number > bn` from every table, in
/// child-first order to preserve foreign-key constraints.
async fn delete_above_in_tx(
    tx: &mut sqlx::Transaction<'_, sqlx::Any>,
    bn: i64,
) -> Result<(), SqlColdError> {
    for table in ["logs", "transactions", "receipts", "signet_events", "zenith_headers", "headers"]
    {
        sqlx::query(&format!("DELETE FROM {table} WHERE block_number > $1"))
            .bind(bn)
            .execute(&mut **tx)
            .await?;
    }
    Ok(())
}

impl SqlColdBackend {
    /// Create a new SQL cold storage backend from an existing [`AnyPool`].
    ///
    /// Auto-detects the database backend and creates all tables if they
    /// do not already exist. Callers must ensure
    /// [`sqlx::any::install_default_drivers`] has been called before
    /// constructing the pool.
    pub async fn new(pool: AnyPool) -> Result<Self, SqlColdError> {
        // Detect backend from a pooled connection.
        let conn = pool.acquire().await?;
        let backend = conn.backend_name().to_owned();
        drop(conn);

        let migration = match backend.as_str() {
            "PostgreSQL" => include_str!("../migrations/001_initial_pg.sql"),
            "SQLite" => include_str!("../migrations/001_initial.sql"),
            other => {
                return Err(SqlColdError::Convert(format!(
                    "unsupported database backend: {other}"
                )));
            }
        };
        // Execute via pool to ensure the migration uses the same
        // connection that subsequent queries will use.
        let is_postgres = backend == "PostgreSQL";
        sqlx::raw_sql(migration).execute(&pool).await?;
        Ok(Self { pool, is_postgres })
    }

    /// Connect to a database URL with explicit pool options.
    ///
    /// Installs the default sqlx drivers on the first call. The database
    /// type is inferred from the URL scheme (`sqlite:` or `postgres:`).
    ///
    /// For in-memory SQLite URLs (`sqlite::memory:` or `mode=memory`),
    /// `max_connections` is forced to 1 regardless of the provided
    /// options, because in-memory databases are per-connection.
    pub async fn connect_with(
        url: &str,
        pool_opts: sqlx::pool::PoolOptions<sqlx::Any>,
    ) -> Result<Self, SqlColdError> {
        sqlx::any::install_default_drivers();
        let pool_opts =
            if is_in_memory_sqlite(url) { pool_opts.max_connections(1) } else { pool_opts };
        let pool: AnyPool = pool_opts.connect(url).await?;
        Self::new(pool).await
    }

    /// Connect to a database URL with default pool settings.
    ///
    /// Convenience wrapper around [`connect_with`](Self::connect_with).
    pub async fn connect(url: &str) -> Result<Self, SqlColdError> {
        Self::connect_with(url, sqlx::pool::PoolOptions::new()).await
    }

    // ========================================================================
    // Specifier resolution
    // ========================================================================

    async fn resolve_header_spec(
        &self,
        spec: HeaderSpecifier,
    ) -> Result<Option<BlockNumber>, SqlColdError> {
        match spec {
            HeaderSpecifier::Number(n) => Ok(Some(n)),
            HeaderSpecifier::Hash(hash) => {
                let row = sqlx::query("SELECT block_number FROM headers WHERE block_hash = $1")
                    .bind(hash)
                    .fetch_optional(&self.pool)
                    .await?;
                Ok(row.map(|r| from_i64(r.get::<i64, _>(COL_BLOCK_NUMBER))))
            }
        }
    }

    // ========================================================================
    // Read helpers
    // ========================================================================

    async fn fetch_header_by_number(
        &self,
        block_num: BlockNumber,
    ) -> Result<Option<SealedHeader>, SqlColdError> {
        let bn = to_i64(block_num);
        let row = sqlx::query("SELECT * FROM headers WHERE block_number = $1")
            .bind(bn)
            .fetch_optional(&self.pool)
            .await?;

        row.map(|r| header_from_row(&r).map(|h| h.seal_slow())).transpose()
    }

    // ========================================================================
    // Write helpers
    // ========================================================================

    async fn insert_block(&self, data: BlockData) -> Result<(), SqlColdError> {
        let mut tx = self.pool.begin().await?;
        write_block_to_tx(&mut tx, data).await?;
        tx.commit().await?;
        Ok(())
    }

    // ========================================================================
    // Streaming helpers
    // ========================================================================

    /// Stream logs using a PostgreSQL REPEATABLE READ transaction.
    ///
    /// The transaction provides a consistent snapshot across all per-block
    /// queries, eliminating the need for anchor-hash reorg detection.
    /// Rows are streamed individually rather than materialised per block.
    #[cfg(feature = "postgres")]
    async fn produce_log_stream_pg(&self, filter: &Filter, params: signet_cold::StreamParams) {
        use tokio_stream::StreamExt;

        /// Unwrap a `Result` or send the error through the stream and return.
        macro_rules! try_stream {
            ($sender:expr, $expr:expr) => {
                match $expr {
                    Ok(v) => v,
                    Err(e) => {
                        let _ = $sender
                            .send(Err(ColdStorageError::backend(SqlColdError::from(e))))
                            .await;
                        return;
                    }
                }
            };
        }

        let signet_cold::StreamParams { from, to, max_logs, sender, deadline } = params;

        // Open a REPEATABLE READ transaction so all per-block queries see a
        // consistent snapshot. This makes reorg detection unnecessary — if a
        // reorg lands mid-stream the transaction still reads the old data.
        let mut tx = try_stream!(sender, self.pool.begin().await);
        try_stream!(
            sender,
            sqlx::query("SET TRANSACTION ISOLATION LEVEL REPEATABLE READ").execute(&mut *tx).await
        );

        // Build the parameterised query once. $1 is the block number
        // (bound per iteration); remaining parameters are the address
        // and topic filters from the user's request.
        let (filter_clause, filter_params) = build_log_filter_clause(filter, 2);
        let data_sql = format!(
            "SELECT l.*, h.block_hash, h.timestamp AS block_timestamp, t.tx_hash, \
               (r.first_log_index + l.log_index) AS block_log_index \
             FROM logs l \
             JOIN headers h ON l.block_number = h.block_number \
             JOIN transactions t ON l.block_number = t.block_number \
               AND l.tx_index = t.tx_index \
             JOIN receipts r ON l.block_number = r.block_number \
               AND l.tx_index = r.tx_index \
             WHERE l.block_number = $1{filter_clause} \
             ORDER BY l.tx_index, l.log_index"
        );

        let mut total = 0usize;

        // Walk through blocks one at a time, streaming matching log rows
        // from each block directly to the channel.
        for block_num in from..=to {
            // Check the deadline before starting each block so we
            // don't begin a new query after the caller's timeout.
            if tokio::time::Instant::now() > deadline {
                let _ = sender.send(Err(ColdStorageError::StreamDeadlineExceeded)).await;
                return;
            }

            let mut query = sqlx::query(&data_sql).bind(to_i64(block_num));
            for param in &filter_params {
                query = query.bind(*param);
            }

            // Stream rows from this block's query. Each row is converted
            // to an RpcLog and sent over the channel immediately rather
            // than being collected into a Vec first.
            let mut stream = query.fetch(&mut *tx);
            while let Some(row_result) = stream.next().await {
                let r = try_stream!(sender, row_result);

                // Enforce the global log limit across all blocks.
                total += 1;
                if total > max_logs {
                    let _ =
                        sender.send(Err(ColdStorageError::TooManyLogs { limit: max_logs })).await;
                    return;
                }

                let log = log_from_row(&r);
                let rpc_log = RpcLog {
                    inner: log,
                    block_hash: Some(r.get(COL_BLOCK_HASH)),
                    block_number: Some(from_i64(r.get::<i64, _>(COL_BLOCK_NUMBER))),
                    block_timestamp: Some(from_i64(r.get::<i64, _>(COL_BLOCK_TIMESTAMP))),
                    transaction_hash: Some(r.get(COL_TX_HASH)),
                    transaction_index: Some(from_i64(r.get::<i64, _>(COL_TX_INDEX))),
                    log_index: Some(from_i64(r.get::<i64, _>(COL_BLOCK_LOG_INDEX))),
                    removed: false,
                };
                // Send the log to the caller. The timeout ensures we
                // stop if the deadline passes while back-pressured.
                match tokio::time::timeout_at(deadline, sender.send(Ok(rpc_log))).await {
                    Ok(Ok(())) => {}
                    Ok(Err(_)) => return, // receiver dropped
                    Err(_) => {
                        let _ = sender.send(Err(ColdStorageError::StreamDeadlineExceeded)).await;
                        return;
                    }
                }
            }

            // Early exit if we've already hit the limit — no need to
            // query the next block.
            if total >= max_logs {
                return;
            }
        }
    }
}

// ============================================================================
// Row → domain type conversion (read path)
// ============================================================================

/// Extract a required BLOB column from a row as a borrowed slice.
fn blob<'r>(r: &'r sqlx::any::AnyRow, col: &str) -> &'r [u8] {
    r.get(col)
}

/// Extract an optional BLOB column from a row as a borrowed slice.
fn opt_blob<'r>(r: &'r sqlx::any::AnyRow, col: &str) -> Option<&'r [u8]> {
    r.get(col)
}

/// Build a [`Header`] from an [`sqlx::any::AnyRow`].
fn header_from_row(r: &sqlx::any::AnyRow) -> Result<Header, SqlColdError> {
    Ok(Header {
        parent_hash: r.get(COL_PARENT_HASH),
        ommers_hash: r.get(COL_OMMERS_HASH),
        beneficiary: r.get(COL_BENEFICIARY),
        state_root: r.get(COL_STATE_ROOT),
        transactions_root: r.get(COL_TRANSACTIONS_ROOT),
        receipts_root: r.get(COL_RECEIPTS_ROOT),
        logs_bloom: Bloom::from_slice(blob(r, COL_LOGS_BLOOM)),
        difficulty: r.get(COL_DIFFICULTY),
        number: from_i64(r.get(COL_BLOCK_NUMBER)),
        gas_limit: from_i64(r.get(COL_GAS_LIMIT)),
        gas_used: from_i64(r.get(COL_GAS_USED)),
        timestamp: from_i64(r.get(COL_TIMESTAMP)),
        extra_data: r.get(COL_EXTRA_DATA),
        mix_hash: r.get(COL_MIX_HASH),
        nonce: r.get(COL_NONCE),
        base_fee_per_gas: r.get::<Option<i64>, _>(COL_BASE_FEE_PER_GAS).map(from_i64),
        withdrawals_root: r.get(COL_WITHDRAWALS_ROOT),
        blob_gas_used: r.get::<Option<i64>, _>(COL_BLOB_GAS_USED).map(from_i64),
        excess_blob_gas: r.get::<Option<i64>, _>(COL_EXCESS_BLOB_GAS).map(from_i64),
        parent_beacon_block_root: r.get(COL_PARENT_BEACON_BLOCK_ROOT),
        requests_hash: r.get(COL_REQUESTS_HASH),
    })
}

/// Build a [`TransactionSigned`] from an [`sqlx::any::AnyRow`].
fn tx_from_row(r: &sqlx::any::AnyRow) -> Result<TransactionSigned, SqlColdError> {
    use alloy::consensus::EthereumTxEnvelope;

    let sig =
        Signature::new(r.get(COL_SIG_R), r.get(COL_SIG_S), r.get::<i32, _>(COL_SIG_Y_PARITY) != 0);

    let tx_type_raw: i32 = r.get(COL_TX_TYPE);
    let tx_type_u8: u8 = tx_type_raw
        .try_into()
        .map_err(|_| SqlColdError::Convert(format!("tx_type out of u8 range: {tx_type_raw}")))?;
    let tx_type = TxType::try_from(tx_type_u8)
        .map_err(|_| SqlColdError::Convert(format!("invalid tx_type: {tx_type_u8}")))?;

    let chain_id: Option<i64> = r.get(COL_CHAIN_ID);
    let nonce = from_i64(r.get(COL_NONCE));
    let gas_limit = from_i64(r.get(COL_GAS_LIMIT));
    let to_addr: Option<Address> = r.get(COL_TO_ADDRESS);
    let value: U256 = r.get(COL_VALUE);
    let input: Bytes = r.get(COL_INPUT);

    match tx_type {
        TxType::Legacy => {
            let tx = TxLegacy {
                chain_id: chain_id.map(from_i64),
                nonce,
                gas_price: decode_u128_required(opt_blob(r, COL_GAS_PRICE), COL_GAS_PRICE)?,
                gas_limit,
                to: to_addr.map_or(TxKind::Create, TxKind::Call),
                value,
                input,
            };
            Ok(EthereumTxEnvelope::Legacy(Signed::new_unhashed(tx, sig)))
        }
        TxType::Eip2930 => {
            let tx = TxEip2930 {
                chain_id: from_i64(
                    chain_id
                        .ok_or_else(|| SqlColdError::Convert("EIP2930 requires chain_id".into()))?,
                ),
                nonce,
                gas_price: decode_u128_required(opt_blob(r, COL_GAS_PRICE), COL_GAS_PRICE)?,
                gas_limit,
                to: to_addr.map_or(TxKind::Create, TxKind::Call),
                value,
                input,
                access_list: decode_access_list_or_empty(opt_blob(r, COL_ACCESS_LIST))?,
            };
            Ok(EthereumTxEnvelope::Eip2930(Signed::new_unhashed(tx, sig)))
        }
        TxType::Eip1559 => {
            let tx = TxEip1559 {
                chain_id: from_i64(
                    chain_id
                        .ok_or_else(|| SqlColdError::Convert("EIP1559 requires chain_id".into()))?,
                ),
                nonce,
                gas_limit,
                max_fee_per_gas: decode_u128_required(
                    opt_blob(r, COL_MAX_FEE_PER_GAS),
                    COL_MAX_FEE_PER_GAS,
                )?,
                max_priority_fee_per_gas: decode_u128_required(
                    opt_blob(r, COL_MAX_PRIORITY_FEE_PER_GAS),
                    COL_MAX_PRIORITY_FEE_PER_GAS,
                )?,
                to: to_addr.map_or(TxKind::Create, TxKind::Call),
                value,
                input,
                access_list: decode_access_list_or_empty(opt_blob(r, COL_ACCESS_LIST))?,
            };
            Ok(EthereumTxEnvelope::Eip1559(Signed::new_unhashed(tx, sig)))
        }
        TxType::Eip4844 => {
            let tx = TxEip4844 {
                chain_id: from_i64(
                    chain_id
                        .ok_or_else(|| SqlColdError::Convert("EIP4844 requires chain_id".into()))?,
                ),
                nonce,
                gas_limit,
                max_fee_per_gas: decode_u128_required(
                    opt_blob(r, COL_MAX_FEE_PER_GAS),
                    COL_MAX_FEE_PER_GAS,
                )?,
                max_priority_fee_per_gas: decode_u128_required(
                    opt_blob(r, COL_MAX_PRIORITY_FEE_PER_GAS),
                    COL_MAX_PRIORITY_FEE_PER_GAS,
                )?,
                to: to_addr
                    .ok_or_else(|| SqlColdError::Convert("EIP4844 requires to_address".into()))?,
                value,
                input,
                access_list: decode_access_list_or_empty(opt_blob(r, COL_ACCESS_LIST))?,
                blob_versioned_hashes: decode_b256_vec(
                    opt_blob(r, COL_BLOB_VERSIONED_HASHES).ok_or_else(|| {
                        SqlColdError::Convert("EIP4844 requires blob_versioned_hashes".into())
                    })?,
                )?,
                max_fee_per_blob_gas: decode_u128_required(
                    opt_blob(r, COL_MAX_FEE_PER_BLOB_GAS),
                    COL_MAX_FEE_PER_BLOB_GAS,
                )?,
            };
            Ok(EthereumTxEnvelope::Eip4844(Signed::new_unhashed(tx, sig)))
        }
        TxType::Eip7702 => {
            let tx = TxEip7702 {
                chain_id: from_i64(
                    chain_id
                        .ok_or_else(|| SqlColdError::Convert("EIP7702 requires chain_id".into()))?,
                ),
                nonce,
                gas_limit,
                max_fee_per_gas: decode_u128_required(
                    opt_blob(r, COL_MAX_FEE_PER_GAS),
                    COL_MAX_FEE_PER_GAS,
                )?,
                max_priority_fee_per_gas: decode_u128_required(
                    opt_blob(r, COL_MAX_PRIORITY_FEE_PER_GAS),
                    COL_MAX_PRIORITY_FEE_PER_GAS,
                )?,
                to: to_addr
                    .ok_or_else(|| SqlColdError::Convert("EIP7702 requires to_address".into()))?,
                value,
                input,
                access_list: decode_access_list_or_empty(opt_blob(r, COL_ACCESS_LIST))?,
                authorization_list: decode_authorization_list(
                    opt_blob(r, COL_AUTHORIZATION_LIST).ok_or_else(|| {
                        SqlColdError::Convert("EIP7702 requires authorization_list".into())
                    })?,
                )?,
            };
            Ok(EthereumTxEnvelope::Eip7702(Signed::new_unhashed(tx, sig)))
        }
    }
}

/// Build a [`RecoveredTx`] from a row that includes `from_address`.
fn recovered_tx_from_row(r: &sqlx::any::AnyRow) -> Result<RecoveredTx, SqlColdError> {
    let sender: Address = r.get(COL_FROM_ADDRESS);
    let tx = tx_from_row(r)?;
    // SAFETY: the sender was recovered at append time and stored in from_address.
    Ok(Recovered::new_unchecked(tx, sender))
}

/// Build a [`Log`] from an [`sqlx::any::AnyRow`].
fn log_from_row(r: &sqlx::any::AnyRow) -> Log {
    let topics = [COL_TOPIC0, COL_TOPIC1, COL_TOPIC2, COL_TOPIC3]
        .into_iter()
        .filter_map(|col| r.get::<Option<B256>, _>(col))
        .collect();
    Log { address: r.get(COL_ADDRESS), data: LogData::new_unchecked(topics, r.get(COL_DATA)) }
}

/// Build a [`DbSignetEvent`] from an [`sqlx::any::AnyRow`].
fn signet_event_from_row(r: &sqlx::any::AnyRow) -> Result<DbSignetEvent, SqlColdError> {
    let event_type = r.get::<i32, _>(COL_EVENT_TYPE) as i16;
    let order = from_i64(r.get(COL_ORDER_INDEX));
    let rollup_chain_id: U256 = r.get(COL_ROLLUP_CHAIN_ID);

    match event_type {
        EVENT_TRANSACT => {
            let sender: Address = r
                .get::<Option<Address>, _>(COL_SENDER)
                .ok_or_else(|| SqlColdError::Convert("Transact requires sender".into()))?;
            let to: Address = r
                .get::<Option<Address>, _>(COL_TO_ADDRESS)
                .ok_or_else(|| SqlColdError::Convert("Transact requires to".into()))?;
            let value: U256 = r
                .get::<Option<U256>, _>(COL_VALUE)
                .ok_or_else(|| SqlColdError::Convert("Transact requires value".into()))?;
            let gas: U256 = r
                .get::<Option<U256>, _>(COL_GAS)
                .ok_or_else(|| SqlColdError::Convert("Transact requires gas".into()))?;
            let max_fee: U256 = r
                .get::<Option<U256>, _>(COL_MAX_FEE_PER_GAS)
                .ok_or_else(|| SqlColdError::Convert("Transact requires max_fee_per_gas".into()))?;
            let data: Bytes = r.get::<Option<Bytes>, _>(COL_DATA).unwrap_or_default();

            Ok(DbSignetEvent::Transact(
                order,
                Transact {
                    rollupChainId: rollup_chain_id,
                    sender,
                    to,
                    value,
                    gas,
                    maxFeePerGas: max_fee,
                    data,
                },
            ))
        }
        EVENT_ENTER => {
            let recipient: Address = r
                .get::<Option<Address>, _>(COL_ROLLUP_RECIPIENT)
                .ok_or_else(|| SqlColdError::Convert("Enter requires rollup_recipient".into()))?;
            let amount: U256 = r
                .get::<Option<U256>, _>(COL_AMOUNT)
                .ok_or_else(|| SqlColdError::Convert("Enter requires amount".into()))?;

            Ok(DbSignetEvent::Enter(
                order,
                Enter { rollupChainId: rollup_chain_id, rollupRecipient: recipient, amount },
            ))
        }
        EVENT_ENTER_TOKEN => {
            let token: Address = r
                .get::<Option<Address>, _>(COL_TOKEN)
                .ok_or_else(|| SqlColdError::Convert("EnterToken requires token".into()))?;
            let recipient: Address =
                r.get::<Option<Address>, _>(COL_ROLLUP_RECIPIENT).ok_or_else(|| {
                    SqlColdError::Convert("EnterToken requires rollup_recipient".into())
                })?;
            let amount: U256 = r
                .get::<Option<U256>, _>(COL_AMOUNT)
                .ok_or_else(|| SqlColdError::Convert("EnterToken requires amount".into()))?;

            Ok(DbSignetEvent::EnterToken(
                order,
                EnterToken {
                    rollupChainId: rollup_chain_id,
                    token,
                    rollupRecipient: recipient,
                    amount,
                },
            ))
        }
        _ => Err(SqlColdError::Convert(format!("invalid event_type: {event_type}"))),
    }
}

/// Build a [`DbZenithHeader`] from an [`sqlx::any::AnyRow`].
fn zenith_header_from_row(r: &sqlx::any::AnyRow) -> Result<DbZenithHeader, SqlColdError> {
    Ok(DbZenithHeader(Zenith::BlockHeader {
        hostBlockNumber: r.get(COL_HOST_BLOCK_NUMBER),
        rollupChainId: r.get(COL_ROLLUP_CHAIN_ID),
        gasLimit: r.get(COL_GAS_LIMIT),
        rewardAddress: r.get(COL_REWARD_ADDRESS),
        blockDataHash: r.get(COL_BLOCK_DATA_HASH),
    }))
}

// ============================================================================
// Domain type → SQL INSERT (write path)
// ============================================================================

/// Write a single block's data into an open SQL transaction.
async fn write_block_to_tx(
    tx: &mut sqlx::Transaction<'_, sqlx::Any>,
    data: BlockData,
) -> Result<(), SqlColdError> {
    let bn = to_i64(data.block_number());

    // Insert header
    let block_hash = data.header.hash_slow();
    let difficulty = &data.header.difficulty;
    sqlx::query(
        "INSERT INTO headers (
            block_number, block_hash, parent_hash, ommers_hash, beneficiary,
            state_root, transactions_root, receipts_root, logs_bloom, difficulty,
            gas_limit, gas_used, timestamp, extra_data, mix_hash, nonce,
            base_fee_per_gas, withdrawals_root, blob_gas_used, excess_blob_gas,
            parent_beacon_block_root, requests_hash
        ) VALUES (
            $1, $2, $3, $4, $5, $6, $7, $8, $9, $10,
            $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21, $22
        )",
    )
    .bind(bn)
    .bind(block_hash)
    .bind(data.header.parent_hash)
    .bind(data.header.ommers_hash)
    .bind(data.header.beneficiary)
    .bind(data.header.state_root)
    .bind(data.header.transactions_root)
    .bind(data.header.receipts_root)
    .bind(data.header.logs_bloom)
    .bind(difficulty)
    .bind(to_i64(data.header.gas_limit))
    .bind(to_i64(data.header.gas_used))
    .bind(to_i64(data.header.timestamp))
    .bind(&data.header.extra_data)
    .bind(data.header.mix_hash)
    .bind(data.header.nonce)
    .bind(data.header.base_fee_per_gas.map(to_i64))
    .bind(data.header.withdrawals_root.as_ref())
    .bind(data.header.blob_gas_used.map(to_i64))
    .bind(data.header.excess_blob_gas.map(to_i64))
    .bind(data.header.parent_beacon_block_root.as_ref())
    .bind(data.header.requests_hash.as_ref())
    .execute(&mut **tx)
    .await?;

    // Insert transactions
    for (idx, recovered_tx) in data.transactions.iter().enumerate() {
        insert_transaction(tx, bn, to_i64(idx as u64), recovered_tx).await?;
    }

    // Insert receipts and logs, computing first_log_index as a running
    // sum of log counts (same algorithm as the MDBX IndexedReceipt path).
    let mut first_log_index = 0i64;
    for (idx, receipt) in data.receipts.iter().enumerate() {
        let tx_idx = to_i64(idx as u64);
        sqlx::query(
            "INSERT INTO receipts (block_number, tx_index, tx_type, success, cumulative_gas_used, first_log_index)
             VALUES ($1, $2, $3, $4, $5, $6)",
        )
        .bind(bn)
        .bind(tx_idx)
        .bind(receipt.tx_type as i32)
        .bind(receipt.inner.status.coerce_status() as i32)
        .bind(to_i64(receipt.inner.cumulative_gas_used))
        .bind(first_log_index)
        .execute(&mut **tx)
        .await?;
        first_log_index += receipt.inner.logs.len() as i64;

        for (log_idx, log) in receipt.inner.logs.iter().enumerate() {
            let topics = log.topics();
            sqlx::query(
                "INSERT INTO logs (block_number, tx_index, log_index, address, topic0, topic1, topic2, topic3, data)
                 VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)",
            )
            .bind(bn)
            .bind(tx_idx)
            .bind(to_i64(log_idx as u64))
            .bind(log.address)
            .bind(topics.first())
            .bind(topics.get(1))
            .bind(topics.get(2))
            .bind(topics.get(3))
            .bind(&log.data.data)
            .execute(&mut **tx)
            .await?;
        }
    }

    // Insert signet events
    for (idx, event) in data.signet_events.iter().enumerate() {
        insert_signet_event(tx, bn, to_i64(idx as u64), event).await?;
    }

    // Insert zenith header
    if let Some(zh) = &data.zenith_header {
        let h = &zh.0;
        sqlx::query(
            "INSERT INTO zenith_headers (
                block_number, host_block_number, rollup_chain_id,
                gas_limit, reward_address, block_data_hash
            ) VALUES ($1, $2, $3, $4, $5, $6)",
        )
        .bind(bn)
        .bind(h.hostBlockNumber)
        .bind(h.rollupChainId)
        .bind(h.gasLimit)
        .bind(h.rewardAddress)
        .bind(h.blockDataHash)
        .execute(&mut **tx)
        .await?;
    }

    Ok(())
}

/// Insert a transaction, binding directly from the source type.
async fn insert_transaction(
    conn: &mut sqlx::AnyConnection,
    bn: i64,
    tx_index: i64,
    recovered: &RecoveredTx,
) -> Result<(), SqlColdError> {
    use alloy::consensus::EthereumTxEnvelope;

    let sender = recovered.signer();
    let tx: &TransactionSigned = recovered;
    let tx_hash = tx.tx_hash();
    let tx_type = tx.tx_type() as i32;

    macro_rules! sig {
        ($s:expr) => {{
            let sig = $s.signature();
            (sig.v() as i32, sig.r(), sig.s())
        }};
    }
    let (sig_y, sig_r, sig_s) = match tx {
        EthereumTxEnvelope::Legacy(s) => sig!(s),
        EthereumTxEnvelope::Eip2930(s) => sig!(s),
        EthereumTxEnvelope::Eip1559(s) => sig!(s),
        EthereumTxEnvelope::Eip4844(s) => sig!(s),
        EthereumTxEnvelope::Eip7702(s) => sig!(s),
    };

    let (chain_id, nonce, gas_limit) = match tx {
        EthereumTxEnvelope::Legacy(s) => {
            (s.tx().chain_id.map(to_i64), to_i64(s.tx().nonce), to_i64(s.tx().gas_limit))
        }
        EthereumTxEnvelope::Eip2930(s) => {
            (Some(to_i64(s.tx().chain_id)), to_i64(s.tx().nonce), to_i64(s.tx().gas_limit))
        }
        EthereumTxEnvelope::Eip1559(s) => {
            (Some(to_i64(s.tx().chain_id)), to_i64(s.tx().nonce), to_i64(s.tx().gas_limit))
        }
        EthereumTxEnvelope::Eip4844(s) => {
            (Some(to_i64(s.tx().chain_id)), to_i64(s.tx().nonce), to_i64(s.tx().gas_limit))
        }
        EthereumTxEnvelope::Eip7702(s) => {
            (Some(to_i64(s.tx().chain_id)), to_i64(s.tx().nonce), to_i64(s.tx().gas_limit))
        }
    };

    let (value, to_addr) = match tx {
        EthereumTxEnvelope::Legacy(s) => (s.tx().value, s.tx().to.to()),
        EthereumTxEnvelope::Eip2930(s) => (s.tx().value, s.tx().to.to()),
        EthereumTxEnvelope::Eip1559(s) => (s.tx().value, s.tx().to.to()),
        EthereumTxEnvelope::Eip4844(s) => (s.tx().value, Some(&s.tx().to)),
        EthereumTxEnvelope::Eip7702(s) => (s.tx().value, Some(&s.tx().to)),
    };

    let input: &[u8] = match tx {
        EthereumTxEnvelope::Legacy(s) => s.tx().input.as_ref(),
        EthereumTxEnvelope::Eip2930(s) => s.tx().input.as_ref(),
        EthereumTxEnvelope::Eip1559(s) => s.tx().input.as_ref(),
        EthereumTxEnvelope::Eip4844(s) => s.tx().input.as_ref(),
        EthereumTxEnvelope::Eip7702(s) => s.tx().input.as_ref(),
    };

    let (gas_price, max_fee, max_priority_fee, max_blob_fee) = match tx {
        EthereumTxEnvelope::Legacy(s) => (Some(encode_u128(s.tx().gas_price)), None, None, None),
        EthereumTxEnvelope::Eip2930(s) => (Some(encode_u128(s.tx().gas_price)), None, None, None),
        EthereumTxEnvelope::Eip1559(s) => (
            None,
            Some(encode_u128(s.tx().max_fee_per_gas)),
            Some(encode_u128(s.tx().max_priority_fee_per_gas)),
            None,
        ),
        EthereumTxEnvelope::Eip4844(s) => (
            None,
            Some(encode_u128(s.tx().max_fee_per_gas)),
            Some(encode_u128(s.tx().max_priority_fee_per_gas)),
            Some(encode_u128(s.tx().max_fee_per_blob_gas)),
        ),
        EthereumTxEnvelope::Eip7702(s) => (
            None,
            Some(encode_u128(s.tx().max_fee_per_gas)),
            Some(encode_u128(s.tx().max_priority_fee_per_gas)),
            None,
        ),
    };

    let (access_list, blob_hashes, auth_list) = match tx {
        EthereumTxEnvelope::Legacy(_) => (None, None, None),
        EthereumTxEnvelope::Eip2930(s) => {
            (Some(encode_access_list(&s.tx().access_list)), None, None)
        }
        EthereumTxEnvelope::Eip1559(s) => {
            (Some(encode_access_list(&s.tx().access_list)), None, None)
        }
        EthereumTxEnvelope::Eip4844(s) => (
            Some(encode_access_list(&s.tx().access_list)),
            Some(encode_b256_vec(&s.tx().blob_versioned_hashes)),
            None,
        ),
        EthereumTxEnvelope::Eip7702(s) => (
            Some(encode_access_list(&s.tx().access_list)),
            None,
            Some(encode_authorization_list(&s.tx().authorization_list)),
        ),
    };

    sqlx::query(
        "INSERT INTO transactions (
            block_number, tx_index, tx_hash, tx_type,
            sig_y_parity, sig_r, sig_s,
            chain_id, nonce, gas_limit, to_address, value, input,
            gas_price, max_fee_per_gas, max_priority_fee_per_gas,
            max_fee_per_blob_gas, blob_versioned_hashes,
            access_list, authorization_list, from_address
        ) VALUES (
            $1, $2, $3, $4, $5, $6, $7, $8, $9, $10,
            $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21
        )",
    )
    .bind(bn)
    .bind(tx_index)
    .bind(tx_hash)
    .bind(tx_type)
    .bind(sig_y)
    .bind(sig_r)
    .bind(sig_s)
    .bind(chain_id)
    .bind(nonce)
    .bind(gas_limit)
    .bind(to_addr)
    .bind(value)
    .bind(input)
    .bind(gas_price.as_ref().map(|v| v.as_slice()))
    .bind(max_fee.as_ref().map(|v| v.as_slice()))
    .bind(max_priority_fee.as_ref().map(|v| v.as_slice()))
    .bind(max_blob_fee.as_ref().map(|v| v.as_slice()))
    .bind(blob_hashes.as_deref())
    .bind(access_list.as_deref())
    .bind(auth_list.as_deref())
    .bind(sender)
    .execute(&mut *conn)
    .await?;

    Ok(())
}

/// Insert a signet event, binding directly from the source type.
async fn insert_signet_event(
    conn: &mut sqlx::AnyConnection,
    block_number: i64,
    event_index: i64,
    event: &DbSignetEvent,
) -> Result<(), SqlColdError> {
    let (event_type, order, chain_id) = match event {
        DbSignetEvent::Transact(o, t) => (0i32, to_i64(*o), &t.rollupChainId),
        DbSignetEvent::Enter(o, e) => (1i32, to_i64(*o), &e.rollupChainId),
        DbSignetEvent::EnterToken(o, e) => (2i32, to_i64(*o), &e.rollupChainId),
    };

    let (value, gas, max_fee, amount) = match event {
        DbSignetEvent::Transact(_, t) => {
            (Some(&t.value), Some(&t.gas), Some(&t.maxFeePerGas), None)
        }
        DbSignetEvent::Enter(_, e) => (None, None, None, Some(&e.amount)),
        DbSignetEvent::EnterToken(_, e) => (None, None, None, Some(&e.amount)),
    };

    sqlx::query(
        "INSERT INTO signet_events (
            block_number, event_index, event_type, order_index,
            rollup_chain_id, sender, to_address, value, gas,
            max_fee_per_gas, data, rollup_recipient, amount, token
        ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14)",
    )
    .bind(block_number)
    .bind(event_index)
    .bind(event_type)
    .bind(order)
    .bind(chain_id)
    .bind(match event {
        DbSignetEvent::Transact(_, t) => Some(&t.sender),
        _ => None,
    })
    .bind(match event {
        DbSignetEvent::Transact(_, t) => Some(&t.to),
        _ => None,
    })
    .bind(value)
    .bind(gas)
    .bind(max_fee)
    .bind(match event {
        DbSignetEvent::Transact(_, t) => Some(&t.data),
        _ => None,
    })
    .bind(match event {
        DbSignetEvent::Enter(_, e) => Some(&e.rollupRecipient),
        DbSignetEvent::EnterToken(_, e) => Some(&e.rollupRecipient),
        _ => None,
    })
    .bind(amount)
    .bind(match event {
        DbSignetEvent::EnterToken(_, e) => Some(&e.token),
        _ => None,
    })
    .execute(&mut *conn)
    .await?;

    Ok(())
}

// ============================================================================
// Log filter helpers
// ============================================================================

/// Append a SQL filter clause for a set of byte-encoded values.
///
/// For a single value, generates ` AND {column} = ${idx}`.
/// For multiple values, generates ` AND {column} IN (${idx}, ...)`.
/// Returns the next available parameter index.
fn append_filter_clause<'a>(
    clause: &mut String,
    params: &mut Vec<&'a [u8]>,
    mut idx: u32,
    column: &str,
    values: impl ExactSizeIterator<Item = &'a [u8]>,
) -> u32 {
    use std::fmt::Write;

    let len = values.len();
    if len == 1 {
        write!(clause, " AND {column} = ${idx}").unwrap();
        values.for_each(|v| params.push(v));
        return idx + 1;
    }
    write!(clause, " AND {column} IN (").unwrap();
    for (i, v) in values.enumerate() {
        if i > 0 {
            clause.push_str(", ");
        }
        write!(clause, "${idx}").unwrap();
        params.push(v);
        idx += 1;
    }
    clause.push(')');
    idx
}

fn build_log_filter_clause(filter: &Filter, start_idx: u32) -> (String, Vec<&[u8]>) {
    let mut clause = String::new();
    let mut params: Vec<&[u8]> = Vec::new();
    let mut idx = start_idx;

    if !filter.address.is_empty() {
        idx = append_filter_clause(
            &mut clause,
            &mut params,
            idx,
            "l.address",
            filter.address.iter().map(|a| a.as_slice()),
        );
    }

    let topic_cols = ["l.topic0", "l.topic1", "l.topic2", "l.topic3"];
    for (i, topic_filter) in filter.topics.iter().enumerate() {
        if topic_filter.is_empty() {
            continue;
        }
        idx = append_filter_clause(
            &mut clause,
            &mut params,
            idx,
            topic_cols[i],
            topic_filter.iter().map(|v| v.as_slice()),
        );
    }

    (clause, params)
}

// ============================================================================
// ColdStorage implementation
// ============================================================================

impl ColdStorageRead for SqlColdBackend {
    async fn get_header(&self, spec: HeaderSpecifier) -> ColdResult<Option<SealedHeader>> {
        let Some(block_num) = self.resolve_header_spec(spec).await? else {
            return Ok(None);
        };
        self.fetch_header_by_number(block_num).await.map_err(ColdStorageError::from)
    }

    async fn get_headers(
        &self,
        specs: Vec<HeaderSpecifier>,
    ) -> ColdResult<Vec<Option<SealedHeader>>> {
        let mut results = Vec::with_capacity(specs.len());
        for spec in specs {
            let header = self.get_header(spec).await?;
            results.push(header);
        }
        Ok(results)
    }

    async fn get_transaction(
        &self,
        spec: TransactionSpecifier,
    ) -> ColdResult<Option<Confirmed<RecoveredTx>>> {
        let row = match spec {
            TransactionSpecifier::Hash(hash) => sqlx::query(
                "SELECT t.*, h.block_hash
                     FROM transactions t
                     JOIN headers h ON t.block_number = h.block_number
                     WHERE t.tx_hash = $1",
            )
            .bind(hash)
            .fetch_optional(&self.pool)
            .await
            .map_err(SqlColdError::from)?,
            TransactionSpecifier::BlockAndIndex { block, index } => sqlx::query(
                "SELECT t.*, h.block_hash
                     FROM transactions t
                     JOIN headers h ON t.block_number = h.block_number
                     WHERE t.block_number = $1 AND t.tx_index = $2",
            )
            .bind(to_i64(block))
            .bind(to_i64(index))
            .fetch_optional(&self.pool)
            .await
            .map_err(SqlColdError::from)?,
            TransactionSpecifier::BlockHashAndIndex { block_hash, index } => sqlx::query(
                "SELECT t.*, h.block_hash
                     FROM transactions t
                     JOIN headers h ON t.block_number = h.block_number
                     WHERE h.block_hash = $1 AND t.tx_index = $2",
            )
            .bind(block_hash)
            .bind(to_i64(index))
            .fetch_optional(&self.pool)
            .await
            .map_err(SqlColdError::from)?,
        };

        let Some(r) = row else {
            return Ok(None);
        };

        let block = from_i64(r.get::<i64, _>(COL_BLOCK_NUMBER));
        let index = from_i64(r.get::<i64, _>(COL_TX_INDEX));
        let block_hash = r.get(COL_BLOCK_HASH);
        let recovered = recovered_tx_from_row(&r).map_err(ColdStorageError::from)?;
        let meta = ConfirmationMeta::new(block, block_hash, index);
        Ok(Some(Confirmed::new(recovered, meta)))
    }

    async fn get_transactions_in_block(&self, block: BlockNumber) -> ColdResult<Vec<RecoveredTx>> {
        let bn = to_i64(block);
        let rows =
            sqlx::query("SELECT * FROM transactions WHERE block_number = $1 ORDER BY tx_index")
                .bind(bn)
                .fetch_all(&self.pool)
                .await
                .map_err(SqlColdError::from)?;

        rows.iter().map(|r| recovered_tx_from_row(r).map_err(ColdStorageError::from)).collect()
    }

    async fn get_transaction_count(&self, block: BlockNumber) -> ColdResult<u64> {
        let bn = to_i64(block);
        let row = sqlx::query("SELECT COUNT(*) as cnt FROM transactions WHERE block_number = $1")
            .bind(bn)
            .fetch_one(&self.pool)
            .await
            .map_err(SqlColdError::from)?;

        Ok(from_i64(row.get::<i64, _>(COL_CNT)))
    }

    async fn get_receipt(&self, spec: ReceiptSpecifier) -> ColdResult<Option<ColdReceipt>> {
        // Resolve to (block, index)
        let (block, index) = match spec {
            ReceiptSpecifier::TxHash(hash) => {
                let row = sqlx::query(
                    "SELECT block_number, tx_index FROM transactions WHERE tx_hash = $1",
                )
                .bind(hash)
                .fetch_optional(&self.pool)
                .await
                .map_err(SqlColdError::from)?;
                let Some(r) = row else { return Ok(None) };
                (
                    from_i64(r.get::<i64, _>(COL_BLOCK_NUMBER)),
                    from_i64(r.get::<i64, _>(COL_TX_INDEX)),
                )
            }
            ReceiptSpecifier::BlockAndIndex { block, index } => (block, index),
        };

        // Combined query: receipt + tx metadata + full header + prior gas.
        // Header columns use standard names (h.*) so header_from_row works.
        // Receipt/tx columns use r_ prefix to avoid name collisions.
        let combined = sqlx::query(
            "SELECT h.*, \
               r.tx_type AS r_tx_type, r.success AS r_success, \
               r.cumulative_gas_used AS r_cumulative_gas_used, \
               r.first_log_index AS r_first_log_index, \
               t.tx_hash AS r_tx_hash, t.from_address AS r_from_address, \
               COALESCE( \
                 (SELECT CAST(MAX(r2.cumulative_gas_used) AS bigint) \
                  FROM receipts r2 \
                  WHERE r2.block_number = r.block_number \
                    AND r2.tx_index < r.tx_index), \
                 0 \
               ) AS prior_gas \
             FROM receipts r \
             JOIN transactions t ON r.block_number = t.block_number \
               AND r.tx_index = t.tx_index \
             JOIN headers h ON r.block_number = h.block_number \
             WHERE r.block_number = $1 AND r.tx_index = $2",
        )
        .bind(to_i64(block))
        .bind(to_i64(index))
        .fetch_optional(&self.pool)
        .await
        .map_err(SqlColdError::from)?;

        let Some(rr) = combined else {
            return Ok(None);
        };

        // Extract header using existing helper (h.* columns are unaliased).
        let header = header_from_row(&rr).map_err(ColdStorageError::from)?.seal_slow();

        // Extract receipt fields from r_ prefixed aliases.
        let tx_hash = rr.get(COL_R_TX_HASH);
        let sender = rr.get(COL_R_FROM_ADDRESS);
        let tx_type: i32 = rr.get(COL_R_TX_TYPE);
        let success = rr.get::<i32, _>(COL_R_SUCCESS) != 0;
        let cumulative_gas_used: i64 = rr.get(COL_R_CUMULATIVE_GAS_USED);
        let first_log_index: u64 = from_i64(rr.get::<i64, _>(COL_R_FIRST_LOG_INDEX));
        let prior_cumulative_gas: u64 = from_i64(rr.get::<i64, _>(COL_PRIOR_GAS));

        // Logs still require a separate query (variable row count).
        let log_rows = sqlx::query(
            "SELECT * FROM logs WHERE block_number = $1 AND tx_index = $2 ORDER BY log_index",
        )
        .bind(to_i64(block))
        .bind(to_i64(index))
        .fetch_all(&self.pool)
        .await
        .map_err(SqlColdError::from)?;

        let logs = log_rows.iter().map(log_from_row).collect();
        let built = build_receipt(tx_type, success, cumulative_gas_used, logs)
            .map_err(ColdStorageError::from)?;
        // Cumulative gas must be non-decreasing within a block; a violation
        // indicates database corruption.
        debug_assert!(
            built.inner.cumulative_gas_used >= prior_cumulative_gas,
            "cumulative gas decreased: {} < {}",
            built.inner.cumulative_gas_used,
            prior_cumulative_gas,
        );
        let gas_used = built.inner.cumulative_gas_used - prior_cumulative_gas;

        let ir = IndexedReceipt { receipt: built, tx_hash, first_log_index, gas_used, sender };
        Ok(Some(ColdReceipt::new(ir, &header, index)))
    }

    async fn get_receipts_in_block(&self, block: BlockNumber) -> ColdResult<Vec<ColdReceipt>> {
        let Some(header) =
            self.fetch_header_by_number(block).await.map_err(ColdStorageError::from)?
        else {
            return Ok(Vec::new());
        };

        let bn = to_i64(block);

        // Fetch receipts joined with tx_hash and from_address
        let receipt_rows = sqlx::query(
            "SELECT r.block_number, r.tx_index, r.tx_type, r.success, \
               r.cumulative_gas_used, r.first_log_index, \
               t.tx_hash, t.from_address \
             FROM receipts r \
             JOIN transactions t ON r.block_number = t.block_number \
               AND r.tx_index = t.tx_index \
             WHERE r.block_number = $1 \
             ORDER BY r.tx_index",
        )
        .bind(bn)
        .fetch_all(&self.pool)
        .await
        .map_err(SqlColdError::from)?;

        let all_log_rows =
            sqlx::query("SELECT * FROM logs WHERE block_number = $1 ORDER BY tx_index, log_index")
                .bind(bn)
                .fetch_all(&self.pool)
                .await
                .map_err(SqlColdError::from)?;

        // Group logs by tx_index
        let mut logs_by_tx: BTreeMap<i64, Vec<Log>> = BTreeMap::new();
        for r in &all_log_rows {
            let tx_idx: i64 = r.get(COL_TX_INDEX);
            logs_by_tx.entry(tx_idx).or_default().push(log_from_row(r));
        }

        let mut first_log_index = 0u64;
        let mut prior_cumulative_gas = 0u64;
        receipt_rows
            .into_iter()
            .enumerate()
            .map(|(idx, rr)| {
                let tx_idx: i64 = rr.get(COL_TX_INDEX);
                let tx_hash = rr.get(COL_TX_HASH);
                let sender = rr.get(COL_FROM_ADDRESS);
                let tx_type = rr.get::<i32, _>(COL_TX_TYPE);
                let success = rr.get::<i32, _>(COL_SUCCESS) != 0;
                let cumulative_gas_used: i64 = rr.get(COL_CUMULATIVE_GAS_USED);
                let logs = logs_by_tx.remove(&tx_idx).unwrap_or_default();
                let receipt = build_receipt(tx_type, success, cumulative_gas_used, logs)
                    .map_err(ColdStorageError::from)?;
                let gas_used = receipt.inner.cumulative_gas_used - prior_cumulative_gas;
                prior_cumulative_gas = receipt.inner.cumulative_gas_used;
                let ir = IndexedReceipt { receipt, tx_hash, first_log_index, gas_used, sender };
                first_log_index += ir.receipt.inner.logs.len() as u64;
                Ok(ColdReceipt::new(ir, &header, idx as u64))
            })
            .collect()
    }

    async fn get_signet_events(
        &self,
        spec: SignetEventsSpecifier,
    ) -> ColdResult<Vec<DbSignetEvent>> {
        let rows = match spec {
            SignetEventsSpecifier::Block(block) => {
                let bn = to_i64(block);
                sqlx::query(
                    "SELECT * FROM signet_events WHERE block_number = $1 ORDER BY event_index",
                )
                .bind(bn)
                .fetch_all(&self.pool)
                .await
                .map_err(SqlColdError::from)?
            }
            SignetEventsSpecifier::BlockRange { start, end } => {
                let s = to_i64(start);
                let e = to_i64(end);
                sqlx::query(
                    "SELECT * FROM signet_events WHERE block_number >= $1 AND block_number <= $2
                     ORDER BY block_number, event_index",
                )
                .bind(s)
                .bind(e)
                .fetch_all(&self.pool)
                .await
                .map_err(SqlColdError::from)?
            }
        };

        rows.iter().map(|r| signet_event_from_row(r).map_err(ColdStorageError::from)).collect()
    }

    async fn get_zenith_header(
        &self,
        spec: ZenithHeaderSpecifier,
    ) -> ColdResult<Option<DbZenithHeader>> {
        let block = match spec {
            ZenithHeaderSpecifier::Number(n) => n,
            ZenithHeaderSpecifier::Range { start, .. } => start,
        };
        let bn = to_i64(block);
        let row = sqlx::query("SELECT * FROM zenith_headers WHERE block_number = $1")
            .bind(bn)
            .fetch_optional(&self.pool)
            .await
            .map_err(SqlColdError::from)?;

        row.map(|r| zenith_header_from_row(&r)).transpose().map_err(ColdStorageError::from)
    }

    async fn get_zenith_headers(
        &self,
        spec: ZenithHeaderSpecifier,
    ) -> ColdResult<Vec<DbZenithHeader>> {
        let rows = match spec {
            ZenithHeaderSpecifier::Number(n) => {
                let bn = to_i64(n);
                sqlx::query("SELECT * FROM zenith_headers WHERE block_number = $1")
                    .bind(bn)
                    .fetch_all(&self.pool)
                    .await
                    .map_err(SqlColdError::from)?
            }
            ZenithHeaderSpecifier::Range { start, end } => {
                let s = to_i64(start);
                let e = to_i64(end);
                sqlx::query(
                    "SELECT * FROM zenith_headers WHERE block_number >= $1 AND block_number <= $2
                     ORDER BY block_number",
                )
                .bind(s)
                .bind(e)
                .fetch_all(&self.pool)
                .await
                .map_err(SqlColdError::from)?
            }
        };

        rows.iter().map(|r| zenith_header_from_row(r).map_err(ColdStorageError::from)).collect()
    }

    async fn get_logs(&self, filter: &Filter, max_logs: usize) -> ColdResult<Vec<RpcLog>> {
        let from = filter.get_from_block().unwrap_or(0);
        let to = filter.get_to_block().unwrap_or(u64::MAX);

        // Build WHERE clause: block range ($1, $2) + address/topic filters.
        let (filter_clause, params) = build_log_filter_clause(filter, 3);
        let where_clause = format!("l.block_number >= $1 AND l.block_number <= $2{filter_clause}");

        // Use LIMIT to cap results. Fetch one extra row to detect overflow
        // without a separate COUNT query. PostgreSQL stops scanning after
        // finding enough rows, making this faster than COUNT in both the
        // under-limit and over-limit cases.
        let limit_idx = 3 + params.len() as u32;
        let data_sql = format!(
            "SELECT l.*, h.block_hash, h.timestamp AS block_timestamp, t.tx_hash, \
               (r.first_log_index + l.log_index) AS block_log_index \
             FROM logs l \
             JOIN headers h ON l.block_number = h.block_number \
             JOIN transactions t ON l.block_number = t.block_number \
               AND l.tx_index = t.tx_index \
             JOIN receipts r ON l.block_number = r.block_number \
               AND l.tx_index = r.tx_index \
             WHERE {where_clause} \
             ORDER BY l.block_number, l.tx_index, l.log_index \
             LIMIT ${limit_idx}"
        );
        let mut query = sqlx::query(&data_sql).bind(to_i64(from)).bind(to_i64(to));
        for param in &params {
            query = query.bind(*param);
        }
        // Clamp to i64::MAX before converting: SQL LIMIT is an i64, and
        // saturating_add avoids overflow when max_logs is usize::MAX.
        let limit = max_logs.saturating_add(1).min(i64::MAX as usize);
        query = query.bind(to_i64(limit as u64));

        let rows = query.fetch_all(&self.pool).await.map_err(SqlColdError::from)?;

        if rows.len() > max_logs {
            return Err(ColdStorageError::TooManyLogs { limit: max_logs });
        }

        rows.into_iter()
            .map(|r| {
                let log = log_from_row(&r);
                Ok(RpcLog {
                    inner: log,
                    block_hash: Some(r.get(COL_BLOCK_HASH)),
                    block_number: Some(from_i64(r.get::<i64, _>(COL_BLOCK_NUMBER))),
                    block_timestamp: Some(from_i64(r.get::<i64, _>(COL_BLOCK_TIMESTAMP))),
                    transaction_hash: Some(r.get(COL_TX_HASH)),
                    transaction_index: Some(from_i64(r.get::<i64, _>(COL_TX_INDEX))),
                    log_index: Some(from_i64(r.get::<i64, _>(COL_BLOCK_LOG_INDEX))),
                    removed: false,
                })
            })
            .collect::<ColdResult<Vec<_>>>()
    }

    async fn produce_log_stream(&self, filter: &Filter, params: signet_cold::StreamParams) {
        #[cfg(feature = "postgres")]
        if self.is_postgres {
            return self.produce_log_stream_pg(filter, params).await;
        }
        signet_cold::produce_log_stream_default(self, filter, params).await;
    }

    async fn get_latest_block(&self) -> ColdResult<Option<BlockNumber>> {
        let row = sqlx::query("SELECT MAX(block_number) as max_bn FROM headers")
            .fetch_one(&self.pool)
            .await
            .map_err(SqlColdError::from)?;
        Ok(row.get::<Option<i64>, _>(COL_MAX_BN).map(from_i64))
    }
}

impl ColdStorageWrite for SqlColdBackend {
    async fn append_block(&mut self, data: BlockData) -> ColdResult<()> {
        self.insert_block(data).await.map_err(ColdStorageError::from)
    }

    async fn append_blocks(&mut self, data: Vec<BlockData>) -> ColdResult<()> {
        let mut tx = self.pool.begin().await.map_err(SqlColdError::from)?;
        for block_data in data {
            write_block_to_tx(&mut tx, block_data).await.map_err(ColdStorageError::from)?;
        }
        tx.commit().await.map_err(SqlColdError::from)?;
        Ok(())
    }

    async fn truncate_above(&mut self, block: BlockNumber) -> ColdResult<()> {
        let bn = to_i64(block);
        let mut tx = self.pool.begin().await.map_err(SqlColdError::from)?;

        delete_above_in_tx(&mut tx, bn).await.map_err(ColdStorageError::from)?;

        tx.commit().await.map_err(SqlColdError::from)?;
        Ok(())
    }
}

impl ColdStorage for SqlColdBackend {
    async fn drain_above(&mut self, block: BlockNumber) -> ColdResult<Vec<Vec<ColdReceipt>>> {
        let bn = to_i64(block);
        let mut tx = self.pool.begin().await.map_err(SqlColdError::from)?;

        // 1. Fetch all headers above block.
        let header_rows =
            sqlx::query("SELECT * FROM headers WHERE block_number > $1 ORDER BY block_number")
                .bind(bn)
                .fetch_all(&mut *tx)
                .await
                .map_err(SqlColdError::from)?;

        if header_rows.is_empty() {
            tx.commit().await.map_err(SqlColdError::from)?;
            return Ok(Vec::new());
        }

        let mut headers: BTreeMap<i64, SealedHeader> = BTreeMap::new();
        for r in &header_rows {
            let num: i64 = r.get(COL_BLOCK_NUMBER);
            let h = header_from_row(r).map_err(ColdStorageError::from)?.seal_slow();
            headers.insert(num, h);
        }

        // 2. Fetch all receipt + tx metadata above block.
        let receipt_rows = sqlx::query(
            "SELECT r.block_number, r.tx_index, r.tx_type, r.success, \
               r.cumulative_gas_used, r.first_log_index, \
               t.tx_hash, t.from_address \
             FROM receipts r \
             JOIN transactions t ON r.block_number = t.block_number \
               AND r.tx_index = t.tx_index \
             WHERE r.block_number > $1 \
             ORDER BY r.block_number, r.tx_index",
        )
        .bind(bn)
        .fetch_all(&mut *tx)
        .await
        .map_err(SqlColdError::from)?;

        // 3. Fetch all logs above block.
        let log_rows = sqlx::query(
            "SELECT * FROM logs WHERE block_number > $1 \
             ORDER BY block_number, tx_index, log_index",
        )
        .bind(bn)
        .fetch_all(&mut *tx)
        .await
        .map_err(SqlColdError::from)?;

        // Group logs by (block_number, tx_index).
        let mut logs_by_block_tx: BTreeMap<(i64, i64), Vec<Log>> = BTreeMap::new();
        for r in &log_rows {
            let block_num: i64 = r.get(COL_BLOCK_NUMBER);
            let tx_idx: i64 = r.get(COL_TX_INDEX);
            logs_by_block_tx.entry((block_num, tx_idx)).or_default().push(log_from_row(r));
        }

        // Group receipt rows by block_number.
        let mut receipts_by_block: BTreeMap<i64, Vec<&sqlx::any::AnyRow>> = BTreeMap::new();
        for r in &receipt_rows {
            let block_num: i64 = r.get(COL_BLOCK_NUMBER);
            receipts_by_block.entry(block_num).or_default().push(r);
        }

        // 4. Assemble ColdReceipts per block.
        let mut all_receipts = Vec::with_capacity(headers.len());
        for (&block_num, header) in &headers {
            let block_receipt_rows = receipts_by_block.remove(&block_num).unwrap_or_default();
            let mut prior_cumulative_gas = 0u64;
            let block_receipts: ColdResult<Vec<ColdReceipt>> = block_receipt_rows
                .into_iter()
                .map(|rr: &sqlx::any::AnyRow| {
                    let tx_idx: i64 = rr.get(COL_TX_INDEX);
                    let tx_hash = rr.get(COL_TX_HASH);
                    let sender = rr.get(COL_FROM_ADDRESS);
                    let tx_type: i32 = rr.get(COL_TX_TYPE);
                    let success = rr.get::<i32, _>(COL_SUCCESS) != 0;
                    let cumulative_gas_used: i64 = rr.get(COL_CUMULATIVE_GAS_USED);
                    let first_log_index = from_i64(rr.get::<i64, _>(COL_FIRST_LOG_INDEX));
                    let logs = logs_by_block_tx.remove(&(block_num, tx_idx)).unwrap_or_default();
                    let receipt = build_receipt(tx_type, success, cumulative_gas_used, logs)
                        .map_err(ColdStorageError::from)?;
                    // Cumulative gas must be non-decreasing within a block; a violation
                    // indicates database corruption.
                    debug_assert!(
                        receipt.inner.cumulative_gas_used >= prior_cumulative_gas,
                        "cumulative gas decreased: {} < {}",
                        receipt.inner.cumulative_gas_used,
                        prior_cumulative_gas,
                    );
                    let gas_used = receipt.inner.cumulative_gas_used - prior_cumulative_gas;
                    prior_cumulative_gas = receipt.inner.cumulative_gas_used;
                    let ir = IndexedReceipt { receipt, tx_hash, first_log_index, gas_used, sender };
                    Ok(ColdReceipt::new(ir, header, from_i64(tx_idx)))
                })
                .collect();
            all_receipts.push(block_receipts?);
        }

        // 5. Delete from all tables (same order as truncate_above).
        delete_above_in_tx(&mut tx, bn).await.map_err(ColdStorageError::from)?;

        tx.commit().await.map_err(SqlColdError::from)?;
        Ok(all_receipts)
    }
}

#[cfg(all(test, feature = "test-utils"))]
mod tests {
    use super::*;
    use signet_cold::conformance::conformance;

    #[tokio::test]
    async fn sqlite_conformance() {
        let backend = SqlColdBackend::connect("sqlite::memory:").await.unwrap();
        conformance(backend).await.unwrap();
    }

    #[tokio::test]
    async fn pg_conformance() {
        let Ok(url) = std::env::var("DATABASE_URL") else {
            eprintln!("skipping pg conformance: DATABASE_URL not set");
            return;
        };
        let backend = SqlColdBackend::connect(&url).await.unwrap();
        conformance(backend).await.unwrap();
    }
}