[be]improve remote execution (#7907)

add remote execution option such as debug session
refactor the command internace in to seperate files

Author: yangw-dev

tools/remote_execution/blast/src/re_cli/cli_helper.py

@@ -0,0 +1,20 @@
+"""Shared utilities for commands."""
+
+from typing import cast
+
+import click
+
+from ..core.core_types import console
+from ..core.k8s_client import K8sClient, K8sConfig
+
+
+def get_client(ctx: click.Context) -> K8sClient:
+    """Lazy-init K8sClient from click context. Call from any command."""
+    if ctx.obj.get("_client") is None:
+        console.print("[Auth] getting K8sConfig")
+        config = K8sConfig(
+            namespace=ctx.obj["_k8s_namespace"],
+            timeout=ctx.obj["_k8s_timeout"],
+        )
+        ctx.obj["_client"] = K8sClient(config)
+    return cast(K8sClient, ctx.obj["_client"])

tools/remote_execution/blast/src/re_cli/commands/__init__.py

@@ -0,0 +1,33 @@
+"""Cancel command."""
+
+import json
+import sys
+
+import click
+
+from ..core.core_types import console
+from . import get_client
+
+
+@click.command()
+@click.argument("run_id", type=str)
+@click.pass_context
+def cancel(ctx, run_id):
+    """Cancel a run and all its tasks."""
+    as_json = ctx.obj.get("as_json", False)
+
+    client = get_client(ctx)
+
+    try:
+        result = client.cancel_run(run_id)
+        if as_json:
+            print(json.dumps({"run_id": run_id, "status": "cancelled", **result}))
+
+        console.print(f"[yellow]○ Run {run_id} cancelled[/yellow]")
+        if result.get("message"):
+            console.print(f"[dim]  {result['message']}[/dim]")
+    except Exception as e:
+        if as_json:
+            print(json.dumps({"error": str(e)}))
+        console.print(f"[red]Error: {e}[/red]")
+        sys.exit(1)