# Copyright 2014 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import os
import re
import tempfile

from pylib import constants
from pylib import cmd_helper


_PROGUARD_CLASS_RE = re.compile(r'\s*?- Program class:\s*([\S]+)$')
_PROGUARD_SUPERCLASS_RE = re.compile(r'\s*?  Superclass:\s*([\S]+)$')
_PROGUARD_SECTION_RE = re.compile(
    r'^(?:Interfaces|Constant Pool|Fields|Methods|Class file attributes) '
    r'\(count = \d+\):$')
_PROGUARD_METHOD_RE = re.compile(r'\s*?- Method:\s*(\S*)[(].*$')
_PROGUARD_ANNOTATION_RE = re.compile(r'\s*?- Annotation \[L(\S*);\]:$')
_PROGUARD_ANNOTATION_CONST_RE = (
    re.compile(r'\s*?- Constant element value.*$'))
_PROGUARD_ANNOTATION_VALUE_RE = re.compile(r'\s*?- \S+? \[(.*)\]$')

_PROGUARD_PATH_SDK = os.path.join(
    constants.ANDROID_SDK_ROOT, 'tools', 'proguard', 'lib', 'proguard.jar')
_PROGUARD_PATH_BUILT = (
    os.path.join(os.environ['ANDROID_BUILD_TOP'], 'external', 'proguard',
                 'lib', 'proguard.jar')
    if 'ANDROID_BUILD_TOP' in os.environ else None)
_PROGUARD_PATH = (
    _PROGUARD_PATH_SDK if os.path.exists(_PROGUARD_PATH_SDK)
    else _PROGUARD_PATH_BUILT)


def Dump(jar_path):
  """Dumps class and method information from a JAR into a dict via proguard.

  Args:
    jar_path: An absolute path to the JAR file to dump.
  Returns:
    A dict in the following format:
      {
        'classes': [
          {
            'class': '',
            'superclass': '',
            'annotations': {},
            'methods': [
              {
                'method': '',
                'annotations': {},
              },
              ...
            ],
          },
          ...
        ],
      }
  """

  with tempfile.NamedTemporaryFile() as proguard_output:
    cmd_helper.RunCmd(['java', '-jar',
                       _PROGUARD_PATH,
                       '-injars', jar_path,
                       '-dontshrink',
                       '-dontoptimize',
                       '-dontobfuscate',
                       '-dontpreverify',
                       '-dump', proguard_output.name])


    results = {
      'classes': [],
    }

    annotation = None
    annotation_has_value = False
    class_result = None
    method_result = None

    for line in proguard_output:
      line = line.strip('\r\n')

      m = _PROGUARD_CLASS_RE.match(line)
      if m:
        class_result = {
          'class': m.group(1).replace('/', '.'),
          'superclass': '',
          'annotations': {},
          'methods': [],
        }
        results['classes'].append(class_result)
        annotation = None
        annotation_has_value = False
        method_result = None
        continue

      if not class_result:
        continue

      m = _PROGUARD_SUPERCLASS_RE.match(line)
      if m:
        class_result['superclass'] = m.group(1).replace('/', '.')
        continue

      m = _PROGUARD_SECTION_RE.match(line)
      if m:
        annotation = None
        annotation_has_value = False
        method_result = None
        continue

      m = _PROGUARD_METHOD_RE.match(line)
      if m:
        method_result = {
          'method': m.group(1),
          'annotations': {},
        }
        class_result['methods'].append(method_result)
        annotation = None
        annotation_has_value = False
        continue

      m = _PROGUARD_ANNOTATION_RE.match(line)
      if m:
        # Ignore the annotation package.
        annotation = m.group(1).split('/')[-1]
        if method_result:
          method_result['annotations'][annotation] = None
        else:
          class_result['annotations'][annotation] = None
        continue

      if annotation:
        if not annotation_has_value:
          m = _PROGUARD_ANNOTATION_CONST_RE.match(line)
          annotation_has_value = bool(m)
        else:
          m = _PROGUARD_ANNOTATION_VALUE_RE.match(line)
          if m:
            if method_result:
              method_result['annotations'][annotation] = m.group(1)
            else:
              class_result['annotations'][annotation] = m.group(1)
          annotation_has_value = None

  return results