2020-07-02 17:06:31 +00:00
|
|
|
--[[ NOTHING INSIDE THIS FILE NEEDS TO BE EDITED BY THE USER. ]]
|
|
|
|
|
2020-11-19 17:12:03 +00:00
|
|
|
--[[
|
|
|
|
/*
|
|
|
|
* IMPORTS
|
|
|
|
*/
|
|
|
|
--]]
|
|
|
|
|
|
|
|
local vim = vim
|
|
|
|
local cmd = vim.api.nvim_command
|
2020-06-22 02:38:18 +00:00
|
|
|
|
2020-11-19 17:12:03 +00:00
|
|
|
--[[
|
|
|
|
/*
|
|
|
|
* VARS
|
|
|
|
*/
|
|
|
|
--]]
|
2020-06-22 02:38:18 +00:00
|
|
|
|
2020-11-19 17:12:03 +00:00
|
|
|
-- These are constants for the indexes in the colors that were defined before.
|
|
|
|
local _NONE = 'NONE'
|
|
|
|
local _PALETTE_256 = 2
|
|
|
|
local _PALETTE_ANSI = 3
|
|
|
|
local _PALETTE_HEX = 1
|
|
|
|
local _TYPE_STRING = 'string'
|
|
|
|
local _TYPE_TABLE = 'table'
|
2020-06-22 02:38:18 +00:00
|
|
|
|
|
|
|
-- Determine which set of colors to use.
|
2020-11-19 17:12:03 +00:00
|
|
|
local _USING_HEX_OR_256 = tonumber(vim.o.t_Co) >= 256
|
2020-09-05 16:30:41 +00:00
|
|
|
or vim.o.termguicolors
|
2020-06-22 19:00:06 +00:00
|
|
|
or vim.fn.has('gui_running')
|
2020-11-19 17:12:03 +00:00
|
|
|
or string.find(vim.env.TERM, '256')
|
2020-06-22 02:38:18 +00:00
|
|
|
|
2020-11-19 17:12:03 +00:00
|
|
|
--[[
|
|
|
|
/*
|
|
|
|
* HELPER FUNCTIONS
|
|
|
|
*/
|
|
|
|
--]]
|
2020-06-22 02:38:18 +00:00
|
|
|
|
2020-11-19 17:12:03 +00:00
|
|
|
-- Add the 'blend' parameter to some highlight command, if there is one.
|
|
|
|
local function blend(command, attributes) -- {{{ †
|
|
|
|
if attributes.blend then -- There is a value for the `highlight-blend` field.
|
|
|
|
command[#command + 1] = ' blend='..attributes.blend
|
|
|
|
end
|
|
|
|
end --}}} ‡
|
2020-06-22 02:38:18 +00:00
|
|
|
|
2020-07-16 12:56:53 +00:00
|
|
|
-- Get the color value of a color variable, or "NONE" as a default.
|
2020-09-06 17:21:30 +00:00
|
|
|
local function get(color, index) -- {{{ †
|
2020-11-19 17:12:03 +00:00
|
|
|
if type(color) == _TYPE_TABLE and color[index] then
|
2020-07-16 12:56:53 +00:00
|
|
|
return color[index]
|
2020-11-19 17:12:03 +00:00
|
|
|
elseif type(color) == _TYPE_STRING then
|
2020-07-16 12:56:53 +00:00
|
|
|
return color
|
|
|
|
else
|
2020-11-19 17:12:03 +00:00
|
|
|
return _NONE
|
2020-07-16 12:56:53 +00:00
|
|
|
end
|
2020-09-06 17:21:30 +00:00
|
|
|
end --}}} ‡
|
2020-07-16 12:56:53 +00:00
|
|
|
|
|
|
|
--[[ If using hex and 256-bit colors, then populate the gui* and cterm* args.
|
|
|
|
If using 16-bit colors, just populate the cterm* args. ]]
|
2020-11-19 17:12:03 +00:00
|
|
|
local colorize = _USING_HEX_OR_256 and function(command, attributes) -- {{{ †
|
2020-09-06 16:33:48 +00:00
|
|
|
command[#command + 1] =
|
2020-11-19 17:12:03 +00:00
|
|
|
' ctermbg='..get(attributes.bg, _PALETTE_256)
|
|
|
|
..' ctermfg='..get(attributes.fg, _PALETTE_256)
|
|
|
|
..' guibg='..get(attributes.bg, _PALETTE_HEX)
|
|
|
|
..' guifg='..get(attributes.fg, _PALETTE_HEX)
|
2020-09-06 16:33:48 +00:00
|
|
|
blend(command, attributes)
|
|
|
|
end or function(command, attributes)
|
|
|
|
command[#command + 1] =
|
2020-11-19 17:12:03 +00:00
|
|
|
' ctermbg='..get(attributes.bg, _PALETTE_ANSI)
|
|
|
|
..' ctermfg='..get(attributes.fg, _PALETTE_ANSI)
|
2020-09-06 16:33:48 +00:00
|
|
|
blend(command, attributes)
|
2020-09-06 17:21:30 +00:00
|
|
|
end --}}} ‡
|
2020-07-16 12:56:53 +00:00
|
|
|
|
2020-11-19 17:12:03 +00:00
|
|
|
-- Detect attribute links and link them
|
|
|
|
local function link_attributes(attributes)
|
|
|
|
for attribute, val in pairs(attributes) do
|
|
|
|
if type(val) == _TYPE_STRING then print() end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2020-07-16 12:56:53 +00:00
|
|
|
-- This function appends `selected_attributes` to the end of `highlight_cmd`.
|
2020-11-19 17:12:03 +00:00
|
|
|
local stylize = _USING_HEX_OR_256 and function(command, style, color) -- {{{ †
|
2020-09-06 16:33:48 +00:00
|
|
|
command[#command + 1] = ' cterm='..style..' gui='..style
|
|
|
|
|
|
|
|
if color then -- There is an undercurl color.
|
2020-11-19 17:12:03 +00:00
|
|
|
command[#command + 1] = ' guisp='..get(color, _PALETTE_HEX)
|
2020-09-06 16:33:48 +00:00
|
|
|
end
|
|
|
|
end or function(command, style)
|
|
|
|
command[#command + 1] = ' cterm='..style
|
2020-09-06 17:21:30 +00:00
|
|
|
end --}}} ‡
|
2020-07-16 12:56:53 +00:00
|
|
|
|
2020-11-19 17:12:03 +00:00
|
|
|
-- Load specific &bg instructions
|
|
|
|
local function use_background_with(attributes)
|
|
|
|
attributes.__index = attributes
|
|
|
|
return setmetatable(attributes[vim.o.background], attributes)
|
|
|
|
end
|
|
|
|
|
|
|
|
--[[
|
|
|
|
/*
|
|
|
|
* MODULE
|
|
|
|
*/
|
|
|
|
--]]
|
|
|
|
|
|
|
|
local highlite = {}
|
|
|
|
|
2020-06-22 02:38:18 +00:00
|
|
|
-- Generate a `:highlight` command from a group and some attributes.
|
2020-11-13 20:43:16 +00:00
|
|
|
function highlite.highlight(highlight_group, attributes) -- {{{ †
|
2020-09-06 18:04:36 +00:00
|
|
|
-- The base highlight command
|
2020-06-22 02:38:18 +00:00
|
|
|
local highlight_cmd = {'hi! ', highlight_group}
|
2020-09-06 18:04:36 +00:00
|
|
|
|
|
|
|
-- Take care of special instructions for certain background colors.
|
|
|
|
if attributes[vim.o.background] then
|
2020-11-19 17:12:03 +00:00
|
|
|
attributes = use_background_with(attributes)
|
2020-09-06 18:04:36 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
-- Determine if there is a highlight link, and if so, assign it.
|
2020-11-19 17:12:03 +00:00
|
|
|
local link = (type(attributes) == _TYPE_STRING) and attributes or attributes.link
|
2020-06-22 02:38:18 +00:00
|
|
|
|
2020-07-16 01:37:03 +00:00
|
|
|
if link then -- `highlight_group` is a link to another group.
|
2020-07-16 12:56:53 +00:00
|
|
|
highlight_cmd[3] = highlight_cmd[2]..' '
|
2020-06-22 02:38:18 +00:00
|
|
|
highlight_cmd[2] = 'link '
|
2020-07-16 01:37:03 +00:00
|
|
|
highlight_cmd[4] = link
|
2020-08-28 17:26:37 +00:00
|
|
|
else -- The `highlight_group` is uniquely defined.
|
2020-11-19 17:12:03 +00:00
|
|
|
link_attributes(attributes)
|
2020-07-16 12:56:53 +00:00
|
|
|
colorize(highlight_cmd, attributes)
|
2020-07-16 03:30:14 +00:00
|
|
|
|
2020-11-19 17:12:03 +00:00
|
|
|
local style = attributes.style or _NONE
|
|
|
|
if type(style) == _TYPE_TABLE then
|
2020-07-16 12:56:53 +00:00
|
|
|
-- Concat all of the entries together with a comma between before styling.
|
2020-09-06 16:33:48 +00:00
|
|
|
stylize(highlight_cmd, table.concat(style, ','), style.color)
|
2020-09-06 17:29:09 +00:00
|
|
|
else -- The style is just a single entry.
|
2020-07-16 12:59:09 +00:00
|
|
|
stylize(highlight_cmd, style)
|
2020-06-20 00:49:05 +00:00
|
|
|
end
|
2020-06-22 02:38:18 +00:00
|
|
|
end
|
2020-06-20 00:49:05 +00:00
|
|
|
|
2020-11-19 17:12:03 +00:00
|
|
|
cmd(table.concat(highlight_cmd))
|
2020-06-22 02:38:18 +00:00
|
|
|
end --}}} ‡
|
|
|
|
|
2020-11-13 20:43:16 +00:00
|
|
|
function highlite:highlight_terminal(terminal_ansi_colors)
|
|
|
|
if self.using_hex_or_256 then for index, color in ipairs(terminal_ansi_colors) do
|
2020-11-19 17:12:03 +00:00
|
|
|
vim.g['terminal_color_'..index] = vim.o.termguicolors and color[_PALETTE_HEX] or color[_PALETTE_256]
|
2020-09-05 16:30:41 +00:00
|
|
|
end end
|
2020-06-20 00:49:05 +00:00
|
|
|
end
|
2020-11-13 20:43:16 +00:00
|
|
|
|
|
|
|
return setmetatable(highlite, {
|
|
|
|
['__call'] = function(self, normal, highlights, terminal_ansi_colors)
|
2020-11-19 17:12:03 +00:00
|
|
|
-- Clear the highlighting.
|
|
|
|
cmd('hi clear')
|
|
|
|
|
|
|
|
-- If the syntax has been enabled, reset it.
|
|
|
|
if vim.fn.exists('syntax_on') then cmd('syntax reset') end
|
|
|
|
|
|
|
|
-- If we aren't using hex nor 256 colorsets.
|
|
|
|
if not _USING_HEX_OR_256 then vim.o.t_Co = 16 end
|
|
|
|
|
2020-11-13 20:43:16 +00:00
|
|
|
-- Highlight the baseline.
|
|
|
|
self.highlight('Normal', normal)
|
|
|
|
|
|
|
|
-- Highlight everything else.
|
|
|
|
for highlight_group, attributes in pairs(highlights) do
|
|
|
|
self.highlight(highlight_group, attributes)
|
|
|
|
end
|
|
|
|
|
|
|
|
-- Set the terminal highlight colors.
|
|
|
|
self:highlight_terminal(terminal_ansi_colors)
|
|
|
|
end
|
|
|
|
})
|