From 0bbe52ed8178cef8f9897720b92ba451b8f26618 Mon Sep 17 00:00:00 2001
From: Oleksandr Smirnov <olexsmir@gmail.com>
Date: Tue, 18 Mar 2025 23:07:30 +0200
Subject: [PATCH] refactor(_utils.ts): all public methods are just adapters

---
 lua/gopher/_utils/{ts/init.lua => ts.lua} | 79 +++++++++++------------
 1 file changed, 37 insertions(+), 42 deletions(-)
 rename lua/gopher/_utils/{ts/init.lua => ts.lua} (51%)

diff --git a/lua/gopher/_utils/ts/init.lua b/lua/gopher/_utils/ts.lua
similarity index 51%
rename from lua/gopher/_utils/ts/init.lua
rename to lua/gopher/_utils/ts.lua
index 02c0d6a..b9f5d6f 100644
--- a/lua/gopher/_utils/ts/init.lua
+++ b/lua/gopher/_utils/ts.lua
@@ -1,19 +1,18 @@
-local ts = {
-  queries = {
-    struct = [[
-      (type_spec name: (type_identifier) @_name
-                 type: (struct_type))
-    ]],
-    func = [[
-      [(function_declaration name: (identifier)       @_name)
-       (method_declaration   name: (field_identifier) @_name)]
-    ]],
+local ts = {}
+local queries = {
+  struct = [[
+    (type_spec name: (type_identifier) @_name
+               type: (struct_type))
+  ]],
+  func = [[
+    [(function_declaration name: (identifier)       @_name)
+     (method_declaration   name: (field_identifier) @_name)]
+  ]],
 
-    -- 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)]],
-  },
+  -- 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)]],
 }
 
 ---@param parent_type string|[string]
@@ -45,7 +44,7 @@ end
 ---@param node TSNode
 ---@param bufnr integer
 ---@return {name:string}
-local function query_and_get_captures(query, node, bufnr)
+local function get_captures(query, node, bufnr)
   local res = {}
   for _, match, _ in query:iter_matches(node, bufnr) do
     for capture_id, captured_node in pairs(match) do
@@ -59,24 +58,28 @@ local function query_and_get_captures(query, node, bufnr)
   return res
 end
 
+---@class gopher.TsResult
+---@field name string
+---@field start_line integer
+---@field end_line integer
+
 ---@param bufnr integer
----@return table
-function ts.get_struct_under_cursor(bufnr)
+---@param parent_type string|[string]
+---@param query string
+---@return gopher.TsResult
+local function do_stuff(bufnr, parent_type, query)
   local node = vim.treesitter.get_node()
   if not node then
     error "No nodes found under cursor"
   end
 
-  --- 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
-  local parent_node = get_parrent_node({ "type_spec", "type_declaration" }, node)
+  local parent_node = get_parrent_node(parent_type, node)
   if not parent_node then
     error "No struct found under cursor"
   end
 
-  local query = vim.treesitter.query.parse("go", ts.queries.struct)
-  local res = query_and_get_captures(query, parent_node, bufnr)
+  local q = vim.treesitter.query.parse("go", query)
+  local res = get_captures(q, parent_node, bufnr)
 
   local start_row, _, end_row, _ = parent_node:range()
   res["start_line"] = start_row + 1
@@ -85,26 +88,18 @@ function ts.get_struct_under_cursor(bufnr)
   return res
 end
 
+---@param bufnr integer
+---@return table
+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
+  return do_stuff(bufnr, { "type_spec", "type_declaration" }, queries.struct)
+end
+
 ---@param bufnr integer
 function ts.get_func_under_cursor(bufnr)
-  local node = vim.treesitter.get_node()
-  if not node then
-    error "No nodes found under cursor"
-  end
-
-  local parent_node = get_parrent_node({ "function_declaration", "method_declaration" }, node)
-  if not parent_node then
-    error "No struct found under cursor"
-  end
-
-  local query = vim.treesitter.query.parse("go", ts.queries.func)
-  local res = query_and_get_captures(query, parent_node, bufnr)
-
-  local start_row, _, end_row, _ = parent_node:range()
-  res["start_line"] = start_row + 1
-  res["end_line"] = end_row + 1
-
-  return res
+  return do_stuff(bufnr, { "function_declaration", "method_declaration" }, queries.func)
 end
 
 return ts