From 9be68966b498fdc3db135c2ecb9743c45a3bf69d Mon Sep 17 00:00:00 2001 From: Kovid Goyal Date: Wed, 8 Jun 2022 18:53:03 +0530 Subject: [PATCH] Also detect ssh kitten run as an overlay --- kittens/ssh/utils.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/kittens/ssh/utils.py b/kittens/ssh/utils.py index 1861102a9..fcae4e24b 100644 --- a/kittens/ssh/utils.py +++ b/kittens/ssh/utils.py @@ -4,7 +4,7 @@ import os import subprocess -from typing import Any, Dict, List +from typing import Any, Dict, List, Sequence from kitty.types import run_once @@ -50,12 +50,14 @@ def ssh_options() -> Dict[str, str]: return ans -def is_kitten_cmdline(q: List[str]) -> bool: +def is_kitten_cmdline(q: Sequence[str]) -> bool: if len(q) < 4: return False if os.path.basename(q[0]).lower() != 'kitty': return False - return q[1:3] == ['+kitten', 'ssh'] or q[1:4] == ['+', 'kitten', 'ssh'] + if q[1:3] == ['+kitten', 'ssh'] or q[1:4] == ['+', 'kitten', 'ssh']: + return True + return q[1:3] == ['+runpy', 'from kittens.runner import main; main()'] and len(q) >= 6 and q[5] == 'ssh' def patch_cmdline(key: str, val: str, argv: List[str]) -> None: