From 3b0f669f4782117661a9603137615f1fae99d128 Mon Sep 17 00:00:00 2001
From: Leonardo Rivera <leodrivera@gmail.com>
Date: Wed, 3 Jun 2026 14:38:00 -0300
Subject: [PATCH] [gree] Fix HEAT_COOL advertised when supports_heat is false;
 restrict YAN swing to vertical (#16199)

---
 esphome/components/gree/gree.cpp | 16 ++++++++++++++++
 esphome/components/gree/gree.h   |  1 +
 2 files changed, 17 insertions(+)

diff --git a/esphome/components/gree/gree.cpp b/esphome/components/gree/gree.cpp
index 705c741dd0f..a794e7721f5 100644
--- a/esphome/components/gree/gree.cpp
+++ b/esphome/components/gree/gree.cpp
@@ -5,7 +5,23 @@ namespace esphome::gree {
 
 static const char *const TAG = "gree.climate";
 
+climate::ClimateTraits GreeClimate::traits() {
+  auto t = climate_ir::ClimateIR::traits();
+  // ClimateIR unconditionally includes HEAT_COOL in the base mode set; remove it when heat is not supported.
+  if (!this->supports_heat_) {
+    auto modes = t.get_supported_modes();
+    modes.erase(climate::CLIMATE_MODE_HEAT_COOL);
+    t.set_supported_modes(modes);
+  }
+  return t;
+}
+
 void GreeClimate::set_model(Model model) {
+  if (model == GREE_YAN) {
+    // YAN only has a vertical vane; the horizontal swing IR bytes are not defined for this model.
+    this->swing_modes_.erase(climate::CLIMATE_SWING_HORIZONTAL);
+    this->swing_modes_.erase(climate::CLIMATE_SWING_BOTH);
+  }
   if (model == GREE_YX1FF) {
     this->fan_modes_.insert(climate::CLIMATE_FAN_QUIET);   // YX1FF 4 speed
     this->presets_.insert(climate::CLIMATE_PRESET_NONE);   // YX1FF sleep mode
diff --git a/esphome/components/gree/gree.h b/esphome/components/gree/gree.h
index 24453750ae3..1eb812ae467 100644
--- a/esphome/components/gree/gree.h
+++ b/esphome/components/gree/gree.h
@@ -94,6 +94,7 @@ class GreeClimate : public climate_ir::ClimateIR {
  protected:
   // Transmit via IR the state of this climate controller.
   void transmit_state() override;
+  climate::ClimateTraits traits() override;
 
   uint8_t operation_mode_();
   uint8_t fan_speed_();