"""API Client for ows-transcoding."""

from typing import Any

import requests


class APIClient:
    """API Client for ows-transcoding."""

    def __init__(self, base_url: str, session: Any | None = None) -> None:
        """Initialize APIClient class."""
        self.base_url = base_url
        self.session = session

    def transcoding_order(
        self, headers: dict[str, Any], data: dict[str, Any]
    ) -> requests.Response:
        """Execute a POST against /transcoding/preset."""
        env = self.base_url.split("://")[-1].split("-")[0]
        endpoint = "{}/transcoding/preset/{}_mezzanine".format(self.base_url, env)
        return requests.post(endpoint, headers=headers, json=data)

    def transcoding_status(self, order_id: int) -> requests.Response:
        """Execute a GET against /transcoding-order/status."""
        endpoint = "{}/transcoding-order/{}/status".format(self.base_url, order_id)
        return requests.get(endpoint)