Get clone-in-kitty working for bash as well

kitty/launch.py

@@ -498,9 +498,11 @@ def parse_opts_for_clone(args: List[str]) -> LaunchCLIOptions:
 
 def clone_and_launch(msg: str, window: Window) -> None:
     import base64
+    import json
 
     from .child import cmdline_of_process
     args = []
+    cmdline: List[str] = []
     env: Optional[Dict[str, str]] = None
     cwd = ''
     pid = -1
@@ -524,16 +526,19 @@ def clone_and_launch(msg: str, window: Window) -> None:
                     env[k] = v
         elif k == 'cwd':
             cwd = v
+        elif k == 'argv':
+            cmdline = json.loads(v)
     opts = parse_opts_for_clone(args)
     if cwd:
         opts.cwd = cwd
     opts.copy_colors = True
-    try:
-        cmdline = cmdline_of_process(pid)
-    except Exception:
-        cmdline = []
-    if not cmdline:
-        cmdline = list(window.child.argv)
+    if pid > -1:
+        try:
+            cmdline = cmdline_of_process(pid)
+        except Exception:
+            cmdline = []
+        if not cmdline:
+            cmdline = list(window.child.argv)
     ssh_kitten_cmdline = window.ssh_kitten_cmdline()
     if ssh_kitten_cmdline:
         from kittens.ssh.main import set_cwd_in_cmdline, set_env_in_cmdline

kitty/shell_integration.py

@@ -2,15 +2,17 @@
 # License: GPLv3 Copyright: 2021, Kovid Goyal <kovid at kovidgoyal.net>
 
 
+import base64
+import json
 import os
 import subprocess
 from contextlib import suppress
 from typing import Dict, List, Optional
 
 from .constants import shell_integration_dir
+from .fast_data_types import get_options
 from .options.types import Options, defaults
 from .utils import log_error, which
-from .fast_data_types import get_options
 
 
 def setup_fish_env(env: Dict[str, str], argv: List[str]) -> None:
@@ -133,6 +135,7 @@ def setup_bash_env(env: Dict[str, str], argv: List[str]) -> None:
         return
     env['ENV'] = os.path.join(shell_integration_dir, 'bash', 'kitty.bash')
     env['KITTY_BASH_INJECT'] = ' '.join(inject)
+    env['KITTY_BASH_ORIGINAL_ARGV'] = base64.standard_b64encode(json.dumps(argv).encode('utf-8')).decode('ascii')
     if posix_env:
         env['KITTY_BASH_POSIX_ENV'] = posix_env
     if rcfile: