gux/lang/src/node.c

#include "node.h"
#include "vec.h"

GUX_ENUM_C(NodeType, NODE_TYPE);

void node_init(struct node* self, enum NodeType type,
               char const* value, int line)
{
  assert(self);
  self->type = type;
  self->value = strdup(value);
  vec_init(&self->children, 1);
  self->line = line;
}

void node_free(struct node* self)
{
  assert(self);
  free(self->value);

  for (size_t i=0; i<self->children.size; i++)
    {
      node_free(self->children.data[i]);
    }

  vec_free_elements(&self->children);
  vec_free(&self->children);
}

struct node* node_add_new_child(struct node* self,
                                enum NodeType type, char const* value)
{
  assert(self);
  struct node* child = malloc(sizeof(struct node));
  node_init(child, type, value, self->line);

  vec_push(&self->children, child);

  return child;
}

struct node* node_add_child(struct node* self, struct node* child)
{
  assert(self);
  vec_push(&self->children, child);
  return child;
}


size_t node_str(struct node* self, char* buffer, size_t size)
{
  assert(self);
  assert(buffer);

  size_t sz = 0;

  sz += snprintf(buffer + sz, size - sz, "%s",
                 NodeTypeStr[self->type] + strlen("NODE_"));

  if (strcmp(self->value, "") != 0)
    {
      sz += snprintf(buffer + sz, size - sz, "[%s]", self->value);
    }

  if (self->children.size > 0)
    {
      sz += snprintf(buffer + sz, size - sz, "(");

      for (size_t i=0; i<self->children.size; i++)
        {
          if (i > 0)
            {
              sz += snprintf(buffer + sz, size - sz, ",");
            }

          sz += node_str(self->children.data[i], buffer + sz, size - sz);
        }

      sz += snprintf(buffer + sz, size - sz, ")");
    }

  return sz;
}