import os
from datetime import datetime

import pystac
from pystac import (
    Asset,
    Catalog,
    Collection,
    Extent,
    Item,
    MediaType,
    SpatialExtent,
    TemporalExtent,
)
from pystac.extensions.label import (
    LabelClasses,
    LabelCount,
    LabelExtension,
    LabelOverview,
    LabelType,
)

from tests import test_data

TEST_LABEL_CATALOG = {
    "country-1": {
        "area-1-1": {
            "dsm": "area-1-1_dsm.tif",
            "ortho": "area-1-1_ortho.tif",
            "labels": "area-1-1_labels.geojson",
        },
        "area-1-2": {
            "dsm": "area-1-2_dsm.tif",
            "ortho": "area-1-2_ortho.tif",
            "labels": "area-1-2_labels.geojson",
        },
    },
    "country-2": {
        "area-2-1": {
            "dsm": "area-2-1_dsm.tif",
            "ortho": "area-2-1_ortho.tif",
            "labels": "area-2-1_labels.geojson",
        },
        "area-2-2": {
            "dsm": "area-2-2_dsm.tif",
            "ortho": "area-2-2_ortho.tif",
            "labels": "area-2-2_labels.geojson",
        },
    },
}

RANDOM_GEOM = {
    "type": "Polygon",
    "coordinates": [
        [
            [-2.5048828125, 3.8916575492899987],
            [-1.9610595703125, 3.8916575492899987],
            [-1.9610595703125, 4.275202171119132],
            [-2.5048828125, 4.275202171119132],
            [-2.5048828125, 3.8916575492899987],
        ]
    ],
}

RANDOM_BBOX = [
    RANDOM_GEOM["coordinates"][0][0][0],
    RANDOM_GEOM["coordinates"][0][0][1],
    RANDOM_GEOM["coordinates"][0][1][0],
    RANDOM_GEOM["coordinates"][0][1][1],
]

RANDOM_EXTENT = Extent(
    spatial=SpatialExtent.from_coordinates(RANDOM_GEOM["coordinates"]),
    temporal=TemporalExtent.from_now(),
)  # noqa: E126


class TestCases:
    __test__ = False

    @staticmethod
    def get_path(rel_path):
        return os.path.abspath(os.path.join(os.path.dirname(__file__), "..", rel_path))

    @staticmethod
    def all_test_catalogs():
        return [
            TestCases.planet_disaster(),
            TestCases.test_case_1(),
            TestCases.test_case_2(),
            TestCases.test_case_3(),
            TestCases.test_case_4(),
            TestCases.test_case_5(),
            TestCases.test_case_7(),
            TestCases.test_case_8(),
        ]

    @staticmethod
    def planet_disaster() -> Collection:
        return pystac.read_file(
            test_data.get_path("data-files/planet-disaster/collection.json")
        )

    @staticmethod
    def test_case_1():
        return Catalog.from_file(
            test_data.get_path("data-files/catalogs/test-case-1/catalog.json")
        )

    @staticmethod
    def test_case_2():
        return Catalog.from_file(
            test_data.get_path("data-files/catalogs/test-case-2/catalog.json")
        )

    @staticmethod
    def test_case_3():
        root_cat = Catalog(
            id="test3", description="test case 3 catalog", title="test case 3 title"
        )

        image_item = Item(
            id="imagery-item",
            geometry=RANDOM_GEOM,
            bbox=RANDOM_BBOX,
            datetime=datetime.utcnow(),
            properties={},
        )

        image_item.add_asset(
            "ortho", Asset(href="some/geotiff.tiff", media_type=MediaType.GEOTIFF)
        )

        overviews = [
            LabelOverview.create(
                "label",
                counts=[LabelCount.create("one", 1), LabelCount.create("two", 2)],
            )
        ]

        label_item = Item(
            id="label-items",
            geometry=RANDOM_GEOM,
            bbox=RANDOM_BBOX,
            datetime=datetime.utcnow(),
            properties={},
        )

        label_extension = LabelExtension.ext(label_item, add_if_missing=True)
        label_extension.apply(
            label_description="ML Labels",
            label_type=LabelType.VECTOR,
            label_properties=["label"],
            label_classes=[LabelClasses.create(classes=["one", "two"], name="label")],
            label_tasks=["classification"],
            label_methods=["manual"],
            label_overviews=overviews,
        )
        label_extension.add_source(image_item, assets=["ortho"])

        root_cat.add_item(image_item)
        root_cat.add_item(label_item)

        return root_cat

    @staticmethod
    def test_case_4():
        """Test case that is based on a local copy of the Tier 1 dataset from
        DrivenData's OpenCities AI Challenge.
        See: https://www.drivendata.org/competitions/60/building-segmentation-disaster-resilience
        """  # noqa: E501
        return Catalog.from_file(
            test_data.get_path("data-files/catalogs/test-case-4/catalog.json")
        )

    @staticmethod
    def test_case_5():
        """Based on a subset of https://cbers.stac.cloud/"""
        return Catalog.from_file(
            test_data.get_path("data-files/catalogs/test-case-5/catalog.json")
        )

    @staticmethod
    def test_case_6():
        """Contains local assets with relative hrefs"""
        return Catalog.from_file(
            test_data.get_path("data-files/catalogs/test-case-6/catalog.json")
        )

    @staticmethod
    def test_case_7():
        """Test case 4 as STAC version 0.8.1"""
        return Catalog.from_file(
            test_data.get_path("data-files/catalogs/label_catalog_0_8_1/catalog.json")
        )

    @staticmethod
    def test_case_8():
        """Planet disaster data example catalog, 1.0.0-beta.2"""
        return pystac.read_file(
            test_data.get_path(
                "data-files/catalogs/" "planet-example-1.0.0-beta.2/collection.json"
            )
        )