typeql 3.8.3-rc0

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

use crate::{
    common::{error::TypeQLError, Spanned},
    parser::{IntoChildNodes, Node, Rule, RuleMatcher},
    value::{
        BooleanLiteral, DateFragment, DateLiteral, DateTimeLiteral, DateTimeTZLiteral, DurationDate, DurationLiteral,
        DurationTime, IntegerLiteral, Literal, NumericLiteral, Sign, SignedDecimalLiteral, SignedDoubleLiteral,
        SignedIntegerLiteral, StringLiteral, TimeFragment, TimeZone, ValueLiteral,
    },
};

pub(super) fn visit_value_literal(node: Node<'_>) -> Literal {
    debug_assert_eq!(node.as_rule(), Rule::value_literal);
    let span = node.span();
    let child = node.into_child();
    let value_literal = match child.as_rule() {
        Rule::quoted_string_literal => ValueLiteral::String(visit_quoted_string_literal(child)),
        Rule::boolean_literal => ValueLiteral::Boolean(BooleanLiteral { value: child.as_str().to_owned() }),
        Rule::signed_integer => ValueLiteral::Integer(visit_signed_integer(child)),
        Rule::signed_decimal => ValueLiteral::Decimal(visit_signed_decimal(child)),
        Rule::signed_double => ValueLiteral::Double(visit_signed_double(child)),

        Rule::datetime_tz_literal => ValueLiteral::DateTimeTz(visit_datetime_tz_literal(child)),
        Rule::datetime_literal => ValueLiteral::DateTime(visit_datetime_literal(child)),
        Rule::date_literal => ValueLiteral::Date(visit_date_literal(child)),
        Rule::duration_literal => ValueLiteral::Duration(visit_duration_literal(child)),

        _ => unreachable!("{}", TypeQLError::IllegalGrammar { input: child.as_str().to_owned() }),
    };
    Literal::new(span, value_literal)
}

fn visit_sign(node: Node<'_>) -> Sign {
    debug_assert_eq!(node.as_rule(), Rule::sign);
    let child = node.into_child();
    match child.as_rule() {
        Rule::PLUS => Sign::Plus,
        Rule::MINUS => Sign::Minus,
        _ => unreachable!("{}", TypeQLError::IllegalGrammar { input: child.as_str().to_owned() }),
    }
}

pub(super) fn visit_integer_literal(node: Node<'_>) -> IntegerLiteral {
    debug_assert_eq!(node.as_rule(), Rule::integer_literal);
    IntegerLiteral { value: node.as_str().to_owned() }
}

pub(super) fn visit_numeric_literal(node: Node<'_>) -> NumericLiteral {
    debug_assert_eq!(node.as_rule(), Rule::numeric_literal);
    NumericLiteral { value: node.as_str().to_owned() }
}

pub(super) fn visit_quoted_string_literal(node: Node<'_>) -> StringLiteral {
    debug_assert_eq!(node.as_rule(), Rule::quoted_string_literal);
    StringLiteral { value: node.as_str().to_owned() }
}

fn visit_signed_integer(node: Node<'_>) -> SignedIntegerLiteral {
    debug_assert_eq!(node.as_rule(), Rule::signed_integer);
    let mut children = node.into_children();
    let first_node = children.consume_any();
    let (sign, integral) = match first_node.as_rule() {
        Rule::sign => (
            Some(visit_sign(first_node)),
            visit_integer_literal(children.consume_expected(Rule::integer_literal)).value,
        ),
        Rule::integer_literal => (None, visit_integer_literal(first_node).value),
        _ => unreachable!("{}", TypeQLError::IllegalGrammar { input: first_node.as_str().to_owned() }),
    };
    debug_assert_eq!(children.try_consume_any(), None);
    SignedIntegerLiteral { sign, integral }
}

fn visit_signed_decimal(node: Node<'_>) -> SignedDecimalLiteral {
    debug_assert_eq!(node.as_rule(), Rule::signed_decimal);
    let mut children = node.into_children();
    let first_node = children.consume_any();
    let (sign, decimal_with_suffix) = match first_node.as_rule() {
        Rule::sign => (Some(visit_sign(first_node)), children.consume_expected(Rule::decimal_literal).as_str()),
        Rule::decimal_literal => (None, first_node.as_str()),
        _ => unreachable!("{}", TypeQLError::IllegalGrammar { input: first_node.as_str().to_owned() }),
    };
    debug_assert_eq!(children.try_consume_any(), None);
    let decimal = decimal_with_suffix.trim_end_matches("dec").to_owned();
    SignedDecimalLiteral { sign, decimal }
}

fn visit_signed_double(node: Node<'_>) -> SignedDoubleLiteral {
    debug_assert_eq!(node.as_rule(), Rule::signed_double);
    let mut children = node.into_children();
    let first_node = children.consume_any();
    let (sign, double) = match first_node.as_rule() {
        Rule::sign => {
            (Some(visit_sign(first_node)), children.consume_expected(Rule::double_literal).as_str().to_owned())
        }
        Rule::double_literal => (None, first_node.as_str().to_owned()),
        _ => unreachable!("{}", TypeQLError::IllegalGrammar { input: first_node.as_str().to_owned() }),
    };
    debug_assert_eq!(children.try_consume_any(), None);
    SignedDoubleLiteral { sign, double }
}

fn visit_datetime_tz_literal(node: Node<'_>) -> DateTimeTZLiteral {
    debug_assert_eq!(node.as_rule(), Rule::datetime_tz_literal);
    let mut children = node.into_children();
    let date = visit_date_fragment(children.consume_expected(Rule::date_fragment));
    let time = visit_time(children.consume_expected(Rule::time));
    let tz_node = children.consume_any();
    let timezone = match tz_node.as_rule() {
        Rule::iana_timezone => TimeZone::IANA(tz_node.as_str().to_owned()),
        Rule::iso8601_timezone_offset => TimeZone::ISO(tz_node.as_str().to_owned()),
        _ => unreachable!("{}", TypeQLError::IllegalGrammar { input: tz_node.as_str().to_owned() }),
    };
    debug_assert_eq!(children.try_consume_any(), None);
    DateTimeTZLiteral { date, time, timezone }
}

fn visit_datetime_literal(node: Node<'_>) -> DateTimeLiteral {
    debug_assert_eq!(node.as_rule(), Rule::datetime_literal);
    let mut children = node.into_children();
    let date = visit_date_fragment(children.consume_expected(Rule::date_fragment));
    let time = visit_time(children.consume_expected(Rule::time));
    debug_assert_eq!(children.try_consume_any(), None);
    DateTimeLiteral { date, time }
}

fn visit_date_literal(node: Node<'_>) -> DateLiteral {
    debug_assert_eq!(node.as_rule(), Rule::date_literal);
    let date = visit_date_fragment(node.into_child());
    DateLiteral { date }
}

fn visit_date_fragment(node: Node<'_>) -> DateFragment {
    debug_assert_eq!(node.as_rule(), Rule::date_fragment);
    let mut children = node.into_children();
    let year = children.consume_expected(Rule::year).as_str().to_owned();
    let month = children.consume_expected(Rule::month).as_str().to_owned();
    let day = children.consume_expected(Rule::day).as_str().to_owned();
    debug_assert_eq!(children.try_consume_any(), None);
    DateFragment { year, month, day }
}

fn visit_time(node: Node<'_>) -> TimeFragment {
    debug_assert_eq!(node.as_rule(), Rule::time);
    let mut children = node.into_children();
    let hour = children.consume_expected(Rule::hour).as_str().to_owned();
    let minute = children.consume_expected(Rule::minute).as_str().to_owned();
    let second = children.try_consume_expected(Rule::second).map(|node| node.as_str().to_owned());
    let second_fraction = children.try_consume_expected(Rule::second_fraction).map(|node| node.as_str().to_owned());
    debug_assert_eq!(children.try_consume_any(), None);
    TimeFragment { hour, minute, second, second_fraction }
}

fn visit_duration_literal(node: Node<'_>) -> DurationLiteral {
    debug_assert_eq!(node.as_rule(), Rule::duration_literal);
    let mut children = node.into_children();
    let child = children.consume_any();
    let duration = match child.as_rule() {
        Rule::duration_weeks => DurationLiteral::Weeks(visit_duration_weeks(child)),
        Rule::duration_date => {
            let date = visit_duration_date(child);
            let time = children.try_consume_expected(Rule::duration_time).map(visit_duration_time);
            DurationLiteral::DateAndTime(date, time)
        }
        Rule::duration_time => {
            let time = visit_duration_time(child);
            DurationLiteral::Time(time)
        }
        _ => unreachable!("{}", TypeQLError::IllegalGrammar { input: child.as_str().to_owned() }),
    };
    debug_assert_eq!(children.try_consume_any(), None);
    duration
}

fn visit_duration_weeks(node: Node<'_>) -> IntegerLiteral {
    debug_assert_eq!(node.as_rule(), Rule::duration_weeks);
    visit_integer_literal(node.into_child())
}

fn visit_duration_date(node: Node<'_>) -> DurationDate {
    debug_assert_eq!(node.as_rule(), Rule::duration_date);
    let mut children = node.into_children();
    let years = children.try_consume_expected(Rule::duration_years).map(visit_duration_years);
    let months = children.try_consume_expected(Rule::duration_months).map(visit_duration_months);
    let days = children.try_consume_expected(Rule::duration_days).map(visit_duration_days);
    debug_assert_eq!(children.try_consume_any(), None);
    DurationDate { years, months, days }
}

fn visit_duration_years(node: Node<'_>) -> IntegerLiteral {
    debug_assert_eq!(node.as_rule(), Rule::duration_years);
    visit_integer_literal(node.into_child())
}

fn visit_duration_months(node: Node<'_>) -> IntegerLiteral {
    debug_assert_eq!(node.as_rule(), Rule::duration_months);
    visit_integer_literal(node.into_child())
}

fn visit_duration_days(node: Node<'_>) -> IntegerLiteral {
    debug_assert_eq!(node.as_rule(), Rule::duration_days);
    visit_integer_literal(node.into_child())
}

fn visit_duration_time(node: Node<'_>) -> DurationTime {
    debug_assert_eq!(node.as_rule(), Rule::duration_time);
    let mut children = node.into_children();
    let hours = children.try_consume_expected(Rule::duration_hours).map(visit_duration_hours);
    let minutes = children.try_consume_expected(Rule::duration_minutes).map(visit_duration_minutes);
    let seconds = children.try_consume_expected(Rule::duration_seconds).map(visit_duration_seconds);
    debug_assert_eq!(children.try_consume_any(), None);
    DurationTime { hours, minutes, seconds }
}

fn visit_duration_hours(node: Node<'_>) -> IntegerLiteral {
    debug_assert_eq!(node.as_rule(), Rule::duration_hours);
    visit_integer_literal(node.into_child())
}

fn visit_duration_minutes(node: Node<'_>) -> IntegerLiteral {
    debug_assert_eq!(node.as_rule(), Rule::duration_minutes);
    visit_integer_literal(node.into_child())
}

fn visit_duration_seconds(node: Node<'_>) -> NumericLiteral {
    debug_assert_eq!(node.as_rule(), Rule::duration_seconds);
    visit_numeric_literal(node.into_child())
}