Adding JNIAdditionalImport annotation.

base/android/jni_generator/jni_generator.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 """Extracts native methods from a Java file and generates the JNI bindings.
 If you change this, please run and update the tests."""
 
 import collections
 import errno
@@ skipping 145 matching lines @@
   _remappings = []
   _implicit_imports = []
 
   @staticmethod
   def SetFullyQualifiedClass(fully_qualified_class):
     JniParams._fully_qualified_class = 'L' + fully_qualified_class
     JniParams._package = '/'.join(fully_qualified_class.split('/')[:-1])
 
   @staticmethod
   def ExtractImportsAndInnerClasses(contents):
+    if not JniParams._package:
+      raise RuntimeError('SetFullyQualifiedClass must be called before '
+                         'ExtractImportsAndInnerClasses')
     contents = contents.replace('\n', '')
     re_import = re.compile(r'import.*?(?P<class>\S*?);')
     for match in re.finditer(re_import, contents):
       JniParams._imports += ['L' + match.group('class').replace('.', '/')]
 
     re_inner = re.compile(r'(class|interface)\s+?(?P<name>\w+?)\W')
     for match in re.finditer(re_inner, contents):
       inner = match.group('name')
       if not JniParams._fully_qualified_class.endswith(inner):
         JniParams._inner_classes += [JniParams._fully_qualified_class + '$' +
                                      inner]
 
+    re_additional_imports = re.compile(
+        r'@JNIAdditionalImport\((?P<class_name>\w+?)\.class\)')
+    for match in re.finditer(re_additional_imports, contents):
+      class_name = match.group('class_name')
+      if '.' in class_name:
+        raise SyntaxError('*.class cannot be used in @JNIAdditionalImport. '
+                          'Only import unqualified outer classes.' % class_name)
+      new_import = 'L%s/%s' % (JniParams._package, class_name)
+      if new_import in JniParams._imports:
+        raise SyntaxError('Do not use JNIAdditionalImport on an already '
+                          'imported class: %s' % (new_import.replace('/', '.')))
+      JniParams._imports += [new_import]
+
   @staticmethod
   def ParseJavaPSignature(signature_line):
     prefix = 'Signature: '
     return '"%s"' % signature_line[signature_line.index(prefix) + len(prefix):]
 
   @staticmethod
   def JavaToJni(param):
     """Converts a java param into a JNI signature type."""
     pod_param_map = {
         'int': 'I',
@@ skipping 1199 matching lines @@
     root_name = os.path.splitext(os.path.basename(input_file))[0]
     output_file = os.path.join(options.output_dir, root_name) + '_jni.h'
   if options.jarjar:
     with open(options.jarjar) as f:
       JniParams.SetJarJarMappings(f.read())
   GenerateJNIHeader(input_file, output_file, options)
 
 
 if __name__ == '__main__':
   sys.exit(main(sys.argv))