ideapad gnome-shell extension

%sudo ALL=(ALL) NOPASSWD: /usr/bin/tee /sys/bus/platform/drivers/ideapad_acpi/VPC????\:??/conservation_mode

//Copyright (C) 2019 - 2022 Laurento Frittella

//This file is part of Lenovo IdeaPad goodies.

//Lenovo IdeaPad goodies is free software: you can redistribute it and/or modify

//it under the terms of the GNU General Public License as published by

//the Free Software Foundation, either version 3 of the License, or

//(at your option) any later version.

//Lenovo IdeaPad goodies is distributed in the hope that it will be useful,

//but WITHOUT ANY WARRANTY; without even the implied warranty of

//MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

//GNU General Public License for more details.

//You should have received a copy of the GNU General Public License

//along with Lenovo IdeaPad goodies.  If not, see <http://www.gnu.org/licenses/>.

//Author: Laurento Frittella <laurento.frittella at gmail dot com>

const ExtensionUtils = imports.misc.extensionUtils;

const { Gio, GObject, Shell, St } = imports.gi;

const Gettext = imports.gettext;

const Main = imports.ui.main;

const Me = ExtensionUtils.getCurrentExtension();

const Util = imports.misc.util;

const Domain = Gettext.domain(Me.metadata['gettext-domain']);

const _ = Domain.gettext;

const PanelMenu = imports.ui.panelMenu;

const PopupMenu = imports.ui.popupMenu;

const aggregateMenu = Main.panel.statusArea.aggregateMenu;

const powerIndicator = _getIndicators(aggregateMenu._power);

const powerMenu = aggregateMenu._power.menu.firstMenuItem.menu;

// MANUAL OVERRIDE

// to disable the auto-discovery more, just set the absolute device path here

// E.g.: "/sys/bus/platform/drivers/ideapad_acpi/VPC2004:00/conservation_mode"

let sys_conservation = null;

const BatteryConservationIndicator = GObject.registerClass(

    {

        GTypeName: 'BatteryConservationIndicator'

    },

    class BatteryConservationIndicator extends PanelMenu.SystemIndicator {

        _init() {

            super._init();

            this._monitor = null;

            this._indicator = this._addIndicator();

            this._indicator.icon_name = "emoji-nature-symbolic";

            powerIndicator.add_child(_getIndicators(this));

            if (sys_conservation !== null) {

                this._item = new PopupMenu.PopupSwitchMenuItem(_("Conservation Mode"), true);

                this._item.connect('toggled', item => {

                    BatteryConservationIndicator._setConservationMode(item.state);

                });

                powerMenu.addMenuItem(this._item);

                // Monitor the changes and show or hide the indicator accordingly.

                const fileM = Gio.file_new_for_path(sys_conservation);

                this._monitor = fileM.monitor(Gio.FileMonitorFlags.NONE, null);

                this._monitor.connect('changed', this._syncStatus.bind(this));

                // Set the initial and proper indicator status.

                this._syncStatus();

            } else {

                this._item = powerMenu.addAction(

                    _("Conservation mode is not available"),

                    function () {}

                );

                this._indicator.visible = false;

            }

        }

        _syncStatus() {

            const status = Shell.get_file_contents_utf8_sync(sys_conservation);

            const active = (status.trim() == "1");

            this._indicator.visible = active;

            this._item.setToggleState(active);

        }

        static _setConservationMode(enabled) {

            const new_status = (enabled) ? "1" : "0";

            Util.spawnCommandLine(`/bin/sh -c 'echo ${new_status} | sudo tee ${sys_conservation} >/dev/null'`);

        }

        destroy() {

            this._indicator.destroy();

            this._item.destroy();

            if (this._monitor !== null) this._monitor.cancel();

        }

    }

);

function _getIndicators(delegate) {

    if (delegate instanceof St.BoxLayout) {

        return delegate;

    }

    return delegate.indicators;

}

function _auto_dev_discovery(search_path) {

    let mod_path = Gio.file_new_for_path(search_path);

    let walker = mod_path.enumerate_children(

        "standard::name",

        Gio.FileQueryInfoFlags.NOFOLLOW_SYMLINKS,

        null);

    let child = null;

    let found = null;

    while ((child = walker.next_file(null))) {

        if (child.get_is_symlink() && child.get_name().startsWith("VPC2004")) {

            // ideapad_device_ids[] from the kernel module ideapad_acpi.c

            found = _auto_dev_discovery(`${search_path}/${child.get_name()}`);

        } else if (child.get_name() == "conservation_mode") {

            log(`IdeaPad device FOUND at ${search_path}`);

            found = `${search_path}/${child.get_name()}`;

        }

        // Stop as soon as the device is found.

        if (found !== null) break;

    }

    return found;

}

function init() {

    ExtensionUtils.initTranslations(Me.metadata['gettext-domain']);

}

let batteryConservationIndicator = null;

function enable() {

    let sysfs_path = "/sys/bus/platform/drivers/ideapad_acpi";

    if (sys_conservation === null) {

        try {

            sys_conservation = _auto_dev_discovery(sysfs_path);

            if (sys_conservation === null) {

                throw new Error("Battery conservation mode not available.");

            }

        } catch (e) {

            logError(e, "ideapad");

        }

    }

    batteryConservationIndicator = new BatteryConservationIndicator();

}

function disable() {

    batteryConservationIndicator.destroy();

    batteryConservationIndicator = null;

}