"""Routes for calendar events, screening, ownership, and estimates."""

from fastapi import APIRouter, Path, Query

from models import ApiResponse
from route_utils import safe, validate_symbol
import calendar_service

router = APIRouter(prefix="/api/v1")

DATE_PATTERN = r"^\d{4}-\d{2}-\d{2}$"


# --- Calendar Events ---


@router.get("/calendar/earnings", response_model=ApiResponse)
@safe
async def earnings_calendar(
    start_date: str | None = Query(
        default=None, pattern=DATE_PATTERN, description="YYYY-MM-DD"
    ),
    end_date: str | None = Query(
        default=None, pattern=DATE_PATTERN, description="YYYY-MM-DD"
    ),
):
    """Get upcoming earnings announcements."""
    data = await calendar_service.get_earnings_calendar(start_date, end_date)
    return ApiResponse(data=data)


@router.get("/calendar/dividends", response_model=ApiResponse)
@safe
async def dividend_calendar(
    start_date: str | None = Query(
        default=None, pattern=DATE_PATTERN, description="YYYY-MM-DD"
    ),
    end_date: str | None = Query(
        default=None, pattern=DATE_PATTERN, description="YYYY-MM-DD"
    ),
):
    """Get upcoming dividend dates."""
    data = await calendar_service.get_dividend_calendar(start_date, end_date)
    return ApiResponse(data=data)


@router.get("/calendar/ipo", response_model=ApiResponse)
@safe
async def ipo_calendar(
    start_date: str | None = Query(
        default=None, pattern=DATE_PATTERN, description="YYYY-MM-DD"
    ),
    end_date: str | None = Query(
        default=None, pattern=DATE_PATTERN, description="YYYY-MM-DD"
    ),
):
    """Get upcoming IPOs."""
    data = await calendar_service.get_ipo_calendar(start_date, end_date)
    return ApiResponse(data=data)


@router.get("/calendar/splits", response_model=ApiResponse)
@safe
async def splits_calendar(
    start_date: str | None = Query(
        default=None, pattern=DATE_PATTERN, description="YYYY-MM-DD"
    ),
    end_date: str | None = Query(
        default=None, pattern=DATE_PATTERN, description="YYYY-MM-DD"
    ),
):
    """Get upcoming stock splits."""
    data = await calendar_service.get_splits_calendar(start_date, end_date)
    return ApiResponse(data=data)


# --- Analyst Estimates ---


@router.get("/stock/{symbol}/estimates", response_model=ApiResponse)
@safe
async def stock_estimates(symbol: str = Path(..., min_length=1, max_length=20)):
    """Get analyst consensus estimates."""
    symbol = validate_symbol(symbol)
    data = await calendar_service.get_analyst_estimates(symbol)
    return ApiResponse(data=data)


@router.get("/stock/{symbol}/share-statistics", response_model=ApiResponse)
@safe
async def stock_share_stats(symbol: str = Path(..., min_length=1, max_length=20)):
    """Get share statistics: float, outstanding, short interest."""
    symbol = validate_symbol(symbol)
    data = await calendar_service.get_share_statistics(symbol)
    return ApiResponse(data=data)


# --- Ownership (SEC, free) ---


@router.get("/stock/{symbol}/sec-insider", response_model=ApiResponse)
@safe
async def stock_sec_insider(symbol: str = Path(..., min_length=1, max_length=20)):
    """Get insider trading data from SEC (Form 4)."""
    symbol = validate_symbol(symbol)
    data = await calendar_service.get_insider_trading(symbol)
    return ApiResponse(data=data)


@router.get("/stock/{symbol}/institutional", response_model=ApiResponse)
@safe
async def stock_institutional(symbol: str = Path(..., min_length=1, max_length=20)):
    """Get institutional holders from SEC 13F filings."""
    symbol = validate_symbol(symbol)
    data = await calendar_service.get_institutional_holders(symbol)
    return ApiResponse(data=data)


# --- Screener ---


@router.get("/screener", response_model=ApiResponse)
@safe
async def stock_screener():
    """Screen stocks using available filters."""
    data = await calendar_service.screen_stocks()
    return ApiResponse(data=data)