Restore the j command

Author: NickCao

jumpstarter/cli/control.py

@@ -0,0 +1,22 @@
+import os
+
+from anyio.from_thread import start_blocking_portal
+
+from jumpstarter.client import client_from_channel
+from jumpstarter.common.grpc import insecure_channel
+
+
+def main():
+    host = os.environ.get("JUMPSTARTER_HOST", None)
+
+    if host is None:
+        raise RuntimeError("j command can only be used under jmp shell")
+
+    with start_blocking_portal() as portal:
+        channel = portal.call(insecure_channel, host)
+        client = portal.call(client_from_channel, channel, portal)
+        client.cli()()
+
+
+if __name__ == "__main__":
+    main()

jumpstarter/cli/j.py

@@ -3,7 +3,6 @@
 import click
 
 from .client import client
-from .control import control_from_env
 from .exporter import exporter
 from .shell import shell
 from .version import version
@@ -17,6 +16,7 @@ def help():
     click.echo(ctx.parent.get_help())
     ctx.exit()
 
+
 @click.group(no_args_is_help=True)
 def main():
     pass
@@ -26,4 +26,3 @@ def main():
 main.add_command(shell)
 main.add_command(exporter)
 main.add_command(client)
-main.add_command(control_from_env(), "control")

jumpstarter/cli/main.py

@@ -10,7 +10,7 @@ class AliasedGroup(click.Group):
         "create": ["cr"],
         "move": ["mv"],
         "config": ["conf"],
-        "delete": ["del"]
+        "delete": ["del"],
     }
 
     def get_command(self, ctx: click.Context, cmd_name: str):

jumpstarter/cli/util/alias.py

@@ -33,6 +33,7 @@ subdirectory = "python"
 
 [tool.poetry.scripts]
 jmp = "jumpstarter.__main__:main"
+j = "jumpstarter.cli.j:main"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.2.2"