from typing import List

from scheduler.block import InputBlock


class TextInput(InputBlock):
    def input(self, text: str):
        self.text = text

    def __call__(self) -> str:
        return self.text


class ListInput(InputBlock):
    def input(self, messages: list):
        self.messages = messages

    def __call__(self) -> list:
        return self.messages


class DictInput(InputBlock):
    def input(self, messages: dict):
        self.messages = messages

    def __call__(self) -> dict:
        return self.messages