import unittest
from io import BytesIO
import os
from pmtiles.writer import Writer
from pmtiles.reader import Reader, MemorySource
from pmtiles.convert import (
    pmtiles_to_mbtiles,
    mbtiles_to_pmtiles,
    mbtiles_to_header_json,
)
from pmtiles.tile import TileType, Compression


class TestConvert(unittest.TestCase):
    def tearDown(self):
        try:
            os.remove("test_tmp.pmtiles")
        except:
            pass
        try:
            os.remove("test_tmp.mbtiles")
        except:
            pass
        try:
            os.remove("test_tmp_2.mbtiles")
        except:
            pass

    def test_roundtrip(self):
        pass
        # with open("test_tmp.pmtiles", "wb") as f:
        #     writer = Writer(f, 7)
        #     writer.write_tile(1, 0, 0, b"0")
        #     writer.write_tile(1, 0, 1, b"1")
        #     writer.write_tile(1, 1, 0, b"2")
        #     writer.write_tile(1, 1, 1, b"3")
        #     writer.write_tile(2, 0, 0, b"4")
        #     writer.write_tile(3, 0, 0, b"5")
        #     writer.write_tile(2, 0, 1, b"6")
        #     writer.write_tile(3, 0, 2, b"7")
        #     writer.finalize({"key": "value"})

        # pmtiles_to_mbtiles("test_tmp.pmtiles", "test_tmp.mbtiles", False)
        # mbtiles_to_pmtiles("test_tmp.mbtiles", "test_tmp_2.pmtiles", 3, False)

    def test_mbtiles_header(self):
        header, json_metadata = mbtiles_to_header_json(
            {
                "name": "test_name",
                "format": "pbf",
                "bounds": "-180.0,-85,180,85",
                "center": "-122.1906,37.7599,11",
                "minzoom": "1",
                "maxzoom": "2",
                "attribution": "<div>abc</div>",
                "compression": "gzip",
                "json": '{"vector_layers":[{"abc":123}],"tilestats":{"def":456}}',
            }
        )
        self.assertEqual(header["min_lon_e7"], -180 * 10000000)
        self.assertTrue(isinstance(header["min_lon_e7"], int))
        self.assertEqual(header["min_lat_e7"], -85 * 10000000)
        self.assertEqual(header["max_lon_e7"], 180 * 10000000)
        self.assertEqual(header["max_lat_e7"], 85 * 10000000)
        self.assertEqual(header["tile_type"], TileType.MVT)
        self.assertEqual(header["center_lon_e7"], -122.1906 * 10000000)
        self.assertEqual(header["center_lat_e7"], 37.7599 * 10000000)
        self.assertEqual(header["center_zoom"], 11)
        self.assertEqual(header["min_zoom"], 1)
        self.assertEqual(header["max_zoom"], 2)
        self.assertEqual(header["tile_compression"], Compression.GZIP)

        self.assertTrue("name" in json_metadata)
        self.assertTrue("format" in json_metadata)
        self.assertTrue("compression" in json_metadata)
        self.assertFalse("center" in json_metadata)
        self.assertFalse("bounds" in json_metadata)
        self.assertFalse("bounds" in json_metadata)