From a71cded95cbaa5150fef7e658155753de1fd1fbf Mon Sep 17 00:00:00 2001
From: Eric Seidel <eseidel@google.com>
Date: Mon, 17 Aug 2015 12:42:27 -0700
Subject: [PATCH] Split roll_versions.py into apk vs. pub

---
 .../{roll_versions.py => roll_apk_version.py} | 45 ++--------
 sky/tools/roll_pub_versions.py                | 83 +++++++++++++++++++
 2 files changed, 88 insertions(+), 40 deletions(-)
 rename sky/tools/{roll_versions.py => roll_apk_version.py} (67%)
 create mode 100755 sky/tools/roll_pub_versions.py

diff --git a/sky/tools/roll_versions.py b/sky/tools/roll_apk_version.py
similarity index 67%
rename from sky/tools/roll_versions.py
rename to sky/tools/roll_apk_version.py
index ccd15e56382..c225310eacc 100755
--- a/sky/tools/roll_versions.py
+++ b/sky/tools/roll_apk_version.py
@@ -5,20 +5,10 @@
 
 import os
 import subprocess
-import yaml
 import xml.etree.ElementTree as ET
+import argparse
 
 
-PUBSPECS = [
-    'sky/packages/sky/pubspec.yaml',
-    'sky/packages/sky_engine/pubspec.yaml',
-    'sky/packages/sky_services/pubspec.yaml',
-]
-
-MANIFESTS = [
-    'examples/demo_launcher/apk/AndroidManifest.xml',
-]
-
 MANIFEST_PREFACE = '''<?xml version="1.0" encoding="utf-8"?>
 <!-- Copyright 2015 The Chromium Authors. All rights reserved.
      Use of this source code is governed by a BSD-style license that can be
@@ -33,13 +23,6 @@ def increment_version(version):
     return '.'.join(pieces)
 
 
-def sort_dict(unsorted):
-    sorted_dict = collections.OrderedDict()
-    for key in sorted(unsorted.keys()):
-        sorted_dict[key] = unsorted[key]
-    return sorted_dict
-
-
 def count_commits(start, end):
     return subprocess.check_output([
         'git', 'rev-list', '%s...%s' % (start, end)]).count('\n')
@@ -49,19 +32,6 @@ def last_commit_to(file_path):
     return subprocess.check_output(['git', 'log', '-1', '--format=%h', file_path]).strip()
 
 
-def update_pubspec(pubspec):
-    # TODO(eseidel): This does not prserve any manual sort-order of the yaml.
-    with open(pubspec, 'r') as stream:
-        spec = yaml.load(stream)
-        old_version = spec['version']
-        spec['version'] = increment_version(old_version)
-        print "%20s  %6s => %6s" % (spec['name'], old_version, spec['version'])
-
-    with open(pubspec, 'w') as stream:
-        yaml.dump(spec, stream=stream, default_flow_style=False)
-    return spec['version']
-
-
 def update_changelog(changelog, pubspec, version):
     old = last_commit_to(pubspec)
     new = last_commit_to('.')
@@ -103,19 +73,14 @@ def update_manifest(manifest):
 
 def main():
     # Should chdir to the root directory.
-
-    print 'Pub packages:'
-    for pubspec in PUBSPECS:
-        new_version = update_pubspec(pubspec)
-        changelog = os.path.join(os.path.dirname(pubspec), 'CHANGELOG.md')
-        update_changelog(changelog, pubspec, new_version)
+    parser = argparse.ArgumentParser()
+    parser.add_argument('manifest')
+    args = parser.parse_args()
 
     # TODO(eseidel): Without this ET uses 'ns0' for 'android' which is wrong.
     ET.register_namespace('android', 'http://schemas.android.com/apk/res/android')
 
-    print 'APKs:'
-    for manifest in MANIFESTS:
-        update_manifest(manifest)
+    update_manifest(args.manifest)
 
 
 if __name__ == '__main__':
diff --git a/sky/tools/roll_pub_versions.py b/sky/tools/roll_pub_versions.py
new file mode 100755
index 00000000000..a23e4c3ed21
--- /dev/null
+++ b/sky/tools/roll_pub_versions.py
@@ -0,0 +1,83 @@
+#!/usr/bin/env python
+# Copyright 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 os
+import subprocess
+import yaml
+
+
+PUBSPECS = [
+    'sky/packages/sky/pubspec.yaml',
+    'sky/packages/sky_engine/pubspec.yaml',
+    'sky/packages/sky_services/pubspec.yaml',
+]
+
+def increment_version(version):
+    pieces = version.split('.')
+    pieces[-1] = str(int(pieces[-1]) + 1)
+    return '.'.join(pieces)
+
+
+def sort_dict(unsorted):
+    sorted_dict = collections.OrderedDict()
+    for key in sorted(unsorted.keys()):
+        sorted_dict[key] = unsorted[key]
+    return sorted_dict
+
+
+def count_commits(start, end):
+    return subprocess.check_output([
+        'git', 'rev-list', '%s...%s' % (start, end)]).count('\n')
+
+
+def last_commit_to(file_path):
+    return subprocess.check_output(['git', 'log', '-1', '--format=%h', file_path]).strip()
+
+
+def update_pubspec(pubspec):
+    # TODO(eseidel): This does not prserve any manual sort-order of the yaml.
+    with open(pubspec, 'r') as stream:
+        spec = yaml.load(stream)
+        old_version = spec['version']
+        spec['version'] = increment_version(old_version)
+        print "%20s  %6s => %6s" % (spec['name'], old_version, spec['version'])
+
+    with open(pubspec, 'w') as stream:
+        yaml.dump(spec, stream=stream, default_flow_style=False)
+    return spec['version']
+
+
+def update_changelog(changelog, pubspec, version):
+    old = last_commit_to(pubspec)
+    new = last_commit_to('.')
+    url = "https://github.com/domokit/mojo/compare/%s...%s" % (old, new)
+    count = count_commits(old, new)
+    message = """## %s
+
+  - %s changes: %s
+
+""" % (version, count, url)
+    prepend_to_file(message, changelog)
+
+
+def prepend_to_file(to_prepend, filepath):
+    with open(filepath, 'r+') as f:
+        content = f.read()
+        f.seek(0, 0)
+        f.write(to_prepend + content)
+
+
+def main():
+    # Should chdir to the root directory.
+
+    print 'Pub packages:'
+    for pubspec in PUBSPECS:
+        new_version = update_pubspec(pubspec)
+        changelog = os.path.join(os.path.dirname(pubspec), 'CHANGELOG.md')
+        update_changelog(changelog, pubspec, new_version)
+
+
+if __name__ == '__main__':
+    main()