from custom_components.ecoflow_cloud.config_flow import EcoflowModel
from custom_components.ecoflow_cloud.devices import BaseDevice, DiagnosticDevice
from custom_components.ecoflow_cloud.devices.delta2 import Delta2
from custom_components.ecoflow_cloud.devices.delta_mini import DeltaMini
from custom_components.ecoflow_cloud.devices.delta_pro import DeltaPro
from custom_components.ecoflow_cloud.devices.river2 import River2
from custom_components.ecoflow_cloud.devices.river2_max import River2Max
from custom_components.ecoflow_cloud.devices.river2_pro import River2Pro
from custom_components.ecoflow_cloud.devices.river_max import RiverMax
from custom_components.ecoflow_cloud.devices.river_pro import RiverPro
from custom_components.ecoflow_cloud.devices.delta_max import DeltaMax
from custom_components.ecoflow_cloud.devices.delta2_max import Delta2Max
from custom_components.ecoflow_cloud.devices.powerstream import PowerStream
from custom_components.ecoflow_cloud.devices.glacier import Glacier
from custom_components.ecoflow_cloud.devices.wave2 import Wave2

devices: dict[str, BaseDevice] = {
    EcoflowModel.DELTA_2.name: Delta2(),
    EcoflowModel.RIVER_2.name: River2(),
    EcoflowModel.RIVER_2_MAX.name: River2Max(),
    EcoflowModel.RIVER_2_PRO.name: River2Pro(),
    EcoflowModel.DELTA_PRO.name: DeltaPro(),
    EcoflowModel.RIVER_MAX.name: RiverMax(),
    EcoflowModel.RIVER_PRO.name: RiverPro(),
    EcoflowModel.DELTA_MINI.name: DeltaMini(),
    EcoflowModel.DELTA_MAX.name: DeltaMax(),
    EcoflowModel.DELTA_2_MAX.name: Delta2Max(),
    EcoflowModel.POWERSTREAM.name: PowerStream(),
    EcoflowModel.GLACIER.name: Glacier(),
    EcoflowModel.WAVE_2.name: Wave2(),
    EcoflowModel.DIAGNOSTIC.name: DiagnosticDevice()
}