#!/usr/bin/env python3
#
# Copyright 2020 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 FeatureParserDelegate(java_cpp_utils.CppConstantParser.Delegate):
  # Ex. 'BASE_FEATURE(kConstantName, "StringNameOfTheFeature", ...);'
  # or 'BASE_FEATURE(ConstantName, ...)'
  # would parse as:
  #   ExtractConstantName() -> 'ConstantName'
  #   ExtractValue() -> '"StringNameOfTheFeature"' or '"ConstantName"'
  # For 3-arg macro: BASE_FEATURE(kMyFeature, "MyFeature", ...)
  _FEATURE_RE_3_ARGS = re.compile(r'BASE_FEATURE\(k([^,]+),')
  # For 2-arg macro: BASE_FEATURE(MyFeature, ...)
  _FEATURE_RE_2_ARGS = re.compile(r'BASE_FEATURE\(([^,]+),')
  _VALUE_RE = re.compile(r'\s*("(?:\\"|[^"])*")\s*,')

  def ExtractConstantName(self, line):
    match = self._FEATURE_RE_3_ARGS.match(line)
    if match:
      return match.group(1)
    match = self._FEATURE_RE_2_ARGS.match(line)
    # A 2-arg macro does not have "k" prefix and does not have a string literal
    # as the second argument.
    if match and not match.group(1).startswith('k') and not self._VALUE_RE.search(line, pos=match.end()):
      return match.group(1)
    return None

  def ExtractValue(self, line):
    match = self._VALUE_RE.search(line)
    if match:
      return match.group(1)
    # If there is no string literal, it must be a 2-arg macro.
    match = self._FEATURE_RE_2_ARGS.match(line)
    if match and not match.group(1).startswith('k'):
      return f'"{match.group(1)}"'
    return None

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


def _GenerateOutput(template, source_paths, features):
  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_features = '\n\n'.join(x.Format() for x in features)

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


def _ParseFeatureFile(path):
  with open(path, encoding='utf-8') as f:
    feature_file_parser = java_cpp_utils.CppConstantParser(
        FeatureParserDelegate(), f.readlines())
  return feature_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)

  features = []
  for source_path in source_paths:
    features.extend(_ParseFeatureFile(source_path))

  output = _GenerateOutput(template, source_paths, features)
  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_FEATURES}" 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_FEATURES}')

  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:])