ccproxy-api 0.1.0__py3-none-any.whl

ccproxy/api/dependencies.py

@@ -0,0 +1,140 @@
+"""Shared dependencies for CCProxy API Server."""
+
+from typing import Annotated
+
+from fastapi import Depends, Request
+from structlog import get_logger
+
+from ccproxy.auth.dependencies import AuthManagerDep, get_auth_manager
+from ccproxy.config.settings import Settings, get_settings
+from ccproxy.core.http import BaseProxyClient
+from ccproxy.observability import PrometheusMetrics, get_metrics
+from ccproxy.observability.storage.duckdb_simple import SimpleDuckDBStorage
+from ccproxy.services.claude_sdk_service import ClaudeSDKService
+from ccproxy.services.credentials.manager import CredentialsManager
+from ccproxy.services.proxy_service import ProxyService
+
+
+logger = get_logger(__name__)
+
+# Type aliases for dependency injection
+SettingsDep = Annotated[Settings, Depends(get_settings)]
+
+
+def get_claude_service(
+    auth_manager: AuthManagerDep,
+) -> ClaudeSDKService:
+    """Get Claude SDK service instance.
+
+    Args:
+        auth_manager: Authentication manager dependency
+
+    Returns:
+        Claude SDK service instance
+    """
+    logger.debug("Creating Claude SDK service instance")
+    # Get global metrics instance
+    metrics = get_metrics()
+
+    return ClaudeSDKService(
+        auth_manager=auth_manager,
+        metrics=metrics,
+    )
+
+
+def get_credentials_manager(
+    settings: SettingsDep,
+) -> CredentialsManager:
+    """Get credentials manager instance.
+
+    Args:
+        settings: Application settings dependency
+
+    Returns:
+        Credentials manager instance
+    """
+    logger.debug("Creating credentials manager instance")
+    return CredentialsManager(config=settings.auth)
+
+
+def get_proxy_service(
+    settings: SettingsDep,
+    credentials_manager: Annotated[
+        CredentialsManager, Depends(get_credentials_manager)
+    ],
+) -> ProxyService:
+    """Get proxy service instance.
+
+    Args:
+        settings: Application settings dependency
+        credentials_manager: Credentials manager dependency
+
+    Returns:
+        Proxy service instance
+    """
+    logger.debug("Creating proxy service instance")
+    # Create HTTP client for proxy
+    from ccproxy.core.http import HTTPXClient
+
+    http_client = HTTPXClient()
+    proxy_client = BaseProxyClient(http_client)
+
+    # Get global metrics instance
+    metrics = get_metrics()
+
+    return ProxyService(
+        proxy_client=proxy_client,
+        credentials_manager=credentials_manager,
+        settings=settings,
+        proxy_mode="full",
+        target_base_url=settings.reverse_proxy.target_url,
+        metrics=metrics,
+    )
+
+
+def get_observability_metrics() -> PrometheusMetrics:
+    """Get observability metrics instance.
+
+    Returns:
+        PrometheusMetrics instance
+    """
+    logger.debug("Getting observability metrics instance")
+    return get_metrics()
+
+
+async def get_log_storage(request: Request) -> SimpleDuckDBStorage | None:
+    """Get log storage from app state.
+
+    Args:
+        request: FastAPI request object
+
+    Returns:
+        SimpleDuckDBStorage instance if available, None otherwise
+    """
+    return getattr(request.app.state, "log_storage", None)
+
+
+async def get_duckdb_storage(request: Request) -> SimpleDuckDBStorage | None:
+    """Get DuckDB storage from app state (backward compatibility).
+
+    Args:
+        request: FastAPI request object
+
+    Returns:
+        SimpleDuckDBStorage instance if available, None otherwise
+    """
+    # Try new name first, then fall back to old name for backward compatibility
+    storage = getattr(request.app.state, "log_storage", None)
+    if storage is None:
+        storage = getattr(request.app.state, "duckdb_storage", None)
+    return storage
+
+
+# Type aliases for service dependencies
+ClaudeServiceDep = Annotated[ClaudeSDKService, Depends(get_claude_service)]
+ProxyServiceDep = Annotated[ProxyService, Depends(get_proxy_service)]
+ObservabilityMetricsDep = Annotated[
+    PrometheusMetrics, Depends(get_observability_metrics)
+]
+LogStorageDep = Annotated[SimpleDuckDBStorage | None, Depends(get_log_storage)]
+DuckDBStorageDep = Annotated[SimpleDuckDBStorage | None, Depends(get_duckdb_storage)]

ccproxy/api/middleware/__init__.py

@@ -0,0 +1,11 @@
+"""API middleware for CCProxy API Server."""
+
+from ccproxy.api.middleware.cors import get_cors_config, setup_cors_middleware
+from ccproxy.api.middleware.errors import setup_error_handlers
+
+
+__all__ = [
+    "setup_cors_middleware",
+    "get_cors_config",
+    "setup_error_handlers",
+]

ccproxy/api/middleware/cors.py

@@ -0,0 +1,55 @@
+"""CORS middleware for CCProxy API Server."""
+
+from typing import Any
+
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from structlog import get_logger
+
+from ccproxy.config.settings import Settings
+
+
+logger = get_logger(__name__)
+
+
+def setup_cors_middleware(app: FastAPI, settings: Settings) -> None:
+    """Setup CORS middleware for the FastAPI application.
+
+    Args:
+        app: FastAPI application instance
+        settings: Application settings containing CORS configuration
+    """
+    logger.debug("cors_middleware_setup_start")
+
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=settings.cors.origins,
+        allow_credentials=settings.cors.credentials,
+        allow_methods=settings.cors.methods,
+        allow_headers=settings.cors.headers,
+        allow_origin_regex=settings.cors.origin_regex,
+        expose_headers=settings.cors.expose_headers,
+        max_age=settings.cors.max_age,
+    )
+
+    logger.debug("cors_middleware_configured", origins=settings.cors.origins)
+
+
+def get_cors_config(settings: Settings) -> dict[str, Any]:
+    """Get CORS configuration dictionary.
+
+    Args:
+        settings: Application settings containing CORS configuration
+
+    Returns:
+        Dictionary containing CORS configuration
+    """
+    return {
+        "allow_origins": settings.cors.origins,
+        "allow_credentials": settings.cors.credentials,
+        "allow_methods": settings.cors.methods,
+        "allow_headers": settings.cors.headers,
+        "allow_origin_regex": settings.cors.origin_regex,
+        "expose_headers": settings.cors.expose_headers,
+        "max_age": settings.cors.max_age,
+    }