from fastapi import APIRouter, Depends, Query
from sqlalchemy.orm import Session
from datetime import datetime, timezone

from ..database import get_db
from .. import models
from ..dependencies import get_current_user
from ..services.candles import refresh_latest_candles
from ..services.btc import get_btc_price_eur

router = APIRouter()


@router.get("/candles")
def get_candles(
    days: str = Query(default="365"),
    db: Session = Depends(get_db),
    current_user: models.User = Depends(get_current_user),
):
    # Refresh candles on every request (no-op if DB is already current)
    refresh_latest_candles(db)

    query = db.query(models.OHLCCandle).order_by(models.OHLCCandle.date.asc())

    if days != "all":
        try:
            limit = int(days)
        except ValueError:
            limit = 365
        # Fetch the most recent `limit` rows
        total = query.count()
        if total > limit:
            query = query.offset(total - limit)

    candles_db = query.all()
    candles = [
        {
            "date": c.date,
            "open": round(c.open, 2),
            "high": round(c.high, 2),
            "low": round(c.low, 2),
            "close": round(c.close, 2),
        }
        for c in candles_db
    ]

    # Patch today's candle with the live price so it tracks intraday movement
    today = datetime.now(tz=timezone.utc).strftime("%Y-%m-%d")
    live_price, _ = get_btc_price_eur()
    if live_price and candles:
        last = candles[-1]
        if last["date"] == today:
            last["close"] = round(live_price, 2)
            last["high"] = round(max(last["high"], live_price), 2)
            last["low"] = round(min(last["low"], live_price), 2)
        elif last["date"] < today:
            # No candle for today yet — create a synthetic one from live price
            prev_close = last["close"]
            candles.append({
                "date": today,
                "open": prev_close,
                "high": round(max(prev_close, live_price), 2),
                "low": round(min(prev_close, live_price), 2),
                "close": round(live_price, 2),
            })

    purchases_db = db.query(models.Purchase).filter(
        models.Purchase.user_id == current_user.id
    ).all()
    purchases = [
        {
            "date": p.created_at.strftime("%Y-%m-%d"),
            "amount_eur": round(p.amount_eur, 2),
            "price_eur": round(p.price_eur, 2),
        }
        for p in purchases_db
    ]

    return {"candles": candles, "purchases": purchases}