This commit is contained in:
Alex Yatskov 2024-08-04 11:33:42 -07:00
parent a4933b4d11
commit 637e3a4c7b

View File

@ -8,6 +8,20 @@ local GuidConfig = {
object_char = 'g', object_char = 'g',
} }
function GuidConfig.setup(config)
if config then
for key, value in pairs(config) do
GuidConfig[key] = config[key] or value
end
end
if GuidConfig.object_char then
for _, mode in ipairs({'x', 'o'}) do
for _, prefix in ipairs({'i', 'a'}) do vim.api.nvim_set_keymap(mode, prefix .. GuidConfig.object_char, ':<C-u>GuidObject<cr>', {noremap = true, silent = true}) end
end
end
end
-- --
-- Guid -- Guid
-- --
@ -80,10 +94,10 @@ function Guid.parse(text)
table.insert(bytes, tonumber(text_byte, 16)) table.insert(bytes, tonumber(text_byte, 16))
end end
return bytes return setmetatable(bytes, Guid)
end end
function Guid.print(guid, style) function Guid:print(style)
if style == '' then if style == '' then
style = GuidConfig.default_style style = GuidConfig.default_style
end end
@ -113,7 +127,7 @@ function Guid.print(guid, style)
return return
end end
local guid_printed = string.format(format, unpack(guid)) local guid_printed = string.format(format, unpack(self))
if style:upper() == style then if style:upper() == style then
guid_printed = guid_printed:upper():gsub('X', 'x') guid_printed = guid_printed:upper():gsub('X', 'x')
end end
@ -125,167 +139,51 @@ function Guid.print(guid, style)
return guid_printed return guid_printed
end end
local GuidPatterns = { --
'{\\s*0x[0-9a-fA-F]\\{8\\},\\s*0x[0-9a-fA-F]\\{4\\},\\s*0x[0-9a-fA-F]\\{4\\},\\s*{\\s*0x[0-9a-fA-F]\\{2\\},\\s*0x[0-9a-fA-F]\\{2\\},\\s*0x[0-9a-fA-F]\\{2\\},\\s*0x[0-9a-fA-F]\\{2\\},\\s*0x[0-9a-fA-F]\\{2\\},\\s*0x[0-9a-fA-F]\\{2\\},\\s*0x[0-9a-fA-F]\\{2\\},\\s*0x[0-9a-fA-F]\\{2\\}\\s*}\\s*}', -- x -- Functions
'(\\s*[0-9a-fA-F]\\{8\\}-[0-9a-fA-F]\\{4\\}-[0-9a-fA-F]\\{4\\}-[0-9a-fA-F]\\{4\\}-[0-9a-fA-F]\\{12\\}\\s*)', -- p --
'{\\s*[0-9a-fA-F]\\{8\\}-[0-9a-fA-F]\\{4\\}-[0-9a-fA-F]\\{4\\}-[0-9a-fA-F]\\{4\\}-[0-9a-fA-F]\\{12\\}\\s*}', -- b
'[0-9a-fA-F]\\{8\\}-[0-9a-fA-F]\\{4\\}-[0-9a-fA-F]\\{4\\}-[0-9a-fA-F]\\{4\\}-[0-9a-fA-F]\\{12\\}', -- d
'[0-9a-fA-F]\\{32\\}', -- n
}
local function setup(config)
if config then
for key, value in pairs(config) do
GuidConfig[key] = config[key] or value
end
end
if GuidConfig.object_char then
for _, mode in ipairs({'x', 'o'}) do
for _, prefix in ipairs({'i', 'a'}) do vim.api.nvim_set_keymap(mode, prefix .. GuidConfig.object_char, ':<C-u>GuidObject<cr>', {noremap = true, silent = true}) end
end
end
end
local function get_cursor_pos() local function get_cursor_pos()
local _, row, col, _ = unpack(vim.fn.getpos('.')) local _, row, col, _ = unpack(vim.fn.getpos('.'))
return {row = row, col = col} return {row = row, col = col}
end end
local function insert_text_at_pos(text, pos) local function guid_insert(style)
local insert_text_at_pos = function(text, pos)
local line = vim.fn.getline(pos.row) local line = vim.fn.getline(pos.row)
local prefix = string.sub(line, 0, pos.col - 1) local prefix = string.sub(line, 0, pos.col - 1)
local suffix = string.sub(line, pos.col) local suffix = string.sub(line, pos.col)
vim.fn.setline(pos.row, prefix .. text .. suffix) vim.fn.setline(pos.row, prefix .. text .. suffix)
end end
local function find_pattern(pattern, flags) local guid = Guid:generate()
local row, col = unpack(vim.fn.searchpos(pattern, flags)) local guid_printed = guid:print(style)
if row ~= 0 or col ~= 0 then
local text = vim.fn.matchstr(vim.fn.getline(row), pattern)
return {row = row, col = col, text = text}
end
end
local function find_pattern_at_pos(pattern, pos, check_col)
for _, flags in ipairs({'cnW', 'bnW'}) do
local match_pos = find_pattern(pattern, flags)
if match_pos and match_pos.row == pos.row and (not check_col or match_pos.col <= pos.col and match_pos.col + #match_pos.text > pos.col) then
return match_pos
end
end
end
local function guid_generate()
-- Generate a pseudo-random GUID according to RFC 4122:
-- https://www.rfc-editor.org/rfc/rfc4122
-- Set all bits to randomly (or pseudo-randomly) chosen values.
local bytes = {}
for i = 1, 16 do
bytes[i] = math.random(0, 255)
end
-- Set the two most significant bits (bits 6 and 7) of the
-- clock_seq_hi_and_reserved to zero and one, respectively.
bytes[9] = bit.band(bit.bor(bytes[9], 0x80), 0xbf)
-- Set the four most significant bits (bits 12 through 15) of the
-- time_hi_and_version field to the 4-bit version number.
bytes[7] = bit.band(bit.bor(bytes[7], 0x40), bit.lshift(4, 4))
return bytes
end
local function guid_parse(text)
local text_stripped = text:gsub('[{}()%-, ]', ''):gsub('0x', '')
assert(#text_stripped == 32)
local bytes = {}
for i = 0, 30, 2 do
local text_byte = text_stripped:sub(1 + i, 2 + i)
table.insert(bytes, tonumber(text_byte, 16))
end
return bytes
end
local function guid_print(guid, style)
if style == '' then
style = GuidConfig.default_style
end
-- Format specifier definition:
-- https://learn.microsoft.com/en-us/dotnet/api/system.guid.tostring?view=net-7.0
local format = nil
local style_lower = style:lower()
if style_lower == 'n' then
-- 00000000000000000000000000000000
format = '%.2x%.2x%.2x%.2x%.2x%.2x%.2x%.2x%.2x%.2x%.2x%.2x%.2x%.2x%.2x%.2x'
elseif style_lower == 'd' then
-- 00000000-0000-0000-0000-000000000000
format = '%.2x%.2x%.2x%.2x-%.2x%.2x-%.2x%.2x-%.2x%.2x-%.2x%.2x%.2x%.2x%.2x%.2x'
elseif style_lower == 'b' then
-- {00000000-0000-0000-0000-000000000000}
format = '{%.2x%.2x%.2x%.2x-%.2x%.2x-%.2x%.2x-%.2x%.2x-%.2x%.2x%.2x%.2x%.2x%.2x}'
elseif style_lower == 'p' then
-- (00000000-0000-0000-0000-000000000000)
format = '(%.2x%.2x%.2x%.2x-%.2x%.2x-%.2x%.2x-%.2x%.2x-%.2x%.2x%.2x%.2x%.2x%.2x)'
elseif style_lower == 'x' then
-- {0x00000000,0x0000,0x0000,{0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00}}
format = '{0x%.2x%.2x%.2x%.2x,0x%.2x%.2x,0x%.2x%.2x,{0x%.2x,0x%.2x,0x%.2x,0x%.2x,0x%.2x,0x%.2x,0x%.2x,0x%.2x}}'
else
vim.api.nvim_notify('Unrecognized GUID format!', vim.log.levels.ERROR, {})
return
end
local guid_printed = string.format(format, unpack(guid))
if style:upper() == style then
guid_printed = guid_printed:upper():gsub('X', 'x')
end
if GuidConfig.comma_space then
guid_printed = guid_printed:gsub(',', ', ')
end
return guid_printed
end
local function guid_insert(style)
local guid_printed = guid_print(guid_generate(), style)
if guid_printed then if guid_printed then
insert_text_at_pos(guid_printed, get_cursor_pos()) insert_text_at_pos(guid_printed, get_cursor_pos())
end end
end end
local function guid_format(style) local function guid_format(style)
for _, guid_pattern in ipairs(GuidPatterns) do local guid_match = Guid.find(get_cursor_pos(), true)
local match_pos = find_pattern_at_pos(guid_pattern, get_cursor_pos(), true) if guid_match then
if match_pos then local line = vim.fn.getline(guid_match.row)
local line = vim.fn.getline(match_pos.row) local line_prefix = line:sub(1, guid_match.col - 1)
local line_prefix = line:sub(1, match_pos.col - 1) local line_suffix = line:sub(guid_match.col + #guid_match.text)
local line_suffix = line:sub(match_pos.col + #match_pos.text) local guid_printed = guid_match.guid:print(style)
local guid_printed = guid_print(guid_parse(match_pos.text), style)
if guid_printed then if guid_printed then
vim.fn.setline(match_pos.row, line_prefix .. guid_printed .. line_suffix) vim.fn.setline(guid_match.row, line_prefix .. guid_printed .. line_suffix)
end end
return else
end
end
vim.api.nvim_notify('No GUID found at cursor!', vim.log.levels.ERROR, {}) vim.api.nvim_notify('No GUID found at cursor!', vim.log.levels.ERROR, {})
end end
end
local function guid_object() local function guid_object()
for _, guid_pattern in ipairs(GuidPatterns) do local guid_match = Guid.find(get_cursor_pos(), false)
local match_pos = find_pattern_at_pos(guid_pattern, get_cursor_pos(), false) if guid_match then
if match_pos then
local cursor_pos = get_cursor_pos() local cursor_pos = get_cursor_pos()
vim.fn.cursor({cursor_pos.row, match_pos.col}) vim.fn.cursor({cursor_pos.row, guid_match.col})
vim.cmd(string.format(':normal! v%dl', #match_pos.text - 1)) vim.cmd(string.format(':normal! v%dl', #guid_match.text - 1))
return
end
end end
end end
@ -293,5 +191,5 @@ return {
guid_format = guid_format, guid_format = guid_format,
guid_insert = guid_insert, guid_insert = guid_insert,
guid_object = guid_object, guid_object = guid_object,
setup = setup, setup = GuidConfig.setup,
} }