hyprpanel

ags/HyprPanel/customModules/storage/computeStorage.ts

@@ -0,0 +1,28 @@
+// @ts-expect-error is a special directive that tells the compiler to use the GTop library
+import GTop from 'gi://GTop';
+
+import { divide } from 'customModules/utils';
+import { Variable as VariableType } from 'types/variable';
+import { GenericResourceData } from 'lib/types/customModules/generic';
+
+export const computeStorage = (round: VariableType<boolean>): GenericResourceData => {
+    try {
+        const currentFsUsage = new GTop.glibtop_fsusage();
+
+        GTop.glibtop_get_fsusage(currentFsUsage, '/');
+
+        const total = currentFsUsage.blocks * currentFsUsage.block_size;
+        const available = currentFsUsage.bavail * currentFsUsage.block_size;
+        const used = total - available;
+
+        return {
+            total,
+            used,
+            free: available,
+            percentage: divide([total, used], round.value),
+        };
+    } catch (error) {
+        console.error('Error calculating RAM usage:', error);
+        return { total: 0, used: 0, percentage: 0, free: 0 };
+    }
+};

ags/HyprPanel/customModules/storage/index.ts

@@ -0,0 +1,68 @@
+import options from 'options';
+import { module } from '../module';
+
+import { formatTooltip, inputHandler, renderResourceLabel } from 'customModules/utils';
+import { computeStorage } from './computeStorage';
+import { BarBoxChild, ResourceLabelType } from 'lib/types/bar';
+import { GenericResourceData } from 'lib/types/customModules/generic';
+import Button from 'types/widgets/button';
+import { LABEL_TYPES } from 'lib/types/defaults/bar';
+import { pollVariable } from 'customModules/PollVar';
+import { Attribute, Child } from 'lib/types/widget';
+
+const { label, labelType, icon, round, leftClick, rightClick, middleClick, pollingInterval } =
+    options.bar.customModules.storage;
+
+const defaultStorageData = { total: 0, used: 0, percentage: 0, free: 0 };
+
+const storageUsage = Variable<GenericResourceData>(defaultStorageData);
+
+pollVariable(storageUsage, [round.bind('value')], pollingInterval.bind('value'), computeStorage, round);
+
+export const Storage = (): BarBoxChild => {
+    const storageModule = module({
+        textIcon: icon.bind('value'),
+        label: Utils.merge(
+            [storageUsage.bind('value'), labelType.bind('value'), round.bind('value')],
+            (storage: GenericResourceData, lblTyp: ResourceLabelType, round: boolean) => {
+                return renderResourceLabel(lblTyp, storage, round);
+            },
+        ),
+        tooltipText: labelType.bind('value').as((lblTyp) => {
+            return formatTooltip('Storage', lblTyp);
+        }),
+        boxClass: 'storage',
+        showLabelBinding: label.bind('value'),
+        props: {
+            setup: (self: Button<Child, Attribute>) => {
+                inputHandler(self, {
+                    onPrimaryClick: {
+                        cmd: leftClick,
+                    },
+                    onSecondaryClick: {
+                        cmd: rightClick,
+                    },
+                    onMiddleClick: {
+                        cmd: middleClick,
+                    },
+                    onScrollUp: {
+                        fn: () => {
+                            labelType.value = LABEL_TYPES[
+                                (LABEL_TYPES.indexOf(labelType.value) + 1) % LABEL_TYPES.length
+                            ] as ResourceLabelType;
+                        },
+                    },
+                    onScrollDown: {
+                        fn: () => {
+                            labelType.value = LABEL_TYPES[
+                                (LABEL_TYPES.indexOf(labelType.value) - 1 + LABEL_TYPES.length) % LABEL_TYPES.length
+                            ] as ResourceLabelType;
+                        },
+                    },
+                });
+            },
+        },
+    });
+
+    return storageModule;
+};