mirror of
https://forge.fsky.io/oneflux/omegafox.git
synced 2026-02-10 04:42:03 -08:00
dc63ec8d3f
Do not copy the Firefox source repo before patching. Manage everything through git instead.
135 lines
4 KiB
Python
135 lines
4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
"""
|
|
GUI for managing Camoufox patches.
|
|
"""
|
|
|
|
import os
|
|
import contextlib
|
|
|
|
import easygui
|
|
from patch import list_files, patch, run
|
|
|
|
|
|
def into_camoufox_dir():
|
|
"""Find and cd to the camoufox-* folder (this is located ..)"""
|
|
os.chdir(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
folders = os.listdir('.')
|
|
for folder in folders:
|
|
if os.path.isdir(folder) and folder.startswith('camoufox-'):
|
|
os.chdir(folder)
|
|
break
|
|
else:
|
|
raise FileNotFoundError('No camoufox-* folder found')
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def temp_cd(path):
|
|
# Temporarily change to a different working directory
|
|
_old_cwd = os.getcwd()
|
|
os.chdir(os.path.abspath(path))
|
|
|
|
try:
|
|
yield
|
|
finally:
|
|
os.chdir(_old_cwd)
|
|
|
|
|
|
def reset_camoufox():
|
|
with temp_cd('..'):
|
|
run('make clean')
|
|
|
|
|
|
def run_patches(reverse=False):
|
|
patch_files = list(list_files('../patches', suffix='*.patch'))
|
|
if reverse:
|
|
title = "Unpatch files"
|
|
else:
|
|
title = "Patch files"
|
|
selected_patches = easygui.multchoicebox(title, "Patches", patch_files)
|
|
if not selected_patches:
|
|
exit()
|
|
|
|
for patch_file in selected_patches:
|
|
patch(patch_file, reverse=reverse)
|
|
|
|
|
|
# GUI Choicebox with options
|
|
choices = [
|
|
"Reset workspace",
|
|
"Select patches",
|
|
"Reverse patches",
|
|
"Find broken patches",
|
|
"See current workspace",
|
|
"Write workspace to patch",
|
|
]
|
|
|
|
"""
|
|
GUI Choicebox with the following options:
|
|
- Reset:
|
|
`make clean`
|
|
- Patch all BUT:
|
|
Checklist of *.patch files in ../patches to exclude. The rest gets patched.
|
|
- Find broken:
|
|
Resets, runs patches, then finds broken patches.
|
|
If all show good error code, show at the top of the message box "All patches applied successfully"
|
|
If any show bad error code, show at the top of the message box "Some patches failed to apply", then the rej output
|
|
"""
|
|
|
|
|
|
def handle_choice(choice):
|
|
match choice:
|
|
case "Reset workspace":
|
|
reset_camoufox()
|
|
easygui.msgbox("Reset completed and bootstrap patches applied.", "Reset Complete")
|
|
|
|
case "Select patches":
|
|
run_patches(reverse=False)
|
|
easygui.msgbox("Patching completed.", "Patching Complete")
|
|
|
|
case "Reverse patches":
|
|
run_patches(reverse=True)
|
|
easygui.msgbox("Unpatching completed.", "Unpatching Complete")
|
|
|
|
case "Find broken patches":
|
|
reset_camoufox()
|
|
|
|
broken_patches = []
|
|
for patch_file in list_files('../patches', suffix='*.patch'):
|
|
cmd = rf'patch -p1 -i "{patch_file}" | tee /dev/stderr | sed -r --quiet \'s/^.*saving rejects to file (.*\.rej)$/\\1/p\''
|
|
result = os.popen(cmd).read().strip()
|
|
print(result)
|
|
if result:
|
|
broken_patches.append((patch_file, result))
|
|
|
|
if not broken_patches:
|
|
easygui.msgbox("All patches applied successfully", "Patching Result")
|
|
else:
|
|
message = "Some patches failed to apply:\n\n"
|
|
for patch_file, rejects in broken_patches:
|
|
message += f"Patch: {patch_file}\nRejects: {rejects}\n\n"
|
|
easygui.textbox("Patching Result", "Failed Patches", message)
|
|
|
|
case "See current workspace":
|
|
result = os.popen('git diff').read()
|
|
easygui.textbox("Diff", "Diff", result)
|
|
|
|
case "Write workspace to patch":
|
|
# Open a file dialog to select a file to write the diff to
|
|
file_path = easygui.filesavebox(
|
|
"Select a file to write the diff to", "Write Diff", filetypes="*.patch"
|
|
)
|
|
if not file_path:
|
|
exit()
|
|
run(f'git diff > {file_path}')
|
|
easygui.msgbox("Diff written to file", "Diff Written")
|
|
|
|
case _:
|
|
print('No choice selected')
|
|
|
|
|
|
if __name__ == "__main__":
|
|
into_camoufox_dir()
|
|
|
|
choice = easygui.choicebox("Select an option:", "Camoufox Dev Tools", choices)
|
|
handle_choice(choice)
|