diff --git a/shell-integration/bash/kitty.bash b/shell-integration/bash/kitty.bash index 22d1a998a..ff3edc886 100644 --- a/shell-integration/bash/kitty.bash +++ b/shell-integration/bash/kitty.bash @@ -308,6 +308,23 @@ builtin unset -f _ksi_main case :$SHELLOPTS: in *:posix:*) ;; *) + +_ksi_transmit_data() { + builtin local data + data="${1//[[:space:]]}" + builtin local pos=0 + builtin local chunk_num=0 + while [ $pos -lt ${#data} ]; do + builtin local chunk="${data:$pos:2048}" + pos=$(($pos+2048)) + builtin printf '\eP@kitty-%s|%s:%s\e\\' "${2}" "${chunk_num}" "${chunk}" + chunk_num=$(($chunk_num+1)) + done + # save history so it is available in new shell + [ "$3" = "save_history" ] && builtin history -a + builtin printf '\eP@kitty-%s|\e\\' "${2}" +} + clone-in-kitty() { builtin local data="shell=bash,pid=$$,cwd=$(builtin printf "%s" "$PWD" | builtin command base64),envfmt=bash,env=$(builtin export | builtin command base64)" while :; do @@ -321,18 +338,84 @@ clone-in-kitty() { esac shift done - data="${data//[[:space:]]}" - builtin local pos=0 - builtin local chunk_num=0 - while [ $pos -lt ${#data} ]; do - builtin local chunk="${data:$pos:2048}" - pos=$(($pos+2048)) - builtin printf '\eP@kitty-clone|%s:%s\e\\' "${chunk_num}" "${chunk}" - chunk_num=$(($chunk_num+1)) + _ksi_transmit_data "$data" "clone" "save_history" +} + +edit-in-kitty() { + builtin local data="" + builtin local ed_filename="" + builtin local usage="Usage: edit-in-kitty [OPTIONS] FILE" + data="cwd=$(builtin printf "%s" "$PWD" | builtin command base64)" + while :; do + case "$1" in + "") break;; + -h|--help) + builtin printf "%s\n\n%s\n\n%s\n" "$usage" "Edit the specified file in a kitty overlay window. Works over SSH as well." "For usage instructions see: https://sw.kovidgoyal.net/kitty/shell-integration/#edit-file" + return + ;; + *) data="$data,a=$(builtin printf "%s" "$1" | builtin command base64)"; ed_filename="$1";; + esac + shift done - # append current commands to history file - builtin history -a - builtin printf '\eP@kitty-clone|\e\\' + [ -z "$ed_filename" ] && { + builtin echo "$usage" > /dev/stderr + return 1 + } + [ -r "$ed_filename" -a -w "$ed_filename" ] || { + builtin echo "$ed_filename is not readable and writable" > /dev/stderr + return 1 + } + [ ! -f "$ed_filename" ] && { + builtin echo "$ed_filename is not a file" > /dev/stderr + return 1 + } + builtin local stat_result="" + stat_result=$(builtin command stat -L --format '%d:%i:%s' "$ed_filename" 2> /dev/null) + [ $? != 0 ] && stat_result=$(builtin command stat -L -f '%d:%i:%z' "$ed_filename" 2> /dev/null) + [ -z "$stat_result" ] && { builtin echo "Failed to stat the file: $ed_filename" > /dev/stderr; return 1; } + data="$data,file_inode=$stat_result" + builtin local file_size=$(builtin echo "$stat_result" | builtin command cut -d: -f3) + [ "$file_size" -gt $((8 * 1024 * 1024)) ] && { builtin echo "File is too large for performant editing"; return 1; } + data="$data,file_data=$(builtin command cat "$ed_filename" | builtin command base64)" + _ksi_transmit_data "$data" "edit" + data="" + builtin echo "Waiting for editing to be completed..." + _ksi_wait_for_complete() { + builtin local started="n" + builtin local line="" + builtin local old_tty_settings=$(builtin command stty -g) + builtin command stty "-echo" + builtin trap -- "builtin command stty '$old_tty_settings'" RETURN + builtin trap -- "builtin command stty '$old_tty_settings'; _ksi_transmit_data 'abort_signaled=interrupt' 'edit'; builtin exit 1;" SIGINT SIGTERM + while :; do + started="n" + while IFS= read -r line; do + if [ "$started" = "y" ]; then + [ "$line" = "UPDATE" ] && break; + [ "$line" = "DONE" ] && { started="done"; break; } + builtin printf "%s\n" "$line" > /dev/stderr; + return 1; + else + [ "$line" = "KITTY_DATA_START" ] && started="y" + fi + done + [ "$started" = "n" ] && continue; + data="" + while IFS= read -r line; do + [ "$line" = "KITTY_DATA_END" ] && break; + data="$data$line" + done + [ -n "$data" -a "$started" != "done" ] && { + builtin echo "Updating $ed_filename..." + builtin printf "%s" "$data" | builtin command base64 -d > "$ed_filename" + } + [ "$started" = "done" ] && break; + done + } + $(_ksi_wait_for_complete > /dev/tty) + builtin local rc=$? + builtin unset -f _ksi_wait_for_complete + return $rc } ;; esac