mambatray/mambatray.py

#!/usr/bin/python3
# mambatray 2
# Copyright (C) 2011-2024 by Silvan Calarco
# Copyright (C) 2011 by michiamophil
# Distributed under the terms of the GPL version 3 FLOSS License
#
import sys
import os
import gettext
import distro
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QIcon, QCursor
from PyQt5.QtWidgets import QSystemTrayIcon, QApplication, QMenu, QMessageBox
from PyQt5.QtCore import QTimer, QProcess

DATADIR = os.path.dirname(os.path.realpath((__file__))) + "/"


class SystemTrayIcon(QSystemTrayIcon):

    menu = None

    def __init__(self, icon):
        QSystemTrayIcon.__init__(self, icon)

        global w, contatore, srpm, lista, NetsrpmsArg, nOre, desktop_session
        nOre = 2
        srpm = [
            "skype", "win32codecs", "msttcorefonts", "b43-firmware",
            "virtualbox-extension-pack", "chromium-widevine", "spotify"
        ]
        NetsrpmsArg = "-c"
        lista = []
        contatore = 0
        self.menu = QMenu()

        try:
            desktop_session = os.path.basename(os.getenv('DESKTOP_SESSION'))
        except:
            desktop_session = 'default'

        if desktop_session == 'default':
            desktop_session = 'plasma'
        elif desktop_session[:3] == 'kde':
            desktop_session = 'kde'

        # refresh packages cache timer
        global refreshPackagesTimer, refreshPackagesCacheProcess,\
            checkSRPMSUpdatesTimer, SRPMUpdateProcess, networkOnlineProcess
        refreshPackagesTimer = QTimer(self)
        checkSRPMSUpdatesTimer = QTimer(self)
        refreshPackagesCacheProcess = QProcess(self)
        SRPMUpdateProcess = QProcess(self)
        networkOnlineProcess = QProcess(self)
        refreshPackagesTimer.timeout.connect(self.refreshPackagesCache)
        refreshPackagesCacheProcess.finished.connect(self.restartRefreshPackagesTimer)
        checkSRPMSUpdatesTimer.timeout.connect(self.SRPMSCheckUpgradeList)

        # software menu
        global softwareManagerProcess, softwareCheckUpgradesProcess,\
            softwareBaseInstallsProcess
        softwareManagerProcess = QProcess(self)
        softwareCheckUpgradesProcess = QProcess(self)
        softwareBaseInstallsProcess = QProcess(self)

        softwareMenu = self.menu.addMenu(QIcon.fromTheme("applications-system"),_("Software"))

        softwareManageAction = softwareMenu.addAction(QIcon.fromTheme("applications-other"),_("Add/Remove software packages..."))
        softwareManageAction.triggered.connect(self.softwareManager)

        softwareCheckUpgradesAction = softwareMenu.addAction(QIcon.fromTheme("system-software-update"),_("Check for updates now..."))
        softwareCheckUpgradesAction.triggered.connect(self.softwareCheckUpgrades)

        softwareBaseInstallsAction = softwareMenu.addAction(QIcon.fromTheme("mambabase"),_("{} components...").format(distro.name()))
        softwareBaseInstallsAction.triggered.connect(self.softwareBaseInstalls)

        # network menu
        global networkFirewallDisableProcess, networkFirewallEnableProcess,\
            networkFirewallConfigureProcess, networkProxyProcess
        networkFirewallDisableProcess = QProcess(self)
        networkFirewallEnableProcess = QProcess(self)
        networkFirewallConfigureProcess = QProcess(self)
        networkProxyProcess = QProcess(self)

        networkMenu = self.menu.addMenu(QIcon.fromTheme("preferences-system-network"),
            _("Network"))

        networkFirewallMenu = networkMenu.addMenu(QIcon.fromTheme("security-medium"),_("Firewall"))
        networkFirewallDisableAction = networkFirewallMenu.addAction(QIcon.fromTheme("security-low"),_("Disable"))
        networkFirewallDisableAction.triggered.connect(self.networkFirewallDisable)
        networkFirewallDisableProcess.finished.connect(self.networkFirewallDisableMessage)
        networkFirewallEnableAction = networkFirewallMenu.addAction(QIcon.fromTheme("security-high"),_("Enable"))
        networkFirewallEnableAction.triggered.connect(self.networkFirewallEnable)
        networkFirewallEnableProcess.finished.connect(self.networkFirewallEnableMessage)
        networkFirewallConfigureAction = networkFirewallMenu.addAction(QIcon.fromTheme("security-medium"),_("Configure..."))
        networkFirewallConfigureAction.triggered.connect(self.networkFirewallConfigure)
        networkProxyAction = networkMenu.addAction(QIcon.fromTheme("network-server"),_("Proxy..."))
        networkProxyAction.triggered.connect(self.networkProxy)
        SRPMUpdateProcess.finished.connect(self.SRPMSCheckUpgrade)

        # about box
        aboutAction = self.menu.addSeparator()
        aboutAction = self.menu.addAction(QIcon.fromTheme("system-logo"),_("About"))
        aboutAction.triggered.connect(self.about)

        self.setToolTip(_("{} control center").format(distro.name()))
        self.activated.connect(self.iconActivated)
        self.setContextMenu(self.menu)

        # the first time refresh packages cache after 5 minutes
        refreshPackagesTimer.start( 5 * 60000 )
        # the first time check for SRPMS updates after 4 minutes
        checkSRPMSUpdatesTimer.start( 4 * 60000 )

    def iconActivated(self, reason):
        if reason in (QSystemTrayIcon.Trigger, QSystemTrayIcon.DoubleClick, QSystemTrayIcon.Context):
            # FIXME: on plasma wayland and left click menu will popup in the center of the screen
            # disabling and only supporting right click.
            if desktop_session != 'plasmawayland':
                self.contextMenu().popup(QCursor.pos())

    def about(self):
        aboutBox = QMessageBox(self.menu)
        aboutBox.setWindowIcon(QIcon.fromTheme("mambatray"))
        aboutBox.setWindowTitle(_("About"))
        aboutBox.setText(
                      "<center><b>"+_("{} control center").format(distro.name()+"</b>"
                      "<br/><br/>"+distro.name(pretty=True)+
                      "<br/><br/><a href=\""+distro.os_release_info()['home_url']+
                      "\">" + distro.os_release_info()['home_url'] + "</a>"
                      "</center>"))
        aboutBox.setTextFormat(Qt.RichText)
        aboutBox.setIconPixmap(QIcon.fromTheme("system-logo").pixmap(32));
        aboutBox.setWindowModality(Qt.NonModal);
        aboutBox.exec()

    def networkOnline(self):
        return networkOnlineProcess.execute("/usr/bin/nm-online",['-q','-x'])

    def softwareManager(self):
        softwareManagerProcess.execute("/usr/bin/plasma-discover")

    def softwareCheckUpgrades(self):
        softwareCheckUpgradesProcess.start("/usr/bin/pkcon", ['refresh', 'force'])

    def refreshPackagesCache(self):
        refreshPackagesTimer.stop()
        refreshPackagesCacheProcess.start("/usr/bin/pkcon", ['refresh', 'force'])

    def restartRefreshPackagesTimer(self,status):
        # check every 4 hours and 31 minutes */
        refreshPackagesTimer.start( (4 * 60 + 31) * 60000)

    def softwareBaseInstalls(self):
        softwareBaseInstallsProcess.start("/usr/bin/mambabase")

    def networkFirewallDisable(self):
        networkFirewallDisableProcess.start("sudo",['/usr/sbin/ufw','disable'])

    def networkFirewallDisableMessage(self,status):
        if (status == QProcess.NormalExit):
            self.showMessage("Firewall", _("Firewall disabled in current session"))
        else:
            self.showMessage("Firewall", _("Could not disable firewall"))

    def networkFirewallEnable(self):
        networkFirewallEnableProcess.start("sudo",['/usr/sbin/ufw','enable'])

    def networkFirewallEnableMessage(self,status):
        if (status == QProcess.NormalExit):
            self.showMessage("Firewall", _("Firewall enabled in current session"))
        else:
            self.showMessage("Firewall", _("Could not enable firewall"))

    def networkFirewallConfigure(self):
        if desktop_session == 'kde' or desktop_session.startswith('plasma'):
            networkFirewallConfigureProcess.execute("/usr/bin/kcmshell5", list(set(['firewall'])))
        else:
            networkFirewallConfigureProcess.execute("/usr/bin/gufw")

    def networkProxy(self):
        if desktop_session.startswith('plasma'):
            networkProxyProcess.execute("/usr/bin/kcmshell5", list(set(['proxy'])))
        elif desktop_session == 'kde':
            networkProxyProcess.execute("/opt/kde/bin/kcmshell4", list(set(['proxy'])))
        elif desktop_session == 'gnome':
            networkProxyProcess.execute("/usr/bin/gnome-control-center", "network")

    def SRPMSCheckUpgradeList(self):
        checkSRPMSUpdatesTimer.stop()
        global contatore, lista, NetsrpmsArg
        if (self.networkOnline() == 0):
            try:
                SRPMUpdateProcess.start("/usr/bin/openmamba-netsrpms",[NetsrpmsArg,srpm[contatore]])
            except:
                contatore = 0
                if NetsrpmsArg == '-c':
                    self.SRPMSAskAndInstall()
                else:
                    lista = []
                    NetsrpmsArg = '-c'
                    checkSRPMSUpdatesTimer.start((nOre * 60) * 60000)

    # run for each netsrpm at the end of SRPMUpdateProcess
    def SRPMSCheckUpgrade(self, status):
        global contatore, lista
        if status == 1:
            lista.append(srpm[contatore])
        contatore += 1
        self.SRPMSCheckUpgradeList()

    # netsrpm updates ask and install
    def SRPMSAskAndInstall(self):
        global NetsrpmsArg, lista
        if not lista == []:
            Lista = self.Lista()
            msg = QMessageBox.question(self.menu,
                (_("Not Open Source {} components").format(distro.name())),
                (_("The following components are available for update: ") +
                "\n\n" + Lista + "\n\n" + _("Ok to proceed?")),
                QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
            if msg == QMessageBox.Yes:
                NetsrpmsArg = '-u'
                self.SRPMSCheckUpgradeList()
            else:
                lista = []
                checkSRPMSUpdatesTimer.start((nOre * 60) * 60000)
        else:
            checkSRPMSUpdatesTimer.start((nOre * 60) * 60000)

    # starting from the list "lista" returns the list in readable format
    # FIXME: make this code more clear and clean
    def Lista(self):
        testo = ''
        for a in lista:
            testo = testo + a + ', '
        return testo[0:len(testo)-2]


if __name__ == '__main__':
    app = QApplication(sys.argv)
    if app.isSessionRestored():
        app.exit(1)
    else:
        app.setQuitOnLastWindowClosed(0)
        gettext.install('mambatray', '/usr/share/locale')
        trayIcon = SystemTrayIcon(QIcon.fromTheme("mambatray"))
        trayIcon.show()
        sys.exit(app.exec())