use crate::text::misc_token::{unstructured, Unstructured};
use crate::text::whitespace::{foldable_line, obs_crlf};
use nom::{
    branch::alt,
    bytes::complete::{tag, tag_no_case, take_while1},
    character::complete::space0,
    combinator::map,
    multi::{fold_many0},
    sequence::{pair, terminated, tuple},
    IResult,
};

#[derive(Debug, PartialEq)]
pub enum CompField<'a, T> {
    Known(T),
    Unknown(Kv<'a>),
    Bad(&'a [u8]),
}

#[derive(Debug, PartialEq, Clone)]
pub struct Kv<'a>(pub &'a [u8], pub Unstructured<'a>);


pub fn header<'a, T>(
    fx: impl Fn(&'a [u8]) -> IResult<&'a [u8], T> + Copy,
) -> impl Fn(&'a [u8]) -> IResult<&'a [u8], (Vec::<T>, Vec::<Kv>, Vec<&'a [u8]>)> {
    move |input| {
            terminated(
                fold_many0(
                    alt((
                        map(fx, CompField::Known),
                        map(opt_field, |(k, v)| CompField::Unknown(Kv(k, v))),
                        map(foldable_line, CompField::Bad),
                    )),
                    || (Vec::<T>::new(), Vec::<Kv>::new(), Vec::<&'a [u8]>::new()),
                    |(mut known, mut unknown, mut bad), item| {
                        match item {
                            CompField::Known(v) => known.push(v),
                            CompField::Unknown(v) => unknown.push(v),
                            CompField::Bad(v) => bad.push(v),
                        };
                        (known, unknown, bad)
                    }
                ),
                obs_crlf,
            )(input)
    }
}

pub fn field_name<'a>(name: &'static [u8]) -> impl Fn(&'a [u8]) -> IResult<&'a [u8], &'a [u8]> {
    move |input| terminated(tag_no_case(name), tuple((space0, tag(b":"), space0)))(input)
}

/// Optional field
///
/// ```abnf
/// field      =   field-name ":" unstructured CRLF
/// field-name =   1*ftext
/// ftext      =   %d33-57 /          ; Printable US-ASCII
///                %d59-126           ;  characters not including
///                                   ;  ":".
/// ```
pub fn opt_field(input: &[u8]) -> IResult<&[u8], (&[u8], Unstructured)> {
    terminated(
        pair(
            terminated(
                take_while1(|c| (0x21..=0x7E).contains(&c) && c != 0x3A),
                tuple((space0, tag(b":"), space0)),
            ),
            unstructured,
        ),
        obs_crlf,
    )(input)
}