from django.shortcuts import reverse

import pytest
from rest_framework.status import (
    HTTP_200_OK,
    HTTP_204_NO_CONTENT,
    HTTP_400_BAD_REQUEST,
    HTTP_404_NOT_FOUND,
)

from baserow.contrib.database.views.handler import ViewHandler
from baserow.contrib.database.views.models import ViewGroupBy
from baserow.contrib.database.views.registries import view_type_registry


@pytest.mark.django_db
def test_list_view_group_bys(api_client, data_fixture):
    user, token = data_fixture.create_user_and_token()
    table_1 = data_fixture.create_database_table(user=user)
    table_2 = data_fixture.create_database_table()
    field_1 = data_fixture.create_text_field(table=table_1)
    field_2 = data_fixture.create_text_field(table=table_1)
    field_3 = data_fixture.create_text_field(table=table_2)
    view_1 = data_fixture.create_grid_view(table=table_1, order=1)
    data_fixture.create_grid_view(table=table_1, order=2)
    view_3 = data_fixture.create_grid_view(table=table_2, order=1)
    group_by_1 = data_fixture.create_view_group_by(view=view_1, field=field_1)
    group_by_2 = data_fixture.create_view_group_by(view=view_1, field=field_2)
    data_fixture.create_view_group_by(view=view_3, field=field_3)

    response = api_client.get(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": view_3.id}),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response.json()["error"] == "ERROR_USER_NOT_IN_GROUP"

    response = api_client.get(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": 999999}),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_404_NOT_FOUND
    assert response.json()["error"] == "ERROR_VIEW_DOES_NOT_EXIST"

    response = api_client.get(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": view_1.id}),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert len(response_json) == 2
    assert response_json[0]["id"] == group_by_1.id
    assert response_json[0]["view"] == view_1.id
    assert response_json[0]["field"] == field_1.id
    assert response_json[0]["order"] == group_by_1.order
    assert response_json[0]["width"] == group_by_1.width
    assert response_json[1]["id"] == group_by_2.id

    response = api_client.delete(
        reverse(
            "api:workspaces:item",
            kwargs={"workspace_id": view_1.table.database.workspace.id},
        ),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_204_NO_CONTENT

    response = api_client.get(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": view_1.id}),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_404_NOT_FOUND
    assert response.json()["error"] == "ERROR_VIEW_DOES_NOT_EXIST"


@pytest.mark.django_db
def test_create_view_group_by(api_client, data_fixture):
    user, token = data_fixture.create_user_and_token()
    table_1 = data_fixture.create_database_table(user=user)
    table_2 = data_fixture.create_database_table()
    field_1 = data_fixture.create_text_field(table=table_1)
    field_2 = data_fixture.create_text_field(table=table_2)
    field_3 = data_fixture.create_text_field(table=table_1)
    field_4 = data_fixture.create_text_field(table=table_1)
    link_row_field = data_fixture.create_link_row_field(table=table_1)
    view_1 = data_fixture.create_grid_view(table=table_1)
    view_2 = data_fixture.create_grid_view(table=table_2)

    response = api_client.post(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": view_2.id}),
        {
            "field": field_2.id,
            "order": "ASC",
        },
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response.json()["error"] == "ERROR_USER_NOT_IN_GROUP"

    response = api_client.post(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": 99999}),
        {"field": field_1.id, "order": "ASC"},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_404_NOT_FOUND
    assert response.json()["error"] == "ERROR_VIEW_DOES_NOT_EXIST"

    response = api_client.post(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": view_1.id}),
        {"field": 9999999, "order": "NOT_EXISTING"},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response_json["error"] == "ERROR_REQUEST_BODY_VALIDATION"
    assert response_json["detail"]["field"][0]["code"] == "does_not_exist"
    assert response_json["detail"]["order"][0]["code"] == "invalid_choice"

    response = api_client.post(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": view_1.id}),
        {
            "field": field_2.id,
            "order": "ASC",
        },
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response_json["error"] == "ERROR_FIELD_NOT_IN_TABLE"

    grid_view_type = view_type_registry.get("grid")
    grid_view_type.can_group_by = False
    response = api_client.post(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": view_1.id}),
        {"field": field_1.id, "order": "ASC"},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response_json["error"] == "ERROR_VIEW_GROUP_BY_NOT_SUPPORTED"
    grid_view_type.can_group_by = True

    response = api_client.post(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": view_1.id}),
        {"field": link_row_field.id, "order": "ASC"},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response_json["error"] == "ERROR_VIEW_GROUP_BY_FIELD_NOT_SUPPORTED"

    response = api_client.post(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": view_1.id}),
        {"field": field_1.id, "order": "ASC", "width": 150},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_200_OK
    assert ViewGroupBy.objects.all().count() == 1
    first = ViewGroupBy.objects.all().first()
    assert response_json["id"] == first.id
    assert response_json["view"] == view_1.id
    assert response_json["field"] == field_1.id
    assert response_json["order"] == "ASC"
    assert response_json["width"] == 150

    response = api_client.post(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": view_1.id}),
        {"field": field_1.id, "order": "ASC"},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response_json["error"] == "ERROR_VIEW_GROUP_BY_FIELD_ALREADY_EXISTS"

    response = api_client.post(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": view_1.id}),
        {"field": field_3.id, "order": "DESC"},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_200_OK
    assert response_json["order"] == "DESC"

    response = api_client.post(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": view_1.id}),
        {
            "field": field_4.id,
        },
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_200_OK
    assert response_json["order"] == "ASC"
    assert response_json["width"] == 200

    assert ViewGroupBy.objects.all().count() == 3


@pytest.mark.django_db
def test_cannot_created_group_by(api_client, data_fixture):
    user, token = data_fixture.create_user_and_token()
    field_1 = data_fixture.create_long_text_field(long_text_enable_rich_text=True)
    view_1 = data_fixture.create_grid_view(user=user)

    response = api_client.post(
        reverse("api:database:views:list_group_bys", kwargs={"view_id": view_1.id}),
        {
            "field": field_1.id,
            "order": "ASC",
        },
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response.json()["error"] == "ERROR_VIEW_GROUP_BY_FIELD_NOT_SUPPORTED"


@pytest.mark.django_db
def test_update_field_to_incompatible_group_by_deletes_group(api_client, data_fixture):
    user, token = data_fixture.create_user_and_token()
    field_1 = data_fixture.create_long_text_field(
        user=user, long_text_enable_rich_text=False
    )
    view_1 = data_fixture.create_grid_view(user=user)
    data_fixture.create_view_group_by(field=field_1, view=view_1)

    url = reverse("api:database:fields:item", kwargs={"field_id": field_1.id})
    response = api_client.patch(
        url,
        {"long_text_enable_rich_text": True},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_200_OK
    assert ViewGroupBy.objects.all().count() == 0


@pytest.mark.django_db
def test_get_view_group_by(api_client, data_fixture):
    user, token = data_fixture.create_user_and_token()
    group_by_1 = data_fixture.create_view_group_by(user=user, order="DESC")
    group_by_2 = data_fixture.create_view_group_by()

    response = api_client.get(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_2.id},
        ),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response.json()["error"] == "ERROR_USER_NOT_IN_GROUP"

    response = api_client.get(
        reverse("api:database:views:group_by_item", kwargs={"view_group_by_id": 99999}),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_404_NOT_FOUND
    assert response.json()["error"] == "ERROR_VIEW_GROUP_BY_DOES_NOT_EXIST"

    response = api_client.get(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_1.id},
        ),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_200_OK
    assert ViewGroupBy.objects.all().count() == 2
    first = ViewGroupBy.objects.get(pk=group_by_1.id)
    assert response_json["id"] == first.id
    assert response_json["view"] == first.view_id
    assert response_json["field"] == first.field_id
    assert response_json["order"] == "DESC"

    response = api_client.delete(
        reverse(
            "api:workspaces:item",
            kwargs={"workspace_id": group_by_1.view.table.database.workspace.id},
        ),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_204_NO_CONTENT

    response = api_client.get(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_1.id},
        ),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_404_NOT_FOUND
    assert response.json()["error"] == "ERROR_VIEW_GROUP_BY_DOES_NOT_EXIST"


@pytest.mark.django_db
def test_update_view_group_by(api_client, data_fixture):
    user, token = data_fixture.create_user_and_token()
    group_by_1 = data_fixture.create_view_group_by(user=user, order="DESC")
    group_by_2 = data_fixture.create_view_group_by()
    group_by_3 = data_fixture.create_view_group_by(view=group_by_1.view, order="ASC")
    field_1 = data_fixture.create_text_field(table=group_by_1.view.table)
    link_row_field = data_fixture.create_link_row_field(table=group_by_1.view.table)
    field_2 = data_fixture.create_text_field()

    response = api_client.patch(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_2.id},
        ),
        {"order": "ASC"},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response.json()["error"] == "ERROR_USER_NOT_IN_GROUP"

    response = api_client.patch(
        reverse("api:database:views:group_by_item", kwargs={"view_group_by_id": 9999}),
        {"order": "ASC"},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_404_NOT_FOUND
    assert response.json()["error"] == "ERROR_VIEW_GROUP_BY_DOES_NOT_EXIST"

    response = api_client.patch(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_1.id},
        ),
        {
            "field": 9999999,
            "order": "EXISTING",
        },
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response_json["error"] == "ERROR_REQUEST_BODY_VALIDATION"
    assert response_json["detail"]["field"][0]["code"] == "does_not_exist"
    assert response_json["detail"]["order"][0]["code"] == "invalid_choice"

    response = api_client.patch(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_1.id},
        ),
        {"field": field_2.id},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response_json["error"] == "ERROR_FIELD_NOT_IN_TABLE"

    response = api_client.patch(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_1.id},
        ),
        {"field": link_row_field.id},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response_json["error"] == "ERROR_VIEW_GROUP_BY_FIELD_NOT_SUPPORTED"

    response = api_client.patch(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_3.id},
        ),
        {"field": group_by_1.field_id},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response_json["error"] == "ERROR_VIEW_GROUP_BY_FIELD_ALREADY_EXISTS"

    response = api_client.patch(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_1.id},
        ),
        {"field": field_1.id, "order": "ASC", "width": 150},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_200_OK
    assert ViewGroupBy.objects.all().count() == 3
    first = ViewGroupBy.objects.get(pk=group_by_1.id)
    assert first.field_id == field_1.id
    assert first.order == "ASC"
    assert response_json["id"] == first.id
    assert response_json["view"] == first.view_id
    assert response_json["field"] == field_1.id
    assert response_json["order"] == "ASC"
    assert response_json["width"] == 150

    response = api_client.patch(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_1.id},
        ),
        {"order": "DESC"},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_200_OK
    first = ViewGroupBy.objects.get(pk=group_by_1.id)
    assert first.field_id == field_1.id
    assert first.order == "DESC"
    assert response_json["id"] == first.id
    assert response_json["view"] == first.view_id
    assert response_json["field"] == field_1.id
    assert response_json["order"] == "DESC"

    response = api_client.patch(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_1.id},
        ),
        {"width": 120},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_200_OK
    first = ViewGroupBy.objects.get(pk=group_by_1.id)
    assert first.field_id == field_1.id
    assert first.order == "DESC"
    assert response_json["id"] == first.id
    assert response_json["view"] == first.view_id
    assert response_json["field"] == field_1.id
    assert response_json["order"] == "DESC"
    assert response_json["width"] == 120

    response = api_client.patch(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_1.id},
        ),
        {},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    response_json = response.json()
    assert response.status_code == HTTP_200_OK
    first = ViewGroupBy.objects.get(pk=group_by_1.id)
    assert first.field_id == field_1.id
    assert first.order == "DESC"
    assert response_json["id"] == first.id
    assert response_json["view"] == first.view_id
    assert response_json["field"] == field_1.id
    assert response_json["order"] == "DESC"
    assert response_json["width"] == 120


@pytest.mark.django_db
def test_delete_view_group_by(api_client, data_fixture):
    user, token = data_fixture.create_user_and_token()
    group_by_1 = data_fixture.create_view_group_by(user=user, order="DESC")
    group_by_2 = data_fixture.create_view_group_by()

    response = api_client.delete(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_2.id},
        ),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_400_BAD_REQUEST
    assert response.json()["error"] == "ERROR_USER_NOT_IN_GROUP"

    response = api_client.delete(
        reverse("api:database:views:group_by_item", kwargs={"view_group_by_id": 9999}),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_404_NOT_FOUND
    assert response.json()["error"] == "ERROR_VIEW_GROUP_BY_DOES_NOT_EXIST"

    response = api_client.delete(
        reverse(
            "api:database:views:group_by_item",
            kwargs={"view_group_by_id": group_by_1.id},
        ),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == 204
    assert ViewGroupBy.objects.all().count() == 1


@pytest.mark.django_db
def test_list_views_including_group_bys(api_client, data_fixture):
    user, token = data_fixture.create_user_and_token()
    table_1 = data_fixture.create_database_table(user=user)
    table_2 = data_fixture.create_database_table()
    field_1 = data_fixture.create_text_field(table=table_1)
    field_2 = data_fixture.create_text_field(table=table_1)
    field_3 = data_fixture.create_text_field(table=table_2)
    view_1 = data_fixture.create_grid_view(table=table_1, order=1)
    view_2 = data_fixture.create_grid_view(table=table_1, order=2)
    view_3 = data_fixture.create_grid_view(table=table_2, order=1)
    group_by_1 = data_fixture.create_view_group_by(view=view_1, field=field_1)
    group_by_2 = data_fixture.create_view_group_by(view=view_1, field=field_2)
    group_by_3 = data_fixture.create_view_group_by(view=view_2, field=field_1)
    data_fixture.create_view_group_by(view=view_3, field=field_3)

    response = api_client.get(
        "{}".format(
            reverse("api:database:views:list", kwargs={"table_id": table_1.id})
        ),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_200_OK
    response_json = response.json()
    assert len(response_json) == 2
    assert "group_bys" not in response_json[0]
    assert "group_bys" not in response_json[1]

    response = api_client.get(
        "{}?include=group_bys".format(
            reverse("api:database:views:list", kwargs={"table_id": table_1.id})
        ),
        HTTP_AUTHORIZATION=f"JWT {token}",
    )
    assert response.status_code == HTTP_200_OK
    response_json = response.json()

    assert len(response_json[0]["group_bys"]) == 2
    assert response_json[0]["group_bys"][0]["id"] == group_by_1.id
    assert response_json[0]["group_bys"][0]["view"] == view_1.id
    assert response_json[0]["group_bys"][0]["field"] == field_1.id
    assert response_json[0]["group_bys"][0]["order"] == group_by_1.order
    assert response_json[0]["group_bys"][0]["width"] == group_by_1.width
    assert response_json[0]["group_bys"][1]["id"] == group_by_2.id
    assert len(response_json[1]["group_bys"]) == 1
    assert response_json[1]["group_bys"][0]["id"] == group_by_3.id


@pytest.mark.django_db
def test_cant_get_view_group_by_when_view_trashed(api_client, data_fixture):
    user, token = data_fixture.create_user_and_token()
    table = data_fixture.create_database_table(user=user)
    view = data_fixture.create_form_view(table=table)

    ViewHandler().delete_view(user, view)

    url = reverse(
        "api:database:views:group_by_item", kwargs={"view_group_by_id": view.id}
    )
    response = api_client.get(url, format="json", HTTP_AUTHORIZATION=f"JWT {token}")
    assert response.status_code == HTTP_404_NOT_FOUND


@pytest.mark.django_db
def test_cant_update_view_group_by_when_view_trashed(api_client, data_fixture):
    user, token = data_fixture.create_user_and_token()
    table = data_fixture.create_database_table(user=user)
    view = data_fixture.create_grid_view(table=table)
    field = data_fixture.create_number_field(user, table=table)

    view_group_by = ViewHandler().create_group_by(user, view, field, "asc", 150)
    ViewHandler().delete_view(user, view)

    url = reverse(
        "api:database:views:group_by_item",
        kwargs={"view_group_by_id": view_group_by.id},
    )

    response = api_client.patch(
        url,
        {"order": "ASC"},
        format="json",
        HTTP_AUTHORIZATION=f"JWT {token}",
    )

    assert response.status_code == HTTP_404_NOT_FOUND


@pytest.mark.django_db
def test_cant_delete_view_group_by_when_view_trashed(api_client, data_fixture):
    user, token = data_fixture.create_user_and_token()
    table = data_fixture.create_database_table(user=user)
    view = data_fixture.create_grid_view(table=table)
    field = data_fixture.create_number_field(user, table=table)

    view_group_by = ViewHandler().create_group_by(user, view, field, "asc", 150)
    ViewHandler().delete_view(user, view)

    url = reverse(
        "api:database:views:group_by_item",
        kwargs={"view_group_by_id": view_group_by.id},
    )

    response = api_client.delete(url, format="json", HTTP_AUTHORIZATION=f"JWT {token}")

    assert response.status_code == HTTP_404_NOT_FOUND