| #!/usr/bin/env python |
| # Copyright 2014 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Generate a CL to roll a DEPS entry to the specified revision number and post |
| it to Rietveld so that the CL will land automatically if it passes the |
| commit-queue's checks. |
| """ |
| |
| import logging |
| import optparse |
| import os |
| import re |
| import sys |
| |
| SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__)) |
| SRC_DIR = os.path.abspath(os.path.join(SCRIPT_DIR, os.pardir)) |
| sys.path.insert(0, os.path.join(SRC_DIR, 'build')) |
| import find_depot_tools |
| import scm |
| import subprocess2 |
| |
| |
| def die_with_error(msg): |
| print >> sys.stderr, msg |
| sys.exit(1) |
| |
| |
| def process_deps(path, project, new_rev, is_dry_run): |
| """Update project_revision to |new_issue|. |
| |
| A bit hacky, could it be made better? |
| """ |
| content = open(path).read() |
| # Hack for Blink to get the AutoRollBot running again. |
| if project == "blink": |
| project = "webkit" |
| old_line = r"(\s+)'%s_revision': '([0-9a-f]{2,40})'," % project |
| new_line = r"\1'%s_revision': '%s'," % (project, new_rev) |
| new_content = re.sub(old_line, new_line, content, 1) |
| old_rev = re.search(old_line, content).group(2) |
| if not old_rev or new_content == content: |
| die_with_error('Failed to update the DEPS file') |
| |
| if not is_dry_run: |
| open(path, 'w').write(new_content) |
| return old_rev |
| |
| |
| class PrintSubprocess(object): |
| """Wrapper for subprocess2 which prints out every command.""" |
| def __getattr__(self, attr): |
| def _run_subprocess2(cmd, *args, **kwargs): |
| print cmd |
| sys.stdout.flush() |
| return getattr(subprocess2, attr)(cmd, *args, **kwargs) |
| return _run_subprocess2 |
| |
| prnt_subprocess = PrintSubprocess() |
| |
| |
| def main(): |
| tool_dir = os.path.dirname(os.path.abspath(__file__)) |
| parser = optparse.OptionParser(usage='%prog [options] <project> <new rev>', |
| description=sys.modules[__name__].__doc__) |
| parser.add_option('-v', '--verbose', action='count', default=0) |
| parser.add_option('--dry-run', action='store_true') |
| parser.add_option('-f', '--force', action='store_true', |
| help='Make destructive changes to the local checkout if ' |
| 'necessary.') |
| parser.add_option('--commit', action='store_true', default=True, |
| help='(default) Put change in commit queue on upload.') |
| parser.add_option('--no-commit', action='store_false', dest='commit', |
| help='Don\'t put change in commit queue on upload.') |
| parser.add_option('-r', '--reviewers', default='', |
| help='Add given users as either reviewers or TBR as' |
| ' appropriate.') |
| parser.add_option('--upstream', default='origin/master', |
| help='(default "%default") Use given start point for change' |
| ' to upload. For instance, if you use the old git workflow,' |
| ' you might set it to "origin/trunk".') |
| parser.add_option('--cc', help='CC email addresses for issue.') |
| parser.add_option('-m', '--message', help='Custom commit message.') |
| |
| options, args = parser.parse_args() |
| logging.basicConfig( |
| level= |
| [logging.WARNING, logging.INFO, logging.DEBUG][ |
| min(2, options.verbose)]) |
| if len(args) != 2: |
| parser.print_help() |
| exit(0) |
| |
| root_dir = os.path.dirname(tool_dir) |
| os.chdir(root_dir) |
| |
| project = args[0] |
| new_rev = args[1] |
| |
| # Silence the editor. |
| os.environ['EDITOR'] = 'true' |
| |
| if options.force and not options.dry_run: |
| prnt_subprocess.check_call(['git', 'clean', '-d', '-f']) |
| prnt_subprocess.call(['git', 'rebase', '--abort']) |
| |
| old_branch = scm.GIT.GetBranch(root_dir) |
| new_branch = '%s_roll' % project |
| |
| if options.upstream == new_branch: |
| parser.error('Cannot set %s as its own upstream.' % new_branch) |
| |
| if old_branch == new_branch: |
| if options.force: |
| if not options.dry_run: |
| prnt_subprocess.check_call(['git', 'checkout', options.upstream, '-f']) |
| prnt_subprocess.call(['git', 'branch', '-D', old_branch]) |
| else: |
| parser.error('Please delete the branch %s and move to a different branch' |
| % new_branch) |
| |
| if not options.dry_run: |
| prnt_subprocess.check_call(['git', 'fetch', 'origin']) |
| prnt_subprocess.call(['git', 'svn', 'fetch']) |
| branch_cmd = ['git', 'checkout', '-b', new_branch, options.upstream] |
| if options.force: |
| branch_cmd.append('-f') |
| prnt_subprocess.check_output(branch_cmd) |
| |
| try: |
| old_rev = process_deps(os.path.join(root_dir, 'DEPS'), project, new_rev, |
| options.dry_run) |
| print '%s roll %s:%s' % (project.title(), old_rev, new_rev) |
| |
| review_field = 'TBR' if options.commit else 'R' |
| commit_msg = options.message or '%s roll %s:%s\n' % (project.title(), |
| old_rev, new_rev) |
| commit_msg += '\n%s=%s\n' % (review_field, options.reviewers) |
| |
| if options.dry_run: |
| print 'Commit message: ' + commit_msg |
| return 0 |
| |
| prnt_subprocess.check_output(['git', 'commit', '-m', commit_msg, 'DEPS']) |
| prnt_subprocess.check_call(['git', 'diff', '--no-ext-diff', |
| options.upstream]) |
| upload_cmd = ['git', 'cl', 'upload', '--bypass-hooks'] |
| if options.commit: |
| upload_cmd.append('--use-commit-queue') |
| if options.reviewers: |
| upload_cmd.append('--send-mail') |
| if options.cc: |
| upload_cmd.extend(['--cc', options.cc]) |
| prnt_subprocess.check_call(upload_cmd) |
| finally: |
| if not options.dry_run: |
| prnt_subprocess.check_output(['git', 'checkout', old_branch]) |
| prnt_subprocess.check_output(['git', 'branch', '-D', new_branch]) |
| return 0 |
| |
| |
| if __name__ == '__main__': |
| sys.exit(main()) |