move hightlight.lua

add a session

Author: cht

README.md

@@ -5,8 +5,10 @@ should not setup or
 
 ```lua
 require("csvtools").setup({
-	before = 5,
-	after = 5,
+	before = 10,
+	after = 10,
+	clearafter = true
+	-- this will clear the highlight of buf after move
 })
 ```
 

lua/csvtools.lua

@@ -1,13 +1,14 @@
 local api = vim.api
-
+local highlight = require("csvtools.highlight")
 local M = {
     winid = nil,
     buf = nil,
     mainwindowbuf = nil,
-    before = 5,
-    after = 5,
+    before = 20,
+    after = 20,
+    clearafter = true,
 }
-local highlight = require("csvtools.topbarhighlight")
+
 function M.NewWindow()
     if M.winid == nil then
         M.mainwindowbuf = vim.api.nvim_get_current_buf()
@@ -33,13 +34,18 @@ function M.NewWindow()
         M.add_mappings()
     end
 end
+
 function M.CloseWindow()
     if M.winid ~= nil then
         vim.api.nvim_win_close(M.winid, true)
         M.winid = nil
         M.buf = nil
     end
 end
+
+--@param line number
+--@param length string
+--@return number number
 local function getrange(line, length)
     local start = 1
     if line - M.before > 1 then
@@ -56,6 +62,9 @@ function M.Highlight()
         M.mainwindowbuf = vim.api.nvim_get_current_buf()
         local line, _ = unpack(vim.api.nvim_win_get_cursor(0))
         local length = vim.api.nvim_buf_line_count(M.mainwindowbuf)
+        if M.clearafter then
+            api.nvim_buf_clear_highlight(M.mainwindowbuf, -1, 0, length)
+        end
         local start, final = getrange(line, length)
         --print(start)
         --print(final)

lua/csvtools/topbarhighlight.lua lua/csvtools/highlight.lualua/csvtools/topbarhighlight.lua renamed to lua/csvtools/highlight.lua

@@ -1,5 +1,8 @@
 local api = vim.api
 local M = {}
+
+--@param buf winbuf
+--@param line string
 function M.highlighttop(buf, line)
     local cout = 1
     for i = 1, #line do
@@ -15,6 +18,9 @@ function M.highlighttop(buf, line)
         end
     end
 end
+
+--@param buf winbuf
+--@param number number
 function M.highlight(buf, number)
     local line = unpack(vim.api.nvim_buf_get_lines(buf, number - 1, number, true))
     --print(line)