Implement function for getting latest compatible controller version

(cherry picked from commit 8024d36)

Author: NickCao

packages/jumpstarter-cli-admin/jumpstarter_cli_admin/controller.py

@@ -0,0 +1,53 @@
+import aiohttp
+import click
+import semver
+from jumpstarter_cli_common.version import get_client_version
+from packaging.version import Version
+
+
+async def get_latest_compatible_controller_version(
+    client_version: str | None = None,
+):
+    if client_version is None:
+        client_version = Version(get_client_version())
+
+    async with aiohttp.ClientSession(
+        raise_for_status=True,
+    ) as session:
+        try:
+            async with session.get(
+                "https://quay.io/api/v1/repository/jumpstarter-dev/helm/jumpstarter/tag/",
+                timeout=aiohttp.ClientTimeout(total=30),
+            ) as resp:
+                resp = await resp.json()
+        except Exception as e:
+            raise click.ClickException(f"Failed to fetch controller versions: {e}") from e
+
+    compatible = set()
+    fallback = set()
+
+    if not isinstance(resp, dict) or "tags" not in resp or not isinstance(resp["tags"], list):
+        raise click.ClickException("Unexpected response fetching controller version")
+
+    for tag in resp["tags"]:
+        if not isinstance(tag, dict) or "name" not in tag:
+            continue  # Skip malformed tag entries
+
+        try:
+            version = semver.VersionInfo.parse(tag["name"])
+        except ValueError:
+            continue  # ignore invalid versions
+
+        if version.major == client_version.major and version.minor == client_version.minor:
+            compatible.add(version)
+        else:
+            fallback.add(version)
+
+    if compatible:
+        selected = max(compatible)
+    elif fallback:
+        selected = max(fallback)
+    else:
+        raise ValueError("No valid controller versions found in the repository")
+
+    return str(selected)

packages/jumpstarter-cli-admin/jumpstarter_cli_admin/install.py

@@ -2,18 +2,12 @@
 
 import asyncclick as click
 from jumpstarter_cli_common.opt import opt_context, opt_kubeconfig
-from jumpstarter_cli_common.version import get_client_version
 from jumpstarter_kubernetes import helm_installed, install_helm_chart
 
+from .controller import get_latest_compatible_controller_version
 from jumpstarter.common.ipaddr import get_ip_address
 
 
-def get_chart_version() -> str:
-    client_version = get_client_version()
-    parts = client_version.split(".")
-    return f"{parts[0].replace('v', '')}.{parts[1]}.{parts[2]}"
-
-
 @click.command
 @click.option("--helm", type=str, help="Path or name of a helm executable", default="helm")
 @click.option("--name", type=str, help="The name of the chart installation", default="jumpstarter")
@@ -34,7 +28,7 @@ def get_chart_version() -> str:
 @click.option("--nodeport", "mode", flag_value="nodeport", help="Use Nodeport routing (recommended)", default=True)
 @click.option("--ingress", "mode", flag_value="ingress", help="Use a Kubernetes ingress")
 @click.option("--route", "mode", flag_value="route", help="Use an OpenShift route")
-@click.option("-v", "--version", help="The version of the service to install", default=get_chart_version())
+@click.option("-v", "--version", help="The version of the service to install", default=None)
 @opt_kubeconfig
 @opt_context
 async def install(
@@ -69,6 +63,9 @@ async def install(
     if router_endpoint is None:
         router_endpoint = f"router.{basedomain}:8083"
 
+    if version is None:
+        version = await get_latest_compatible_controller_version()
+
     click.echo(f'Installing Jumpstarter service v{version} in namespace "{namespace}" with Helm\n')
     click.echo(f"Chart URI: {chart}")
     click.echo(f"Chart Version: {version}")

packages/jumpstarter-cli-admin/pyproject.toml

@@ -6,7 +6,13 @@ authors = [{ name = "Kirk Brauer", email = "kbrauer@hatci.com" }]
 readme = "README.md"
 license = "Apache-2.0"
 requires-python = ">=3.11"
-dependencies = ["jumpstarter-cli-common", "jumpstarter-kubernetes"]
+dependencies = [
+    "aiohttp>=3.11.18",
+    "jumpstarter-cli-common",
+    "jumpstarter-kubernetes",
+    "packaging>=25.0",
+    "semver~=2.13",
+]
 
 [dependency-groups]
 dev = [