"""
Strategy optimizer API router.
"""
from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.orm import Session

from app.api.deps import CurrentUser
from app.core.database import get_db
from app.schemas.optimizer import (
    DEFAULT_GRIDS,
    STRATEGY_TYPES,
    OptimizeRequest,
    OptimizeResponse,
)
from app.services.optimizer import OptimizerService

router = APIRouter(prefix="/api/optimizer", tags=["optimizer"])


@router.post("", response_model=OptimizeResponse)
async def run_optimization(
    request: OptimizeRequest,
    current_user: CurrentUser,
    db: Session = Depends(get_db),
):
    """Run grid-search optimization for a strategy."""
    if request.strategy_type not in STRATEGY_TYPES:
        raise HTTPException(
            status_code=400,
            detail=f"Unknown strategy_type: {request.strategy_type}. "
            f"Valid types: {STRATEGY_TYPES}",
        )

    service = OptimizerService(db)
    try:
        return service.optimize(request)
    except ValueError as e:
        raise HTTPException(status_code=400, detail=str(e))


@router.get("/presets/{strategy_type}")
async def get_preset(
    strategy_type: str,
    current_user: CurrentUser,
):
    """Get default parameter grid preset for a strategy type."""
    if strategy_type not in DEFAULT_GRIDS:
        raise HTTPException(
            status_code=404,
            detail=f"No preset for strategy_type: {strategy_type}",
        )
    return {
        "strategy_type": strategy_type,
        "param_grid": DEFAULT_GRIDS[strategy_type],
    }


@router.get("/presets")
async def list_presets(
    current_user: CurrentUser,
):
    """List all available strategy presets."""
    return {
        "presets": {
            st: DEFAULT_GRIDS[st] for st in STRATEGY_TYPES
        }
    }