reduct-base 1.19.9

// Copyright 2021-2026 ReductSoftware UG
// Licensed under the Apache License, Version 2.0

use crate::error::ReductError;
use crate::{unprocessable_entity, Labels};
use http::{HeaderMap, HeaderValue};

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct RecordHeader {
    pub content_length: u64,
    pub content_type: String,
    pub labels: Labels,
}

/// Parse a batched header into a content length, content type, and labels.
///
/// # Arguments
///
/// * `header` - The batched header to parse.
///
/// # Returns
///
/// * `content_length` - The content length of the batched header.
/// * `content_type` - The content type of the batched header.
/// * `labels` - The labels of the batched header.
pub fn parse_batched_header(header: &str) -> Result<RecordHeader, ReductError> {
    let (content_length, rest) = header
        .split_once(',')
        .ok_or(unprocessable_entity!("Invalid batched header"))?;
    let content_length = content_length
        .trim()
        .parse::<u64>()
        .map_err(|_| unprocessable_entity!("Invalid content length"))?;

    let (content_type, rest) = rest
        .split_once(',')
        .unwrap_or((rest, "application/octet-stream"));

    let content_type = if content_type.is_empty() {
        "application/octet-stream".to_string()
    } else {
        content_type.trim().to_string()
    };

    let mut labels = Labels::new();
    let mut rest = rest.to_string();
    while let Some(pair) = rest.split_once('=') {
        let (key, value) = pair;
        let key = key.trim();

        rest = if value.starts_with('\"') {
            let value = value[1..].to_string();
            let (value, rest) = value
                .split_once('\"')
                .ok_or(unprocessable_entity!("Invalid batched header"))?;
            labels.insert(key.trim().to_string(), value.trim().to_string());
            rest.trim_start_matches(',').trim().to_string()
        } else if let Some(ret) = value.split_once(',') {
            let (value, rest) = ret;
            labels.insert(key.trim().to_string(), value.trim().to_string());
            rest.trim().to_string()
        } else {
            labels.insert(key.to_string(), value.trim().to_string());
            break;
        };
    }

    Ok(RecordHeader {
        content_length,
        content_type,
        labels,
    })
}

pub fn sort_headers_by_time(headers: &HeaderMap) -> Result<Vec<(u64, HeaderValue)>, ReductError> {
    let sorted_headers: Vec<_> = headers
        .clone()
        .into_iter()
        .filter(|(name, _)| name.is_some())
        .map(|(name, value)| (name.unwrap().to_string(), value))
        .filter(|(name, _)| name.starts_with("x-reduct-time-"))
        .map(|(key, value)| (key[14..].parse::<u64>().ok(), (key, value)))
        .collect();

    for (time, (key, _)) in &sorted_headers {
        if time.is_none() {
            return Err(unprocessable_entity!(
                "Invalid header '{}': must be an unix timestamp in microseconds",
                key
            ));
        }
    }

    let mut sorted_headers: Vec<(u64, HeaderValue)> = sorted_headers
        .into_iter()
        .map(|(time, (_key, value))| (time.unwrap(), value))
        .collect();
    sorted_headers.sort_by(|(ts1, _), (ts2, _)| ts1.cmp(ts2));
    Ok(sorted_headers)
}

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

    #[rstest]
    fn test_parse_batched_header_row() {
        let header = "123, text/plain, label1=value1, label2=value2";
        let RecordHeader {
            content_length,
            content_type,
            labels,
        } = parse_batched_header(header).unwrap();
        assert_eq!(content_length, 123);
        assert_eq!(content_type, "text/plain");
        assert_eq!(labels.len(), 2);
        assert_eq!(labels.get("label1"), Some(&"value1".to_string()));
        assert_eq!(labels.get("label2"), Some(&"value2".to_string()));
    }

    #[rstest]
    fn test_parse_batched_header_row_quotes() {
        let header = "123, text/plain, label1=\"[1, 2, 3]\", label2=\"value2\"";
        let RecordHeader {
            content_length,
            content_type,
            labels,
        } = parse_batched_header(header).unwrap();
        assert_eq!(content_length, 123);
        assert_eq!(content_type, "text/plain");
        assert_eq!(labels.len(), 2);
        assert_eq!(labels.get("label1"), Some(&"[1, 2, 3]".to_string()));
        assert_eq!(labels.get("label2"), Some(&"value2".to_string()));
    }

    #[rstest]
    fn test_parse_header_no_labels() {
        let header = "123, text/plain";
        let RecordHeader {
            content_length,
            content_type,
            labels,
        } = parse_batched_header(header).unwrap();
        assert_eq!(content_length, 123);
        assert_eq!(content_type, "text/plain");
        assert_eq!(labels.len(), 0);
    }

    #[rstest]
    #[case("")]
    #[case("xxx")]
    fn test_parse_header_bad_header(#[case] header: &str) {
        let err = parse_batched_header(header).err().unwrap();
        assert_eq!(err, unprocessable_entity!("Invalid batched header"));
    }

    #[rstest]
    fn test_parse_header_with_computed_label() {
        let RecordHeader { labels, .. } =
            parse_batched_header("123, text/plain, @label1=value1, label2=value2").unwrap();

        assert_eq!(labels.get("@label1"), Some(&"value1".to_string()));
        assert_eq!(labels.get("label2"), Some(&"value2".to_string()));
    }
}