63 lines
2.0 KiB
Python
Executable File
63 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# Copyright 2022 The Chromium Authors
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
"""Generates APKs for use on system images."""
|
|
|
|
import argparse
|
|
import os
|
|
import pathlib
|
|
import tempfile
|
|
import shutil
|
|
import sys
|
|
import zipfile
|
|
|
|
_DIR_SOURCE_ROOT = str(pathlib.Path(__file__).parents[2])
|
|
sys.path.append(os.path.join(_DIR_SOURCE_ROOT, 'build', 'android', 'gyp'))
|
|
from util import build_utils
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--input', required=True, help='Input path')
|
|
parser.add_argument('--output', required=True, help='Output path')
|
|
parser.add_argument('--bundle-wrapper', help='APK operations script path')
|
|
parser.add_argument('--fuse-apk',
|
|
help='Create single .apk rather than using apk splits',
|
|
action='store_true')
|
|
args = parser.parse_args()
|
|
|
|
if not args.bundle_wrapper:
|
|
shutil.copyfile(args.input, args.output)
|
|
return
|
|
|
|
with tempfile.NamedTemporaryFile(suffix='.apks') as tmp_file:
|
|
cmd = [
|
|
args.bundle_wrapper, 'build-bundle-apks', '--output-apks', tmp_file.name
|
|
]
|
|
cmd += ['--build-mode', 'system' if args.fuse_apk else 'system_apks']
|
|
|
|
# Creates a .apks zip file that contains the system image APK(s).
|
|
build_utils.CheckOutput(cmd)
|
|
|
|
if args.fuse_apk:
|
|
with zipfile.ZipFile(tmp_file.name) as z:
|
|
pathlib.Path(args.output).write_bytes(z.read('system/system.apk'))
|
|
return
|
|
|
|
# Rename .apk files and remove toc.pb to make it clear that system apks
|
|
# should not be installed via bundletool.
|
|
with zipfile.ZipFile(tmp_file.name) as z_input, \
|
|
zipfile.ZipFile(args.output, 'w') as z_output:
|
|
for info in z_input.infolist():
|
|
if info.filename.endswith('.apk'):
|
|
data = z_input.read(info)
|
|
info.filename = (info.filename.replace('splits/',
|
|
'').replace('-master', ''))
|
|
z_output.writestr(info, data)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|