Chrome dialog audit

tools/ui/dialog_audit.py

+#!/usr/bin/env python
+# 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.
+
+"""
+Systematically search the Chrome codebase for dialog invocations.
+"""
+
+import subprocess
+import sys
+import re
+import mmap
+import os
+
+def InheritsRe(klass):
+  return r'public.*[: ]{}[, ]'.format(klass)
+
+def ChildRe(klass):
+  return r'class ([\w :]+)\s*:[^{]*' + klass + r'[^{]*[{]'
+
+def FindFilesInheritingFromClass(klass):
+  process = subprocess.Popen(['git', 'grep', '-l', InheritsRe(klass), '--',
+                              '*.h', '*.mm', '*.cc'],
+                             stdout=subprocess.PIPE)
+  return process.communicate()[0].splitlines();
+
+def ToLine(lines, offset):
+  line = 0
+  pos = 0
+  while pos < offset[0]:
+    pos += len(lines[line])
+    line += 1
+  return line
+
+def FindClassesInFilesInheritingFromClass(files, klass):
+  children = []
+  child_re = re.compile(ChildRe(klass), re.MULTILINE)
+  for filename in files:
+    with open(filename, 'r+') as f:
+      lines = f.readlines()
+      text = " ".join([l[:-1] for l in lines])
+      found = child_re.finditer(text)
+      before = len(children)
+      for child in found:
+        child_klass = child.group(1).split()[-1]
+        children.append((child_klass, filename, ToLine(lines, child.span()),
+                         re.sub(' +', ' ', child.group(0)), klass))
+      if len(children) == before:
+        sys.stderr.write(
+            "ERROR: Couldn't find a child of {} in {}\n".format(klass,
+                                                                filename))
+  return children
+
+def HuntForDecendants(result, parent, klasses, depth=''):
+  for klass in klasses:
+    result.append(klass)
+    sys.stderr.write('{}{}'.format(depth, klass[0]))
+    files = FindFilesInheritingFromClass(klass[0])
+    children = FindClassesInFilesInheritingFromClass(files, klass[0]);
+    sys.stderr.write( ' has {} children\n'.format(len(children)))
+    HuntForDecendants(result, klass, children, depth + '  ')
+
+def DefaultHunt():
+  seed = 'DialogDelegate'
+  #seed = 'DialogDelegateView'
+  #seed = 'LocationBarBubbleDelegateView'
+  result = []
+  HuntForDecendants(result, 'Seed', [(seed, '', 0, '', 'Seed')])
+  return result
+
+def ToUrl(filename, klass, line):
+  return 'https://cs.chromium.org/chromium/src/{}?q={}&l={}'.format(
+      filename, klass, line)
+
+def Process(data):
+  cells = eval("".join(open('tools/ui/cells.cache').readlines()))
+  cellmap = {}
+  for row in range(len(cells)):
+    if cells[row][0] == '' and cells[row][1] != '':
+      print "Unmatched: {} {}".format(row, cells[row][1])
+    else:
+      cellmap[cells[row][0]] = row
+  ignored = []
+  parents = {}
+  missing = {}
+  for (klass, filename, line, match, parent) in data:
+    if klass in cellmap:
+      row = cells[cellmap[klass]]
+      cells[cellmap[klass]] = (row[1], klass, )
+      print 'Matched: {}'.format(klass)
+      continue
+
+    # Ignore test files
+    if filename.find('test.') > 0:
+      ignored.append((klass, filename))
+      continue
+
+    # Track classes that have children and ignore them.
+    parents.setdefault(parent, [('f', 'c', 0), []])[1].append(klass)
+    if parent in missing:
+      parents[parent][0] = missing[parent]
+      del missing[parent]
+      continue
+
+    missing[klass] = (filename, klass, line)
+
+  sys.stderr.write('\nIgnored (Tests):')
+  for ignore in ignored:
+    sys.stderr.write(' {}({})'.format(ignore[0], os.path.basename(ignore[1])))
+  sys.stderr.write('\n')
+
+  sys.stderr.write('\nIgnored (Parents):')
+  for pk in parents:
+    parent = parents[pk]
+    filename, klass, line = parent[0]
+    children = " ".join(parent[1])
+    basename = os.path.basename(filename)
+    sys.stderr.write('\n\t {}({}): {}'.format(pk, basename, children))
+  sys.stderr.write('\n')
+
+  # Sort by filename.
+  missing_list = missing.values()
+  missing_list.sort();
+
+  for m in missing_list:
+    (filename, klass, line) = m
+    print 'Missing: {:40} from {}'.format(klass, ToUrl(filename, klass, line))
+
+def main():
+  try:
+    data = eval(open('tools/ui/dialog_audit.cache').readline())
+    Process(data)
+  except IOError as e:
+    data = DefaultHunt()
+    print data
+    sys.stderr.write("""
+Try `tools/ui/dialog_audit.py > tools/ui/dialog_audit.cache.new && \
+mv tools/ui/dialog_audit.cache.new tools/ui/dialog_audit.cache`
+""")
+
+if __name__ == '__main__':
+  sys.exit(main())