#!/usr/bin/env python3
#
# Copyright 2013 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 os
import posixpath
import re
import sys
import zipfile

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

_CHROMIUM_SRC = os.path.join(os.path.dirname(__file__), os.pardir, os.pardir,
                             os.pardir)
_LLVM_CLANG_PATH = os.path.join(_CHROMIUM_SRC, 'third_party', 'llvm-build',
                                'Release+Asserts', 'bin', 'clang')

def _ParsePackageName(data):
  m = re.search(r'^\s*package\s+(.*?)\s*;', data, re.MULTILINE)
  return m.group(1) if m else ''


def ProcessJavaFile(template, defines, include_dirs):
  clang_cmd = [
      _LLVM_CLANG_PATH,
      '-E',  # stop after preprocessing.
      '-CC',  # Keep comments
      '-DANDROID',  # Specify ANDROID define for pre-processor.
      '-x',
      'c-header',  # treat sources as C header files
      '-P',  # disable line markers, i.e. '#line 309'
      '-Wno-gcc-install-dir-libstdcxx',  # don't worry about libstdc++ includes
  ]
  clang_cmd.extend('-D' + x for x in defines)
  clang_cmd.extend('-I' + x for x in include_dirs)
  data = build_utils.CheckOutput(clang_cmd + [template])
  package_name = _ParsePackageName(data)
  if not package_name:
    raise Exception('Could not find java package of ' + template)
  return package_name, data


def main(args):
  args = build_utils.ExpandFileArgs(args)

  parser = argparse.ArgumentParser()
  parser.add_argument('--include-dirs', help='GN list of include directories.')
  parser.add_argument('--output', help='Path for .srcjar.')
  parser.add_argument('--define',
                      action='append',
                      dest='defines',
                      help='List of -D args')
  parser.add_argument('templates', nargs='+', help='Template files.')
  options = parser.parse_args(args)

  options.defines = action_helpers.parse_gn_list(options.defines)
  options.include_dirs = action_helpers.parse_gn_list(options.include_dirs)
  with action_helpers.atomic_output(options.output) as f:
    with zipfile.ZipFile(f, 'w') as z:
      for template in options.templates:
        package_name, data = ProcessJavaFile(template, options.defines,
                                             options.include_dirs)
        zip_path = posixpath.join(
            package_name.replace('.', '/'),
            os.path.splitext(os.path.basename(template))[0]) + '.java'
        zip_helpers.add_to_zip_hermetic(z, zip_path, data=data)


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