#!/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.
"""Creates a binary profile from an HRF + dex + mapping."""

import argparse
import sys

from util import build_utils
import action_helpers


def main(args):
    parser = argparse.ArgumentParser(description=__doc__)
    action_helpers.add_depfile_arg(parser)
    parser.add_argument(
        '--output-profile', required=True, help='Path to output binary profile.'
    )
    parser.add_argument(
        '--output-metadata',
        required=True,
        help='Path to output binary profile metadata.',
    )
    parser.add_argument(
        '--profgen', required=True, help='Path to profgen binary.'
    )
    parser.add_argument(
        '--dex',
        required=True,
        help='Path to a zip containing release dex files.',
    )
    parser.add_argument(
        '--proguard-mapping', help='Path to proguard mapping for release dex.'
    )
    parser.add_argument(
        '--input-profile-path',
        required=True,
        help='Path to HRF baseline profile to apply.',
    )
    options = parser.parse_args(build_utils.ExpandFileArgs(args))

    cmd = [
        options.profgen,
        'bin',
        options.input_profile_path,
        '-o',
        options.output_profile,
        '-om',
        options.output_metadata,
        '-a',
        options.dex,
    ]

    if options.proguard_mapping:
        cmd += [
            '-m',
            options.proguard_mapping,
        ]

    build_utils.CheckOutput(cmd, env={'JAVA_HOME': build_utils.JAVA_HOME})
    action_helpers.write_depfile(
        options.depfile, options.output_profile, inputs=[options.dex]
    )


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