olexsmir/gopher.nvim
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat(comment): add comment on a variable

Browse source
This commit is contained in:
Oleksandr Smirnov 2025-09-04 15:52:48 +03:00
parent 9bffa2c212
commit 094afb2c21
No known key found for this signature in database
7 changed files with 45 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
lua/gopher/_utils/ts.lua
View file

@ -27,6 +27,11 @@ local queries = {
name: (type_identifier) @_name
type: (interface_type))
]],
var = [[
[(var_declaration (var_spec name: (identifier) @_name))
(short_var_declaration
left: (expression_list (identifier) @_name @_var))]
]],
}
---@param parent_type string[]
@ -147,4 +152,9 @@ function ts.get_interface_under_cursor(bufnr)
return do_stuff(bufnr, { "type_declaration" }, queries.interface)
end
---@param bufnr integer
function ts.get_variable_under_cursor(bufnr)
return do_stuff(bufnr, { "var_declaration", "short_var_declaration" }, queries.var)
end
return ts

5
lua/gopher/comment.lua
View file

@ -25,6 +25,11 @@ local function generate(bufnr, line)
return u.indent(line, s_res.indent) .. "// " .. s_res.name .. " "
end
local v_ok, v_res = pcall(ts.get_variable_under_cursor, bufnr)
if v_ok then
return u.indent(line, v_res.indent) .. "// " .. v_res.name .. " "
end
local f_ok, f_res = pcall(ts.get_func_under_cursor, bufnr)
if f_ok then
return u.indent(line, f_res.indent) .. "// " .. f_res.name .. " "