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

Add comment generator (#10)

Browse source 
* feat(_utils.ts): get package

* feat(_utils.ts): get interface

* feat(_utils.ts): update type annotations

* feat(comments): first naive implementation

* docs: add comment [skip ci]
This commit is contained in:
Smirnov Oleksandr 2022-06-30 12:00:13 +03:00 • committed by GitHub
parent 40a2839eab
commit 0fcdceeb89
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 97 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

39
lua/gopher/_utils/ts/init.lua
View file

@ -4,6 +4,8 @@ local M = {
querys = {
struct_block = [[((type_declaration (type_spec name:(type_identifier) @struct.name type: (struct_type)))@struct.declaration)]],
em_struct_block = [[(field_declaration name:(field_identifier)@struct.name type: (struct_type)) @struct.declaration]],
package = [[(package_clause (package_identifier)@package.name)@package.clause]],
interface = [[((type_declaration (type_spec name:(type_identifier) @interface.name type:(interface_type)))@interface.declaration)]],
method_name = [[((method_declaration receiver: (parameter_list)@method.receiver name: (field_identifier)@method.name body:(block))@method.declaration)]],
func = [[((function_declaration name: (identifier)@function.name) @function.declaration)]],
},
@ -21,7 +23,7 @@ end
---@param row string
---@param col string
---@param bufnr string
---@param bufnr string|nil
---@return table|nil
function M.get_struct_node_at_pos(row, col, bufnr)
local query = M.querys.struct_block .. " " .. M.querys.em_struct_block
@ -36,7 +38,7 @@ end
---@param row string
---@param col string
---@param bufnr string
---@param bufnr string|nil
---@return table|nil
function M.get_func_method_node_at_pos(row, col, bufnr)
local query = M.querys.func .. " " .. M.querys.method_name
@ -49,4 +51,37 @@ function M.get_func_method_node_at_pos(row, col, bufnr)
end
end
---@param row string
---@param col string
---@param bufnr string|nil
---@return table|nil
function M.get_package_node_at_pos(row, col, bufnr)
-- stylua: ignore
if row > 10 then return end
local query = M.querys.package
local bufn = bufnr or vim.api.nvim_get_current_buf()
local ns = nodes.nodes_at_cursor(query, get_name_defaults(), bufn, row, col)
if ns == nil then
u.notify("package not found", "warn")
return nil
else
return ns[#ns]
end
end
---@param row string
---@param col string
---@param bufnr string|nil
---@return table|nil
function M.get_interface_node_at_pos(row, col, bufnr)
local query = M.querys.interface
local bufn = bufnr or vim.api.nvim_get_current_buf()
local ns = nodes.nodes_at_cursor(query, get_name_defaults(), bufn, row, col)
if ns == nil then
u.notify("interface not found", "warn")
else
return ns[#ns]
end
end
return M