From cf7bf010f9b940c4cfe15162894142abc7404abe Mon Sep 17 00:00:00 2001
From: TolaMironcenko <mironcenkotola@gmail.com>
Date: Tue, 18 Mar 2025 22:26:46 +0700
Subject: [PATCH] added battery health and cycles count

---
 widget/menus/BatteryMenu/BatteryMenu.tsx | 18 +++++-------
 widget/menus/BatteryMenu/helpers.ts      | 37 ++++++++++++++++++++++++
 2 files changed, 45 insertions(+), 10 deletions(-)
 create mode 100644 widget/menus/BatteryMenu/helpers.ts

diff --git a/widget/menus/BatteryMenu/BatteryMenu.tsx b/widget/menus/BatteryMenu/BatteryMenu.tsx
index d71f41c..f7cd282 100644
--- a/widget/menus/BatteryMenu/BatteryMenu.tsx
+++ b/widget/menus/BatteryMenu/BatteryMenu.tsx
@@ -1,17 +1,15 @@
 import { Astal, App, Gtk } from "astal/gtk3";
 import Battery from "gi://AstalBattery";
-import { bind } from "astal";
+import { bind, Variable } from "astal";
+import {cycles_count, health} from "./helpers.ts";
 
 const BatteryMenu = () => {
   const { TOP, RIGHT } = Astal.WindowAnchor;
-  const { START, END } = Gtk.Align;
+  const { START, END, CENTER } = Gtk.Align;
   const bat = Battery.get_default();
-  console.log(
-    bat.charging,
-    bat.capacity,
-    bat.energyFull,
-    bat.energy_full_design,
-  );
+  const batcycles = Variable<string>("").poll(5000, () => cycles_count());
+  const bathealth = Variable<string>("").poll(5000, () => health());
+  
   return (
     <window
       anchor={TOP | RIGHT}
@@ -55,7 +53,7 @@ const BatteryMenu = () => {
               halign={START}
               label="Health:"
             />
-            <label hexpand halign={END} label="92.5 %" />
+            <label hexpand halign={END} label={bathealth()} />
           </box>
           <box>
             <label
@@ -64,7 +62,7 @@ const BatteryMenu = () => {
               halign={START}
               label="Charge cycles:"
             />
-            <label hexpand halign={END} label="325" />
+            <label hexpand halign={END} label={batcycles()} />
           </box>
           <box>
             <label
diff --git a/widget/menus/BatteryMenu/helpers.ts b/widget/menus/BatteryMenu/helpers.ts
new file mode 100644
index 0000000..4968406
--- /dev/null
+++ b/widget/menus/BatteryMenu/helpers.ts
@@ -0,0 +1,37 @@
+import { GLib } from "astal"
+
+const cycles_count = (): string => {
+  try {
+  const [success, cycles] = GLib.file_get_contents("/sys/class/power_supply/BAT0/cycle_count");
+  if (!success || !cycles) {
+    throw new Error("Failed to read /sys/class/ or the file content is null.");
+  }
+  const cyclesinfo = new TextDecoder("utf-8").decode(cycles);
+  return cyclesinfo.replace(/\n$/, '');
+  } catch (error) {
+    console.error("Error get battery cycles_count");
+    return "";
+  }
+}
+
+const health = (): string => {
+  try {
+    const [success, charge_full] = GLib.file_get_contents("/sys/class/power_supply/BAT0/charge_full");
+    if (!success || !charge_full) {
+      throw new Error("Failed to read /sys/class/ or the file content is null.");
+    }
+    const [success_chagre_full_design, charge_full_design] = GLib.file_get_contents("/sys/class/power_supply/BAT0/charge_full_design");
+    if (!success_chagre_full_design || !charge_full_design) {
+      throw new Error("Failed to read /sys/class/ or the file content is null.");
+    }
+    const chargefullnow = parseInt(new TextDecoder("utf-8").decode(charge_full));
+    const chargefulldesign = parseInt(new TextDecoder("utf-8").decode(charge_full_design));
+    const percents = Math.round(chargefullnow * 100 / chargefulldesign);
+    return percents.toString() + " %";
+  } catch (error) {
+    console.error("error get battery health");
+    return "";
+  }
+}
+
+export {cycles_count, health};