56 lines
1.6 KiB
Lua
Executable File
56 lines
1.6 KiB
Lua
Executable File
-- taken from https://github.com/ibhagwan/nvim-lua
|
|
|
|
local M = {}
|
|
|
|
M.opts = function(desc)
|
|
if desc then
|
|
return { silent = true, desc = desc}
|
|
else
|
|
return { silent = true }
|
|
end
|
|
end
|
|
|
|
M.sudo_exec = function(cmd, print_output)
|
|
vim.fn.inputsave()
|
|
local password = vim.fn.inputsecret("Password: ")
|
|
vim.fn.inputrestore()
|
|
if not password or #password == 0 then
|
|
print("Invalid password, sudo aborted")
|
|
return false
|
|
end
|
|
local out = vim.fn.system(string.format("sudo -p '' -S %s", cmd), password)
|
|
if vim.v.shell_error ~= 0 then
|
|
print("\r\n")
|
|
print(out)
|
|
return false
|
|
end
|
|
if print_output then print("\r\n", out) end
|
|
return true
|
|
end
|
|
|
|
M.sudo_write = function(tmpfile, filepath)
|
|
if not tmpfile then tmpfile = vim.fn.tempname() end
|
|
if not filepath then filepath = vim.fn.expand("%") end
|
|
if not filepath or #filepath == 0 then
|
|
print("E32: No file name")
|
|
return
|
|
end
|
|
-- `bs=1048576` is equivalent to `bs=1M` for GNU dd or `bs=1m` for BSD dd
|
|
-- Both `bs=1M` and `bs=1m` are non-POSIX
|
|
local cmd = string.format("dd if=%s of=%s bs=1048576",
|
|
vim.fn.shellescape(tmpfile),
|
|
vim.fn.shellescape(filepath))
|
|
-- no need to check error as this fails the entire function
|
|
vim.api.nvim_exec(string.format("write! %s", tmpfile), true)
|
|
if M.sudo_exec(cmd) then
|
|
-- refreshes the buffer and prints the "written" message
|
|
vim.cmd.checktime()
|
|
-- exit command mode
|
|
vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes(
|
|
"<Esc>", true, false, true), "n", true)
|
|
end
|
|
vim.fn.delete(tmpfile)
|
|
end
|
|
|
|
return M
|