Init Project

Author: TwIStOy

.stylua.toml

@@ -0,0 +1,4 @@
+indent_type = "Spaces"
+indent_width = 2
+column_width = 80
+call_parentheses = "NoSingleTable"

lua/luasnip-snippets/init.lua

@@ -0,0 +1,50 @@
+local M = {}
+
+-- dummy
+---@class LuaSnip.Snippet
+
+---@alias SnippetOrBuilder LuaSnip.Snippet|fun():SnippetOrBuilder
+
+---Register snippets to luasnip.
+---@param ft string filetype
+---@param snippets SnippetOrBuilder[]
+local function add_snippets(ft, snippets)
+  local ls = require("luasnip")
+  local ret = {}
+  for _, snippet in ipairs(snippets) do
+    if type(snippet) == "function" then
+      snippet = snippet()
+    end
+    ret[#ret + 1] = snippet
+  end
+  ls.add_snippets(ft, ret)
+end
+
+---Load snippets from ft module.
+---@param ft string
+---@return SnippetOrBuilder[]
+local function load_snippets(ft)
+  local snippets = require("luasnip-snippets.snippets." .. ft)
+  if type(snippets) == "function" then
+    snippets = snippets()
+  end
+  return snippets
+end
+
+---Load and register snippets.
+---@param fts string[]
+local function load_and_add_snippet(fts)
+  for _, ft in ipairs(fts) do
+    local snippets = load_snippets(ft)
+    add_snippets(ft, snippets)
+  end
+end
+
+function M.setup()
+  -- register snippets
+  load_and_add_snippet {
+    "rust",
+  }
+end
+
+return M

lua/luasnip-snippets/snippets/cpp/init.lua

@@ -0,0 +1,9 @@
+local Utils = require("luasnip-snippets.utils")
+
+local function setup()
+  return Utils.concat_snippets("luasnip-snippets.snippets.rust", {
+    "postfix",
+  })
+end
+
+return setup

lua/luasnip-snippets/snippets/rust/init.lua

@@ -0,0 +1,49 @@
+local ls = require("luasnip")
+local f = ls.function_node
+local tsp = require("luasnip.extras.treesitter_postfix")
+local Utils = require("luasnip-snippets.utils")
+
+local expr_node_types = {
+  "struct_expression",
+  "call_expression",
+  "identifier",
+}
+
+---@param trig string
+---@param expand string
+local function expr_tsp(trig, expand)
+  local name = ("(%s) %s"):format(trig, expand)
+  local dscr = ("Wrap expression with %s"):format(expand)
+  local replaced = expand:gsub("?", "%%s")
+
+  return tsp.treesitter_postfix({
+    trig = trig,
+    name = name,
+    dscr = dscr,
+    wordTrig = false,
+    reparseBuffer = nil,
+    matchTSNode = tsp.builtin.tsnode_matcher.find_topmost_types(
+      expr_node_types,
+      trig
+    ),
+  }, {
+    f(function(_, parent)
+      return Utils.replace_all(parent.snippet.env.LS_TSMATCH, replaced)
+    end, {}),
+  })
+end
+
+return {
+  expr_tsp(".rc", "Rc::new(?)"),
+  expr_tsp(".arc", "Arc::new(?)"),
+  expr_tsp(".box", "Box::new(?)"),
+  expr_tsp(".mu", "Mutex::new(?)"),
+  expr_tsp(".rw", "RwLock::new(?)"),
+  expr_tsp(".cell", "Cell::new(?)"),
+  expr_tsp(".refcell", "RefCell::new(?)"),
+  expr_tsp(".ref", "&?"),
+  expr_tsp(".refm", "&mut ?"),
+  expr_tsp(".ok", "Ok(?)"),
+  expr_tsp(".err", "Err(?)"),
+  expr_tsp(".some", "Some(?)"),
+}

lua/luasnip-snippets/snippets/rust/postfix.lua

@@ -0,0 +1,40 @@
+local M = {}
+
+---Replace all occurences of %s in template with match.
+---@param match string|string[]
+---@param template string
+---@return string[]
+function M.replace_all(match, template)
+  ---@type string
+  local match_str = ""
+  if type(match) == "table" then
+    match_str = table.concat(match, "\n")
+  else
+    match_str = match
+  end
+
+  local ret = template:gsub("%%s", match_str)
+  local ret_lines = vim.split(ret, "\n", {
+    trimempty = false,
+  })
+
+  return ret_lines
+end
+
+---Load and concat snippets.
+---@param base string
+---@param snippets string[]
+---@return LuaSnip.Snippet[]
+function M.concat_snippets(base, snippets)
+  local ret = {}
+  for _, snippet in ipairs(snippets) do
+    local snippet_module = require(base .. "." .. snippet)
+    if type(snippet_module) == "function" then
+      snippet_module = snippet_module()
+    end
+    vim.list_extend(ret, snippet_module)
+  end
+  return ret
+end
+
+return M