brk_computer 0.3.0-beta.6

use brk_traversable::Traversable;
use brk_types::{Bitcoin, Cents, Dollars, Height, Sats, Version};
use vecdb::{LazyVecFrom1, ReadableCloneableVec, UnaryTransform, VecIndex};

use crate::internal::ValuePerBlock;

/// Fully lazy value type at height level.
///
/// All fields are lazy transforms from existing sources - no storage.
#[derive(Clone, Traversable)]
pub struct LazyValue<I: VecIndex> {
    pub btc: LazyVecFrom1<I, Bitcoin, I, Sats>,
    pub sats: LazyVecFrom1<I, Sats, I, Sats>,
    pub usd: LazyVecFrom1<I, Dollars, I, Dollars>,
    pub cents: LazyVecFrom1<I, Cents, I, Cents>,
}

impl LazyValue<Height> {
    pub(crate) fn from_block_source<
        SatsTransform,
        BitcoinTransform,
        CentsTransform,
        DollarsTransform,
    >(
        name: &str,
        source: &ValuePerBlock,
        version: Version,
    ) -> Self
    where
        SatsTransform: UnaryTransform<Sats, Sats>,
        BitcoinTransform: UnaryTransform<Sats, Bitcoin>,
        CentsTransform: UnaryTransform<Cents, Cents>,
        DollarsTransform: UnaryTransform<Dollars, Dollars>,
    {
        let sats = LazyVecFrom1::transformed::<SatsTransform>(
            &format!("{name}_sats"),
            version,
            source.sats.height.read_only_boxed_clone(),
        );

        let btc = LazyVecFrom1::transformed::<BitcoinTransform>(
            name,
            version,
            source.sats.height.read_only_boxed_clone(),
        );

        let cents = LazyVecFrom1::transformed::<CentsTransform>(
            &format!("{name}_cents"),
            version,
            source.cents.height.read_only_boxed_clone(),
        );

        let usd = LazyVecFrom1::transformed::<DollarsTransform>(
            &format!("{name}_usd"),
            version,
            source.usd.height.read_only_boxed_clone(),
        );

        Self {
            btc,
            sats,
            usd,
            cents,
        }
    }
}