rediff-patches.py

   1 #!/usr/bin/python3
   2
   3 # rediff-patches.py name.spec
   4
   5 import argparse
   6 import collections
   7 import logging
   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
  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 ]
  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)
  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 ]
  52         logging.debug("running %s" % repr(cmd))
  53         try:
  54             subprocess.check_call(cmd, cwd=builddir, stdout=f, stderr=sys.stderr,
  55                                   env={'LC_ALL': 'C.UTF-8'}, timeout=600)
  56         except subprocess.CalledProcessError as err:
  57             if err.returncode != 1:
  58                 raise
  59     logging.info("rediff generated as %s" % output)
  60
  61 def main():
  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
  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
  94     applied_patches = collections.OrderedDict()
  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
 107     for patch_nr in applied_patches.keys():
 108         if args.patches and patch_nr not in args.patches:
 109             continue
 110         patch_name = patches[patch_nr]
 111         logging.info("*** patch %d: %s" % (patch_nr, patch_name))
 112
 113         tempspec = prepare_spec(r, patch_nr, before=True)
 114         unpack(tempspec.name, builddir)
 115         tempspec.close()
 116         os.rename(appbuilddir, appbuilddir + ".org")
 117
 118         tempspec = prepare_spec(r, patch_nr, before=False)
 119         unpack(tempspec.name, builddir)
 120         tempspec.close()
 121
 122         diff(appbuilddir + ".org", appbuilddir, builddir, os.path.join(topdir, os.path.join(appsourcedir, patch_name + ".rediff")))
 123
 124         shutil.rmtree(builddir)
 125     tempdir.cleanup()
 126
 127 if __name__ == '__main__':
 128     main()