Updates to compose_down command to match docker compose behavior (Issue: containers#267)

Signed-off-by: Yashodhan Pise <[email protected]>

Author: me-coder

podman_compose.py

@@ -2854,12 +2854,30 @@ def get_volume_names(compose, cnt):
 
 @cmd_run(podman_compose, "down", "tear down entire stack")
 async def compose_down(compose: PodmanCompose, args):
+    # get_excluded fails as no-deps is not a supported cli arg
     excluded = get_excluded(compose, args)
     podman_args = []
     timeout_global = getattr(args, "timeout", None)
     containers = list(reversed(compose.containers))
 
-    down_tasks = []
+    # Get list of currently running containers
+    running_cnt_names = (
+                            (
+                                await compose.podman.output(
+                                    [],
+                                    "ps",
+                                    [
+                                        "--filter",
+                                        f"label=io.podman.compose.project={compose.project_name}",
+                                        "-a",
+                                        "--format",
+                                        "{{ .Names }}",
+                                    ],
+                                )
+                            )
+                            .decode("utf-8")
+                            .splitlines()
+                        )
 
     for cnt in containers:
         if cnt["_service"] in excluded:
@@ -2871,38 +2889,44 @@ async def compose_down(compose: PodmanCompose, args):
             timeout = str_to_seconds(timeout_str)
         if timeout is not None:
             podman_stop_args.extend(["-t", str(timeout)])
-        down_tasks.append(
-            asyncio.create_task(
-                compose.podman.run([], "stop", [*podman_stop_args, cnt["name"]]), name=cnt["name"]
-            )
-        )
-    await asyncio.gather(*down_tasks)
+        await compose.podman.run([], "stop", [*podman_stop_args, cnt["name"]])
+
     for cnt in containers:
         if cnt["_service"] in excluded:
             continue
         await compose.podman.run([], "rm", [cnt["name"]])
+        if cnt["name"] in running_cnt_names:
+            running_cnt_names.remove(cnt["name"])
+
+    # The logic is updated based on docker compose documentation:
+    # `--remove-orphans`: Remove containers for services not defined in the Compose file
+    # Ref: https://docs.docker.com/reference/cli/docker/compose/down/#options
+    orphan_cnt_names = []
+    for cnt in running_cnt_names:
+        if not any(f"{compose.project_name}_{service}_" in cnt for service in compose.all_services):
+            orphan_cnt_names.append(cnt)
+            running_cnt_names.remove(cnt)
+
+    # We list the containers and remove them from running container list
+    # However, we stop them only if porvided with CLI arg `--remove-orphans`
     if args.remove_orphans:
-        names = (
-            (
-                await compose.podman.output(
-                    [],
-                    "ps",
-                    [
-                        "--filter",
-                        f"label=io.podman.compose.project={compose.project_name}",
-                        "-a",
-                        "--format",
-                        "{{ .Names }}",
-                    ],
-                )
-            )
-            .decode("utf-8")
-            .splitlines()
-        )
-        for name in names:
+        for name in orphan_cnt_names:
             await compose.podman.run([], "stop", [*podman_args, name])
-        for name in names:
+        for name in orphan_cnt_names:
             await compose.podman.run([], "rm", [name])
+
+    for cnt in running_cnt_names:
+        # This logic goes away if the containers list can be updated accordingly at source
+        # Clear containers not formed out of the current compose file definitions
+        # E.g.: By using CLI `up --scale <APP>=<NUM>` option
+        podman_stop_args = [*podman_args]
+        if timeout_global is not None:
+            podman_stop_args.extend(["-t", str(timeout_global)])
+        await compose.podman.run( [], "stop", [*podman_stop_args, cnt])
+
+    for cnt in running_cnt_names:
+        await compose.podman.run([], "rm", [cnt])
+
     if args.volumes:
         vol_names_to_keep = set()
         for cnt in containers: