# main.py
from fastapi import FastAPI
from fastapi.responses import JSONResponse
import yfinance as yf
from datetime import datetime, timedelta
import pytz

app = FastAPI()

@app.get("/stock/{name}")
async def get_stock_history(name: str):
    tz = pytz.timezone('Asia/Kolkata')
    today = datetime.now(tz)

    dates = {
        "Current": today,
        "1 Month Ago": today - timedelta(days=30),
        "6 Months Ago": today - timedelta(days=182),
        "1 Year Ago": today - timedelta(days=365),
        "3 Years Ago": today - timedelta(days=3*365),
    }

    try:
        ticker = yf.Ticker(name)
        start_date = (today - timedelta(days=4*365)).strftime('%Y-%m-%d')
        end_date = (today + timedelta(days=1)).strftime('%Y-%m-%d')

        hist = ticker.history(start=start_date, end=end_date)

        if hist.empty:
            return JSONResponse(content={"error": f"No data found for {name}"}, status_code=404)

        if hist.index.tz is None:
            hist.index = hist.index.tz_localize(tz)
        else:
            hist.index = hist.index.tz_convert(tz)

        def get_closest_price(date):
            available_dates = hist.index[hist.index <= date]
            if not available_dates.empty:
                closest_date = available_dates[-1]
                return float(hist.loc[closest_date]['Close']), closest_date.date()
            else:
                return None, None

        result = {}
        for label, date in dates.items():
            price, actual_date = get_closest_price(date)
            result[label] = {
                "price": f"{price:.2f}" if price is not None else "No data",
                "date": str(actual_date) if actual_date else None
            }

        latest_date = hist.index[-1].date()
        if latest_date < today.date():
            result["Warning"] = f"Data for {today.date()} not available. Latest is from {latest_date}."

        return result

    except Exception as e:
        return JSONResponse(content={"error": str(e)}, status_code=500)