From 02d3202026d799c6ad6f09fa8881841b23d18ee0 Mon Sep 17 00:00:00 2001
From: tronikos <tronikos@users.noreply.github.com>
Date: Sat, 7 Feb 2026 02:08:28 -0800
Subject: [PATCH] add deprecated functions

---
 esphome/components/water_heater/water_heater.h | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/esphome/components/water_heater/water_heater.h b/esphome/components/water_heater/water_heater.h
index 6677de7d36..2ea97e7c4c 100644
--- a/esphome/components/water_heater/water_heater.h
+++ b/esphome/components/water_heater/water_heater.h
@@ -92,6 +92,21 @@ class WaterHeaterCall {
   const optional<bool> &get_away() const { return this->away_; }
   const optional<bool> &get_on() const { return this->on_; }
 
+  ESPDEPRECATED("set_state() is deprecated, use set_on() and set_away() instead. (Removed in 2026.8.0)", "2026.2.0")
+  void set_state(uint32_t state) {
+    this->set_away((state & WATER_HEATER_STATE_AWAY) != 0);
+    this->set_on((state & WATER_HEATER_STATE_ON) != 0);
+  }
+  ESPDEPRECATED("get_state() is deprecated, use is_on() and is_away() instead. (Removed in 2026.8.0)", "2026.2.0")
+  uint32_t get_state() const {
+    uint32_t state = 0;
+    if (this->away_.value_or(false))
+      state |= WATER_HEATER_STATE_AWAY;
+    if (this->on_.value_or(false))
+      state |= WATER_HEATER_STATE_ON;
+    return state;
+  }
+
  protected:
   void validate_();
   WaterHeater *parent_;