blob: 61e0b5d30ea2c0a0a9750bb9910b727dbf42f1dc [file] [log] [blame]
#!/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:
` --revision-mapping='{"src/v8": "got_v8_revision"}' --
sync --gclientfile=.gclient`
import contextlib
import json
import optparse
import os
import subprocess
import sys
import tempfile
def temp_filename(suffix='', prefix='tmp'):
output_fd, output_file = tempfile.mkstemp(suffix=suffix, prefix=prefix)
yield 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, json.dumps(revision))
def main():
parser = optparse.OptionParser(
description=('Runs gclient and annotates the output with any '
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(['--output-json', f])
p = subprocess.Popen(cmd)
if p.returncode == 0:
revisions = parse_got_revision(f, revision_mapping)
return p.returncode
if __name__ == '__main__':