import tempfile
import unittest
import zipfile
from pathlib import Path

from src.kml_parser import parse_kml_file


KML_TEXT = """<?xml version="1.0" encoding="UTF-8"?>
<kml xmlns="http://www.opengis.net/kml/2.2">
  <Document>
    <Placemark>
      <name>Site boundary</name>
      <Polygon>
        <outerBoundaryIs>
          <LinearRing>
            <coordinates>
              70.2450,21.0020,0 70.2460,21.0020,0 70.2460,21.0030,0 70.2450,21.0030,0 70.2450,21.0020,0
            </coordinates>
          </LinearRing>
        </outerBoundaryIs>
      </Polygon>
    </Placemark>
  </Document>
</kml>
"""


class KmlParserTests(unittest.TestCase):
    def test_kml_polygon_creates_wgs84_selection(self):
        with tempfile.TemporaryDirectory() as tmp:
            path = Path(tmp) / "site.kml"
            path.write_text(KML_TEXT, encoding="utf-8")

            selection = parse_kml_file(path)

        self.assertEqual(selection.selection_type, "kml_boundary")
        self.assertEqual(selection.coordinate_mode, "wgs84")
        self.assertGreater(selection.area_sqm or 0, 0)
        self.assertAlmostEqual(selection.anchor_lat or 0, 21.0024, places=3)
        self.assertAlmostEqual(selection.anchor_lon or 0, 70.2454, places=3)
        self.assertIn("legal/cadastral", " ".join(selection.limitations))

    def test_kmz_polygon_creates_selection(self):
        with tempfile.TemporaryDirectory() as tmp:
            path = Path(tmp) / "site.kmz"
            with zipfile.ZipFile(path, "w") as archive:
                archive.writestr("doc.kml", KML_TEXT)

            selection = parse_kml_file(path)

        self.assertEqual(selection.selection_type, "kml_boundary")
        self.assertEqual(selection.source_files, ["site.kmz"])

    def test_kml_without_polygon_is_rejected(self):
        with tempfile.TemporaryDirectory() as tmp:
            path = Path(tmp) / "empty.kml"
            path.write_text("<kml><Document /></kml>", encoding="utf-8")

            with self.assertRaises(ValueError):
                parse_kml_file(path)


if __name__ == "__main__":
    unittest.main()