/src/main/java/org/rythmengine/internal/compiler/TemplateCompiler.java
Java | 346 lines | 244 code | 40 blank | 62 comment | 51 complexity | 620b431fcaf962709712521b3029ba50 MD5 | raw file
- /**
- * Copyright (C) 2013-2016 The Rythm Engine project
- * for LICENSE and other details see:
- * https://github.com/rythmengine/rythmengine
- */
- package org.rythmengine.internal.compiler;
- /*-
- * #%L
- * Rythm Template Engine
- * %%
- * Copyright (C) 2017 - 2021 OSGL (Open Source General Library)
- * %%
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- * #L%
- */
- import org.rythmengine.RythmEngine;
- import org.rythmengine.exception.CompileException;
- import org.rythmengine.logger.ILogger;
- import org.rythmengine.logger.Logger;
- import org.eclipse.jdt.core.compiler.IProblem;
- import org.eclipse.jdt.internal.compiler.*;
- import org.eclipse.jdt.internal.compiler.Compiler;
- import org.eclipse.jdt.internal.compiler.ast.CompilationUnitDeclaration;
- import org.eclipse.jdt.internal.compiler.classfmt.ClassFileReader;
- import org.eclipse.jdt.internal.compiler.classfmt.ClassFormatException;
- import org.eclipse.jdt.internal.compiler.env.ICompilationUnit;
- import org.eclipse.jdt.internal.compiler.env.INameEnvironment;
- import org.eclipse.jdt.internal.compiler.env.NameEnvironmentAnswer;
- import org.eclipse.jdt.internal.compiler.impl.CompilerOptions;
- import org.eclipse.jdt.internal.compiler.problem.DefaultProblemFactory;
- import java.util.*;
- /**
- * Created by IntelliJ IDEA.
- * User: luog
- * Date: 18/01/12
- * Time: 7:10 PM
- * To change this template use File | Settings | File Templates.
- */
- public class TemplateCompiler {
- private final static ILogger logger = Logger.get(TemplateCompiler.class);
- private RythmEngine engine() {
- return classCache.engine;
- }
- Map<String, Boolean> packagesCache = new HashMap<String, Boolean>();
- // -- util methods
- private String getTemplateByClassName(String className) {
- TemplateClass tc = engine().classes().getByClassName(className);
- return null == tc ? null : tc.getKey().toString();
- }
- // -- the following code comes from PlayFramework 1.2
- TemplateClassManager classCache;
- Map<String, String> settings;
- /**
- * Try to guess the magic configuration options
- */
- public TemplateCompiler(TemplateClassManager classCache) {
- this.classCache = classCache;
- this.settings = new HashMap<String, String>();
- this.settings.put(CompilerOptions.OPTION_ReportMissingSerialVersion, CompilerOptions.IGNORE);
- this.settings.put(CompilerOptions.OPTION_ReportDeprecation, CompilerOptions.IGNORE);
- this.settings.put(CompilerOptions.OPTION_ReportUnusedImport, CompilerOptions.IGNORE);
- this.settings.put(CompilerOptions.OPTION_Encoding, "UTF-8");
- this.settings.put(CompilerOptions.OPTION_LineNumberAttribute, CompilerOptions.GENERATE);
- this.settings.put(CompilerOptions.OPTION_SourceFileAttribute, CompilerOptions.GENERATE);
- this.settings.put(CompilerOptions.OPTION_LocalVariableAttribute, CompilerOptions.GENERATE);
- String javaVersion = CompilerOptions.VERSION_1_6;
- String provisionedJavaVersion = System.getProperty("java.version");
- if (provisionedJavaVersion.startsWith("1.7")) {
- javaVersion = CompilerOptions.VERSION_1_7;
- } else if (provisionedJavaVersion.startsWith("1.8")) {
- javaVersion = CompilerOptions.VERSION_1_8;
- }
- this.settings.put(CompilerOptions.OPTION_Source, javaVersion);
- this.settings.put(CompilerOptions.OPTION_TargetPlatform, javaVersion);
- this.settings.put(CompilerOptions.OPTION_PreserveUnusedLocal, CompilerOptions.PRESERVE);
- this.settings.put(CompilerOptions.OPTION_Compliance, javaVersion);
- }
- /**
- * Something to compile
- */
- final class CompilationUnit implements ICompilationUnit {
- final private String clazzName;
- final private String fileName;
- final private char[] typeName;
- final private char[][] packageName;
- CompilationUnit(String pClazzName) {
- clazzName = pClazzName;
- if (pClazzName.contains("$")) {
- pClazzName = pClazzName.substring(0, pClazzName.indexOf("$"));
- }
- fileName = pClazzName.replace('.', '/') + ".java";
- int dot = pClazzName.lastIndexOf('.');
- if (dot > 0) {
- typeName = pClazzName.substring(dot + 1).toCharArray();
- } else {
- typeName = pClazzName.toCharArray();
- }
- StringTokenizer st = new StringTokenizer(pClazzName, ".");
- packageName = new char[st.countTokens() - 1][];
- for (int i = 0; i < packageName.length; i++) {
- packageName[i] = st.nextToken().toCharArray();
- }
- }
- @Override
- public boolean ignoreOptionalProblems() {
- return false;
- }
- public char[] getFileName() {
- return fileName.toCharArray();
- }
- public char[] getContents() {
- if (null == classCache) {
- throw new NullPointerException("classCache is null");
- }
- TemplateClass tc = classCache.getByClassName(clazzName);
- if (null == tc) {
- throw new NullPointerException("Error get java source content for " + clazzName + ": template class is null");
- }
- if (null == tc.javaSource) {
- throw new NullPointerException("Error get java source content for " + clazzName + ": java source is null");
- }
- return tc.javaSource.toCharArray();
- //return classCache.getByClassName(clazzName).javaSource.toCharArray();
- }
- public char[] getMainTypeName() {
- return typeName;
- }
- public char[][] getPackageName() {
- return packageName;
- }
- }
- final Set<String> notFoundTypes = new HashSet<String>();
- /**
- * Please compile this className
- */
- @SuppressWarnings("deprecation")
- public void compile(String[] classNames) {
- ICompilationUnit[] compilationUnits = new CompilationUnit[classNames.length];
- for (int i = 0; i < classNames.length; i++) {
- compilationUnits[i] = new CompilationUnit(classNames[i]);
- }
- IErrorHandlingPolicy policy = DefaultErrorHandlingPolicies.exitOnFirstError();
- IProblemFactory problemFactory = new DefaultProblemFactory(Locale.ENGLISH);
- /**
- * To find types ...
- */
- INameEnvironment nameEnvironment = new INameEnvironment() {
- public NameEnvironmentAnswer findType(final char[][] compoundTypeName) {
- final StringBuffer result = new StringBuffer();
- for (int i = 0; i < compoundTypeName.length; i++) {
- if (i != 0) {
- result.append('.');
- }
- result.append(compoundTypeName[i]);
- }
- return findType(result.toString());
- }
- public NameEnvironmentAnswer findType(final char[] typeName, final char[][] packageName) {
- final StringBuffer result = new StringBuffer();
- for (int i = 0; i < packageName.length; i++) {
- result.append(packageName[i]);
- result.append('.');
- }
- result.append(typeName);
- return findType(result.toString());
- }
- private NameEnvironmentAnswer findStandType(final String name) throws ClassFormatException {
- if (notFoundTypes.contains(name)) {
- return null;
- }
- RythmEngine engine = engine();
- byte[] bytes = engine.classLoader().getClassDefinition(name);
- if (bytes != null) {
- ClassFileReader classFileReader = new ClassFileReader(bytes, name.toCharArray(), true);
- return new NameEnvironmentAnswer(classFileReader, null);
- }
- if (engine.isProdMode()) {
- notFoundTypes.add(name);
- } else if (name.matches("^(java\\.|play\\.|com\\.greenlaw110\\.).*")) {
- notFoundTypes.add(name);
- }
- return null;
- }
- private NameEnvironmentAnswer findType(final String name) {
- try {
- if (!name.contains(TemplateClass.CN_SUFFIX)) {
- return findStandType(name);
- }
- char[] fileName = name.toCharArray();
- TemplateClass templateClass = classCache.getByClassName(name);
- // TemplateClass exists
- if (templateClass != null) {
- if (templateClass.javaByteCode != null) {
- ClassFileReader classFileReader = new ClassFileReader(templateClass.javaByteCode, fileName, true);
- return new NameEnvironmentAnswer(classFileReader, null);
- }
- // Cascade compilation
- ICompilationUnit compilationUnit = new CompilationUnit(name);
- return new NameEnvironmentAnswer(compilationUnit, null);
- }
- // So it's a standard class
- return findStandType(name);
- } catch (ClassFormatException e) {
- // Something very very bad
- throw new RuntimeException(e);
- }
- }
- @Override
- public boolean isPackage(char[][] parentPackageName, char[] packageName) {
- // Rebuild something usable
- StringBuilder sb = new StringBuilder();
- if (parentPackageName != null) {
- for (char[] p : parentPackageName) {
- sb.append(new String(p));
- sb.append(".");
- }
- }
- sb.append(new String(packageName));
- String name = sb.toString();
- if (packagesCache.containsKey(name)) {
- return packagesCache.get(name).booleanValue();
- }
- // Check if thera a .java or .class for this resource
- if (engine().classLoader().getClassDefinition(name) != null) {
- packagesCache.put(name, false);
- return false;
- }
- if (engine().classes().getByClassName(name) != null) {
- packagesCache.put(name, false);
- return false;
- }
- packagesCache.put(name, true);
- return true;
- }
- public void cleanup() {
- }
- };
- /**
- * Compilation result
- */
- ICompilerRequestor compilerRequestor = new ICompilerRequestor() {
- public void acceptResult(CompilationResult result) {
- // If error
- if (result.hasErrors()) {
- for (IProblem problem : result.getErrors()) {
- int line = problem.getSourceLineNumber();
- String message = problem.getMessage();
- throw CompileException.compilerException(String.valueOf(result.compilationUnit.getMainTypeName()), line, message);
- }
- }
- // Something has been compiled
- ClassFile[] clazzFiles = result.getClassFiles();
- for (int i = 0; i < clazzFiles.length; i++) {
- final ClassFile clazzFile = clazzFiles[i];
- final char[][] compoundName = clazzFile.getCompoundName();
- final StringBuffer clazzName = new StringBuffer();
- for (int j = 0; j < compoundName.length; j++) {
- if (j != 0) {
- clazzName.append('.');
- }
- clazzName.append(compoundName[j]);
- }
- if (logger.isTraceEnabled()) {
- logger.trace("Compiled %s", getTemplateByClassName(clazzName.toString()));
- }
- String cn = clazzName.toString();
- TemplateClass tc = classCache.getByClassName(cn);
- if (null == tc) {
- int pos = cn.indexOf("$");
- if (-1 != pos) {
- // inner class
- TemplateClass root = classCache.getByClassName(cn.substring(0, pos));
- tc = TemplateClass.createInnerClass(cn, clazzFile.getBytes(), root);
- tc.delayedEnhance(root);
- classCache.add(tc);
- } else {
- throw new RuntimeException("Cannot find class by name: " + cn);
- }
- } else {
- tc.compiled(clazzFile.getBytes());
- }
- }
- }
- };
- /**
- * The JDT compiler
- */
- Compiler jdtCompiler = new Compiler(nameEnvironment, policy, settings, compilerRequestor, problemFactory) {
- @Override
- protected void handleInternalException(Throwable e, CompilationUnitDeclaration ud, CompilationResult result) {
- }
- };
- // Go !
- jdtCompiler.compile(compilationUnits);
- }
- }