#!/usr/bin/env python3
# Copyright (c) 2018, the R8 project authors. Please see the AUTHORS file
# for details. All rights reserved. Use of this source code is governed by a
# BSD-style license that can be found in the LICENSE file.

import optparse
import os
import shutil
import subprocess
import sys
import time

import utils
import zip_utils

USAGE = 'usage: %prog [options] <apk>'


def parse_options():
    parser = optparse.OptionParser(usage=USAGE)
    parser.add_option('--keystore',
                      help='keystore file (default ~/.android/app.keystore)',
                      default='~/.android/app.keystore')
    parser.add_option('--sign',
                      help='Sign the passed in apk.',
                      default=False,
                      action='store_true')
    parser.add_option('--use_apksigner',
                      help='Use apksigner to sign.',
                      default=False,
                      action='store_true')
    parser.add_option('--output',
                      help='Where to put the signed apk.)',
                      default=None)

    (options, args) = parser.parse_args()
    if len(args) != 1:
        parser.error('Expected <apk> argument, got: ' + ' '.join(args))
    apk = args[0]
    return options, apk


def add_baseline_profile_to_apk(apk, baseline_profile,
                                baseline_profile_metadata, tmp_dir):
    if baseline_profile is None:
        return apk
    ts = time.time_ns()
    dest_apk = os.path.join(tmp_dir, 'app-%s.apk' % ts)
    dest_apk_aligned = os.path.join(tmp_dir, 'app-aligned-%s.apk' % ts)
    dest_apk_signed = os.path.join(tmp_dir, 'app-signed-%s.apk' % ts)
    shutil.copy2(apk, dest_apk)
    zip_utils.remove_files_from_zip(
        ['assets/dexopt/baseline.prof', 'assets/dexopt/baseline.profm'],
        dest_apk)
    zip_utils.add_file_to_zip(baseline_profile, 'assets/dexopt/baseline.prof',
                              dest_apk)
    if baseline_profile_metadata is not None:
        zip_utils.add_file_to_zip(baseline_profile_metadata,
                                  'assets/dexopt/baseline.profm', dest_apk)
    align(dest_apk, dest_apk_aligned)
    sign_with_apksigner(dest_apk_aligned, dest_apk_signed)
    return dest_apk_signed


def align(apk, aligned_apk):
    zipalign_path = ('zipalign' if 'build_tools' in os.environ.get('PATH') else
                     os.path.join(utils.getAndroidBuildTools(), 'zipalign'))
    cmd = [zipalign_path, '-f', '-p', '4', apk, aligned_apk]
    utils.RunCmd(cmd, quiet=True, logging=False)
    return aligned_apk


def default_keystore():
    return os.path.join(os.getenv('HOME'), '.android', 'app.keystore')


def get_min_api(apk):
    aapt = os.path.join(utils.getAndroidBuildTools(), 'aapt')
    cmd = [aapt, 'dump', 'badging', apk]
    stdout = subprocess.check_output(cmd).decode('utf-8').strip()
    for line in stdout.splitlines():
        if line.startswith('sdkVersion:\''):
            return int(line[len('sdkVersion:\''):-1])
    raise ValueError('Unexpected stdout: %s' % stdout)


def sign(unsigned_apk, signed_apk, keystore, quiet=False, logging=True):
    utils.Print('Signing (ignore the warnings)', quiet=quiet)
    cmd = ['zip', '-d', unsigned_apk, 'META-INF/*']
    utils.RunCmd(cmd, quiet=quiet, logging=logging, fail=False)
    cmd = [
        'jarsigner', '-sigalg', 'SHA1withRSA', '-digestalg', 'SHA1',
        '-keystore', keystore, '-storepass', 'android', '-signedjar',
        signed_apk, unsigned_apk, 'androiddebugkey'
    ]
    utils.RunCmd(cmd, quiet=quiet)


def sign_with_apksigner(unsigned_apk,
                        signed_apk,
                        keystore=None,
                        password='android',
                        quiet=False,
                        logging=True):
    cmd = [
        os.path.join(utils.getAndroidBuildTools(), 'apksigner'), 'sign', '-v',
        '--ks', keystore or default_keystore(), '--ks-pass', 'pass:' + password,
        '--min-sdk-version', '19', '--out', signed_apk, '--v2-signing-enabled',
        unsigned_apk
    ]
    utils.RunCmd(cmd, quiet=quiet, logging=logging)
    return signed_apk


def main():
    (options, apk) = parse_options()
    if options.sign:
        if not options.output:
            print('When signing you must specify an output apk')
            return 1
        if not options.keystore:
            print('When signing you must specify a keystore')
            return 1
        if options.use_apksigner:
            sign_with_apksigner(apk, options.output, options.keystore)
        else:
            sign(apk, options.output, options.keystore)
    return 0


if __name__ == '__main__':
    sys.exit(main())