Index: build/android/pylib/utils/apk_helper.py |
diff --git a/build/android/pylib/utils/apk_helper.py b/build/android/pylib/utils/apk_helper.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..a556e7b31e66806bea8a65df7d7ca3825800bea6 |
--- /dev/null |
+++ b/build/android/pylib/utils/apk_helper.py |
@@ -0,0 +1,131 @@ |
+# Copyright (c) 2013 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. |
+ |
+"""Module containing utilities for apk packages.""" |
+ |
+import os.path |
+import re |
+ |
+from pylib import cmd_helper |
+from pylib import constants |
+from pylib.sdk import aapt |
+ |
+ |
+_AAPT_PATH = os.path.join(constants.ANDROID_SDK_TOOLS, 'aapt') |
+_MANIFEST_ATTRIBUTE_RE = re.compile( |
+ r'\s*A: ([^\(\)= ]*)\([^\(\)= ]*\)="(.*)" \(Raw: .*\)$') |
+_MANIFEST_ELEMENT_RE = re.compile(r'\s*(?:E|N): (\S*) .*$') |
+_PACKAGE_NAME_RE = re.compile(r'package: .*name=\'(\S*)\'') |
+_SPLIT_NAME_RE = re.compile(r'package: .*split=\'(\S*)\'') |
+ |
+ |
+def GetPackageName(apk_path): |
+ """Returns the package name of the apk.""" |
+ return ApkHelper(apk_path).GetPackageName() |
+ |
+ |
+# TODO(jbudorick): Deprecate and remove this function once callers have been |
+# converted to ApkHelper.GetInstrumentationName |
+def GetInstrumentationName(apk_path): |
+ """Returns the name of the Instrumentation in the apk.""" |
+ return ApkHelper(apk_path).GetInstrumentationName() |
+ |
+ |
+def _ParseManifestFromApk(apk_path): |
+ aapt_output = aapt.Dump('xmltree', apk_path, 'AndroidManifest.xml') |
+ |
+ parsed_manifest = {} |
+ node_stack = [parsed_manifest] |
+ indent = ' ' |
+ |
+ for line in aapt_output[1:]: |
+ if len(line) == 0: |
+ continue |
+ |
+ indent_depth = 0 |
+ while line[(len(indent) * indent_depth):].startswith(indent): |
+ indent_depth += 1 |
+ |
+ node_stack = node_stack[:indent_depth] |
+ node = node_stack[-1] |
+ |
+ m = _MANIFEST_ELEMENT_RE.match(line[len(indent) * indent_depth:]) |
+ if m: |
+ if not m.group(1) in node: |
+ node[m.group(1)] = {} |
+ node_stack += [node[m.group(1)]] |
+ continue |
+ |
+ m = _MANIFEST_ATTRIBUTE_RE.match(line[len(indent) * indent_depth:]) |
+ if m: |
+ if not m.group(1) in node: |
+ node[m.group(1)] = [] |
+ node[m.group(1)].append(m.group(2)) |
+ continue |
+ |
+ return parsed_manifest |
+ |
+ |
+class ApkHelper(object): |
+ def __init__(self, apk_path): |
+ self._apk_path = apk_path |
+ self._manifest = None |
+ self._package_name = None |
+ self._split_name = None |
+ |
+ def GetActivityName(self): |
+ """Returns the name of the Activity in the apk.""" |
+ manifest_info = self._GetManifest() |
+ try: |
+ activity = ( |
+ manifest_info['manifest']['application']['activity'] |
+ ['android:name'][0]) |
+ except KeyError: |
+ return None |
+ if '.' not in activity: |
+ activity = '%s.%s' % (self.GetPackageName(), activity) |
+ elif activity.startswith('.'): |
+ activity = '%s%s' % (self.GetPackageName(), activity) |
+ return activity |
+ |
+ def GetInstrumentationName( |
+ self, default='android.test.InstrumentationTestRunner'): |
+ """Returns the name of the Instrumentation in the apk.""" |
+ manifest_info = self._GetManifest() |
+ try: |
+ return manifest_info['manifest']['instrumentation']['android:name'][0] |
+ except KeyError: |
+ return default |
+ |
+ def GetPackageName(self): |
+ """Returns the package name of the apk.""" |
+ if self._package_name: |
+ return self._package_name |
+ |
+ aapt_output = aapt.Dump('badging', self._apk_path) |
+ for line in aapt_output: |
+ m = _PACKAGE_NAME_RE.match(line) |
+ if m: |
+ self._package_name = m.group(1) |
+ return self._package_name |
+ raise Exception('Failed to determine package name of %s' % self._apk_path) |
+ |
+ def GetSplitName(self): |
+ """Returns the name of the split of the apk.""" |
+ if self._split_name: |
+ return self._split_name |
+ |
+ aapt_output = aapt.Dump('badging', self._apk_path) |
+ for line in aapt_output: |
+ m = _SPLIT_NAME_RE.match(line) |
+ if m: |
+ self._split_name = m.group(1) |
+ return self._split_name |
+ return None |
+ |
+ def _GetManifest(self): |
+ if not self._manifest: |
+ self._manifest = _ParseManifestFromApk(self._apk_path) |
+ return self._manifest |
+ |