From 75d442923300632663c84dde0bc1389bf5a97efb Mon Sep 17 00:00:00 2001
From: ia74 <68617740+ia74@users.noreply.github.com>
Date: Tue, 12 Aug 2025 17:31:53 -0500
Subject: [PATCH] more granularity on what the activity is

---
 custom_components/roomba_rest980/vacuum.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/custom_components/roomba_rest980/vacuum.py b/custom_components/roomba_rest980/vacuum.py
index b1c5127..fb4cbc2 100644
--- a/custom_components/roomba_rest980/vacuum.py
+++ b/custom_components/roomba_rest980/vacuum.py
@@ -63,13 +63,14 @@ class RoombaVacuum(CoordinatorEntity, StateVacuumEntity):
             self._attr_activity = VacuumActivity.IDLE
         if not_ready and not_ready > 0:
             self._attr_activity = VacuumActivity.ERROR
-        if cycle in ["clean", "quick", "spot", "train"]:
+        if cycle in ["clean", "quick", "spot", "train"] or phase in {"hwMidMsn"}:
             self._attr_activity = VacuumActivity.CLEANING
-        if cycle in ["evac", "dock"]:  # Emptying Roomba Bin to Dock, Entering Dock
+        if cycle in ["evac", "dock"] or phase in {
+            "charge",
+        }:  # Emptying Roomba Bin to Dock, Entering Dock
             self._attr_activity = VacuumActivity.DOCKED
         if phase in {
             "hmUsrDock",
-            "hwMidMsn",
             "hmPostMsn",
         }:  # Sent Home, Mid Dock, Final Dock
             self._attr_activity = VacuumActivity.RETURNING