chore: minor

Mungg1818 · Aug 5, 2024 · 7453d12 · 7453d12
1 parent 314acde
commit 7453d12
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/core/site_battery.go b/core/site_battery.go
@@ -40,11 +40,15 @@ func (site *Site) requiredBatteryMode(batteryGridChargeActive bool, rate api.Rat
 	var res api.BatteryMode
 	batMode := site.GetBatteryMode()
 
+	mapper := func(s api.BatteryMode) api.BatteryMode {
+		return map[bool]api.BatteryMode{false: s, true: api.BatteryUnknown}[batMode == s]
+	}
+
 	switch {
 	case batteryGridChargeActive:
-		res = map[bool]api.BatteryMode{false: api.BatteryCharge, true: api.BatteryUnknown}[batMode == api.BatteryCharge]
-	case !batteryGridChargeActive && site.dischargeControlActive(rate) && batMode != api.BatteryHold:
-		res = api.BatteryHold
+		res = mapper(api.BatteryCharge)
+	case site.dischargeControlActive(rate):
+		res = mapper(api.BatteryHold)
 	case batteryModeModified(batMode):
 		res = api.BatteryNormal
 	}