extension.js

/*
 *
 *  CPUPower for GNOME Shell preferences
 *  - Creates a widget to set the preferences of the cpupower extension
 *
 * Copyright (C) 2015
 *     Martin Koppehel <martin.koppehel@st.ovgu.de>
 *
 * This file is part of the gnome-shell extension cpupower.
 *
 * gnome-shell extension cpupower 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.
 *
 * gnome-shell extension cpupower 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 gnome-shell extension cpupower.  If not, see
 * <http://www.gnu.org/licenses/>.
 *
 */

const ExtensionUtils = imports.misc.extensionUtils;
const Me = ExtensionUtils.getCurrentExtension();
const EXTENSIONDIR = Me.dir.get_path();
const Convenience = Me.imports.src.convenience;
const Main = imports.ui.main;
const Config = imports.misc.config;
const ByteArray = imports.byteArray;

const utils = Me.imports.src.utils;
const checkInstalled = Me.imports.src.utils.checkInstalled;
const notinstalled = Me.imports.src.notinstalled;
const update = Me.imports.src.update;
const indicator = Me.imports.src.indicator;

const Gio = imports.gi.Gio;
const GLib = imports.gi.GLib;

let indicatorInstance;
/* exported init */
function init(_meta) {
    Convenience.initTranslations("gnome-shell-extension-cpupower");
}

function enableIndicator(instance) {
    Main.panel.addToStatusArea("cpupower", instance.mainButton);
    instance.enable();
}

let cpupowerProxy;
let extensionReloadSignalHandler;
/* exported enable */
function enable() {
    const interfaceBinary = GLib.file_get_contents(`${EXTENSIONDIR}/schemas/io.github.martin31821.cpupower.dbus.xml`)[1];
    let interfaceXml;
    if (parseFloat(Config.PACKAGE_VERSION.substring(0, 4)) > 3.28) {
        interfaceXml = ByteArray.toString(interfaceBinary);
    } else {
        interfaceXml = interfaceBinary.toString();
    }
    const CpupowerProxy = Gio.DBusProxy.makeProxyWrapper(interfaceXml);

    cpupowerProxy = new CpupowerProxy(
        Gio.DBus.session,
        "io.github.martin31821.cpupower",
        "/io/github/martin31821/cpupower",
    );

    extensionReloadSignalHandler = cpupowerProxy.connectSignal("ExtensionReloadRequired", () => {
        log("Reloading cpupower");
        disable();
        enable();
    });

    try {
        checkInstalled((installed, exitCode) => {
            if (!installed) {
                switch (exitCode) {
                case utils.INSTALLER_NEEDS_UPDATE:
                    indicatorInstance = new update.UpdateIndicator(update.UPDATE, function (success) {
                        if (success) {
                            // reenable the extension to allow immediate operation.
                            disable();
                            enable();
                        }
                    }, (inst) => enableIndicator(inst));
                    break;
                case utils.INSTALLER_NEEDS_SECURITY_UPDATE:
                    indicatorInstance = new update.UpdateIndicator(update.SECURITY_UPDATE, function (success) {
                        if (success) {
                            // reenable the extension to allow immediate operation.
                            disable();
                            enable();
                        }
                    }, (inst) => enableIndicator(inst));
                    break;
                default:
                    indicatorInstance = new notinstalled.NotInstalledIndicator(exitCode, function (success) {
                        if (success) {
                            // reenable the extension to allow immediate operation.
                            disable();
                            enable();
                        }
                    }, (inst) => enableIndicator(inst));
                    break;
                }
            } else {
                indicatorInstance = new indicator.CPUFreqIndicator((inst) => enableIndicator(inst));
            }
        });
    } catch (e) {
        logError(e.message);
    }
}

/* exported disable */
function disable() {
    if (indicatorInstance) {
        indicatorInstance.disable();
        indicatorInstance.destroy();
    }

    if (cpupowerProxy && extensionReloadSignalHandler) {
        cpupowerProxy.disconnectSignal(extensionReloadSignalHandler);

        cpupowerProxy = null;
        extensionReloadSignalHandler = null;
    }
}