use std::cell::RefCell;
use crate::ast::model::Model;
use crate::{parse_append, parse_container_node_variables, parse_container_node_variables_cleanup, parse_insert, parse_insert_keyword, parse_insert_punctuation, parse_set_identifier_and_string_path, parse_set_optional};
use crate::parser::parse_availability_end::parse_availability_end;
use crate::parser::parse_availability_flag::parse_availability_flag;
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_field::parse_field;
use crate::parser::parse_handler_group::parse_handler_declaration;
use crate::parser::parse_include_handler_from_template::parse_include_handler_from_template;
use crate::parser::parse_partial_field::parse_partial_field;
use crate::parser::parse_span::parse_span;
use crate::parser::parser_context::ParserContext;
use crate::parser::pest_parser::{Pair, Rule};
pub(super) fn parse_model_declaration(pair: Pair<'_>, context: &ParserContext) -> Model {
let (
span,
path,
mut string_path,
mut children,
define_availability,
actual_availability
) = parse_container_node_variables!(pair, context, named, availability);
let mut inside_block = false;
let mut comment = None;
let mut decorators = vec![];
let mut empty_decorator_spans = vec![];
let mut empty_field_decorator_spans = vec![];
let mut unattached_field_decorators = vec![];
let mut identifier = 0;
let mut fields = vec![];
let mut partial_fields = vec![];
let mut handlers = vec![];
let mut handler_inclusions = vec![];
for current in pair.into_inner() {
match current.as_rule() {
Rule::MODEL_KEYWORD => parse_insert_keyword!(context, current, children, "model"),
Rule::BLOCK_CLOSE => parse_insert_punctuation!(context, current, children, "}"),
Rule::BLOCK_OPEN => {
parse_insert_punctuation!(context, current, children, "{");
inside_block = true;
},
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(¤t));
parse_append!(parse_doc_comment(current, context), children);
},
Rule::double_comment_block => parse_append!(parse_code_comment(current, context), children),
Rule::decorator => if inside_block {
unattached_field_decorators.push(parse_decorator(current, context));
} else {
parse_insert!(parse_decorator(current, context), children, decorators);
},
Rule::empty_decorator => if inside_block {
empty_field_decorator_spans.push(parse_span(¤t));
} else {
empty_decorator_spans.push(parse_span(¤t));
},
Rule::identifier => parse_set_identifier_and_string_path!(context, current, children, identifier, string_path),
Rule::field_declaration => parse_insert!(parse_field(current, context), children, fields),
Rule::partial_field => parse_insert!(parse_partial_field(current, context), children, partial_fields),
Rule::handler_declaration => parse_insert!(parse_handler_declaration(current, context, true), children, handlers),
Rule::availability_start => parse_append!(parse_availability_flag(current, context), children),
Rule::availability_end => parse_append!(parse_availability_end(current, context), children),
Rule::include_handler_from_template => parse_insert!(parse_include_handler_from_template(current, context), children, handler_inclusions),
_ => context.insert_unparsed(parse_span(¤t)),
}
}
parse_container_node_variables_cleanup!(context, named);
Model {
span,
path,
string_path,
children,
define_availability,
actual_availability,
comment,
identifier,
fields,
partial_fields,
decorators,
empty_decorator_spans,
empty_field_decorator_spans,
unattached_field_decorators,
handlers,
handler_inclusions,
resolved: RefCell::new(None),
}
}