feat(tree): add language getter

docs/classes/tree_sitter.Tree.rst

@@ -15,4 +15,7 @@ Tree
    ----------
 
    .. autoattribute:: included_ranges
+   .. autoattribute:: language
+
+      .. versionadded:: 0.23.0
    .. autoattribute:: root_node

tests/test_parser.py

@@ -145,6 +145,7 @@ def read(byte_position, _):
 
         self.assertEqual(snake_node.type, "string")
         self.assertEqual(snake.decode("utf16"), "🐍")
+        self.assertIs(tree.language, self.javascript)
 
 
     def test_parse_invalid_encoding(self):

tree_sitter/__init__.pyi

@@ -158,6 +158,8 @@ class Tree:
     def root_node(self) -> Node: ...
     @property
     def included_ranges(self) -> list[Range]: ...
+    @property
+    def language(self) -> Language: ...
     def root_node_with_offset(
         self,
         offset_bytes: int,
@@ -175,6 +177,7 @@ class Tree:
     ) -> None: ...
     def walk(self) -> TreeCursor: ...
     def changed_ranges(self, new_tree: Tree) -> list[Range]: ...
+    # TODO(0.24): add copy methods
 
 @final
 class TreeCursor:

tree_sitter/binding/parser.c

@@ -175,6 +175,7 @@ PyObject *parser_parse(Parser *self, PyObject *args, PyObject *kwargs) {
         return NULL;
     }
     tree->tree = new_tree;
+    tree->language = self->language;
     tree->source = keep_text ? source_or_callback : Py_None;
     Py_INCREF(tree->source);
     return PyObject_Init((PyObject *)tree, state->tree_type);

tree_sitter/binding/tree.c

@@ -127,6 +127,11 @@ PyObject *tree_get_included_ranges(Tree *self, PyObject *Py_UNUSED(args)) {
     return result;
 }
 
+PyObject *tree_get_language(Tree *self, PyObject *Py_UNUSED(args)) {
+    Py_INCREF(self->language);
+    return self->language;
+}
+
 PyDoc_STRVAR(tree_root_node_with_offset_doc,
              "root_node_with_offset(self, offset_bytes, offset_extent, /)\n--\n\n"
              "Get the root node of the syntax tree, but with its position shifted "

tree_sitter/binding/tree_cursor.c

@@ -162,17 +162,16 @@ PyObject *tree_cursor_reset_to(TreeCursor *self, PyObject *args) {
     Py_RETURN_NONE;
 }
 
-PyObject *tree_cursor_copy(PyObject *self, PyObject *Py_UNUSED(args)) {
+PyObject *tree_cursor_copy(TreeCursor *self, PyObject *Py_UNUSED(args)) {
     ModuleState *state = GET_MODULE_STATE(self);
-    TreeCursor *origin = (TreeCursor *)self;
     TreeCursor *copied = PyObject_New(TreeCursor, state->tree_cursor_type);
     if (copied == NULL) {
         return NULL;
     }
 
-    Py_INCREF(origin->tree);
-    copied->tree = origin->tree;
-    copied->cursor = ts_tree_cursor_copy(&origin->cursor);
+    Py_INCREF(self->tree);
+    copied->tree = self->tree;
+    copied->cursor = ts_tree_cursor_copy(&self->cursor);
     return PyObject_Init((PyObject *)copied, state->tree_cursor_type);
 }
 

tree_sitter/binding/types.h

@@ -24,6 +24,7 @@ typedef struct {
     PyObject_HEAD
     TSTree *tree;
     PyObject *source;
+    PyObject *language;
 } Tree;
 
 typedef struct {