zeldhash-parser 0.5.1

use anyhow::{Context, Result};
use protoblock::runtime::config::FetcherConfig;

use crate::cli::{
    ProtoblockOptions, DEFAULT_PROTOBLOCK_MAX_BATCH_SIZE_MB, DEFAULT_PROTOBLOCK_REORG_WINDOW_SIZE,
    DEFAULT_PROTOBLOCK_RPC_PASSWORD, DEFAULT_PROTOBLOCK_RPC_URL, DEFAULT_PROTOBLOCK_RPC_USER,
    DEFAULT_PROTOBLOCK_THREAD_COUNT,
};

use super::overlay::Overlay;

#[derive(Debug, Clone)]
pub struct ProtoblockSettings {
    fetcher: FetcherConfig,
}

impl ProtoblockSettings {
    pub fn fetcher_config(&self) -> &FetcherConfig {
        &self.fetcher
    }
}

impl ProtoblockOptions {
    pub fn build(self, start_height: u64) -> Result<ProtoblockSettings> {
        let mut builder = FetcherConfig::builder();

        let rpc_url = self
            .rpc_url
            .unwrap_or_else(|| DEFAULT_PROTOBLOCK_RPC_URL.to_string());
        let rpc_user = self
            .rpc_user
            .unwrap_or_else(|| DEFAULT_PROTOBLOCK_RPC_USER.to_string());
        let rpc_password = self
            .rpc_password
            .unwrap_or_else(|| DEFAULT_PROTOBLOCK_RPC_PASSWORD.to_string());
        let reorg_window_size = self
            .reorg_window_size
            .unwrap_or(DEFAULT_PROTOBLOCK_REORG_WINDOW_SIZE);
        let thread_count = self.thread_count.unwrap_or(DEFAULT_PROTOBLOCK_THREAD_COUNT);
        let max_batch_size_mb = self
            .max_batch_size_mb
            .unwrap_or(DEFAULT_PROTOBLOCK_MAX_BATCH_SIZE_MB);

        builder = builder
            .rpc_url(rpc_url)
            .rpc_user(rpc_user)
            .rpc_password(rpc_password)
            .thread_count(thread_count)
            .max_batch_size_mb(max_batch_size_mb)
            .reorg_window_size(reorg_window_size)
            .start_height(start_height);

        if let Some(timeout) = self.rpc_timeout {
            builder = builder.rpc_timeout(timeout);
        }
        if let Some(interval) = self.metrics_interval {
            builder = builder.metrics_interval(interval);
        }
        if let Some(queue) = self.queue_max_size_mb {
            builder = builder.queue_max_size_mb(queue);
        }
        if let Some(backoff) = self.tip_idle_backoff {
            builder = builder.tip_idle_backoff(backoff);
        }
        if let Some(refresh) = self.tip_refresh_interval {
            builder = builder.tip_refresh_interval(refresh);
        }
        if let Some(bytes) = self.rpc_max_request_body_bytes {
            builder = builder.rpc_max_request_body_bytes(bytes);
        }
        if let Some(bytes) = self.rpc_max_response_body_bytes {
            builder = builder.rpc_max_response_body_bytes(bytes);
        }

        let fetcher = builder
            .build()
            .context("failed to build Protoblock FetcherConfig")?;

        Ok(ProtoblockSettings { fetcher })
    }
}

impl Overlay for ProtoblockOptions {
    fn overlay(self, overrides: Self) -> Self {
        Self {
            rpc_url: overrides.rpc_url.or(self.rpc_url),
            rpc_user: overrides.rpc_user.or(self.rpc_user),
            rpc_password: overrides.rpc_password.or(self.rpc_password),
            thread_count: overrides.thread_count.or(self.thread_count),
            max_batch_size_mb: overrides.max_batch_size_mb.or(self.max_batch_size_mb),
            reorg_window_size: overrides.reorg_window_size.or(self.reorg_window_size),
            rpc_timeout: overrides.rpc_timeout.or(self.rpc_timeout),
            metrics_interval: overrides.metrics_interval.or(self.metrics_interval),
            queue_max_size_mb: overrides.queue_max_size_mb.or(self.queue_max_size_mb),
            tip_idle_backoff: overrides.tip_idle_backoff.or(self.tip_idle_backoff),
            tip_refresh_interval: overrides.tip_refresh_interval.or(self.tip_refresh_interval),
            rpc_max_request_body_bytes: overrides
                .rpc_max_request_body_bytes
                .or(self.rpc_max_request_body_bytes),
            rpc_max_response_body_bytes: overrides
                .rpc_max_response_body_bytes
                .or(self.rpc_max_response_body_bytes),
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn overlay_prefers_override_when_present() {
        let base = ProtoblockOptions {
            rpc_url: Some("base".to_string()),
            thread_count: Some(2),
            max_batch_size_mb: Some(5),
            ..Default::default()
        };
        let overrides = ProtoblockOptions {
            rpc_url: Some("override".to_string()),
            thread_count: None,
            queue_max_size_mb: Some(50),
            ..Default::default()
        };

        let merged = base.overlay(overrides);

        assert_eq!(merged.rpc_url, Some("override".to_string()));
        assert_eq!(
            merged.thread_count,
            Some(2),
            "keeps base when override missing"
        );
        assert_eq!(merged.queue_max_size_mb, Some(50));
    }

    #[test]
    fn build_uses_defaults_for_missing_options() {
        let options = ProtoblockOptions::default();
        let settings = options.build(0).expect("build settings");

        let config = settings.fetcher_config();
        assert_eq!(config.start_height(), 0);
        // Verify defaults are applied
        assert_eq!(config.thread_count(), DEFAULT_PROTOBLOCK_THREAD_COUNT);
    }

    #[test]
    fn build_applies_custom_start_height() {
        let options = ProtoblockOptions::default();
        let settings = options.build(12345).expect("build settings");

        assert_eq!(settings.fetcher_config().start_height(), 12345);
    }

    #[test]
    fn build_applies_optional_rpc_timeout() {
        use std::time::Duration;

        let options = ProtoblockOptions {
            rpc_timeout: Some(Duration::from_secs(60)),
            ..Default::default()
        };
        let settings = options.build(0).expect("build settings");

        assert_eq!(
            settings.fetcher_config().rpc_timeout(),
            Duration::from_secs(60)
        );
    }

    #[test]
    fn build_applies_optional_tip_refresh_interval() {
        use std::time::Duration;

        let options = ProtoblockOptions {
            tip_refresh_interval: Some(Duration::from_secs(30)),
            ..Default::default()
        };
        let settings = options.build(0).expect("build settings");

        assert_eq!(
            settings.fetcher_config().tip_refresh_interval(),
            Duration::from_secs(30)
        );
    }

    #[test]
    fn build_applies_optional_tip_idle_backoff() {
        use std::time::Duration;

        let options = ProtoblockOptions {
            tip_idle_backoff: Some(Duration::from_millis(500)),
            ..Default::default()
        };
        let settings = options.build(0).expect("build settings");

        assert_eq!(
            settings.fetcher_config().tip_idle_backoff(),
            Duration::from_millis(500)
        );
    }

    #[test]
    fn build_applies_optional_metrics_interval() {
        use std::time::Duration;

        let options = ProtoblockOptions {
            metrics_interval: Some(Duration::from_secs(10)),
            ..Default::default()
        };
        let settings = options.build(0).expect("build settings");

        assert_eq!(
            settings.fetcher_config().metrics_interval(),
            Duration::from_secs(10)
        );
    }

    #[test]
    fn build_applies_optional_queue_max_size_mb() {
        let options = ProtoblockOptions {
            queue_max_size_mb: Some(512),
            ..Default::default()
        };
        let settings = options.build(0).expect("build settings");

        assert_eq!(settings.fetcher_config().queue_max_size_mb(), 512);
    }

    #[test]
    fn build_applies_optional_rpc_max_request_body_bytes() {
        let options = ProtoblockOptions {
            rpc_max_request_body_bytes: Some(1024 * 1024),
            ..Default::default()
        };
        let settings = options.build(0).expect("build settings");

        assert_eq!(
            settings.fetcher_config().rpc_max_request_body_bytes(),
            1024 * 1024
        );
    }

    #[test]
    fn build_applies_optional_rpc_max_response_body_bytes() {
        // Must be at least ~8MB to fit a full Bitcoin block
        let options = ProtoblockOptions {
            rpc_max_response_body_bytes: Some(16 * 1024 * 1024),
            ..Default::default()
        };
        let settings = options.build(0).expect("build settings");

        assert_eq!(
            settings.fetcher_config().rpc_max_response_body_bytes(),
            16 * 1024 * 1024
        );
    }

    #[test]
    fn overlay_handles_all_optional_fields() {
        use std::time::Duration;

        let base = ProtoblockOptions {
            rpc_timeout: Some(Duration::from_secs(30)),
            metrics_interval: Some(Duration::from_secs(5)),
            tip_idle_backoff: Some(Duration::from_millis(100)),
            tip_refresh_interval: Some(Duration::from_secs(10)),
            rpc_max_request_body_bytes: Some(100),
            rpc_max_response_body_bytes: Some(200),
            ..Default::default()
        };
        let overrides = ProtoblockOptions {
            rpc_timeout: Some(Duration::from_secs(60)),
            // Leave others as None to ensure base values are preserved
            ..Default::default()
        };

        let merged = base.overlay(overrides);

        assert_eq!(merged.rpc_timeout, Some(Duration::from_secs(60)));
        assert_eq!(merged.metrics_interval, Some(Duration::from_secs(5)));
        assert_eq!(merged.tip_idle_backoff, Some(Duration::from_millis(100)));
        assert_eq!(merged.tip_refresh_interval, Some(Duration::from_secs(10)));
        assert_eq!(merged.rpc_max_request_body_bytes, Some(100));
        assert_eq!(merged.rpc_max_response_body_bytes, Some(200));
    }
}