import json
from abc import ABC, abstractmethod
from lib.common import rest_client
from lib.services.redis.base_client import BaseClient


class NodeClient(BaseClient):
    """Service client for Node API"""

    def __init__(self, auth_provider, client, **kwargs):
        super().__init__(auth_provider=auth_provider, **kwargs)
        self.client = client

    def _parse_response(self, resp, body, schema):
        """Parse and validate response."""
        # Handle empty response bodies (e.g., validation endpoints)
        if not body or body.strip() == '':
            body_data = {}
        else:
            body_data = json.loads(body)

        self.validate_response(schema, resp, body_data)

        # Return format expected by tests: {'status': int, 'body': dict}
        return {
            'status': resp.status,
            'body': body_data
        }

    def get_node_redis_info(self, service_name, node_name):
        """Get Redis Info of the Node."""
        url = self.client.get_node_redis_info_path(self.project_name, service_name, node_name)
        resp, body = self.get(url)
        return self._parse_response(
            resp, body, self.client.get_schema().get_node_redis_info
        )


class NodeClientInterface(ABC):
    """Interface for cloud-specific implementations."""

    @abstractmethod
    def get_schema(self):
        """Return response schemas module."""
        pass

    @abstractmethod
    def get_node_redis_info_path(self, project_name, service_name, node_name):
        """Return API endpoint path for node redis info."""
        pass