Add Python snippet example.
parent
2cfa18616b
commit
dd9a4b5493
|
@ -1,8 +1,13 @@
|
||||||
local s = require('luasnip').snippet
|
local c = require('luasnip').choice_node
|
||||||
|
local d = require('luasnip').dynamic_node
|
||||||
local i = require('luasnip').insert_node
|
local i = require('luasnip').insert_node
|
||||||
|
local sn = require('luasnip').snippet_node
|
||||||
|
local t = require('luasnip').text_node
|
||||||
|
|
||||||
|
local s = require('luasnip').snippet
|
||||||
local fmt = require('luasnip.extras.fmt').fmt
|
local fmt = require('luasnip.extras.fmt').fmt
|
||||||
|
|
||||||
local v = require('utils.snippets').visual_dynamic_node
|
local VISUAL = require('utils.luasnip').VISUAL
|
||||||
|
|
||||||
return {
|
return {
|
||||||
s(
|
s(
|
||||||
|
@ -10,7 +15,44 @@ return {
|
||||||
fmt([[
|
fmt([[
|
||||||
for {} in {}:
|
for {} in {}:
|
||||||
{}]],
|
{}]],
|
||||||
{ i(1, ''), i(2, ''), v(3) }
|
{
|
||||||
|
c(1, {
|
||||||
|
i(1, 'i'),
|
||||||
|
i(2, 'k'),
|
||||||
|
i(3, 'v'),
|
||||||
|
sn(4, { i(1, 'k'), t(', '), i(2, 'v') }),
|
||||||
|
}),
|
||||||
|
d(2, function(args, _, old_state)
|
||||||
|
local default = i(nil, 'val')
|
||||||
|
local snip = old_state or default
|
||||||
|
|
||||||
|
if args[1][1] == 'i' then
|
||||||
|
snip = sn(nil, c(1, {
|
||||||
|
{ t('range('), i(1, 'n'), t(')') },
|
||||||
|
default,
|
||||||
|
}))
|
||||||
|
elseif args[1][1] == 'k' then
|
||||||
|
snip = sn(nil, c(1, {
|
||||||
|
{ i(1, 'dict'), t('.keys()') },
|
||||||
|
default,
|
||||||
|
}))
|
||||||
|
elseif args[1][1] == 'v' then
|
||||||
|
snip = sn(nil, c(1, {
|
||||||
|
{ i(1, 'dict'), t('.values()') },
|
||||||
|
default,
|
||||||
|
}))
|
||||||
|
elseif args[1][1] == 'k, v' then
|
||||||
|
snip = sn(nil, c(1, {
|
||||||
|
{ i(1, 'dict'), t('.items()') },
|
||||||
|
default,
|
||||||
|
}))
|
||||||
|
end
|
||||||
|
|
||||||
|
snip.old_state = snip
|
||||||
|
return snip
|
||||||
|
end, { 1 }),
|
||||||
|
VISUAL,
|
||||||
|
}
|
||||||
)
|
)
|
||||||
),
|
),
|
||||||
}
|
}
|
||||||
|
|
|
@ -2,7 +2,7 @@ local M = {}
|
||||||
|
|
||||||
local cmp = require('cmp')
|
local cmp = require('cmp')
|
||||||
local cmp_buffer = require('cmp_buffer')
|
local cmp_buffer = require('cmp_buffer')
|
||||||
local luasnip = require("luasnip")
|
local luasnip = require('luasnip')
|
||||||
|
|
||||||
function M.setup()
|
function M.setup()
|
||||||
cmp.setup {
|
cmp.setup {
|
||||||
|
@ -12,9 +12,8 @@ function M.setup()
|
||||||
end,
|
end,
|
||||||
},
|
},
|
||||||
sources = {
|
sources = {
|
||||||
{
|
{ name = 'luasnip' },
|
||||||
name = 'nvim_lsp',
|
{ name = 'nvim_lsp' },
|
||||||
},
|
|
||||||
{
|
{
|
||||||
name = 'buffer',
|
name = 'buffer',
|
||||||
option = {
|
option = {
|
||||||
|
@ -31,29 +30,24 @@ function M.setup()
|
||||||
},
|
},
|
||||||
sorting = {
|
sorting = {
|
||||||
comparators = {
|
comparators = {
|
||||||
function (...)
|
function(...)
|
||||||
-- This also sorts completion results coming from other sources (e.g.
|
-- This also sorts completion results coming from other sources.
|
||||||
-- LSPs).
|
|
||||||
return cmp_buffer:compare_locality(...)
|
return cmp_buffer:compare_locality(...)
|
||||||
end,
|
end,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
mapping = {
|
mapping = {
|
||||||
['<c-n>'] = cmp.mapping(function(fallback)
|
['<tab>'] = cmp.mapping(function(fallback)
|
||||||
if cmp.visible() then cmp.select_next_item() else fallback() end
|
if cmp.get_active_entry() then cmp.confirm() else fallback() end
|
||||||
end, { 'i', 's' }),
|
|
||||||
|
|
||||||
['<c-p>'] = cmp.mapping(function(fallback)
|
|
||||||
if cmp.visible() then cmp.select_prev_item() else fallback() end
|
|
||||||
end, { 'i', 's' }),
|
end, { 'i', 's' }),
|
||||||
|
|
||||||
['<c-l>'] = cmp.mapping(function(fallback)
|
['<c-l>'] = cmp.mapping(function(fallback)
|
||||||
if cmp.visible() then cmp.abort() else fallback() end
|
if cmp.visible() then cmp.abort() else fallback() end
|
||||||
end, { 'i', 's' }),
|
end, { 'i', 's' }),
|
||||||
|
|
||||||
['<tab>'] = cmp.mapping(function(fallback)
|
['<c-n>'] = cmp.mapping(function(fallback)
|
||||||
if cmp.get_active_entry() then
|
if cmp.visible() then
|
||||||
cmp.confirm()
|
cmp.select_next_item()
|
||||||
elseif luasnip.expand_or_locally_jumpable() then
|
elseif luasnip.expand_or_locally_jumpable() then
|
||||||
luasnip.expand_or_jump()
|
luasnip.expand_or_jump()
|
||||||
else
|
else
|
||||||
|
@ -61,8 +55,10 @@ function M.setup()
|
||||||
end
|
end
|
||||||
end, { 'i', 's' }),
|
end, { 'i', 's' }),
|
||||||
|
|
||||||
['<s-tab>'] = cmp.mapping(function(fallback)
|
['<c-p>'] = cmp.mapping(function(fallback)
|
||||||
if luasnip.jumpable(-1) then
|
if cmp.visible() then
|
||||||
|
cmp.select_prev_item()
|
||||||
|
elseif luasnip.locally_jumpable(-1) then
|
||||||
luasnip.jump(-1)
|
luasnip.jump(-1)
|
||||||
else
|
else
|
||||||
fallback()
|
fallback()
|
||||||
|
|
|
@ -0,0 +1,64 @@
|
||||||
|
local M = {}
|
||||||
|
|
||||||
|
local luasnip = require('luasnip')
|
||||||
|
local types = require('luasnip.util.types')
|
||||||
|
local f = require('luasnip').function_node
|
||||||
|
|
||||||
|
M.VISUAL = f(function(_, snip)
|
||||||
|
local res, env = {}, snip.env
|
||||||
|
for _, ele in ipairs(env.LS_SELECT_RAW) do
|
||||||
|
table.insert(res, ele)
|
||||||
|
end
|
||||||
|
return res
|
||||||
|
end, {})
|
||||||
|
|
||||||
|
function M.setup()
|
||||||
|
luasnip.config.setup {
|
||||||
|
region_check_events = 'InsertEnter',
|
||||||
|
delete_check_events = 'InsertLeave',
|
||||||
|
store_selection_keys = '<tab>',
|
||||||
|
ext_opts = {
|
||||||
|
[types.snippet] = {
|
||||||
|
active = {
|
||||||
|
virt_text = { { '●', 'DiagnosticWarn' } },
|
||||||
|
},
|
||||||
|
},
|
||||||
|
[types.choiceNode] = {
|
||||||
|
active = {
|
||||||
|
virt_text = { { '⧨', 'DiagnosticHint' } },
|
||||||
|
-- Include in case one of our choice options is an empty string.
|
||||||
|
hl_group = 'DiagnosticOk',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
vim.keymap.set(
|
||||||
|
{ 'i', 's' },
|
||||||
|
'<c-e>',
|
||||||
|
function()
|
||||||
|
if luasnip.choice_active() then
|
||||||
|
luasnip.change_choice(1)
|
||||||
|
end
|
||||||
|
end,
|
||||||
|
{ silent = true }
|
||||||
|
)
|
||||||
|
vim.keymap.set(
|
||||||
|
{ 'i', 's' },
|
||||||
|
'<c-y>',
|
||||||
|
function()
|
||||||
|
if luasnip.choice_active() then
|
||||||
|
luasnip.change_choice(-1)
|
||||||
|
end
|
||||||
|
end,
|
||||||
|
{ silent = true }
|
||||||
|
)
|
||||||
|
-- Allow aborting the active snippet at any point in time.
|
||||||
|
vim.keymap.set(
|
||||||
|
{ 'n', 'i', 's' },
|
||||||
|
'<c-l>',
|
||||||
|
'<cmd>LuaSnipUnlinkCurrent<cr>'
|
||||||
|
)
|
||||||
|
end
|
||||||
|
|
||||||
|
return M
|
|
@ -1,28 +0,0 @@
|
||||||
local M = {}
|
|
||||||
|
|
||||||
local luasnip = require('luasnip')
|
|
||||||
local d = luasnip.dynamic_node
|
|
||||||
local i = luasnip.insert_node
|
|
||||||
local sn = luasnip.snippet_node
|
|
||||||
|
|
||||||
-- Creates an insertion node with a default value of whatever was stored in the
|
|
||||||
-- visual selection prior to entering insert mode (refer to the configuration
|
|
||||||
-- field `store_selection_keys`).
|
|
||||||
function M.visual_dynamic_node(index, default)
|
|
||||||
return d(index, function(_, parent)
|
|
||||||
local res, env = {}, parent.snippet.env
|
|
||||||
if type(env.LS_SELECT_RAW) ~= 'table' then
|
|
||||||
return sn(nil, { i(1, default or '') }, '')
|
|
||||||
end
|
|
||||||
for k, v in ipairs(env.LS_SELECT_RAW) do
|
|
||||||
local indent = env.CUSTOM_POS[2] - 1
|
|
||||||
table.insert(res, (k == 1 and v) or string.rep(' ', indent) .. v)
|
|
||||||
end
|
|
||||||
if table.concat(res):match('^%s*$') then
|
|
||||||
return sn(nil, { i(1, default or '') }, '')
|
|
||||||
end
|
|
||||||
return sn(nil, { i(1, res) }, '')
|
|
||||||
end, {})
|
|
||||||
end
|
|
||||||
|
|
||||||
return M
|
|
|
@ -30,6 +30,7 @@ let
|
||||||
luasnip = {
|
luasnip = {
|
||||||
plugin = pkgs.vimPlugins.luasnip;
|
plugin = pkgs.vimPlugins.luasnip;
|
||||||
config = ''
|
config = ''
|
||||||
|
require('utils.luasnip').setup()
|
||||||
${config.programs.neovim.nvim-snippets}
|
${config.programs.neovim.nvim-snippets}
|
||||||
'';
|
'';
|
||||||
};
|
};
|
||||||
|
|
Loading…
Reference in New Issue