add overflowtext

Author: cht

ftdetect/csv-tools.vim

@@ -1,4 +1,5 @@
 au BufRead,BufNewFile *.csv,*.dat,*.tsv,*.tab set filetype=csv
 au FileType csv command! -buffer CloseTopWindow lua require"csvtools".CloseWindow()
 au FileType csv command! -buffer TopWindow lua require"csvtools".NewWindow()
+autocmd! InsertEnter *.csv,*.bat,*.tsv,*.tab lua require"csvtools".deleteMark()
 "au FIiletype csv autocmd CursorMoved csv lua require"csvtools".Highlight()

lua/csvtools.lua

@@ -1,14 +1,26 @@
 local api = vim.api
 local highlight = require("csvtools.highlight")
+local overflow = require("csvtools.overflowtext")
 local M = {
     winid = nil,
     buf = nil,
     mainwindowbuf = nil,
+	header = {},
     before = 20,
     after = 20,
     clearafter = true,
+	overflowtext = {
+		markid = nil,
+		ns_id = nil,
+		id = nil,
+	}
 }
-
+function M.printheader()
+	--for count = 1, #M.header do
+	--	print(M.header[count])
+	--end
+	return M.header
+end
 function M.NewWindow()
     if M.winid == nil then
         M.mainwindowbuf = vim.api.nvim_get_current_buf()
@@ -26,9 +38,7 @@ function M.NewWindow()
         local win = vim.api.nvim_get_current_win()
         api.nvim_buf_set_lines(buf, 0, -1, false, { messages })
         api.nvim_win_set_buf(win, buf)
-        highlight.highlighttop(buf, messages)
-        --api.nvim_buf_add_highlight(buf, -1, 'WhidHeader',0,0,1)
-        --api.nvim_buf_add_highlight(buf, -1, 'WhidSubHeader', 0, 1, 2)
+        M.header = highlight.highlighttop(buf, messages)
         M.winid = win
         M.buf = buf
         M.add_mappings()
@@ -39,7 +49,8 @@ function M.CloseWindow()
     if M.winid ~= nil then
         vim.api.nvim_win_close(M.winid, true)
         M.winid = nil
-        M.buf = nil
+  M.buf = nil
+		M.header = {}
     end
 end
 
@@ -61,13 +72,15 @@ function M.Highlight()
     if vim.o.filetype == "csv" then
         M.mainwindowbuf = vim.api.nvim_get_current_buf()
         local line, _ = unpack(vim.api.nvim_win_get_cursor(0))
+		--print(line)
         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)
+		M.overflowtext= overflow.OverFlow(line,M.header)
         for i = start, line, 1 do
             highlight.highlight(M.mainwindowbuf, i)
         end
@@ -76,6 +89,13 @@ function M.Highlight()
         end
     end
 end
+function M.deleteMark()
+	vim.api.nvim_buf_del_extmark(
+		M.overflowtext.markid,
+		M.overflowtext.ns_id,
+		M.overflowtext.id
+	)
+end
 function M.add_mappings()
     M.mainwindowbuf = vim.api.nvim_get_current_buf()
     --print(M.mainwindowbuf)
@@ -84,8 +104,8 @@ function M.add_mappings()
     vim.api.nvim_buf_set_keymap(M.mainwindowbuf, "n", "<leader>tf", ":lua require'csvtools'.NewWindow()<cr>", opts)
     vim.api.nvim_buf_set_keymap(M.buf, "n", "<leader>td", ":lua require'csvtools'.CloseWindow()<cr>", opts)
     vim.api.nvim_buf_set_keymap(M.mainwindowbuf, "n", "<leader>td", ":lua require'csvtools'.CloseWindow()<cr>", opts)
-    vim.api.nvim_buf_set_keymap(M.mainwindowbuf, "n", "<up>", ":lua require'csvtools'.Highlight()<cr>:-1<cr>", opts)
-    vim.api.nvim_buf_set_keymap(M.mainwindowbuf, "n", "<down>", ":lua require'csvtools'.Highlight()<cr>:+1<cr>", opts)
+    vim.api.nvim_buf_set_keymap(M.mainwindowbuf, "n", "<up>", ":-1<cr>:lua require'csvtools'.Highlight()<cr>", opts)
+    vim.api.nvim_buf_set_keymap(M.mainwindowbuf, "n", "<down>", ":+1<cr>:lua require'csvtools'.Highlight()<cr>", opts)
 end
 function M.setup(opts)
     M = vim.tbl_deep_extend("force", M, opts)

lua/csvtools/highlight.lua

@@ -4,19 +4,25 @@ local M = {}
 --@param buf winbuf
 --@param line string
 function M.highlighttop(buf, line)
+	local header = {}
     local cout = 1
+	local length = 0
     for i = 1, #line do
         if line:sub(i, i) ~= "|" then
+			length = length + 1
             if cout % 2 == 0 then
                 --print(line:sub(i, i))
                 api.nvim_buf_add_highlight(buf, -1, "WhidHeader", 0, i - 1, i)
             else
                 api.nvim_buf_add_highlight(buf, -1, "WhidSubHeader", 0, i - 1, i)
             end
         else
+			table.insert(header,length)
+			length = 0
             cout = cout + 1
         end
     end
+	return header
 end
 
 --@param buf winbuf

lua/csvtools/overflowtext.lua

@@ -0,0 +1,57 @@
+local M = {}
+local function Split(s, delimiter)
+    local result = {};
+    for match in (s..delimiter):gmatch("(.-)"..delimiter) do
+        table.insert(result, match);
+    end
+    return result;
+end
+local ns_id = vim.api.nvim_create_namespace('demo')
+function M.OverFlow(line_num, header)
+	local bnr = vim.fn.bufnr('%')
+	local line = unpack(vim.api.nvim_buf_get_lines(0, line_num-1  , line_num  , true))
+	local output = Split(line,',')
+	for count=1,#header do
+		if output[count] == nil then
+			output[count] = ''
+		end
+		local len = string.len(output[count])
+		if len < header[count] then
+			for _=1, header[count]-len do
+				output[count] = output[count]..' '
+			end
+		elseif len > header[count] then
+			--print(len,"  ",header[count])
+			output[count] = output[count]:sub(1,header[count]-2)
+			output[count] = output[count]..'..'
+			--print("len=" , string.len(output[count]))
+		end
+		--print(header[count])
+	end
+	local virt_text = {}
+	for count=1,#output do
+		if count % 2 == 0 then
+			table.insert(virt_text,{output[count],"WhidHeader"})
+		else
+			table.insert(virt_text,{output[count],"WhidSubHeader"})
+		end
+		table.insert(virt_text,{"|"})
+	--table.insert(virt_text,{text,"WhidHeader"})
+		--print(output[count])
+	end
+	local opts = {
+	  end_line = 10,
+	  id = 1,
+	  virt_text = virt_text,
+	  virt_text_pos = 'overlay',
+	  -- virt_text_win_col = 20,
+	}
+	--highlighttop2(bnr, text)
+	--print("sss")
+	return {
+		markid = vim.api.nvim_buf_set_extmark(bnr, ns_id, line_num - 1 , 0, opts),
+		ns_id = ns_id,
+		id = 1,
+	}
+end
+return M