from typing import Union, Dict, List, Optional, Any
from Services.Redis.Models.base import RedisRow


class RedisResponse:
    """Base class for Redis response handling."""

    def __init__(
        self,
        status: bool,
        message: str,
        data: Any = None,
        error: Optional[str] = None,
    ):
        self.status = status
        self.message = message
        self.data = data

        if isinstance(data, dict):
            self.data_type = "dict"
        elif isinstance(data, list):
            self.data_type = "list"
        elif data is None:
            self.data_type = None
        self.error = error

    def as_dict(self) -> Dict:
        data = self.all
        main_dict = {
            "status": self.status,
            "message": self.message,
            "count": self.count,
            "dataType": self.data_type,
        }
        if isinstance(data, RedisRow):
            return {"data": {data.keys: data.data}, **main_dict}
        elif isinstance(data, list):
            return {"data": {row.keys: row.data for row in data}, **main_dict}

    @property
    def all(self) -> Union[Optional[List[RedisRow]]]:
        return self.data or []

    @property
    def count(self) -> int:
        return len(self.all)

    @property
    def first(self) -> Union[RedisRow, None]:
        return self.data[0] if self.data else None