import json
from pathlib import Path

from enroll.diff import compare_harvests


def _write_bundle(root: Path, state: dict, artifacts: dict[str, bytes]) -> None:
    root.mkdir(parents=True, exist_ok=True)
    (root / "state.json").write_text(json.dumps(state, indent=2), encoding="utf-8")
    for rel, data in artifacts.items():
        p = root / rel
        p.parent.mkdir(parents=True, exist_ok=True)
        p.write_bytes(data)


def test_diff_includes_usr_local_custom_files(tmp_path: Path):
    old = tmp_path / "old"
    new = tmp_path / "new"

    old_state = {
        "host": {"hostname": "h1", "os": "debian"},
        "users": {
            "role_name": "users",
            "users": [],
            "managed_files": [],
            "excluded": [],
            "notes": [],
        },
        "services": [],
        "package_roles": [],
        "manual_packages": ["curl"],
        "manual_packages_skipped": [],
        "etc_custom": {
            "role_name": "etc_custom",
            "managed_files": [],
            "excluded": [],
            "notes": [],
        },
        "usr_local_custom": {
            "role_name": "usr_local_custom",
            "managed_files": [
                {
                    "path": "/usr/local/etc/myapp.conf",
                    "src_rel": "usr/local/etc/myapp.conf",
                    "owner": "root",
                    "group": "root",
                    "mode": "0644",
                    "reason": "usr_local_etc_custom",
                }
            ],
            "excluded": [],
            "notes": [],
        },
    }
    new_state = {
        **old_state,
        "manual_packages": ["curl", "htop"],
        "usr_local_custom": {
            "role_name": "usr_local_custom",
            "managed_files": [
                {
                    "path": "/usr/local/etc/myapp.conf",
                    "src_rel": "usr/local/etc/myapp.conf",
                    "owner": "root",
                    "group": "root",
                    "mode": "0644",
                    "reason": "usr_local_etc_custom",
                },
                {
                    "path": "/usr/local/bin/myscript",
                    "src_rel": "usr/local/bin/myscript",
                    "owner": "root",
                    "group": "root",
                    "mode": "0755",
                    "reason": "usr_local_bin_script",
                },
            ],
            "excluded": [],
            "notes": [],
        },
    }

    _write_bundle(
        old,
        old_state,
        {
            "artifacts/usr_local_custom/usr/local/etc/myapp.conf": b"myapp=1\n",
        },
    )
    _write_bundle(
        new,
        new_state,
        {
            "artifacts/usr_local_custom/usr/local/etc/myapp.conf": b"myapp=2\n",
            "artifacts/usr_local_custom/usr/local/bin/myscript": b"#!/bin/sh\necho hi\n",
        },
    )

    report, has_changes = compare_harvests(str(old), str(new))
    assert has_changes is True

    # Packages: htop was added.
    assert report["packages"]["added"] == ["htop"]

    # Files: /usr/local/etc/myapp.conf should be detected as changed (content sha differs).
    changed_paths = {c["path"] for c in report["files"]["changed"]}
    assert "/usr/local/etc/myapp.conf" in changed_paths

    # Files: new script was added.
    added_paths = {a["path"] for a in report["files"]["added"]}
    assert "/usr/local/bin/myscript" in added_paths