jsonpath-rust 1.0.4

use crate::parser::errors::JsonPathError;
use crate::parser::Parsed;
use std::fmt::{Display, Formatter};

/// Represents a JSONPath query with a list of segments.
#[derive(Debug, Clone, PartialEq)]
pub struct JpQuery {
    pub segments: Vec<Segment>,
}

impl JpQuery {
    pub fn new(segments: Vec<Segment>) -> Self {
        JpQuery { segments }
    }
}

impl Display for JpQuery {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "${}",
            self.segments
                .iter()
                .map(|s| s.to_string())
                .collect::<String>()
        )
    }
}
/// Enum representing different types of segments in a JSONPath query.
#[derive(Debug, Clone, PartialEq)]
pub enum Segment {
    /// Represents a descendant segment.
    Descendant(Box<Segment>),
    /// Represents a selector segment.
    Selector(Selector),
    /// Represents multiple selectors.
    Selectors(Vec<Selector>),
}

impl Segment {
    pub fn name(name: &str) -> Self {
        Segment::Selector(Selector::Name(name.to_string()))
    }
}

impl Display for Segment {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            Segment::Descendant(s) => write!(f, "..{}", s),
            Segment::Selector(selector) => write!(f, "{}", selector),
            Segment::Selectors(selectors) => write!(
                f,
                "{}",
                selectors.iter().map(|s| s.to_string()).collect::<String>()
            ),
        }
    }
}
/// Enum representing different types of selectors in a JSONPath query.
#[derive(Debug, Clone, PartialEq)]
pub enum Selector {
    /// Represents a name selector.
    Name(String),
    /// Represents a wildcard selector.
    Wildcard,
    /// Represents an index selector.
    Index(i64),
    /// Represents a slice selector.
    Slice(Option<i64>, Option<i64>, Option<i64>),
    /// Represents a filter selector.
    Filter(Filter),
}

pub fn slice_from((start, end, step): (Option<i64>, Option<i64>, Option<i64>)) -> Selector {
    Selector::Slice(start, end, step)
}

impl Display for Selector {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            Selector::Name(name) => write!(f, "{}", name),
            Selector::Wildcard => write!(f, "*"),
            Selector::Index(index) => write!(f, "{}", index),
            Selector::Slice(start, end, step) => write!(
                f,
                "{}:{}:{}",
                start.unwrap_or(0),
                end.unwrap_or(0),
                step.unwrap_or(1)
            ),
            Selector::Filter(filter) => write!(f, "[?{}]", filter),
        }
    }
}
/// Enum representing different types of filters in a JSONPath query.
#[derive(Debug, Clone, PartialEq)]
pub enum Filter {
    /// Represents a logical OR filter.
    Or(Vec<Filter>),
    /// Represents a logical AND filter.
    And(Vec<Filter>),
    /// Represents an atomic filter.
    Atom(FilterAtom),
}

impl Display for Filter {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        let items_to_str = |items: &Vec<Filter>, sep: &str| {
            items
                .iter()
                .map(|f| f.to_string())
                .collect::<Vec<_>>()
                .join(sep)
        };

        match self {
            Filter::Or(filters) => write!(f, "{}", items_to_str(filters, " || ")),
            Filter::And(filters) => write!(f, "{}", items_to_str(filters, " && ")),
            Filter::Atom(atom) => write!(f, "{}", atom),
        }
    }
}

/// Enum representing different types of atomic filters in a JSONPath query.
#[derive(Debug, Clone, PartialEq)]
pub enum FilterAtom {
    /// Represents a nested filter with an optional NOT flag.
    Filter { expr: Box<Filter>, not: bool },
    /// Represents a test filter with an optional NOT flag.
    Test { expr: Box<Test>, not: bool },
    /// Represents a comparison filter.
    Comparison(Box<Comparison>),
}

impl FilterAtom {
    pub fn filter(expr: Filter, not: bool) -> Self {
        FilterAtom::Filter {
            expr: Box::new(expr),
            not,
        }
    }

    pub fn test(expr: Test, not: bool) -> Self {
        FilterAtom::Test {
            expr: Box::new(expr),
            not,
        }
    }

    pub fn cmp(cmp: Box<Comparison>) -> Self {
        FilterAtom::Comparison(cmp)
    }
}

impl Display for FilterAtom {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            FilterAtom::Filter { expr, not } => {
                if *not {
                    write!(f, "!{}", expr)
                } else {
                    write!(f, "{}", expr)
                }
            }
            FilterAtom::Test { expr, not } => {
                if *not {
                    write!(f, "!{}", expr)
                } else {
                    write!(f, "{}", expr)
                }
            }
            FilterAtom::Comparison(cmp) => write!(f, "{}", cmp),
        }
    }
}
/// Enum representing different types of comparisons in a JSONPath query.
#[derive(Debug, Clone, PartialEq)]
pub enum Comparison {
    /// Represents an equality comparison.
    Eq(Comparable, Comparable),
    /// Represents a non-equality comparison.
    Ne(Comparable, Comparable),
    /// Represents a greater-than comparison.
    Gt(Comparable, Comparable),
    /// Represents a greater-than-or-equal-to comparison.
    Gte(Comparable, Comparable),
    /// Represents a less-than comparison.
    Lt(Comparable, Comparable),
    /// Represents a less-than-or-equal-to comparison.
    Lte(Comparable, Comparable),
}

impl Comparison {
    pub fn try_new(op: &str, left: Comparable, right: Comparable) -> Parsed<Self> {
        match op {
            "==" => Ok(Comparison::Eq(left, right)),
            "!=" => Ok(Comparison::Ne(left, right)),
            ">" => Ok(Comparison::Gt(left, right)),
            ">=" => Ok(Comparison::Gte(left, right)),
            "<" => Ok(Comparison::Lt(left, right)),
            "<=" => Ok(Comparison::Lte(left, right)),
            _ => Err(JsonPathError::InvalidJsonPath(format!(
                "Invalid comparison operator: {}",
                op
            ))),
        }
    }

    pub fn vals(&self) -> (&Comparable, &Comparable) {
        match self {
            Comparison::Eq(left, right) => (left, right),
            Comparison::Ne(left, right) => (left, right),
            Comparison::Gt(left, right) => (left, right),
            Comparison::Gte(left, right) => (left, right),
            Comparison::Lt(left, right) => (left, right),
            Comparison::Lte(left, right) => (left, right),
        }
    }
}

impl Display for Comparison {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            Comparison::Eq(left, right) => write!(f, "{} == {}", left, right),
            Comparison::Ne(left, right) => write!(f, "{} != {}", left, right),
            Comparison::Gt(left, right) => write!(f, "{} > {}", left, right),
            Comparison::Gte(left, right) => write!(f, "{} >= {}", left, right),
            Comparison::Lt(left, right) => write!(f, "{} < {}", left, right),
            Comparison::Lte(left, right) => write!(f, "{} <= {}", left, right),
        }
    }
}

/// Enum representing different types of comparable values in a JSONPath query.
#[derive(Debug, Clone, PartialEq)]
pub enum Comparable {
    /// Represents a literal value.
    Literal(Literal),
    /// Represents a function.
    Function(TestFunction),
    /// Represents a singular query.
    SingularQuery(SingularQuery),
}

impl Display for Comparable {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            Comparable::Literal(literal) => write!(f, "{}", literal),
            Comparable::Function(func) => write!(f, "{}", func),
            Comparable::SingularQuery(query) => write!(f, "{}", query),
        }
    }
}

/// Enum representing different types of singular queries in a JSONPath query.
#[derive(Debug, Clone, PartialEq)]
pub enum SingularQuery {
    /// Represents a current node query.
    Current(Vec<SingularQuerySegment>),
    /// Represents a root node query.
    Root(Vec<SingularQuerySegment>),
}

impl Display for SingularQuery {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            SingularQuery::Current(segments) => write!(
                f,
                "@.{}",
                segments.iter().map(|s| s.to_string()).collect::<String>()
            ),
            SingularQuery::Root(segments) => write!(
                f,
                "$.{}",
                segments.iter().map(|s| s.to_string()).collect::<String>()
            ),
        }
    }
}

/// Enum representing different types of singular query segments in a JSONPath query.
#[derive(Debug, Clone, PartialEq)]
pub enum SingularQuerySegment {
    /// Represents an index segment.
    Index(i64),
    /// Represents a name segment.
    Name(String),
}

impl Display for SingularQuerySegment {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            SingularQuerySegment::Index(index) => write!(f, "{}", index),
            SingularQuerySegment::Name(name) => write!(f, "{}", name),
        }
    }
}

/// Enum representing different types of tests in a JSONPath query.
#[derive(Debug, Clone, PartialEq)]
pub enum Test {
    /// Represents a relative query.
    RelQuery(Vec<Segment>),
    /// Represents an absolute query.
    AbsQuery(JpQuery),
    /// Represents a function test.
    Function(Box<TestFunction>),
}

impl Test {
    pub fn is_res_bool(&self) -> bool {
        match self {
            Test::RelQuery(_) => false,
            Test::AbsQuery(_) => false,
            Test::Function(func) => match **func {
                TestFunction::Length(_) => false,
                TestFunction::Value(_) => false,
                TestFunction::Count(_) => false,
                TestFunction::Custom(_, _) => true,
                TestFunction::Search(_, _) => true,
                TestFunction::Match(_, _) => true,
            },
        }
    }
}

impl Display for Test {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            Test::RelQuery(segments) => write!(
                f,
                "{}",
                segments.iter().map(|s| s.to_string()).collect::<String>()
            ),
            Test::AbsQuery(query) => write!(f, "{}", query),
            Test::Function(func) => write!(f, "{}", func),
        }
    }
}

/// Enum representing different types of test functions in a JSONPath query.
#[derive(Debug, Clone, PartialEq)]
pub enum TestFunction {
    /// Represents a custom function.
    Custom(String, Vec<FnArg>),
    /// Represents a length function.
    Length(Box<FnArg>),
    /// Represents a value function.
    Value(FnArg),
    /// Represents a count function.
    Count(FnArg),
    /// Represents a search function.
    Search(FnArg, FnArg),
    /// Represents a match function.
    Match(FnArg, FnArg),
}

impl TestFunction {
    pub fn try_new(name: &str, args: Vec<FnArg>) -> Parsed<Self> {
        fn with_node_type_validation<'a>(
            a: &'a FnArg,
            name: &str,
        ) -> Result<&'a FnArg, JsonPathError> {
            if a.is_lit() {
                Err(JsonPathError::InvalidJsonPath(format!(
                    "Invalid argument for the function `{}`: expected a node, got a literal",
                    name
                )))
            } else if a.is_filter() {
                Err(JsonPathError::InvalidJsonPath(format!(
                    "Invalid argument for the function `{}`: expected a node, got a filter",
                    name
                )))
            } else {
                Ok(a)
            }
        }

        match (name, args.as_slice()) {
            ("length", [a]) => Ok(TestFunction::Length(Box::new(a.clone()))),
            ("value", [a]) => Ok(TestFunction::Value(a.clone())),
            ("count", [a]) => Ok(TestFunction::Count(
                with_node_type_validation(a, name)?.clone(),
            )),
            ("search", [a, b]) => Ok(TestFunction::Search(a.clone(), b.clone())),
            ("match", [a, b]) => Ok(TestFunction::Match(a.clone(), b.clone())),
            ("length" | "value" | "count" | "match" | "search", args) => {
                Err(JsonPathError::InvalidJsonPath(format!(
                    "Invalid number of arguments for the function `{}`: got {}",
                    name,
                    args.len()
                )))
            }
            (custom, _) => Ok(TestFunction::Custom(custom.to_string(), args)),
        }
    }

    pub fn is_comparable(&self) -> bool {
        match self {
            TestFunction::Length(_) => true,
            TestFunction::Value(_) => true,
            TestFunction::Count(_) => true,
            TestFunction::Custom(_, _) => false,
            TestFunction::Search(_, _) => false,
            TestFunction::Match(_, _) => false,
        }
    }
}

impl Display for TestFunction {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            TestFunction::Custom(name, args) => write!(
                f,
                "{}({})",
                name,
                args.iter().map(|a| a.to_string()).collect::<String>()
            ),
            TestFunction::Length(arg) => write!(f, "length({})", arg),
            TestFunction::Value(arg) => write!(f, "value({})", arg),
            TestFunction::Count(arg) => write!(f, "count({})", arg),
            TestFunction::Search(arg1, arg2) => write!(f, "search({}, {})", arg1, arg2),
            TestFunction::Match(arg1, arg2) => write!(f, "match({}, {})", arg1, arg2),
        }
    }
}

/// Enum representing different types of function arguments in a JSONPath query.
#[derive(Debug, Clone, PartialEq)]
pub enum FnArg {
    /// Represents a literal argument.
    Literal(Literal),
    /// Represents a test argument.
    Test(Box<Test>),
    /// Represents a filter argument.
    Filter(Filter),
}

impl FnArg {
    pub fn is_lit(&self) -> bool {
        matches!(self, FnArg::Literal(_))
    }
    pub fn is_filter(&self) -> bool {
        matches!(self, FnArg::Filter(_))
    }
}

impl Display for FnArg {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            FnArg::Literal(literal) => write!(f, "{}", literal),
            FnArg::Test(test) => write!(f, "{}", test),
            FnArg::Filter(filter) => write!(f, "{}", filter),
        }
    }
}

/// Enum representing different types of literal values in a JSONPath query.
#[derive(Debug, Clone, PartialEq)]
pub enum Literal {
    /// Represents an integer literal.
    Int(i64),
    /// Represents a floating-point literal.
    Float(f64),
    /// Represents a string literal.
    String(String),
    /// Represents a boolean literal.
    Bool(bool),
    /// Represents a null literal.
    Null,
}

impl Display for Literal {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        match self {
            Literal::Int(val) => write!(f, "{}", val),
            Literal::Float(val) => write!(f, "{}", val),
            Literal::String(val) => write!(f, "\"{}\"", val),
            Literal::Bool(val) => write!(f, "{}", val),
            Literal::Null => write!(f, "null"),
        }
    }
}