import pytest

from specklepy.core.api.operations import deserialize, serialize
from specklepy.objects.geometry import Point, Polyline
from specklepy.objects.models.units import Units
from specklepy.objects.primitive import Interval


@pytest.fixture
def open_square_coords():
    return [
        0.0,
        0.0,
        0.0,  # point 1
        1.0,
        0.0,
        0.0,  # point 2
        1.0,
        1.0,
        0.0,  # point 3
        0.0,
        1.0,
        0.0,  # point 4
    ]


@pytest.fixture
def closed_square_coords():
    return [
        0.0,
        0.0,
        0.0,  # point 1
        1.0,
        0.0,
        0.0,  # point 2
        1.0,
        1.0,
        0.0,  # point 3
        0.0,
        1.0,
        0.0,  # point 4
        0.0,
        0.0,
        0.0,  # point 5 (same as point 1)
    ]


@pytest.fixture
def sample_polyline(open_square_coords):
    return Polyline(value=open_square_coords, units=Units.m)


def test_polyline_creation(open_square_coords):
    polyline = Polyline(value=open_square_coords, units=Units.m)
    assert polyline.value == open_square_coords
    assert polyline.units == Units.m.value


def test_polyline_domain(sample_polyline):
    assert isinstance(sample_polyline.domain, Interval)
    assert sample_polyline.domain.start == 0.0
    assert sample_polyline.domain.end == 1.0


def test_polyline_is_closed(open_square_coords, closed_square_coords):
    open_poly = Polyline(value=open_square_coords, units=Units.m)
    closed_poly = Polyline(value=closed_square_coords, units=Units.m)

    assert not open_poly.is_closed()
    assert closed_poly.is_closed()


def test_polyline_is_closed_with_tolerance(open_square_coords):
    almost_closed = open_square_coords + [
        0.0,
        0.0,
        0.001,
    ]  # last point slightly above start
    poly = Polyline(value=almost_closed, units=Units.m)

    assert not poly.is_closed(tolerance=1e-6)
    assert poly.is_closed(tolerance=0.01)


def test_polyline_length_open(sample_polyline):
    sample_polyline.length = sample_polyline.calculate_length()
    assert sample_polyline.length == 3.0


def test_polyline_length_closed(closed_square_coords):
    polyline = Polyline(value=closed_square_coords, units=Units.m)
    polyline.length = polyline.calculate_length()
    assert polyline.length == 4.0


def test_polyline_get_points(sample_polyline):
    points = sample_polyline.get_points()

    assert len(points) == 4
    assert all(isinstance(p, Point) for p in points)
    assert all(p.units == Units.m.value for p in points)

    # Create expected points
    expected_points = [
        Point(x=0.0, y=0.0, z=0.0, units=Units.m),
        Point(x=1.0, y=0.0, z=0.0, units=Units.m),
        Point(x=1.0, y=1.0, z=0.0, units=Units.m),
        Point(x=0.0, y=1.0, z=0.0, units=Units.m),
    ]

    # Check coordinates match
    for actual, expected in zip(points, expected_points, strict=False):
        assert actual.x == expected.x
        assert actual.y == expected.y
        assert actual.z == expected.z


def test_polyline_invalid_coordinates():
    invalid_coords = [0.0, 0.0, 0.0, 1.0, 1.0]  # missing one coordinate
    with pytest.raises(ValueError):
        polyline = Polyline(value=invalid_coords, units=Units.m)
        polyline.get_points()


def test_polyline_units(open_square_coords):
    polyline = Polyline(value=open_square_coords, units=Units.m)
    assert polyline.units == Units.m.value

    polyline.units = "mm"
    assert polyline.units == "mm"


def test_polyline_serialization(sample_polyline):
    serialized = serialize(sample_polyline)
    deserialized = deserialize(serialized)

    assert isinstance(deserialized, Polyline)
    assert deserialized.value == sample_polyline.value
    assert deserialized.units == sample_polyline.units
    assert deserialized.domain.start == sample_polyline.domain.start
    assert deserialized.domain.end == sample_polyline.domain.end