Add option to highlight the whole line

Author: JafarAbdi

lua/nvim-treesitter-refactor.lua

@@ -17,6 +17,7 @@ function M.init()
         enable = false,
         disable = {},
         is_supported = queries.has_locals,
+        highlight_eol = false,
       },
       smart_rename = {
         module_path = "nvim-treesitter-refactor.smart_rename",

lua/nvim-treesitter-refactor/highlight_current_scope.lua

@@ -1,5 +1,6 @@
 -- This module highlights the current scope of at the cursor position
 
+local configs = require "nvim-treesitter.configs"
 local ts_utils = require "nvim-treesitter.ts_utils"
 local locals = require "nvim-treesitter.locals"
 local api = vim.api
@@ -16,10 +17,21 @@ function M.highlight_current_scope(bufnr)
   local current_scope = locals.containing_scope(node_at_point, bufnr)
 
   if current_scope then
+    -- Highlight range [start_line, end_line) 0-based
+    local highlighter = function(start_line, end_line)
+      local config = configs.get_module "refactor.highlight_current_scope"
+      vim.api.nvim_buf_set_extmark(bufnr, current_scope_namespace, math.max(vim.fn.line("w0") - 1, start_line), 0, {
+        end_row = math.min(vim.fn.line("w$"), end_line),
+        end_col = 0,
+        hl_group = "TSCurrentScope",
+        hl_eol = config.highlight_eol,
+      })
+    end
     local start_line, _, end_line, _ = current_scope:range()
 
     if start_line ~= 0 or end_line ~= vim.fn.line("$") then
-      ts_utils.highlight_node(current_scope, bufnr, current_scope_namespace, "TSCurrentScope")
+      highlighter(start_line, vim.fn.line(".") - 1)
+      highlighter(vim.fn.line("."), end_line + 1)
     end
   end
 end