Merge 91932596413fe4f791c65c36d17cd4786b92fd87 into ec0d6d26ceaadd93d6824630ba587b31e442214d

This commit is contained in:
jtrv 2025-01-01 14:58:14 +01:00 committed by GitHub
commit acb53795f4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,12 +1,15 @@
# save the current buffer to its file as root using `sudo`
# (optionally pass the user password to sudo if not cached)
# prompt and pass the user password to sudo if not cached
define-command -hidden sudo-write-cached-password %{
# toggle fprint mode
declare-option bool fprint_mode false
define-command -hidden sudo-write-cached %{
# easy case: the password was already cached, so we don't need any tricky handling
eval -save-regs f %{
reg f %sh{ mktemp -t XXXXXX }
evaluate-commands -save-regs f %{
set-register f %sh{ mktemp -t XXXXXX }
write! %reg{f}
eval %sh{
evaluate-commands %sh{
sudo -n -- dd if="$kak_main_reg_f" of="$kak_buffile" >/dev/null 2>&1
if [ $? -eq 0 ]; then
echo "edit!"
@ -18,12 +21,12 @@ define-command -hidden sudo-write-cached-password %{
}
}
define-command -hidden sudo-write-prompt-password %{
define-command -hidden sudo-write-prompt %{
prompt -password 'Password:' %{
eval -save-regs r %{
eval -draft -save-regs 'tf|"' %{
reg t %val{buffile}
reg f %sh{ mktemp -t XXXXXX }
evaluate-commands -save-regs r %{
evaluate-commands -draft -save-regs 'tf|"' %{
set-register t %val{buffile}
set-register f %sh{ mktemp -t XXXXXX }
write! %reg{f}
# write the password in a buffer in order to pass it through STDIN to sudo
@ -31,9 +34,9 @@ define-command -hidden sudo-write-prompt-password %{
# through the shell scope's environment or interpolating it inside the shell string
# 'exec |' is pretty much the only way to pass data over STDIN
edit -scratch '*sudo-password-tmp*'
reg '"' "%val{text}"
exec <a-P>
reg | %{
set-register '"' "%val{text}"
execute-keys <a-P>
set-register | %{
sudo -S -- dd if="$kak_main_reg_f" of="$kak_main_reg_t" > /dev/null 2>&1
if [ $? -eq 0 ]; then
printf 'edit!'
@ -42,30 +45,68 @@ define-command -hidden sudo-write-prompt-password %{
fi
rm -f "$kak_main_reg_f"
}
exec '|<ret>'
exec -save-regs '' '%"ry'
execute-keys '|<ret>'
execute-keys -save-regs '' '%"ry'
delete-buffer! '*sudo-password-tmp*'
}
eval %reg{r}
evaluate-commands %reg{r}
}
}
}
define-command -hidden sudo-write-fprint %{
evaluate-commands -save-regs f %{
set-register f %sh{ mktemp -t XXXXXX }
write! %reg{f}
evaluate-commands %sh{
# if fprint is enabled we use -S to bypass the password prompt
sudo -S -- dd if="$kak_main_reg_f" of="$kak_buffile" >/dev/null 2>&1
if [ $? -eq 0 ]; then
echo "edit!"
else
echo 'fail "Unknown failure"'
fi
rm -f "$kak_main_reg_f"
}
}
}
define-command sudo-write -docstring "Write the content of the buffer using sudo" %{
eval %sh{
evaluate-commands %sh{
# tricky posix-way of getting the first character of a variable
# no subprocess!
if [ "${kak_buffile%"${kak_buffile#?}"}" != "/" ]; then
# not entirely foolproof as a scratch buffer may start with '/', but good enough
printf 'fail "Not a file"'
echo 'fail "Not a file"'
exit
fi
# check if fprint is enabled
elif "$kak_opt_fprint_mode"; then
echo sudo-write-fprint
# check if the password is cached
if sudo -n true > /dev/null 2>&1; then
printf sudo-write-cached-password
elif sudo -n true > /dev/null 2>&1; then
echo sudo-write-cached
else
printf sudo-write-prompt-password
echo sudo-write-prompt
fi
}
}
# SAME AS ABOVE WITH 'QUIT' APPENEDED
define-command sudo-write-quit -docstring "Write the content of the buffer using sudo, then quit" %{
evaluate-commands %sh{
if [ "${kak_buffile%"${kak_buffile#?}"}" != "/" ]; then
echo 'fail "Not a file"'
exit
elif "$kak_opt_fprint_mode"; then
echo "sudo-write-fprint"
elif sudo -n true > /dev/null 2>&1; then
echo "sudo-write-cached"
else
echo "sudo-write-prompt"
fi
echo "quit"
}
}