"""wasi-testsuite adapter for the in-tree Zig-based wamr CLI.

Mirrors `wasm-micro-runtime.py` from upstream wasi-testsuite but reads the
runtime path from the `WAMR` environment variable (set by
`zig build wasi-testsuite`) instead of `IWASM`. Lives outside the vendored
submodule (`tests/wasi-testsuite/`) so this repo can edit it without a
fork. Keep the contract aligned with upstream — any drift will silently
break the runner.

The wamr CLI uses a Wasmtime-shaped subcommand layout
(`wamr run <file.wasm>`, `wamr version`), so the adapter inserts `run` /
`version` after the binary path.
"""

import os
import shlex
import shutil
import subprocess
import tempfile
from pathlib import Path
from typing import Dict, List, Tuple


WAMR = shlex.split(os.getenv("WAMR", "wamr"))


def get_name() -> str:
    return "wamr-zig"


def get_version() -> str:
    result = subprocess.run(
        WAMR + ["version"],
        encoding="UTF-8",
        capture_output=True,
        check=True,
    )
    output = result.stdout.splitlines()[0].split(" ")
    return output[1]


def get_wasi_versions() -> List[str]:
    # Declares both Preview 1 (covered by `zig build wasi-testsuite`) and
    # Preview 3 (covered by `zig build wasi-p3-testsuite`, issue #489) so
    # the upstream `UnsupportedWasiTestExcludeFilter` doesn't auto-skip
    # the wasm32-wasip3 fixtures. Per-test gating for incomplete P3
    # adapter coverage lives in `tests/wasi-p3-testsuite-skip.json`.
    return ["wasm32-wasip1", "wasm32-wasip3"]


def get_wasi_worlds() -> List[str]:
    # Declares both `wasi:cli/command` (every cli/filesystem/sockets
    # fixture) and `wasi:http/service` (the `http-service` fixture,
    # which exports `wasi:http/incoming-handler@0.3.0.handle`). Without
    # the http world declaration the upstream
    # `UnsupportedWasiTestExcludeFilter` auto-skips http-service
    # before any of our dispatch glue gets a chance to run. (#570)
    return ["wasi:cli/command", "wasi:http/service"]


def _isolate_preopens(dirs: List[Tuple[Path, str]]) -> List[Tuple[Path, str]]:
    """Snapshot each preopen host directory into a fresh tempdir so a
    filesystem test that mutates the mapped directory doesn't pollute
    state for subsequent tests in the run (the upstream runner reuses
    the same host paths across invocations). The tempdirs are leaked
    intentionally — they're tiny and the per-suite TMPDIR is cleared
    between CI invocations. (#564.)
    """
    isolated: List[Tuple[Path, str]] = []
    for host, guest in dirs:
        host_path = Path(host)
        if not host_path.is_dir():
            isolated.append((host, guest))
            continue
        snapshot = Path(tempfile.mkdtemp(prefix="wamr-zig-fs-"))
        # `copytree(..., dirs_exist_ok=True)` lets us land into the
        # just-created mkdtemp root rather than under a child dir.
        shutil.copytree(host_path, snapshot, dirs_exist_ok=True, symlinks=True)
        isolated.append((snapshot, guest))
    return isolated


def compute_argv(
    test_path: str,
    args_env_dirs: Tuple[List[str], Dict[str, str], List[Tuple[Path, str]]],
    proposals: List[str],
    wasi_world: str,
    wasi_version: str,
) -> List[str]:
    argv: List[str] = []
    argv += WAMR
    argv += ["run"]
    args, env, dirs = args_env_dirs

    for k, v in env.items():
        argv += ["--env", f"{k}={v}"]

    for host, guest in _isolate_preopens(dirs):
        argv += ["--map-dir", f"{host}::{guest}"]

    # wasi:sockets fixtures need an explicit allow-list to escape the
    # adapter's default deny-all posture. Localhost is sufficient for
    # every wasi-testsuite sockets fixture (they all bind/connect to
    # 127.0.0.1 / ::1). (#520 wave 2)
    if "sockets" in proposals:
        argv += ["--allow-net", "127.0.0.0/8"]
        argv += ["--allow-net", "::1/128"]

    # `wasi:http/service` fixtures (`http-service.wasm`) export
    # `wasi:http/incoming-handler@0.3.0.handle` and expect the host
    # to bind a TCP listener, accept on it, and route incoming HTTP
    # over the guest export. Bare `--listen` selects an ephemeral
    # 127.0.0.1:0 bind and triggers `announce_listening` — the wamr
    # CLI prints `http://<host>:<port>` to stderr so the
    # wasi-testsuite `TestCaseRunner.get_http_server` URL-scrape
    # succeeds. (#570)
    if wasi_world == "wasi:http/service":
        argv += ["--listen"]

    argv += [test_path]
    argv += args
    return argv