#  Copyright (c) PARRRATE T&V 2021. All rights reserved.

from bu4.evaluation.av.evtype import evtype
from bu4.evaluation.constructs.eilambda import EILambda
from bu4.evaluation.constructs.evaluable import Evaluable
from bu4.indexing.constructs.indexed import Indexed
from bu4.parsing.codes import CODE_MAKE
from bu4.transform.states.aftertransform import AfterTransform
from bu4.transform.states.transformfinished import TransformFinished
from bu4.transform.states.transformstart import TransformStart
from bu4.transform.states.transformstate import TransformState
from bu4.transform.targets.atlambda import ATLambda

__all__ = ('ILambda',)


class ILambda(Indexed):
    def __init__(self, value: Indexed, table: list[int], *, memoize: bool):
        self.value = value
        self.table = table
        self.memoize = memoize

    def attach(self, ev: evtype) -> Evaluable:
        return EILambda(
            [ev[i] for i in self.table],
            self.value,
            memoize=self.memoize
        )

    def __str__(self):
        return f'(0){self.value}'

    def bytes(self) -> TransformState[bytes]:
        return AfterTransform(
            TransformStart(lambda: self.value.bytes()),
            ATLambda(
                lambda value: TransformFinished(
                    bytes([CODE_MAKE, self.memoize, len(self.table), *self.table, *value])
                )
            )
        )