from __future__ import annotations

import abc
from typing import Any, Final

from moto.stepfunctions.parser.asl.component.common.string.string_expression import (
    StringVariableSample,
)
from moto.stepfunctions.parser.asl.component.state.choice.comparison.comparison_operator_type import (
    ComparisonOperatorType,
)
from moto.stepfunctions.parser.asl.component.state.choice.comparison.comparison_type import (
    Comparison,
)
from moto.stepfunctions.parser.asl.component.state.choice.comparison.operator.factory import (
    OperatorFactory,
)
from moto.stepfunctions.parser.asl.component.state.choice.comparison.operator.operator import (
    Operator,
)
from moto.stepfunctions.parser.asl.eval.environment import Environment


class ComparisonFunc(Comparison, abc.ABC):
    operator_type: Final[ComparisonOperatorType]

    def __init__(self, operator_type: ComparisonOperatorType):
        self.operator_type = operator_type


class ComparisonFuncValue(ComparisonFunc):
    value: Final[Any]

    def __init__(self, operator_type: ComparisonOperatorType, value: Any):
        super().__init__(operator_type=operator_type)
        self.value = value

    def _eval_body(self, env: Environment) -> None:
        operator: Operator = OperatorFactory.get(self.operator_type)
        operator.eval(env=env, value=self.value)


class ComparisonFuncStringVariableSample(ComparisonFuncValue):
    _COMPARISON_FUNC_VAR_VALUE: Final[str] = "$"
    string_variable_sample: Final[StringVariableSample]

    def __init__(
        self,
        operator_type: ComparisonOperatorType,
        string_variable_sample: StringVariableSample,
    ):
        super().__init__(
            operator_type=operator_type, value=self._COMPARISON_FUNC_VAR_VALUE
        )
        self.string_variable_sample = string_variable_sample

    def _eval_body(self, env: Environment) -> None:
        self.string_variable_sample.eval(env=env)
        super()._eval_body(env=env)
        # Purge the outcome of the variable sampling form the
        # stack as operators do not digest the input value.
        del env.stack[-2]
