Pull latest datasette image and wait for readiness before opening browser

Author: nezhar

src/vibepod/commands/logs.py

@@ -2,6 +2,9 @@
 
 from __future__ import annotations
 
+import time
+import urllib.error
+import urllib.request
 import webbrowser
 from pathlib import Path
 from typing import Annotated
@@ -16,6 +19,24 @@
 app = typer.Typer(help="View logs and traffic UI")
 
 
+_HEALTH_TIMEOUT = 30
+_HEALTH_INTERVAL = 0.5
+
+
+def _wait_for_datasette(port: int) -> None:
+    url = f"http://localhost:{port}/-/healthz"
+    deadline = time.monotonic() + _HEALTH_TIMEOUT
+    while time.monotonic() < deadline:
+        try:
+            urllib.request.urlopen(url, timeout=2)  # noqa: S310
+            return
+        except urllib.error.HTTPError:
+            return  # server responded — healthy enough
+        except (urllib.error.URLError, OSError):
+            time.sleep(_HEALTH_INTERVAL)
+    warning("Datasette did not become healthy in time — opening browser anyway")
+
+
 @app.command("start")
 def logs_start(
     port: Annotated[int | None, typer.Option("--port", help="Datasette host port")] = None,
@@ -39,17 +60,27 @@ def logs_start(
         error(str(exc))
         raise typer.Exit(EXIT_DOCKER_NOT_RUNNING) from exc
 
+    info("Checking for datasette image updates…")
+    updated = manager.pull_if_newer(datasette_image)
+    if updated:
+        info("New image available — restarting datasette")
+        existing = manager.find_datasette()
+        if existing:
+            existing.remove(force=True)
+
     info(f"Starting Datasette on http://localhost:{datasette_port}")
     manager.ensure_datasette(
         image=datasette_image,
         logs_db_path=logs_db_path,
         proxy_db_path=proxy_db_path,
         port=datasette_port,
     )
+
+    _wait_for_datasette(datasette_port)
     success("Datasette is ready")
 
     if not no_open:
-        webbrowser.open(f"http://localhost:{datasette_port}")
+        webbrowser.open(f"http://localhost:{datasette_port}/-/dashboards")
 
 
 @app.command("stop")

src/vibepod/core/docker.py

@@ -70,6 +70,31 @@ def pull_image(self, image: str) -> None:
         except APIError as exc:
             raise DockerClientError(f"Failed to pull image {image}: {exc}") from exc
 
+    def pull_if_newer(self, image: str) -> bool:
+        """Pull *image* and return True if the local image was updated.
+
+        Returns False when the image is already up to date, when the pull
+        fails (e.g. no network / private registry), or when the image only
+        exists locally and cannot be found on a registry.
+        """
+        try:
+            old_id: str | None
+            try:
+                old_id = self.client.images.get(image).id
+            except NotFound:
+                old_id = None
+
+            self.client.images.pull(image)
+
+            try:
+                new_id = self.client.images.get(image).id
+            except NotFound:
+                return False
+
+            return bool(old_id != new_id)
+        except APIError:
+            return False
+
     def ensure_network(self, name: str) -> None:
         try:
             self.client.networks.get(name)