136 lines
3.6 KiB
Lua
136 lines
3.6 KiB
Lua
require("plenary.async").tests.add_to_env()
|
|
local cache = require("oil.cache")
|
|
local test_adapter = require("oil.adapters.test")
|
|
local util = require("oil.util")
|
|
local M = {}
|
|
|
|
M.reset_editor = function()
|
|
require("oil").setup({
|
|
columms = {},
|
|
adapters = {
|
|
["oil-test://"] = "test",
|
|
},
|
|
prompt_save_on_select_new_entry = false,
|
|
})
|
|
vim.cmd.tabonly({ mods = { silent = true } })
|
|
for i, winid in ipairs(vim.api.nvim_tabpage_list_wins(0)) do
|
|
if i > 1 then
|
|
vim.api.nvim_win_close(winid, true)
|
|
end
|
|
end
|
|
vim.api.nvim_win_set_buf(0, vim.api.nvim_create_buf(false, true))
|
|
for _, bufnr in ipairs(vim.api.nvim_list_bufs()) do
|
|
vim.api.nvim_buf_delete(bufnr, { force = true })
|
|
end
|
|
cache.clear_everything()
|
|
test_adapter.test_clear()
|
|
end
|
|
|
|
local function throwiferr(err, ...)
|
|
if err then
|
|
error(err)
|
|
else
|
|
return ...
|
|
end
|
|
end
|
|
|
|
M.await = function(fn, nargs, ...)
|
|
return throwiferr(a.wrap(fn, nargs)(...))
|
|
end
|
|
|
|
M.wait_for_autocmd = a.wrap(function(autocmd, cb)
|
|
local opts = {
|
|
pattern = "*",
|
|
nested = true,
|
|
once = true,
|
|
}
|
|
if type(autocmd) == "table" then
|
|
opts = vim.tbl_extend("force", opts, autocmd)
|
|
autocmd = autocmd[1]
|
|
opts[1] = nil
|
|
end
|
|
opts.callback = vim.schedule_wrap(cb)
|
|
|
|
vim.api.nvim_create_autocmd(autocmd, opts)
|
|
end, 2)
|
|
|
|
M.wait_oil_ready = a.wrap(function(cb)
|
|
util.run_after_load(0, vim.schedule_wrap(cb))
|
|
end, 1)
|
|
|
|
---@param actions string[]
|
|
---@param timestep integer
|
|
M.feedkeys = function(actions, timestep)
|
|
timestep = timestep or 10
|
|
a.util.sleep(timestep)
|
|
for _, action in ipairs(actions) do
|
|
a.util.sleep(timestep)
|
|
local escaped = vim.api.nvim_replace_termcodes(action, true, false, true)
|
|
vim.api.nvim_feedkeys(escaped, "m", true)
|
|
end
|
|
a.util.sleep(timestep)
|
|
-- process pending keys until the queue is empty.
|
|
-- Note that this will exit insert mode.
|
|
vim.api.nvim_feedkeys("", "x", true)
|
|
a.util.sleep(timestep)
|
|
end
|
|
|
|
M.actions = {
|
|
---Open oil and wait for it to finish rendering
|
|
---@param args string[]
|
|
open = function(args)
|
|
vim.schedule(function()
|
|
vim.cmd.Oil({ args = args })
|
|
-- If this buffer was already open, manually dispatch the autocmd to finish the wait
|
|
if vim.b.oil_ready then
|
|
vim.api.nvim_exec_autocmds("User", {
|
|
pattern = "OilEnter",
|
|
modeline = false,
|
|
data = { buf = vim.api.nvim_get_current_buf() },
|
|
})
|
|
end
|
|
end)
|
|
M.wait_for_autocmd({ "User", pattern = "OilEnter" })
|
|
end,
|
|
|
|
---Save all changes and wait for operation to complete
|
|
save = function()
|
|
vim.schedule_wrap(require("oil").save)({ confirm = false })
|
|
M.wait_for_autocmd({ "User", pattern = "OilMutationComplete" })
|
|
end,
|
|
|
|
---@param bufnr? integer
|
|
reload = function(bufnr)
|
|
M.await(require("oil.view").render_buffer_async, 3, bufnr or 0)
|
|
end,
|
|
|
|
---Move cursor to a file or directory in an oil buffer
|
|
---@param filename string
|
|
focus = function(filename)
|
|
local lines = vim.api.nvim_buf_get_lines(0, 0, -1, true)
|
|
local search = " " .. filename .. "$"
|
|
for i, line in ipairs(lines) do
|
|
if line:match(search) then
|
|
vim.api.nvim_win_set_cursor(0, { i, 0 })
|
|
return
|
|
end
|
|
end
|
|
error("Could not find file " .. filename)
|
|
end,
|
|
}
|
|
|
|
---Get the raw list of filenames from an unmodified oil buffer
|
|
---@param bufnr? integer
|
|
---@return string[]
|
|
M.parse_entries = function(bufnr)
|
|
bufnr = bufnr or 0
|
|
if vim.bo[bufnr].modified then
|
|
error("parse_entries doesn't work on a modified oil buffer")
|
|
end
|
|
local lines = vim.api.nvim_buf_get_lines(bufnr, 0, -1, true)
|
|
return vim.tbl_map(function(line)
|
|
return line:match("^/%d+ +(.+)$")
|
|
end, lines)
|
|
end
|
|
|
|
return M
|