34 lines
927 B
Lua
34 lines
927 B
Lua
local nodes = require "gopher._utils.ts.nodes"
|
|
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]],
|
|
},
|
|
}
|
|
|
|
---@return table
|
|
local function get_name_defaults()
|
|
return {
|
|
["func"] = "function",
|
|
["if"] = "if",
|
|
["else"] = "else",
|
|
["for"] = "for",
|
|
}
|
|
end
|
|
|
|
---@param row any
|
|
---@param col any
|
|
---@param bufnr any
|
|
---@return table|nil
|
|
function M.get_struct_node_at_pos(row, col, bufnr)
|
|
local query = M.querys.struct_block .. " " .. M.querys.em_struct_block
|
|
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
|
|
print "struct not found"
|
|
else
|
|
return ns[#ns]
|
|
end
|
|
end
|
|
|
|
return M
|