LorenEteval_Furious/Furious/Action/Settings.py

# Copyright (C) 2023  Loren Eteval <loren.eteval@proton.me>
#
# This file is part of Furious.
#
# This program 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.
#
# This program 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 this program.  If not, see <https://www.gnu.org/licenses/>.

from Furious.Gui.Action import Action, Seperator
from Furious.Widget.Widget import Menu
from Furious.Utility.Constants import APP, PLATFORM, ADMINISTRATOR_NAME
from Furious.Utility.Utility import Switch, bootstrapIcon, isAdministrator, isVPNMode
from Furious.Utility.Translator import gettext as _
from Furious.Utility.StartupOnBoot import StartupOnBoot


class SettingsChildAction(Action):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

    def triggeredCallback(self, checked):
        if self.textCompare('Startup On Boot'):
            if checked:
                StartupOnBoot.on_()

                APP().StartupOnBoot = Switch.ON_
            else:
                StartupOnBoot.off()

                APP().StartupOnBoot = Switch.OFF
        if self.textCompare('Show Progress Bar When Connecting'):
            if checked:
                APP().ShowProgressBarWhenConnecting = Switch.ON_
            else:
                APP().ShowProgressBarWhenConnecting = Switch.OFF
        if self.textCompare('Show Tab And Spaces In Editor'):
            if checked:
                APP().ShowTabAndSpacesInEditor = Switch.ON_
            else:
                APP().ShowTabAndSpacesInEditor = Switch.OFF

            # Reference
            ServerWidget = APP().ServerWidget

            currentFocus = ServerWidget.currentFocus

            if currentFocus >= 0:
                ServerWidget.saveScrollBarValue(currentFocus)

            ServerWidget.showTabAndSpacesIfNecessary()

            if currentFocus >= 0:
                ServerWidget.restoreScrollBarValue(currentFocus)


class RoutingChildAction(Action):
    def __init__(self, **kwargs):
        super().__init__(_('Routing Settings...'), **kwargs)

    def triggeredCallback(self, checked):
        APP().RoutesWidget.show()


class VPNModeAction(Action):
    def __init__(self, **kwargs):
        super().__init__(
            _('VPN Mode')
            if isAdministrator()
            else _(f'VPN Mode Disabled ({ADMINISTRATOR_NAME})'),
            **kwargs,
        )

        if not isAdministrator():
            self.setDisabled(True)

    def triggeredCallback(self, checked):
        assert isAdministrator()

        if checked:
            APP().VPNMode = Switch.ON_

            if APP().isConnected():
                if isVPNMode():
                    if PLATFORM == 'Windows' or PLATFORM == 'Darwin':
                        # Currently VPN Mode is only supported on Windows and macOS
                        APP().tray.ConnectAction.startTun2socks(
                            successCallback=lambda: APP().tray.showMessage(
                                _('VPN mode started')
                            )
                        )
        else:
            if APP().isConnected():
                if isVPNMode():
                    if PLATFORM == 'Windows' or PLATFORM == 'Darwin':
                        # Currently VPN Mode is only supported on Windows and macOS
                        APP().tray.ConnectAction.stopTun2socks()
                        APP().tray.showMessage(_('VPN mode stopped'))

            APP().VPNMode = Switch.OFF


class TorRelaySettingsChildAction(Action):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

    def triggeredCallback(self, checked):
        APP().TorRelayWidget.open()


class SettingsAction(Action):
    def __init__(self):
        super().__init__(
            _('Settings'),
            icon=bootstrapIcon('gear-wide-connected.svg'),
            menu=Menu(
                SettingsChildAction(
                    _('Startup On Boot'),
                    checkable=True,
                    checked=APP().StartupOnBoot == Switch.ON_,
                ),
                SettingsChildAction(
                    _('Show Progress Bar When Connecting'),
                    checkable=True,
                    checked=APP().ShowProgressBarWhenConnecting == Switch.ON_,
                ),
                SettingsChildAction(
                    _('Show Tab And Spaces In Editor'),
                    checkable=True,
                    checked=APP().ShowTabAndSpacesInEditor == Switch.ON_,
                ),
                VPNModeAction(
                    checkable=True,
                    checked=APP().VPNMode == Switch.ON_,
                )
                if PLATFORM == 'Windows' or PLATFORM == 'Darwin'
                else None,
                Seperator(),
                RoutingChildAction(),
                Seperator(),
                TorRelaySettingsChildAction(
                    _('Tor Relay Settings...'), icon=bootstrapIcon('incognito.svg')
                ),
            ),
            useActionGroup=False,
        )

    def getVPNModeAction(self):
        if PLATFORM == 'Windows' or PLATFORM == 'Darwin':
            # 3rd action
            return self._menu.actions()[3]
        else:
            return None