rt/vm/src/main/java/org/apidesign/vm4brwsr/ExportedSymbols.java
author Jaroslav Tulach <jaroslav.tulach@apidesign.org>
Mon, 26 May 2014 10:18:57 +0200
branchclosure
changeset 1592 f0433f790aaa
parent 1586 d4ee65642d8d
child 1787 ea12a3bb4b33
permissions -rw-r--r--
Export all java.** packages
     1 /**
     2  * Back 2 Browser Bytecode Translator
     3  * Copyright (C) 2012 Jaroslav Tulach <jaroslav.tulach@apidesign.org>
     4  *
     5  * This program is free software: you can redistribute it and/or modify
     6  * it under the terms of the GNU General Public License as published by
     7  * the Free Software Foundation, version 2 of the License.
     8  *
     9  * This program is distributed in the hope that it will be useful,
    10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    12  * GNU General Public License for more details.
    13  *
    14  * You should have received a copy of the GNU General Public License
    15  * along with this program. Look for COPYING file in the top folder.
    16  * If not, see http://opensource.org/licenses/GPL-2.0.
    17  */
    18 package org.apidesign.vm4brwsr;
    19 
    20 import java.io.IOException;
    21 import java.io.InputStream;
    22 import java.util.HashMap;
    23 import java.util.Map;
    24 import org.apidesign.bck2brwsr.core.ExtraJavaScript;
    25 import org.apidesign.vm4brwsr.ByteCodeParser.AnnotationParser;
    26 import org.apidesign.vm4brwsr.ByteCodeParser.ClassData;
    27 import org.apidesign.vm4brwsr.ByteCodeParser.FieldData;
    28 import org.apidesign.vm4brwsr.ByteCodeParser.MethodData;
    29 
    30 @ExtraJavaScript(processByteCode = false, resource="")
    31 final class ExportedSymbols {
    32     private final Bck2Brwsr.Resources resources;
    33     private final StringArray exported;
    34     private final Map<Object, Boolean> isMarkedAsExportedCache;
    35 
    36     ExportedSymbols(final Bck2Brwsr.Resources resources, StringArray explicitlyExported) {
    37         this.resources = resources;
    38         this.exported = explicitlyExported;
    39 
    40         isMarkedAsExportedCache = new HashMap<Object, Boolean>();
    41     }
    42 
    43     boolean isExported(ClassData classData) throws IOException {
    44         if (exported.contains(classData.getClassName())) {
    45             return true;
    46         }
    47         return classData.isPublic() && isMarkedAsExportedPackage(
    48                                            classData.getPkgName())
    49                    || isMarkedAsExported(classData);
    50     }
    51 
    52     boolean isExported(MethodData methodData) throws IOException {
    53         return isAccessible(methodData.access) && isExported(methodData.cls)
    54                    || isMarkedAsExported(methodData);
    55     }
    56 
    57     boolean isExported(FieldData fieldData) throws IOException {
    58         if (
    59             isAccessible(fieldData.access) && 
    60             isExported(fieldData.cls) || isMarkedAsExported(fieldData)
    61         ) {
    62             return true;
    63         }
    64         if (
    65             fieldData.isStatic() && fieldData.getName().equals("$VALUES") &&
    66             "java/lang/Enum".equals(fieldData.cls.getSuperClassName())
    67         ) {
    68             // enum values need to be exported
    69             return true;
    70         }
    71         return false;
    72     }
    73 
    74     private boolean isMarkedAsExportedPackage(String pkgName) {
    75         if (pkgName == null) {
    76             return false;
    77         }
    78         if (pkgName.startsWith("java/")) {
    79             return true;
    80         }
    81 
    82         final Boolean cachedValue = isMarkedAsExportedCache.get(pkgName);
    83         if (cachedValue != null) {
    84             return cachedValue;
    85         }
    86 
    87         final boolean newValue = resolveIsMarkedAsExportedPackage(pkgName);
    88         isMarkedAsExportedCache.put(pkgName, newValue);
    89 
    90         return newValue;
    91     }
    92 
    93     private boolean isMarkedAsExported(ClassData classData)
    94             throws IOException {
    95         final Boolean cachedValue = isMarkedAsExportedCache.get(classData);
    96         if (cachedValue != null) {
    97             return cachedValue;
    98         }
    99 
   100         final boolean newValue =
   101                 isMarkedAsExported(classData.findAnnotationData(true),
   102                                    classData);
   103         isMarkedAsExportedCache.put(classData, newValue);
   104 
   105         return newValue;
   106     }
   107 
   108     private boolean isMarkedAsExported(MethodData methodData)
   109             throws IOException {
   110         return isMarkedAsExported(methodData.findAnnotationData(true),
   111                                   methodData.cls);
   112     }
   113 
   114     private boolean isMarkedAsExported(FieldData fieldData)
   115             throws IOException {
   116         return isMarkedAsExported(fieldData.findAnnotationData(true),
   117                                   fieldData.cls);
   118     }
   119 
   120     private boolean resolveIsMarkedAsExportedPackage(String pkgName) {
   121         if (exported.contains(pkgName + '/')) {
   122             return true;
   123         }
   124         try {
   125             final InputStream is =
   126                     resources.get(pkgName + "/package-info.class");
   127             if (is == null) {
   128                 return false;
   129             }
   130 
   131             try {
   132                 final ClassData pkgInfoClass = new ClassData(is);
   133                 return isMarkedAsExported(
   134                                pkgInfoClass.findAnnotationData(true),
   135                                pkgInfoClass);
   136             } finally {
   137                 is.close();
   138             }
   139         } catch (final IOException e) {
   140             return false;
   141         }
   142     }
   143 
   144     static boolean isMarkedAsExported(byte[] arrData, ClassData cd)
   145             throws IOException {
   146         if (arrData == null) {
   147             return false;
   148         }
   149 
   150         final boolean[] found = { false };
   151         final AnnotationParser annotationParser =
   152                 new AnnotationParser(false, false) {
   153                     @Override
   154                     protected void visitAnnotationStart(
   155                             String type,
   156                             boolean top) {
   157                         if (top && type.equals("Lorg/apidesign/bck2brwsr"
   158                                                    + "/core/Exported;")) {
   159                             found[0] = true;
   160                         }
   161                     }
   162                 };
   163         annotationParser.parse(arrData, cd);
   164         return found[0];
   165     }
   166 
   167     private static boolean isAccessible(int access) {
   168         return (access & (ByteCodeParser.ACC_PUBLIC
   169                               | ByteCodeParser.ACC_PROTECTED)) != 0;
   170     }
   171 }