Merge branch 'main' into chore/tests

Author: alukach

.github/workflows/cicd.yaml

@@ -0,0 +1,11 @@
+on:
+  - push
+
+jobs:
+  lint:
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v3
+      - uses: pre-commit/[email protected]

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
   - repo: https://github.com/abravalheri/validate-pyproject
-    rev: v0.12.1
+    rev: v0.23
     hooks:
       - id: validate-pyproject
 

pyproject.toml

@@ -29,7 +29,7 @@ known_first_party = ["stac_auth_proxy"]
 profile = "black"
 
 [tool.ruff]
-ignore = ["E501"]
+ignore = ["E501", "D203", "D212"]
 select = ["D", "E", "F"]
 
 [build-system]

src/stac_auth_proxy/__init__.py

@@ -1,4 +1,5 @@
-"""STAC Auth Proxy package.
+"""
+STAC Auth Proxy package.
 
 This package contains the components for the STAC authentication and proxying system.
 It includes FastAPI routes for handling authentication, authorization, and interaction

src/stac_auth_proxy/__main__.py

@@ -1,3 +1,5 @@
+"""Main module for the STAC Auth Proxy."""
+
 import uvicorn
 from uvicorn.config import LOGGING_CONFIG
 

src/stac_auth_proxy/app.py

@@ -1,4 +1,5 @@
-"""STAC Auth Proxy API.
+"""
+STAC Auth Proxy API.
 
 This module defines the FastAPI application for the STAC Auth Proxy, which handles
 authentication, authorization, and proxying of requests to some internal STAC API.
@@ -16,6 +17,7 @@
 
 
 def create_app(settings: Optional[Settings] = None) -> FastAPI:
+    """FastAPI Application Factory."""
     settings = settings or Settings()
 
     app = FastAPI(openapi_url=None)

src/stac_auth_proxy/config.py

@@ -1,3 +1,5 @@
+"""Configuration for the STAC Auth Proxy."""
+
 from typing import Optional, TypeAlias
 
 from pydantic.networks import HttpUrl
@@ -7,7 +9,9 @@
 
 
 class Settings(BaseSettings):
-    upstream_url: HttpUrl = "https://earth-search.aws.element84.com/v1"
+    """Configuration settings for the STAC Auth Proxy."""
+
+    upstream_url: HttpUrl = HttpUrl(url="https://earth-search.aws.element84.com/v1")
     oidc_discovery_url: HttpUrl
 
     # Endpoints
@@ -26,4 +30,6 @@ class Settings(BaseSettings):
     openapi_spec_endpoint: Optional[str] = None
 
     class Config:
+        """Pydantic configuration."""
+
         env_prefix = "STAC_AUTH_PROXY_"

src/stac_auth_proxy/handlers.py

@@ -1,3 +1,5 @@
+"""Custom request handlers."""
+
 import logging
 from dataclasses import dataclass
 
@@ -12,14 +14,14 @@
 
 @dataclass
 class OpenApiSpecHandler:
+    """Handler for OpenAPI spec requests."""
+
     proxy: ReverseProxy
     oidc_config_url: str
     auth_scheme_name: str = "oidcAuth"
 
     async def dispatch(self, req: Request, res: Response):
-        """Proxy the OpenAPI spec from the upstream STAC API, updating it with OIDC security
-        requirements.
-        """
+        """Proxy the OpenAPI spec from the upstream STAC API, updating it with OIDC security requirements."""
         oidc_spec_response = await self.proxy.proxy_request(req)
         openapi_spec = oidc_spec_response.json()
 

src/stac_auth_proxy/middleware.py

@@ -1,11 +1,16 @@
+"""Custom middleware."""
+
 import time
 
 from fastapi import Request, Response
 from starlette.middleware.base import BaseHTTPMiddleware
 
 
 class AddProcessTimeHeaderMiddleware(BaseHTTPMiddleware):
+    """Middleware to add a header with the process time to the response."""
+
     async def dispatch(self, request: Request, call_next) -> Response:
+        """Add a header with the process time to the response."""
         start_time = time.perf_counter()
         response = await call_next(request)
         process_time = time.perf_counter() - start_time

src/stac_auth_proxy/proxy.py

@@ -1,3 +1,5 @@
+"""Tooling to manage the reverse proxying of requests to an upstream STAC API."""
+
 import logging
 import time
 from dataclasses import dataclass
@@ -13,16 +15,20 @@
 
 @dataclass
 class ReverseProxy:
+    """Reverse proxy functionality."""
+
     upstream: str
     client: httpx.AsyncClient = None
 
     def __post_init__(self):
+        """Initialize the HTTP client."""
         self.client = self.client or httpx.AsyncClient(
             base_url=self.upstream,
             timeout=httpx.Timeout(timeout=15.0),
         )
 
     async def proxy_request(self, request: Request, *, stream=False) -> httpx.Response:
+        """Proxy a request to the upstream STAC API."""
         headers = MutableHeaders(request.headers)
 
         # https://github.com/fastapi/fastapi/discussions/7382#discussioncomment-5136466