"""Go UPM system tests — go install + go mod download via bare prebake."""

import os
import subprocess
import pytest
from probes import binary_exists


def _yml(actions_yaml):
    return """\
version: "1.0"
environment:
  builder: "baremetal"
bootstrap:
  user: "vulcan"
security:
  drop_after: never
dependencies:
  config:
    repology_endpoint: none
  system:
    pacman:
      packages:
  user:
    go:
""" + _indent(actions_yaml, 6)


def _indent(text, spaces):
    pad = " " * spaces
    return "\n".join(pad + line if line.strip() else line
                     for line in text.splitlines())


def _query(cmd, timeout=60):
    try:
        r = subprocess.run(cmd, capture_output=True, text=True,
                           timeout=timeout, env={**os.environ, "HOME": "/root"})
        return {"success": r.returncode == 0, "stdout": r.stdout, "stderr": r.stderr}
    except Exception as e:
        return {"success": False, "stdout": "", "stderr": str(e)}


def _run_prebake(run_baker, config_path, timeout, operation, expected):
    r = run_baker("prebake", config_path, timeout=timeout,
                  env={"RUST_LOG": os.environ.get("RUST_LOG", "info")})
    full_log = "[stdout]\n{}\n[stderr]\n{}".format(r["stdout"], r["stderr"])
    return r, full_log


def _check(r, full_log, operation, expected):
    if r["success"]:
        return
    diagnosis = ""
    try:
        from llm import judge_log
        j = judge_log(operation, expected, full_log)
        diagnosis = "\n  LLM diagnosis: [{}] {}".format(j.verdict, j.reason)
    except Exception:
        pass
    pytest.fail("{} failed ({} chars):\n{}\n{}".format(
        operation, len(full_log), full_log, diagnosis))


def _gopath_bin(name):
    """Return the expected $GOPATH/bin path or $HOME/go/bin fallback."""
    gopath = os.environ.get("GOPATH", "")
    home = os.environ.get("HOME", "/root")
    if not gopath:
        gopath = os.path.join(home, "go")
    return os.path.join(gopath, "bin", name)


# ── go install ──────────────────────────────────────────────────────

def test_go_install(run_baker, work_dir):
    """go install a small known tool, verify binary exists."""
    cfg = os.path.join(work_dir, "go.yml")
    with open(cfg, "w") as f:
        f.write(_yml("""\
proxy: "https://goproxy.cn,direct"
packages: ["golang.org/x/tools/cmd/goimports@latest"]
"""))

    r, log = _run_prebake(run_baker, cfg, timeout=300,
                          operation="go install goimports",
                          expected="goimports binary appears in GOPATH/bin")
    _check(r, log, "go install", "goimports installed")

    bin_path = _gopath_bin("goimports")
    assert binary_exists(bin_path), f"goimports not found at {bin_path}"


# ── go mod download ─────────────────────────────────────────────────

def test_go_mod_download(run_baker, work_dir):
    """Create a minimal go.mod, download deps, verify cache populated."""
    mod_dir = os.path.join(work_dir, "testmod")
    os.makedirs(mod_dir, exist_ok=True)
    with open(os.path.join(mod_dir, "go.mod"), "w") as f:
        f.write("module test\n\ngo 1.21\n\nrequire golang.org/x/text v0.14.0\n")

    cfg = os.path.join(work_dir, "go.yml")
    with open(cfg, "w") as f:
        f.write(_yml("""\
proxy: "https://goproxy.cn,direct"
modules: ["{mod}"]
""".format(mod=os.path.join(mod_dir, "go.mod"))))

    r, log = _run_prebake(run_baker, cfg, timeout=180,
                          operation="go mod download",
                          expected="GOMODCACHE has x/text cached")
    _check(r, log, "go mod download", "dependencies downloaded to module cache")

    # Check GOMODCACHE or default $GOPATH/pkg/mod
    gomodcache = os.environ.get("GOMODCACHE", "")
    if not gomodcache:
        gopath = os.environ.get("GOPATH",
                                os.path.join(os.environ.get("HOME", "/root"), "go"))
        gomodcache = os.path.join(gopath, "pkg", "mod")
    cache_dir = os.path.join(gomodcache, "golang.org", "x")
    assert os.path.isdir(cache_dir), f"no module cache at {cache_dir}"


# ── go install + mod download ───────────────────────────────────────

def test_go_full(run_baker, work_dir):
    """go install a tool AND download project deps."""
    mod_dir = os.path.join(work_dir, "testmod2")
    os.makedirs(mod_dir, exist_ok=True)
    with open(os.path.join(mod_dir, "go.mod"), "w") as f:
        f.write("module test\n\ngo 1.21\n")

    cfg = os.path.join(work_dir, "go.yml")
    with open(cfg, "w") as f:
        f.write(_yml("""\
proxy: "https://goproxy.cn,direct"
packages: ["golang.org/x/tools/cmd/goimports@latest"]
modules: ["{mod}"]
env:
  GOMODCACHE: "/tmp/gocache"
""".format(mod=os.path.join(mod_dir, "go.mod"))))

    r, log = _run_prebake(run_baker, cfg, timeout=300,
                          operation="go install + go mod download",
                          expected="goimports installed, module cache populated")
    _check(r, log, "go full", "tool installed + deps downloaded")

    assert binary_exists(_gopath_bin("goimports")), "goimports missing"
    assert os.path.isdir("/tmp/gocache"), "GOMODCACHE not created"