1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
local ls = require("luasnip")
local s = ls.snippet
local i = ls.insert_node
local fmt = require("luasnip.extras.fmt").fmt
local rep = require("luasnip.extras").rep
local f = ls.function_node -- Import function_node for dynamic content
-- Function to get the filename without the path
local function get_filename()
local filepath = vim.api.nvim_buf_get_name(0)
local filename = vim.fn.fnamemodify(filepath, ":t:r") -- Get filename without path and extension
return filename:gsub("_", " ") -- Replace underscores with spaces
end
local neetcode_snippet = s(
"neetcode",
fmt(
[[
"""
Question
{1}
"""
from typing import Dict
class Solution:
"""
A class to solve {file_name} from {2}
"""
def {3}(self, {4}) -> {5}:
"""
{file_name} from the input {parameters_type} using a {func_name} approach.
Args:
{parameters_type}: {6}
Returns:
{return_type}: {7}
"""
return {8}
def print(self, examples: Dict[str, {parameters_type}]) -> None:
for name, example in examples.items():
result = self.{3}(example)
print(f"{{name}}: {{result}}")
def main(self):
cases = {{
"case1" : {9},
"case2" : {10},
"case3" : {11},
}}
self.print(cases)
solution = Solution()
solution.main()
"""
Solution
url: {12}
video: {13}
1. {func_name}
time: {14}
space: {15}
code:
```python
{16}
```
"""
]],
{
i(1, '"Describe the question here"'),
file_name = f(get_filename), -- Insert the filename dynamicall
i(2, '"Describe the class here"'),
i(3, '"Function name"'), -- Primary insert node for method name
func_name = rep(3), -- Repeat the method name
i(4, '"parameters"'),
parameters_type = rep(4),
i(5, '"return_type"'),
return_type = rep(5),
i(6, '"parameters_desc"'),
i(7, '"return_type_desc"'),
i(8, '"return"'),
i(9, '"case1"'),
i(10, '"case2"'),
i(11, '"case3"'),
i(12, '"url"'),
i(13, '"video"'),
i(14, '"time_complexity"'),
i(15, '"space_complexity"'),
i(16, '"code"'),
}
)
)
-- Add the snippets for multiple filetypes
ls.add_snippets("python", { neetcode_snippet })
|
