Add color_mode field which should be supplied for HomeAssistant

Author: sidoh

dist/index.html.gz.h

@@ -104,30 +104,16 @@ void HomeAssistantDiscoveryClient::addConfig(const char* alias, const BulbId& bu
       break;
   }
 
-  // These bulbs support RGB color
-  switch (bulbId.deviceType) {
-    case REMOTE_TYPE_FUT089:
-    case REMOTE_TYPE_RGB:
-    case REMOTE_TYPE_RGB_CCT:
-    case REMOTE_TYPE_RGBW:
-      config[F("rgb")] = true;
-      break;
-    default:
-      break; //nothing
+  // Flag RGB support
+  if (MiLightRemoteTypeHelpers::supportsRgb(bulbId.deviceType)) {
+    config[F("rgb")] = true;
   }
 
-  // These bulbs support adjustable white values
-  switch (bulbId.deviceType) {
-    case REMOTE_TYPE_CCT:
-    case REMOTE_TYPE_FUT089:
-    case REMOTE_TYPE_FUT091:
-    case REMOTE_TYPE_RGB_CCT:
-      config[GroupStateFieldNames::COLOR_TEMP] = true;
-      config[F("max_mirs")] = COLOR_TEMP_MAX_MIREDS;
-      config[F("min_mirs")] = COLOR_TEMP_MIN_MIREDS;
-      break;
-    default:
-      break; //nothing
+  // Flag adjustable color temp support
+  if (MiLightRemoteTypeHelpers::supportsColorTemp(bulbId.deviceType)) {
+    config[GroupStateFieldNames::COLOR_TEMP] = true;
+    config[F("max_mirs")] = COLOR_TEMP_MAX_MIREDS;
+    config[F("min_mirs")] = COLOR_TEMP_MIN_MIREDS;
   }
 
   String message;

lib/MQTT/HomeAssistantDiscoveryClient.cpp

@@ -241,6 +241,7 @@ bool GroupState::isSetField(GroupStateField field) const {
     case GroupStateField::COLOR_TEMP:
       return isSetKelvin();
     case GroupStateField::BULB_MODE:
+    case GroupStateField::COLOR_MODE:
       return isSetBulbMode();
     default:
       Serial.print(F("WARNING: tried to check if unknown field was set: "));
@@ -855,6 +856,29 @@ void GroupState::applyField(JsonObject partialState, const BulbId& bulbId, Group
         partialState[GroupStateFieldNames::BULB_MODE] = BULB_MODE_NAMES[getBulbMode()];
         break;
 
+      // For HomeAssistant. Should report:
+      //   1. "brightness" if no color temp/rgb support
+      //   2. "rgb" if RGB or RGBW bulb and in color mode
+      //   3. "color_temp" if WW or RGBW and in color temp mode
+      //   4. "onoff" if in night mode
+      case GroupStateField::COLOR_MODE:
+        if (
+          MiLightRemoteTypeHelpers::supportsRgb(bulbId.deviceType) 
+          && getBulbMode() == BULB_MODE_COLOR
+        ) {
+          partialState[GroupStateFieldNames::COLOR_MODE] = F("rgb");
+        } else if (
+          MiLightRemoteTypeHelpers::supportsColorTemp(bulbId.deviceType) 
+          && getBulbMode() == BULB_MODE_WHITE
+        ) {
+          partialState[GroupStateFieldNames::COLOR_MODE] = F("color_temp");
+        } else if (getBulbMode() == BULB_MODE_NIGHT) {
+          partialState[GroupStateFieldNames::COLOR_MODE] = F("onoff");
+        } else {
+          partialState[GroupStateFieldNames::COLOR_MODE] = F("brightness");
+        }
+        break;
+
       case GroupStateField::COLOR:
         if (getBulbMode() == BULB_MODE_COLOR) {
           applyColor(partialState);

lib/MiLightState/GroupState.cpp

@@ -20,7 +20,8 @@ static const char* STATE_NAMES[] = {
   GroupStateFieldNames::GROUP_ID,
   GroupStateFieldNames::DEVICE_TYPE,
   GroupStateFieldNames::OH_COLOR,
-  GroupStateFieldNames::HEX_COLOR
+  GroupStateFieldNames::HEX_COLOR,
+  GroupStateFieldNames::COLOR_MODE,
 };
 
 GroupStateField GroupStateFieldHelpers::getFieldByName(const char* name) {

lib/Types/GroupStateField.cpp

@@ -24,6 +24,9 @@ namespace GroupStateFieldNames {
   static const char HEX_COLOR[] = "hex_color";
   static const char COMMAND[] = "command";
   static const char COMMANDS[] = "commands";
+
+  // For use with HomeAssistant
+  static const char COLOR_MODE[] = "color_mode";
 };
 
 enum class GroupStateField {
@@ -45,7 +48,8 @@ enum class GroupStateField {
   GROUP_ID,
   DEVICE_TYPE,
   OH_COLOR,
-  HEX_COLOR
+  HEX_COLOR,
+  COLOR_MODE,
 };
 
 class GroupStateFieldHelpers {

lib/Types/GroupStateField.h

@@ -65,4 +65,28 @@ const String MiLightRemoteTypeHelpers::remoteTypeToString(const MiLightRemoteTyp
       Serial.println(type);
       return "unknown";
   }
+}
+
+const bool MiLightRemoteTypeHelpers::supportsRgb(const MiLightRemoteType type) {
+  switch (type) {
+    case REMOTE_TYPE_FUT089:
+    case REMOTE_TYPE_RGB:
+    case REMOTE_TYPE_RGB_CCT:
+    case REMOTE_TYPE_RGBW:
+      return true;
+    default:
+      return false;
+  }
+}
+
+const bool MiLightRemoteTypeHelpers::supportsColorTemp(const MiLightRemoteType type) {
+  switch (type) {
+    case REMOTE_TYPE_CCT:
+    case REMOTE_TYPE_FUT089:
+    case REMOTE_TYPE_FUT091:
+    case REMOTE_TYPE_RGB_CCT:
+      return true;
+    default:
+      return false;
+  }
 }

lib/Types/MiLightRemoteType.cpp

@@ -17,4 +17,6 @@ class MiLightRemoteTypeHelpers {
 public:
   static const MiLightRemoteType remoteTypeFromString(const String& type);
   static const String remoteTypeToString(const MiLightRemoteType type);
+  static const bool supportsRgb(const MiLightRemoteType type);
+  static const bool supportsColorTemp(const MiLightRemoteType type);
 };

lib/Types/MiLightRemoteType.h

@@ -390,7 +390,8 @@ var GROUP_STATE_KEYS = [
   "group_id",
   "device_type",
   "oh_color",
-  "hex_color"
+  "hex_color",
+  "color_mode"
 ];
 
 var LED_MODES = [