ASTNodeJumpPlacementChecker.cpp

#include <ASTNodeJumpPlacementChecker.hpp>

#include <PEGGrammar.hpp>

#include <vector>

void
ASTNodeJumpPlacementChecker::_checkJumpPlacement(ASTNode& n, bool is_inside_loop)
{
  if (n.is<language::for_statement>() or n.is<language::do_while_statement>() or n.is<language::while_statement>()) {
    for (auto& child : n.children) {
      this->_checkJumpPlacement(*child, true);
    }
  } else if (n.is<language::break_kw>() or n.is<language::continue_kw>()) {
    if (not is_inside_loop) {
      std::ostringstream error_message;
      error_message << "unexpected '" << rang::fgB::red << n.string() << rang::fg::reset
                    << "' outside of loop or switch statement";
      throw parse_error(error_message.str(), std::vector{n.begin()});
    }
  } else {
    for (auto& child : n.children) {
      this->_checkJumpPlacement(*child, is_inside_loop);
    }
  }
}

ASTNodeJumpPlacementChecker::ASTNodeJumpPlacementChecker(ASTNode& n)
{
  Assert(n.is_root());
  this->_checkJumpPlacement(n, false);
}