import pytest

from webdriver import WebElement

from tests.support.classic.asserts import assert_error, assert_success
from tests.support.image import png_dimensions
from . import element_dimensions, take_element_screenshot


def test_no_top_browsing_context(session, closed_window):
    response = take_element_screenshot(session, "foo")
    assert_error(response, "no such window")


def test_no_browsing_context(session, closed_frame, inline):
    session.url = inline("<input>")
    element = session.find.css("input", all=False)

    response = take_element_screenshot(session, element.id)
    screenshot = assert_success(response)

    assert png_dimensions(screenshot) == element_dimensions(session, element)


def test_no_such_element_with_invalid_value(session):
    element = WebElement(session, "foo")

    response = take_element_screenshot(session, element.id)
    assert_error(response, "no such element")


def test_no_such_element_with_shadow_root(session, get_test_page):
    session.url = get_test_page()

    element = session.find.css("custom-element", all=False)

    result = take_element_screenshot(session, element.shadow_root.id)
    assert_error(result, "no such element")


@pytest.mark.parametrize("closed", [False, True], ids=["open", "closed"])
def test_no_such_element_from_other_window_handle(session, inline, closed):
    session.url = inline("<div id='parent'><p/>")
    element = session.find.css("#parent", all=False)

    new_handle = session.new_window()

    if closed:
        session.window.close()

    session.window_handle = new_handle

    response = take_element_screenshot(session, element.id)
    assert_error(response, "no such element")


@pytest.mark.parametrize("closed", [False, True], ids=["open", "closed"])
def test_no_such_element_from_other_frame(session, get_test_page, closed):
    session.url = get_test_page(as_frame=True)

    frame = session.find.css("iframe", all=False)
    session.switch_to_frame(frame)

    element = session.find.css("div", all=False)

    session.switch_to_parent_frame()

    if closed:
        session.execute_script("arguments[0].remove();", args=[frame])

    response = take_element_screenshot(session, element.id)
    assert_error(response, "no such element")


@pytest.mark.parametrize("as_frame", [False, True], ids=["top_context", "child_context"])
def test_stale_element_reference(session, stale_element, as_frame):
    element = stale_element("input#text", as_frame=as_frame)

    result = take_element_screenshot(session, element.id)
    assert_error(result, "stale element reference")


def test_format_and_dimensions(session, inline):
    session.url = inline("<input>")
    element = session.find.css("input", all=False)

    response = take_element_screenshot(session, element.id)
    screenshot = assert_success(response)

    assert png_dimensions(screenshot) == element_dimensions(session, element)


def test_clip_huge_element_to_viewport(session, inline):
    width = "32768px"
    height = "32768px"

    session.url = inline(f"""
        <style>
            body {{ margin: 0; }}
        </style>
        <div style='width: {width}; height: {height}; background-color: black;'></div>
    """)
    element = session.find.css("div", all=False)

    response = take_element_screenshot(session, element.id)

    screenshot = assert_success(response)

    viewport = session.execute_script("""
        return {
            width: window.innerWidth,
            // The element is scrolled into view first, which causes the page to scroll to the bottom.
            // This means the rectangle intersection logic will only capture the viewport height
            // without the scrollbar height. Therefore, we use visualViewport.height.
            height: window.visualViewport.height,
            devicePixelRatio: window.devicePixelRatio
        };
    """)

    expected_width = round(viewport["width"] * viewport["devicePixelRatio"])
    expected_height = round(viewport["height"] * viewport["devicePixelRatio"])

    # 5. Assert the screenshot was clipped to the viewport size
    assert png_dimensions(screenshot) == (expected_width, expected_height)