1#!/usr/bin/env python
2# Copyright 2014 The Chromium Authors. All rights reserved.
3# Use of this source code is governed by a BSD-style license that can be
4# found in the LICENSE file.
5
6"""Rolls swarming_client.
7
8While it is currently hard coded for swarming_client/, it is potentially
9modifiable to allow different dependencies. Works only with git checkout and git
10dependencies.
11"""
12
13import optparse
14import os
15import re
16import subprocess
17import sys
18
19SRC_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
20
21
22def is_pristine(root, merge_base='origin/master'):
23  """Returns True is a git checkout is pristine."""
24  cmd = ['git', 'diff', '--ignore-submodules', merge_base]
25  return not (
26      subprocess.check_output(cmd, cwd=root).strip() or
27      subprocess.check_output(cmd + ['--cached'], cwd=root).strip())
28
29
30def roll(deps_dir, key, reviewer, bug):
31  if not is_pristine(SRC_ROOT):
32    print >> sys.stderr, 'Ensure %s is clean first.' % SRC_ROOT
33    return 1
34
35  full_dir = os.path.join(SRC_ROOT, deps_dir)
36  head = subprocess.check_output(
37      ['git', 'rev-parse', 'HEAD'], cwd=full_dir).strip()
38  deps = os.path.join(SRC_ROOT, 'DEPS')
39  with open(deps, 'rb') as f:
40    deps_content = f.read()
41
42  if not head in deps_content:
43    print('Warning: %s is not checked out at the expected revision in DEPS' %
44          deps_dir)
45    # It happens if the user checked out a branch in the dependency by himself.
46    # Fall back to reading the DEPS to figure out the original commit.
47    for i in deps_content.splitlines():
48      m = re.match(r'\s+"' + key + '": "([a-z0-9]{40})",', i)
49      if m:
50        head = m.group(1)
51        break
52    else:
53      print >> sys.stderr, 'Expected to find commit %s for %s in DEPS' % (
54          head, key)
55      return 1
56
57  print('Found old revision %s' % head)
58
59  subprocess.check_call(['git', 'fetch', 'origin'], cwd=full_dir)
60  master = subprocess.check_output(
61      ['git', 'rev-parse', 'origin/master'], cwd=full_dir).strip()
62  print('Found new revision %s' % master)
63
64  if master == head:
65    print('No revision to roll!')
66    return 1
67
68  commit_range = '%s..%s' % (head[:9], master[:9])
69  logs = subprocess.check_output(
70      ['git', 'log', commit_range, '--date=short', '--format=%ad %ae %s'],
71      cwd=full_dir).strip()
72  logs = logs.replace('@chromium.org', '')
73  cmd = (
74      'git log %s --date=short --format=\'%%ad %%ae %%s\' | '
75      'sed \'s/@chromium\.org//\'') % commit_range
76
77  msg = (
78      'Roll %s/ to %s.\n'
79      '\n'
80      '$ %s\n'
81      '%s\n\n'
82      'R=%s\n'
83      'BUG=%s') % (
84          deps_dir,
85          master,
86          cmd,
87          logs,
88          reviewer,
89          bug)
90
91  print('Commit message:')
92  print('\n'.join('    ' + i for i in msg.splitlines()))
93  deps_content = deps_content.replace(head, master)
94  with open(deps, 'wb') as f:
95    f.write(deps_content)
96  subprocess.check_call(['git', 'add', 'DEPS'], cwd=SRC_ROOT)
97  subprocess.check_call(['git', 'commit', '-m', msg], cwd=SRC_ROOT)
98  print('Run:')
99  print('  git cl upl --send-mail')
100  return 0
101
102
103def main():
104  parser = optparse.OptionParser(description=sys.modules[__name__].__doc__)
105  parser.add_option(
106      '-r', '--reviewer', default='',
107      help='To specify multiple reviewers, use comma separated list, e.g. '
108           '-r joe,jack,john. Defaults to @chromium.org')
109  parser.add_option('-b', '--bug', default='')
110  options, args = parser.parse_args()
111  if args:
112    parser.error('Unknown argument %s' % args)
113  if not options.reviewer:
114    parser.error('Pass a reviewer right away with -r/--reviewer')
115
116  reviewers = options.reviewer.split(',')
117  for i, r in enumerate(reviewers):
118    if not '@' in r:
119      reviewers[i] = r + '@chromium.org'
120
121  return roll(
122      'tools/swarming_client',
123      'swarming_revision',
124      ','.join(reviewers),
125      options.bug)
126
127
128if __name__ == '__main__':
129  sys.exit(main())
130