from apollo_utils.core.utils.ext_enum import ExtEnum

from server.db.models.messages import FeedMessage, Message, PushMessage


class MessagesType(ExtEnum):
    MESSAGES = "messages"
    PUSH_MESSAGES = "push_messages"
    FEED_MESSAGES = "feed_messages"


MESSAGE_TYPE_TABLE_MAP = {
    MessagesType.MESSAGES: Message,
    MessagesType.FEED_MESSAGES: FeedMessage,
    MessagesType.PUSH_MESSAGES: PushMessage,
}


class AccountsDeleteInclude(ExtEnum):
    DEVICES = "devices"


class AccountsGetInclude(ExtEnum):
    ID = "id"
    USER_ID = "user_id"