go.nvim/lua/go/env.lua
ray-x 981adf9b39 Features:
1) count the overall test coverage per-file and per-project
2) add gotestsum as a test runner option
3) handle windows GOPATH and
4) minor fix: stop GoFmt if no formatter is available
2022-07-01 19:15:31 +10:00

82 lines
1.8 KiB
Lua

-- env fileread
local util = require("go.utils")
local log = util.log
local M = {}
local vfn = vim.fn
local sep = require("go.utils").sep()
function M.envfile(f)
local workfolder = vim.lsp.buf.list_workspace_folders()[1] or vfn.getcwd()
local goenv = workfolder .. sep .. (f or ".env")
if vfn.filereadable(goenv) == 1 then
return goenv
end
end
function M.append(env, val)
local oldval = vfn.getenv(env)
if val == vim.NIL or string.find(oldval, val) then
return
end
if oldval == vim.NIL then
util.notify("failed to get env var: " .. env)
end
if oldval:find(val) then -- presented
return
end
local newval = oldval .. ":" .. val
vfn.setenv(env, newval)
end
function M.load_env(env, setToEnv)
setToEnv = setToEnv or true
env = env or M.envfile()
if vfn.filereadable(env) == 0 then
return false
end
local lines = util.lines_from(env)
local envs = {}
for _, line in ipairs(lines) do
for k, v in string.gmatch(line, "([%w_]+)=([%w%c%p%z]+)") do
envs[k] = v
end
end
log(envs)
if setToEnv then
for key, val in pairs(envs) do
vfn.setenv(key, val)
end
end
return envs
end
-- best effort to enabl $GOBIN
function M.setup()
local home = "HOME"
if util.is_windows() then
home = "USERPROFILE"
end
local gohome = vfn.getenv("GOHOME")
local gobin = vfn.getenv("GOBIN")
local user_home = vfn.getenv(home)
if gobin == vim.NIL then
if gohome == vim.NIL then
if user_home == vim.NIL then
util.notify("failed to setup $GOBIN")
return
end
gobin = user_home .. sep .. "go" .. sep .. "bin"
else
local gohome1 = vim.split(gohome, ":")[1]
gobin = gohome1 .. require("go.utils").sep() .. "bin"
vfn.setenv("GOBIN", gobin)
end
end
M.append("PATH", gobin)
end
return M