from PIL import Image

from src.postprocess import SIZE_PRESETS, add_badge, add_watermark, fit_canvas
from src.providers.openai_client import orientation_for, resolve_size


def _img(w=1024, h=1536, color="#888888"):
    return Image.new("RGB", (w, h), color)


def test_fit_canvas_cover_exact_size():
    out = fit_canvas(_img(), (900, 1200), "Обрезать по краям")
    assert out.size == (900, 1200)


def test_fit_canvas_contain_pads_with_background():
    out = fit_canvas(_img(1024, 1024), (1920, 1080), "Вписать целиком", "#FFFFFF")
    assert out.size == (1920, 1080)
    assert out.getpixel((5, 5)) == (255, 255, 255)  # поле залито


def test_fit_canvas_passthrough():
    src = _img(512, 512)
    assert fit_canvas(src, None).size == (512, 512)


def test_badge_keeps_size_and_changes_pixels():
    src = _img(900, 1200, "#FFFFFF")
    out = add_badge(src, "Держит тепло 12 часов", position="Слева сверху", fill="#E4287C")
    assert out.size == src.size
    assert out.tobytes() != src.tobytes()


def test_empty_badge_is_noop():
    src = _img(200, 200)
    assert add_badge(src, "   ") is src


def test_watermark_keeps_size():
    src = _img(600, 800)
    assert add_watermark(src, "черновик").size == (600, 800)


def test_orientation_and_size_matrix():
    assert orientation_for(900, 1200) == "portrait"
    assert orientation_for(1920, 1080) == "landscape"
    assert orientation_for(1000, 1000) == "square"
    assert resolve_size("gpt-image-1", "portrait") == "1024x1536"
    assert resolve_size("dall-e-3", "landscape") == "1792x1024"


def test_all_presets_are_valid():
    for name, value in SIZE_PRESETS.items():
        assert value is None or (len(value) == 2 and all(v > 0 for v in value)), name