proxy.py

"""API methods to support Proxy endpoints."""

# Standard Python Libraries
from typing import Optional

# Third-Party Libraries
from fastapi import Request
from fastapi.responses import RedirectResponse, Response
import httpx


# Helper function to proxy requests
async def proxy_request(
    request: Request,
    target_url: str,
    path: Optional[str] = None,
    cookie_name: Optional[str] = None,
):
    """Proxy requests to the specified target URL with optional cookie handling."""
    headers = dict(request.headers)

    # Include specified cookie in the headers if present
    if cookie_name:
        cookies = request.cookies.get(cookie_name)
        if cookies:
            headers["Cookie"] = "{}={}".format(cookie_name, cookies)

    # Send the request to the target
    async with httpx.AsyncClient(timeout=httpx.Timeout(90.0)) as client:
        proxy_response = await client.request(
            method=request.method,
            url="{}/{}".format(target_url, path),
            headers=headers,
            params=request.query_params,
            content=await request.body(),
        )
    # Adjust response headers
    proxy_response_headers = dict(proxy_response.headers)
    for header in ["content-encoding", "transfer-encoding", "content-length"]:
        proxy_response_headers.pop(header, None)

    return Response(
        content=proxy_response.content,
        status_code=proxy_response.status_code,
        headers=proxy_response_headers,
    )


async def matomo_proxy_handler(
    request: Request,
    path: str,
    MATOMO_URL: str,
):
    """
    Handle Matomo-specific proxy logic.

    Includes public paths, font redirects, and authentication for private paths.
    """
    # Redirect font requests to CDN
    font_paths = {
        "/plugins/Morpheus/fonts/matomo.woff2": "https://cdn.jsdelivr.net/gh/matomo-org/matomo@5.2.1/plugins/Morpheus/fonts/matomo.woff2",
        "/plugins/Morpheus/fonts/matomo.woff": "https://cdn.jsdelivr.net/gh/matomo-org/matomo@5.2.1/plugins/Morpheus/fonts/matomo.woff",
        "/plugins/Morpheus/fonts/matomo.ttf": "https://cdn.jsdelivr.net/gh/matomo-org/matomo@5.2.1/plugins/Morpheus/fonts/matomo.ttf",
    }
    if path in font_paths:
        return RedirectResponse(url=font_paths[path])

    return await proxy_request(
        request=request,
        target_url=MATOMO_URL,
        path=path,
    )