ccproxy-api 0.1.0__py3-none-any.whl

ccproxy/config/validators.py

@@ -0,0 +1,231 @@
+"""Configuration validation utilities."""
+
+import re
+from pathlib import Path
+from typing import Any, Optional, Union
+from urllib.parse import urlparse
+
+
+class ConfigValidationError(Exception):
+    """Configuration validation error."""
+
+    pass
+
+
+def validate_host(host: str) -> str:
+    """Validate host address.
+
+    Args:
+        host: Host address to validate
+
+    Returns:
+        The validated host address
+
+    Raises:
+        ConfigValidationError: If host is invalid
+    """
+    if not host:
+        raise ConfigValidationError("Host cannot be empty")
+
+    # Allow localhost, IP addresses, and domain names
+    if host in ["localhost", "0.0.0.0", "127.0.0.1"]:
+        return host
+
+    # Basic IP address validation
+    if re.match(r"^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$", host):
+        parts = host.split(".")
+        if all(0 <= int(part) <= 255 for part in parts):
+            return host
+        raise ConfigValidationError(f"Invalid IP address: {host}")
+
+    # Basic domain name validation
+    if re.match(r"^[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$", host):
+        return host
+
+    return host  # Allow other formats for flexibility
+
+
+def validate_port(port: int | str) -> int:
+    """Validate port number.
+
+    Args:
+        port: Port number to validate
+
+    Returns:
+        The validated port number
+
+    Raises:
+        ConfigValidationError: If port is invalid
+    """
+    if isinstance(port, str):
+        try:
+            port = int(port)
+        except ValueError as e:
+            raise ConfigValidationError(f"Port must be a valid integer: {port}") from e
+
+    if not isinstance(port, int):
+        raise ConfigValidationError(f"Port must be an integer: {port}")
+
+    if port < 1 or port > 65535:
+        raise ConfigValidationError(f"Port must be between 1 and 65535: {port}")
+
+    return port
+
+
+def validate_url(url: str) -> str:
+    """Validate URL format.
+
+    Args:
+        url: URL to validate
+
+    Returns:
+        The validated URL
+
+    Raises:
+        ConfigValidationError: If URL is invalid
+    """
+    if not url:
+        raise ConfigValidationError("URL cannot be empty")
+
+    try:
+        result = urlparse(url)
+        if not result.scheme or not result.netloc:
+            raise ConfigValidationError(f"Invalid URL format: {url}")
+    except Exception as e:
+        raise ConfigValidationError(f"Invalid URL: {url}") from e
+
+    return url
+
+
+def validate_path(path: str | Path) -> Path:
+    """Validate file path.
+
+    Args:
+        path: Path to validate
+
+    Returns:
+        The validated Path object
+
+    Raises:
+        ConfigValidationError: If path is invalid
+    """
+    if isinstance(path, str):
+        path = Path(path)
+
+    if not isinstance(path, Path):
+        raise ConfigValidationError(f"Path must be a string or Path object: {path}")
+
+    return path
+
+
+def validate_log_level(level: str) -> str:
+    """Validate log level.
+
+    Args:
+        level: Log level to validate
+
+    Returns:
+        The validated log level
+
+    Raises:
+        ConfigValidationError: If log level is invalid
+    """
+    valid_levels = ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]
+    level = level.upper()
+
+    if level not in valid_levels:
+        raise ConfigValidationError(
+            f"Invalid log level: {level}. Must be one of: {valid_levels}"
+        )
+
+    return level
+
+
+def validate_cors_origins(origins: list[str]) -> list[str]:
+    """Validate CORS origins.
+
+    Args:
+        origins: List of origin URLs to validate
+
+    Returns:
+        The validated list of origins
+
+    Raises:
+        ConfigValidationError: If any origin is invalid
+    """
+    if not isinstance(origins, list):
+        raise ConfigValidationError("CORS origins must be a list")
+
+    validated_origins = []
+    for origin in origins:
+        if origin == "*":
+            validated_origins.append(origin)
+        else:
+            validated_origins.append(validate_url(origin))
+
+    return validated_origins
+
+
+def validate_timeout(timeout: int | float) -> int | float:
+    """Validate timeout value.
+
+    Args:
+        timeout: Timeout value to validate
+
+    Returns:
+        The validated timeout value
+
+    Raises:
+        ConfigValidationError: If timeout is invalid
+    """
+    if not isinstance(timeout, int | float):
+        raise ConfigValidationError(f"Timeout must be a number: {timeout}")
+
+    if timeout <= 0:
+        raise ConfigValidationError(f"Timeout must be positive: {timeout}")
+
+    return timeout
+
+
+def validate_config_dict(config: dict[str, Any]) -> dict[str, Any]:
+    """Validate configuration dictionary.
+
+    Args:
+        config: Configuration dictionary to validate
+
+    Returns:
+        The validated configuration dictionary
+
+    Raises:
+        ConfigValidationError: If configuration is invalid
+    """
+    if not isinstance(config, dict):
+        raise ConfigValidationError("Configuration must be a dictionary")
+
+    validated_config: dict[str, Any] = {}
+
+    # Validate specific fields if present
+    if "host" in config:
+        validated_config["host"] = validate_host(config["host"])
+
+    if "port" in config:
+        validated_config["port"] = validate_port(config["port"])
+
+    if "target_url" in config:
+        validated_config["target_url"] = validate_url(config["target_url"])
+
+    if "log_level" in config:
+        validated_config["log_level"] = validate_log_level(config["log_level"])
+
+    if "cors_origins" in config:
+        validated_config["cors_origins"] = validate_cors_origins(config["cors_origins"])
+
+    if "timeout" in config:
+        validated_config["timeout"] = validate_timeout(config["timeout"])
+
+    # Copy other fields without validation
+    for key, value in config.items():
+        if key not in validated_config:
+            validated_config[key] = value
+
+    return validated_config

ccproxy/core/__init__.py

@@ -0,0 +1,274 @@
+"""Core abstractions for the CCProxy API."""
+
+from ccproxy.core.async_utils import (
+    async_cache_result,
+    async_timer,
+    gather_with_concurrency,
+    get_package_dir,
+    get_root_package_name,
+    patched_typing,
+    retry_async,
+    run_in_executor,
+    safe_await,
+    wait_for_condition,
+)
+from ccproxy.core.constants import (
+    ANTHROPIC_API_BASE_PATH,
+    AUTH_HEADER,
+    CHAT_COMPLETIONS_ENDPOINT,
+    CONFIG_FILE_NAMES,
+    CONTENT_TYPE_HEADER,
+    CONTENT_TYPE_JSON,
+    CONTENT_TYPE_STREAM,
+    CONTENT_TYPE_TEXT,
+    DEFAULT_DOCKER_IMAGE,
+    DEFAULT_DOCKER_TIMEOUT,
+    DEFAULT_MAX_TOKENS,
+    DEFAULT_MODEL,
+    DEFAULT_RATE_LIMIT,
+    DEFAULT_STREAM,
+    DEFAULT_TEMPERATURE,
+    DEFAULT_TIMEOUT,
+    DEFAULT_TOP_P,
+    EMAIL_PATTERN,
+    ENV_PREFIX,
+    ERROR_MSG_INTERNAL_ERROR,
+    ERROR_MSG_INVALID_REQUEST,
+    ERROR_MSG_INVALID_TOKEN,
+    ERROR_MSG_MODEL_NOT_FOUND,
+    ERROR_MSG_RATE_LIMIT_EXCEEDED,
+    JSON_EXTENSIONS,
+    LOG_LEVELS,
+    MAX_MESSAGE_LENGTH,
+    MAX_PROMPT_LENGTH,
+    MAX_TOOL_CALLS,
+    MESSAGES_ENDPOINT,
+    MODELS_ENDPOINT,
+    OPENAI_API_BASE_PATH,
+    REQUEST_ID_HEADER,
+    STATUS_BAD_GATEWAY,
+    STATUS_BAD_REQUEST,
+    STATUS_CREATED,
+    STATUS_FORBIDDEN,
+    STATUS_INTERNAL_ERROR,
+    STATUS_NOT_FOUND,
+    STATUS_OK,
+    STATUS_RATE_LIMITED,
+    STATUS_SERVICE_UNAVAILABLE,
+    STATUS_UNAUTHORIZED,
+    STREAM_EVENT_CONTENT_BLOCK_DELTA,
+    STREAM_EVENT_CONTENT_BLOCK_START,
+    STREAM_EVENT_CONTENT_BLOCK_STOP,
+    STREAM_EVENT_MESSAGE_DELTA,
+    STREAM_EVENT_MESSAGE_START,
+    STREAM_EVENT_MESSAGE_STOP,
+    TOML_EXTENSIONS,
+    URL_PATTERN,
+    UUID_PATTERN,
+    YAML_EXTENSIONS,
+)
+from ccproxy.core.errors import (
+    MiddlewareError,
+    ProxyAuthenticationError,
+    ProxyConnectionError,
+    ProxyError,
+    ProxyTimeoutError,
+    TransformationError,
+)
+from ccproxy.core.http import (
+    BaseProxyClient,
+    HTTPClient,
+    HTTPConnectionError,
+    HTTPError,
+    HTTPTimeoutError,
+    HTTPXClient,
+)
+from ccproxy.core.interfaces import (
+    APIAdapter,
+    MetricExporter,
+    StreamTransformer,
+    TokenStorage,
+)
+from ccproxy.core.interfaces import (
+    RequestTransformer as IRequestTransformer,
+)
+from ccproxy.core.interfaces import (
+    ResponseTransformer as IResponseTransformer,
+)
+from ccproxy.core.interfaces import (
+    TransformerProtocol as ITransformerProtocol,
+)
+from ccproxy.core.middleware import (
+    BaseMiddleware,
+    CompositeMiddleware,
+    MiddlewareChain,
+    MiddlewareProtocol,
+    NextMiddleware,
+)
+from ccproxy.core.proxy import (
+    BaseProxy,
+    HTTPProxy,
+    ProxyProtocol,
+    WebSocketProxy,
+)
+from ccproxy.core.transformers import (
+    BaseTransformer,
+    ChainedTransformer,
+    RequestTransformer,
+    ResponseTransformer,
+    TransformerProtocol,
+)
+from ccproxy.core.types import (
+    MiddlewareConfig,
+    ProxyConfig,
+    ProxyMethod,
+    ProxyRequest,
+    ProxyResponse,
+    TransformContext,
+)
+from ccproxy.core.types import (
+    ProxyProtocol as ProxyProtocolEnum,
+)
+from ccproxy.core.validators import (
+    ValidationError,
+    validate_choice,
+    validate_dict,
+    validate_email,
+    validate_list,
+    validate_non_empty_string,
+    validate_path,
+    validate_port,
+    validate_range,
+    validate_timeout,
+    validate_url,
+    validate_uuid,
+)
+
+
+__all__ = [
+    # Proxy abstractions
+    "BaseProxy",
+    "HTTPProxy",
+    "WebSocketProxy",
+    "ProxyProtocol",
+    # HTTP client abstractions
+    "HTTPClient",
+    "BaseProxyClient",
+    "HTTPError",
+    "HTTPTimeoutError",
+    "HTTPConnectionError",
+    "HTTPXClient",
+    # Interface abstractions
+    "APIAdapter",
+    "MetricExporter",
+    "IRequestTransformer",
+    "IResponseTransformer",
+    "StreamTransformer",
+    "TokenStorage",
+    "ITransformerProtocol",
+    # Transformer abstractions
+    "BaseTransformer",
+    "RequestTransformer",
+    "ResponseTransformer",
+    "TransformerProtocol",
+    "ChainedTransformer",
+    # Middleware abstractions
+    "BaseMiddleware",
+    "MiddlewareChain",
+    "MiddlewareProtocol",
+    "CompositeMiddleware",
+    "NextMiddleware",
+    # Error types
+    "ProxyError",
+    "TransformationError",
+    "MiddlewareError",
+    "ProxyConnectionError",
+    "ProxyTimeoutError",
+    "ProxyAuthenticationError",
+    "ValidationError",
+    # Type definitions
+    "ProxyRequest",
+    "ProxyResponse",
+    "TransformContext",
+    "ProxyMethod",
+    "ProxyProtocolEnum",
+    "ProxyConfig",
+    "MiddlewareConfig",
+    # Async utilities
+    "async_cache_result",
+    "async_timer",
+    "gather_with_concurrency",
+    "get_package_dir",
+    "get_root_package_name",
+    "patched_typing",
+    "retry_async",
+    "run_in_executor",
+    "safe_await",
+    "wait_for_condition",
+    # Constants
+    "ANTHROPIC_API_BASE_PATH",
+    "AUTH_HEADER",
+    "CHAT_COMPLETIONS_ENDPOINT",
+    "CONFIG_FILE_NAMES",
+    "CONTENT_TYPE_HEADER",
+    "CONTENT_TYPE_JSON",
+    "CONTENT_TYPE_STREAM",
+    "CONTENT_TYPE_TEXT",
+    "DEFAULT_DOCKER_IMAGE",
+    "DEFAULT_DOCKER_TIMEOUT",
+    "DEFAULT_MAX_TOKENS",
+    "DEFAULT_MODEL",
+    "DEFAULT_RATE_LIMIT",
+    "DEFAULT_STREAM",
+    "DEFAULT_TEMPERATURE",
+    "DEFAULT_TIMEOUT",
+    "DEFAULT_TOP_P",
+    "EMAIL_PATTERN",
+    "ENV_PREFIX",
+    "ERROR_MSG_INTERNAL_ERROR",
+    "ERROR_MSG_INVALID_REQUEST",
+    "ERROR_MSG_INVALID_TOKEN",
+    "ERROR_MSG_MODEL_NOT_FOUND",
+    "ERROR_MSG_RATE_LIMIT_EXCEEDED",
+    "JSON_EXTENSIONS",
+    "LOG_LEVELS",
+    "MAX_MESSAGE_LENGTH",
+    "MAX_PROMPT_LENGTH",
+    "MAX_TOOL_CALLS",
+    "MESSAGES_ENDPOINT",
+    "MODELS_ENDPOINT",
+    "OPENAI_API_BASE_PATH",
+    "REQUEST_ID_HEADER",
+    "STATUS_BAD_GATEWAY",
+    "STATUS_BAD_REQUEST",
+    "STATUS_CREATED",
+    "STATUS_FORBIDDEN",
+    "STATUS_INTERNAL_ERROR",
+    "STATUS_NOT_FOUND",
+    "STATUS_OK",
+    "STATUS_RATE_LIMITED",
+    "STATUS_SERVICE_UNAVAILABLE",
+    "STATUS_UNAUTHORIZED",
+    "STREAM_EVENT_CONTENT_BLOCK_DELTA",
+    "STREAM_EVENT_CONTENT_BLOCK_START",
+    "STREAM_EVENT_CONTENT_BLOCK_STOP",
+    "STREAM_EVENT_MESSAGE_DELTA",
+    "STREAM_EVENT_MESSAGE_START",
+    "STREAM_EVENT_MESSAGE_STOP",
+    "TOML_EXTENSIONS",
+    "URL_PATTERN",
+    "UUID_PATTERN",
+    "YAML_EXTENSIONS",
+    # Validators
+    "validate_choice",
+    "validate_dict",
+    "validate_email",
+    "validate_list",
+    "validate_non_empty_string",
+    "validate_path",
+    "validate_port",
+    "validate_range",
+    "validate_timeout",
+    "validate_url",
+    "validate_uuid",
+]