Script for exporting WPT

third_party/WebKit/Tools/Scripts/webkitpy/w3c/local_wpt.py

Index: third_party/WebKit/Tools/Scripts/webkitpy/w3c/local_wpt.py
diff --git a/third_party/WebKit/Tools/Scripts/webkitpy/w3c/local_wpt.py b/third_party/WebKit/Tools/Scripts/webkitpy/w3c/local_wpt.py
new file mode 100644
index 0000000000000000000000000000000000000000..2590f7be291c9445a8ab42e42bfb34912a08fc8d
--- /dev/null
+++ b/third_party/WebKit/Tools/Scripts/webkitpy/w3c/local_wpt.py
@@ -0,0 +1,96 @@
+# 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.
+
+"""A utility class for interacting with a local checkout of the web-platform-tests."""
+
+import logging
+
+WPT_REPO_URL = 'https://chromium.googlesource.com/external/w3c/web-platform-tests.git'
+WPT_TMP_DIR = '/tmp/wpt'
+CHROMIUM_WPT_DIR = 'third_party/WebKit/LayoutTests/imported/wpt/'
+_log = logging.getLogger(__name__)
+
+
+class LocalWPT(object):
+
+    def __init__(self, host, path=WPT_TMP_DIR, no_fetch=False, use_github=False):
+        """
+        Args:
+            host: A Host object.
+            path: Optional, the directory where LocalWPT will check out web-platform-tests.
+            no_fetch: Optional, passing true will skip updating the local WPT.
+                Intended for use only in development after fetching once.
+            use_github: Optional, passing true will check if the GitHub remote is enabled
+                (necessary for later pull request steps).
+        """
+        self.host = host
+        self.path = path
+
+        if no_fetch:
+            _log.info('Skipping remote WPT fetch')
+            return
+
+        if self.host.filesystem.exists(self.path):
+            _log.info('WPT checkout exists at %s, fetching latest', self.path)
+            self.run(['git', 'checkout', 'master'])
+            self.run(['git', 'fetch', '--all'])
+            self.run(['git', 'merge', '--ff-only', 'origin/master'])
+        else:
+            _log.info('Cloning %s into %s', WPT_REPO_URL, self.path)
+            self.host.executive.run_command(['git', 'clone', WPT_REPO_URL, self.path])
+
+        if use_github and 'github' not in self.run(['git', 'remote']):
+            raise Exception('Need to set up remote "github"')
+
+    def run(self, command, **kwargs):
+        """Runs a command in the local WPT directory."""
+        return self.host.executive.run_command(command, cwd=self.path, **kwargs)
+
+    def most_recent_chromium_commit(self):
+        """Goes back in WPT commit history and gets the most recent commit
+        that contains 'Cr-Commit-Position:'
+        """
+        sha = self.run(['git', 'rev-list', 'HEAD', '-n', '1', '--grep=Cr-Commit-Position'])
+        if not sha:
+            return None, None
+
+        sha = sha.strip()
+        position = self.run(['git', 'footers', '--position', sha])
+        position = position.strip()
+        assert position
+        return sha, position
+
+    def clean(self):
+        self.run(['git', 'reset', '--hard', 'HEAD'])
+        self.run(['git', 'clean', '-fdx'])
+        self.run(['git', 'checkout', 'master'])
+
+    def all_branches(self):
+        return self.run(['git', 'branch', '-a']).splitlines()
+
+    def create_branch_with_patch(self, branch_name, message, patch):
+        self.clean()
+        all_branches = self.all_branches()
+        remote_branch_name = 'remotes/github/%s' % branch_name
+
+        if branch_name in all_branches:
+            _log.info('Local branch %s already exists, deleting', branch_name)
+            self.run(['git', 'branch', '-D', branch_name])
+
+        if remote_branch_name in all_branches:
+            _log.info('Remote branch %s already exists, deleting', branch_name)
+            # TODO(jeffcarp): Investigate what happens when remote branch exists
+            self.run(['git', 'push', 'github', ':{}'.format(branch_name)])
+
+        _log.info('Creating local branch %s', branch_name)
+        self.run(['git', 'checkout', '-b', branch_name])
+
+        # Remove Chromium WPT directory prefix
+        patch = patch.replace(CHROMIUM_WPT_DIR, '')
+
+        # TODO(jeffcarp): Use git am -p<n> where n is len(CHROMIUM_WPT_DIR.split(/'))
+        # or something not off-by-one.
+        self.run(['git', 'apply', '-'], input=patch)
+        self.run(['git', 'commit', '-am', message])
+        self.run(['git', 'push', 'github', branch_name])