1 files changed, 101 insertions, 0 deletions
diff --git a/catapult/systrace/systrace/update_systrace_trace_viewer.py b/catapult/systrace/systrace/update_systrace_trace_viewer.py
new file mode 100755
index 00000000..71eece03
--- /dev/null
+++ b/catapult/systrace/systrace/update_systrace_trace_viewer.py
@@ -0,0 +1,101 @@
+#!/usr/bin/env python
+
+# Copyright (c) 2015 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 codecs
+import optparse
+import os
+import re
+import subprocess
+import sys
+
+catapult_path = os.path.abspath(
+    os.path.join(os.path.dirname(__file__), '../..'))
+sys.path.append(os.path.join(catapult_path, 'tracing'))
+from tracing_build import vulcanize_trace_viewer
+
+SYSTRACE_TRACE_VIEWER_HTML_FILE_ = 'systrace_trace_viewer.html'
+CATAPULT_REV_ = 'CATAPULT_REV'
+NO_AUTO_UPDATE_ = 'NO_AUTO_UPDATE'
+
+
+def create_catapult_rev_str_(revision):
+    return '<!--' + CATAPULT_REV_ + '=' + str(revision) + '-->'
+
+def get_catapult_rev_in_file_():
+  assert os.path.exists(SYSTRACE_TRACE_VIEWER_HTML_FILE_)
+  rev = ''
+  with open(SYSTRACE_TRACE_VIEWER_HTML_FILE_, 'r') as f:
+    lines = f.readlines()
+    for line in lines[::-1]:
+      if CATAPULT_REV_ in line:
+        tokens = line.split(CATAPULT_REV_)
+        rev = re.sub('[=\->]', '', tokens[1]).strip()
+        break
+  return rev
+
+def get_catapult_rev_in_git_():
+  try:
+    return subprocess.check_output(
+        ['git', 'rev-parse', 'HEAD'],
+        cwd=os.path.dirname(os.path.abspath(__file__))).strip()
+  except subprocess.CalledProcessError:
+    return ''
+
+
+def update(no_auto_update=False, no_min=False):
+  """Update the systrace trace viewer html file.
+
+  When the html file exists, do not update the file if
+  1. the revision is NO_AUTO_UPDATE_;
+  2. or the revision is not changed.
+
+  Args:
+    no_auto_update: If true, force updating the file with revision
+                    NO_AUTO_UPDATE_. Future updates will be skipped unless this
+                    argument is true again.
+    no_min:         If true, skip minification when updating the file.
+  """
+  new_rev = None
+  if no_auto_update:
+    new_rev = NO_AUTO_UPDATE_
+  else:
+    new_rev = get_catapult_rev_in_git_()
+    if not new_rev:
+      return
+
+    if os.path.exists(SYSTRACE_TRACE_VIEWER_HTML_FILE_):
+      rev_in_file = get_catapult_rev_in_file_()
+      if rev_in_file == NO_AUTO_UPDATE_ or rev_in_file == new_rev:
+        return
+
+  # Generate the vulcanized result.
+  output_html_file = SYSTRACE_TRACE_VIEWER_HTML_FILE_
+  with codecs.open(output_html_file, encoding='utf-8', mode='w') as f:
+    vulcanize_trace_viewer.WriteTraceViewer(
+        f,
+        config_name='systrace',
+        minify=(not no_min),
+        output_html_head_and_body=False)
+    f.write(create_catapult_rev_str_(new_rev))
+  print 'Generated %s with revision %s.' % (output_html_file, new_rev)
+
+
+def main():
+  parser = optparse.OptionParser()
+  parser.add_option('--no-auto-update', dest='no_auto_update',
+                    default=False, action='store_true', help='force update the '
+                    'systrace trace viewer html file. Future auto updates will '
+                    'be skipped unless this flag is specified again.')
+  parser.add_option('--no-min', dest='no_min', default=False,
+                    action='store_true', help='skip minification')
+  # pylint: disable=unused-variable
+  options, unused_args = parser.parse_args(sys.argv[1:])
+
+  update(options.no_auto_update, options.no_min)
+
+
+if __name__ == '__main__':
+  main()