From cf72524b2f7e3868ea74037004913af6f935f126 Mon Sep 17 00:00:00 2001
From: kiyan42 <yazdani.kiyan@protonmail.com>
Date: Fri, 19 Jun 2020 14:13:23 +0200
Subject: [PATCH] add winnr to get_node_at_cursor

---
 README.md                        | 2 +-
 doc/nvim-treesitter.txt          | 3 ++-
 lua/nvim-treesitter/ts_utils.lua | 4 ++--
 3 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index b2193103a..787a2a200 100644
--- a/README.md
+++ b/README.md
@@ -137,7 +137,7 @@ you can get some utility functions with
 ```lua
 local ts_utils = require 'nvim-treesitter.ts_utils'
 ```
-More information is available in neovim documentation (`:help nvim-treesitter-utils`).
+More information is available in the help file (`:help nvim-treesitter-utils`).
 
 ## Supported Languages
 
diff --git a/doc/nvim-treesitter.txt b/doc/nvim-treesitter.txt
index 27e4f0bb5..bb53e182e 100644
--- a/doc/nvim-treesitter.txt
+++ b/doc/nvim-treesitter.txt
@@ -99,7 +99,8 @@ Nvim treesitter has some wrapper functions that you can retrieve with:
 <
 Methods
 
-get_node_at_cursor()                              *ts_utils.get_node_at_cursor*
+get_node_at_cursor(winnr)                         *ts_utils.get_node_at_cursor*
+    winnr will be 0 if nil
     returns the node under the cursor
 
 get_node_text(node, bufnr)                        *ts_utils.get_node_text*
diff --git a/lua/nvim-treesitter/ts_utils.lua b/lua/nvim-treesitter/ts_utils.lua
index a85934c0b..687a3c539 100644
--- a/lua/nvim-treesitter/ts_utils.lua
+++ b/lua/nvim-treesitter/ts_utils.lua
@@ -206,8 +206,8 @@ function M.previous_scope(node)
   end
 end
 
-function M.get_node_at_cursor()
-  local cursor = api.nvim_win_get_cursor(0)
+function M.get_node_at_cursor(winnr)
+  local cursor = api.nvim_win_get_cursor(winnr or 0)
   local root = parsers.get_parser().tree:root()
   return root:named_descendant_for_range(cursor[1]-1,cursor[2],cursor[1]-1,cursor[2])
 end