import marimo

__generated_with = "0.23.3"
app = marimo.App()


@app.cell
def _():
    import marimo as mo

    mo.md(
        """
        # Verify vega-lite tooltip fix

        This notebook reproduces the numeric-tooltip bug fixed by
        [vega-lite#9707](https://github.com/vega/vega-lite/pull/9707) (released in
        vega-lite **6.4.2**).

        The bug: when hovering near a point on a chart that uses
        `selection_point(nearest=True)`, the tooltip would show `undefined`
        instead of the actual numeric value.

        **Hover over the points below.** The tooltip should display real
        numbers — not `undefined`.
        """
    )
    return (mo,)


@app.cell
def _():
    import altair as alt
    import pandas as pd

    return alt, pd


@app.cell
def _(alt, pd):
    df = pd.DataFrame(
        {
            "x": list(range(20)),
            "y": [v * v for v in range(20)],
        }
    )

    nearest = alt.selection_point(
        fields=["x"], nearest=True, on="mouseover", empty=False
    )

    line = alt.Chart(df).mark_line().encode(x="x:Q", y="y:Q")
    points = (
        alt.Chart(df)
        .mark_point(size=80, filled=True)
        .encode(
            x="x:Q",
            y="y:Q",
            opacity=alt.condition(nearest, alt.value(1), alt.value(0)),
            tooltip=["x:Q", "y:Q"],
        )
        .add_params(nearest)
    )

    chart = (line + points).properties(
        title="Hover near points — tooltip should show numbers, not `undefined`"
    )
    chart
    return


@app.cell
def _(mo):
    mo.md("""
    ## Datasource panel chart

    The same pattern is used by the datasource preview chart for date
    columns (see `DateChartBuilder` in `marimo/_data/charts.py`). If the
    upgrade is good, hovering anywhere along the area below should show
    the count and date — never `undefined`.
    """)
    return


@app.cell
def _(alt, pd):
    import datetime as _dt

    dates = pd.DataFrame(
        {
            "date": [
                _dt.date(2024, 1, 1) + _dt.timedelta(days=i)
                for i in range(60)
            ],
            "count": [i % 7 + 1 for i in range(60)],
        }
    )

    nearest_d = alt.selection_point(
        fields=["date"], nearest=True, on="mouseover", empty=False
    )

    base = alt.Chart(dates)
    area = base.mark_area(opacity=0.4).encode(x="date:T", y="count:Q")
    pts = (
        base.mark_point(size=80, filled=True)
        .encode(
            x="date:T",
            y="count:Q",
            opacity=alt.condition(nearest_d, alt.value(1), alt.value(0)),
            tooltip=[
                alt.Tooltip("date:T"),
                alt.Tooltip("count:Q", format=",.0f"),
            ],
        )
        .add_params(nearest_d)
    )
    (area + pts).properties(width="container", height=200)
    return


if __name__ == "__main__":
    app.run()