cesr-rs 0.4.0

use crate::core::counter::CounterCodeV1;
use crate::core::counter::CounterCodeV2;
#[cfg(feature = "alloc")]
#[allow(
    unused_imports,
    reason = "alloc prelude items; subset used per cfg/feature combination"
)]
use alloc::{format, vec, vec::Vec};
use bytes::Bytes;

use crate::stream::error::ParseError;
use crate::stream::parse::skip_counter;
use crate::stream::parse::skip_indexer;
use crate::stream::parse::skip_matter;

use super::types::TransIdxSigGroups;

pub(super) fn parse(input: &Bytes, count: u32) -> Result<(TransIdxSigGroups, Bytes), ParseError> {
    let mut offset = 0;
    for _ in 0..count {
        offset += skip_matter(&input[offset..])?;
        offset += skip_matter(&input[offset..])?;
        offset += skip_matter(&input[offset..])?;
        let counter_slice = &input[offset..];
        let counter_size = skip_counter(counter_slice)?;
        let (code, sub_count, _) = crate::stream::parse::parse_counter(counter_slice)?;
        if code != CounterCodeV1::ControllerIdxSigs {
            return Err(ParseError::Malformed(format!(
                "expected -A counter inside -F group, got {}",
                code.as_str()
            )));
        }
        offset += counter_size;
        for _ in 0..sub_count {
            offset += skip_indexer(&input[offset..])?;
        }
    }
    let raw = input.slice(..offset);
    let rest = input.slice(offset..);
    Ok((TransIdxSigGroups::new(raw, count, false), rest))
}

pub(super) fn parse_v2(
    input: &Bytes,
    count: u32,
) -> Result<(TransIdxSigGroups, Bytes), ParseError> {
    let mut offset = 0;
    for _ in 0..count {
        offset += skip_matter(&input[offset..])?;
        offset += skip_matter(&input[offset..])?;
        offset += skip_matter(&input[offset..])?;
        let counter_slice = &input[offset..];
        let counter_size = skip_counter(counter_slice)?;
        let (code, sub_count, _) = crate::stream::parse::parse_counter_v2(counter_slice)?;
        if code != CounterCodeV2::ControllerIdxSigs {
            return Err(ParseError::Malformed(format!(
                "expected -K counter inside -X group (V2), got {}",
                code.as_str()
            )));
        }
        offset += counter_size;
        for _ in 0..sub_count {
            offset += skip_indexer(&input[offset..])?;
        }
    }
    let raw = input.slice(..offset);
    let rest = input.slice(offset..);
    Ok((TransIdxSigGroups::new(raw, count, true), rest))
}

#[cfg(test)]
#[allow(
    clippy::unwrap_used,
    clippy::expect_used,
    clippy::panic,
    clippy::as_conversions,
    reason = "test code: panics and type conversions acceptable"
)]
mod tests {
    use super::*;
    use crate::core::counter::CounterCodeV1;
    use crate::core::indexer::IndexerBuilder;
    use crate::core::indexer::code::IndexedSigCode;
    use base64::{Engine, engine::general_purpose as b64};
    use core::num::NonZeroUsize;

    fn build_ed25519_qb64() -> Vec<u8> {
        let raw = [0xAB_u8; 32];
        let ps = 1_usize;
        let mut padded = vec![0u8; ps];
        padded.extend_from_slice(&raw);
        let payload_b64 = b64::URL_SAFE_NO_PAD.encode(&padded);
        format!("D{}", &payload_b64[ps..]).into_bytes()
    }

    fn build_blake3_256_qb64() -> Vec<u8> {
        let raw = [0xCD_u8; 32];
        let ps = 1_usize;
        let mut padded = vec![0u8; ps];
        padded.extend_from_slice(&raw);
        let payload_b64 = b64::URL_SAFE_NO_PAD.encode(&padded);
        format!("E{}", &payload_b64[ps..]).into_bytes()
    }

    fn build_siger_qb64(index: u32) -> Vec<u8> {
        IndexerBuilder::new()
            .with_code(IndexedSigCode::Ed25519)
            .with_index(index)
            .unwrap()
            .with_raw(&[0u8; 64])
            .unwrap()
            .to_qb64()
            .into_bytes()
    }

    fn build_counter_qb64(code: CounterCodeV1, count: u32) -> Vec<u8> {
        let hard = code.as_str();
        let ss = code.soft_size();
        let ss_nz = NonZeroUsize::new(ss).unwrap();
        let soft = crate::b64::encode_int(count, ss_nz);
        format!("{hard}{soft}").into_bytes()
    }

    #[test]
    fn parse_zero_elements() {
        let (group, rest) = parse(&Bytes::new(), 0).unwrap();
        assert_eq!(group.count(), 0);
        assert!(rest.is_empty());
    }

    #[test]
    fn parse_one_group_with_two_sigs() {
        let mut input = build_ed25519_qb64();
        input.extend_from_slice(&build_ed25519_qb64());
        input.extend_from_slice(&build_blake3_256_qb64());
        input.extend_from_slice(&build_counter_qb64(CounterCodeV1::ControllerIdxSigs, 2));
        input.extend_from_slice(&build_siger_qb64(0));
        input.extend_from_slice(&build_siger_qb64(1));

        let (group, rest) = parse(&Bytes::copy_from_slice(&input), 1).unwrap();
        assert_eq!(group.count(), 1);
        let elem = group.iter().next().unwrap().unwrap();
        assert_eq!(elem.3.count() as usize, 2);
        assert!(rest.is_empty());
    }

    #[test]
    fn trailing_bytes_preserved() {
        let mut input = build_ed25519_qb64();
        input.extend_from_slice(&build_ed25519_qb64());
        input.extend_from_slice(&build_blake3_256_qb64());
        input.extend_from_slice(&build_counter_qb64(CounterCodeV1::ControllerIdxSigs, 1));
        input.extend_from_slice(&build_siger_qb64(0));
        input.extend_from_slice(b"TAIL");

        let (group, rest) = parse(&Bytes::copy_from_slice(&input), 1).unwrap();
        assert_eq!(group.count(), 1);
        assert_eq!(rest, Bytes::from_static(b"TAIL"));
    }
}