Added option node_context

doc/twilight.txt

@@ -1,4 +1,4 @@
-*twilight.txt*          For NVIM v0.5.0         Last change: 2022 September 05
+*twilight.txt*          For NVIM v0.5.0          Last change: 2022 November 23
 
 ==============================================================================
 Table of Contents                                 *twilight-table-of-contents*

lua/twilight/config.lua

@@ -11,6 +11,7 @@ local defaults = {
     inactive = false, -- when true, other windows will be fully dimmed (unless they contain the same buffer)
   },
   context = 10, -- amount of lines we will try to show arounc the current line
+  node_context = 0, -- amount of extra nodes to show. Only node types mention in expand are considered
   treesitter = true, -- use treesitter when available for the filetype
   -- treesitter is used to automatically expand the visible text,
   -- but you can further control the types of nodes that should always be fully expanded

lua/twilight/view.lua

@@ -150,16 +150,24 @@ end
 
 function M.get_expand_root(node, opts)
   opts = opts or {}
+  local left_context = config.options.node_context
   local root
   while node do
     if config.expand[node:type()] then
       if opts.first then
         return node
       end
+      if left_context <= 0 then
+          return node
+      end
+      left_context = left_context - 1
       root = node
     end
     node = node:parent()
   end
+  if left_context > 0 then
+      return
+  end
   return root
 end
 
@@ -173,6 +181,11 @@ function M.get_context(buf, line)
     end
     local from, to = M.expand(buf, line, line, line)
 
+    if config.options.node_context > 0 then
+      local lcount = vim.api.nvim_buf_line_count(buf)
+      return 2, lcount  + 2
+    end
+
     while to - from < config.options.context do
       local pf, pt, pnode = M.expand(buf, from, to, from - 1)
       local nf, nt, nnode = M.expand(buf, from, to, to + 1)