from pathlib import Path
import json
import os

from pyucc.core import countings_impl as ci


def make_item_variant(key_map):
    # helper to create dict with varying keys
    return key_map


def test_map_pygount_json_item_variants():
    item1 = {
        "raw_total_lines": 10,
        "code": 7,
        "comment": 2,
        "blank": 1,
        "language": "Python",
        "filename": "a.py",
    }
    res1 = ci._map_pygount_json_item(item1)
    assert res1["physical_lines"] == 10
    assert res1["code_lines"] == 7
    assert res1["comment_lines"] == 2
    assert res1["blank_lines"] == 1
    assert res1["language"] == "Python"
    assert res1["file"].endswith("a.py")

    # variant with different key names
    item2 = {
        "n_lines": 4,
        "n_code": 3,
        "n_comment": 1,
        "n_blank": 0,
        "lang": "C",
        "file": "b.c",
    }
    res2 = ci._map_pygount_json_item(item2)
    assert res2["physical_lines"] == 4
    assert res2["code_lines"] == 3
    assert res2["comment_lines"] == 1
    assert res2["blank_lines"] == 0
    assert res2["language"] == "C"


def test_analyze_file_counts_fallback(tmp_path):
    p = tmp_path / "sample.txt"
    data = "line1\n\nline3\nline4\n\n"
    p.write_text(data)
    res = ci.analyze_file_counts(p)
    # pygount counts this as 4 physical lines (doesn't count final empty line)
    assert res["physical_lines"] >= 4  # May be 4 (pygount) or 5 (fallback)
    assert res["blank_lines"] >= 1
    assert "file" in res


def test_analyze_paths_with_missing(tmp_path):
    files = [tmp_path / "nope.py", tmp_path / "also.py"]
    # none exist
    out = ci.analyze_paths(files)
    assert isinstance(out, list)
    assert len(out) == 2
    assert "error" in out[0] or "error" in out[1]