From c0dffe2012d13e57b9d2042fece5a427cec92adf Mon Sep 17 00:00:00 2001
From: NiLuJe <ninuje@gmail.com>
Date: Fri, 7 Oct 2022 17:33:57 +0200
Subject: [PATCH] Kobo: Add a debug option to turn the charging LED on PM entry
 failure (#9601)

This should be fairly unintrusive, but still noticeable enough without having to watch the battery gauge like a hawk, or resort to actual monitoring via kernel logs and/or usb meter and/or multimeter...
---
 frontend/apps/filemanager/filemanagermenu.lua | 11 +++++
 frontend/device/kobo/device.lua               | 41 +++++++++++++++++--
 2 files changed, 48 insertions(+), 4 deletions(-)

diff --git a/frontend/apps/filemanager/filemanagermenu.lua b/frontend/apps/filemanager/filemanagermenu.lua
index 369b683af..32fa8fdb9 100644
--- a/frontend/apps/filemanager/filemanagermenu.lua
+++ b/frontend/apps/filemanager/filemanagermenu.lua
@@ -705,6 +705,17 @@ To:
             FontList:dumpFontList()
         end,
     })
+    if Device:isKobo() and Device:canToggleChargingLED() then
+        table.insert(self.menu_items.developer_options.sub_item_table, {
+            text = _("Turn on the LED on PM entry failure"),
+            checked_func = function()
+                return G_reader_settings:isTrue("pm_debug_entry_failure")
+            end,
+            callback = function()
+                G_reader_settings:toggle("pm_debug_entry_failure")
+            end,
+        })
+    end
 
     self.menu_items.cloud_storage = {
         text = _("Cloud storage"),
diff --git a/frontend/device/kobo/device.lua b/frontend/device/kobo/device.lua
index 7e8b63051..1166ce39a 100644
--- a/frontend/device/kobo/device.lua
+++ b/frontend/device/kobo/device.lua
@@ -232,7 +232,7 @@ local KoboDaylight = Kobo:extend{
 -- Kobo Aura H2O:
 local KoboDahlia = Kobo:extend{
     model = "Kobo_dahlia",
-    canToggleChargingLED = no,  -- Possibly weird interactions with Nickel
+    canToggleChargingLED = yes,
     led_uses_channel_3 = true,
     hasFrontlight = yes,
     touch_phoenix_protocol = true,
@@ -776,6 +776,15 @@ function Kobo:init()
         self.restoreKeyRepeat = NOP
     end
 
+    -- Detect the NTX charging LED sysfs knob
+    if util.pathExists("/sys/devices/platform/ntx_led/lit") then
+        self.ntx_lit_sysfs_knob = "/sys/devices/platform/ntx_led/lit"
+    elseif util.pathExists("/sys/devices/platform/pmic_light.1/lit") then
+        self.ntx_lit_sysfs_knob = "/sys/devices/platform/pmic_light.1/lit"
+    else
+        self.canToggleChargingLED = no
+    end
+
     -- NOP unsupported methods
     if not self:canToggleChargingLED() then
         self.toggleChargingLED = NOP
@@ -1019,8 +1028,16 @@ function Kobo:standby(max_duration)
 
     if ret then
         logger.info("Kobo standby: zZz zZz zZz zZz... And woke up!")
+        if G_reader_settings:isTrue("pm_debug_entry_failure") then
+            -- NOTE: This is a debug option where we coopt the charging LED, hence us not using setupChargingLED here.
+            --       (It's called on resume anyway).
+            self:toggleChargingLED(false)
+        end
     else
         logger.warn("Kobo standby: the kernel refused to enter standby!")
+        if G_reader_settings:isTrue("pm_debug_entry_failure") then
+            self:toggleChargingLED(true)
+        end
     end
 
     if max_duration then
@@ -1123,10 +1140,16 @@ function Kobo:suspend()
 
     if ret then
         logger.info("Kobo suspend: ZzZ ZzZ ZzZ... And woke up!")
+        if G_reader_settings:isTrue("pm_debug_entry_failure") then
+            self:toggleChargingLED(false)
+        end
     else
         logger.warn("Kobo suspend: the kernel refused to enter suspend!")
         -- Reset state-extended back to 0 since we are giving up.
         writeToSys("0", "/sys/power/state-extended")
+        if G_reader_settings:isTrue("pm_debug_entry_failure") then
+            self:toggleChargingLED(true)
+        end
     end
 
     -- NOTE: Ideally, we'd need a way to warn the user that suspending
@@ -1231,17 +1254,27 @@ function Kobo:toggleChargingLED(toggle)
     if toggle == nil then
         return
     end
+    -- Don't do anything if the state is already correct
+    -- NOTE: What happens to the LED when attempting/successfully entering PM is... kind of a mess.
+    --       On a H2O, even *attempting* to enter PM will kill the light (and it'll stay off).
+    --       On a Forma, a failed attempt will *not* affect the light, but a successful one *will* kill it,
+    --       be that standby or suspend, but it'll be restored on wakeup...
+    --       On sunxi, PM appears to have zero effect on the LED.
+    if self.charging_led_state == toggle then
+        return
+    end
+    self.charging_led_state = toggle
+    logger.dbg("Kobo: Turning the charging LED", toggle and "on" or "off")
 
     -- NOTE: While most/all Kobos actually have a charging LED, and it can usually be fiddled with in a similar fashion,
     --       we've seen *extremely* weird behavior in the past when playing with it on older devices (c.f., #5479).
     --       In fact, Nickel itself doesn't provide this feature on said older devices
     --       (when it does, it's an option in the Energy saving settings),
     --       which is why we also limit ourselves to "true" on devices where this was tested.
-    --       FWIW, on older devices, the knob is at "/sys/devices/platform/pmic_light.1/lit".
     -- c.f., drivers/misc/ntx_misc_light.c
-    local f = io.open("/sys/devices/platform/ntx_led/lit", "we")
+    local f = io.open(self.ntx_lit_sysfs_knob, "we")
     if not f then
-        logger.err("cannot open /sys/devices/platform/ntx_led/lit for writing!")
+        logger.err("cannot open", self.ntx_lit_sysfs_knob, "for writing!")
         return false
     end
     -- Relying on LFs is mildly more elegant than spamming f:flush() calls ;).