import { HassEntity } from 'home-assistant-js-websocket';

export const binarySensorIcon = (state?: string, entity?: HassEntity) => {
  const isOff = state === 'off';
  switch (entity?.attributes.device_class) {
    case 'battery':
      return isOff ? 'mdi:battery' : 'mdi:battery-outline';
    case 'battery_charging':
      return isOff ? 'mdi:battery' : 'mdi:battery-charging';
    case 'cold':
      return isOff ? 'mdi:thermometer' : 'mdi:snowflake';
    case 'connectivity':
      return isOff ? 'mdi:close-network-outline' : 'mdi:check-network-outline';
    case 'door':
      return isOff ? 'mdi:door-closed' : 'mdi:door-open';
    case 'garage_door':
      return isOff ? 'mdi:garage' : 'mdi:garage-open';
    case 'power':
      return isOff ? 'mdi:power-plug-off' : 'mdi:power-plug';
    case 'gas':
    case 'problem':
    case 'safety':
    case 'tamper':
      return isOff ? 'mdi:check-circle' : 'mdi:alert-circle';
    case 'smoke':
      return isOff ? 'mdi:check-circle' : 'mdi:smoke';
    case 'heat':
      return isOff ? 'mdi:thermometer' : 'mdi:fire';
    case 'light':
      return isOff ? 'mdi:brightness5' : 'mdi:brightness-7';
    case 'lock':
      return isOff ? 'mdi:lock' : 'mdi:lock-open';
    case 'moisture':
      return isOff ? 'mdi:water-off' : 'mdi:water';
    case 'motion':
      return isOff ? 'mdi:motion-sensor-off' : 'mdi:motion-sensor';
    case 'occupancy':
      return isOff ? 'mdi:home-outline' : 'mdi:home';
    case 'opening':
      return isOff ? 'mdi:square' : 'mdi:square-outline';
    case 'plug':
      return isOff ? 'mdi:power-plug-off' : 'mdi:power-plug';
    case 'presence':
      return isOff ? 'mdi:home-outline' : 'mdi:home';
    case 'running':
      return isOff ? 'mdi:stop' : 'mdi:play';
    case 'sound':
      return isOff ? 'mdi:music-note-off' : 'mdi:music-note';
    case 'update':
      return isOff ? 'mdi:package' : 'mdi:package-up';
    case 'vibration':
      return isOff ? 'mdi:crop-portrait' : 'mdi:vibrate';
    case 'window':
      return isOff ? 'mdi:window-closed' : 'mdi:window-open';
    default:
      return isOff ? 'mdi:radiobox-blank' : 'mdi:checkbox-marked-circle';
  }
};