fix: add check for correct type of geojson input

geojson_aoi/parser.py

@@ -3,10 +3,14 @@
 import json
 import logging
 from pathlib import Path
-from typing import Any, Literal
-
-AllowedInputTypes = Literal[
-    "Polygon", "MultiPolygon", "Feature", "FeatureCollection", "GeometryCollection"
+from typing import Any
+
+AllowedInputTypes = [
+    "Polygon",
+    "MultiPolygon",
+    "Feature",
+    "FeatureCollection",
+    "GeometryCollection",
 ]
 Coordinate = float | int
 PointGeom = tuple[Coordinate, Coordinate]
@@ -113,8 +117,8 @@ def split_multigeom(
 
 
 def _ensure_right_hand_rule(
-    coordinates: list[list[PointGeom]],
-) -> list[list[PointGeom]]:
+    coordinates: PolygonGeom,
+) -> PolygonGeom:
     """Ensure the outer ring follows the right-hand rule (clockwise)."""
 
     def is_clockwise(ring: list[PointGeom]) -> bool:
@@ -136,7 +140,7 @@ def is_clockwise(ring: list[PointGeom]) -> bool:
     return coordinates
 
 
-def _create_convex_hull(polygons: list[list[PointGeom]]) -> list[PointGeom]:
+def _create_convex_hull(polygons: PolygonGeom) -> list[PointGeom]:
     """Create a convex hull from a list of polygons."""
     from itertools import chain
 
@@ -314,10 +318,17 @@ def parse_aoi(
     else:
         raise ValueError("GeoJSON input must be a valid dict, str, or bytes")
 
+    print(geojson_parsed)
+    print("type" in geojson_parsed)
+
     # Throw error if no data
-    if geojson_parsed is None or geojson_parsed == {}:
+    if geojson_parsed is None or geojson_parsed == {} or "type" not in geojson_parsed:
         raise ValueError("Provided GeoJSON is empty")
 
+    # Throw error if wrong geometry type
+    if geojson_parsed["type"] not in AllowedInputTypes:
+        raise ValueError(f"The GeoJSON type must be one of: {AllowedInputTypes}")
+
     # Convert to FeatureCollection
     featcol = geojson_to_featcol(geojson_parsed)
     if not featcol.get("features", []):

tests/test_parser.py

@@ -183,6 +183,9 @@ def test_invalid_input():
     with pytest.raises(ValueError, match="Provided GeoJSON is empty"):
         parse_aoi("{}")
 
+    with pytest.raises(ValueError, match="The GeoJSON type must be one of:"):
+        parse_aoi({"type": "Point"})
+
 
 def test_file_input(tmp_path):
     """GeoJSON file input for parse_aoi function."""