mirror of https://github.com/ray-x/go.nvim
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
1.8 KiB
Lua
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
|