Merge pull request #162 from quantum5/dnssec-rectify

feat(rectify): implement DNSSEC rectify API proxying

Author: rwxd

powerdns_api_proxy/proxy.py

@@ -416,6 +416,30 @@ async def zone_notification(server_id: str, zone_id: str, X_API_Key: str = Heade
     return JSONResponse(content=pdns_response.data, status_code=status_code)
 
 
+@router_pdns.put("/servers/{server_id}/zones/{zone_id}/rectify")
+async def zone_rectification(
+    response: Response, server_id: str, zone_id: str, X_API_Key: str = Header()
+):
+    """
+    Rectify the zone data.
+
+    <https://doc.powerdns.com/authoritative/http-api/zone.html#put--servers-server_id-zones-zone_id-rectify>
+    """
+    environment = get_environment_for_token(config, X_API_Key)
+    if not check_pdns_zone_allowed(environment, zone_id):
+        logger.info(f"Zone {zone_id} not allowed for environment {environment.name}")
+        raise ZoneNotAllowedException()
+    resp = await pdns.put(f"/api/v1/servers/{server_id}/zones/{zone_id}/rectify")
+    pdns_response = await handle_pdns_response(resp)
+    status_code = pdns_response.raise_for_error()
+
+    # PUT operations often return 204 No Content
+    if status_code == HTTPStatus.NO_CONTENT:
+        return Response(status_code=HTTPStatus.NO_CONTENT)
+
+    return JSONResponse(content=pdns_response.data, status_code=status_code)
+
+
 @router_pdns.get("/servers/{server_id}/search-data")
 async def search_data(
     server_id: str,

tests/unit/proxy_test.py

@@ -118,6 +118,7 @@ def test_api_post_missing_token(path, fixture_patch_dummy_config, fixture_patch_
 put_routes = [
     "/api/v1/servers/localhost/zones/test-zone.example.com.",
     "/api/v1/servers/localhost/zones/test-zone.example.com./notify",
+    "/api/v1/servers/localhost/zones/test-zone.example.com./rectify",
 ]