/*
 * SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <hello@stalw.art>
 *
 * SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
 */

use std::fmt::Write;

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Arguments {
    pub tag: String,
    pub batch_size: u32,
    pub batch_range: Option<(u32, u32)>,
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Response {
    pub ranges: Vec<(u32, u32)>,
}

impl Response {
    pub fn serialize(self, tag: &str) -> Vec<u8> {
        let mut buf = String::with_capacity(32 + (self.ranges.len() * 16));
        let _ = write!(&mut buf, "* UIDBATCHES (TAG \"{tag}\")");
        for (pos, (high, low)) in self.ranges.iter().enumerate() {
            let _ = write!(&mut buf, "{}{high}:{low}", if pos == 0 { ' ' } else { ',' });
        }
        buf.push_str("\r\n");
        buf.into_bytes()
    }
}

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

    #[test]
    fn serialize_uidbatches() {
        assert_eq!(
            String::from_utf8(
                Response {
                    ranges: vec![(215295, 99696), (99695, 20351), (20350, 7830), (7829, 1)],
                }
                .serialize("A143")
            )
            .unwrap(),
            concat!(
                "* UIDBATCHES (TAG \"A143\") ",
                "215295:99696,99695:20351,20350:7830,7829:1\r\n"
            )
        );

        assert_eq!(
            String::from_utf8(Response { ranges: vec![] }.serialize("A144")).unwrap(),
            "* UIDBATCHES (TAG \"A144\")\r\n"
        );
    }
}