backyard-parser 0.1.10

Parse PHP code to AST node.
Documentation
use backyard_lexer::token::TokenType;
use backyard_nodes::{
  BodyType,
  DeclareArgumentNode,
  DeclareNode,
  Location,
  Node,
  utils::{ IntoBoxedNode, IntoBoxedOptionNode },
};

use crate::{
  error::ParserError,
  parser::{ LoopArgument, Parser },
  utils::{ match_pattern, Lookup, LookupResult },
};

use super::{ block::BlockParser, comment::CommentParser, identifier::IdentifierParser };

#[derive(Debug, Clone)]
pub struct DeclareParser;

impl DeclareParser {
  pub fn test<'arena, 'a>(
    parser: &mut Parser<'arena, 'a>,
    _: &mut LoopArgument
  ) -> Option<std::vec::Vec<LookupResult<'arena>>> {
    match_pattern(
      parser,
      &[Lookup::Equal(&[TokenType::Declare]), Lookup::Equal(&[TokenType::LeftParenthesis])]
    )
  }

  pub fn parse<'arena, 'a, 'b>(
    parser: &mut Parser<'arena, 'a>,
    matched: std::vec::Vec<LookupResult<'arena>>,
    start_loc: Location,
    _: &mut LoopArgument<'arena, 'b>
  ) -> Result<Node<'arena>, ParserError> {
    if let [_, _] = matched.as_slice() {
      let arguments = parser.get_children(
        &mut LoopArgument::new(
          parser.arena,
          "declare",
          &[TokenType::Comma],
          &[TokenType::RightParenthesis],
          &[
            (DeclareArgumentParser::test, DeclareArgumentParser::parse),
            (CommentParser::test, CommentParser::parse),
          ]
        )
      )?;
      let body_type: BodyType = if let Ok(close) = parser.get_token(parser.position) {
        match close.token_type {
          TokenType::Colon => BodyType::Short,
          TokenType::LeftCurlyBracket => BodyType::Basic,
          _ => BodyType::Empty,
        }
      } else {
        BodyType::Empty
      };
      let body = match body_type {
        BodyType::Empty => None,
        BodyType::Basic => Some(BlockParser::new_block(parser)?),
        BodyType::Short => Some(BlockParser::new_short(parser, &[TokenType::EndDeclare])?),
      };
      return Ok(
        DeclareNode::loc(
          arguments,
          body.into_boxed(parser.arena),
          body_type,
          parser.gen_loc(start_loc)
        )
      );
    }
    Err(ParserError::Internal)
  }
}

#[derive(Debug, Clone)]
pub struct DeclareArgumentParser;

impl DeclareArgumentParser {
  pub fn test<'arena, 'a>(
    parser: &mut Parser<'arena, 'a>,
    _: &mut LoopArgument
  ) -> Option<std::vec::Vec<LookupResult<'arena>>> {
    match_pattern(
      parser,
      &[Lookup::Equal(&[TokenType::UnqualifiedName]), Lookup::Equal(&[TokenType::Assignment])]
    )
  }

  pub fn parse<'arena, 'a, 'b>(
    parser: &mut Parser<'arena, 'a>,
    matched: std::vec::Vec<LookupResult<'arena>>,
    start_loc: Location,
    _: &mut LoopArgument<'arena, 'b>
  ) -> Result<Node<'arena>, ParserError> {
    if let [name, _] = matched.as_slice() {
      let name = IdentifierParser::from_token(name.as_equal(parser)?);
      if
        let Some(value) = parser.get_statement(
          &mut LoopArgument::with_tokens(
            parser.arena,
            "declare_argument",
            &[TokenType::Comma],
            &[TokenType::RightParenthesis]
          )
        )?
      {
        return Ok(
          DeclareArgumentNode::loc(
            name.into_boxed(parser.arena),
            value.into_boxed(parser.arena),
            parser.gen_loc(start_loc)
          )
        );
      }
    }
    Err(ParserError::Internal)
  }
}