stac_auth_proxy.middleware

Custom middleware.

OpenApiMiddleware dataclass

Bases: JsonResponseMiddleware

Middleware to add the OpenAPI spec to the response.

Parameters:

Name	Type	Description	Default
app	Callable[list, Awaitable[None]]		required
openapi_spec_path	str		required
oidc_discovery_url	str		required
private_endpoints	dict[str, Sequence[Literal[GET, POST, PUT, DELETE, PATCH]]]		required
public_endpoints	dict[str, Sequence[Literal[GET, POST, PUT, DELETE, PATCH]]]		required
default_public	bool		required
root_path	str		''
auth_scheme_name	str		'oidcAuth'
auth_scheme_override	dict | None		None
json_content_type_expr	str		'application/(vnd\\.oai\\.openapi\\+json?|json)'

Source code in src/stac_auth_proxy/middleware/UpdateOpenApiMiddleware.py

@dataclass(frozen=True)
class OpenApiMiddleware(JsonResponseMiddleware):
    """Middleware to add the OpenAPI spec to the response."""

    app: ASGIApp
    openapi_spec_path: str
    oidc_discovery_url: str
    private_endpoints: EndpointMethods
    public_endpoints: EndpointMethods
    default_public: bool
    root_path: str = ""
    auth_scheme_name: str = "oidcAuth"
    auth_scheme_override: Optional[dict] = None

    json_content_type_expr: str = r"application/(vnd\.oai\.openapi\+json?|json)"

    def should_transform_response(self, request: Request, scope: Scope) -> bool:
        """Only transform responses for the OpenAPI spec path."""
        return (
            all(
                re.match(expr, val)
                for expr, val in [
                    (self.openapi_spec_path, request.url.path),
                    (
                        self.json_content_type_expr,
                        Headers(scope=scope).get("content-type", ""),
                    ),
                ]
            )
            and 200 <= scope["status"] < 300
        )

    def transform_json(self, data: dict[str, Any], request: Request) -> dict[str, Any]:
        """Augment the OpenAPI spec with auth information."""
        # Add servers field with root path if root_path is set
        if self.root_path:
            data["servers"] = [{"url": self.root_path}]

        # Add security scheme
        components = data.setdefault("components", {})
        securitySchemes = components.setdefault("securitySchemes", {})
        securitySchemes[self.auth_scheme_name] = self.auth_scheme_override or {
            "type": "openIdConnect",
            "openIdConnectUrl": self.oidc_discovery_url,
        }

        # Add security to private endpoints
        for path, method_config in data["paths"].items():
            for method, config in method_config.items():
                if method == "options":
                    # OPTIONS requests are not authenticated, https://fetch.spec.whatwg.org/#cors-protocol-and-credentials
                    continue
                match = find_match(
                    path,
                    method,
                    self.private_endpoints,
                    self.public_endpoints,
                    self.default_public,
                )
                if match.is_private:
                    config.setdefault("security", []).append(
                        {self.auth_scheme_name: match.required_scopes}
                    )
        return data

should_transform_response(request: Request, scope: Scope) -> bool

Only transform responses for the OpenAPI spec path.

Source code in src/stac_auth_proxy/middleware/UpdateOpenApiMiddleware.py

def should_transform_response(self, request: Request, scope: Scope) -> bool:
    """Only transform responses for the OpenAPI spec path."""
    return (
        all(
            re.match(expr, val)
            for expr, val in [
                (self.openapi_spec_path, request.url.path),
                (
                    self.json_content_type_expr,
                    Headers(scope=scope).get("content-type", ""),
                ),
            ]
        )
        and 200 <= scope["status"] < 300
    )

transform_json(data: dict[str, Any], request: Request) -> dict[str, Any]

Augment the OpenAPI spec with auth information.

Source code in src/stac_auth_proxy/middleware/UpdateOpenApiMiddleware.py

def transform_json(self, data: dict[str, Any], request: Request) -> dict[str, Any]:
    """Augment the OpenAPI spec with auth information."""
    # Add servers field with root path if root_path is set
    if self.root_path:
        data["servers"] = [{"url": self.root_path}]

    # Add security scheme
    components = data.setdefault("components", {})
    securitySchemes = components.setdefault("securitySchemes", {})
    securitySchemes[self.auth_scheme_name] = self.auth_scheme_override or {
        "type": "openIdConnect",
        "openIdConnectUrl": self.oidc_discovery_url,
    }

    # Add security to private endpoints
    for path, method_config in data["paths"].items():
        for method, config in method_config.items():
            if method == "options":
                # OPTIONS requests are not authenticated, https://fetch.spec.whatwg.org/#cors-protocol-and-credentials
                continue
            match = find_match(
                path,
                method,
                self.private_endpoints,
                self.public_endpoints,
                self.default_public,
            )
            if match.is_private:
                config.setdefault("security", []).append(
                    {self.auth_scheme_name: match.required_scopes}
                )
    return data