From f57341a4b69398ff0c58686e66c2f4138be164aa Mon Sep 17 00:00:00 2001
From: Quentin Rasmont <qrasmont@gmail.com>
Date: Sun, 1 May 2022 21:13:47 +0200
Subject: feat(treesitter): upstream node_length() as a node method

Util from the nvim-treesitter project.
---
 src/nvim/lua/treesitter.c | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'src/nvim/lua/treesitter.c')

diff --git a/src/nvim/lua/treesitter.c b/src/nvim/lua/treesitter.c
index 083042ad98..8803fd452c 100644
--- a/src/nvim/lua/treesitter.c
+++ b/src/nvim/lua/treesitter.c
@@ -90,6 +90,7 @@ static struct luaL_Reg node_meta[] = {
   { "prev_named_sibling", node_prev_named_sibling },
   { "named_children", node_named_children },
   { "root", node_root },
+  { "byte_length", node_byte_length },
 
   { NULL, NULL }
 };
@@ -1111,6 +1112,20 @@ static int node_root(lua_State *L)
   return 1;
 }
 
+static int node_byte_length(lua_State *L)
+{
+  TSNode node;
+  if (!node_check(L, 1, &node)) {
+    return 0;
+  }
+
+  uint32_t start_byte = ts_node_start_byte(node);
+  uint32_t end_byte = ts_node_end_byte(node);
+
+  lua_pushnumber(L, end_byte - start_byte);
+  return 1;
+}
+
 /// assumes the match table being on top of the stack
 static void set_match(lua_State *L, TSQueryMatch *match, int nodeidx)
 {
-- 
cgit