from typing import Callable

from apollo_messages_views.config import EVENT_VIEW
from apollo_messages_views.constants.base import EventView
from apollo_messages_views.handlers.feed_messages import handler as feed_handler
from apollo_messages_views.handlers.push_messages import handler as push_handler

EVENT_VIEW_TO_HANDLER = {
    EventView.FEED.value: feed_handler,
    EventView.PUSH.value: push_handler
}


def get_event_handler() -> Callable:
    return EVENT_VIEW_TO_HANDLER[EVENT_VIEW]