2021-09-20 13:11:41 +00:00
|
|
|
-- Copyright (c) 2020-2021 shadmansaleh
|
|
|
|
-- MIT license, see LICENSE for more details.
|
2021-10-10 18:04:55 +00:00
|
|
|
local require = require('lualine_require').require
|
2022-01-02 11:38:39 +00:00
|
|
|
local Tab = require('lualine.components.tabs.tab')
|
2021-09-25 17:15:42 +00:00
|
|
|
local M = require('lualine.component'):extend()
|
2022-01-02 11:38:39 +00:00
|
|
|
local highlight = require('lualine.highlight')
|
2021-09-20 13:11:41 +00:00
|
|
|
|
|
|
|
local default_options = {
|
|
|
|
max_length = 0,
|
|
|
|
mode = 0,
|
2021-09-21 06:26:19 +00:00
|
|
|
tabs_color = {
|
|
|
|
active = nil,
|
|
|
|
inactive = nil,
|
|
|
|
},
|
2021-09-20 13:11:41 +00:00
|
|
|
}
|
|
|
|
|
2021-10-12 14:04:47 +00:00
|
|
|
-- This function is duplicated in buffers
|
|
|
|
---returns the proper hl for tab in section. used for setting default highlights
|
|
|
|
---@param section string name of section tabs component is in
|
|
|
|
---@param is_active boolean
|
|
|
|
---@return string hl name
|
2021-09-20 13:11:41 +00:00
|
|
|
local function get_hl(section, is_active)
|
|
|
|
local suffix = is_active and '_normal' or '_inactive'
|
|
|
|
local section_redirects = {
|
|
|
|
lualine_x = 'lualine_c',
|
|
|
|
lualine_y = 'lualine_b',
|
|
|
|
lualine_z = 'lualine_a',
|
|
|
|
}
|
|
|
|
if section_redirects[section] then
|
|
|
|
section = highlight.highlight_exists(section .. suffix) and section or section_redirects[section]
|
|
|
|
end
|
|
|
|
return section .. suffix
|
|
|
|
end
|
|
|
|
|
2021-09-25 17:15:42 +00:00
|
|
|
function M:init(options)
|
|
|
|
M.super.init(self, options)
|
2021-09-20 13:11:41 +00:00
|
|
|
default_options.tabs_color = {
|
|
|
|
active = get_hl(options.self.section, true),
|
|
|
|
inactive = get_hl(options.self.section, false),
|
|
|
|
}
|
2021-09-25 17:15:42 +00:00
|
|
|
self.options = vim.tbl_deep_extend('keep', self.options or {}, default_options)
|
2021-09-20 13:11:41 +00:00
|
|
|
-- stylua: ignore
|
2021-09-25 17:15:42 +00:00
|
|
|
self.highlights = {
|
2021-09-20 13:11:41 +00:00
|
|
|
active = highlight.create_component_highlight_group(
|
2021-09-25 17:15:42 +00:00
|
|
|
self.options.tabs_color.active,
|
2021-09-20 13:11:41 +00:00
|
|
|
'tabs_active',
|
2021-09-25 17:15:42 +00:00
|
|
|
self.options
|
2021-09-20 13:11:41 +00:00
|
|
|
),
|
|
|
|
inactive = highlight.create_component_highlight_group(
|
2021-09-25 17:15:42 +00:00
|
|
|
self.options.tabs_color.inactive,
|
2021-09-20 13:11:41 +00:00
|
|
|
'tabs_active',
|
2021-09-25 17:15:42 +00:00
|
|
|
self.options
|
2021-09-20 13:11:41 +00:00
|
|
|
),
|
|
|
|
}
|
|
|
|
end
|
|
|
|
|
2021-09-25 17:15:42 +00:00
|
|
|
function M:update_status()
|
2021-09-20 13:11:41 +00:00
|
|
|
local data = {}
|
|
|
|
local tabs = {}
|
2022-02-07 16:13:42 +00:00
|
|
|
for nr, id in ipairs(vim.api.nvim_list_tabpages()) do
|
|
|
|
tabs[#tabs + 1] = Tab { tabId = id, tabnr = nr, options = self.options, highlights = self.highlights }
|
2021-09-20 13:11:41 +00:00
|
|
|
end
|
2021-10-12 14:04:47 +00:00
|
|
|
-- mark the first, last, current, before current, after current tabpages
|
|
|
|
-- for rendering
|
2021-10-08 14:27:12 +00:00
|
|
|
local current = vim.fn.tabpagenr()
|
2021-09-20 13:11:41 +00:00
|
|
|
tabs[1].first = true
|
|
|
|
tabs[#tabs].last = true
|
|
|
|
if tabs[current] then
|
|
|
|
tabs[current].current = true
|
|
|
|
end
|
|
|
|
if tabs[current - 1] then
|
|
|
|
tabs[current - 1].beforecurrent = true
|
|
|
|
end
|
|
|
|
if tabs[current + 1] then
|
|
|
|
tabs[current + 1].aftercurrent = true
|
|
|
|
end
|
|
|
|
|
|
|
|
local max_length = self.options.max_length
|
2021-10-29 13:07:11 +00:00
|
|
|
if type(max_length) == 'function' then
|
|
|
|
max_length = max_length(self)
|
|
|
|
end
|
2021-10-29 13:13:17 +00:00
|
|
|
|
2021-09-20 13:11:41 +00:00
|
|
|
if max_length == 0 then
|
|
|
|
max_length = math.floor(vim.o.columns / 3)
|
|
|
|
end
|
|
|
|
local total_length
|
|
|
|
for i, tab in pairs(tabs) do
|
|
|
|
if tab.current then
|
|
|
|
current = i
|
|
|
|
end
|
|
|
|
end
|
|
|
|
local current_tab = tabs[current]
|
2021-10-12 14:04:47 +00:00
|
|
|
-- start drawing from current tab and draw left and right of it until
|
|
|
|
-- all tabpages are drawn or max_length has been reached.
|
|
|
|
if current_tab == nil then -- maybe redundent code
|
2022-02-07 16:13:42 +00:00
|
|
|
local t = Tab {
|
|
|
|
tabId = vim.api.nvim_get_current_tabpage(),
|
|
|
|
tabnr = vim.fn.tabpagenr(),
|
|
|
|
options = self.options,
|
|
|
|
highlights = self.highlights,
|
|
|
|
}
|
2021-09-20 13:11:41 +00:00
|
|
|
t.current = true
|
|
|
|
t.last = true
|
|
|
|
data[#data + 1] = t:render()
|
|
|
|
else
|
|
|
|
data[#data + 1] = current_tab:render()
|
|
|
|
total_length = current_tab.len
|
|
|
|
local i = 0
|
|
|
|
local before, after
|
|
|
|
while true do
|
|
|
|
i = i + 1
|
|
|
|
before = tabs[current - i]
|
|
|
|
after = tabs[current + i]
|
|
|
|
local rendered_before, rendered_after
|
|
|
|
if before == nil and after == nil then
|
|
|
|
break
|
|
|
|
end
|
2021-10-12 14:04:47 +00:00
|
|
|
-- draw left most undrawn tab if fits in max_length
|
2021-09-20 13:11:41 +00:00
|
|
|
if before then
|
|
|
|
rendered_before = before:render()
|
|
|
|
total_length = total_length + before.len
|
|
|
|
if total_length > max_length then
|
|
|
|
break
|
|
|
|
end
|
|
|
|
table.insert(data, 1, rendered_before)
|
|
|
|
end
|
2021-10-12 14:04:47 +00:00
|
|
|
-- draw right most undrawn tab if fits in max_length
|
2021-09-20 13:11:41 +00:00
|
|
|
if after then
|
|
|
|
rendered_after = after:render()
|
|
|
|
total_length = total_length + after.len
|
|
|
|
if total_length > max_length then
|
|
|
|
break
|
|
|
|
end
|
|
|
|
data[#data + 1] = rendered_after
|
|
|
|
end
|
|
|
|
end
|
2021-10-12 14:04:47 +00:00
|
|
|
-- draw elipsis (...) on relevent sides if all tabs don't fit in max_length
|
2021-09-20 13:11:41 +00:00
|
|
|
if total_length > max_length then
|
|
|
|
if before ~= nil then
|
|
|
|
before.ellipse = true
|
|
|
|
before.first = true
|
|
|
|
table.insert(data, 1, before:render())
|
|
|
|
end
|
|
|
|
if after ~= nil then
|
|
|
|
after.ellipse = true
|
|
|
|
after.last = true
|
|
|
|
data[#data + 1] = after:render()
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
return table.concat(data)
|
|
|
|
end
|
|
|
|
|
2021-11-18 02:12:33 +00:00
|
|
|
function M:draw()
|
|
|
|
self.status = ''
|
|
|
|
self.applied_separator = ''
|
|
|
|
|
|
|
|
if self.options.cond ~= nil and self.options.cond() ~= true then
|
|
|
|
return self.status
|
|
|
|
end
|
|
|
|
local status = self:update_status()
|
|
|
|
if type(status) == 'string' and #status > 0 then
|
|
|
|
self.status = status
|
|
|
|
self:apply_section_separators()
|
|
|
|
self:apply_separator()
|
|
|
|
end
|
|
|
|
return self.status
|
|
|
|
end
|
|
|
|
|
2022-01-02 11:38:39 +00:00
|
|
|
vim.cmd([[
|
2021-09-20 13:11:41 +00:00
|
|
|
function! LualineSwitchTab(tabnr, mouseclicks, mousebutton, modifiers)
|
|
|
|
execute a:tabnr . "tabnext"
|
|
|
|
endfunction
|
2022-02-07 15:26:25 +00:00
|
|
|
|
|
|
|
function! LualineRenameTab(...)
|
|
|
|
if a:0 == 1
|
|
|
|
let t:tabname = a:1
|
|
|
|
else
|
|
|
|
unlet t:tabname
|
|
|
|
end
|
|
|
|
redrawtabline
|
|
|
|
endfunction
|
|
|
|
|
|
|
|
command! -nargs=? LualineRenameTab call LualineRenameTab("<args>")
|
2022-01-02 11:38:39 +00:00
|
|
|
]])
|
2021-09-20 13:11:41 +00:00
|
|
|
|
2021-09-25 17:15:42 +00:00
|
|
|
return M
|