]>
Commit | Line | Data |
---|---|---|
7eacbddd AM |
1 | #!/usr/bin/python3 |
2 | ||
3 | # rediff-patches.py name.spec | |
4 | ||
1e57e977 AM |
5 | import argparse |
6 | import collections | |
7 | import logging | |
7eacbddd AM |
8 | import os |
9 | import re | |
10 | import rpm | |
11 | import shutil | |
12 | import subprocess | |
13 | import sys | |
14 | import tempfile | |
15 | ||
16 | RPMBUILD_ISPATCH = (1<<1) | |
17 | ||
ce75a2bd AM |
18 | def prepare_spec(r, patch_nr, before=False): |
19 | tempspec = tempfile.NamedTemporaryFile() | |
20 | re_patch = re.compile(r'^%patch(?P<patch_number>\d+)\w*') | |
21 | for line in r.parsed.split('\n'): | |
22 | m = re_patch.match(line) | |
23 | if m: | |
24 | patch_number = int(m.group('patch_number')) | |
25 | if patch_nr == patch_number: | |
26 | if before: | |
27 | tempspec.write(b"exit 0\n# here was patch%d\n" % patch_nr) | |
28 | else: | |
29 | line = re.sub(r'#.*', "", line) | |
30 | tempspec.write(b"%s\nexit 0\n" % line.encode('utf-8')) | |
31 | continue | |
32 | tempspec.write(b"%s\n" % line.encode('utf-8')) | |
33 | tempspec.flush() | |
34 | return tempspec | |
35 | ||
36 | def unpack(spec, builddir): | |
37 | cmd = [ 'rpmbuild', '-bp', | |
38 | '--define', '_builddir %s' % builddir, | |
39 | '--define', '_enable_debug_packages 0', | |
40 | '--define', '_default_patch_fuzz 2', | |
41 | spec ] | |
1e57e977 AM |
42 | logging.debug("running %s" % repr(cmd)) |
43 | subprocess.check_call(cmd, stdout=sys.stderr, stderr=sys.stderr, | |
44 | env={'LC_ALL': 'C.UTF-8'}, timeout=600) | |
7eacbddd AM |
45 | |
46 | def diff(diffdir_org, diffdir, builddir, output): | |
47 | diffdir_org = os.path.basename(diffdir_org) | |
48 | diffdir = os.path.basename(diffdir) | |
49 | ||
50 | with open(output, 'wt') as f: | |
51 | cmd = [ 'diff', '-urNp', '-x', '*.orig', diffdir_org, diffdir ] | |
1e57e977 | 52 | logging.debug("running %s" % repr(cmd)) |
7eacbddd | 53 | try: |
1e57e977 AM |
54 | subprocess.check_call(cmd, cwd=builddir, stdout=f, stderr=sys.stderr, |
55 | env={'LC_ALL': 'C.UTF-8'}, timeout=600) | |
7eacbddd AM |
56 | except subprocess.CalledProcessError as err: |
57 | if err.returncode != 1: | |
58 | raise | |
1e57e977 AM |
59 | logging.info("rediff generated as %s" % output) |
60 | ||
c65b587f | 61 | def main(): |
1e57e977 AM |
62 | parser = parser = argparse.ArgumentParser(description='rediff patches to avoid fuzzy hunks') |
63 | parser.add_argument('spec', type=str, help='spec file name') | |
64 | parser.add_argument('-p', '--patches', type=str, help='comma separated list of patch numbers to rediff') | |
65 | parser.add_argument('-v', '--verbose', help='increase output verbosity', action='store_true') | |
66 | args = parser.parse_args() | |
67 | ||
68 | logging.basicConfig(level=logging.INFO) | |
69 | rpm.setVerbosity(rpm.RPMLOG_ERR) | |
70 | ||
71 | if args.verbose: | |
72 | logging.basicConfig(level=logging.DEBUG) | |
73 | rpm.setVerbosity(rpm.RPMLOG_DEBUG) | |
74 | ||
75 | if args.patches: | |
76 | args.patches = [int(x) for x in args.patches.split(',')] | |
77 | ||
78 | specfile = args.spec | |
c65b587f AM |
79 | |
80 | tempdir = tempfile.TemporaryDirectory(dir="/dev/shm") | |
81 | topdir = tempdir.name | |
82 | builddir = os.path.join(topdir, 'BUILD') | |
83 | ||
84 | rpm.addMacro("_builddir", builddir) | |
85 | ||
86 | r = rpm.spec(specfile) | |
87 | ||
88 | patches = {} | |
89 | ||
90 | for (name, nr, flags) in r.sources: | |
91 | if flags & RPMBUILD_ISPATCH: | |
92 | patches[nr] = name | |
93 | ||
1e57e977 | 94 | applied_patches = collections.OrderedDict() |
c65b587f AM |
95 | re_patch = re.compile(r'^%patch(?P<patch_number>\d+)\w*(?P<patch_args>.*)') |
96 | for line in r.parsed.split('\n'): | |
97 | m = re_patch.match(line) | |
98 | if not m: | |
99 | continue | |
100 | patch_nr = int(m.group('patch_number')) | |
101 | patch_args = m.group('patch_args') | |
102 | applied_patches[patch_nr] = patch_args | |
103 | ||
104 | appsourcedir = rpm.expandMacro("%{_sourcedir}") | |
105 | appbuilddir = rpm.expandMacro("%{_builddir}/%{?buildsubdir}") | |
106 | ||
ce75a2bd | 107 | for patch_nr in applied_patches.keys(): |
1e57e977 | 108 | if args.patches and patch_nr not in args.patches: |
c65b587f | 109 | continue |
1e57e977 AM |
110 | patch_name = patches[patch_nr] |
111 | logging.info("*** patch %d: %s" % (patch_nr, patch_name)) | |
ce75a2bd AM |
112 | |
113 | tempspec = prepare_spec(r, patch_nr, before=True) | |
114 | unpack(tempspec.name, builddir) | |
115 | tempspec.close() | |
c65b587f | 116 | os.rename(appbuilddir, appbuilddir + ".org") |
ce75a2bd AM |
117 | |
118 | tempspec = prepare_spec(r, patch_nr, before=False) | |
119 | unpack(tempspec.name, builddir) | |
120 | tempspec.close() | |
121 | ||
c65b587f | 122 | diff(appbuilddir + ".org", appbuilddir, builddir, os.path.join(topdir, os.path.join(appsourcedir, patch_name + ".rediff"))) |
ce75a2bd | 123 | |
c65b587f AM |
124 | shutil.rmtree(builddir) |
125 | tempdir.cleanup() | |
126 | ||
127 | if __name__ == '__main__': | |
128 | main() |