paper-tonic/lua/highlite.lua

194 lines
5.6 KiB
Lua
Raw Normal View History

2020-07-02 17:06:31 +00:00
--[[ NOTHING INSIDE THIS FILE NEEDS TO BE EDITED BY THE USER. ]]
--[[
/*
* IMPORTS
*/
--]]
local vim = vim
local api = vim.api
local cmd = api.nvim_command
local fn = vim.fn
2020-06-22 02:38:18 +00:00
--[[
/*
* VARS
*/
--]]
2020-06-22 02:38:18 +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 21:42:31 +00:00
local _USE_HEX = vim.o.termguicolors
local _USE_256 = tonumber(vim.o.t_Co) >= 256
or string.find(vim.env.TERM, '256')
2020-06-22 02:38:18 +00:00
--[[
/*
* HELPER FUNCTIONS
*/
--]]
2020-06-22 02:38:18 +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-11-19 21:53:35 +00:00
-- filter a highlight group's style information
local function filter_group_style(value)
return value ~= 'background'
and value ~= 'blend'
and value ~= 'foreground'
and value ~= 'special'
end
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) -- {{{ †
if type(color) == _TYPE_TABLE and color[index] then
2020-07-16 12:56:53 +00:00
return color[index]
elseif type(color) == _TYPE_STRING then
2020-07-16 12:56:53 +00:00
return color
else
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 21:42:31 +00:00
local colorize = _USE_HEX and function(command, attributes) -- {{{ †
command[#command+1]=' guibg='..get(attributes.bg, _PALETTE_HEX)..' guifg='..get(attributes.fg, _PALETTE_HEX)
end or _USE_256 and function(command, attributes)
command[#command+1]=' ctermbg='..get(attributes.bg, _PALETTE_256)..' ctermfg='..get(attributes.fg, _PALETTE_256)
end or function(command, attributes)
2020-11-19 21:42:31 +00:00
command[#command+1]=' ctermbg='..get(attributes.bg, _PALETTE_ANSI)..' ctermfg='..get(attributes.fg, _PALETTE_ANSI)
2020-09-06 17:21:30 +00:00
end --}}} ‡
2020-07-16 12:56:53 +00:00
-- This function appends `selected_attributes` to the end of `highlight_cmd`.
2020-11-19 21:42:31 +00:00
local stylize = _USE_HEX and function(command, style, color)
command[#command+1]=' gui='..style
if color then -- There is an undercurl color.
command[#command + 1] = ' guisp='..get(color, _PALETTE_HEX)
end
end or function(command, style)
command[#command + 1] = ' cterm='..style
2020-11-19 21:42:31 +00:00
end
2020-07-16 12:56:53 +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-11-19 21:42:31 +00:00
function highlite.group(group_name)
local no_errors, group_definition = pcall(api.nvim_get_hl_by_name, group_name, vim.o.termguicolors)
if not no_errors then group_definition = {} end
2020-11-19 21:53:35 +00:00
-- the string.fmt expression to convert RGB to HEX
2020-11-19 21:42:31 +00:00
local fmt = '#%06x'
2020-11-19 21:53:35 +00:00
-- the style of the highlight group
local style = vim.tbl_filter(filter_group_style, vim.tbl_keys(group_definition))
if group_definition.special then
style.color = string.format(fmt, group_definition.special)
2020-11-19 21:42:31 +00:00
end
return {
['fg'] = group_definition.foreground and string.format(fmt, group_definition.foreground) or _NONE,
['bg'] = group_definition.background and string.format(fmt, group_definition.background) or _NONE,
['blend'] = group_definition.blend,
['style'] = style or _NONE
}
end
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
2020-11-19 21:42:31 +00:00
if type(attributes) == _TYPE_STRING then -- `highlight_group` is a link to another group.
highlight_cmd[3] = highlight_cmd[2]
2020-06-22 02:38:18 +00:00
highlight_cmd[2] = 'link '
highlight_cmd[4] = ' '
2020-11-19 21:42:31 +00:00
highlight_cmd[5] = attributes
2020-08-28 17:26:37 +00:00
else -- The `highlight_group` is uniquely defined.
2020-11-19 21:42:31 +00:00
-- Take care of special instructions for certain background colors.
if attributes[vim.o.background] then
attributes = use_background_with(attributes)
end
2020-07-16 12:56:53 +00:00
colorize(highlight_cmd, attributes)
2020-11-19 21:42:31 +00:00
blend(highlight_cmd, attributes)
2020-07-16 03:30:14 +00:00
local style = attributes.style or _NONE
2020-11-19 21:42:31 +00:00
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.
stylize(highlight_cmd, table.concat(style, ','), style.color)
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
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
vim.g['terminal_color_'..index] = vim.o.termguicolors and color[_PALETTE_HEX] or color[_PALETTE_256]
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 21:42:31 +00:00
-- save the colors_name before syntax reset
local color_name = vim.g.colors_name
-- Clear the highlighting.
cmd('hi clear')
-- If the syntax has been enabled, reset it.
if fn.exists('syntax_on') then cmd('syntax reset') end
2020-11-19 21:42:31 +00:00
-- replace the colors_name
vim.g.colors_name = color_name
color_name = nil
-- If we aren't using hex nor 256 colorsets.
2020-11-19 21:42:31 +00:00
if not (_USE_HEX or _USE_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
2020-11-19 21:42:31 +00:00
if type(attributes) == 'function' then
attributes = attributes(highlights)
end
2020-11-13 20:43:16 +00:00
self.highlight(highlight_group, attributes)
end
-- Set the terminal highlight colors.
self:highlight_terminal(terminal_ansi_colors)
end
})