import sys
from typing import TYPE_CHECKING

if TYPE_CHECKING or sys.version_info >= (3, 12):
    from . import utils as utils
    from ._augmented_lagrangian import (
        AugmentedLagrangianConfig as AugmentedLagrangianConfig,
    )
    from ._cost import Cost as Cost
    from ._covariance import CovarianceEstimator as CovarianceEstimator
    from ._covariance import (
        LinearSolverCovarianceEstimatorConfig as LinearSolverCovarianceEstimatorConfig,
    )
    from ._lie_group_variables import SE2Var as SE2Var
    from ._lie_group_variables import SE3Var as SE3Var
    from ._lie_group_variables import SO2Var as SO2Var
    from ._lie_group_variables import SO3Var as SO3Var
    from ._problem import AnalyzedLeastSquaresProblem as AnalyzedLeastSquaresProblem
    from ._problem import LeastSquaresProblem as LeastSquaresProblem
    from ._solvers import ConjugateGradientConfig as ConjugateGradientConfig
    from ._solvers import SolveSummary as SolveSummary
    from ._solvers import TerminationConfig as TerminationConfig
    from ._solvers import TrustRegionConfig as TrustRegionConfig
    from ._variables import Var as Var
    from ._variables import VarValues as VarValues
elif sys.version_info >= (3, 10):
    from ._py310 import utils as utils
    from ._py310._augmented_lagrangian import (
        AugmentedLagrangianConfig as AugmentedLagrangianConfig,
    )
    from ._py310._cost import Cost as Cost
    from ._py310._covariance import CovarianceEstimator as CovarianceEstimator
    from ._py310._covariance import (
        LinearSolverCovarianceEstimatorConfig as LinearSolverCovarianceEstimatorConfig,
    )
    from ._py310._lie_group_variables import SE2Var as SE2Var
    from ._py310._lie_group_variables import SE3Var as SE3Var
    from ._py310._lie_group_variables import SO2Var as SO2Var
    from ._py310._lie_group_variables import SO3Var as SO3Var
    from ._py310._problem import (
        AnalyzedLeastSquaresProblem as AnalyzedLeastSquaresProblem,
    )
    from ._py310._problem import LeastSquaresProblem as LeastSquaresProblem
    from ._py310._solvers import ConjugateGradientConfig as ConjugateGradientConfig
    from ._py310._solvers import SolveSummary as SolveSummary
    from ._py310._solvers import TerminationConfig as TerminationConfig
    from ._py310._solvers import TrustRegionConfig as TrustRegionConfig
    from ._py310._variables import Var as Var
    from ._py310._variables import VarValues as VarValues
else:
    assert False


__version__ = "0.0.0"

# Some shims for backwards compatibility.
if not TYPE_CHECKING:
    import sys
    from typing import Any

    # Create a descriptor for Factor that shows a deprecation warning when accessed
    class _FactorDescriptor:
        def __get__(self, obj: Any, objtype: Any = None) -> Any:
            utils.print_deprecation_warning(
                "Import detected for [yellow]jaxls.Factor[/yellow], which has been renamed:",
                "[yellow]jaxls.Factor[/yellow] has been renamed to [green]jaxls.Cost[/green]. "
                "See migration guide at [blue]https://github.com/brentyi/jaxls/pull/36[/blue] for details. "
                "To suppress this warning, you can also downgrade jaxls: [bold]pip install git+https://github.com/brentyi/jaxls.git@21219e08676e13bb481230907e1cbc486ee284c7[/bold].",
            )

            # Standard warning is now handled by print_deprecation_warning
            return Cost

    # Create a descriptor for FactorGraph that shows a deprecation warning when accessed
    class _FactorGraphDescriptor:
        def __get__(self, obj: Any, objtype: Any = None) -> Any:
            utils.print_deprecation_warning(
                "Import detected for [yellow]jaxls.FactorGraph[/yellow], which has been renamed:",
                "[yellow]jaxls.FactorGraph.make(...)[/yellow] has been replaced with [green]jaxls.LeastSquaresProblem(...).analyze()[/green]. "
                "See migration guide at [blue]https://github.com/brentyi/jaxls/pull/36[/blue] for details. "
                "To suppress this warning, you can also downgrade jaxls: [bold]pip install git+https://github.com/brentyi/jaxls.git@21219e08676e13bb481230907e1cbc486ee284c7[/bold].",
            )

            class FactorGraph:
                @staticmethod
                def make(factors, variables, use_onp=False):
                    return LeastSquaresProblem(
                        costs=factors, variables=variables
                    ).analyze(use_onp)

            return FactorGraph

    # Use the descriptors to trigger warnings when Factor or FactorGraph are accessed
    sys.modules[__name__].__class__ = type(
        sys.modules[__name__].__class__.__name__,
        (sys.modules[__name__].__class__,),
        {
            "Factor": _FactorDescriptor(),
            "FactorGraph": _FactorGraphDescriptor(),
        },
    )