Work on enabling shell integration over ssh

shell-integration/ssh/bootstrap.sh

@@ -83,3 +83,153 @@ fi
 
 # If a command was passed to SSH execute it here
 EXEC_CMD
+
+shell_integration_dir="$HOME/SHELL_INTEGRATION_DIR"
+
+login_shell_is_ok() {
+    if [ -z "$login_shell" ] || [ ! -x "$login_shell" ]; then return 1; fi
+    case "$login_shell" in
+        *sh) return 0;
+    esac
+    return 1;
+}
+
+detect_python() {
+    python=$(command -v python3)
+    if [ -z "$python" ]; then python=$(command -v python2); fi
+    if [ -z "$python" ]; then python=python; fi
+    if [ -z "$python" || ! -x "$python" ]; then return 1; fi
+    return 0;
+}
+
+using_getent() {
+    cmd=$(command -v getent)
+    if [ -n "$cmd" ]; then 
+        output=$($cmd passwd $USER 2>/dev/null)
+        if [ $? = 0 ]; then 
+            login_shell=$(echo $output | cut -d: -f7);
+            if login_shell_is_ok; then return 0; fi
+        fi
+    fi
+    return 1;
+}
+
+using_id() {
+    cmd=$(command -v id)
+    if [ -n "$cmd" ]; then 
+        output=$($cmd -P $USER 2>/dev/null)
+        if [ $? = 0 ]; then 
+            login_shell=$(echo $output | cut -d: -f7);
+            if login_shell_is_ok; then return 0; fi
+        fi
+    fi
+    return 1;
+}
+
+using_passwd() {
+    cmd=$(command -v grep)
+    if [ -n "$cmd" ]; then 
+        output=$($cmd "^$USER:" /etc/passwd 2>/dev/null)
+        if [ $? = 0 ]; then 
+            login_shell=$(echo $output | cut -d: -f7);
+            if login_shell_is_ok; then return 0; fi
+        fi
+    fi
+    return 1;
+}
+
+using_python() {
+    if detect_python; then
+        output=$($python -c "import pwd, os; print(pwd.getpwuid(os.geteuid()).pw_shell)")
+        if [ $? = 0 ]; then 
+            login_shell=$output; 
+            if login_shell_is_ok; then return 0; fi
+        fi
+    fi
+    return 1
+}
+
+execute_with_python() {
+    if detect_python; then
+        exec $python -c "import os; os.execl('$login_shell', '-' '$shell_name')"
+    fi
+    return 1;
+}
+
+LOGIN_SHELL="OVERRIDE_LOGIN_SHELL"
+if [ -n "$LOGIN_SHELL" ]; then
+    login_shell="$LOGIN_SHELL"
+else
+    using_getent || using_id || using_python || using_passwd || die "Could not detect login shell";
+fi
+shell_name=$(basename $login_shell)
+
+export KITTY_SHELL_INTEGRATION="SHELL_INTEGRATION_VALUE"
+
+exec_bash_with_integration() {
+    export ENV="$shell_integration_dir/bash/kitty.bash"
+    export KITTY_BASH_INJECT="1"
+    exec "$login_shell" "--posix"
+}
+
+exec_zsh_with_integration() {
+    zdotdir="$ZDOTDIR"
+    if [ -z "$zdotdir" ]; then 
+        zdotdir=~; 
+        unset KITTY_ORIG_ZDOTDIR
+    else
+        export KITTY_ORIG_ZDOTDIR="$zdotdir"
+    fi
+    # dont prevent zsh-new-user from running
+    if [ -e "$zdotdir/.zshrc" || -e "$zdotdir/.zshenv" || -e "$zdotdir/.zprofile" || -e "$zdotdir/.zlogin" ]; then
+        export ZDOTDIR="$shell_integration_dir/zsh"
+        exec "$login_shell" "-l"
+    fi
+}
+
+exec_fish_with_integration() {
+    if [ -z "$XDG_DATA_DIRS" ]; then
+        export XDG_DATA_DIRS="$shell_integration_dir"
+    else
+        export XDG_DATA_DIRS="$shell_integration_dir:$XDG_DATA_DIRS"
+    fi
+    export KITTY_FISH_XDG_DATA_DIR="$shell_integration_dir"
+    exec "$login_shell" "-l"
+}
+
+exec_with_shell_integration() {
+    case "$login_shell" in
+        *"zsh")
+            exec_zsh_with_integration
+            ;;
+        *"bash")
+            exec_bash_with_integration
+            ;;
+        *"fish")
+            exec_fish_with_integration
+            ;;
+    esac
+}
+
+case "$KITTY_SHELL_INTEGRATION" in
+    "")
+        unset KITTY_SHELL_INTEGRATION
+        ;;
+    *"no-rc"*)
+        ;;
+    *)
+        exec_with_shell_integration
+        unset KITTY_SHELL_INTEGRATION
+        ;;
+esac
+
+# We need to pass the first argument to the executed program with a leading -
+# to make sure the shell executes as a login shell. Note that not all shells
+# support exec -a so we use the below to try to detect such shells
+shell_name=$(basename $login_shell)
+if [ -z "$PIPESTATUS" ]; then
+    # the dash shell does not support exec -a and also does not define PIPESTATUS
+    execute_with_python
+    exec $login_shell "-l"
+fi
+exec -a "-$shell_name" $login_shell