fix PyByteArray_Concat

Author: MrPrezident

tree_sitter/binding/node.c

@@ -589,6 +589,7 @@ PyObject *node_get_text(Node *self, void *Py_UNUSED(payload)) {
         }
 
         TSPoint start_point = ts_node_start_point(self->node);
+        TSPoint end_point = ts_node_end_point(self->node);
 
         TSPoint current_point = start_point;
         size_t current_offset = start_offset;
@@ -619,12 +620,23 @@ PyObject *node_get_text(Node *self, void *Py_UNUSED(payload)) {
                 return NULL;
             }
 
-            if (PyByteArray_Concat(collected_bytes, rv) == NULL) {
-                Py_XDECREF(rv);
+            PyObject *rv_bytearray = PyByteArray_FromObject(rv);
+            if (rv_bytearray == NULL) {
                 Py_XDECREF(collected_bytes);
+                Py_XDECREF(rv);
                 return NULL;
             }
-            Py_XDECREF(rv);
+
+            PyObject *new_collected_bytes = PyByteArray_Concat(collected_bytes, rv_bytearray);
+            Py_DECREF(rv_bytearray);
+            Py_DECREF(collected_bytes);
+
+            if (new_collected_bytes == NULL) {
+                Py_XDECREF(rv);
+                return NULL;
+            }
+
+            collected_bytes = new_collected_bytes;
 
             // Update current_point and current_offset
             Py_ssize_t bytes_read = PyBytes_Size(rv);
@@ -639,7 +651,20 @@ PyObject *node_get_text(Node *self, void *Py_UNUSED(payload)) {
             current_offset += bytes_read;
         }
 
-        result = PyBytes_FromObject(collected_bytes);
+        PyObject *start_byte = PyLong_FromUnsignedLong(0);
+        PyObject *end_byte = PyLong_FromUnsignedLong(end_offset - start_offset);
+        PyObject *slice = PySlice_New(start_byte, end_byte, NULL);
+        Py_XDECREF(start_byte);
+        Py_XDECREF(end_byte);
+
+        if (slice == NULL) {
+            Py_DECREF(collected_bytes);
+            return NULL;
+        }
+
+        result = PyObject_GetItem(collected_bytes, slice);
+        Py_DECREF(slice);
+
         Py_XDECREF(collected_bytes);
     }
     return result;