mirror of
https://github.com/zoriya/virtcolumn.nvim.git
synced 2026-05-27 14:51:45 +00:00
style: reformat
This commit is contained in:
wongxy
+114
-124
@@ -3,158 +3,148 @@
|
||||
--- vim.g.virtcolumn_priority = 10
|
||||
|
||||
local api, fn = vim.api, vim.fn
|
||||
local ffi = require "ffi"
|
||||
local ffi = require('ffi')
|
||||
|
||||
ffi.cdef "int curwin_col_off(void);"
|
||||
ffi.cdef('int curwin_col_off(void);')
|
||||
---@diagnostic disable-next-line: undefined-field
|
||||
local curwin_col_off = ffi.C.curwin_col_off
|
||||
|
||||
local NS = api.nvim_create_namespace "virtcolumn"
|
||||
local NS = api.nvim_create_namespace('virtcolumn')
|
||||
|
||||
---@param cc string
|
||||
---@return number[]
|
||||
local function parse_items(cc)
|
||||
local textwidth = vim.o.textwidth
|
||||
---@type number[]
|
||||
local items = {}
|
||||
for _, c in ipairs(vim.split(cc, ",")) do
|
||||
local item
|
||||
if c and c ~= "" then
|
||||
if vim.startswith(c, "+") then
|
||||
if textwidth ~= 0 then
|
||||
item = textwidth + tonumber(c:sub(2))
|
||||
end
|
||||
elseif vim.startswith(cc, "-") then
|
||||
if textwidth ~= 0 then
|
||||
item = textwidth - tonumber(c:sub(2))
|
||||
end
|
||||
else
|
||||
item = tonumber(c)
|
||||
end
|
||||
end
|
||||
if item and item > 0 then
|
||||
table.insert(items, item)
|
||||
end
|
||||
local textwidth = vim.o.textwidth
|
||||
---@type number[]
|
||||
local items = {}
|
||||
for _, c in ipairs(vim.split(cc, ',')) do
|
||||
local item
|
||||
if c and c ~= '' then
|
||||
if vim.startswith(c, '+') then
|
||||
if textwidth ~= 0 then item = textwidth + tonumber(c:sub(2)) end
|
||||
elseif vim.startswith(cc, '-') then
|
||||
if textwidth ~= 0 then item = textwidth - tonumber(c:sub(2)) end
|
||||
else
|
||||
item = tonumber(c)
|
||||
end
|
||||
end
|
||||
table.sort(items, function(a, b)
|
||||
return a > b
|
||||
end)
|
||||
return items
|
||||
if item and item > 0 then table.insert(items, item) end
|
||||
end
|
||||
table.sort(items, function(a, b)
|
||||
return a > b
|
||||
end)
|
||||
return items
|
||||
end
|
||||
|
||||
local function _refresh()
|
||||
local curbuf = api.nvim_get_current_buf()
|
||||
if not api.nvim_buf_is_loaded(curbuf) then
|
||||
return
|
||||
end
|
||||
|
||||
local items = vim.b.virtcolumn_items or vim.w.virtcolumn_items
|
||||
local local_cc = api.nvim_get_option_value("cc", { scope = "local" })
|
||||
if not items or local_cc ~= "" then
|
||||
items = parse_items(local_cc)
|
||||
api.nvim_set_option_value("cc", "", { scope = "local" })
|
||||
end
|
||||
vim.b.virtcolumn_items = items
|
||||
vim.w.virtcolumn_items = items
|
||||
|
||||
api.nvim_buf_clear_namespace(curbuf, NS, 0, -1)
|
||||
|
||||
local win_width = api.nvim_win_get_width(0) - curwin_col_off()
|
||||
items = vim.tbl_filter(function(item)
|
||||
return win_width > item
|
||||
end, items)
|
||||
|
||||
if #items == 0 then
|
||||
return
|
||||
end
|
||||
|
||||
local debounce = math.floor(api.nvim_win_get_height(0) * 0.6)
|
||||
local offset = vim.fn.line "w0"
|
||||
-- Avoid flickering caused by winscrolled_timer
|
||||
offset = (offset <= debounce and 1 or offset - debounce) - 1 -- convert to 0-based
|
||||
|
||||
-- Avoid flickering caused by winscrolled_timer
|
||||
-- ↓↓↓↓↓↓↓↓↓↓↓
|
||||
local lines = api.nvim_buf_get_lines(curbuf, offset, vim.fn.line "w$" + debounce, false)
|
||||
local tabstop = vim.opt.tabstop:get()
|
||||
local char = vim.g.virtcolumn_char or "▕"
|
||||
local priority = vim.g.virtcolumn_priority or 10
|
||||
|
||||
for i = 1, #lines do
|
||||
for _, item in ipairs(items) do
|
||||
local line = lines[i]:gsub("\t", string.rep(" ", tabstop))
|
||||
if api.nvim_strwidth(line) < item or fn.strpart(line, item - 1, 1) == " " then
|
||||
api.nvim_buf_set_extmark(curbuf, NS, i + offset - 1, 0, {
|
||||
virt_text = { { char, "VirtColumn" } },
|
||||
hl_mode = "combine",
|
||||
virt_text_win_col = item - 1,
|
||||
priority = priority,
|
||||
})
|
||||
end
|
||||
end
|
||||
local curbuf = api.nvim_get_current_buf()
|
||||
if not api.nvim_buf_is_loaded(curbuf) then return end
|
||||
|
||||
local items = vim.b.virtcolumn_items or vim.w.virtcolumn_items
|
||||
local local_cc = api.nvim_get_option_value('cc', { scope = 'local' })
|
||||
if not items or local_cc ~= '' then
|
||||
items = parse_items(local_cc)
|
||||
api.nvim_set_option_value('cc', '', { scope = 'local' })
|
||||
end
|
||||
vim.b.virtcolumn_items = items
|
||||
vim.w.virtcolumn_items = items
|
||||
|
||||
api.nvim_buf_clear_namespace(curbuf, NS, 0, -1)
|
||||
|
||||
local win_width = api.nvim_win_get_width(0) - curwin_col_off()
|
||||
items = vim.tbl_filter(function(item)
|
||||
return win_width > item
|
||||
end, items)
|
||||
|
||||
if #items == 0 then return end
|
||||
|
||||
local debounce = math.floor(api.nvim_win_get_height(0) * 0.6)
|
||||
local offset = vim.fn.line('w0')
|
||||
-- Avoid flickering caused by winscrolled_timer
|
||||
offset = (offset <= debounce and 1 or offset - debounce) - 1 -- convert to 0-based
|
||||
|
||||
-- Avoid flickering caused by winscrolled_timer
|
||||
-- ↓↓↓↓↓↓↓↓↓↓↓
|
||||
local lines = api.nvim_buf_get_lines(curbuf, offset, vim.fn.line('w$') + debounce, false)
|
||||
local tabstop = vim.opt.tabstop:get()
|
||||
local char = vim.g.virtcolumn_char or '▕'
|
||||
local priority = vim.g.virtcolumn_priority or 10
|
||||
|
||||
for i = 1, #lines do
|
||||
for _, item in ipairs(items) do
|
||||
local line = lines[i]:gsub('\t', string.rep(' ', tabstop))
|
||||
if api.nvim_strwidth(line) < item or fn.strpart(line, item - 1, 1) == ' ' then
|
||||
api.nvim_buf_set_extmark(curbuf, NS, i + offset - 1, 0, {
|
||||
virt_text = { { char, 'VirtColumn' } },
|
||||
hl_mode = 'combine',
|
||||
virt_text_win_col = item - 1,
|
||||
priority = priority,
|
||||
})
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
-- Avoid unnecessary refreshing as much as possible lcoallfdafffadf
|
||||
local winscrolled_timer
|
||||
local textchanged_timer
|
||||
local function refresh(args)
|
||||
---@type string
|
||||
local event = args.event or ""
|
||||
if event == "WinScrolled" then
|
||||
if winscrolled_timer and winscrolled_timer:is_active() then
|
||||
winscrolled_timer:stop()
|
||||
winscrolled_timer:close()
|
||||
end
|
||||
winscrolled_timer = vim.defer_fn(_refresh, 100)
|
||||
elseif event:match "TextChanged" then
|
||||
if textchanged_timer and textchanged_timer:is_active() then
|
||||
textchanged_timer:stop()
|
||||
textchanged_timer:close()
|
||||
end
|
||||
local lines_count = vim.fn.line "$"
|
||||
local delay
|
||||
if lines_count ~= vim.b.virtcolumn_lines_count then
|
||||
vim.b.virtcolumn_lines_count = lines_count
|
||||
delay = 15
|
||||
else
|
||||
delay = 150
|
||||
end
|
||||
textchanged_timer = vim.defer_fn(_refresh, delay)
|
||||
else
|
||||
_refresh()
|
||||
---@type string
|
||||
local event = args.event or ''
|
||||
if event == 'WinScrolled' then
|
||||
if winscrolled_timer and winscrolled_timer:is_active() then
|
||||
winscrolled_timer:stop()
|
||||
winscrolled_timer:close()
|
||||
end
|
||||
winscrolled_timer = vim.defer_fn(_refresh, 100)
|
||||
elseif event:match('TextChanged') then
|
||||
if textchanged_timer and textchanged_timer:is_active() then
|
||||
textchanged_timer:stop()
|
||||
textchanged_timer:close()
|
||||
end
|
||||
local lines_count = vim.fn.line('$')
|
||||
local delay
|
||||
if lines_count ~= vim.b.virtcolumn_lines_count then
|
||||
vim.b.virtcolumn_lines_count = lines_count
|
||||
delay = 15
|
||||
else
|
||||
delay = 150
|
||||
end
|
||||
textchanged_timer = vim.defer_fn(_refresh, delay)
|
||||
else
|
||||
_refresh()
|
||||
end
|
||||
end
|
||||
|
||||
local function set_hl()
|
||||
local cc_bg = api.nvim_get_hl_by_name("ColorColumn", true).background
|
||||
if cc_bg then
|
||||
api.nvim_set_hl(0, "VirtColumn", { fg = cc_bg, default = true })
|
||||
else
|
||||
vim.cmd [[hi default link VirtColumn NonText]]
|
||||
end
|
||||
local cc_bg = api.nvim_get_hl_by_name('ColorColumn', true).background
|
||||
if cc_bg then
|
||||
api.nvim_set_hl(0, 'VirtColumn', { fg = cc_bg, default = true })
|
||||
else
|
||||
vim.cmd([[hi default link VirtColumn NonText]])
|
||||
end
|
||||
end
|
||||
|
||||
local group = api.nvim_create_augroup("virtcolumn", {})
|
||||
local group = api.nvim_create_augroup('virtcolumn', {})
|
||||
api.nvim_create_autocmd({
|
||||
"WinScrolled",
|
||||
"TextChanged",
|
||||
"TextChangedI",
|
||||
"BufWinEnter",
|
||||
"InsertLeave",
|
||||
"InsertEnter",
|
||||
"FileChangedShellPost",
|
||||
'WinScrolled',
|
||||
'TextChanged',
|
||||
'TextChangedI',
|
||||
'BufWinEnter',
|
||||
'InsertLeave',
|
||||
'InsertEnter',
|
||||
'FileChangedShellPost',
|
||||
}, { group = group, callback = refresh })
|
||||
api.nvim_create_autocmd("OptionSet", {
|
||||
group = group,
|
||||
callback = function()
|
||||
vim.b.virtcolumn_items = nil
|
||||
vim.w.virtcolumn_items = nil
|
||||
_refresh()
|
||||
end,
|
||||
pattern = "colorcolumn",
|
||||
api.nvim_create_autocmd('OptionSet', {
|
||||
group = group,
|
||||
callback = function()
|
||||
vim.b.virtcolumn_items = nil
|
||||
vim.w.virtcolumn_items = nil
|
||||
_refresh()
|
||||
end,
|
||||
pattern = 'colorcolumn',
|
||||
})
|
||||
api.nvim_create_autocmd("ColorScheme", { group = group, callback = set_hl })
|
||||
api.nvim_create_autocmd('ColorScheme', { group = group, callback = set_hl })
|
||||
|
||||
pcall(set_hl)
|
||||
pcall(_refresh)
|
||||
|
||||
Reference in New Issue
Block a user