feat(cpp): add selected code wrapper #define

Author: TwIStOy

README.md

@@ -153,10 +153,11 @@ Snippets with `*` are available only when `vim_snippet` is enabled.
 |    `itf`    | Expands to a struct with default virtual destruction.                                              |         No          |     |
 |    `pvf`    | Expands to a pure virtual function declaration.                                                    |         No          |     |
 |   `qcls`    | Expands to a class inherts from QObject.                                                           |         No          | Yes |
-|    `#if`    | Expands to `#if` statement.                                                                        | After cut selection | No  |
-|    `if`     | Expands to `if` statement.                                                                         | After cut selection | No  |
-|    `do`     | Expands to `do ... while (0)` statement.                                                           | After cut selection | No  |
-|   `while`   | Expands to `while` statement.                                                                      | After cut selection | No  |
+|    `#if`    | Wrap selected code in `#if ... #endif` block.                                                      | After cut selection | No  |
+|    `if`     | Wrap selected code in `if (...)` block.                                                            | After cut selection | No  |
+|    `do`     | Wrap selected code in `do ... while(0)` block.                                                     | After cut selection | No  |
+|   `while`   | Wrap selected code in `while (...)` block.                                                         | After cut selection | No  |
+|    `#de`    | Wrap selected code in `#define ...` block.                                                         | After cut selection | No  |
 
 #### Auto-snippets
 

lua/luasnip-snippets/snippets/cpp/selection.lua

@@ -34,6 +34,20 @@ local function fix_leading_whitespace(lines, indent)
   return ret
 end
 
+local function add_trailing_slash(lines)
+  local ret = {}
+  local max_len = 0
+  for _, line in ipairs(lines) do
+    max_len = math.max(max_len, #line)
+  end
+  for _, line in ipairs(lines) do
+    local len = #line
+    local diff = max_len - len
+    table.insert(ret, line .. string.rep(" ", diff) .. " \\")
+  end
+  return ret
+end
+
 local function has_select_raw_fn(_, _, _)
   return Utils.get_buf_var(0, "LUASNIP_SELECT_RAW") ~= nil
 end
@@ -43,7 +57,7 @@ return {
   snippet {
     "#if",
     name = "(#if) #if ... #endif",
-    dscr = "Expands to #if ... #endif",
+    dscr = "Wrap selected code in #if ... #endif block",
     mode = "bw",
     lang = "cpp",
     cond = has_select_raw,
@@ -68,7 +82,7 @@ return {
   snippet {
     "if",
     name = "(if) if (...) { ... }",
-    dscr = "Expands to if expression",
+    dscr = "Wrap selected code in if (...) { ... } block",
     mode = "bw",
     lang = "cpp",
     cond = has_select_raw,
@@ -92,7 +106,7 @@ return {
   snippet {
     "do",
     name = "(do) do { ... } while (0)",
-    dscr = "Expands to do ... while(0) expression",
+    dscr = "Wrap selected code in do { ... } while (0) block",
     mode = "bw",
     lang = "cpp",
     cond = has_select_raw,
@@ -115,7 +129,7 @@ return {
   snippet {
     "while",
     name = "(while) while (...) { ... }",
-    dscr = "Expands to while (...) { ... } expression",
+    dscr = "Wrap selected code in while (...) { ... } block",
     mode = "bw",
     lang = "cpp",
     cond = has_select_raw,
@@ -135,4 +149,27 @@ return {
       }
     ),
   },
+
+  snippet {
+    "#de",
+    name = "(#de) #define ...",
+    dscr = "Wrap selected code in #define block",
+    mode = "bw",
+    lang = "cpp",
+    cond = has_select_raw,
+    nodes = fmta(
+      [[
+      #define <name>() \
+        <selected><cursor>
+      ]],
+      {
+        name = i(1, "condition"),
+        cursor = i(0),
+        selected = f(function(_, snip)
+          local _, env = {}, snip.env
+          return fix_leading_whitespace(add_trailing_slash(env.LS_SELECT_RAW))
+        end),
+      }
+    ),
+  },
 }