from __future__ import annotations

from typing import Any

import pandas as pd
import polars as pl
import pyarrow as pa
from typing_extensions import reveal_type

import narwhals as nw

data: dict[str, Any] = {"a": [1, 2, 3]}

impl_1 = nw.from_native(pd.DataFrame(data)).implementation
impl_2 = nw.from_native(pl.DataFrame(data)).implementation
impl_3 = nw.from_native(pa.table(data)).implementation

reveal_type(impl_1)  # Type of "impl_1" is "Literal[Implementation.PANDAS]"
reveal_type(impl_2)  # Type of "impl_2" is "Literal[Implementation.POLARS]"
reveal_type(impl_3)  # Type of "impl_3" is "Literal[Implementation.PYARROW]"

if TYPE_CHECKING:
    from narwhals.typing import IntoFrameT


def agnostic_function(df_native: IntoFrameT) -> IntoFrameT:
    frame = nw.from_native(df_native)
    reveal_type(frame.implementation)  # Type of "frame.implementation" is "Literal[Implementation.PANDAS, Implementation.CUDF, Implementation.MODIN, Implementation.PYARROW, Implementation.POLARS, Implementation.PYSPARK, Implementation.SQLFRAME, Implementation.PYSPARK_CONNECT, Implementation.DASK, Implementation.DUCKDB, Implementation.IBIS]"
    return frame.to_native()

if TYPE_CHECKING:
    from narwhals.typing import IntoDataFrameT


def agnostic_function(df_native: IntoDataFrameT) -> IntoDataFrameT:
    frame = nw.from_native(df_native)
    reveal_type(frame.implementation)  # Type of "frame.implementation" is "Literal[Implementation.PANDAS, Implementation.CUDF, Implementation.MODIN, Implementation.PYARROW, Implementation.POLARS]"
    return frame.to_native()

if TYPE_CHECKING:
    from narwhals.typing import IntoLazyFrameT


def agnostic_function(df_native: IntoLazyFrameT) -> IntoLazyFrameT:
    frame = nw.from_native(df_native)
    reveal_type(frame.implementation)  # Type of "frame.implementation" is "Literal[Implementation.PYSPARK, Implementation.SQLFRAME, Implementation.PYSPARK_CONNECT, Implementation.DASK, Implementation.DUCKDB, Implementation.IBIS, Implementation.POLARS]"
    return frame.to_native()