100 lines
3.2 KiB
Python
100 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
import sys
|
|
import glob
|
|
import subprocess
|
|
import os
|
|
import tempfile
|
|
|
|
# Get the directory and construct the patch file pattern
|
|
directory = sys.argv[1]
|
|
patch_pattern = f"{directory}*.patch"
|
|
|
|
# Find patch files matching the pattern
|
|
patches = glob.glob(patch_pattern)
|
|
|
|
|
|
def raise_error(error_msg):
|
|
sys.stderr.write(error_msg + '\n')
|
|
sys.exit(1)
|
|
|
|
|
|
patches = sorted(os.listdir(directory))
|
|
for patch in patches:
|
|
if not patch.endswith('.patch'):
|
|
raise_error(
|
|
f'Patch file {patch} found in directory {directory} does not end in ".patch" - rename the patch file'
|
|
)
|
|
|
|
|
|
# Exit if no patches are found
|
|
if not patches:
|
|
error_message = (
|
|
f"\nERROR: Extension patching enabled, but no patches found in '{directory}'. "
|
|
"Please make sure APPLY_PATCHES is only enabled when there are actually patches present. "
|
|
"See .github/patches/extensions/README.md for more details."
|
|
)
|
|
raise_error(error_message)
|
|
|
|
|
|
current_dir = os.getcwd()
|
|
print(f"Applying patches at '{current_dir}'")
|
|
print(f"Resetting patches in {directory}\n")
|
|
|
|
# capture the current diff
|
|
diff_proc = subprocess.run(["git", "diff"], capture_output=True, check=True)
|
|
prev_diff = diff_proc.stdout
|
|
|
|
output_proc = subprocess.run(["git", "diff", "--numstat"], capture_output=True, check=True)
|
|
prev_output_lines = output_proc.stdout.decode('utf8').split('\n')
|
|
prev_output_lines.sort()
|
|
|
|
subprocess.run(["git", "clean", "-f"], check=True)
|
|
subprocess.run(["git", "reset", "--hard", "HEAD"], check=True)
|
|
|
|
|
|
def apply_patch(patch_file):
|
|
ARGUMENTS = ["patch", "-p1", "--forward", "-i"]
|
|
arguments = []
|
|
arguments.extend(ARGUMENTS)
|
|
arguments.append(patch_file)
|
|
try:
|
|
subprocess.run(arguments, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
except subprocess.CalledProcessError as e:
|
|
arguments[1:1] = ['-d', current_dir]
|
|
command = " ".join(arguments)
|
|
print(f"Failed to apply patch, command to reproduce locally:\n{command}")
|
|
print("\nError output:")
|
|
print(e.stderr.decode('utf-8'))
|
|
print("\nStandard output:")
|
|
print(e.stdout.decode('utf-8'))
|
|
print("Exiting")
|
|
exit(1)
|
|
|
|
|
|
# Apply each patch file using patch
|
|
for patch in patches:
|
|
print(f"Applying patch: {patch}\n")
|
|
apply_patch(os.path.join(directory, patch))
|
|
|
|
# all patches have applied - check the current diff
|
|
output_proc = subprocess.run(["git", "diff", "--numstat"], capture_output=True, check=True)
|
|
output_lines = output_proc.stdout.decode('utf8').split('\n')
|
|
output_lines.sort()
|
|
|
|
if len(output_lines) <= len(prev_output_lines) and prev_output_lines != output_lines:
|
|
print("Detected local changes - rolling back patch application")
|
|
|
|
subprocess.run(["git", "clean", "-f"], check=True)
|
|
subprocess.run(["git", "reset", "--hard", "HEAD"], check=True)
|
|
with tempfile.NamedTemporaryFile() as f:
|
|
f.write(prev_diff)
|
|
apply_patch(f.name)
|
|
|
|
print("--------------------------------------------------")
|
|
print("Generate a patch file using the following command:")
|
|
print("--------------------------------------------------")
|
|
print(f"(cd {os.getcwd()} && git diff > {os.path.join(directory, 'fix.patch')})")
|
|
print("--------------------------------------------------")
|
|
|
|
exit(1)
|