add support for raster mbtiles

Author: hambsch

server/guppy/endpoints/endpoints_rio_tiler.py

@@ -39,7 +39,10 @@ def get_tile_for_layer(layer_name: str, style: str, db: Session, z: int, x: int,
     t = time.time()
     file_path = validate_layer_and_get_file_path(db, layer_name)
 
-    result = get_tile(file_path, z, x, y, style, values, colors)
+    if file_path.endswith('.mbtiles'):
+        result = get_tile_from_mbtiles(file_path, z, x, y)
+    else:
+        result = get_tile(file_path, z, x, y, style, values, colors)
     log_cache_info(t)
     return result
 
@@ -124,6 +127,35 @@ def generate_colormap(min_val, max_val, value_points, colors):
     return final_colormap
 
 
+def get_tile_from_mbtiles(file_path: str, z: int, x: int, y: int) -> Response:
+    """
+    Args:
+        file_path: A string representing the path to the file.
+        z: An integer representing the zoom level.
+        x: An integer representing the x-coordinate of the tile.
+        y: An integer representing the y-coordinate of the tile.
+
+    Returns:
+        A Response object containing the rendered tile image in PNG format, or raises an HTTPException with a status code of 404 and a corresponding detail message.
+
+    """
+    y = (1 << z) - 1 - y
+    try:
+        uri = f'file:{file_path}?mode=ro'
+        with sqlite3.connect(uri, uri=True) as conn:
+            cursor = conn.cursor()
+            cursor.execute("SELECT tile_data FROM tiles WHERE zoom_level=? AND tile_column=? AND tile_row=?", (z, x, y))
+            tile = cursor.fetchone()
+            if tile:
+                return Response(bytes(tile[0]), media_type="image/png")
+            else:
+                return None
+    except Exception as e:
+        raise HTTPException(status_code=404, detail=str(e))
+    except TileOutsideBounds:
+        raise HTTPException(status_code=204, detail=f"Tile out of bounds {z} {x} {y}")
+
+
 @lru_cache(maxsize=128)
 def get_tile(file_path: str, z: int, x: int, y: int, style: str = None, values: str = None, colors: str = None) -> Response:
     """

server/guppy/routes/tiles_router.py

@@ -22,7 +22,7 @@ async def get_vector_tile(layer_name: str, z: int, x: int, y: int, db: Session =
 @router.get(r"/raster/{layer_name}/{z}/{x}/{y}.png", responses={200: {"content": {"image/png": {}}, "description": "Return an image.", }}, response_class=Response,
             description="Read COG and return a png tile")
 async def get_raster_tile(layer_name: str, z: int, x: int, y: int,
-                          style: str = Query(..., description=f"Style should be '<b>shader_rgba</b>', '<b>custom</b>' or one of {list(cmap.data.keys())} values. <br><br>"
+                          style: str = Query(default=None, description=f"Style should be '<b>shader_rgba</b>', '<b>custom</b>' or one of {list(cmap.data.keys())} values. <br><br>"
                                                               f"If '<b>custom</b>', extra parameters values and colors are needed like:<br> values=1.23,80.35,190.587&colors=255,0,0,255_0,255,0,255_0,0,255,255 <br>"
                                                               f"so values are comma seperated, and colors are r,g,b,a and _ seperated."),
                           values: str = None, colors: str = None,