# Copyright (C) 2018-2020  The Software Heritage developers
# See the AUTHORS file at the top-level directory of this distribution
# License: GNU Affero General Public License version 3, or any later version
# See top-level LICENSE file for more information

import random
from urllib.parse import quote

from hypothesis import given

from swh.model.identifiers import CONTENT, DIRECTORY, RELEASE, REVISION, SNAPSHOT
from swh.model.model import Origin
from swh.web.common.identifiers import gen_swhid
from swh.web.common.utils import reverse
from swh.web.tests.django_asserts import assert_contains
from swh.web.tests.strategies import (
    content,
    directory,
    origin,
    release,
    revision,
    snapshot,
)
from swh.web.tests.utils import check_html_get_response


@given(content())
def test_content_id_browse(client, content):
    cnt_sha1_git = content["sha1_git"]
    swhid = gen_swhid(CONTENT, cnt_sha1_git)

    for swhid_ in (swhid, swhid.upper()):
        url = reverse("browse-swhid", url_args={"swhid": swhid_})

        query_string = "sha1_git:" + cnt_sha1_git
        content_browse_url = reverse(
            "browse-content", url_args={"query_string": query_string}
        )

        resp = check_html_get_response(client, url, status_code=302)
        assert resp["location"] == content_browse_url


@given(directory())
def test_directory_id_browse(client, directory):
    swhid = gen_swhid(DIRECTORY, directory)

    for swhid_ in (swhid, swhid.upper()):
        url = reverse("browse-swhid", url_args={"swhid": swhid_})

        directory_browse_url = reverse(
            "browse-directory", url_args={"sha1_git": directory}
        )

        resp = check_html_get_response(client, url, status_code=302)
        assert resp["location"] == directory_browse_url


@given(revision())
def test_revision_id_browse(client, revision):
    swhid = gen_swhid(REVISION, revision)

    for swhid_ in (swhid, swhid.upper()):
        url = reverse("browse-swhid", url_args={"swhid": swhid_})

        revision_browse_url = reverse(
            "browse-revision", url_args={"sha1_git": revision}
        )

        resp = check_html_get_response(client, url, status_code=302)
        assert resp["location"] == revision_browse_url

        query_params = {"origin_url": "https://github.com/user/repo"}
        url = reverse(
            "browse-swhid", url_args={"swhid": swhid_}, query_params=query_params
        )

        revision_browse_url = reverse(
            "browse-revision",
            url_args={"sha1_git": revision},
            query_params=query_params,
        )

        resp = check_html_get_response(client, url, status_code=302)
        assert resp["location"] == revision_browse_url


@given(release())
def test_release_id_browse(client, release):
    swhid = gen_swhid(RELEASE, release)

    for swhid_ in (swhid, swhid.upper()):
        url = reverse("browse-swhid", url_args={"swhid": swhid_})

        release_browse_url = reverse("browse-release", url_args={"sha1_git": release})

        resp = check_html_get_response(client, url, status_code=302)
        assert resp["location"] == release_browse_url

        query_params = {"origin_url": "https://github.com/user/repo"}

        url = reverse(
            "browse-swhid", url_args={"swhid": swhid_}, query_params=query_params
        )

        release_browse_url = reverse(
            "browse-release", url_args={"sha1_git": release}, query_params=query_params
        )

        resp = check_html_get_response(client, url, status_code=302)
        assert resp["location"] == release_browse_url


@given(snapshot())
def test_snapshot_id_browse(client, snapshot):
    swhid = gen_swhid(SNAPSHOT, snapshot)

    for swhid_ in (swhid, swhid.upper()):
        url = reverse("browse-swhid", url_args={"swhid": swhid_})

        snapshot_browse_url = reverse(
            "browse-snapshot", url_args={"snapshot_id": snapshot}
        )

        resp = check_html_get_response(client, url, status_code=302)
        assert resp["location"] == snapshot_browse_url

        query_params = {"origin_url": "https://github.com/user/repo"}

        url = reverse(
            "browse-swhid", url_args={"swhid": swhid_}, query_params=query_params
        )

        release_browse_url = reverse(
            "browse-snapshot",
            url_args={"snapshot_id": snapshot},
            query_params=query_params,
        )

        resp = check_html_get_response(client, url, status_code=302)
        assert resp["location"] == release_browse_url


@given(release())
def test_bad_id_browse(client, release):
    swhid = f"swh:1:foo:{release}"
    url = reverse("browse-swhid", url_args={"swhid": swhid})

    check_html_get_response(client, url, status_code=400)


@given(content())
def test_content_id_optional_parts_browse(client, archive_data, content):
    cnt_sha1_git = content["sha1_git"]
    origin_url = "https://github.com/user/repo"

    archive_data.origin_add([Origin(url=origin_url)])

    swhid = gen_swhid(
        CONTENT, cnt_sha1_git, metadata={"lines": "4-20", "origin": origin_url},
    )
    url = reverse("browse-swhid", url_args={"swhid": swhid})

    query_string = "sha1_git:" + cnt_sha1_git
    content_browse_url = reverse(
        "browse-content",
        url_args={"query_string": query_string},
        query_params={"origin_url": origin_url},
    )
    content_browse_url += "#L4-L20"

    resp = check_html_get_response(client, url, status_code=302)
    assert resp["location"] == content_browse_url


@given(release())
def test_origin_id_not_resolvable(client, release):
    swhid = "swh:1:ori:8068d0075010b590762c6cb5682ed53cb3c13deb"
    url = reverse("browse-swhid", url_args={"swhid": swhid})

    check_html_get_response(client, url, status_code=400)


@given(origin())
def test_legacy_swhid_browse(archive_data, client, origin):
    snapshot = archive_data.snapshot_get_latest(origin["url"])
    revision = archive_data.snapshot_get_head(snapshot)
    directory = archive_data.revision_get(revision)["directory"]
    directory_content = archive_data.directory_ls(directory)
    directory_file = random.choice(
        [e for e in directory_content if e["type"] == "file"]
    )
    legacy_swhid = gen_swhid(
        CONTENT,
        directory_file["checksums"]["sha1_git"],
        metadata={"origin": origin["url"]},
    )

    url = reverse("browse-swhid", url_args={"swhid": legacy_swhid})

    resp = check_html_get_response(client, url, status_code=302)
    resp = check_html_get_response(
        client, resp["location"], status_code=200, template_used="browse/content.html"
    )

    swhid = gen_swhid(
        CONTENT,
        directory_file["checksums"]["sha1_git"],
        metadata={
            "origin": origin["url"],
            "visit": gen_swhid(SNAPSHOT, snapshot["id"]),
            "anchor": gen_swhid(REVISION, revision),
        },
    )

    assert_contains(resp, swhid)

    # also check legacy SWHID URL with trailing slash
    url = reverse("browse-swhid-legacy", url_args={"swhid": swhid})

    resp = check_html_get_response(client, url, status_code=302)
    resp = check_html_get_response(
        client, resp["location"], status_code=200, template_used="browse/content.html"
    )

    assert_contains(resp, swhid)


@given(directory())
def test_browse_swhid_special_characters_escaping(client, archive_data, directory):
    origin = "http://example.org/?project=abc;"
    archive_data.origin_add([Origin(url=origin)])
    origin_swhid_escaped = quote(origin, safe="/?:@&")
    origin_swhid_url_escaped = quote(origin, safe="/:@;")
    swhid = gen_swhid(DIRECTORY, directory, metadata={"origin": origin_swhid_escaped})
    url = reverse("browse-swhid", url_args={"swhid": swhid})

    resp = check_html_get_response(client, url, status_code=302)
    assert origin_swhid_url_escaped in resp["location"]