# Copyright 2013 The Flutter 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 sys

import argparse
import os
import zipfile
import subprocess


def run_command_checked(command):
  try:
    subprocess.check_output(command, stderr=subprocess.STDOUT, text=True)
  except subprocess.CalledProcessError as cpe:
    print(cpe.output)
    raise cpe


def main():
  parser = argparse.ArgumentParser()

  parser.add_argument('--aapt2-bin', type=str, required=True, help='The path to the aapt2 binary.')
  parser.add_argument(
      '--zipalign-bin', type=str, required=True, help='The path to the zipalign binary.'
  )
  parser.add_argument(
      '--apksigner-bin', type=str, required=True, help='The path to the apksigner binary.'
  )
  parser.add_argument(
      '--android-manifest', type=str, required=True, help='The path to the AndroidManifest.xml.'
  )
  parser.add_argument('--android-jar', type=str, required=True, help='The path to android.jar.')
  parser.add_argument('--output-path', type=str, required=True, help='The path to the output apk.')
  parser.add_argument(
      '--library', type=str, required=True, help='The path to the library to put in the apk.'
  )
  parser.add_argument(
      '--keystore', type=str, required=True, help='The path to the debug keystore to sign the apk.'
  )
  parser.add_argument(
      '--gen-dir', type=str, required=True, help='The directory for generated files.'
  )
  parser.add_argument(
      '--android-abi', type=str, required=True, help='The android ABI of the library.'
  )

  args = parser.parse_args()

  library_file = os.path.basename(args.library)
  apk_name = os.path.basename(args.output_path)

  unaligned_apk_path = os.path.join(args.gen_dir, '%s.unaligned' % apk_name)
  unsigned_apk_path = os.path.join(args.gen_dir, '%s.unsigned' % apk_name)
  apk_path = args.output_path

  # Create the skeleton of the APK using aapt2.
  aapt2_command = [
      args.aapt2_bin,
      'link',
      '-I',
      args.android_jar,
      '--manifest',
      args.android_manifest,
      '-o',
      unaligned_apk_path,
  ]
  run_command_checked(aapt2_command)

  # Stuff the library in the APK which is just a regular ZIP file. Libraries are not compressed.
  with zipfile.ZipFile(unaligned_apk_path, 'a', compression=zipfile.ZIP_STORED) as zipf:
    zipf.write(args.library, 'lib/%s/%s' % (args.android_abi, library_file))

  # Align the dylib to a page boundary.
  zipalign_command = [
      args.zipalign_bin,
      '-p',  # Page align the dylib
      '-f',  # overwrite output if exists
      '4',  # 32-bit alignment
      unaligned_apk_path,
      unsigned_apk_path,
  ]
  run_command_checked(zipalign_command)

  # Sign the APK.
  apksigner_command = [
      args.apksigner_bin, 'sign', '--ks', args.keystore, '--ks-pass', 'pass:android', '--out',
      apk_path, unsigned_apk_path
  ]
  run_command_checked(apksigner_command)

  return 0


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