#
# This file is part of pretix (Community Edition).
#
# Copyright (C) 2014-2020  Raphael Michel and contributors
# Copyright (C) 2020-today pretix GmbH and contributors
#
# This program is free software: you can redistribute it and/or modify it under the terms of the GNU Affero General
# Public License as published by the Free Software Foundation in version 3 of the License.
#
# ADDITIONAL TERMS APPLY: Pursuant to Section 7 of the GNU Affero General Public License, additional terms are
# applicable granting you additional permissions and placing additional restrictions on your usage of this software.
# Please refer to the pretix LICENSE file to obtain the full terms applicable to this work. If you did not receive
# this file, see <https://pretix.eu/about/en/license>.
#
# This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied
# warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License for more
# details.
#
# You should have received a copy of the GNU Affero General Public License along with this program.  If not, see
# <https://www.gnu.org/licenses/>.
#
import zoneinfo
from datetime import datetime

import pytest
from django.core.files.base import ContentFile
from django_scopes import scopes_disabled
from tests.const import SAMPLE_PNG

TEST_ORGANIZER_RES = {
    "name": "Dummy",
    "slug": "dummy",
    "public_url": "http://example.com/dummy/",
    "plugins": ["pretix.plugins.banktransfer"],
}


@pytest.mark.django_db
def test_organizer_list(token_client, organizer):
    resp = token_client.get('/api/v1/organizers/')
    assert resp.status_code == 200
    assert TEST_ORGANIZER_RES in resp.data['results']


@pytest.mark.django_db
def test_organizer_detail(token_client, organizer):
    resp = token_client.get('/api/v1/organizers/{}/'.format(organizer.slug))
    assert resp.status_code == 200
    assert TEST_ORGANIZER_RES == resp.data


@pytest.mark.django_db
def test_organizer_patch(token_client, organizer):
    with scopes_disabled():
        # An event needs to exist for the backwards-compatibility mechanism in get_all_plugins to trigger
        event = organizer.events.create(
            name="Event", slug="e2", live=True,
            date_from=datetime(2020, 1, 10, 16, 0, tzinfo=zoneinfo.ZoneInfo("UTC")),
            date_to=datetime(2020, 1, 10, 17, 0, tzinfo=zoneinfo.ZoneInfo("UTC")),
        )
    resp = token_client.patch(
        '/api/v1/organizers/{}/'.format(organizer.slug),
        {
            'slug': 'willbeignored',
            'name': 'Willbeignored',
            'plugins': ['tests.testdummyorga', 'tests.testdummyhybrid']
        },
        format='json',
    )
    assert resp.status_code == 200
    assert resp.data['slug'] == 'dummy'
    assert resp.data['name'] == 'Dummy'
    assert set(resp.data['plugins']) == {'tests.testdummyorga', 'tests.testdummyhybrid'}

    resp = token_client.patch(
        '/api/v1/organizers/{}/'.format(organizer.slug),
        {
            'slug': 'willbeignored',
            'name': 'Willbeignored',
            'plugins': ['pretix.plugins.statistics']
        },
        format='json',
    )
    assert resp.status_code == 400
    assert resp.data == {
        "plugins": ["Plugin cannot be enabled on this level: 'pretix.plugins.statistics'."]
    }

    event.plugins = "tests.testdummyhybrid,tests.testdummy"
    event.save()
    resp = token_client.patch(
        '/api/v1/organizers/{}/'.format(organizer.slug),
        {
            'slug': 'willbeignored',
            'name': 'Willbeignored',
            'plugins': ['tests.testdummyorga']
        },
        format='json',
    )
    assert resp.status_code == 200

    event.refresh_from_db()
    assert event.plugins == "tests.testdummy"


@pytest.mark.django_db
def test_patch_settings(token_client, organizer):
    organizer.settings.event_list_type = 'week'
    resp = token_client.patch(
        '/api/v1/organizers/{}/settings/'.format(organizer.slug),
        {
            'event_list_type': 'list'
        },
        format='json'
    )
    assert resp.status_code == 200
    assert resp.data['event_list_type'] == "list"
    organizer.settings.flush()
    assert organizer.settings.event_list_type == 'list'

    resp = token_client.patch(
        '/api/v1/organizers/{}/settings/'.format(organizer.slug),
        {
            'event_list_type': None,
        },
        format='json'
    )
    assert resp.status_code == 200
    assert resp.data['event_list_type'] == "list"
    organizer.settings.flush()
    assert organizer.settings.event_list_type == 'list'

    resp = token_client.put(
        '/api/v1/organizers/{}/settings/'.format(organizer.slug),
        {
            'event_list_type': 'put-not-allowed'
        },
        format='json'
    )
    assert resp.status_code == 405

    resp = token_client.patch(
        '/api/v1/organizers/{}/settings/'.format(organizer.slug),
        {
            'primary_color': 'invalid-color'
        },
        format='json'
    )
    assert resp.status_code == 400

    resp = token_client.patch(
        '/api/v1/organizers/{}/settings/'.format(organizer.slug),
        {
            'primary_color': '#ff0000'
        },
        format='json'
    )
    assert resp.status_code == 200


@pytest.mark.django_db
def test_patch_organizer_settings_file(token_client, organizer):
    r = token_client.post(
        '/api/v1/upload',
        data={
            'media_type': 'image/png',
            'file': ContentFile(SAMPLE_PNG)
        },
        format='upload',
        HTTP_CONTENT_DISPOSITION='attachment; filename="file.png"',
    )
    assert r.status_code == 201
    file_id_png = r.data['id']

    r = token_client.post(
        '/api/v1/upload',
        data={
            'media_type': 'application/pdf',
            'file': ContentFile('invalid pdf content')
        },
        format='upload',
        HTTP_CONTENT_DISPOSITION='attachment; filename="file.pdf"',
    )
    assert r.status_code == 201
    file_id_pdf = r.data['id']

    resp = token_client.patch(
        '/api/v1/organizers/{}/settings/'.format(organizer.slug),
        {
            'organizer_logo_image': 'invalid'
        },
        format='json'
    )
    assert resp.status_code == 400
    assert resp.data == {
        'organizer_logo_image': ['The submitted file ID was not found.']
    }

    resp = token_client.patch(
        '/api/v1/organizers/{}/settings/'.format(organizer.slug),
        {
            'organizer_logo_image': file_id_pdf
        },
        format='json'
    )
    assert resp.status_code == 400
    assert resp.data == {
        'organizer_logo_image': ['The submitted file has a file type that is not allowed in this field.']
    }

    resp = token_client.patch(
        '/api/v1/organizers/{}/settings/'.format(organizer.slug,),
        {
            'organizer_logo_image': file_id_png
        },
        format='json'
    )
    assert resp.status_code == 200
    assert resp.data['organizer_logo_image'].startswith('http')

    resp = token_client.patch(
        '/api/v1/organizers/{}/settings/'.format(organizer.slug),
        {
            'organizer_logo_image': None
        },
        format='json'
    )
    assert resp.status_code == 200
    assert resp.data['organizer_logo_image'] is None