feat(cpp): add ns

TwIStOy · TwIStOy · commit 942d42beca78 · 2024-01-03T22:04:07.000+08:00
diff --git a/README.md b/README.md
@@ -46,12 +46,13 @@ ls.setup({
 
 #### Normal Snippets
 
-|    Trig    | Desc                                                                                             | Context Required |
-| :--------: | ------------------------------------------------------------------------------------------------ | :--------------: |
-|    `fn`    | Expand to lambda function in argument list or function body, otherwise expand to normal function |        No        |
-| `\|trans`  | Expand to ranges::views::transform pipe.                                                         |        No        |
-| `\|filter` | Expand to ranges::views::filter pipe.                                                            |        No        |
-|   `cpo`    | Expand to customize point object.                                                                |        No        |
+|    Trig     | Desc                                                                                             | Context Required |
+| :---------: | ------------------------------------------------------------------------------------------------ | :--------------: |
+|    `fn`     | Expand to lambda function in argument list or function body, otherwise expand to normal function |        No        |
+|  `\|trans`  | Expand to ranges::views::transform pipe.                                                         |        No        |
+| `\|filter`  | Expand to ranges::views::filter pipe.                                                            |        No        |
+|    `cpo`    | Expand to customize point object.                                                                |        No        |
+| `ns%s(%S+)` | Expand to namespace block (including comments).                                                  |        No        |
 
 #### Auto-snippets
 
@@ -115,6 +116,7 @@ ls.setup({
 |   `.ts`   | Switch indent's coding style between `CamelCase` and `snake_case`. |      `indent`      |
 |   `.sc`   | Wraps with `static_cast<>(?)`                                      |     `any_expr`     |
 | `.single` | Wraps with `ranges::views::single(?)`                              |     `any_expr`     |
+|   `.in`   | Expands to `if (...find)` statements.                              |     `any_expr`     |
 
 </details>
 
diff --git a/lua/luasnip-snippets/snippets/cpp/default.lua b/lua/luasnip-snippets/snippets/cpp/default.lua
@@ -207,4 +207,36 @@ return {
       return quick_type(shortcut)
     end),
   }),
+
+  snippet {
+    "ns%s+(%S+)",
+    name = "namespace",
+    dscr = "namespace",
+    mode = "br",
+    nodes = fmta(
+      [[
+      namespace <name> {
+      <body>
+      }  // namespace <name>
+      ]],
+      {
+        body = i(0),
+        name = f(function(_, snip)
+          local parts = vim.split(snip.captures[1], "::", {
+            plain = true,
+            trimempty = true,
+          })
+          local names = {}
+          for _, part in ipairs(parts) do
+            local nest_parts = vim.split(part, ".", {
+              plain = true,
+              trimempty = true,
+            })
+            vim.list_extend(names, nest_parts)
+          end
+          return table.concat(names, "::")
+        end),
+      }
+    ),
+  },
 }
diff --git a/lua/luasnip-snippets/snippets/cpp/postfix.lua b/lua/luasnip-snippets/snippets/cpp/postfix.lua
@@ -3,6 +3,7 @@ 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 fmta = require("luasnip.extras.fmt").fmta
 local i = require("luasnip-snippets.nodes").insert_node
 
 local expr_query = [[
@@ -101,7 +102,7 @@ return {
   tsp.treesitter_postfix(
     {
       trig = ".sc",
-      name = "static_cast<TYPE>(?)",
+      name = "(.sc) static_cast<TYPE>(?)",
       dscr = "Wraps an expression with static_cast<TYPE>(?)",
       wordTrig = false,
       reparseBuffer = "live",
@@ -123,4 +124,32 @@ return {
       }
     )
   ),
+
+  tsp.treesitter_postfix(
+    {
+      trig = ".in",
+      name = "(.in) if (...find)",
+      dscr = "Expands to an if-expr to find an element in map-like object",
+      wordTrig = false,
+      reparseBuffer = "live",
+      matchTSNode = {
+        query = expr_query,
+        query_lang = "cpp",
+      },
+    },
+    fmta(
+      [[
+      if (auto it = <expr>.find(<key>); it != <expr>.end()) {
+        <cursor>
+      }
+      ]],
+      {
+        cursor = i(0),
+        key = i(1, "Key"),
+        expr = f(function(_, parent)
+          return Utils.replace_all(parent.snippet.env.LS_TSMATCH, "%s")
+        end, {}),
+      }
+    )
+  ),
 }
