feat(comment): add comment on a struct field
This commit is contained in:
parent
3cd45c45a3
commit
9bffa2c212
9 changed files with 100 additions and 7 deletions
|
|
@ -11,6 +11,9 @@ local queries = {
|
|||
right: (expression_list (composite_literal
|
||||
type: (struct_type))))]
|
||||
]],
|
||||
struct_field = [[
|
||||
(field_declaration name: (field_identifier) @_name)
|
||||
]],
|
||||
func = [[
|
||||
[(function_declaration name: (identifier) @_name)
|
||||
(method_declaration name: (field_identifier) @_name)
|
||||
|
|
@ -80,16 +83,14 @@ local function do_stuff(bufnr, parent_type, query)
|
|||
error "No treesitter parser found for go"
|
||||
end
|
||||
|
||||
local node = vim.treesitter.get_node {
|
||||
bufnr = bufnr,
|
||||
}
|
||||
local node = vim.treesitter.get_node { bufnr = bufnr }
|
||||
if not node then
|
||||
error "No nodes found under cursor"
|
||||
error "No nodes found under the cursor"
|
||||
end
|
||||
|
||||
local parent_node = get_parrent_node(parent_type, node)
|
||||
if not parent_node then
|
||||
error "No parent node found under cursor"
|
||||
error "No parent node found under the cursor"
|
||||
end
|
||||
|
||||
local q = vim.treesitter.query.parse("go", query)
|
||||
|
|
@ -107,11 +108,12 @@ end
|
|||
---@param bufnr integer
|
||||
function ts.get_struct_under_cursor(bufnr)
|
||||
--- should be both type_spec and type_declaration
|
||||
--- because in cases like `type ( T struct{}, U strict{} )`
|
||||
--- i will be choosing always last struct in the list
|
||||
--- because in cases like `type ( T struct{}, U struct{} )`
|
||||
---
|
||||
--- var_declaration is for cases like `var x struct{}`
|
||||
--- short_var_declaration is for cases like `x := struct{}{}`
|
||||
---
|
||||
--- it always chooses last struct type in the list
|
||||
return do_stuff(bufnr, {
|
||||
"type_spec",
|
||||
"type_declaration",
|
||||
|
|
@ -120,6 +122,11 @@ function ts.get_struct_under_cursor(bufnr)
|
|||
}, queries.struct)
|
||||
end
|
||||
|
||||
---@param bufnr integer
|
||||
function ts.get_struct_field_under_cursor(bufnr)
|
||||
return do_stuff(bufnr, { "field_declaration" }, queries.struct_field)
|
||||
end
|
||||
|
||||
---@param bufnr integer
|
||||
function ts.get_func_under_cursor(bufnr)
|
||||
--- since this handles both and funcs and methods we should check for both parent nodes
|
||||
|
|
|
|||
|
|
@ -15,6 +15,11 @@ local comment = {}
|
|||
---@return string
|
||||
---@dochide
|
||||
local function generate(bufnr, line)
|
||||
local sf_ok, sf_res = pcall(ts.get_struct_field_under_cursor, bufnr)
|
||||
if sf_ok then
|
||||
return u.indent(line, sf_res.indent) .. "// " .. sf_res.name .. " "
|
||||
end
|
||||
|
||||
local s_ok, s_res = pcall(ts.get_struct_under_cursor, bufnr)
|
||||
if s_ok then
|
||||
return u.indent(line, s_res.indent) .. "// " .. s_res.name .. " "
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue