#!/usr/bin/env python3
# Copyright 2018 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Generate linker version scripts for Chrome on Android shared libraries."""

import argparse
import os

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

_SCRIPT_HEADER = """\
# AUTO-GENERATED FILE.  DO NOT MODIFY.
#
# See: %s

{
  global:
""" % os.path.relpath(__file__, build_utils.DIR_SOURCE_ROOT)

_SCRIPT_FOOTER = """\
  local:
    *;
};
"""


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--output',
        required=True,
        help='Path to output linker version script file.',
    )
    parser.add_argument(
        '--jni-multiplexing',
        action='store_true',
        help='Export only the JNI methods generated by multiplexing',
    )
    parser.add_argument(
        '--export-fortesting-java-symbols',
        action='store_true',
        help='Export Java_*_ForTesting JNI methods',
    )
    options = parser.parse_args()

    # JNI_OnLoad is always exported.
    # CrashpadHandlerMain() is the entry point to the Crashpad handler, required
    # for libcrashpad_handler_trampoline.so.
    # NativeChildProcessService_onCreate export is for native-only services,
    # which have a symbol exposed in our AndroidManifest which Android calls.
    symbol_list = [
        'CrashpadHandlerMain',
        'JNI_OnLoad',
        'NativeChildProcessService_onCreate',
    ]

    if options.export_fortesting_java_symbols or options.jni_multiplexing:
        symbol_list.append('Java_*')
    else:
        # The linker uses unix shell globbing patterns, not regex. So, we have to
        # include everything that doesn't end in "ForTest(ing)" with this set of
        # globs.
        symbol_list.append('Java_*[^F]orTesting')
        symbol_list.append('Java_*[^o]rTesting')
        symbol_list.append('Java_*[^r]Testing')
        symbol_list.append('Java_*[^T]esting')
        symbol_list.append('Java_*[^e]sting')
        symbol_list.append('Java_*[^s]ting')
        symbol_list.append('Java_*[^t]ing')
        symbol_list.append('Java_*[^i]ng')
        symbol_list.append('Java_*[^n]g')
        symbol_list.append('Java_*[^F]orTest')
        symbol_list.append('Java_*[^o]rTest')
        symbol_list.append('Java_*[^r]Test')
        symbol_list.append('Java_*[^T]est')
        symbol_list.append('Java_*[^e]st')
        symbol_list.append('Java_*[^s]t')
        symbol_list.append('Java_*[^gt]')

    script_content = [_SCRIPT_HEADER]
    for symbol in symbol_list:
        script_content.append('    %s;\n' % symbol)
    script_content.append(_SCRIPT_FOOTER)

    script = ''.join(script_content)

    with action_helpers.atomic_output(options.output, mode='w') as f:
        f.write(script)


if __name__ == '__main__':
    main()