#!/usr/bin/env python3

# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Generates AndroidManifest.xml for a -Stub.apk."""

import argparse
import pathlib
import re
import sys
from xml.etree import ElementTree

from util import build_utils
from util import manifest_utils

_MAIN_TEMPLATE = """\
<?xml version="1.0" encoding="utf-8"?>
<manifest xmlns:android="http://schemas.android.com/apk/res/android"
    package="will.be.replaced">
  {}<application android:label="APK Stub">{}</application>
</manifest>
"""

_STATIC_LIBRARY_TEMPLATE = """
    <static-library android:name="{}" android:version="{}" />
"""


def main():
  parser = argparse.ArgumentParser()
  parser.add_argument('--static-library-name')
  parser.add_argument('--static-library-version')
  parser.add_argument('--output', required=True)
  parser.add_argument('--use-permissions-from')
  args = parser.parse_args(build_utils.ExpandFileArgs(sys.argv[1:]))

  if args.use_permissions_from:
    manifest_node = manifest_utils.ParseManifest(args.use_permissions_from)[1]
    permission_nodes = manifest_node.findall('uses-permission')
    permission_nodes += manifest_node.findall('uses-permission-sdk-23')
    sb = []
    for node in permission_nodes:
      sb.append(ElementTree.tostring(node).decode('utf8').rstrip() + '\n  ')
    sb.sort()
    permissions_part = ''.join(sb)
    permissions_part = re.sub(r'tools:ignore=".*?"\s*', '', permissions_part)
    permissions_part = re.sub(r'xmlns:tools=".*?"\s*', '', permissions_part)
  else:
    permissions_part = ''

  static_library_part = ''
  if args.static_library_name:
    static_library_part = _STATIC_LIBRARY_TEMPLATE.format(
        args.static_library_name, args.static_library_version)

  data = _MAIN_TEMPLATE.format(permissions_part, static_library_part)
  pathlib.Path(args.output).write_text(data, encoding='utf8')


if __name__ == '__main__':
  main()