from typing import List

from apollo_utils.job.clients.base.client import BaseApiClient

from juno_email_messages.clients.base_config import ApiConfig
from juno_email_messages.constants import USER_DATA_HEADER, Service


class UserDataClientConfig(ApiConfig):
    service = Service.USER_DATA


class UserDataClient(UserDataClientConfig, BaseApiClient):
    def get_messages(self, id_list: List[int]) -> List[dict]:
        result = self.send_request(
            "api/service/messages/list/", method="POST", data={"id": id_list}, headers=USER_DATA_HEADER, get_json=True
        )
        return result["data"]