feat(rust): add new snippet 'pm'

Author: TwIStOy

README.md

@@ -231,13 +231,15 @@ tmss! -> absl::flat_hash_map<std::string, std::string>
 
 #### Normal Snippets
 
-| Trig | Desc                                                                                               | Context Required |
-| :--: | -------------------------------------------------------------------------------------------------- | :--------------: |
-| `fn` | Expands to lambda function in argument list or function body, otherwise expand to normal function. |        No        |
-| `pc` | Expands to `pub(crate)`.                                                                           |        No        |
-| `ps` | Expands to `pub(super)`.                                                                           |        No        |
-| `ii` | Expands to `#[inline]`.                                                                            |        No        |
-| `ia` | Expands to `#[inline(always)]`.                                                                    |        No        |
+| Trig  | Desc                                                                                                                                         | Context Required |
+| :---: | -------------------------------------------------------------------------------------------------------------------------------------------- | :--------------: |
+| `fn`  | Expands to lambda function in argument list or function body, otherwise expand to normal function.                                           |        No        |
+| `pc`  | Expands to `pub(crate)`.                                                                                                                     |        No        |
+| `ps`  | Expands to `pub(super)`.                                                                                                                     |        No        |
+| `ii`  | Expands to `#[inline]`.                                                                                                                      |        No        |
+| `ia`  | Expands to `#[inline(always)]`.                                                                                                              |        No        |
+| `tfn` | Expands to a test function. `#[test]` or `#[tokio::test]` supported. With `snippet.rust.rstest_support` enabled, `#[rstest]` also supported. |        No        |
+| `pm`  | Expands to a public method definition.                                                                                                       |  In impl block   |
 
 #### Postfix Snippets
 
@@ -285,12 +287,11 @@ tmss! -> absl::flat_hash_map<std::string, std::string>
 
 #### Normal Snippets
 
-| Trig  | Desc                                                                                                                                         | Context Required |
-| :---: | -------------------------------------------------------------------------------------------------------------------------------------------- | :--------------: |
-| `fn`  | Expands to function definition.                                                                                                              |        No        |
-| `wfn` | Expands to function definition returns a widget.                                                                                             |        No        |
-| `afn` | Expands to an async function definition.                                                                                                     |        No        |
-| `tfn` | Expands to a test function. `#[test]` or `#[tokio::test]` supported. With `snippet.rust.rstest_support` enabled, `#[rstest]` also supported. |        No        |
+| Trig  | Desc                                             | Context Required |
+| :---: | ------------------------------------------------ | :--------------: |
+| `fn`  | Expands to function definition.                  |        No        |
+| `wfn` | Expands to function definition returns a widget. |        No        |
+| `afn` | Expands to an async function definition.         |        No        |
 
 #### Auto-snippets
 

lua/luasnip-snippets/snippets/rust/impl_block.lua

@@ -0,0 +1,79 @@
+local UtilsTS = require("luasnip-snippets.utils.treesitter")
+local ls = require("luasnip")
+local t = ls.text_node
+local fmta = require("luasnip.extras.fmt").fmta
+local i = require("luasnip-snippets.nodes").insert_node
+local c = require("luasnip-snippets.nodes").choice_node
+
+local function require_impl_block(_, _, match, captures)
+  local row, col = unpack(vim.api.nvim_win_get_cursor(0))
+  local buf = vim.api.nvim_get_current_buf()
+
+  return UtilsTS.invoke_after_reparse_buffer(buf, match, function(parser, _)
+    local pos = {
+      row - 1,
+      col - #match,
+    }
+    local node = parser:named_node_for_range {
+      pos[1],
+      pos[2],
+      pos[1],
+      pos[2],
+    }
+
+    local ret = {
+      trigger = match,
+      capture = captures,
+      env_override = {
+        IMPL_ITEM_START = UtilsTS.start_pos(
+          UtilsTS.find_first_parent(node, { "impl_item" })
+        ),
+        FUNCTION_ITEM_START = UtilsTS.start_pos(
+          UtilsTS.find_first_parent(node, { "function_item" })
+        ),
+        CLOSURE_EXPRESSION_START = UtilsTS.start_pos(
+          UtilsTS.find_first_parent(node, { "closure_expression" })
+        ),
+      },
+    }
+
+    vim.api.nvim_win_set_cursor(0, { row, col })
+
+    if
+      ret.env_override.IMPL_ITEM_START ~= nil
+      and ret.env_override.FUNCTION_ITEM_START == nil
+      and ret.env_override.CLOSURE_EXPRESSION_START == nil
+    then
+      return ret
+    end
+
+    return nil
+  end)
+end
+
+return {
+  ls.s(
+    {
+      trig = "pm",
+      wordTrig = true,
+      name = "(pm) pub method",
+      resolveExpandParams = require_impl_block,
+    },
+    fmta(
+      [[
+      pub fn <name>(<_self>) {
+        <body>
+      }
+      ]],
+      {
+        body = i(0),
+        name = i(1, "new_fn", { desc = "function name" }),
+        _self = c(2, {
+          t("&self"),
+          t("&mut self"),
+          t("self"),
+        }, { desc = "self" }),
+      }
+    )
+  ),
+}

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

@@ -6,6 +6,7 @@ local function setup()
     "postfix",
     "lambda_fn",
     "test_fn",
+    "impl_block",
   })
 end