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


class NodeGroupClient(BaseClient):
    """Service client for NodeGroup 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 list_node_groups(self, service_name, **params):
        """Get node groups of the service."""
        url = self.client.get_node_groups_path(self.project_name, service_name)
        if params:
            url += f"?{urllib.urlencode(params)}"
        resp, body = self.get(url)
        return self._parse_response(
            resp, body, self.client.get_schema().list_node_groups
        )

    def scale_nodes(self, service_name, scale):
        """Scale nodes of the node groups in the service."""
        url = self.client.get_scale_nodes_path(self.project_name, service_name)
        put_body = json.dumps({"scale": scale})
        resp, body = self.put(url, put_body)
        return self._parse_response(
            resp, body, self.client.get_schema().scale_nodes
        )

    def update_server_type(self, service_name, server_type):
        """Update server type of the service."""
        url = self.client.get_server_type_path(self.project_name, service_name)
        put_body = json.dumps({"serverType": server_type})
        resp, body = self.put(url, put_body)
        return self._parse_response(
            resp, body, self.client.get_schema().update_server_type
        )

    def failover_node_group(self, service_name, node_group_name, primary):
        """Failover node group."""
        url = self.client.get_node_group_failover_path(self.project_name, service_name, node_group_name)
        put_body = json.dumps({"primary": primary})
        resp, body = self.put(url, put_body)
        return self._parse_response(
            resp, body, self.client.get_schema().failover_node_group
        )


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

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

    @abstractmethod
    def get_node_groups_path(self, project_name, service_name):
        """Return API endpoint path for node groups."""
        pass

    @abstractmethod
    def get_scale_nodes_path(self, project_name, service_name):
        """Return API endpoint path for scaling nodes."""
        pass

    @abstractmethod
    def get_server_type_path(self, project_name, service_name):
        """Return API endpoint path for updating server type."""
        pass

    @abstractmethod
    def get_node_group_failover_path(self, project_name, service_name, node_group_name):
        """Return API endpoint path for node group failover."""
        pass