"""Unit: image_id / tag / push helpers in
bot_bottle.backend.docker.util (PRD 0023 chunk 4c additions).

Tests mock `subprocess.run` and assert on argv shape + parsing.
The actual docker round-trip is covered by the chunk 4c
integration smoke."""

from __future__ import annotations

import subprocess
import tempfile
import unittest
from pathlib import Path
from unittest.mock import patch

from bot_bottle.backend.docker import util as docker_mod
from bot_bottle.workspace import WorkspacePlan


def _ok(stdout: str = "", stderr: str = "") -> subprocess.CompletedProcess:
    return subprocess.CompletedProcess(
        args=[], returncode=0, stdout=stdout, stderr=stderr,
    )


def _fail(stderr: str = "boom") -> subprocess.CompletedProcess:
    return subprocess.CompletedProcess(
        args=[], returncode=1, stdout="", stderr=stderr,
    )


class TestImageId(unittest.TestCase):
    def test_strips_trailing_newline(self):
        # docker image inspect --format ... emits a trailing newline.
        with patch.object(
            docker_mod.subprocess, "run",
            return_value=_ok(stdout="sha256:abcdef\n"),
        ) as run:
            self.assertEqual(
                "sha256:abcdef", docker_mod.image_id("bot-bottle-claude:latest")
            )
        argv = run.call_args.args[0]
        self.assertEqual(
            ["docker", "image", "inspect", "--format", "{{.Id}}", "bot-bottle-claude:latest"],
            argv,
        )

    def test_dies_on_inspect_failure(self):
        with patch.object(
            docker_mod.subprocess, "run", return_value=_fail("No such image"),
        ), patch.object(
            docker_mod, "die", side_effect=SystemExit("die"),
        ) as die:
            with self.assertRaises(SystemExit):
                docker_mod.image_id("missing:tag")
            die.assert_called_once()
            self.assertIn("missing:tag", die.call_args.args[0])


class TestSave(unittest.TestCase):
    def test_save_runs_docker_save(self):
        with patch.object(
            docker_mod.subprocess, "run", return_value=_ok(),
        ) as run:
            docker_mod.save("bot-bottle-claude:latest", "/tmp/img.tar")
        argv = run.call_args.args[0]
        self.assertEqual(
            ["docker", "save", "bot-bottle-claude:latest", "-o", "/tmp/img.tar"],
            argv,
        )


class TestBuildImageWithCwd(unittest.TestCase):
    def test_uses_workspace_plan_paths(self):
        with tempfile.TemporaryDirectory(prefix="bb-docker-cwd.") as tmp:
            workspace = WorkspacePlan(
                enabled=True,
                host_path=Path(tmp),
                guest_home="/guest/home",
                guest_path="/guest/home/workspace",
                workdir="/guest/home/workspace",
            )
            with patch.object(docker_mod.subprocess, "run") as run:
                docker_mod.build_image_with_cwd("derived:tag", "base:tag", workspace)

        argv = run.call_args.args[0]
        dockerfile = run.call_args.kwargs["input"]
        self.assertEqual(["docker", "build", "-t", "derived:tag", "-f", "-", tmp], argv)
        self.assertIn("FROM base:tag\n", dockerfile)
        self.assertIn("COPY --chown=node:node . /guest/home/workspace\n", dockerfile)
        self.assertIn("WORKDIR /guest/home/workspace\n", dockerfile)


if __name__ == "__main__":
    unittest.main()