Bot to update buildbucket.config daily

Bug: skia:6591
Change-Id: Ic2ec5b033965ca073a00e272693b1c5e1d0ad69f
Reviewed-on: https://skia-review.googlesource.com/16541
Commit-Queue: Ravi Mistry <rmistry@google.com>
Reviewed-by: Eric Boren <borenet@google.com>

1 files changed, 91 insertions, 0 deletions

diff --git a/infra/bots/update_meta_config.py b/infra/bots/update_meta_config.py
new file mode 100644
index 0000000000..663d170c33
--- /dev/null
+++ b/infra/bots/update_meta_config.py
@@ -0,0 +1,91 @@
+# Copyright 2017 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.
+
+"""Update meta/config of the specified Skia repo."""
+
+
+import argparse
+import json
+import os
+import subprocess
+import sys
+import urllib2
+
+import git_utils
+
+
+SKIA_COMMITTER_EMAIL = 'update-meta-config@skia.org'
+SKIA_COMMITTER_NAME = 'Update Meta Config'
+SKIA_REPO_TEMPLATE = 'https://skia.googlesource.com/%s.git'
+
+CQ_INCLUDE_CHROMIUM_TRYBOTS = (
+    'master.tryserver.blink:linux_trusty_blink_rel,linux_trusty_blink_dbg;'
+    'master.tryserver.chromium.linux:linux_optional_gpu_tests_rel;'
+    'master.tryserver.chromium.mac:mac_optional_gpu_tests_rel;'
+    'master.tryserver.chromium.win:win_optional_gpu_tests_rel;'
+    'master.tryserver.chromium.android:android_optional_gpu_tests_rel'
+)
+
+
+def addChromiumTrybots(f):
+  for master_section in CQ_INCLUDE_CHROMIUM_TRYBOTS.split(';'):
+    master, bots = master_section.split(':')
+    f.write('[bucket "%s"]\n' % master)
+    for bot in bots.split(','):
+      f.write('\tbuilder = %s\n' % bot)
+
+
+def main(gitcookies, repo_name, tasks_json):
+  skia_repo = SKIA_REPO_TEMPLATE % repo_name
+  with git_utils.NewGitCheckout(repository=skia_repo):
+    # Fetch and checkout the meta/config branch.
+    subprocess.check_call(['git', 'fetch', skia_repo, 'refs/meta/config:cfg'])
+    subprocess.check_call(['git', 'checkout', 'cfg'])
+
+    # Create list of tryjobs from tasks_json.
+    tryjobs = []
+    with open(tasks_json) as tasks_json:
+      data = json.load(tasks_json)
+      for job in data['jobs'].keys():
+        if not job.startswith('Upload-'):
+          tryjobs.append(job)
+    tryjobs.sort()
+
+    # Write to buildbucket.config.
+    buildbucket_config = os.path.join(os.getcwd(), 'buildbucket.config')
+    with open(buildbucket_config, 'w') as f:
+
+      if repo_name == 'skia':
+        addChromiumTrybots(f)
+
+      # Adding all Skia jobs.
+      f.write('[bucket "skia.primary"]\n')
+      for job in tryjobs:
+        f.write('\tbuilder = ' + job + '\n')
+
+    # Push the change as the update-meta-config user.
+    config_dict = {
+      'user.name': SKIA_COMMITTER_NAME,
+      'user.email': SKIA_COMMITTER_EMAIL,
+      'http.cookiefile': gitcookies,
+    }
+    with git_utils.GitLocalConfig(config_dict):
+      subprocess.check_call(['git', 'add', 'buildbucket.config'])
+      try:
+        subprocess.check_call(
+            ['git', 'commit', '-m', 'Update builders in buildbucket.config'])
+      except subprocess.CalledProcessError:
+        print 'No changes to buildbucket.config'
+        return
+
+      subprocess.check_call(['git', 'push', skia_repo, 'cfg:refs/meta/config'])
+
+
+if '__main__' == __name__:
+  parser = argparse.ArgumentParser()
+  parser.add_argument("--gitcookies")
+  parser.add_argument("--repo_name")
+  parser.add_argument("--tasks_json")
+  args = parser.parse_args()
+  main(args.gitcookies, args.repo_name, args.tasks_json)