@herb-tools/node 0.8.10 → 0.9.0

package/extension/libherb/ast_node.c

@@ -2,8 +2,9 @@
 #include "include/ast_nodes.h"
 #include "include/errors.h"
 #include "include/position.h"
-#include "include/token.h"
 #include "include/util.h"
+#include "include/util/hb_allocator.h"
+#include "include/util/hb_string.h"
 #include "include/visitor.h"
 
 #include <prism.h>
@@ -14,7 +15,14 @@ size_t ast_node_sizeof(void) {
   return sizeof(struct AST_NODE_STRUCT);
 }
 
-void ast_node_init(AST_NODE_T* node, const ast_node_type_T type, position_T start, position_T end, hb_array_T* errors) {
+void ast_node_init(
+  AST_NODE_T* node,
+  const ast_node_type_T type,
+  position_T start,
+  position_T end,
+  hb_array_T* errors,
+  hb_allocator_T* allocator
+) {
   if (!node) { return; }
 
   node->type = type;
@@ -22,18 +30,20 @@ void ast_node_init(AST_NODE_T* node, const ast_node_type_T type, position_T star
   node->location.end = end;
 
   if (errors == NULL) {
-    node->errors = hb_array_init(8);
+    node->errors = hb_array_init(0, allocator);
   } else {
     node->errors = errors;
   }
 }
 
-AST_LITERAL_NODE_T* ast_literal_node_init_from_token(const token_T* token) {
-  AST_LITERAL_NODE_T* literal = malloc(sizeof(AST_LITERAL_NODE_T));
+AST_LITERAL_NODE_T* ast_literal_node_init_from_token(const token_T* token, hb_allocator_T* allocator) {
+  AST_LITERAL_NODE_T* literal = hb_allocator_alloc(allocator, sizeof(AST_LITERAL_NODE_T));
 
-  ast_node_init(&literal->base, AST_LITERAL_NODE, token->location.start, token->location.end, NULL);
+  if (!literal) { return NULL; }
 
-  literal->content = herb_strdup(token->value);
+  ast_node_init(&literal->base, AST_LITERAL_NODE, token->location.start, token->location.end, NULL, allocator);
+
+  literal->content = hb_string_copy(token->value, allocator);
 
   return literal;
 }

package/extension/libherb/ast_node.h

@@ -5,19 +5,25 @@
 #include "errors.h"
 #include "position.h"
 #include "token_struct.h"
+#include "util/hb_allocator.h"
 
-void ast_node_init(AST_NODE_T* node, ast_node_type_T type, position_T start, position_T end, hb_array_T* errors);
-void ast_node_free(AST_NODE_T* node);
+void ast_node_init(
+  AST_NODE_T* node,
+  ast_node_type_T type,
+  position_T start,
+  position_T end,
+  hb_array_T* errors,
+  hb_allocator_T* allocator
+);
+void ast_node_free(AST_NODE_T* node, hb_allocator_T* allocator);
 
-AST_LITERAL_NODE_T* ast_literal_node_init_from_token(const token_T* token);
+AST_LITERAL_NODE_T* ast_literal_node_init_from_token(const token_T* token, hb_allocator_T* allocator);
 
 size_t ast_node_sizeof(void);
 size_t ast_node_child_count(AST_NODE_T* node);
 
 ast_node_type_T ast_node_type(const AST_NODE_T* node);
 
-char* ast_node_name(AST_NODE_T* node);
-
 void ast_node_set_start(AST_NODE_T* node, position_T position);
 void ast_node_set_end(AST_NODE_T* node, position_T position);