210 lines
7.0 KiB
Lua
210 lines
7.0 KiB
Lua
-- Author : swytch
|
||
-- Created : Friday Mar 12, 2021 21:45:21 CET
|
||
-- License : GPLv3
|
||
-- Description : neovim statusline file
|
||
-- based on github/nihilistkitten's work
|
||
|
||
|
||
local function gen_section(hl_string, items)
|
||
local out = ""
|
||
for _, item in pairs(items) do
|
||
if item ~= "" then
|
||
out = out .. " " .. item
|
||
end
|
||
end
|
||
if out ~="" then
|
||
return hl_string .. out .. " "
|
||
else
|
||
return out
|
||
end
|
||
end
|
||
|
||
-- sensible names for vim modes
|
||
local function get_mode(m)
|
||
local mode_group = {
|
||
["n"] = "Normal",
|
||
["no"] = "O-Pending",
|
||
["nov"] = "O-Pending",
|
||
["noV"] = "O-Pending",
|
||
["no"] = "O-Pending",
|
||
["niI"] = "Normal",
|
||
["niR"] = "Normal",
|
||
["niV"] = "Normal",
|
||
["v"] = "Visual",
|
||
["V"] = "V-Line",
|
||
[""] = "V-Block",
|
||
["s"] = "Select",
|
||
["S"] = "S-Line",
|
||
[""] = "S-Block",
|
||
["i"] = "Insert",
|
||
["ic"] = "Insert",
|
||
["ix"] = "Insert",
|
||
["R"] = "Replace",
|
||
["Rc"] = "Replace",
|
||
["Rv"] = "V-Replace",
|
||
["Rx"] = "Replace",
|
||
["c"] = "Command",
|
||
["cv"] = "Ex",
|
||
["ce"] = "Ex",
|
||
["r"] = "Prompt",
|
||
["rm"] = "Prompt",
|
||
["r?"] = "Prompt",
|
||
["!"] = "Shell",
|
||
["t"] = "Terminal",
|
||
}
|
||
return mode_group[m] or "None"
|
||
end
|
||
|
||
local function get_mode_display_name(m)
|
||
local mode = {
|
||
["Normal"] = "[ NRM ]",
|
||
["O-Pending"] = "[ OTR ]",
|
||
["Visual"] = "[ VSL ]",
|
||
["V-Line"] = "[ V·L ]",
|
||
["V-Block"] = "[ V·B ]",
|
||
["Select"] = "[ SEL ]",
|
||
["S-Line"] = "[ S·L ]",
|
||
["S-Block"] = "[ S·B ]",
|
||
["Insert"] = "[ INS ]",
|
||
["Replace"] = "[ RPL ]",
|
||
["Command"] = "[ CMD ]",
|
||
["Prompt"] = "[ PMT ]",
|
||
["Shell"] = "[ SHL ]",
|
||
["Terminal"] = "[ TRM ]",
|
||
["None"] = "[ --- ]"
|
||
}
|
||
return mode[m]
|
||
end
|
||
|
||
-- get the highlight group for a mode group
|
||
local function get_mode_color(m)
|
||
local color = {
|
||
["Normal"] = "%#NormalMode#",
|
||
["O-Pending"] = "%#NormalMode#",
|
||
["Visual"] = "%#VisualMode#",
|
||
["V-Line"] = "%#VisualMode#",
|
||
["V-Block"] = "%#VisualMode#",
|
||
["Select"] = "%#NormalMode#",
|
||
["S-Line"] = "%#NormalMode#",
|
||
["S-Block"] = "%#NormalMode#",
|
||
["Insert"] = "%#InsertMode#",
|
||
["Replace"] = "%#ReplaceMode#",
|
||
["Command"] = "%#CommandMode#",
|
||
["Prompt"] = "%#NormalMode#",
|
||
["Shell"] = "%#NormalMode#",
|
||
["Terminal"] = "%#NormalMode#",
|
||
["None"] = "%#NormalMode#"
|
||
}
|
||
return color[m]
|
||
end
|
||
|
||
-- from https://github.com/nvim-lua/lsp-status.nvim/blob/master/lua/lsp-status/diagnostics.lua
|
||
local function get_lsp_diagnostics(bufnr)
|
||
local result = {}
|
||
local levels = {
|
||
errors = "Error",
|
||
warnings = "Warn",
|
||
info = "Info",
|
||
hints = "Hint"
|
||
}
|
||
|
||
for k, level in pairs(levels) do
|
||
local count = 0
|
||
local diags = vim.diagnostic.get(bufnr, { severity = level })
|
||
for _, _ in pairs(diags) do
|
||
count = count + 1
|
||
end
|
||
result[k] = count
|
||
end
|
||
|
||
return result
|
||
end
|
||
|
||
local function process_diagnostics(prefix, n, hl)
|
||
if n > 0 then
|
||
return hl .. prefix .. n
|
||
else
|
||
return ""
|
||
end
|
||
end
|
||
|
||
local function spell_check()
|
||
if vim.wo.spell then
|
||
local lang = vim.o.spelllang
|
||
return "[SPELL=" .. lang .. "]"
|
||
else
|
||
return ""
|
||
end
|
||
end
|
||
|
||
local function statusline_focused()
|
||
local diagnostics = get_lsp_diagnostics()
|
||
local mode = vim.fn.mode()
|
||
local mg = get_mode(mode)
|
||
local accent_color = get_mode_color(mg)
|
||
|
||
local left = table.concat {
|
||
gen_section(accent_color, {get_mode_display_name(mg)}),
|
||
gen_section("%#Middle#", {"%f"}),
|
||
gen_section("%#Bottom#", {"%m", "%r"}),
|
||
gen_section(
|
||
"%#Alert#",
|
||
{
|
||
process_diagnostics(
|
||
globals.sign_error .. " ",
|
||
diagnostics.errors,
|
||
"%#DiagnosticVirtualTextError#"
|
||
),
|
||
process_diagnostics(
|
||
globals.sign_warning .. " ",
|
||
diagnostics.warnings,
|
||
"%#DiagnosticVirtualTextWarn#"
|
||
),
|
||
process_diagnostics(
|
||
globals.sign_info .. " ",
|
||
diagnostics.info,
|
||
"%#DiagnosticVirtualTextInfo#"
|
||
)
|
||
}
|
||
)
|
||
}
|
||
local right = table.concat {
|
||
gen_section(
|
||
"%#Bottom#",
|
||
{
|
||
spell_check(),
|
||
vim.bo.filetype
|
||
}
|
||
),
|
||
gen_section("%#Middle#", {"%03.p%%"}),
|
||
gen_section("%#Top#", {"-%03.c-"})
|
||
}
|
||
|
||
return table.concat {
|
||
left,
|
||
"%#Statusline#",
|
||
"%=",
|
||
right
|
||
}
|
||
|
||
end
|
||
|
||
local function statusline_not_focused()
|
||
return table.concat {
|
||
gen_section("%#StatuslineNF#", {"%f", "%m"}),
|
||
"%=",
|
||
gen_section("%#StatuslineNF#", {"%03.p%%"}),
|
||
gen_section("%#StatuslineNF#", {"-%03.c-"})
|
||
}
|
||
end
|
||
|
||
function _G.gen_statusline()
|
||
if vim.g.statusline_winid == vim.fn.win_getid() then
|
||
return statusline_focused()
|
||
else
|
||
return statusline_not_focused()
|
||
end
|
||
end
|
||
|
||
vim.o.statusline = "%!luaeval(\"gen_statusline()\")"
|