Implemented disable brave update feature

Author: maxisoft

.idea/codeStyles/codeStyleConfig.xml

@@ -17,8 +17,9 @@
 from bravepatcher.patcher import Patcher
 from bravepatcher.pattern import PatternData, PatternDownloader
 from bravepatcher.static_data import default_pattern_data
-from bravepatcher.utils.brave import get_brave_path, kill_all_brave, find_chrome_dll, get_brave_release_asset_url, \
-    get_brave_for_chrome_dll
+from bravepatcher.utils.brave import (
+    find_chrome_dll, get_brave_for_chrome_dll, get_brave_path,
+    get_brave_release_asset_url, kill_all_brave)
 
 app = typer.Typer(no_args_is_help=True, add_completion=False)
 

.idea/csv-plugin.xml

@@ -21,7 +21,8 @@
 from bravepatcher.static_data import default_pattern_data
 from bravepatcher.utils import open_folder_in_explorer
 from bravepatcher.utils.brave import (
-    find_chrome_dll, get_brave_for_chrome_dll, get_brave_path, kill_all_brave)
+    UpdateDisabler, find_chrome_dll, get_brave_for_chrome_dll, get_brave_path,
+    get_brave_updater_for_chrome_dll, kill_all_brave)
 
 try:
     chrome_dll = find_chrome_dll(get_brave_path()) or ""
@@ -313,6 +314,26 @@ def on_stop_brave(window):
         sg.popup_error(f"{e}", keep_on_top=True, modal=True)
 
 
+@event_handler("Enable Updates")
+def on_enable_updates(window):
+    chrome_dll_path = Path(window['chrome_dll'].get())
+    try:
+        updater_exe = get_brave_updater_for_chrome_dll(chrome_dll_path)
+        UpdateDisabler(updater_exe).enable_update()
+    except Exception as e:
+        sg.popup_error(f"{e}", keep_on_top=True, modal=True)
+
+
+@event_handler("Disable Updates")
+def on_disable_updates(window):
+    chrome_dll_path = Path(window['chrome_dll'].get())
+    try:
+        updater_exe = get_brave_updater_for_chrome_dll(chrome_dll_path)
+        UpdateDisabler(updater_exe).disable_update()
+    except Exception as e:
+        sg.popup_error(f"{e}", keep_on_top=True, modal=True)
+
+
 def _is_nuitka():
     return "__compiled__" in globals()
 

bravepatcher/cli.py

@@ -1,6 +1,6 @@
-from .Patcher import Patcher  # noqa: F401, I005
+from .exceptions import \
+    MemorySearchTooManyMatchException  # noqa: F401, I005; noqa: F401, I005
 from .exceptions import (MemorySearchException, MemorySearchNotFoundException,  # noqa: F401, I005
                          PatchError, PatchException)  # noqa: F401, I005
-from .exceptions import \
-    MemorySearchTooManyMatchException  # noqa: F401, I005
 from .models import PatchedSegment, PatcherResult  # noqa: F401, I005
+from .Patcher import Patcher  # noqa: F401, I005

bravepatcher/gui.py

@@ -4,7 +4,6 @@
 from io import BytesIO
 from zlib import decompress
 
-
 _default_pattern_data = b"%PATTERN_PLACEHOLDER%"
 
 default_pattern_data_cache = None

bravepatcher/patcher/__init__.py

@@ -0,0 +1,22 @@
+from pathlib import Path
+from typing import Optional
+
+from ..firewall import AbstractFirewallHelper, WindowsFirewallHelper
+
+
+class UpdateDisabler:
+    rule_name = "BravePatcher - Block Brave update"
+
+    def __init__(self, brave_path: Path, firewall_helper: Optional[AbstractFirewallHelper] = None):
+        self.brave_path = brave_path
+        if firewall_helper is None:
+            firewall_helper = WindowsFirewallHelper()
+        self.firewall_helper = firewall_helper
+
+    def disable_update(self):
+        if not self.firewall_helper.has_rule(self.rule_name):
+            self.firewall_helper.block_program(self.brave_path, self.rule_name)
+
+    def enable_update(self):
+        if self.firewall_helper.has_rule(self.rule_name):
+            self.firewall_helper.allow_program(self.brave_path, self.rule_name)

bravepatcher/static_data_template.py

@@ -1,3 +1,5 @@
-from .fs import find_chrome_dll, get_brave_for_chrome_dll, get_brave_path  # noqa: F401
+from .fs import (find_chrome_dll, get_brave_for_chrome_dll,  # noqa: F401
+                 get_brave_path, get_brave_updater_for_chrome_dll)
 from .gitrepo import BraveChannel, get_brave_release_asset_url  # noqa: F401
 from .runtime import kill_all_brave  # noqa: F401
+from .UpdateDisabler import UpdateDisabler  # noqa: F401

bravepatcher/utils/brave/UpdateDisabler.py

@@ -63,8 +63,15 @@ def find_chrome_dll(brave_path: Optional[Path]) -> Optional[Path]:
     return res
 
 
-def get_brave_for_chrome_dll(chrome_dll_path: Path):
+def get_brave_for_chrome_dll(chrome_dll_path: Path) -> Path:
     if not _is_windows:
         return Path()
 
     return chrome_dll_path.parent.parent / "brave.exe"
+
+
+def get_brave_updater_for_chrome_dll(chrome_dll_path: Path) -> Path:
+    if not _is_windows:
+        return Path()
+
+    return chrome_dll_path.parent.parent.parent.parent / "Update" / "BraveUpdate.exe"

bravepatcher/utils/brave/__init__.py

@@ -0,0 +1,16 @@
+from abc import ABC, abstractmethod
+from pathlib import Path
+
+
+class AbstractFirewallHelper(ABC):
+    @abstractmethod
+    def allow_program(self, path: Path, name: str):
+        pass
+
+    @abstractmethod
+    def block_program(self, path: Path, name: str):
+        pass
+
+    @abstractmethod
+    def has_rule(self, name: str) -> bool:
+        pass