split out chromium_mojo recipe changes from chromium_mojo master cl

scripts/slave/recipe_modules/chromium/api.py

 # 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.
 
 import re
 
 from slave import recipe_api
 from slave import recipe_util
 
 from . import builders
@@ skipping 95 matching lines @@
                 "MAJOR=37\nMINOR=0\nBUILD=2021\nPATCH=0\n"))).stdout
     return self.version
 
   def set_build_properties(self, props):
     self._build_properties = props
 
   def add_builders(self, builders):
     """Adds builders to our builder map"""
     self._builders.update(builders)
 
+  def configure_bot(self, builders_dict, additional_configs=None):
+    """Sets up the configurations and gclient to be ready for bot update.
+
+    builders_dict is a dict of mastername -> 'builders' -> buildername ->
+        bot_config.
+
+    The current mastername and buildername are looked up from the
+    build properties; we then apply the configs specified in bot_config
+    as appropriate.
+
+    Returns a tuple of (buildername, bot_config) for subsequent use in
+       the recipe.
+    """
+    additional_configs = additional_configs or []
+
+    # TODO: crbug.com/358481 . The build_config should probably be a property
+    # passed in from the slave config, but that doesn't exist today, so we
+    # need a lookup mechanism to map bot name to build_config.
+    mastername = self.m.properties.get('mastername')
+    buildername = self.m.properties.get('buildername')
+    master_dict = builders_dict.get(mastername, {})
+    bot_config = master_dict.get('builders', {}).get(buildername)
+
+    self.set_config('chromium', **bot_config.get('chromium_config_kwargs', {}))
+
+    for c in bot_config.get('chromium_apply_config', []):
+      self.apply_config(c)
+
+    for c in additional_configs:
+      self.apply_config(c)
+
+    # Note that we have to call gclient.set_config() and apply_config() *after*
+    # calling chromium.set_config(), above, because otherwise the chromium
+    # call would reset the gclient config to its defaults.
+    self.m.gclient.set_config('chromium')
+    for c in bot_config.get('gclient_apply_config', []):
+      self.m.gclient.apply_config(c)
+
+    if bot_config.get('set_component_rev'):
+      # If this is a component build and the main revision is e.g. blink,
+      # webrtc, or v8, the custom deps revision of this component must be
+      # dynamically set to either:
+      # (1) 'revision' from the waterfall, or
+      # (2) 'HEAD' for forced builds with unspecified 'revision'.
+      component_rev = self.m.properties.get('revision') or 'HEAD'
+      dep = bot_config.get('set_component_rev')
+      self.m.gclient.c.revisions[dep['name']] = dep['rev_str'] % component_rev
+
+    if self.m.tryserver.is_tryserver:
+      self.m.step.auto_resolve_conflicts = True
+
+    return (buildername, bot_config)
+
   def compile(self, targets=None, name=None,
               force_clobber=False, **kwargs):
     """Return a compile.py invocation."""
     targets = targets or self.c.compile_py.default_targets.as_jsonish()
     assert isinstance(targets, (list, tuple))
 
     args = [
       '--target', self.c.build_config_fs,
       '--src-dir', self.m.path['checkout'],
     ]
@@ skipping 549 matching lines @@
   def get_compile_targets_for_scripts(self):
     return self.m.python(
         name='get compile targets for scripts',
         script=self.m.path['checkout'].join(
             'testing', 'scripts', 'get_compile_targets.py'),
         args=[
             '--output', self.m.json.output(),
             '--',
         ] + self.get_common_args_for_scripts(),
         step_test_data=lambda: self.m.json.test_api.output({}))