Fix Tree.printDotGraph ignoring fd

Fixes #202
tree-sitter · Jul 5, 2024 · f81f3d0 · f81f3d0
1 parent d07a2ad
commit f81f3d0
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/parser.cc b/src/parser.cc
@@ -274,7 +274,7 @@ Napi::Value Parser::SetLogger(const Napi::CallbackInfo &info) {
 
 Napi::Value Parser::PrintDotGraphs(const Napi::CallbackInfo &info) {
   bool should_print = true;
-  int32_t fd = fileno(stderr);
+  int fd = fileno(stderr);
 
   if (info.Length() > 0) {
     if (!info[0].IsBoolean()) {

diff --git a/src/tree.cc b/src/tree.cc
@@ -231,7 +231,16 @@ Napi::Value Tree::GetEditedRange(const Napi::CallbackInfo &info) {
 }
 
 Napi::Value Tree::PrintDotGraph(const Napi::CallbackInfo &info) {
-  ts_tree_print_dot_graph(tree_, fileno(stderr));
+  int fd = fileno(stderr);
+
+  if (info.Length() > 1) {
+    if (!info[1].IsNumber()) {
+      throw TypeError::New(info.Env(), "Second argument must be a number");
+    }
+    fd = info[1].As<Number>().Int32Value();
+  }
+
+  ts_tree_print_dot_graph(tree_, fd);
   return info.This();
 }