Add UMA metrics for blocked plugins.

chrome/tools/extract_actions.py

 #!/usr/bin/python
 # Copyright (c) 2011 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.
 
 """Extract UserMetrics "actions" strings from the Chrome source.
 
 This program generates the list of known actions we expect to see in the
 user behavior logs.  It walks the Chrome source, looking for calls to
 UserMetrics functions, extracting actions and warning on improper calls,
@@ skipping 287 matching lines @@
 
 def WalkDirectory(root_path, actions, extensions, callback):
   for path, dirs, files in os.walk(root_path):
     if '.svn' in dirs:
       dirs.remove('.svn')
     for file in files:
       ext = os.path.splitext(file)[1]
       if ext in extensions:
         callback(os.path.join(path, file), actions)
 
+def GrepForMsgActions(path, actions):
+  """Grep a source file for ViewHostMsg_UserMetricsRecordAction.
+
+  Arguments:
+    path: path to the file
+    actions: set of actions to add to
+  """
+  # We look for the ViewHostMsg_UserMetricsRecordAction constructor.
+  # This should be on one line.
+  action_re = re.compile(
+      r'[^a-zA-Z]ViewHostMsg_UserMetricsRecordAction\("([^"]*)')
+  line_number = 0
+  for line in open(path):
+    match = action_re.search(line)
+    if match:  # Plain call to RecordAction
+      actions.add(match.group(1))
+
 def AddLiteralActions(actions):
   """Add literal actions specified via calls to UserMetrics functions.
 
   Arguments:
     actions: set of actions to add to.
   """
   EXTENSIONS = ('.cc', '.mm', '.c', '.m')
 
   # Walk the source tree to process all .cc files.
   chrome_root = os.path.join(path_utils.ScriptDir(), '..')
   WalkDirectory(chrome_root, actions, EXTENSIONS, GrepForActions)
   content_root = os.path.join(path_utils.ScriptDir(), '..', '..', 'content')
   WalkDirectory(content_root, actions, EXTENSIONS, GrepForActions)
   webkit_root = os.path.join(path_utils.ScriptDir(), '..', '..', 'webkit')
   WalkDirectory(os.path.join(webkit_root, 'glue'), actions, EXTENSIONS,
                 GrepForActions)
   WalkDirectory(os.path.join(webkit_root, 'port'), actions, EXTENSIONS,
                 GrepForActions)
 
 def AddWebUIActions(actions):
   """Add user actions defined in WebUI files.
 
   Arguments:
     actions: set of actions to add to.
   """
   resources_root = os.path.join(path_utils.ScriptDir(), '..', 'browser',
                                 'resources')
   WalkDirectory(resources_root, actions, ('.html'), GrepForWebUIActions)
 
+def AddMsgActions(actions):
+  """Add user actions sent via ViewHostMsg_UserMetricsRecordAction.
+
+  Arguments:
+    actions: set of actions to add to.
+  """
+  EXTENSIONS = ('.cc', '.mm', '.c', '.m')
+
+  chrome_renderer_root = os.path.join(path_utils.ScriptDir(), '..', 'renderer')
+  content_renderer_root = os.path.join(path_utils.ScriptDir(), '..', '..',
+      'content', 'renderer')
+  WalkDirectory(chrome_renderer_root, actions, EXTENSIONS, GrepForMsgActions)
+  WalkDirectory(content_renderer_root, actions, EXTENSIONS, GrepForMsgActions)
+
 def main(argv):
   if '--hash' in argv:
     hash_output = True
   else:
     hash_output = False
     print >>sys.stderr, "WARNING: If you added new UMA tags, you must" + \
            " use the --hash option to update chromeactions.txt."
   # if we do a hash output, we want to only append NEW actions, and we know
   # the file we want to work on
   actions = set()
 
   chromeactions_path = os.path.join(path_utils.ScriptDir(), "chromeactions.txt")
 
   if hash_output:
     f = open(chromeactions_path)
     for line in f:
       part = line.rpartition("\t")
       part = part[2].strip()
       actions.add(part)
     f.close()
 
 
   AddComputedActions(actions)
   # TODO(fmantek): bring back webkit editor actions.
   # AddWebKitEditorActions(actions)
   AddAboutFlagsActions(actions)
   AddWebUIActions(actions)
+  AddMsgActions(actions)
 
   AddLiteralActions(actions)
 
   # print "Scanned {0} number of files".format(number_of_files_total)
   # print "Found {0} entries".format(len(actions))
 
   AddClosedSourceActions(actions)
   AddChromeOSActions(actions)
 
   if hash_output:
     f = open(chromeactions_path, "w")
 
 
   # Print out the actions as a sorted list.
   for action in sorted(actions):
     if hash_output:
       hash = hashlib.md5()
       hash.update(action)
       print >>f, '0x%s\t%s' % (hash.hexdigest()[:16], action)
     else:
       print action
 
   if hash_output:
     print "Done. Do not forget to add chromeactions.txt to your changelist"
 
 if '__main__' == __name__:
   main(sys.argv)