use alloc::{boxed::Box, vec::Vec};

use crate::errors::PicoError;

pub fn static_alloc<T>(value: T) -> &'static mut T {
    Box::leak(Box::new(value))
}

pub fn try_buffer<T: Default>(capacity: usize) -> Result<Vec<T>, PicoError> {
    try_buffer_with(capacity, Default::default)
}

fn try_buffer_with<T>(capacity: usize, f: impl FnMut() -> T) -> Result<Vec<T>, PicoError> {
    let mut buffer = Vec::new();

    buffer.try_reserve_exact(capacity)?;

    buffer.resize_with(capacity, f);

    Ok(buffer)
}

pub fn try_static_buffer_with<T>(
    capacity: usize,
    f: impl FnMut() -> T,
) -> Result<&'static mut [T], PicoError> {
    Ok(try_buffer_with(capacity, f)?.leak())
}

pub fn try_static_timestamped_block_vecs<T: Default>(
    block_num: usize,
    block_capacity: usize,
) -> Result<&'static mut [(i64, Vec<T>)], PicoError> {
    let mut blocks = Vec::new();

    blocks.try_reserve_exact(block_num)?;

    for _ in 0..block_num {
        let block = try_buffer(block_capacity)?;

        blocks.push((0, block));
    }

    Ok(blocks.leak())
}