blob: d467a810773c10e84b8398fbf96eb3db569b3288 [file] [log] [blame]
#!/usr/bin/env python3
# Copyright (c) 2019, the R8 project authors. Please see the AUTHORS file
# for details. All rights reserved. Use of this source code is governed by a
# BSD-style license that can be found in the LICENSE file.
# Convenience script for triggering bots on specific commits.
import json
import git_utils
import optparse
import os
import re
import subprocess
import sys
import urllib
from urllib.request import urlopen
import utils
LUCI_SCHEDULE = os.path.join(utils.REPO_ROOT, 'infra', 'config', 'global',
'generated', 'luci-scheduler.cfg')
# Trigger lines have the format:
# triggers: "BUILDER_NAME"
TRIGGERS_RE = r'^ triggers: "(\w.*)"'
DESUGAR_BOT = 'lib_desugar-archive'
def ParseOptions():
result = optparse.OptionParser()
result.add_option('--release',
help='Run on the release branch builders.',
default=False, action='store_true')
result.add_option('--cl',
help='Run the specified cl on the bots. This should be '
'the full url, e.g., '
'https://r8-review.googlesource.com/c/r8/+/37420/1')
result.add_option('--desugar',
help='Run the library desugar and archiving bot.',
default=False, action='store_true')
result.add_option('--builder', help='Trigger specific builder')
return result.parse_args()
def get_builders():
is_release = False
main_builders = []
release_builders = []
with open(LUCI_SCHEDULE, 'r') as fp:
lines = fp.readlines()
for line in lines:
if 'branch-gitiles' in line:
is_release = True
if 'main-gitiles-trigger' in line:
is_release = False
match = re.match(TRIGGERS_RE, line)
if match:
builder = match.group(1)
if is_release:
assert 'release' in builder, builder
release_builders.append(builder)
else:
assert 'release' not in builder, builder
main_builders.append(builder)
print('Desugar builder:\n ' + DESUGAR_BOT)
print('Main builders:\n ' + '\n '.join(main_builders))
print('Release builders:\n ' + '\n '.join(release_builders))
return (main_builders, release_builders)
def sanity_check_url(url):
a = urlopen(url)
if a.getcode() != 200:
raise Exception('Url: %s \n returned %s' % (url, a.getcode()))
def trigger_builders(builders, commit):
commit_url = 'https://r8.googlesource.com/r8/+/%s' % commit
sanity_check_url(commit_url)
for builder in builders:
cmd = ['bb', 'add', 'r8/ci/%s' % builder , '-commit', commit_url]
subprocess.check_call(cmd)
def trigger_cl(builders, cl_url):
for builder in builders:
cmd = ['bb', 'add', 'r8/ci/%s' % builder , '-cl', cl_url]
subprocess.check_call(cmd)
def Main():
(options, args) = ParseOptions()
if len(args) != 1 and not options.cl and not options.desugar:
print('Takes exactly one argument, the commit to run')
return 1
if options.cl and options.release:
print('You can\'t run cls on the release bots')
return 1
if options.cl and options.desugar:
print('You can\'t run cls on the desugar bot')
return 1
commit = None if (options.cl or options.desugar) else args[0]
(main_builders, release_builders) = get_builders()
builders = release_builders if options.release else main_builders
if options.builder:
builder = options.builder
assert builder in main_builders or builder in release_builders
builders = [options.builder]
if options.desugar:
builders = [DESUGAR_BOT]
commit = git_utils.GetHeadRevision(utils.REPO_ROOT, use_main=True)
if options.cl:
trigger_cl(builders, options.cl)
else:
assert commit
trigger_builders(builders, commit)
if __name__ == '__main__':
sys.exit(Main())