diff --git a/lua/gopher/comment.lua b/lua/gopher/comment.lua index 94f0b43..01f971b 100644 --- a/lua/gopher/comment.lua +++ b/lua/gopher/comment.lua @@ -18,25 +18,24 @@ end ---@return string ---@private local function generate(bufnr) - local sok, sres = pcall(ts.get_struct_under_cursor, bufnr) - vim.print(sok, sres) - if sok then - return template(sres.name) + local s_ok, s_res = pcall(ts.get_struct_under_cursor, bufnr) + if s_ok then + return template(s_res.name) end - local fok, fres = pcall(ts.get_func_under_cursor, bufnr) - if fok then - return template(fres.name) + local f_ok, f_res = pcall(ts.get_func_under_cursor, bufnr) + if f_ok then + return template(f_res.name) end - local iok, ires = pcall(ts.get_interface_under_cursor, bufnr) - if iok then - return template(ires.name) + local i_ok, i_res = pcall(ts.get_interface_under_cursor, bufnr) + if i_ok then + return template(i_res.name) end - local pok, pres = pcall(ts.get_package_under_cursor, bufnr) - if pok then - return "// Package " .. pres.name .. " provides " + local p_ok, p_res = pcall(ts.get_package_under_cursor, bufnr) + if p_ok then + return "// Package " .. p_res.name .. " provides " end return "// " diff --git a/spec/fixtures/comment/interface_input.go b/spec/fixtures/comment/interface_input.go index b82f837..c065125 100644 --- a/spec/fixtures/comment/interface_input.go +++ b/spec/fixtures/comment/interface_input.go @@ -1,3 +1,3 @@ package main -type Testinter interface{} +type Testinger interface{} diff --git a/spec/fixtures/comment/interface_output.go b/spec/fixtures/comment/interface_output.go index ca9e0dd..c065125 100644 --- a/spec/fixtures/comment/interface_output.go +++ b/spec/fixtures/comment/interface_output.go @@ -1,4 +1,3 @@ package main -// Testinter -type Testinter interface{} +type Testinger interface{}