polserver/pol-core/bscript/compiler/ast/Node.cpp

#include "Node.h"

#include <format/format.h>

namespace Pol::Bscript::Compiler
{
Node::Node( const SourceLocation& source_location, NodeVector children )
  : children( std::move( children ) ), source_location( source_location )
{
}

Node::Node( const SourceLocation& source_location, std::unique_ptr<Node> child )
  : source_location( source_location )
{
  children.reserve( 1 );
  children.push_back( std::move( child ) );
}

Node::Node( const SourceLocation& source_location ) : children(), source_location( source_location )
{
}

std::string Node::describe() const
{
  fmt::Writer w;
  describe_to( w );
  return w.str();
}

std::string Node::to_string_tree() const
{
  fmt::Writer w;
  w << *this;
  return w.str();
}

void Node::debug( const std::string& msg ) const
{
  source_location.debug( msg );
}

void Node::internal_error( const std::string& msg ) const
{
  source_location.internal_error( msg );
}

void Node::describe_tree_to_indented( fmt::Writer& w, const Node& node, unsigned indent )
{
  w << std::string( indent * 2, ' ' ) << "- ";
  node.describe_to( w );
  w << "\n";
  for ( const auto& child : node.children )
  {
    if ( child )
      describe_tree_to_indented( w, *child, indent + 1 );
    else
      w << std::string( ( indent + 1 ) * 2, ' ' ) << "- [deleted]\n";
  }
}

fmt::Writer& operator<<( fmt::Writer& w, const Node& node )
{
  node.describe_tree_to_indented( w, node, 0 );
  return w;
}

}  // namespace Pol::Bscript::Compiler