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

import argparse
import re
import sys
import zipfile

from util import build_utils  # pylint: disable=unused-import
from util import java_cpp_utils
import action_helpers  # build_utils adds //build to sys.path.
import zip_helpers


class StringParserDelegate(java_cpp_utils.CppConstantParser.Delegate):
    STRING_RE = re.compile(r'\s*(?:inline )?const(?:expr)? char k(.*)\[\]\s*=')
    VALUE_RE = re.compile(r'\s*("(?:\"|[^"])*")\s*;')

    def ExtractConstantName(self, line):
        match = StringParserDelegate.STRING_RE.match(line)
        return match.group(1) if match else None

    def ExtractValue(self, line):
        match = StringParserDelegate.VALUE_RE.search(line)
        return match.group(1) if match else None

    def CreateJavaConstant(self, name, value, comments):
        return java_cpp_utils.JavaString(name, value, comments)


def _GenerateOutput(template, source_paths, strings):
    description_template = """
    // This following string constants were inserted by
    //     {SCRIPT_NAME}
    // From
    //     {SOURCE_PATHS}

"""
    values = {
        'SCRIPT_NAME': java_cpp_utils.GetScriptName(),
        'SOURCE_PATHS': ',\n    //     '.join(source_paths),
    }
    description = description_template.format(**values)
    native_strings = '\n\n'.join(x.Format() for x in strings)

    # TODO(agrieve): Remove {{ and }} from input templates.
    template = template.replace('{{', '{').replace('}}', '}')
    return template.replace('{NATIVE_STRINGS}', description + native_strings)


def _ParseStringFile(path):
    with open(path, encoding='utf-8') as f:
        lines = f.readlines()
        lines = java_cpp_utils.ProcessListMacros(lines)
        string_file_parser = java_cpp_utils.CppConstantParser(
            StringParserDelegate(), lines
        )
    return string_file_parser.Parse()


def _Generate(source_paths, template):
    package, class_name = java_cpp_utils.ParseTemplateFile(template)
    output_path = java_cpp_utils.GetJavaFilePath(package, class_name)
    strings = []
    for source_path in source_paths:
        strings.extend(_ParseStringFile(source_path))

    output = _GenerateOutput(template, source_paths, strings)
    return output, output_path


def _Main(argv):
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--srcjar',
        required=True,
        help='The path at which to generate the .srcjar file',
    )
    parser.add_argument(
        '--template',
        help='The template file with which to generate the Java '
        'class. Must have "{NATIVE_STRINGS}" somewhere in '
        'the template.',
    )
    parser.add_argument('--class-name', help='FQN of Java class to generate')
    parser.add_argument(
        'inputs', nargs='+', help='Input file(s)', metavar='INPUTFILE'
    )
    args = parser.parse_args(argv)

    if args.template:
        with open(args.template, encoding='utf-8') as f:
            template = f.read()
    else:
        template = java_cpp_utils.CreateDefaultTemplate(
            args.class_name, '{NATIVE_STRINGS}'
        )

    with action_helpers.atomic_output(args.srcjar) as f:
        with zipfile.ZipFile(f, 'w', zipfile.ZIP_STORED) as srcjar:
            data, path = _Generate(args.inputs, template)
            zip_helpers.add_to_zip_hermetic(srcjar, path, data=data)


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