from typing import Dict, Any, Literal

from ..classes import CacheMemcached, CacheRedis


def create_cache_client(
    config: Dict[str, Any],
    engine: Literal["memcached", "redis"] | None = None,
) -> CacheMemcached | CacheRedis:
    if engine not in ["memcached", "redis"] or engine is None:
        raise KeyError(f"Incorrect cache engine provided. Expected 'memcached' or 'redis', got '{engine}'")

    if "cache" not in config or engine not in config["cache"]:
        raise KeyError(
            f"Cache configuration is invalid. Please check if all keys are set (engine: '{engine}')"
        )

    match engine:
        case "memcached":
            return CacheMemcached.from_config(config["cache"][engine])
        case "redis":
            return CacheRedis.from_config(config["cache"][engine])
        case _:
            raise KeyError(f"Cache implementation for the engine '{engine}' is not present.")