Chromium Code Reviews| Index: scripts/slave/recipe_modules/chromium_checkout/api.py |
| diff --git a/scripts/slave/recipe_modules/chromium_checkout/api.py b/scripts/slave/recipe_modules/chromium_checkout/api.py |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..8de28003b29af264f91927154ff9c1f1e17514d7 |
| --- /dev/null |
| +++ b/scripts/slave/recipe_modules/chromium_checkout/api.py |
| @@ -0,0 +1,78 @@ |
| +# Copyright 2016 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. |
| + |
| +from recipe_engine import recipe_api |
| + |
| + |
| +class ChromiumCheckoutApi(recipe_api.RecipeApi): |
| + def __init__(self, *args, **kwargs): |
| + super(ChromiumCheckoutApi, self).__init__(*args, **kwargs) |
| + # Keep track of working directory (which contains the checkout). |
| + # None means "default value". |
| + self._working_dir = None |
| + |
| + @property |
| + def working_dir(self): |
| + """Returns parent directory of the checkout. |
| + |
| + Requires |ensure_checkout| to be called first. |
|
Sergiy Byelozyorov
2016/08/12 13:22:24
perhaps add an assert?
|
| + """ |
| + return self._working_dir |
| + |
| + def get_checkout_dir(self, bot_config): |
| + """Returns directory where checkout can be created. |
| + |
| + None means to use default checkout directory. |
| + """ |
| + try: |
| + builder_cache = self.m.path['builder_cache'] |
| + except KeyError: # no-op if builder cache is not set up. |
| + return None |
| + else: |
| + sanitized_buildername = ''.join( |
| + c if c.isalnum() else '_' for c in self.m.properties['buildername']) |
| + checkout_dir = builder_cache.join( |
| + bot_config.get('checkout_dir', sanitized_buildername)) |
| + self.m.shutil.makedirs('checkout path', checkout_dir) |
| + return checkout_dir |
| + |
| + def get_files_affected_by_patch(self, relative_to='src/', cwd=None): |
| + """Returns list of POSIX paths of files affected by patch for "analyze". |
| + |
| + Paths are relative to `relative_to` which for analyze should be 'src/'. |
| + """ |
| + patch_root = self.m.gclient.calculate_patch_root( |
| + self.m.properties.get('patch_project')) |
| + if not cwd and self.working_dir: |
| + cwd = self.working_dir.join(patch_root) |
|
Sergiy Byelozyorov
2016/08/12 13:22:24
This would require having working_dir initialized.
Sergiy Byelozyorov
2016/08/12 13:41:41
As Pawel pointed out, the if self._working_dir che
|
| + files = self.m.tryserver.get_files_affected_by_patch(patch_root, cwd=cwd) |
| + for i, path in enumerate(files): |
| + path = str(path) |
| + assert path.startswith(relative_to) |
| + files[i] = path[len(relative_to):] |
| + return files |
| + |
| + def ensure_checkout(self, bot_config, root_solution_revision=None, |
| + force=False): |
| + """Wrapper for bot_update.ensure_checkout with chromium-specific additions. |
| + """ |
| + if self.m.platform.is_win: |
| + self.m.chromium.taskkill() |
| + |
| + kwargs = {} |
| + self._working_dir = self.get_checkout_dir(bot_config) |
| + if self._working_dir: |
| + kwargs['cwd'] = self._working_dir |
| + |
| + # Bot Update re-uses the gclient configs. |
| + update_step = self.m.bot_update.ensure_checkout( |
| + patch_root=bot_config.get('patch_root'), |
| + root_solution_revision=root_solution_revision, |
| + clobber=bot_config.get('clobber', False), |
| + force=force, **kwargs) |
| + assert update_step.json.output['did_run'] |
| + # HACK(dnj): Remove after 'crbug.com/398105' has landed |
| + self.m.chromium.set_build_properties(update_step.json.output['properties']) |
| + |
| + return update_step |