advanced_settings.py

#!/usr/bin/env python

import wx

import images as images
from runtime import *


# ============================================================================
#                               Class AdvancedSettings
# ============================================================================
class AdvancedSettings(wx.Dialog):
    def __init__(self, *args, **kwargs):
        wx.Dialog.__init__(self, *args, **kwargs)
        self.SetTitle("Advanced Configuration Settings")
        self.before = get_advanced_options()

        vSizer = wx.BoxSizer(wx.VERTICAL)
        warning_sizer = wx.BoxSizer(wx.HORIZONTAL)
        warning_text = '''WARNING!
This is advanced configuration.
Unless you know what you are doing,
you should not be enabling it.

YOU AND YOU ALONE ARE RESPONSIBLE FOR ANYTHING THAT HAPPENS TO YOUR DEVICE.
THIS TOOL IS CODED WITH THE EXPRESS ASSUMPTION THAT YOU ARE FAMILIAR WITH
ADB, MAGISK, ANDROID, AND ROOT.
IT IS YOUR RESPONSIBILITY TO ENSURE THAT YOU KNOW WHAT YOU ARE DOING.
'''
        # warning label
        self.warning_label = wx.StaticText(parent=self, id=wx.ID_ANY, label=warning_text, pos=wx.DefaultPosition, size=wx.DefaultSize, style=wx.ALIGN_CENTER_HORIZONTAL)
        self.warning_label.Wrap(-1)
        self.warning_label.SetForegroundColour(wx.Colour(255, 0, 0))
        warning_sizer.Add(self.warning_label, proportion=0, flag=wx.LEFT|wx.RIGHT|wx.EXPAND, border=80)
        vSizer.Add(warning_sizer, proportion=0, flag=wx.EXPAND, border=5)

        # advanced options
        advanced_options_sizer = wx.BoxSizer(wx.HORIZONTAL)
        self.advanced_options_checkbox = wx.CheckBox(parent=self, id=wx.ID_ANY, label=u"Enable Advanced Options", pos=wx.DefaultPosition, size=wx.DefaultSize, style=0)
        self.advanced_options_checkbox.SetValue(get_advanced_options())
        self.advanced_options_checkbox.SetToolTip(u"Expert mode")
        advanced_options_sizer.Add(self.advanced_options_checkbox, proportion=0, flag=wx.ALL, border=5)
        vSizer.Add(advanced_options_sizer, proportion=0, flag=wx.EXPAND, border=5)

        # static line
        staticline = wx.StaticLine(parent=self, id=wx.ID_ANY, pos=wx.DefaultPosition, size=wx.DefaultSize, style=wx.LI_HORIZONTAL)
        vSizer.Add(staticline, proportion=0, flag=wx.EXPAND, border=5)

        # gap
        # vSizer.Add((0, 20), proportion=0, flag=0, border=5)

        NUMROWS = 9
        fgs1 = wx.FlexGridSizer(rows=NUMROWS, cols=2, vgap=10, hgap=10)
        # this makes the second column expandable (index starts at 0)
        fgs1.AddGrowableCol(1, 1)

        # Magisk Package name
        package_name_label = wx.StaticText(parent=self, label=u"Magisk Package Name:")
        self.package_name = wx.TextCtrl(parent=self, id=-1, size=(-1, -1))
        self.package_name.SetToolTip(u"If you have hidden Magisk,\nset this to the hidden package name.")
        self.package_name.SetValue(str(get_magisk_package()))
        self.reset_magisk_pkg = wx.BitmapButton(parent=self, id=wx.ID_ANY, bitmap=wx.NullBitmap, pos=wx.DefaultPosition, size=wx.DefaultSize, style=wx.BU_AUTODRAW)
        self.reset_magisk_pkg.SetBitmap(images.scan_24.GetBitmap())
        self.reset_magisk_pkg.SetToolTip(u"Resets package name to default: com.topjohnwu.magisk")
        package_name_sizer = wx.BoxSizer(orient=wx.HORIZONTAL)
        package_name_sizer.Add(self.package_name, proportion=1, flag=wx.ALL, border=0)
        package_name_sizer.Add(self.reset_magisk_pkg, proportion=0, flag=wx.LEFT|wx.ALIGN_CENTER_VERTICAL, border=5)

        # only add if we're on linux
        if sys.platform.startswith("linux"):
            # Linux File Explorer
            file_explorer_label = wx.StaticText(self, label=u"Linux File Explorer:")
            file_explorer_label.SetSize(self.package_name.GetSize())
            self.file_explorer = wx.TextCtrl(self, -1, size=(300, -1))
            self.file_explorer.SetToolTip(u"Set full path to File Explorer.\nDefault: Nautilus")
            self.file_explorer.SetValue(str(get_file_explorer()))
            file_explorer_sizer = wx.BoxSizer(orient=wx.HORIZONTAL)
            file_explorer_sizer.Add((20, 0), proportion=0, flag=wx.ALL, border=5)
            file_explorer_sizer.Add(self.file_explorer, proportion=0, flag=wx.LEFT, border=10)

            # Linux Shell
            shell_label = wx.StaticText(parent=self, label=u"Linux Shell:")
            shell_label.SetSize(self.package_name.GetSize())
            self.shell = wx.TextCtrl(parent=self, id=wx.ID_ANY, size=(300, -1))
            self.shell.SetToolTip(u"Set full path to Linux Shell.\nDefault: gnome-terminal")
            self.shell.SetValue(str(get_linux_shell()))
            shell_sizer = wx.BoxSizer(orient=wx.HORIZONTAL)
            shell_sizer.Add((20, 0), proportion=0, flag=wx.ALL, border=5)
            shell_sizer.Add(self.shell, proportion=0, flag=wx.LEFT, border=10)

        # Offer Patch methods
        self.patch_methods_checkbox = wx.CheckBox(parent=self, id=wx.ID_ANY, label=u"Offer Patch Methods", pos=wx.DefaultPosition, size=wx.DefaultSize, style=0)
        self.patch_methods_checkbox.SetValue(get_patch_methods_settings())
        self.patch_methods_checkbox.SetToolTip(u"When patching the choice of method is presented.")
        self.recovery_patch_checkbox = wx.CheckBox(parent=self, id=wx.ID_ANY, label=u"Patching Recovery Partition", pos=wx.DefaultPosition, size=wx.DefaultSize, style=0)
        self.recovery_patch_checkbox.SetValue(get_recovery_patch_settings())
        self.recovery_patch_checkbox.SetToolTip(u"Enabling this will show an option to patch a recovery partition.\nThis should be kept disabled unless you have an old device.\n(most A-only devices launched with Android 9, legacy SAR)")

        # Use Busybox Shell
        self.use_busybox_shell_checkbox = wx.CheckBox(parent=self, id=wx.ID_ANY, label=u"Use Busybox Shell", pos=wx.DefaultPosition, size=wx.DefaultSize, style=0)
        self.use_busybox_shell_checkbox.SetValue(get_use_busybox_shell_settings())
        self.use_busybox_shell_checkbox.SetToolTip(u"When creating a patch, if this is checked, busybox ash will be used as shell.")

        # Check for updates options
        self.check_for_update_checkbox = wx.CheckBox(parent=self, id=wx.ID_ANY, label=u"Check for updates", pos=wx.DefaultPosition, size=wx.DefaultSize, style=0)
        self.check_for_update_checkbox.SetValue(get_update_check())
        self.check_for_update_checkbox.SetToolTip(u"Checks for available updates on startup")

        # Force codepage
        self.force_codepage_checkbox = wx.CheckBox(parent=self, id=wx.ID_ANY, label=u"Force codepage to:", pos=wx.DefaultPosition, size=wx.DefaultSize, style=0)
        self.force_codepage_checkbox.SetValue(get_codepage_setting())
        self.force_codepage_checkbox.SetToolTip(u"Uses specified code page instead of system code page")
        self.code_page = wx.TextCtrl(parent=self, id=wx.ID_ANY, size=(-1, -1))
        self.code_page.SetValue(str(get_codepage_value()))

        # Use Custom Font
        self.use_custom_font_checkbox = wx.CheckBox(parent=self, id=wx.ID_ANY, label=u"Use Custom Fontface", pos=wx.DefaultPosition, size=wx.DefaultSize, style=0)
        self.use_custom_font_checkbox.SetValue(get_customize_font())
        self.use_custom_font_checkbox.SetToolTip(u"Use custom font for monospace fonts\nMight require PixelFlasher restart to properly apply to the Console window.")

        # Font Selection
        fonts = wx.FontEnumerator()
        fonts.EnumerateFacenames(wx.FONTENCODING_SYSTEM, fixedWidthOnly=True)
        font_list = fonts.GetFacenames(wx.FONTENCODING_SYSTEM, fixedWidthOnly=True)
        self.font = wx.ListBox(parent=self, id=wx.ID_ANY, size=(300, 100), choices=font_list)
        self.font_size = wx.SpinCtrl(parent=self, id=wx.ID_ANY, min=6, max=50, initial=get_pf_font_size())
        self.sample = wx.StaticText(parent=self, id=wx.ID_ANY, label="Sample ")
        fonts_sizer = wx.BoxSizer(wx.HORIZONTAL)
        fonts_sizer.Add(self.font, proportion=0, flag=wx.ALL|wx.ALIGN_CENTER_VERTICAL, border=0)
        fonts_sizer.Add(self.font_size, proportion=0, flag=wx.ALL|wx.ALIGN_CENTER_VERTICAL, border=5)
        fonts_sizer.Add(self.sample, proportion=1, flag=wx.ALL|wx.ALIGN_CENTER_VERTICAL, border=5)
        self.font.SetSelection(-1)
        self.font.SetStringSelection(get_pf_font_face())
        self.font_size.SetToolTip('Select font size')
        self._onFontSelect(None)

        # add the widgets to the grid in two columns, first fix size, the second expandable.
        fgs1.Add(package_name_label, 0, wx.EXPAND)
        fgs1.Add(package_name_sizer, 1, wx.EXPAND)

        # only add if we're on linux
        if sys.platform.startswith("linux"):
            fgs1.Add(file_explorer_label, 0, wx.EXPAND)
            fgs1.Add(file_explorer_sizer, 1, wx.EXPAND)

            fgs1.Add(shell_label, 0, wx.EXPAND)
            fgs1.Add(shell_sizer, 1, wx.EXPAND)

        fgs1.Add(self.patch_methods_checkbox, 0, wx.EXPAND)
        fgs1.Add(self.recovery_patch_checkbox, 0, wx.EXPAND)

        fgs1.Add(self.use_busybox_shell_checkbox, 0, wx.EXPAND)
        fgs1.Add((0, 0))

        fgs1.Add(self.check_for_update_checkbox, 0, wx.EXPAND)
        fgs1.Add((0, 0))

        fgs1.Add(self.force_codepage_checkbox, 0, wx.EXPAND)
        fgs1.Add(self.code_page, 1, wx.EXPAND)

        fgs1.Add(self.use_custom_font_checkbox, 0, wx.EXPAND)
        fgs1.Add(fonts_sizer, 1, wx.EXPAND)

        # add flexgrid to vSizer
        vSizer.Add(fgs1, proportion=0, flag=wx.ALL | wx.EXPAND, border=20)

        # gap
        vSizer.Add((0, 20), proportion=0, flag=0, border=5)

        # buttons
        buttons_sizer = wx.BoxSizer(wx.HORIZONTAL)
        buttons_sizer.Add((0, 0), proportion=1, flag=wx.EXPAND, border=5)
        self.ok_button = wx.Button(parent=self, id=wx.ID_ANY, label=u"OK", pos=wx.DefaultPosition, size=wx.DefaultSize, style=0)
        buttons_sizer.Add(self.ok_button, proportion=0, flag=wx.ALL, border=20)
        self.cancel_button = wx.Button(parent=self, id=wx.ID_ANY, label=u"Cancel", pos=wx.DefaultPosition, size=wx.DefaultSize, style=0)
        buttons_sizer.Add(self.cancel_button, proportion=0, flag=wx.ALL, border=20)
        buttons_sizer.Add((0, 0), proportion=1, flag=wx.EXPAND, border=5)
        vSizer.Add(buttons_sizer, proportion=0, flag=wx.EXPAND, border=5)

        self.SetSizer(vSizer)
        self.Layout()

        # Connect Events
        self.ok_button.Bind(wx.EVT_BUTTON, self._onOk)
        self.cancel_button.Bind(wx.EVT_BUTTON, self._onCancel)
        self.font.Bind(wx.EVT_LISTBOX, self._onFontSelect)
        self.font_size.Bind(wx.EVT_SPINCTRL, self._onFontSelect)
        self.reset_magisk_pkg.Bind(wx.EVT_BUTTON, self._onResetMagiskPkg)
        self.patch_methods_checkbox.Bind(wx.EVT_CHECKBOX, self._on_offer_patch_methods)
        self.use_custom_font_checkbox.Bind(wx.EVT_CHECKBOX, self._on_use_custom_fontface)

        # Enable / Disable Widgets
        self.enable_disable_widgets()

        # Autosize the dialog
        self.SetSizerAndFit(vSizer)

    def enable_disable_widgets(self):
        if self.patch_methods_checkbox.GetValue():
            self.recovery_patch_checkbox.Enable()
        else:
            self.recovery_patch_checkbox.Disable()
        if self.use_custom_font_checkbox.GetValue():
            self.font.Enable()
            self.font_size.Enable()
            self.sample.Enable()
        else:
            self.font.Disable()
            self.font_size.Disable()
            self.sample.Disable()

    def _on_offer_patch_methods(self, event):
        self.enable_disable_widgets()

    def _on_use_custom_fontface(self, event):
        self.enable_disable_widgets()

    def _onFontSelect(self, evt):
        facename = self.font.GetStringSelection()
        size = self.font_size.GetValue()
        font = wx.Font(size, family=wx.DEFAULT, style=wx.NORMAL, weight=wx.NORMAL, underline=False, faceName=facename)
        self.sample.SetLabel(facename)
        self.sample.SetFont(font)
        self.Refresh()

    def _onResetMagiskPkg(self, e):
        self.package_name.Label = 'com.topjohnwu.magisk'

    def _onCancel(self, e):
        set_advanced_options(self.before)
        self.EndModal(wx.ID_CANCEL)

    def _onOk(self, e):
        if self.advanced_options_checkbox.GetValue() != self.before:
            print(f"Setting Enable Advanced Options to: {self.advanced_options_checkbox.GetValue()}")
        set_advanced_options(self.advanced_options_checkbox.GetValue())

        if self.patch_methods_checkbox.GetValue() != get_patch_methods_settings():
            print(f"Setting Offer Patch Methods to: {self.patch_methods_checkbox.GetValue()}")
        set_patch_methods_settings(self.patch_methods_checkbox.GetValue())

        if self.recovery_patch_checkbox.GetValue() != get_recovery_patch_settings():
            print(f"Setting Offer Patch Methods to: {self.recovery_patch_checkbox.GetValue()}")
        set_recovery_patch_settings(self.recovery_patch_checkbox.GetValue())

        if self.use_busybox_shell_checkbox.GetValue() != get_use_busybox_shell_settings():
            print(f"Setting Use Busybox Shell to: {self.use_busybox_shell_checkbox.GetValue()}")
        set_use_busybox_shell_settings(self.use_busybox_shell_checkbox.GetValue())

        if self.check_for_update_checkbox.GetValue() != get_update_check():
            print(f"Setting Check for updates to: {self.check_for_update_checkbox.GetValue()}")
        set_update_check(self.check_for_update_checkbox.GetValue())

        if self.package_name.GetValue():
            if self.package_name.GetValue() != get_magisk_package():
                print(f"Setting Magisk Package Name to: {self.package_name.GetValue()}")
            set_magisk_package(self.package_name.GetValue())

        with contextlib.suppress(Exception):
            if self.file_explorer.GetValue() != get_file_explorer():
                print(f"Setting Linux File Explorer to: {self.file_explorer.GetValue()}")
            set_file_explorer(self.file_explorer.GetValue())

        with contextlib.suppress(Exception):
            if self.shell.GetValue() != get_linux_shell():
                print(f"Setting Linux Shell to: {self.shell.GetValue()}")
            set_linux_shell(self.shell.GetValue())

        if self.force_codepage_checkbox.GetValue():
            if self.code_page.GetValue() and self.code_page.GetValue().isnumeric():
                set_codepage_setting(self.force_codepage_checkbox.GetValue())
                set_codepage_value(int(self.code_page.GetValue()))
            else:
                set_codepage_setting(False)
                set_codepage_value('')
        else:
            set_codepage_setting(False)
            set_codepage_value('')

        if self.use_custom_font_checkbox.GetValue() != get_customize_font():
            set_customize_font(self.use_custom_font_checkbox.GetValue())
            if self.font.GetStringSelection() != get_pf_font_face():
                print(f"Setting Application Font to: {self.font.GetStringSelection()}")
                set_pf_font_face(self.font.GetStringSelection())

            if self.font_size.GetValue() != get_pf_font_size():
                print(f"Setting Application Font Size to: {self.font_size.GetValue()}")
                set_pf_font_size(int(self.font_size.GetValue()))

        self.EndModal(wx.ID_OK)