import time
import random
import logging
import json
import uuid
from typing import Callable, Optional, Any
from pydantic import BaseModel, create_model, ValidationError

_log = logging.getLogger(__name__)

# --- レイヤー1: 指数バックオフとジッター ---
def with_retry(
    fn: Callable[..., str],
    args: dict,
    max_attempts: int = 3,
    base_delay: float = 1.0,
) -> str:
    for attempt in range(max_attempts):
        try:
            return fn(**args)
        except Exception as e:
            if attempt == max_attempts - 1:
                raise
            delay = base_delay * (2 ** attempt) + random.uniform(0, 0.5)
            _log.warning("Attempt %d failed (%s) - retrying in %.1fs", attempt + 1, e, delay)
            time.sleep(delay)

# --- レイヤー2: サーキットブレーカー ---
class CircuitBreaker:
    CLOSED, OPEN, HALF_OPEN = "closed", "open", "half-open"

    def __init__(self, failure_threshold: int = 3, reset_timeout: float = 30.0) -> None:
        self.failure_threshold = failure_threshold
        self.reset_timeout = reset_timeout
        self._failures = 0
        self._state = self.CLOSED
        self._opened_at: Optional[float] = None

    def call(self, fn: Callable[..., str], args: dict) -> str:
        if self._state == self.OPEN:
            elapsed = time.time() - (self._opened_at or 0.0)
            if elapsed < self.reset_timeout:
                raise RuntimeError(
                    f"Circuit open - service unavailable (resets in {self.reset_timeout - elapsed:.0f}s)"
                )
            self._state = self.HALF_OPEN

        try:
            result = fn(**args)
            if self._state == self.HALF_OPEN:
                self._reset()
            return result
        except Exception:
            self._failures += 1
            if self._failures >= self.failure_threshold:
                self._state = self.OPEN
                self._opened_at = time.time()
            raise

    def _reset(self) -> None:
        self._failures = 0
        self._state = self.CLOSED
        self._opened_at = None

# --- レイヤー4: 構造化バリデーション ---
_VALIDATORS: dict[str, type[BaseModel]] = {}
_TYPE_MAP = {
    "string": str,
    "integer": int,
    "number": float,
    "boolean": bool,
}

def call_tool(name: str, args: dict) -> str:
    # 実際のツール実行プレースホルダー
    return f"Success: {name} executed with {args}"

def validated_call(name: str, args: dict) -> str:
    validator = _VALIDATORS.get(name)
    if validator is None:
        return call_tool(name, args)
    try:
        validated = validator(**args)
        return call_tool(name, validated.model_dump(exclude_none=True))
    except ValidationError as e:
        return f"Invalid arguments for '{name}': {e}"

# --- レイヤー5: トレースとマスキング ---
def traced_call(name: str, args: dict, fn: Callable[..., str]) -> str:
    sanitized = {
        k: "***" if any(w in k.lower() for w in ("key", "secret", "token", "password")) else v
        for k, v in args.items()
    }
    start = time.time()
    try:
        result = fn(**args)
        _log.info(
            "tool=%s args=%s result=%r duration=%.3fs",
            name, json.dumps(sanitized), str(result)[:120], time.time() - start,
        )
        return result
    except Exception as e:
        _log.error(
            "tool=%s args=%s error=%s duration=%.3fs",
            name, json.dumps(sanitized), e, time.time() - start,
        )
        raise

# --- ディスパッチャーの合成 ---
def _make_dispatcher(
    breakers: dict[str, CircuitBreaker],
    max_retries: int,
) -> Callable[[str, dict], str]:
    def dispatch(name: str, args: dict) -> str:
        def core(**kw) -> str:
            return validated_call(name, kw)

        def retried(**kw) -> str:
            return with_retry(core, kw, max_attempts=max_retries)

        def guarded(**kw) -> str:
            return breakers[name].call(retried, kw)

        return traced_call(name, args, guarded)

    return dispatch

# 1. 構造化バリデーションによる自己修正のトリガー
2026-06-13 10:42:01,102 [INFO] tool=charge_card args={"amount": "forty-nine", "card_token": "tok_123"} result='Invalid arguments for "charge_card": amount must be a float' duration=0.012s
2026-06-13 10:42:02,450 [INFO] LLM detected validation error. Retrying with corrected arguments...
2026-06-13 10:42:03,115 [INFO] tool=charge_card args={"amount": 49.00, "card_token": "tok_123"} result='Success: charge_card executed' duration=0.189s

# 2. 下流サービス障害に伴う指数バックオフの作動
2026-06-13 10:42:05,201 [WARNING] Attempt 1 failed (503 Service Unavailable) - retrying in 1.2s
2026-06-13 10:42:07,412 [WARNING] Attempt 2 failed (503 Service Unavailable) - retrying in 2.5s
2026-06-13 10:42:10,920 [ERROR] tool=send_notification args={"email": "[email protected]"} error=503 Service Unavailable duration=1.002s

# 3. 連続失敗によるサーキットブレーカーの作動（OPEN状態への遷移）
2026-06-13 10:42:11,005 [ERROR] tool=send_notification args={"email": "[email protected]"} error=Circuit open - service unavailable (resets in 30s) duration=0.001s