import datetime

import pytest
from fatcat_openapi_client import *
from fixtures import *


def test_webcapture(api):

    eg = quick_eg(api)
    r1 = ReleaseEntity(title="test webcapture release", ext_ids=ReleaseExtIds())
    r1edit = api.create_release(eg.editgroup_id, r1)

    # using datetime.datetime in this object construction to make exact object
    # comparisons easier. it is also possible to pass an ISO datetime string
    # (including timezone)
    now = datetime.datetime.now(datetime.timezone.utc)
    wc1 = WebcaptureEntity(
        original_url="http://example.site",
        timestamp=now,
        cdx=[
            WebcaptureCdxLine(
                surt="site,example,)/data/thing.tar.gz",
                timestamp=now,
                url="http://example.site/data/thing.tar.gz",
                mimetype="application/gzip",
                status_code=200,
                size=1234,
                sha1="455face3598611458efe1f072e58624790a67266",
                sha256="c7b49f3e84cd1b7cb0b0e3e9f632b7be7e21b4dc229df23331f880a8a7dfa75a",
            ),
            WebcaptureCdxLine(
                surt="site,example,)/README.md",
                timestamp=now,
                url="http://example.site/README.md",
                mimetype="text/markdown",
                status_code=200,
                size=4321,
                sha1="455face3598611458efe1f072e58624790a67266",
                sha256="429bcafa4d3d0072d5b2511e12c85c1aac1d304011d1c406da14707f7b9cd905",
            ),
        ],
        archive_urls=[
            FileUrl(rel="wayback", url="https://web.archive.org/web/"),
        ],
        release_ids=[r1edit.ident],
        extra=dict(c=1, b=2),
        edit_extra=dict(test_key="webcaptures rule"),
    )

    wc1edit = api.create_webcapture(eg.editgroup_id, wc1)
    assert wc1edit.extra == wc1.edit_extra
    api.accept_editgroup(eg.editgroup_id)
    wc2 = api.get_webcapture(wc1edit.ident)

    # get revision
    wc2_rev = api.get_webcapture_revision(wc1edit.revision)
    assert wc1edit.revision == wc2_rev.revision
    assert wc2.revision == wc2_rev.revision
    assert wc2.timestamp == wc2_rev.timestamp

    # check that fields match
    for i in range(len(wc1.archive_urls)):
        assert wc1.archive_urls[i].rel == wc2.archive_urls[i].rel
        assert wc1.archive_urls[i].url == wc2.archive_urls[i].url
    assert wc1.cdx[0] == wc2.cdx[0]
    assert wc1.cdx[0].size == wc2.cdx[0].size
    assert wc1.cdx == wc2.cdx
    assert wc1.release_ids == wc2.release_ids
    assert wc1.timestamp == wc2.timestamp
    assert wc1.original_url == wc2.original_url
    assert wc1.extra == wc2.extra

    # check release expansion
    r1 = api.get_release(r1edit.ident, expand="webcaptures")
    assert r1.webcaptures[0].cdx == wc1.cdx

    # get redirects (none)
    assert api.get_webcapture_redirects(wc2.ident) == []

    # delete
    eg = quick_eg(api)
    api.delete_webcapture(eg.editgroup_id, wc2.ident)
    api.accept_editgroup(eg.editgroup_id)
    wc2 = api.get_webcapture(wc2.ident)
    assert wc2.state == "deleted"


def test_webcapture_examples(api):
    wc3 = api.get_webcapture("aaaaaaaaaaaaa53xaaaaaaaaam")
    assert wc3.releases is None
    wc3 = api.get_webcapture("aaaaaaaaaaaaa53xaaaaaaaaam", expand="releases")

    assert wc3.cdx[0].surt == "org,asheesh)/"
    assert wc3.cdx[1].sha1 == "a637f1d27d9bcb237310ed29f19c07e1c8cf0aa5"
    assert wc3.archive_urls[1].rel == "warc"
    assert wc3.releases[0].ident
    assert wc3.releases[0].abstracts is None
    assert wc3.releases[0].refs is None


def test_bad_webcapture(api):

    eg = quick_eg(api)

    # good (as a template)
    good = WebcaptureEntity(
        original_url="http://example.site/123.jpg",
        timestamp="2012-01-02T03:04:05Z",
        cdx=[
            WebcaptureCdxLine(
                surt="site,example,)/123.jpg",
                url="http://example.site/123.jpg",
                sha1="455face3598611458efe1f072e58624790a67266",
                timestamp="2012-01-02T03:04:05Z",
            )
        ],
    )

    bad_list = [
        # uncomment to "test the test"
        # good,
        # CDX timestamp format
        WebcaptureEntity(
            original_url="http://example.site/123.jpg",
            timestamp="2012-01-02T03:04:05Z",
            cdx=[
                WebcaptureCdxLine(
                    surt="site,example,)/123.jpg",
                    url="http://example.site/123.jpg",
                    sha1="455face3598611458efe1f072e58624790a67266",
                    size=123,
                    timestamp="20120102030405",
                )
            ],
        ),
        # CDX timestamp format (int)
        WebcaptureEntity(
            original_url="http://example.site/123.jpg",
            timestamp="2012-01-02T03:04:05Z",
            cdx=[
                WebcaptureCdxLine(
                    surt="site,example,)/123.jpg",
                    url="http://example.site/123.jpg",
                    sha1="455face3598611458efe1f072e58624790a67266",
                    timestamp=20120102030405,
                )
            ],
        ),
        # negative size
        WebcaptureEntity(
            original_url="http://example.site/123.jpg",
            timestamp="2012-01-02T03:04:05Z",
            cdx=[
                WebcaptureCdxLine(
                    surt="site,example,)/123.jpg",
                    url="http://example.site/123.jpg",
                    sha1="455face3598611458efe1f072e58624790a67266",
                    size=-123,
                    timestamp="20120102030405",
                )
            ],
        ),
    ]

    api.create_webcapture(eg.editgroup_id, good)
    for b in bad_list:
        with pytest.raises(fatcat_openapi_client.rest.ApiException):
            api.create_webcapture(eg.editgroup_id, b)

    with pytest.raises(ValueError):
        # missing/empty CDX url
        WebcaptureEntity(
            original_url="http://example.site/123.jpg",
            timestamp="2012-01-02T03:04:05Z",
            cdx=[
                WebcaptureCdxLine(
                    # url="http://example.site/123.jpg",
                    surt="site,example,)/123.jpg",
                    sha1="455face3598611458efe1f072e58624790a67266",
                    timestamp="2012-01-02T03:04:05Z",
                )
            ],
        )

    with pytest.raises(ValueError):
        # missing/empty CDX timestamp
        WebcaptureEntity(
            original_url="http://example.site/123.jpg",
            timestamp="2012-01-02T03:04:05Z",
            cdx=[
                WebcaptureCdxLine(
                    url="http://example.site/123.jpg",
                    surt="site,example,)/123.jpg",
                    sha1="455face3598611458efe1f072e58624790a67266",
                    # timestamp="2012-01-02T03:04:05Z",
                )
            ],
        )