keyvalues-parser 0.2.3

use std::borrow::Cow;

use crate::{error::Result, Obj, PartialVdf, Value, Vdf};

use pest::{iterators::Pair as PestPair, Atomicity, RuleType};

// TODO: rename `PartialVdf` to `TopLevelVdf` and have it hold a `Vdf` instead of flattening it out

mod escaped;
mod raw;

#[expect(deprecated)]
pub use escaped::parse as escaped_parse;
pub use escaped::PestError as EscapedPestError;
#[expect(deprecated)]
pub use raw::parse as raw_parse;
pub use raw::PestError as RawPestError;

type BoxedState<'a, R> = Box<pest::ParserState<'a, R>>;
type ParseResult<'a, R> = pest::ParseResult<BoxedState<'a, R>>;

#[inline]
fn whitespace<R: RuleType>(s: BoxedState<'_, R>) -> ParseResult<'_, R> {
    s.atomic(Atomicity::Atomic, |s| {
        s.match_string(" ")
            .or_else(|s| s.match_string("\t"))
            .or_else(|s| s.match_string("\r"))
            .or_else(|s| s.match_string("\n"))
    })
}

#[inline]
fn any<R: RuleType>(s: BoxedState<'_, R>) -> ParseResult<'_, R> {
    s.skip(1)
}

fn soi<R: RuleType>(s: BoxedState<'_, R>) -> ParseResult<'_, R> {
    s.start_of_input()
}

#[inline]
fn comment<R: RuleType>(s: BoxedState<'_, R>) -> ParseResult<'_, R> {
    s.atomic(Atomicity::Atomic, |s| {
        s.sequence(|s| {
            s.match_string("//").and_then(|s| {
                s.repeat(|s| {
                    s.sequence(|s| s.lookahead(false, |s| s.match_string("\n")).and_then(any))
                })
            })
        })
    })
}

#[inline]
fn skip<R: RuleType>(s: BoxedState<'_, R>) -> ParseResult<'_, R> {
    if s.atomicity() == Atomicity::NonAtomic {
        s.sequence(|s| {
            s.repeat(whitespace).and_then(|s| {
                s.repeat(|s| s.sequence(|s| comment(s).and_then(|s| s.repeat(whitespace))))
            })
        })
    } else {
        Ok(s)
    }
}

// unfortunate hack to re-use most of the code that consumes the pest parser produced by our two
// separate grammars :/
macro_rules! common_parsing {
    ($parse_fn:ident, $rule:ty, $parse_escaped:expr) => {
        fn parse_(s: &str) -> Result<PartialVdf<'_>> {
            let mut full_grammar = $parse_fn(s)?;

            // There can be multiple base macros before the initial pair
            let mut bases = Vec::new();
            loop {
                let pair = full_grammar.next().unwrap();
                if let <$rule>::base_macro = pair.as_rule() {
                    let base_path_string = pair.into_inner().next().unwrap();
                    let base_path = match base_path_string.as_rule() {
                        <$rule>::quoted_raw_string => base_path_string.into_inner().next().unwrap(),
                        <$rule>::unquoted_string => base_path_string,
                        _ => unreachable!("Prevented by grammar"),
                    }
                    .as_str();
                    bases.push(Cow::from(base_path));
                } else {
                    let (key, value) = parse_pair(pair);
                    return Ok(PartialVdf { key, value, bases });
                }
            }
        }

        fn parse_pair(grammar_pair: PestPair<'_, $rule>) -> (Cow<'_, str>, Value<'_>) {
            // Structure: pair
            //            \ key   <- Desired
            //            \ value <- Desired
            if let <$rule>::pair = grammar_pair.as_rule() {
                // Parse out the key and value
                let mut grammar_pair_innards = grammar_pair.into_inner();
                let grammar_string = grammar_pair_innards.next().unwrap();
                let key = parse_string(grammar_string);

                let grammar_value = grammar_pair_innards.next().unwrap();
                let value = Value::from(grammar_value);

                (key, value)
            } else {
                unreachable!("Prevented by grammar");
            }
        }

        fn parse_string(grammar_string: PestPair<'_, $rule>) -> Cow<'_, str> {
            match grammar_string.as_rule() {
                // Structure: quoted_string
                //            \ "
                //            \ quoted_inner <- Desired
                //            \ "
                <$rule>::quoted_string => {
                    let quoted_inner = grammar_string.into_inner().next().unwrap();
                    if $parse_escaped {
                        parse_escaped_string(quoted_inner)
                    } else {
                        Cow::from(quoted_inner.as_str())
                    }
                }
                // Structure: unquoted_string <- Desired
                <$rule>::unquoted_string => {
                    let s = grammar_string.as_str();
                    Cow::from(s)
                }
                _ => unreachable!("Prevented by grammar"),
            }
        }

        // Note: there can be a slight performance win here by having the grammar skip capturing
        // quoted_inner and instead just slice off the starting and ending '"', but I'm going to pass since
        // it seems like a hack for a ~4% improvement
        fn parse_escaped_string(inner: PestPair<'_, $rule>) -> Cow<'_, str> {
            let s = inner.as_str();

            if s.contains('\\') {
                // Escaped version won't be quite as long, but it will likely be close
                let mut escaped = String::with_capacity(s.len());
                let mut it = s.chars();

                while let Some(ch) = it.next() {
                    if ch == '\\' {
                        // Character is escaped so check the next character to figure out the full
                        // character
                        match it.next() {
                            Some('n') => escaped.push('\n'),
                            Some('r') => escaped.push('\r'),
                            Some('t') => escaped.push('\t'),
                            Some('\\') => escaped.push('\\'),
                            Some('\"') => escaped.push('\"'),
                            _ => unreachable!("Prevented by grammar"),
                        }
                    } else {
                        escaped.push(ch)
                    }
                }

                Cow::from(escaped)
            } else {
                Cow::from(s)
            }
        }

        impl<'a> From<PestPair<'a, $rule>> for Value<'a> {
            fn from(grammar_value: PestPair<'a, $rule>) -> Self {
                // Structure: value is ( obj | quoted_string | unquoted_string )
                match grammar_value.as_rule() {
                    // Structure: ( quoted_string | unquoted_string )
                    <$rule>::quoted_string | <$rule>::unquoted_string => {
                        Self::Str(parse_string(grammar_value))
                    }
                    // Structure: obj
                    //            \ pair* <- Desired
                    <$rule>::obj => {
                        let mut obj = Obj::new();
                        for grammar_pair in grammar_value.into_inner() {
                            let (key, value) = parse_pair(grammar_pair);
                            let entry = obj.entry(key).or_default();
                            (*entry).push(value);
                        }

                        Self::Obj(obj)
                    }
                    _ => unreachable!("Prevented by grammar"),
                }
            }
        }
    };
}

// expose ^^ macro to the rest of the crate
pub(crate) use common_parsing;

impl<'a> Vdf<'a> {
    /// Attempts to parse VDF text to a [`Vdf`]
    pub fn parse(s: &'a str) -> Result<Self> {
        Ok(Self::from(PartialVdf::parse(s)?))
    }

    pub fn parse_raw(s: &'a str) -> Result<Self> {
        Ok(Self::from(PartialVdf::parse_raw(s)?))
    }
}

impl<'a> PartialVdf<'a> {
    /// Attempts to parse VDF text to a [`Vdf`]
    pub fn parse(s: &'a str) -> Result<Self> {
        #[expect(deprecated)]
        escaped_parse(s)
    }

    pub fn parse_raw(s: &'a str) -> Result<Self> {
        #[expect(deprecated)]
        raw_parse(s)
    }
}