| # 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. | |
| import logging | |
| from webkitpy.w3c.local_wpt import LocalWPT | |
| from webkitpy.w3c.common import exportable_commits_since | |
| from webkitpy.w3c.wpt_github import WPTGitHub | |
| _log = logging.getLogger(__name__) | |
| class TestExporter(object): | |
| def __init__(self, host, gh_user, gh_token, dry_run=False): | |
| self.host = host | |
| self.wpt_github = WPTGitHub(host, gh_user, gh_token) | |
| self.dry_run = dry_run | |
| self.local_wpt = LocalWPT(self.host, gh_token) | |
| self.local_wpt.fetch() | |
| def run(self): | |
| """Query in-flight pull requests, then merge PR or create one. | |
| This script assumes it will be run on a regular interval. On | |
| each invocation, it will either attempt to merge or attempt to | |
| create a PR, never both. | |
| """ | |
| pull_requests = self.wpt_github.in_flight_pull_requests() | |
| if len(pull_requests) == 1: | |
| self.merge_in_flight_pull_request(pull_requests.pop()) | |
| elif len(pull_requests) > 1: | |
| _log.error(pull_requests) | |
| # TODO(jeffcarp): Print links to PRs | |
| raise Exception('More than two in-flight PRs!') | |
| else: | |
| self.export_first_exportable_commit() | |
| def merge_in_flight_pull_request(self, pull_request): | |
| """Attempt to merge an in-flight PR. | |
| Args: | |
| pull_request: a PR object returned from the GitHub API. | |
| """ | |
| _log.info('In-flight PR found: #%d', pull_request['number']) | |
| _log.info(pull_request['title']) | |
| # TODO(jeffcarp): Check the PR status here (for Travis CI, etc.) | |
| if self.dry_run: | |
| _log.info('[dry_run] Would have attempted to merge PR') | |
| return | |
| _log.info('Merging...') | |
| self.wpt_github.merge_pull_request(pull_request['number']) | |
| _log.info('PR merged! Deleting branch.') | |
| self.wpt_github.delete_remote_branch('chromium-export-try') | |
| _log.info('Branch deleted!') | |
| def export_first_exportable_commit(self): | |
| """Looks for exportable commits in Chromium, creates PR if found.""" | |
| wpt_commit, chromium_commit = self.local_wpt.most_recent_chromium_commit() | |
| assert chromium_commit, 'No Chromium commit found, this is impossible' | |
| wpt_behind_master = self.local_wpt.commits_behind_master(wpt_commit) | |
| _log.info('\nLast Chromium export commit in web-platform-tests:') | |
| _log.info('web-platform-tests@%s', wpt_commit) | |
| _log.info('(%d behind web-platform-tests@origin/master)', wpt_behind_master) | |
| _log.info('\nThe above WPT commit points to the following Chromium commit:') | |
| _log.info('chromium@%s', chromium_commit.sha) | |
| _log.info('(%d behind chromium@origin/master)', chromium_commit.num_behind_master()) | |
| exportable_commits = exportable_commits_since(chromium_commit.sha, self.host, self.local_wpt) | |
| if not exportable_commits: | |
| _log.info('No exportable commits found in Chromium, stopping.') | |
| return | |
| _log.info('Found %d exportable commits in Chromium:', len(exportable_commits)) | |
| for commit in exportable_commits: | |
| _log.info('- %s %s', commit, commit.subject()) | |
| outbound_commit = exportable_commits[0] | |
| _log.info('Picking the earliest commit and creating a PR') | |
| _log.info('- %s %s', outbound_commit.sha, outbound_commit.subject()) | |
| patch = outbound_commit.format_patch() | |
| message = outbound_commit.message() | |
| author = outbound_commit.author() | |
| if self.dry_run: | |
| _log.info('[dry_run] Stopping before creating PR') | |
| _log.info('\n\n[dry_run] message:') | |
| _log.info(message) | |
| _log.info('\n\n[dry_run] patch:') | |
| _log.info(patch) | |
| return | |
| remote_branch_name = self.local_wpt.create_branch_with_patch(message, patch, author) | |
| response_data = self.wpt_github.create_pr( | |
| remote_branch_name=remote_branch_name, | |
| desc_title=outbound_commit.subject(), | |
| body=outbound_commit.body()) | |
| _log.info('Create PR response: %s', response_data) | |
| if response_data: | |
| data, status_code = self.wpt_github.add_label(response_data['number']) | |
| _log.info('Add label response (status %s): %s', status_code, data) |