from typing import Union

from dependency_injector import containers, providers

from .datetime_sampler import DatetimeSampler
from .factory import sampler_factory
from .inverse_transform_sampler import (
    HistogramBucket,
    InverseTransformSampler,
    TopNItem,
)
from .numeric_samplers import DecimalSampler, FloatSampler, IntSampler
from .text_sampler import TextSampler

Sampler = Union[DatetimeSampler, DecimalSampler, IntSampler, TextSampler]


class SamplerContainer(containers.DeclarativeContainer):
    sampler = providers.Factory(sampler_factory)


# from .checkpointable_prng import CheckpointablePRNG
# from .tidb_types import create_tidb_bucket, create_tidb_topn_item

__all__ = [
    "InverseTransformSampler",
    "HistogramBucket",
    "TopNItem",
    # "create_tidb_bucket",
    # "create_tidb_topn_item",
    "TextSampler",
    "IntSampler",
    "FloatSampler",
    "DecimalSampler",
    "DatetimeSampler",
    # "CheckpointablePRNG",
]