40 lines
866 B
Lua
40 lines
866 B
Lua
local ls = require('luasnip')
|
|
local sn = ls.snippet_node
|
|
local i = ls.insert_node
|
|
|
|
local M = {}
|
|
|
|
M.fill = function(_, parent, args)
|
|
local title = ''
|
|
local chars = {}
|
|
|
|
if type(parent.snippet.env.LS_SELECT_RAW) == 'table' then
|
|
for _, ele in ipairs(parent.snippet.env.LS_SELECT_RAW) do
|
|
title = title .. ele .. ' '
|
|
end
|
|
end
|
|
|
|
for str in string.gmatch(args, '([^%%]+)') do
|
|
table.insert(chars, str)
|
|
end
|
|
|
|
local snip = chars[1] .. ' ' .. title
|
|
|
|
local _, c = unpack(vim.api.nvim_win_get_cursor(0))
|
|
|
|
while #snip < (vim.bo.tw - c - 1) do
|
|
snip = snip .. chars[2]
|
|
end
|
|
|
|
return snip
|
|
end
|
|
|
|
M.get_visual = function(_, parent)
|
|
if #parent.snippet.env.LS_SELECT_RAW > 0 then
|
|
return sn(nil, i(1, parent.snippet.env.LS_SELECT_RAW))
|
|
else -- If LS_SELECT_RAW is empty, return a blank insert node
|
|
return sn(nil, i(1))
|
|
end
|
|
end
|
|
|
|
return M
|