Add got_revision support for apply_issue.
BUG=170427
Review URL: https://codereview.chromium.org/60613005
git-svn-id: svn://svn.chromium.org/chrome/trunk/tools/depot_tools@236101 0039d316-1c4b-4281-b951-d872f2087c98
diff --git a/annotated_gclient.py b/annotated_gclient.py
new file mode 100755
index 0000000..1ed0f71
--- /dev/null
+++ b/annotated_gclient.py
@@ -0,0 +1,89 @@
+#!/usr/bin/env python
+# Copyright 2013 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.
+
+"""Wraps gclient calls with annotated output.
+
+Note that you will have to use -- to stop option parsing for gclient flags.
+
+To run `gclient sync --gclientfile=.gclient` and annotate got_v8_revision:
+ `annotated_gclient.py --revision-mapping='{"src/v8": "got_v8_revision"}' --
+ sync --gclientfile=.gclient`
+"""
+
+import contextlib
+import json
+import optparse
+import os
+import subprocess
+import sys
+import tempfile
+
+
+@contextlib.contextmanager
+def temp_filename(suffix='', prefix='tmp'):
+ output_fd, output_file = tempfile.mkstemp(suffix=suffix, prefix=prefix)
+ os.close(output_fd)
+
+ yield output_file
+
+ try:
+ os.remove(output_file)
+ except OSError as e:
+ print 'Error cleaning up temp file %s: %s' % (output_file, e)
+
+
+def parse_got_revision(filename, revision_mapping):
+ result = {}
+ with open(filename) as f:
+ data = json.load(f)
+
+ for path, info in data['solutions'].iteritems():
+ # gclient json paths always end with a slash
+ path = path.rstrip('/')
+ if path in revision_mapping:
+ propname = revision_mapping[path]
+ result[propname] = info['revision']
+
+ return result
+
+
+def emit_buildprops(got_revisions):
+ for prop, revision in got_revisions.iteritems():
+ print '@@@SET_BUILD_PROPERTY@%s@%s@@@' % (prop, revision)
+
+
+def main():
+ parser = optparse.OptionParser(
+ description=('Runs gclient and annotates the output with any '
+ 'got_revisions.'))
+ parser.add_option('--revision-mapping', default='{}',
+ help='json dict of directory-to-property mappings.')
+ parser.add_option('--suffix', default='gclient',
+ help='tempfile suffix')
+ opts, args = parser.parse_args()
+
+ revision_mapping = json.loads(opts.revision_mapping)
+
+ if not args:
+ parser.error('Must provide arguments to gclient.')
+
+ if any(a.startswith('--output-json') for a in args):
+ parser.error('Can\'t call annotated_gclient with --output-json.')
+
+ with temp_filename(opts.suffix) as f:
+ cmd = ['gclient']
+ cmd.extend(args)
+ cmd.extend(['--output-json', f])
+ p = subprocess.Popen(cmd)
+ p.wait()
+
+ if p.returncode == 0:
+ revisions = parse_got_revision(f, revision_mapping)
+ emit_buildprops(revisions)
+ return p.returncode
+
+
+if __name__ == '__main__':
+ sys.exit(main())