Handle signals in j command

(cherry picked from commit 9d64e6f)

Author: NickCao

packages/jumpstarter-cli/jumpstarter_cli/j.py

@@ -1,24 +1,48 @@
+import concurrent
 import sys
+from contextlib import ExitStack
 
 import asyncclick as click
-from jumpstarter_cli_common.exceptions import handle_exceptions
+from anyio import create_task_group, get_cancelled_exc_class, run, to_thread
+from anyio.from_thread import BlockingPortal
+from jumpstarter_cli_common.exceptions import async_handle_exceptions, leaf_exceptions
+from jumpstarter_cli_common.signal import signal_handler
 
-from jumpstarter.utils.env import env
+from jumpstarter.utils.env import env_async
+
+
+async def j_async():
+    @async_handle_exceptions
+    async def cli():
+        async with BlockingPortal() as portal:
+            with ExitStack() as stack:
+                async with env_async(portal, stack) as client:
+                    async with client.log_stream_async():
+                        await to_thread.run_sync(lambda: client.cli()(standalone_mode=False))
+
+    try:
+        async with create_task_group() as tg:
+            tg.start_soon(signal_handler, tg.cancel_scope)
+
+            try:
+                await cli()
+            finally:
+                tg.cancel_scope.cancel()
+
+    except* click.ClickException as excgroup:
+        for exc in leaf_exceptions(excgroup):
+            exc.show()
+
+        sys.exit(1)
+    except* (
+        get_cancelled_exc_class(),
+        concurrent.futures._base.CancelledError,
+    ) as _:
+        sys.exit(2)
 
 
 def j():
-    with env() as client:
-
-        @handle_exceptions
-        def cli():
-            with client.log_stream():
-                client.cli()(standalone_mode=False)
-
-        try:
-            cli()
-        except click.ClickException as e:
-            e.show()
-            sys.exit(1)
+    run(j_async)
 
 
 if __name__ == "__main__":