#!/usr/bin/env python3
import os, json, requests
import yfinance as yf
import pandas as pd
import numpy as np

TICKER       = os.getenv("TICKER", "ETH-USD")
INTERVAL     = os.getenv("YF_INTERVAL", "1d")
PERIOD       = os.getenv("YF_PERIOD", "2y")
INITIAL_CASH = float(os.getenv("INITIAL_CASH", "1000000"))
PAYLOAD_OUT  = os.getenv("PAYLOAD_OUT", "quantmaxi_payload.json")

def build_payload() -> dict:
    data = yf.download(TICKER, interval=INTERVAL, period=PERIOD,
                       auto_adjust=False, progress=False)
    if data is None or data.empty:
        raise RuntimeError(f"No data for {TICKER}")
    if isinstance(data.columns, pd.MultiIndex):
        data.columns = data.columns.get_level_values(0)

    data["SMA20"] = data["Close"].rolling(20).mean()
    data["SMA50"] = data["Close"].rolling(50).mean()

    data["Signal"] = 0
    data.loc[(data["SMA20"] > data["SMA50"]) &
             (data["SMA20"].shift(1) <= data["SMA50"].shift(1)), "Signal"] = 1
    data.loc[(data["SMA20"] < data["SMA50"]) &
             (data["SMA20"].shift(1) >= data["SMA50"].shift(1)), "Signal"] = -1

    data["Position"] = data["Signal"].shift(1).fillna(0)
    data["Units"] = (data["Position"].diff().fillna(0) *
                     INITIAL_CASH / data["Close"]).cumsum()
    data["Holdings"] = data["Units"] * data["Close"]
    data["Cash"] = INITIAL_CASH - (data["Units"].diff().fillna(0) *
                                    data["Close"]).cumsum()
    data["Equity"] = data["Holdings"] + data["Cash"]

    data.index = pd.to_datetime(data.index, utc=True)
    data["time"] = (data.index.view("int64") // 10**9).astype("int64")

    cols = ["time","Open","High","Low","Close","Volume",
            "SMA20","SMA50","Signal","Holdings","Cash","Equity"]
    data = data[cols]
    data.replace([np.inf, -np.inf], np.nan, inplace=True)
    data.fillna(0, inplace=True)
    data.sort_values("time", inplace=True)

    return {
        "ticker": TICKER, "interval": INTERVAL,
        "columns": cols,
        "data": data.to_dict(orient="records"),
        "metadata": {
            "ticker": TICKER, "interval": INTERVAL,
            "Open":     {"plot": True, "type": "candlestick", "component": "open"},
            "High":     {"plot": True, "type": "candlestick", "component": "high"},
            "Low":      {"plot": True, "type": "candlestick", "component": "low"},
            "Close":    {"plot": True, "type": "candlestick", "component": "close"},
            "SMA20":    {"plot": True, "type": "line", "color": "#f07b01", "paneIndex": 0},
            "SMA50":    {"plot": True, "type": "line", "color": "#2962ff", "paneIndex": 0},
            "Volume":   {"plot": True, "type": "histogram",
                         "upColor": "#089981", "downColor": "#f23645", "paneIndex": 0},
            "Equity":   {"plot": True, "type": "area", "color": "#7c83fd", "paneIndex": 1},
            "Holdings": {"plot": True, "type": "line", "color": "#22c55e", "paneIndex": 2},
        },
    }

def upload(payload: dict) -> None:
    token = os.getenv("API_TOKEN", "").strip()
    chart_name = os.getenv("CHART_NAME", "default").strip() or "default"
    if not token:
        print("No API_TOKEN -> upload skipped (JSON saved locally).")
        return
    resp = requests.post(
        "https://www.quantmaxi.com/api/ingest",
        headers={"Authorization": f"Bearer {token}",
                 "Content-Type": "application/json"},
        json={"chartName": chart_name, "payload": payload},
        timeout=60)
    print(f"Upload: {resp.status_code} {resp.text}")

if __name__ == "__main__":
    payload = build_payload()
    with open(PAYLOAD_OUT, "w") as f:
        json.dump(payload, f, indent=2)
    print(f"Saved {PAYLOAD_OUT}")
    upload(payload)

# /api/ingest  (token auth, wrapped body)
curl -X POST "https://www.quantmaxi.com/api/ingest" \
  -H "Authorization: Bearer $API_TOKEN" \
  -H "Content-Type: application/json" \
  -d '{"chartName":"default","payload":{...}}'

# /api/charts/by-id/{id}/ingest  (token auth, raw payload body)
curl -X POST "https://www.quantmaxi.com/api/charts/by-id/<ID>/ingest" \
  -H "Authorization: Bearer $API_TOKEN" \
  -H "Content-Type: application/json" \
  -d '{...payload...}'

# /api/charts/{name}/ingest  (browser session only, raw payload body)
curl -X POST "https://www.quantmaxi.com/api/charts/default/ingest" \
  -H "Content-Type: application/json" \
  -d '{...payload...}'