aboutsummaryrefslogtreecommitdiffstats
path: root/lua/alpha/snips/ft/lua.lua
blob: 5ea789ee0273b1ffa64006363cc45c24d5157d74 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
local ls = require("luasnip")

-- local snippet = ls.s
local snippet_from_nodes = ls.sn
local c = ls.choice_node
local d = ls.dynamic_node
local i = ls.insert_node
local f = ls.function_node
local t = ls.text_node

local fmt = require("luasnip.extras.fmt").fmt

local require_var = function(args, _)
	local text = args[1][1] or ""
	local split = vim.split(text, ".", { plain = true })

	local options = {}
	for len = 0, #split - 1 do
		table.insert(options, t(table.concat(vim.list_slice(split, #split - len, #split), "_")))
	end

	return snippet_from_nodes(nil, {
		c(1, options),
	})
end

local newline = function(text)
	return t({ "", text })
end

return {
	req = fmt([[local {} = require("{}")]], {
		d(2, require_var, { 1 }),
		i(1),
	}),
	lf = {
		desc = "table function",
		"local ",
		i(1),
		" = function(",
		i(2),
		")",
		newline("  "),
		i(0),
		newline("end"),
	},

	preq = fmt(
		[[local ok, {} = pcall(require, "{}")
if not ok then
    return
end]],
		{
            f(function(args)
                return args[1][1]
            end, { 1 }),
            i(1)
        }
	),
}