From 91e3609fbfeb72f45677cacbcfcf96d36ade44b5 Mon Sep 17 00:00:00 2001
From: Max Brunsfeld <maxbrunsfeld@gmail.com>
Date: Sat, 2 Apr 2016 22:18:44 -0700
Subject: [PATCH] Write to file directly from stack debugging function

---
 src/runtime/parser.c |  5 +---
 src/runtime/stack.c  | 64 +++++++++++---------------------------------
 src/runtime/stack.h  |  3 ++-
 3 files changed, 19 insertions(+), 53 deletions(-)

diff --git a/src/runtime/parser.c b/src/runtime/parser.c
index 6e9fb9cb..8a23d6f1 100644
--- a/src/runtime/parser.c
+++ b/src/runtime/parser.c
@@ -28,11 +28,8 @@
 
 #define LOG_STACK()                                                  \
   if (self->print_debugging_graphs) {                                \
-    char *graph_string =                                             \
-      ts_stack_dot_graph(self->stack, self->language->symbol_names); \
-    fputs(graph_string, stderr);                                     \
+    ts_stack_print_dot_graph(self->stack, self->language->symbol_names, stderr); \
     fputs("\n\n", stderr);                                           \
-    ts_free(graph_string);                                           \
   }
 
 #define SYM_NAME(symbol) ts_language_symbol_name(self->language, symbol)
diff --git a/src/runtime/stack.c b/src/runtime/stack.c
index ec935845..af3e1ee7 100644
--- a/src/runtime/stack.c
+++ b/src/runtime/stack.c
@@ -548,12 +548,9 @@ void ts_stack_delete(Stack *self) {
   ts_free(self);
 }
 
-size_t ts_stack__write_dot_graph(Stack *self, char *string, size_t n,
-                                 const char **symbol_names) {
-  char *cursor = string;
-  char **s = n > 0 ? &cursor : &string;
-  cursor += snprintf(*s, n, "digraph stack {\n");
-  cursor += snprintf(*s, n, "rankdir=\"RL\";\n");
+int ts_stack_print_dot_graph(Stack *self, const char **symbol_names, FILE *f) {
+  fprintf(f, "digraph stack {\n");
+  fprintf(f, "rankdir=\"RL\";\n");
 
   Array(StackNode *)visited_nodes;
   array_init(&visited_nodes);
@@ -563,7 +560,7 @@ size_t ts_stack__write_dot_graph(Stack *self, char *string, size_t n,
     StackNode *node = self->heads.contents[i];
     size_t color_count = sizeof(COLORS) / sizeof(COLORS[0]);
     const char *color = COLORS[i % color_count];
-    cursor += snprintf(*s, n, "node_%p [color=%s];\n", node, color);
+    fprintf(f, "node_%p [color=%s];\n", node, color);
     array_push(&self->pop_paths, ((PopPath){.node = node }));
   }
 
@@ -586,33 +583,29 @@ size_t ts_stack__write_dot_graph(Stack *self, char *string, size_t n,
         continue;
       all_paths_done = false;
 
-      cursor += snprintf(*s, n, "node_%p [label=", node);
+      fprintf(f, "node_%p [label=", node);
       if (node->state == ts_parse_state_error)
-        cursor += snprintf(*s, n, "\"?\"");
+        fprintf(f, "\"?\"");
       else
-        cursor += snprintf(*s, n, "%d", node->state);
-      cursor += snprintf(*s, n, "];\n");
+        fprintf(f, "%d", node->state);
+      fprintf(f, "];\n");
 
       for (int j = 0; j < node->successor_count; j++) {
         StackLink successor = node->successors[j];
-        cursor +=
-          snprintf(*s, n, "node_%p -> node_%p [label=\"", node, successor.node);
+        fprintf(f, "node_%p -> node_%p [label=\"", node, successor.node);
 
         if (successor.tree->symbol == ts_builtin_sym_error) {
-          cursor += snprintf(*s, n, "ERROR");
+          fprintf(f, "ERROR");
         } else {
           const char *name = symbol_names[successor.tree->symbol];
           for (const char *c = name; *c; c++) {
-            if (*c == '\"' || *c == '\\') {
-              **s = '\\';
-              cursor++;
-            }
-            **s = *c;
-            cursor++;
+            if (*c == '\"' || *c == '\\')
+              fprintf(f, "\\");
+            fprintf(f, "%c", *c);
           }
         }
 
-        cursor += snprintf(*s, n, "\"];\n");
+        fprintf(f, "\"];\n");
 
         if (j == 0) {
           path->node = successor.node;
@@ -629,35 +622,10 @@ size_t ts_stack__write_dot_graph(Stack *self, char *string, size_t n,
     }
   }
 
-  cursor += snprintf(*s, n, "}\n");
+  fprintf(f, "}\n");
 
   array_delete(&visited_nodes);
-  return cursor - string;
 
 error:
-  array_delete(&visited_nodes);
-  return (size_t)-1;
-}
-
-char *ts_stack_dot_graph(Stack *self, const char **symbol_names) {
-  static char SCRATCH[1];
-  char *result = NULL;
-  size_t size = ts_stack__write_dot_graph(self, SCRATCH, 0, symbol_names) + 1;
-  if (size == (size_t)-1)
-    goto error;
-
-  result = ts_malloc(size * sizeof(char));
-  if (!result)
-    goto error;
-
-  size = ts_stack__write_dot_graph(self, result, size, symbol_names);
-  if (size == (size_t)-1)
-    goto error;
-
-  return result;
-
-error:
-  if (result)
-    ts_free(result);
-  return NULL;
+  return -1;
 }
diff --git a/src/runtime/stack.h b/src/runtime/stack.h
index 3a1b6b2d..f352af69 100644
--- a/src/runtime/stack.h
+++ b/src/runtime/stack.h
@@ -8,6 +8,7 @@ extern "C" {
 #include "tree_sitter/parser.h"
 #include "runtime/array.h"
 #include "runtime/tree.h"
+#include <stdio.h>
 
 typedef struct Stack Stack;
 
@@ -118,7 +119,7 @@ void ts_stack_clear(Stack *);
 void ts_stack_set_tree_selection_callback(Stack *, void *,
                                           TreeSelectionFunction);
 
-char *ts_stack_dot_graph(Stack *, const char **);
+int ts_stack_print_dot_graph(Stack *, const char **, FILE *);
 
 #ifdef __cplusplus
 }