nixos-configs/modules/ags/config/ts/bar/items/razer-stats.ts

import { execAsync, interval } from 'resource:///com/github/Aylur/ags/utils.js';

const { Box, Icon, Label } = Widget;

const RAZER_POLL = 10000;
const LOW_BATT = 20;

const RazerBat = Variable({
    battery: 0,
    charging: false,
    sleeping: false,
    disconnected: true,
});


const fetchInfo = () => {
    execAsync([
        'bash',
        '-c',
        "polychromatic-cli -n 'Razer Naga Pro (Wired)' -k" +
        ' || ' +
        "polychromatic-cli -n 'Razer Naga Pro (Wireless)' -k",

    ]).then((out) => {
        const batteryMatches = out.split('\n')
            .filter((i) => i.includes('Battery'))[0]
            .match(/[0-9]+/);

        let sleeping = false;
        let battery = batteryMatches !== null ?
            parseInt(batteryMatches[0]) :
            0;

        // Don't set to zero when in sleep mode
        if (battery === 0 && RazerBat.value.battery > 10) {
            battery = RazerBat.value.battery;
            sleeping = true;
        }

        // If 'Wireless' isn't in the logs it means the cmd found 'Wired'
        const charging = !out.includes('Wireless');

        RazerBat.value = {
            battery,
            charging,
            sleeping,
            disconnected: false,
        };
    }).catch(() => {
        RazerBat.value.disconnected = true;
    });
};

interval(RAZER_POLL, fetchInfo);

export default () => {
    const percentage = Label({ vpack: 'center' });

    const icon = Icon({ hpack: 'start' })
        .hook(RazerBat, (self) => {
            const v = RazerBat.value;

            percentage.visible = !(v.disconnected || v.charging);
            percentage.label = `${v.battery}%`;

            self.icon = v.disconnected ?
                'content-loading-symbolic' :
                'mouse-razer-symbolic';
            self.setCss(
                v.disconnected || v.charging ?
                    'margin-right: -5px;' :
                    '',
            );

            self.toggleClassName(
                'high',
                v.battery > 66,
            );
            self.toggleClassName(
                'medium',
                v.battery > LOW_BATT && v.battery <= 66,
            );
            self.toggleClassName(
                'low',
                v.battery <= LOW_BATT,
            );
        });

    return Box({
        class_name: 'razer',
        children: [icon, percentage],
    });
};