♻️ refactor: include waiting for containers' startup in docker_start

Author: bassemkaroui

template/duties.py.jinja

@@ -4,6 +4,7 @@ import os
 import shutil
 {%- if dockerfile %}
 import sys
+import time
 from contextlib import contextmanager
 {%- endif %}
 from pathlib import Path
@@ -98,22 +99,6 @@ def _is_interactive() -> bool:
     return sys.stdin.isatty() and os.environ.get("CI", "").lower() != "true"
 
 
-def _wait_for_dev_containers_start(ctx: Context) -> None:
-    """Start the dev containers and wait until they are fully up.
-
-    This function triggers the `docker_start` duty with the `dev` profile,
-    then continuously checks the containers status until `{{ project_name }}-dev`
-    is running.
-    """
-    docker_start.run(env="dev")
-    while True:
-        output = ctx.run(
-            ["docker", "compose", "top"], silent=True, allow_overrides=False, capture=True
-        )
-        if "{{ project_name }}-dev" in output:
-            break
-
-
 @contextmanager
 def _setup_container(ctx: Context) -> Iterator[None]:
     """Ensure the dev Docker container is running for the duration of a context block.
@@ -129,13 +114,13 @@ def _setup_container(ctx: Context) -> Iterator[None]:
     )
     if build_docker_images:
         docker_build.run(env="dev")
-        _wait_for_dev_containers_start(ctx)
+        docker_start.run(env="dev")
     else:
         output = ctx.run(
             ["docker", "compose", "top"], silent=True, allow_overrides=False, capture=True
         )
         if "{{ project_name }}-dev" not in output:
-            _wait_for_dev_containers_start(ctx)
+            docker_start.run(env="dev")
     try:
         yield
     finally:
@@ -309,6 +294,14 @@ def docker_start(ctx: Context, env: str | None = None) -> None:
         ctx.run(
             ["docker", "compose", "--profile", "prod", "up", "-d"], title="Starting prod containers"
         )
+    while True:
+        output = ctx.run(
+            ["docker", "compose", "top"], silent=True, allow_overrides=False, capture=True
+        )
+        service = "{{ project_name }}" if env == "prod" else "{{ project_name }}-dev"
+        if service in output:
+            break
+        time.sleep(0.5)
 
 
 @duty(