Merge pull request #45 from VibePod/proxy-latest

Add proxy and datasette autopull for latest tag

Author: nezhar

src/vibepod/commands/logs.py

@@ -13,7 +13,7 @@
 
 from vibepod.constants import EXIT_DOCKER_NOT_RUNNING
 from vibepod.core.config import get_config
-from vibepod.core.docker import DockerClientError, DockerManager
+from vibepod.core.docker import DockerClientError, DockerManager, _is_latest_tag
 from vibepod.utils.console import error, info, success, warning
 
 app = typer.Typer(help="View logs and traffic UI")
@@ -60,13 +60,14 @@ 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)
+    if _is_latest_tag(datasette_image):
+        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(

src/vibepod/commands/proxy.py

@@ -9,7 +9,7 @@
 
 from vibepod.constants import EXIT_DOCKER_NOT_RUNNING
 from vibepod.core.config import get_config
-from vibepod.core.docker import DockerClientError, DockerManager
+from vibepod.core.docker import DockerClientError, DockerManager, _is_latest_tag
 from vibepod.utils.console import error, info, success, warning
 
 app = typer.Typer(help="Manage the HTTP(S) proxy")
@@ -42,6 +42,10 @@ def proxy_start() -> None:
 
     manager.ensure_network(network_name)
 
+    if _is_latest_tag(proxy_image):
+        info("Checking for proxy image updates…")
+        manager.pull_if_newer(proxy_image)
+
     info("Starting proxy")
     manager.ensure_proxy(
         image=proxy_image,

src/vibepod/commands/run.py

@@ -23,7 +23,7 @@
     resolve_agent_name,
 )
 from vibepod.core.config import get_config
-from vibepod.core.docker import DockerClientError, DockerManager
+from vibepod.core.docker import DockerClientError, DockerManager, _is_latest_tag
 from vibepod.core.session_logger import SessionLogger
 from vibepod.utils.console import error, info, success, warning
 
@@ -323,6 +323,9 @@ def run(
             .resolve()
         )
 
+        if _is_latest_tag(proxy_image):
+            manager.pull_if_newer(proxy_image)
+
         manager.ensure_proxy(
             image=proxy_image,
             db_path=proxy_db_path,

src/vibepod/core/docker.py

@@ -41,6 +41,12 @@ class DockerClientError(RuntimeError):
     """Raised for Docker availability or lifecycle errors."""
 
 
+def _is_latest_tag(image: str) -> bool:
+    """Return True when *image* uses the ``latest`` tag (explicitly or by omission)."""
+    name = image.split("/")[-1]
+    return ":" not in name or name.endswith(":latest")
+
+
 def _normalize_command(value: Any) -> list[str]:
     """Normalize Docker command/entrypoint values to a list of strings."""
     if value is None: