#!/usr/bin/env python3
#
# Copyright 2014 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Invokes Android's aidl"""

import argparse
import os
import re
import sys
import shutil
import zipfile

from util import build_utils
import action_helpers  # build_utils adds //build to sys.path.
import zip_helpers


def _GetAidlCmd(options):
    cmd = [options.aidl_path]
    cmd += ['-p' + s for s in options.imports]
    cmd += ['-I' + s for s in options.includes]
    cmd += ['--omit_invocation']
    return cmd


def do_native(options, files):

    for i, f in enumerate(files):
        with build_utils.TempDir() as temp_dir:
            aidl_cmd = _GetAidlCmd(options)
            aidl_cmd += ['--lang=ndk']
            aidl_cmd += ['-h', options.header_output_dir, '-o', temp_dir]
            aidl_cmd += [f]
            build_utils.CheckOutput(aidl_cmd)

            found_outputs = build_utils.FindInDirectory(temp_dir, '*.cpp')
            assert len(found_outputs) == 1, '\n'.join(found_outputs)
            shutil.move(found_outputs[0], options.cpp_output[i])


def main(argv):
    parser = argparse.ArgumentParser()
    parser.add_argument('--aidl-path', help='Path to the aidl binary.')
    parser.add_argument('--imports', help='Files to import.')
    parser.add_argument(
        '--header-output-dir', help='Optional header file output location.'
    )
    parser.add_argument(
        '--cpp-output',
        help='Optional cpp file output location.',
        action='append',
    )
    parser.add_argument(
        '--includes', help='Directories to add as import search paths.'
    )
    parser.add_argument('--srcjar', help='Path for srcjar output.')
    parser.add_argument('files', nargs='+')
    action_helpers.add_depfile_arg(parser)
    options = parser.parse_args(argv[1:])
    args = options.files

    options.imports = action_helpers.parse_gn_list(options.imports)
    options.includes = action_helpers.parse_gn_list(options.includes)

    if options.header_output_dir or options.cpp_output:
        if not (options.header_output_dir and options.cpp_output):
            parser.error(
                'Native generation requires header-output-dir and cpp-output'
            )

    with build_utils.TempDir() as temp_dir:
        for f in args:
            classname = os.path.splitext(os.path.basename(f))[0]
            output = os.path.join(temp_dir, classname + '.java')
            aidl_cmd = _GetAidlCmd(options)
            aidl_cmd += [f, output]
            build_utils.CheckOutput(aidl_cmd)

        with action_helpers.atomic_output(options.srcjar) as f:
            with zipfile.ZipFile(f, 'w') as srcjar:
                for path in build_utils.FindInDirectory(temp_dir, '*.java'):
                    with open(path, encoding='utf-8') as fileobj:
                        data = fileobj.read()
                    pkg_name = re.search(
                        r'^\s*package\s+(.*?)\s*;', data, re.M
                    ).group(1)
                    arcname = '%s/%s' % (
                        pkg_name.replace('.', '/'),
                        os.path.basename(path),
                    )
                    zip_helpers.add_to_zip_hermetic(srcjar, arcname, data=data)

    if options.header_output_dir:
        do_native(options, args)

    if options.depfile:
        include_files = []
        for include_dir in options.includes:
            include_files += build_utils.FindInDirectory(include_dir, '*.aidl')
        action_helpers.write_depfile(
            options.depfile, options.srcjar, include_files
        )


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