aboutsummaryrefslogtreecommitdiffstats
path: root/lua/alpha/snips/ft/lua.lua
diff options
context:
space:
mode:
Diffstat (limited to 'lua/alpha/snips/ft/lua.lua')
-rw-r--r--lua/alpha/snips/ft/lua.lua46
1 files changed, 46 insertions, 0 deletions
diff --git a/lua/alpha/snips/ft/lua.lua b/lua/alpha/snips/ft/lua.lua
new file mode 100644
index 0000000..539751e
--- /dev/null
+++ b/lua/alpha/snips/ft/lua.lua
@@ -0,0 +1,46 @@
+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 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",
+ },
+}