Add postfixes

Author: TwIStOy

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

@@ -4,6 +4,7 @@ local function setup()
   return Utils.concat_snippets("luasnip-snippets.snippets.cpp", {
     "statements",
     "lambda_fn",
+    "postfix",
   })
 end
 

lua/luasnip-snippets/snippets/cpp/lambda_fn.lua

@@ -3,7 +3,6 @@ local UtilsTS = require("luasnip-snippets.utils.treesitter")
 local d = ls.dynamic_node
 local sn = ls.snippet_node
 local t = ls.text_node
-local f = ls.function_node
 local fmta = require("luasnip.extras.fmt").fmta
 local CppCommons = require("luasnip-snippets.snippets.cpp.commons")
 local i = ls.insert_node
@@ -55,7 +54,7 @@ local function is_qualified_function(node)
   return false
 end
 
-local function inject_expanding_environment(_, line_to_cursor, match, captures)
+local function inject_expanding_environment(_, _, match, captures)
   local row, col = unpack(vim.api.nvim_win_get_cursor(0))
   local buf = vim.api.nvim_get_current_buf()
 

lua/luasnip-snippets/snippets/cpp/postfix.lua

@@ -0,0 +1,85 @@
+local ls = require("luasnip")
+local f = ls.function_node
+local tsp = require("luasnip.extras.treesitter_postfix")
+local Utils = require("luasnip-snippets.utils")
+local fmt = require("luasnip.extras.fmt").fmt
+local i = require("luasnip-snippets.nodes").insert_node
+local c = require("luasnip-snippets.nodes").choice_node
+local t = ls.text_node
+
+local expr_query = [[
+[
+  (call_expression)
+  (identifier)
+  (template_function)
+  (subscript_expression)
+  (field_expression)
+  (user_defined_literal)
+] @prefix
+]]
+
+---@param trig string
+---@param expand string
+---@param dscr string?
+local function expr_tsp(trig, expand, dscr)
+  local name = ("(%s) %s"):format(trig, expand)
+  if dscr == nil then
+    dscr = ("Wraps an expression with %s"):format(expand)
+  else
+    dscr = dscr:format(expand)
+  end
+  local replaced = expand:gsub("?", "%%s")
+
+  return tsp.treesitter_postfix({
+    trig = trig,
+    name = name,
+    dscr = dscr,
+    wordTrig = false,
+    reparseBuffer = "live",
+    matchTSNode = {
+      query = expr_query,
+      query_lang = "cpp",
+    },
+  }, {
+    f(function(_, parent)
+      return Utils.replace_all(parent.snippet.env.LS_TSMATCH, replaced)
+    end, {}),
+  })
+end
+
+return {
+  expr_tsp(
+    ".be",
+    "?.begin(), ?.end()",
+    "Completes an expr with both begin() and end()"
+  ),
+  expr_tsp(".mv", "std::move(?)"),
+  expr_tsp(".fwd", "std::forward<decltype(?)>(?)"),
+  expr_tsp(".val", "std::declval<?>()"),
+  expr_tsp(".dt", "decltype(?)"),
+  tsp.treesitter_postfix(
+    {
+      trig = ".sc",
+      name = "static_cast<TYPE>(?)",
+      dscr = "Wraps an expression with static_cast<TYPE>(?)",
+      wordTrig = false,
+      reparseBuffer = "live",
+      matchTSNode = {
+        query = expr_query,
+        query_lang = "cpp",
+      },
+    },
+    fmt(
+      [[
+      static_cast<{body}>({expr}){end}
+      ]],
+      {
+        body = i(1),
+        expr = f(function(_, parent)
+          return Utils.replace_all(parent.snippet.env.LS_TSMATCH, "%s")
+        end, {}),
+        ["end"] = i(0),
+      }
+    )
+  ),
+}

lua/luasnip-snippets/snippets/cpp/statements.lua

@@ -3,11 +3,7 @@ local UtilsTS = require("luasnip-snippets.utils.treesitter")
 local d = ls.dynamic_node
 local sn = ls.snippet_node
 local t = ls.text_node
-local f = ls.function_node
 local fmta = require("luasnip.extras.fmt").fmta
-local CppCommons = require("luasnip-snippets.snippets.cpp.commons")
-local i = ls.insert_node
-local c = ls.choice_node
 
 local function inject_class_name(_, line_to_cursor, match, captures)
   -- check if at the line begin

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

@@ -36,7 +36,7 @@ local expr_node_types = {
 ---@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 dscr = ("Wraps an expression with %s"):format(expand)
   local replaced = expand:gsub("?", "%%s")
 
   return tsp.treesitter_postfix({