from functools import partial

import pytest
from conftest import DB, Response
from hircine.db.models import Namespace, Tag


@pytest.fixture
def query_comic_tags(schema_execute):
    query = """
    query comicTags($forFilter: Boolean) {
        comicTags(forFilter: $forFilter) {
            edges {
                __typename
                id
                name
            }
            count
        }
    }
    """

    def wrapper(q):
        async def _execute(for_filter=False):
            return await schema_execute(q, {"forFilter": for_filter})

        return _execute

    return wrapper(query)


def build_item(namespace, tag):
    nid, tid = "", ""
    nname, tname = "", ""

    if namespace:
        nid, nname = namespace.id, namespace.name

    if tag:
        tid, tname = tag.id, tag.name

    item = {
        "__typename": "ComicTag",
        "id": f"{nid}:{tid}",
        "name": f"{nname}:{tname}",
    }

    return item


@pytest.mark.anyio
async def test_query_comic_tags_cross(query_comic_tags):
    ns_foo = Namespace(id=1, name="foo")
    ns_bar = Namespace(id=2, name="bar")
    tag_qoo = Tag(id=1, name="qoo", namespaces=[ns_foo, ns_bar])
    tag_qar = Tag(id=2, name="qar", namespaces=[ns_foo, ns_bar])

    await DB.add_all(ns_foo, ns_bar)
    await DB.add_all(tag_qoo, tag_qar)

    builder = partial(build_item)

    response = Response(await query_comic_tags())
    assert response.data["edges"] == [
        builder(ns_bar, tag_qar),
        builder(ns_bar, tag_qoo),
        builder(ns_foo, tag_qar),
        builder(ns_foo, tag_qoo),
    ]


@pytest.mark.anyio
async def test_query_comic_tags_restricted_namespace(query_comic_tags):
    ns_foo = Namespace(id=1, name="foo")
    ns_bar = Namespace(id=2, name="bar")
    tag_qoo = Tag(id=1, name="qoo", namespaces=[ns_bar])
    tag_qar = Tag(id=2, name="qar", namespaces=[ns_foo])

    await DB.add_all(ns_foo, ns_bar)
    await DB.add_all(tag_qoo, tag_qar)

    builder = partial(build_item)

    response = Response(await query_comic_tags())
    assert response.data["edges"] == [
        builder(ns_bar, tag_qoo),
        builder(ns_foo, tag_qar),
    ]


@pytest.mark.anyio
async def test_query_comic_tag_matchers_cross(query_comic_tags):
    ns_foo = Namespace(id=1, name="foo")
    ns_bar = Namespace(id=2, name="bar")
    tag_qoo = Tag(id=1, name="qoo", namespaces=[ns_foo, ns_bar])
    tag_qar = Tag(id=2, name="qar", namespaces=[ns_foo, ns_bar])

    await DB.add_all(ns_foo, ns_bar, tag_qoo, tag_qar)

    builder = partial(build_item)

    response = Response(await query_comic_tags(for_filter=True))
    assert response.data["edges"] == [
        builder(ns_bar, None),
        builder(ns_foo, None),
        builder(None, tag_qar),
        builder(None, tag_qoo),
        builder(ns_bar, tag_qar),
        builder(ns_bar, tag_qoo),
        builder(ns_foo, tag_qar),
        builder(ns_foo, tag_qoo),
    ]


@pytest.mark.anyio
async def test_query_comic_tag_matchers_restricted_namespace(query_comic_tags):
    ns_foo = Namespace(id=1, name="foo")
    ns_bar = Namespace(id=2, name="bar")
    tag_qoo = Tag(id=1, name="qoo", namespaces=[ns_bar])
    tag_qar = Tag(id=2, name="qar", namespaces=[ns_foo])

    await DB.add_all(ns_foo, ns_bar, tag_qoo, tag_qar)

    builder = partial(build_item)

    response = Response(await query_comic_tags(for_filter=True))
    assert response.data["edges"] == [
        builder(ns_bar, None),
        builder(ns_foo, None),
        builder(None, tag_qar),
        builder(None, tag_qoo),
        builder(ns_bar, tag_qoo),
        builder(ns_foo, tag_qar),
    ]