from typing import Annotated, Any

import msgspec
from hypothesis import given
from hypothesis import strategies as st
from hypothesis.strategies import SearchStrategy
from msgspec import Struct
from msgspec.structs import asdict

from polyfactory.factories.msgspec_factory import MsgspecFactory
from polyfactory.field_meta import FieldMeta


class Foo(Struct):
    foo: Annotated[int, msgspec.Meta(ge=100)]


def handle_constrained_int(field_meta: FieldMeta) -> SearchStrategy[int]:
    return st.integers(min_value=field_meta.constraints.get("ge"))


class FooFactory(MsgspecFactory[Foo]):
    __model__ = Foo

    @classmethod
    def create_hypothesis_strategy(cls) -> SearchStrategy[Foo]:
        st_kwargs: dict[str, SearchStrategy[Any]] = {}
        for field in cls.get_model_fields():
            if field.annotation is int:
                st_kwargs[field.name] = handle_constrained_int(field)

        return st.builds(cls.__model__, **st_kwargs)


polyfactory_foo_st = FooFactory.create_hypothesis_strategy()
hypothesis_foo_st = st.builds(Foo)


@given(polyfactory_foo_st)
def test_polyfactory_foo_st(foo: Foo):
    foo_dict = asdict(foo)
    _ = msgspec.convert(foo_dict, Foo)


@given(hypothesis_foo_st)
def test_hypothesis_foo_st(foo: Foo):
    foo_dict = asdict(foo)
    _ = msgspec.convert(foo_dict, Foo) # this will fail the msgspec validation


if __name__ == "__main__":
    # test_polyfactory_foo_st()
    test_hypothesis_foo_st()