e9f2eef5e7
* refactor(ts_utils): i dont know why event it was here * fix: typos * fix(struct_tags)!: remove statement that i used for debug * refactor(ts_util): start from scratch * refactor(struct_tags): use new ts_util * fixup! refactor(struct_tags): use new ts_util * test(struct_tags): add support for multiple structs * fix(gotests): use new api * fix(impl): refactor some logic, use new api * docs(ts): add an explanation * refactor(_utils.ts): all public methods are just adapters * fix(comment): now it works * fixup! refactor(_utils.ts): all public methods are just adapters * fixup! fixup! refactor(_utils.ts): all public methods are just adapters * test(comment): e2e * tests(comment): fix * refactor(utils.ts): fix, docs * test(comment): fix tests again * fix(tests/comments): well, now i fell stupid * refactor(ts): add assert just to be sure that all elements are in the result * fix(ts): type annotations * fix(ts): pass bufnr to vim.treesitter.get_node * chore(ci): disable nightly * chore(ci): reorganize
55 lines
1.3 KiB
Lua
55 lines
1.3 KiB
Lua
---@toc_entry Generate comments
|
|
---@tag gopher.nvim-comments
|
|
---@usage Execute `:GoCmt` to generate a comment for the current function/method/struct/etc on this line.
|
|
---@text This module provides a way to generate comments for Go code.
|
|
|
|
local ts = require "gopher._utils.ts"
|
|
local log = require "gopher._utils.log"
|
|
local comment = {}
|
|
|
|
---@param name string
|
|
---@return string
|
|
---@private
|
|
local function template(name)
|
|
return "// " .. name .. " "
|
|
end
|
|
|
|
---@param bufnr integer
|
|
---@return string
|
|
---@private
|
|
local function generate(bufnr)
|
|
local s_ok, s_res = pcall(ts.get_struct_under_cursor, bufnr)
|
|
if s_ok then
|
|
return template(s_res.name)
|
|
end
|
|
|
|
local f_ok, f_res = pcall(ts.get_func_under_cursor, bufnr)
|
|
if f_ok then
|
|
return template(f_res.name)
|
|
end
|
|
|
|
local i_ok, i_res = pcall(ts.get_interface_under_cursor, bufnr)
|
|
if i_ok then
|
|
return template(i_res.name)
|
|
end
|
|
|
|
local p_ok, p_res = pcall(ts.get_package_under_cursor, bufnr)
|
|
if p_ok then
|
|
return "// Package " .. p_res.name .. " provides "
|
|
end
|
|
|
|
return "// "
|
|
end
|
|
|
|
function comment.comment()
|
|
local bufnr = vim.api.nvim_get_current_buf()
|
|
local cmt = generate(bufnr)
|
|
log.debug("generated comment: " .. cmt)
|
|
|
|
local pos = vim.fn.getcurpos()[2]
|
|
vim.fn.append(pos - 1, cmt)
|
|
vim.fn.setpos(".", { 0, pos, #cmt })
|
|
vim.cmd "startinsert!"
|
|
end
|
|
|
|
return comment
|