from src.libs.custom_logger import get_custom_logger
import json
from src.api_clients.base_client import BaseClient

logger = get_custom_logger(__name__)


class REST(BaseClient):
    def __init__(self, rest_port, rest_host="127.0.0.1"):
        self._rest_port = rest_port
        self._rest_host = rest_host

    def rest_call(self, method, endpoint, payload=None):
        url = f"http://{self._rest_host}:{self._rest_port}/{endpoint}"
        headers = {"Content-Type": "application/json", "Connection": "close"}
        return self.make_request(method, url, headers=headers, data=payload)

    def rest_call_text(self, method, endpoint, payload=None):
        url = f"http://{self._rest_host}:{self._rest_port}/{endpoint}"
        headers = {"accept": "text/plain", "Connection": "close"}
        return self.make_request(method, url, headers=headers, data=payload)

    def info(self):
        status_response = self.rest_call("get", "cryptarchia/info")
        return status_response.json()

    def send_dispersal_request(self, data):
        return self.rest_call("post", "disperse-data", json.dumps(data))

    def send_get_range(self, query):
        response = self.rest_call("post", "da/get-range", json.dumps(query))
        return response.json()