2022-12-01 06:38:45 +08:00
|
|
|
local Config = require("lazy.core.config")
|
|
|
|
|
2022-11-21 06:04:56 +08:00
|
|
|
local M = {}
|
|
|
|
|
2022-12-01 06:38:45 +08:00
|
|
|
---@diagnostic disable-next-line: no-unknown
|
|
|
|
local uv = vim.loop
|
|
|
|
|
|
|
|
---@class ProcessOpts
|
|
|
|
---@field args string[]
|
|
|
|
---@field cwd? string
|
|
|
|
---@field on_line? fun(string)
|
|
|
|
---@field on_exit? fun(ok:boolean, output:string)
|
|
|
|
---@field timeout? number
|
2022-11-21 06:04:56 +08:00
|
|
|
|
2022-12-01 06:38:45 +08:00
|
|
|
---@param opts? ProcessOpts
|
2022-11-21 06:04:56 +08:00
|
|
|
function M.spawn(cmd, opts)
|
|
|
|
opts = opts or {}
|
2022-12-01 06:44:10 +08:00
|
|
|
opts.timeout = opts.timeout or (Config.options.git and Config.options.git.timeout * 1000)
|
2022-12-01 06:38:45 +08:00
|
|
|
|
2022-11-21 06:04:56 +08:00
|
|
|
local env = {
|
|
|
|
"GIT_TERMINAL_PROMPT=0",
|
|
|
|
"GIT_SSH_COMMAND=ssh -oBatchMode=yes",
|
|
|
|
}
|
|
|
|
|
|
|
|
for key, value in
|
2022-12-01 06:38:45 +08:00
|
|
|
pairs(uv.os_environ() --[[@as string[] ]])
|
2022-11-21 06:04:56 +08:00
|
|
|
do
|
|
|
|
table.insert(env, key .. "=" .. value)
|
|
|
|
end
|
|
|
|
|
2022-12-01 06:38:45 +08:00
|
|
|
local stdout = uv.new_pipe()
|
|
|
|
local stderr = uv.new_pipe()
|
2022-11-21 06:04:56 +08:00
|
|
|
|
|
|
|
local output = ""
|
|
|
|
---@type vim.loop.Process
|
|
|
|
local handle = nil
|
|
|
|
|
2022-12-01 06:38:45 +08:00
|
|
|
local timeout
|
|
|
|
local killed = false
|
|
|
|
if opts.timeout then
|
|
|
|
timeout = uv.new_timer()
|
|
|
|
timeout:start(opts.timeout, 0, function()
|
|
|
|
if handle and not handle:is_closing() then
|
|
|
|
killed = true
|
|
|
|
uv.process_kill(handle, "sigint")
|
|
|
|
end
|
|
|
|
end)
|
|
|
|
end
|
|
|
|
|
|
|
|
handle = uv.spawn(cmd, {
|
2022-11-21 06:04:56 +08:00
|
|
|
stdio = { nil, stdout, stderr },
|
|
|
|
args = opts.args,
|
|
|
|
cwd = opts.cwd,
|
|
|
|
env = env,
|
2022-12-01 06:38:45 +08:00
|
|
|
}, function(exit_code, signal)
|
|
|
|
if timeout then
|
|
|
|
timeout:stop()
|
|
|
|
timeout:close()
|
|
|
|
end
|
2022-11-21 06:04:56 +08:00
|
|
|
handle:close()
|
|
|
|
stdout:close()
|
|
|
|
stderr:close()
|
2022-12-01 06:38:45 +08:00
|
|
|
local check = uv.new_check()
|
2022-11-21 06:04:56 +08:00
|
|
|
check:start(function()
|
|
|
|
if not stdout:is_closing() or not stderr:is_closing() then
|
|
|
|
return
|
|
|
|
end
|
|
|
|
check:stop()
|
|
|
|
if opts.on_exit then
|
|
|
|
output = output:gsub("[^\r\n]+\r", "")
|
2022-12-01 06:38:45 +08:00
|
|
|
if killed then
|
|
|
|
output = output .. "\n" .. "Process was killed because it reached the timeout"
|
|
|
|
end
|
2022-11-21 06:04:56 +08:00
|
|
|
|
|
|
|
vim.schedule(function()
|
2022-12-01 06:38:45 +08:00
|
|
|
opts.on_exit(exit_code == 0 and signal == 0, output)
|
2022-11-21 06:04:56 +08:00
|
|
|
end)
|
|
|
|
end
|
|
|
|
end)
|
|
|
|
end)
|
|
|
|
|
|
|
|
if not handle then
|
|
|
|
if opts.on_exit then
|
|
|
|
opts.on_exit(false, "Failed to spawn process " .. cmd .. " " .. vim.inspect(opts))
|
|
|
|
end
|
|
|
|
return
|
|
|
|
end
|
|
|
|
|
2022-11-28 14:36:32 +08:00
|
|
|
---@param data? string
|
2022-11-21 06:04:56 +08:00
|
|
|
local function on_output(err, data)
|
|
|
|
assert(not err, err)
|
|
|
|
|
|
|
|
if data then
|
|
|
|
output = output .. data:gsub("\r\n", "\n")
|
|
|
|
local lines = vim.split(vim.trim(output:gsub("\r$", "")):gsub("[^\n\r]+\r", ""), "\n")
|
|
|
|
|
|
|
|
if opts.on_line then
|
|
|
|
vim.schedule(function()
|
|
|
|
opts.on_line(lines[#lines])
|
|
|
|
end)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2022-12-01 06:38:45 +08:00
|
|
|
uv.read_start(stdout, on_output)
|
|
|
|
uv.read_start(stderr, on_output)
|
2022-11-21 06:04:56 +08:00
|
|
|
|
|
|
|
return handle
|
|
|
|
end
|
|
|
|
|
|
|
|
return M
|