Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame^] | 1 | #!/usr/bin/env python |
| 2 | # ex:ts=4:sw=4:sts=4:et |
| 3 | # -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- |
| 4 | # |
| 5 | # Copyright 2011 Intel Corporation |
| 6 | # Authored-by: Yu Ke <ke.yu@intel.com> |
| 7 | # Paul Eggleton <paul.eggleton@intel.com> |
| 8 | # Richard Purdie <richard.purdie@intel.com> |
| 9 | # |
| 10 | # This program is free software; you can redistribute it and/or modify |
| 11 | # it under the terms of the GNU General Public License version 2 as |
| 12 | # published by the Free Software Foundation. |
| 13 | # |
| 14 | # This program is distributed in the hope that it will be useful, |
| 15 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 16 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 17 | # GNU General Public License for more details. |
| 18 | # |
| 19 | # You should have received a copy of the GNU General Public License along |
| 20 | # with this program; if not, write to the Free Software Foundation, Inc., |
| 21 | # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
| 22 | |
| 23 | import fnmatch |
| 24 | import os, sys |
| 25 | import optparse |
| 26 | import logging |
| 27 | import subprocess |
| 28 | import tempfile |
| 29 | import ConfigParser |
| 30 | import re |
| 31 | from collections import OrderedDict |
| 32 | from string import Template |
| 33 | |
| 34 | __version__ = "0.2.1" |
| 35 | |
| 36 | def logger_create(): |
| 37 | logger = logging.getLogger("") |
| 38 | loggerhandler = logging.StreamHandler() |
| 39 | loggerhandler.setFormatter(logging.Formatter("[%(asctime)s] %(message)s","%H:%M:%S")) |
| 40 | logger.addHandler(loggerhandler) |
| 41 | logger.setLevel(logging.INFO) |
| 42 | return logger |
| 43 | |
| 44 | logger = logger_create() |
| 45 | |
| 46 | def get_current_branch(repodir=None): |
| 47 | try: |
| 48 | if not os.path.exists(os.path.join(repodir if repodir else '', ".git")): |
| 49 | # Repo not created yet (i.e. during init) so just assume master |
| 50 | return "master" |
| 51 | branchname = runcmd("git symbolic-ref HEAD 2>/dev/null", repodir).strip() |
| 52 | if branchname.startswith("refs/heads/"): |
| 53 | branchname = branchname[11:] |
| 54 | return branchname |
| 55 | except subprocess.CalledProcessError: |
| 56 | return "" |
| 57 | |
| 58 | class Configuration(object): |
| 59 | """ |
| 60 | Manages the configuration |
| 61 | |
| 62 | For an example config file, see combo-layer.conf.example |
| 63 | |
| 64 | """ |
| 65 | def __init__(self, options): |
| 66 | for key, val in options.__dict__.items(): |
| 67 | setattr(self, key, val) |
| 68 | |
| 69 | def readsection(parser, section, repo): |
| 70 | for (name, value) in parser.items(section): |
| 71 | if value.startswith("@"): |
| 72 | self.repos[repo][name] = eval(value.strip("@")) |
| 73 | else: |
| 74 | # Apply special type transformations for some properties. |
| 75 | # Type matches the RawConfigParser.get*() methods. |
| 76 | types = {'signoff': 'boolean'} |
| 77 | if name in types: |
| 78 | value = getattr(parser, 'get' + types[name])(section, name) |
| 79 | self.repos[repo][name] = value |
| 80 | |
| 81 | def readglobalsection(parser, section): |
| 82 | for (name, value) in parser.items(section): |
| 83 | if name == "commit_msg": |
| 84 | self.commit_msg_template = value |
| 85 | |
| 86 | logger.debug("Loading config file %s" % self.conffile) |
| 87 | self.parser = ConfigParser.ConfigParser() |
| 88 | with open(self.conffile) as f: |
| 89 | self.parser.readfp(f) |
| 90 | |
| 91 | # initialize default values |
| 92 | self.commit_msg_template = "Automatic commit to update last_revision" |
| 93 | |
| 94 | self.repos = {} |
| 95 | for repo in self.parser.sections(): |
| 96 | if repo == "combo-layer-settings": |
| 97 | # special handling for global settings |
| 98 | readglobalsection(self.parser, repo) |
| 99 | else: |
| 100 | self.repos[repo] = {} |
| 101 | readsection(self.parser, repo, repo) |
| 102 | |
| 103 | # Load local configuration, if available |
| 104 | self.localconffile = None |
| 105 | self.localparser = None |
| 106 | self.combobranch = None |
| 107 | if self.conffile.endswith('.conf'): |
| 108 | lcfile = self.conffile.replace('.conf', '-local.conf') |
| 109 | if os.path.exists(lcfile): |
| 110 | # Read combo layer branch |
| 111 | self.combobranch = get_current_branch() |
| 112 | logger.debug("Combo layer branch is %s" % self.combobranch) |
| 113 | |
| 114 | self.localconffile = lcfile |
| 115 | logger.debug("Loading local config file %s" % self.localconffile) |
| 116 | self.localparser = ConfigParser.ConfigParser() |
| 117 | with open(self.localconffile) as f: |
| 118 | self.localparser.readfp(f) |
| 119 | |
| 120 | for section in self.localparser.sections(): |
| 121 | if '|' in section: |
| 122 | sectionvals = section.split('|') |
| 123 | repo = sectionvals[0] |
| 124 | if sectionvals[1] != self.combobranch: |
| 125 | continue |
| 126 | else: |
| 127 | repo = section |
| 128 | if repo in self.repos: |
| 129 | readsection(self.localparser, section, repo) |
| 130 | |
| 131 | def update(self, repo, option, value, initmode=False): |
| 132 | # If the main config has the option already, that is what we |
| 133 | # are expected to modify. |
| 134 | if self.localparser and not self.parser.has_option(repo, option): |
| 135 | parser = self.localparser |
| 136 | section = "%s|%s" % (repo, self.combobranch) |
| 137 | conffile = self.localconffile |
| 138 | if initmode and not parser.has_section(section): |
| 139 | parser.add_section(section) |
| 140 | else: |
| 141 | parser = self.parser |
| 142 | section = repo |
| 143 | conffile = self.conffile |
| 144 | parser.set(section, option, value) |
| 145 | with open(conffile, "w") as f: |
| 146 | parser.write(f) |
| 147 | self.repos[repo][option] = value |
| 148 | |
| 149 | def sanity_check(self, initmode=False): |
| 150 | required_options=["src_uri", "local_repo_dir", "dest_dir", "last_revision"] |
| 151 | if initmode: |
| 152 | required_options.remove("last_revision") |
| 153 | msg = "" |
| 154 | missing_options = [] |
| 155 | for name in self.repos: |
| 156 | for option in required_options: |
| 157 | if option not in self.repos[name]: |
| 158 | msg = "%s\nOption %s is not defined for component %s" %(msg, option, name) |
| 159 | missing_options.append(option) |
| 160 | # Sanitize dest_dir so that we do not have to deal with edge cases |
| 161 | # (unset, empty string, double slashes) in the rest of the code. |
| 162 | # It not being set will still be flagged as error because it is |
| 163 | # listed as required option above; that could be changed now. |
| 164 | dest_dir = os.path.normpath(self.repos[name].get("dest_dir", ".")) |
| 165 | self.repos[name]["dest_dir"] = "." if not dest_dir else dest_dir |
| 166 | if msg != "": |
| 167 | logger.error("configuration file %s has the following error: %s" % (self.conffile,msg)) |
| 168 | if self.localconffile and 'last_revision' in missing_options: |
| 169 | logger.error("local configuration file %s may be missing configuration for combo branch %s" % (self.localconffile, self.combobranch)) |
| 170 | sys.exit(1) |
| 171 | |
| 172 | # filterdiff is required by action_splitpatch, so check its availability |
| 173 | if subprocess.call("which filterdiff > /dev/null 2>&1", shell=True) != 0: |
| 174 | logger.error("ERROR: patchutils package is missing, please install it (e.g. # apt-get install patchutils)") |
| 175 | sys.exit(1) |
| 176 | |
| 177 | def runcmd(cmd,destdir=None,printerr=True,out=None): |
| 178 | """ |
| 179 | execute command, raise CalledProcessError if fail |
| 180 | return output if succeed |
| 181 | """ |
| 182 | logger.debug("run cmd '%s' in %s" % (cmd, os.getcwd() if destdir is None else destdir)) |
| 183 | if not out: |
| 184 | out = os.tmpfile() |
| 185 | err = out |
| 186 | else: |
| 187 | err = os.tmpfile() |
| 188 | try: |
| 189 | subprocess.check_call(cmd, stdout=out, stderr=err, cwd=destdir, shell=isinstance(cmd, str)) |
| 190 | except subprocess.CalledProcessError,e: |
| 191 | err.seek(0) |
| 192 | if printerr: |
| 193 | logger.error("%s" % err.read()) |
| 194 | raise e |
| 195 | |
| 196 | err.seek(0) |
| 197 | output = err.read() |
| 198 | logger.debug("output: %s" % output ) |
| 199 | return output |
| 200 | |
| 201 | def action_init(conf, args): |
| 202 | """ |
| 203 | Clone component repositories |
| 204 | Check git is initialised; if not, copy initial data from component repos |
| 205 | """ |
| 206 | for name in conf.repos: |
| 207 | ldir = conf.repos[name]['local_repo_dir'] |
| 208 | if not os.path.exists(ldir): |
| 209 | logger.info("cloning %s to %s" %(conf.repos[name]['src_uri'], ldir)) |
| 210 | subprocess.check_call("git clone %s %s" % (conf.repos[name]['src_uri'], ldir), shell=True) |
| 211 | if not os.path.exists(".git"): |
| 212 | runcmd("git init") |
| 213 | if conf.history: |
| 214 | # Need a common ref for all trees. |
| 215 | runcmd('git commit -m "initial empty commit" --allow-empty') |
| 216 | startrev = runcmd('git rev-parse master').strip() |
| 217 | |
| 218 | for name in conf.repos: |
| 219 | repo = conf.repos[name] |
| 220 | ldir = repo['local_repo_dir'] |
| 221 | branch = repo.get('branch', "master") |
| 222 | lastrev = repo.get('last_revision', None) |
| 223 | if lastrev and lastrev != "HEAD": |
| 224 | initialrev = lastrev |
| 225 | if branch: |
| 226 | if not check_rev_branch(name, ldir, lastrev, branch): |
| 227 | sys.exit(1) |
| 228 | logger.info("Copying data from %s at specified revision %s..." % (name, lastrev)) |
| 229 | else: |
| 230 | lastrev = None |
| 231 | initialrev = branch |
| 232 | logger.info("Copying data from %s..." % name) |
| 233 | # Sanity check initialrev and turn it into hash (required for copying history, |
| 234 | # because resolving a name ref only works in the component repo). |
| 235 | rev = runcmd('git rev-parse %s' % initialrev, ldir).strip() |
| 236 | if rev != initialrev: |
| 237 | try: |
| 238 | refs = runcmd('git show-ref -s %s' % initialrev, ldir).split('\n') |
| 239 | if len(set(refs)) > 1: |
| 240 | # Happens for example when configured to track |
| 241 | # "master" and there is a refs/heads/master. The |
| 242 | # traditional behavior from "git archive" (preserved |
| 243 | # here) it to choose the first one. This might not be |
| 244 | # intended, so at least warn about it. |
| 245 | logger.warn("%s: initial revision '%s' not unique, picking result of rev-parse = %s" % |
| 246 | (name, initialrev, refs[0])) |
| 247 | initialrev = rev |
| 248 | except: |
| 249 | # show-ref fails for hashes. Skip the sanity warning in that case. |
| 250 | pass |
| 251 | initialrev = rev |
| 252 | dest_dir = repo['dest_dir'] |
| 253 | if dest_dir != ".": |
| 254 | extract_dir = os.path.join(os.getcwd(), dest_dir) |
| 255 | if not os.path.exists(extract_dir): |
| 256 | os.makedirs(extract_dir) |
| 257 | else: |
| 258 | extract_dir = os.getcwd() |
| 259 | file_filter = repo.get('file_filter', "") |
| 260 | exclude_patterns = repo.get('file_exclude', '').split() |
| 261 | def copy_selected_files(initialrev, extract_dir, file_filter, exclude_patterns, ldir, |
| 262 | subdir=""): |
| 263 | # When working inside a filtered branch which had the |
| 264 | # files already moved, we need to prepend the |
| 265 | # subdirectory to all filters, otherwise they would |
| 266 | # not match. |
| 267 | if subdir == '.': |
| 268 | subdir = '' |
| 269 | elif subdir: |
| 270 | subdir = os.path.normpath(subdir) |
| 271 | file_filter = ' '.join([subdir + '/' + x for x in file_filter.split()]) |
| 272 | exclude_patterns = [subdir + '/' + x for x in exclude_patterns] |
| 273 | # To handle both cases, we cd into the target |
| 274 | # directory and optionally tell tar to strip the path |
| 275 | # prefix when the files were already moved. |
| 276 | subdir_components = len(subdir.split(os.path.sep)) if subdir else 0 |
| 277 | strip=('--strip-components=%d' % subdir_components) if subdir else '' |
| 278 | # TODO: file_filter wild cards do not work (and haven't worked before either), because |
| 279 | # a) GNU tar requires a --wildcards parameter before turning on wild card matching. |
| 280 | # b) The semantic is not as intendend (src/*.c also matches src/foo/bar.c, |
| 281 | # in contrast to the other use of file_filter as parameter of "git archive" |
| 282 | # where it only matches .c files directly in src). |
| 283 | files = runcmd("git archive %s %s | tar -x -v %s -C %s %s" % |
| 284 | (initialrev, subdir, |
| 285 | strip, extract_dir, file_filter), |
| 286 | ldir) |
| 287 | if exclude_patterns: |
| 288 | # Implement file removal by letting tar create the |
| 289 | # file and then deleting it in the file system |
| 290 | # again. Uses the list of files created by tar (easier |
| 291 | # than walking the tree). |
| 292 | for file in files.split('\n'): |
| 293 | for pattern in exclude_patterns: |
| 294 | if fnmatch.fnmatch(file, pattern): |
| 295 | os.unlink(os.path.join(*([extract_dir] + ['..'] * subdir_components + [file]))) |
| 296 | break |
| 297 | |
| 298 | if not conf.history: |
| 299 | copy_selected_files(initialrev, extract_dir, file_filter, exclude_patterns, ldir) |
| 300 | else: |
| 301 | # First fetch remote history into local repository. |
| 302 | # We need a ref for that, so ensure that there is one. |
| 303 | refname = "combo-layer-init-%s" % name |
| 304 | runcmd("git branch -f %s %s" % (refname, initialrev), ldir) |
| 305 | runcmd("git fetch %s %s" % (ldir, refname)) |
| 306 | runcmd("git branch -D %s" % refname, ldir) |
| 307 | # Make that the head revision. |
| 308 | runcmd("git checkout -b %s %s" % (name, initialrev)) |
| 309 | # Optional: cut the history by replacing the given |
| 310 | # start point(s) with commits providing the same |
| 311 | # content (aka tree), but with commit information that |
| 312 | # makes it clear that this is an artifically created |
| 313 | # commit and nothing the original authors had anything |
| 314 | # to do with. |
| 315 | since_rev = repo.get('since_revision', '') |
| 316 | if since_rev: |
| 317 | committer = runcmd('git var GIT_AUTHOR_IDENT').strip() |
| 318 | # Same time stamp, no name. |
| 319 | author = re.sub('.* (\d+ [+-]\d+)', r'unknown <unknown> \1', committer) |
| 320 | logger.info('author %s' % author) |
| 321 | for rev in since_rev.split(): |
| 322 | # Resolve in component repo... |
| 323 | rev = runcmd('git log --oneline --no-abbrev-commit -n1 %s' % rev, ldir).split()[0] |
| 324 | # ... and then get the tree in current |
| 325 | # one. The commit should be in both repos with |
| 326 | # the same tree, but better check here. |
| 327 | tree = runcmd('git show -s --pretty=format:%%T %s' % rev).strip() |
| 328 | with tempfile.NamedTemporaryFile() as editor: |
| 329 | editor.write('''cat >$1 <<EOF |
| 330 | tree %s |
| 331 | author %s |
| 332 | committer %s |
| 333 | |
| 334 | %s: squashed import of component |
| 335 | |
| 336 | This commit copies the entire set of files as found in |
| 337 | %s %s |
| 338 | |
| 339 | For more information about previous commits, see the |
| 340 | upstream repository. |
| 341 | |
| 342 | Commit created by combo-layer. |
| 343 | EOF |
| 344 | ''' % (tree, author, committer, name, name, since_rev)) |
| 345 | editor.flush() |
| 346 | os.environ['GIT_EDITOR'] = 'sh %s' % editor.name |
| 347 | runcmd('git replace --edit %s' % rev) |
| 348 | |
| 349 | # Optional: rewrite history to change commit messages or to move files. |
| 350 | if 'hook' in repo or dest_dir != ".": |
| 351 | filter_branch = ['git', 'filter-branch', '--force'] |
| 352 | with tempfile.NamedTemporaryFile() as hookwrapper: |
| 353 | if 'hook' in repo: |
| 354 | # Create a shell script wrapper around the original hook that |
| 355 | # can be used by git filter-branch. Hook may or may not have |
| 356 | # an absolute path. |
| 357 | hook = repo['hook'] |
| 358 | hook = os.path.join(os.path.dirname(conf.conffile), '..', hook) |
| 359 | # The wrappers turns the commit message |
| 360 | # from stdin into a fake patch header. |
| 361 | # This is good enough for changing Subject |
| 362 | # and commit msg body with normal |
| 363 | # combo-layer hooks. |
| 364 | hookwrapper.write('''set -e |
| 365 | tmpname=$(mktemp) |
| 366 | trap "rm $tmpname" EXIT |
| 367 | echo -n 'Subject: [PATCH] ' >>$tmpname |
| 368 | cat >>$tmpname |
| 369 | if ! [ $(tail -c 1 $tmpname | od -A n -t x1) == '0a' ]; then |
| 370 | echo >>$tmpname |
| 371 | fi |
| 372 | echo '---' >>$tmpname |
| 373 | %s $tmpname $GIT_COMMIT %s |
| 374 | tail -c +18 $tmpname | head -c -4 |
| 375 | ''' % (hook, name)) |
| 376 | hookwrapper.flush() |
| 377 | filter_branch.extend(['--msg-filter', 'bash %s' % hookwrapper.name]) |
| 378 | if dest_dir != ".": |
| 379 | parent = os.path.dirname(dest_dir) |
| 380 | if not parent: |
| 381 | parent = '.' |
| 382 | # May run outside of the current directory, so do not assume that .git exists. |
| 383 | filter_branch.extend(['--tree-filter', 'mkdir -p .git/tmptree && mv $(ls -1 -a | grep -v -e ^.git$ -e ^.$ -e ^..$) .git/tmptree && mkdir -p %s && mv .git/tmptree %s' % (parent, dest_dir)]) |
| 384 | filter_branch.append('HEAD') |
| 385 | runcmd(filter_branch) |
| 386 | runcmd('git update-ref -d refs/original/refs/heads/%s' % name) |
| 387 | repo['rewritten_revision'] = runcmd('git rev-parse HEAD').strip() |
| 388 | repo['stripped_revision'] = repo['rewritten_revision'] |
| 389 | # Optional filter files: remove everything and re-populate using the normal filtering code. |
| 390 | # Override any potential .gitignore. |
| 391 | if file_filter or exclude_patterns: |
| 392 | runcmd('git rm -rf .') |
| 393 | if not os.path.exists(extract_dir): |
| 394 | os.makedirs(extract_dir) |
| 395 | copy_selected_files('HEAD', extract_dir, file_filter, exclude_patterns, '.', |
| 396 | subdir=dest_dir) |
| 397 | runcmd('git add --all --force .') |
| 398 | if runcmd('git status --porcelain'): |
| 399 | # Something to commit. |
| 400 | runcmd(['git', 'commit', '-m', |
| 401 | '''%s: select file subset |
| 402 | |
| 403 | Files from the component repository were chosen based on |
| 404 | the following filters: |
| 405 | file_filter = %s |
| 406 | file_exclude = %s''' % (name, file_filter or '<empty>', repo.get('file_exclude', '<empty>'))]) |
| 407 | repo['stripped_revision'] = runcmd('git rev-parse HEAD').strip() |
| 408 | |
| 409 | if not lastrev: |
| 410 | lastrev = runcmd('git rev-parse %s' % initialrev, ldir).strip() |
| 411 | conf.update(name, "last_revision", lastrev, initmode=True) |
| 412 | |
| 413 | if not conf.history: |
| 414 | runcmd("git add .") |
| 415 | else: |
| 416 | # Create Octopus merge commit according to http://stackoverflow.com/questions/10874149/git-octopus-merge-with-unrelated-repositoies |
| 417 | runcmd('git checkout master') |
| 418 | merge = ['git', 'merge', '--no-commit'] |
| 419 | for name in conf.repos: |
| 420 | repo = conf.repos[name] |
| 421 | # Use branch created earlier. |
| 422 | merge.append(name) |
| 423 | # Root all commits which have no parent in the common |
| 424 | # ancestor in the new repository. |
| 425 | for start in runcmd('git log --pretty=format:%%H --max-parents=0 %s' % name).split('\n'): |
| 426 | runcmd('git replace --graft %s %s' % (start, startrev)) |
| 427 | try: |
| 428 | runcmd(merge) |
| 429 | except Exception, error: |
| 430 | logger.info('''Merging component repository history failed, perhaps because of merge conflicts. |
| 431 | It may be possible to commit anyway after resolving these conflicts. |
| 432 | |
| 433 | %s''' % error) |
| 434 | # Create MERGE_HEAD and MERGE_MSG. "git merge" itself |
| 435 | # does not create MERGE_HEAD in case of a (harmless) failure, |
| 436 | # and we want certain auto-generated information in the |
| 437 | # commit message for future reference and/or automation. |
| 438 | with open('.git/MERGE_HEAD', 'w') as head: |
| 439 | with open('.git/MERGE_MSG', 'w') as msg: |
| 440 | msg.write('repo: initial import of components\n\n') |
| 441 | # head.write('%s\n' % startrev) |
| 442 | for name in conf.repos: |
| 443 | repo = conf.repos[name] |
| 444 | # <upstream ref> <rewritten ref> <rewritten + files removed> |
| 445 | msg.write('combo-layer-%s: %s %s %s\n' % (name, |
| 446 | repo['last_revision'], |
| 447 | repo['rewritten_revision'], |
| 448 | repo['stripped_revision'])) |
| 449 | rev = runcmd('git rev-parse %s' % name).strip() |
| 450 | head.write('%s\n' % rev) |
| 451 | |
| 452 | if conf.localconffile: |
| 453 | localadded = True |
| 454 | try: |
| 455 | runcmd("git rm --cached %s" % conf.localconffile, printerr=False) |
| 456 | except subprocess.CalledProcessError: |
| 457 | localadded = False |
| 458 | if localadded: |
| 459 | localrelpath = os.path.relpath(conf.localconffile) |
| 460 | runcmd("grep -q %s .gitignore || echo %s >> .gitignore" % (localrelpath, localrelpath)) |
| 461 | runcmd("git add .gitignore") |
| 462 | logger.info("Added local configuration file %s to .gitignore", localrelpath) |
| 463 | logger.info("Initial combo layer repository data has been created; please make any changes if desired and then use 'git commit' to make the initial commit.") |
| 464 | else: |
| 465 | logger.info("Repository already initialised, nothing to do.") |
| 466 | |
| 467 | |
| 468 | def check_repo_clean(repodir): |
| 469 | """ |
| 470 | check if the repo is clean |
| 471 | exit if repo is dirty |
| 472 | """ |
| 473 | output=runcmd("git status --porcelain", repodir) |
| 474 | r = re.compile('\?\? patch-.*/') |
| 475 | dirtyout = [item for item in output.splitlines() if not r.match(item)] |
| 476 | if dirtyout: |
| 477 | logger.error("git repo %s is dirty, please fix it first", repodir) |
| 478 | sys.exit(1) |
| 479 | |
| 480 | def check_patch(patchfile): |
| 481 | f = open(patchfile) |
| 482 | ln = f.readline() |
| 483 | of = None |
| 484 | in_patch = False |
| 485 | beyond_msg = False |
| 486 | pre_buf = '' |
| 487 | while ln: |
| 488 | if not beyond_msg: |
| 489 | if ln == '---\n': |
| 490 | if not of: |
| 491 | break |
| 492 | in_patch = False |
| 493 | beyond_msg = True |
| 494 | elif ln.startswith('--- '): |
| 495 | # We have a diff in the commit message |
| 496 | in_patch = True |
| 497 | if not of: |
| 498 | print('WARNING: %s contains a diff in its commit message, indenting to avoid failure during apply' % patchfile) |
| 499 | of = open(patchfile + '.tmp', 'w') |
| 500 | of.write(pre_buf) |
| 501 | pre_buf = '' |
| 502 | elif in_patch and not ln[0] in '+-@ \n\r': |
| 503 | in_patch = False |
| 504 | if of: |
| 505 | if in_patch: |
| 506 | of.write(' ' + ln) |
| 507 | else: |
| 508 | of.write(ln) |
| 509 | else: |
| 510 | pre_buf += ln |
| 511 | ln = f.readline() |
| 512 | f.close() |
| 513 | if of: |
| 514 | of.close() |
| 515 | os.rename(patchfile + '.tmp', patchfile) |
| 516 | |
| 517 | def drop_to_shell(workdir=None): |
| 518 | if not sys.stdin.isatty(): |
| 519 | print "Not a TTY so can't drop to shell for resolution, exiting." |
| 520 | return False |
| 521 | |
| 522 | shell = os.environ.get('SHELL', 'bash') |
| 523 | print('Dropping to shell "%s"\n' \ |
| 524 | 'When you are finished, run the following to continue:\n' \ |
| 525 | ' exit -- continue to apply the patches\n' \ |
| 526 | ' exit 1 -- abort\n' % shell); |
| 527 | ret = subprocess.call([shell], cwd=workdir) |
| 528 | if ret != 0: |
| 529 | print "Aborting" |
| 530 | return False |
| 531 | else: |
| 532 | return True |
| 533 | |
| 534 | def check_rev_branch(component, repodir, rev, branch): |
| 535 | try: |
| 536 | actualbranch = runcmd("git branch --contains %s" % rev, repodir, printerr=False) |
| 537 | except subprocess.CalledProcessError as e: |
| 538 | if e.returncode == 129: |
| 539 | actualbranch = "" |
| 540 | else: |
| 541 | raise |
| 542 | |
| 543 | if not actualbranch: |
| 544 | logger.error("%s: specified revision %s is invalid!" % (component, rev)) |
| 545 | return False |
| 546 | |
| 547 | branches = [] |
| 548 | branchlist = actualbranch.split("\n") |
| 549 | for b in branchlist: |
| 550 | branches.append(b.strip().split(' ')[-1]) |
| 551 | |
| 552 | if branch not in branches: |
| 553 | logger.error("%s: specified revision %s is not on specified branch %s!" % (component, rev, branch)) |
| 554 | return False |
| 555 | return True |
| 556 | |
| 557 | def get_repos(conf, repo_names): |
| 558 | repos = [] |
| 559 | for name in repo_names: |
| 560 | if name.startswith('-'): |
| 561 | break |
| 562 | else: |
| 563 | repos.append(name) |
| 564 | for repo in repos: |
| 565 | if not repo in conf.repos: |
| 566 | logger.error("Specified component '%s' not found in configuration" % repo) |
| 567 | sys.exit(1) |
| 568 | |
| 569 | if not repos: |
| 570 | repos = conf.repos |
| 571 | |
| 572 | return repos |
| 573 | |
| 574 | def action_pull(conf, args): |
| 575 | """ |
| 576 | update the component repos only |
| 577 | """ |
| 578 | repos = get_repos(conf, args[1:]) |
| 579 | |
| 580 | # make sure all repos are clean |
| 581 | for name in repos: |
| 582 | check_repo_clean(conf.repos[name]['local_repo_dir']) |
| 583 | |
| 584 | for name in repos: |
| 585 | repo = conf.repos[name] |
| 586 | ldir = repo['local_repo_dir'] |
| 587 | branch = repo.get('branch', "master") |
| 588 | logger.info("update branch %s of component repo %s in %s ..." % (branch, name, ldir)) |
| 589 | if not conf.hard_reset: |
| 590 | # Try to pull only the configured branch. Beware that this may fail |
| 591 | # when the branch is currently unknown (for example, after reconfiguring |
| 592 | # combo-layer). In that case we need to fetch everything and try the check out |
| 593 | # and pull again. |
| 594 | try: |
| 595 | runcmd("git checkout %s" % branch, ldir, printerr=False) |
| 596 | except subprocess.CalledProcessError: |
| 597 | output=runcmd("git fetch", ldir) |
| 598 | logger.info(output) |
| 599 | runcmd("git checkout %s" % branch, ldir) |
| 600 | runcmd("git pull --ff-only", ldir) |
| 601 | else: |
| 602 | output=runcmd("git pull --ff-only", ldir) |
| 603 | logger.info(output) |
| 604 | else: |
| 605 | output=runcmd("git fetch", ldir) |
| 606 | logger.info(output) |
| 607 | runcmd("git checkout %s" % branch, ldir) |
| 608 | runcmd("git reset --hard FETCH_HEAD", ldir) |
| 609 | |
| 610 | def action_update(conf, args): |
| 611 | """ |
| 612 | update the component repos |
| 613 | generate the patch list |
| 614 | apply the generated patches |
| 615 | """ |
| 616 | components = [arg.split(':')[0] for arg in args[1:]] |
| 617 | revisions = {} |
| 618 | for arg in args[1:]: |
| 619 | if ':' in arg: |
| 620 | a = arg.split(':', 1) |
| 621 | revisions[a[0]] = a[1] |
| 622 | repos = get_repos(conf, components) |
| 623 | |
| 624 | # make sure combo repo is clean |
| 625 | check_repo_clean(os.getcwd()) |
| 626 | |
| 627 | import uuid |
| 628 | patch_dir = "patch-%s" % uuid.uuid4() |
| 629 | if not os.path.exists(patch_dir): |
| 630 | os.mkdir(patch_dir) |
| 631 | |
| 632 | # Step 1: update the component repos |
| 633 | if conf.nopull: |
| 634 | logger.info("Skipping pull (-n)") |
| 635 | else: |
| 636 | action_pull(conf, ['arg0'] + components) |
| 637 | |
| 638 | for name in repos: |
| 639 | revision = revisions.get(name, None) |
| 640 | repo = conf.repos[name] |
| 641 | ldir = repo['local_repo_dir'] |
| 642 | dest_dir = repo['dest_dir'] |
| 643 | branch = repo.get('branch', "master") |
| 644 | repo_patch_dir = os.path.join(os.getcwd(), patch_dir, name) |
| 645 | |
| 646 | # Step 2: generate the patch list and store to patch dir |
| 647 | logger.info("Generating patches from %s..." % name) |
| 648 | top_revision = revision or branch |
| 649 | if not check_rev_branch(name, ldir, top_revision, branch): |
| 650 | sys.exit(1) |
| 651 | if dest_dir != ".": |
| 652 | prefix = "--src-prefix=a/%s/ --dst-prefix=b/%s/" % (dest_dir, dest_dir) |
| 653 | else: |
| 654 | prefix = "" |
| 655 | if repo['last_revision'] == "": |
| 656 | logger.info("Warning: last_revision of component %s is not set, starting from the first commit" % name) |
| 657 | patch_cmd_range = "--root %s" % top_revision |
| 658 | rev_cmd_range = top_revision |
| 659 | else: |
| 660 | if not check_rev_branch(name, ldir, repo['last_revision'], branch): |
| 661 | sys.exit(1) |
| 662 | patch_cmd_range = "%s..%s" % (repo['last_revision'], top_revision) |
| 663 | rev_cmd_range = patch_cmd_range |
| 664 | |
| 665 | file_filter = repo.get('file_filter',"") |
| 666 | |
| 667 | patch_cmd = "git format-patch -N %s --output-directory %s %s -- %s" % \ |
| 668 | (prefix,repo_patch_dir, patch_cmd_range, file_filter) |
| 669 | output = runcmd(patch_cmd, ldir) |
| 670 | logger.debug("generated patch set:\n%s" % output) |
| 671 | patchlist = output.splitlines() |
| 672 | |
| 673 | rev_cmd = "git rev-list --no-merges %s -- %s" % (rev_cmd_range, file_filter) |
| 674 | revlist = runcmd(rev_cmd, ldir).splitlines() |
| 675 | |
| 676 | # Step 3: Call repo specific hook to adjust patch |
| 677 | if 'hook' in repo: |
| 678 | # hook parameter is: ./hook patchpath revision reponame |
| 679 | count=len(revlist)-1 |
| 680 | for patch in patchlist: |
| 681 | runcmd("%s %s %s %s" % (repo['hook'], patch, revlist[count], name)) |
| 682 | count=count-1 |
| 683 | |
| 684 | # Step 3a: Filter out unwanted files and patches. |
| 685 | exclude = repo.get('file_exclude', '') |
| 686 | if exclude: |
| 687 | filter = ['filterdiff', '-p1'] |
| 688 | for path in exclude.split(): |
| 689 | filter.append('-x') |
| 690 | filter.append('%s/%s' % (dest_dir, path) if dest_dir != '.' else path) |
| 691 | for patch in patchlist[:]: |
| 692 | filtered = patch + '.tmp' |
| 693 | with open(filtered, 'w') as f: |
| 694 | runcmd(filter + [patch], out=f) |
| 695 | # Now check for empty patches. |
| 696 | if runcmd(['filterdiff', '--list', filtered]): |
| 697 | # Possibly modified. |
| 698 | os.unlink(patch) |
| 699 | os.rename(filtered, patch) |
| 700 | else: |
| 701 | # Empty, ignore it. Must also remove from revlist. |
| 702 | with open(patch, 'r') as f: |
| 703 | fromline = f.readline() |
| 704 | if not fromline: |
| 705 | # Patch must have been empty to start with. No need |
| 706 | # to remove it. |
| 707 | continue |
| 708 | m = re.match(r'''^From ([0-9a-fA-F]+) .*\n''', fromline) |
| 709 | rev = m.group(1) |
| 710 | logger.debug('skipping empty patch %s = %s' % (patch, rev)) |
| 711 | os.unlink(patch) |
| 712 | os.unlink(filtered) |
| 713 | patchlist.remove(patch) |
| 714 | revlist.remove(rev) |
| 715 | |
| 716 | # Step 4: write patch list and revision list to file, for user to edit later |
| 717 | patchlist_file = os.path.join(os.getcwd(), patch_dir, "patchlist-%s" % name) |
| 718 | repo['patchlist'] = patchlist_file |
| 719 | f = open(patchlist_file, 'w') |
| 720 | count=len(revlist)-1 |
| 721 | for patch in patchlist: |
| 722 | f.write("%s %s\n" % (patch, revlist[count])) |
| 723 | check_patch(os.path.join(patch_dir, patch)) |
| 724 | count=count-1 |
| 725 | f.close() |
| 726 | |
| 727 | # Step 5: invoke bash for user to edit patch and patch list |
| 728 | if conf.interactive: |
| 729 | print('You may now edit the patch and patch list in %s\n' \ |
| 730 | 'For example, you can remove unwanted patch entries from patchlist-*, so that they will be not applied later' % patch_dir); |
| 731 | if not drop_to_shell(patch_dir): |
| 732 | sys.exit(1) |
| 733 | |
| 734 | # Step 6: apply the generated and revised patch |
| 735 | apply_patchlist(conf, repos) |
| 736 | runcmd("rm -rf %s" % patch_dir) |
| 737 | |
| 738 | # Step 7: commit the updated config file if it's being tracked |
| 739 | relpath = os.path.relpath(conf.conffile) |
| 740 | try: |
| 741 | output = runcmd("git status --porcelain %s" % relpath, printerr=False) |
| 742 | except: |
| 743 | # Outside the repository |
| 744 | output = None |
| 745 | if output: |
| 746 | logger.info("Committing updated configuration file") |
| 747 | if output.lstrip().startswith("M"): |
| 748 | |
| 749 | # create the "components" string |
| 750 | component_str = "all components" |
| 751 | if len(components) > 0: |
| 752 | # otherwise tell which components were actually changed |
| 753 | component_str = ", ".join(components) |
| 754 | |
| 755 | # expand the template with known values |
| 756 | template = Template(conf.commit_msg_template) |
| 757 | raw_msg = template.substitute(components = component_str) |
| 758 | |
| 759 | # sanitize the string before using it in command line |
| 760 | msg = raw_msg.replace('"', '\\"') |
| 761 | |
| 762 | runcmd('git commit -m "%s" %s' % (msg, relpath)) |
| 763 | |
| 764 | def apply_patchlist(conf, repos): |
| 765 | """ |
| 766 | apply the generated patch list to combo repo |
| 767 | """ |
| 768 | for name in repos: |
| 769 | repo = conf.repos[name] |
| 770 | lastrev = repo["last_revision"] |
| 771 | prevrev = lastrev |
| 772 | |
| 773 | # Get non-blank lines from patch list file |
| 774 | patchlist = [] |
| 775 | if os.path.exists(repo['patchlist']) or not conf.interactive: |
| 776 | # Note: we want this to fail here if the file doesn't exist and we're not in |
| 777 | # interactive mode since the file should exist in this case |
| 778 | with open(repo['patchlist']) as f: |
| 779 | for line in f: |
| 780 | line = line.rstrip() |
| 781 | if line: |
| 782 | patchlist.append(line) |
| 783 | |
| 784 | ldir = conf.repos[name]['local_repo_dir'] |
| 785 | branch = conf.repos[name].get('branch', "master") |
| 786 | branchrev = runcmd("git rev-parse %s" % branch, ldir).strip() |
| 787 | |
| 788 | if patchlist: |
| 789 | logger.info("Applying patches from %s..." % name) |
| 790 | linecount = len(patchlist) |
| 791 | i = 1 |
| 792 | for line in patchlist: |
| 793 | patchfile = line.split()[0] |
| 794 | lastrev = line.split()[1] |
| 795 | patchdisp = os.path.relpath(patchfile) |
| 796 | if os.path.getsize(patchfile) == 0: |
| 797 | logger.info("(skipping %d/%d %s - no changes)" % (i, linecount, patchdisp)) |
| 798 | else: |
| 799 | cmd = "git am --keep-cr %s-p1 %s" % ('-s ' if repo.get('signoff', True) else '', patchfile) |
| 800 | logger.info("Applying %d/%d: %s" % (i, linecount, patchdisp)) |
| 801 | try: |
| 802 | runcmd(cmd) |
| 803 | except subprocess.CalledProcessError: |
| 804 | logger.info('Running "git am --abort" to cleanup repo') |
| 805 | runcmd("git am --abort") |
| 806 | logger.error('"%s" failed' % cmd) |
| 807 | logger.info("Please manually apply patch %s" % patchdisp) |
| 808 | logger.info("Note: if you exit and continue applying without manually applying the patch, it will be skipped") |
| 809 | if not drop_to_shell(): |
| 810 | if prevrev != repo['last_revision']: |
| 811 | conf.update(name, "last_revision", prevrev) |
| 812 | sys.exit(1) |
| 813 | prevrev = lastrev |
| 814 | i += 1 |
| 815 | # Once all patches are applied, we should update |
| 816 | # last_revision to the branch head instead of the last |
| 817 | # applied patch. The two are not necessarily the same when |
| 818 | # the last commit is a merge commit or when the patches at |
| 819 | # the branch head were intentionally excluded. |
| 820 | # |
| 821 | # If we do not do that for a merge commit, the next |
| 822 | # combo-layer run will only exclude patches reachable from |
| 823 | # one of the merged branches and try to re-apply patches |
| 824 | # from other branches even though they were already |
| 825 | # copied. |
| 826 | # |
| 827 | # If patches were intentionally excluded, the next run will |
| 828 | # present them again instead of skipping over them. This |
| 829 | # may or may not be intended, so the code here is conservative |
| 830 | # and only addresses the "head is merge commit" case. |
| 831 | if lastrev != branchrev and \ |
| 832 | len(runcmd("git show --pretty=format:%%P --no-patch %s" % branch, ldir).split()) > 1: |
| 833 | lastrev = branchrev |
| 834 | else: |
| 835 | logger.info("No patches to apply from %s" % name) |
| 836 | lastrev = branchrev |
| 837 | |
| 838 | if lastrev != repo['last_revision']: |
| 839 | conf.update(name, "last_revision", lastrev) |
| 840 | |
| 841 | def action_splitpatch(conf, args): |
| 842 | """ |
| 843 | generate the commit patch and |
| 844 | split the patch per repo |
| 845 | """ |
| 846 | logger.debug("action_splitpatch") |
| 847 | if len(args) > 1: |
| 848 | commit = args[1] |
| 849 | else: |
| 850 | commit = "HEAD" |
| 851 | patchdir = "splitpatch-%s" % commit |
| 852 | if not os.path.exists(patchdir): |
| 853 | os.mkdir(patchdir) |
| 854 | |
| 855 | # filerange_root is for the repo whose dest_dir is root "." |
| 856 | # and it should be specified by excluding all other repo dest dir |
| 857 | # like "-x repo1 -x repo2 -x repo3 ..." |
| 858 | filerange_root = "" |
| 859 | for name in conf.repos: |
| 860 | dest_dir = conf.repos[name]['dest_dir'] |
| 861 | if dest_dir != ".": |
| 862 | filerange_root = '%s -x "%s/*"' % (filerange_root, dest_dir) |
| 863 | |
| 864 | for name in conf.repos: |
| 865 | dest_dir = conf.repos[name]['dest_dir'] |
| 866 | patch_filename = "%s/%s.patch" % (patchdir, name) |
| 867 | if dest_dir == ".": |
| 868 | cmd = "git format-patch -n1 --stdout %s^..%s | filterdiff -p1 %s > %s" % (commit, commit, filerange_root, patch_filename) |
| 869 | else: |
| 870 | cmd = "git format-patch --no-prefix -n1 --stdout %s^..%s -- %s > %s" % (commit, commit, dest_dir, patch_filename) |
| 871 | runcmd(cmd) |
| 872 | # Detect empty patches (including those produced by filterdiff above |
| 873 | # that contain only preamble text) |
| 874 | if os.path.getsize(patch_filename) == 0 or runcmd("filterdiff %s" % patch_filename) == "": |
| 875 | os.remove(patch_filename) |
| 876 | logger.info("(skipping %s - no changes)", name) |
| 877 | else: |
| 878 | logger.info(patch_filename) |
| 879 | |
| 880 | def action_error(conf, args): |
| 881 | logger.info("invalid action %s" % args[0]) |
| 882 | |
| 883 | actions = { |
| 884 | "init": action_init, |
| 885 | "update": action_update, |
| 886 | "pull": action_pull, |
| 887 | "splitpatch": action_splitpatch, |
| 888 | } |
| 889 | |
| 890 | def main(): |
| 891 | parser = optparse.OptionParser( |
| 892 | version = "Combo Layer Repo Tool version %s" % __version__, |
| 893 | usage = """%prog [options] action |
| 894 | |
| 895 | Create and update a combination layer repository from multiple component repositories. |
| 896 | |
| 897 | Action: |
| 898 | init initialise the combo layer repo |
| 899 | update [components] get patches from component repos and apply them to the combo repo |
| 900 | pull [components] just pull component repos only |
| 901 | splitpatch [commit] generate commit patch and split per component, default commit is HEAD""") |
| 902 | |
| 903 | parser.add_option("-c", "--conf", help = "specify the config file (conf/combo-layer.conf is the default).", |
| 904 | action = "store", dest = "conffile", default = "conf/combo-layer.conf") |
| 905 | |
| 906 | parser.add_option("-i", "--interactive", help = "interactive mode, user can edit the patch list and patches", |
| 907 | action = "store_true", dest = "interactive", default = False) |
| 908 | |
| 909 | parser.add_option("-D", "--debug", help = "output debug information", |
| 910 | action = "store_true", dest = "debug", default = False) |
| 911 | |
| 912 | parser.add_option("-n", "--no-pull", help = "skip pulling component repos during update", |
| 913 | action = "store_true", dest = "nopull", default = False) |
| 914 | |
| 915 | parser.add_option("--hard-reset", |
| 916 | help = "instead of pull do fetch and hard-reset in component repos", |
| 917 | action = "store_true", dest = "hard_reset", default = False) |
| 918 | |
| 919 | parser.add_option("-H", "--history", help = "import full history of components during init", |
| 920 | action = "store_true", default = False) |
| 921 | |
| 922 | options, args = parser.parse_args(sys.argv) |
| 923 | |
| 924 | # Dispatch to action handler |
| 925 | if len(args) == 1: |
| 926 | logger.error("No action specified, exiting") |
| 927 | parser.print_help() |
| 928 | elif args[1] not in actions: |
| 929 | logger.error("Unsupported action %s, exiting\n" % (args[1])) |
| 930 | parser.print_help() |
| 931 | elif not os.path.exists(options.conffile): |
| 932 | logger.error("No valid config file, exiting\n") |
| 933 | parser.print_help() |
| 934 | else: |
| 935 | if options.debug: |
| 936 | logger.setLevel(logging.DEBUG) |
| 937 | confdata = Configuration(options) |
| 938 | initmode = (args[1] == 'init') |
| 939 | confdata.sanity_check(initmode) |
| 940 | actions.get(args[1], action_error)(confdata, args[1:]) |
| 941 | |
| 942 | if __name__ == "__main__": |
| 943 | try: |
| 944 | ret = main() |
| 945 | except Exception: |
| 946 | ret = 1 |
| 947 | import traceback |
| 948 | traceback.print_exc(5) |
| 949 | sys.exit(ret) |