Extract test directory owner email addresses from W3CImportExpectations.

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

Index: third_party/WebKit/Tools/Scripts/webkitpy/w3c/directory_owners_extractor.py
diff --git a/third_party/WebKit/Tools/Scripts/webkitpy/w3c/directory_owners_extractor.py b/third_party/WebKit/Tools/Scripts/webkitpy/w3c/directory_owners_extractor.py
new file mode 100644
index 0000000000000000000000000000000000000000..b6c9329c428b926996ffa7c77b9f9ff042b7f744
--- /dev/null
+++ b/third_party/WebKit/Tools/Scripts/webkitpy/w3c/directory_owners_extractor.py
@@ -0,0 +1,75 @@
+# 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.
+
+import re
+
+from webkitpy.common.system.filesystem import FileSystem
+from webkitpy.common.webkit_finder import WebKitFinder
+
+
+class DirectoryOwnersExtractor(object):
+
+    def __init__(self, filesystem=None):
+        self.filesystem = filesystem or FileSystem
+        self.finder = WebKitFinder(filesystem)
+        self.owner_map = None
+
+    def read_owner_map(self):
+        """Reads the W3CImportExpectations file and returns a map of directories to owners."""
+        input_path = self.finder.path_from_webkit_base('LayoutTests', 'W3CImportExpectations')
+        input_contents = self.filesystem.read_text_file(input_path)
+        self.owner_map = self.lines_to_owner_map(input_contents.splitlines())
+
+    def lines_to_owner_map(self, lines):
+        current_owners = []
+        owner_map = {}
+        for line in lines:
+            owners = self.extract_owners(line)
+            if owners:
+                current_owners = owners
+            directory = self.extract_directory(line)
+            if current_owners and directory:
+                owner_map[directory] = current_owners
+        return owner_map
+
+    @staticmethod
+    def extract_owners(line):
+        """Extracts owner email addresses listed on a line."""
+        match = re.match(r'##? Owners?: (?P<addresses>.*)', line)
+        if not match or not match.group('addresses'):
+            return None
+        email_part = match.group('addresses')
+        addresses = [email_part] if ',' not in email_part else re.split(r',\s*', email_part)
+        addresses = [s for s in addresses if re.match(r'\S+@\S+', s)]
+        return addresses or None
+
+    @staticmethod
+    def extract_directory(line):
+        match = re.match(r'# ?(?P<directory>\S+) \[ (Pass|Skip) \]', line)
+        if match and match.group('directory'):
+            return match.group('directory')
+        match = re.match(r'(?P<directory>\S+) \[ Pass \]', line)
+        if match and match.group('directory'):
+            return match.group('directory')
+        return None
+
+    def list_owners(self, changed_files):
+        """Looks up the owners for the given set of changed files.
+
+        Args:
+            changed_files: A list of file paths relative to the repository root.
+
+        Returns:
+            A dict mapping (owner) email addresses to (owned) directories.
+        """
+        tests = [self.finder.layout_test_name(path) for path in changed_files]
+        print tests
+        tests = [t for t in tests if t is not None]
+        email_map = {}
+        for directory, owners in self.owner_map.iteritems():
+            owned_tests = [t for t in tests if t.startswith(directory)]
+            if owned_tests:
+                for owner in owners:
+                    email_map[owner] = directory
+        return email_map