from chain_service.database.models.progress_chain import BaseProgressAction

from .base import BaseProgressActionService
from .wait_progress_action import WaitProgressActionService
from .comment_progress_action import CommentProgressActionService


class ProgressActionServiceFactory:

    def __call__(
        self, progress_action: BaseProgressAction
    ) -> BaseProgressActionService:

        match progress_action.action_type:

            case "wait":
                return WaitProgressActionService()

            case "comment":
                return CommentProgressActionService()