"""Probe endpoints mounted ONLY in the test app (under /v1/_test/) to exercise the rate-limit contract:
row accounting (json/csv/parquet), quota_exempt, request_cost, requires_key, key-required tags, clamp_limit.
"""
from __future__ import annotations

import numpy as np
import pandas as pd
from fastapi import FastAPI, Query, Request

from core.responses import clamp_limit, frame_response, json_response, parse_format


def install(app: FastAPI) -> None:
    def frame(request: Request, n: int = Query(10), format: str = Query("json"), limit: int | None = Query(None)):
        lim = clamp_limit(limit, 10_000, 1_000_000, request) if limit is not None else n
        rows = min(n, lim)
        df = pd.DataFrame({"i": np.arange(rows), "x": np.linspace(0, 1, rows) if rows else []})
        return frame_response(df, parse_format(format), request=request)

    def exempt(request: Request, n: int = Query(1000)):
        request.state.quota_exempt = True
        return json_response([{"i": i} for i in range(n)])

    def expensive(request: Request):
        request.state.request_cost = 2
        return json_response({"ok": True})

    def needkey(request: Request):
        request.state.requires_key = True
        return json_response({"secret": True})

    def tagged():
        return json_response({"stream": True})

    def boom():
        raise RuntimeError("kaboom")

    app.router.add_api_route("/v1/_test/frame", frame, methods=["GET"], include_in_schema=False)
    app.router.add_api_route("/v1/_test/exempt", exempt, methods=["GET"], include_in_schema=False)
    app.router.add_api_route("/v1/_test/expensive", expensive, methods=["GET"], include_in_schema=False)
    app.router.add_api_route("/v1/_test/needkey", needkey, methods=["GET"], include_in_schema=False)
    app.router.add_api_route("/v1/_test/tagged", tagged, methods=["GET"], tags=["stream"], include_in_schema=False)
    app.router.add_api_route("/v1/_test/boom", boom, methods=["GET"], include_in_schema=False)