teo-parser 0.3.0

use crate::ast::handler::{HandlerDeclaration, HandlerGroupDeclaration, HandlerInputFormat};
use crate::{parse_append, parse_container_node_variables, parse_container_node_variables_cleanup, parse_insert, parse_insert_keyword, parse_insert_punctuation, parse_set, parse_set_identifier_and_string_path, parse_set_optional};
use crate::parser::parse_code_comment::parse_code_comment;
use crate::parser::parse_doc_comment::parse_doc_comment;
use crate::parser::parse_decorator::parse_decorator;
use crate::parser::parse_empty_decorator::parse_empty_decorator;
use crate::parser::parse_span::parse_span;
use crate::parser::parse_type_expression::parse_type_expression;
use crate::parser::parser_context::ParserContext;
use crate::parser::pest_parser::{Pair, Rule};

pub(super) fn parse_handler_group_declaration(pair: Pair<'_>, context: &ParserContext) -> HandlerGroupDeclaration {
    let (
        span,
        path,
        mut string_path,
        mut children,
        define_availability,
        actual_availability
    ) = parse_container_node_variables!(pair, context, named, availability);
    let mut comment = None;
    let mut identifier = 0;
    let mut handler_declarations = vec![];
    let mut inside_block = false;
    let mut empty_decorators = vec![];
    let mut unattached_decorators = vec![];
    let mut decorators = vec![];
    for current in pair.into_inner() {
        match current.as_rule() {
            Rule::DECLARE_KEYWORD => parse_insert_keyword!(context, current, children, "declare"),
            Rule::HANDLER_KEYWORD => parse_insert_keyword!(context, current, children, "handler"),
            Rule::GROUP_KEYWORD => parse_insert_keyword!(context, current, children, "group"),
            Rule::COLON => parse_insert_punctuation!(context, current, children, ":"),
            Rule::BLOCK_OPEN => {
                parse_insert_punctuation!(context, current, children, "{");
                inside_block = true;
            },
            Rule::BLOCK_CLOSE => parse_insert_punctuation!(context, current, children, "}"),
            Rule::triple_comment_block => if !inside_block {
                parse_set_optional!(parse_doc_comment(current, context), children, comment)
            } else {
                context.insert_unattached_doc_comment(parse_span(&current));
                parse_append!(parse_doc_comment(current, context), children);
            },
            Rule::double_comment_block => parse_append!(parse_code_comment(current, context), children),
            Rule::identifier => parse_set_identifier_and_string_path!(context, current, children, identifier, string_path),
            Rule::handler_declaration => parse_insert!(parse_handler_declaration(current, context, true), children, handler_declarations),
            Rule::decorator => if inside_block {
                parse_insert!(parse_decorator(current, context), children, unattached_decorators)
            } else {
                parse_insert!(parse_decorator(current, context), children, decorators)
            },
            Rule::empty_decorator => parse_insert!(parse_empty_decorator(current, context), children, empty_decorators),
            _ => context.insert_unparsed(parse_span(&current)),
        }
    }
    parse_container_node_variables_cleanup!(context, named);
    HandlerGroupDeclaration {
        span,
        path,
        string_path,
        children,
        define_availability,
        actual_availability,
        comment,
        identifier,
        handler_declarations,
        empty_decorators,
        unattached_decorators,
        decorators,
    }
}

pub(super) fn parse_handler_declaration(pair: Pair<'_>, context: &ParserContext, inside_group: bool) -> HandlerDeclaration {
    let (
        span,
        path,
        mut string_path,
        mut children,
        define_availability,
        actual_availability
    ) = parse_container_node_variables!(pair, context, named, availability);
    let mut comment = None;
    let mut decorators = vec![];
    let mut identifier = 0;
    let mut input_type = None;
    let mut output_type = 0;
    let mut input_format: HandlerInputFormat = HandlerInputFormat::Json;
    let mut inside_block = false;
    let mut inside_paren = false;
    let mut empty_decorators = vec![];
    let mut nonapi = false;
    for current in pair.into_inner() {
        match current.as_rule() {
            Rule::BLOCK_OPEN => {
                parse_insert_punctuation!(context, current, children, "{");
                inside_block = true;
            },
            Rule::BLOCK_CLOSE => parse_insert_punctuation!(context, current, children, "}"),
            Rule::DECLARE_KEYWORD => parse_insert_keyword!(context, current, children, "declare"),
            Rule::NONAPI_KEYWORD => {
                nonapi = true;
                parse_insert_keyword!(context, current, children, "nonapi");
            },
            Rule::HANDLER_KEYWORD => parse_insert_keyword!(context, current, children, "handler"),
            Rule::triple_comment_block => if !inside_block {
                parse_set_optional!(parse_doc_comment(current, context), children, comment)
            } else {
                context.insert_unattached_doc_comment(parse_span(&current));
                parse_append!(parse_doc_comment(current, context), children);
            },
            Rule::double_comment_block => parse_append!(parse_code_comment(current, context), children),
            Rule::identifier => parse_set_identifier_and_string_path!(context, current, children, identifier, string_path),
            Rule::type_expression => if !inside_paren {
                parse_set!(parse_type_expression(current, context), children, output_type);
            } else {
                parse_set_optional!(parse_type_expression(current, context), children, input_type);
            },
            Rule::PAREN_OPEN => {
                inside_paren = true;
                parse_insert_punctuation!(context, current, children, "(");
            },
            Rule::PAREN_CLOSE => {
                inside_paren = false;
                parse_insert_punctuation!(context, current, children, ")");
            },
            Rule::COLON => parse_insert_punctuation!(context, current, children, ":"),
            Rule::req_type => if current.as_str() == "form" {
                input_format = HandlerInputFormat::Form;
                parse_insert_keyword!(context, current, children, "form")
            } else {
                parse_insert_keyword!(context, current, children, "json")
            },
            Rule::decorator => parse_insert!(parse_decorator(current, context), children, decorators),
            Rule::empty_decorator => parse_insert!(parse_empty_decorator(current, context), children, empty_decorators),
            _ => context.insert_unparsed(parse_span(&current)),
        }
    }
    parse_container_node_variables_cleanup!(context, named);
    HandlerDeclaration {
        span,
        path,
        string_path,
        children,
        define_availability,
        actual_availability,
        comment,
        decorators,
        empty_decorators,
        identifier,
        input_type,
        output_type,
        input_format,
        nonapi,
        inside_group,
    }
}