go.nvim/lua/go/alternate.lua

37 lines
911 B
Lua
Raw Normal View History

local M = {}
function M.alternate()
local file = vim.fn.expand('%')
local alt_file = ""
if #file <= 1 then
vim.notify("no buffer name", vim.lsp.log_levels.ERROR)
return
end
2022-06-01 11:29:13 +00:00
local s = string.find(file, "_test%.go$")
local s2 = string.find(file, "%.go$")
if s ~= nil then
alt_file = string.gsub(file, "_test.go", ".go")
elseif s2 ~= nil then
alt_file = vim.fn.expand('%:r') .. "_test.go"
else
vim.notify('not a go file', vim.lsp.log_levels.ERROR)
end
return alt_file
end
function M.switch(bang, cmd)
local alt_file = M.alternate()
if not vim.fn.filereadable(alt_file) and not vim.fn.bufexists(alt_file) and not bang then
vim.notify("couldn't find " .. alt_file, vim.lsp.log_levels.ERROR)
return
elseif #cmd <= 1 then
local ocmd = "e " .. alt_file
vim.cmd(ocmd)
else
local ocmd = cmd .. " " .. alt_file
vim.cmd(ocmd)
end
end
return M