import logging
from typing import Type

from gehomesdk.erd import ErdApplianceType

from .base import ApplianceApi
from .oven import OvenApi
from .fridge import FridgeApi
from .dishwasher import DishwasherApi
from .washer import WasherApi
from .dryer import DryerApi
from .washer_dryer import WasherDryerApi

_LOGGER = logging.getLogger(__name__)

def get_appliance_api_type(appliance_type: ErdApplianceType) -> Type:
    _LOGGER.debug(f"Found device type: {appliance_type}")
    """Get the appropriate appliance type"""
    if appliance_type == ErdApplianceType.OVEN:
        return OvenApi
    if appliance_type == ErdApplianceType.FRIDGE:
        return FridgeApi
    if appliance_type == ErdApplianceType.DISH_WASHER:
        return DishwasherApi
    if appliance_type == ErdApplianceType.WASHER:
        return WasherApi
    if appliance_type == ErdApplianceType.DRYER:
        return DryerApi
    if appliance_type == ErdApplianceType.COMBINATION_WASHER_DRYER:
        return WasherDryerApi
    # Fallback
    return ApplianceApi