Ptidej-v5.2-master/Ptidej Reporting Tests/src/net/sf/jasperreports/engine/design/JRJdtCompiler.java
/*
 * ============================================================================
 * GNU Lesser General Public License
 * ============================================================================
 *
 * JasperReports - Free Java report-generating library.
 * Copyright (C) 2001-2006 JasperSoft Corporation http://www.jaspersoft.com
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 * 
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307, USA.
 * 
 * JasperSoft Corporation
 * 303 Second Street, Suite 450 North
 * San Francisco, CA 94107
 * http://www.jaspersoft.com
 */
package net.sf.jasperreports.engine.design;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Properties;
import net.sf.jasperreports.engine.JRException;
import net.sf.jasperreports.engine.JRReport;
import net.sf.jasperreports.engine.JRRuntimeException;
import net.sf.jasperreports.engine.util.JRClassLoader;
import net.sf.jasperreports.engine.util.JRProperties;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.eclipse.jdt.core.compiler.IProblem;
import org.eclipse.jdt.internal.compiler.ClassFile;
import org.eclipse.jdt.internal.compiler.CompilationResult;
import org.eclipse.jdt.internal.compiler.Compiler;
import org.eclipse.jdt.internal.compiler.DefaultErrorHandlingPolicies;
import org.eclipse.jdt.internal.compiler.ICompilerRequestor;
import org.eclipse.jdt.internal.compiler.IErrorHandlingPolicy;
import org.eclipse.jdt.internal.compiler.IProblemFactory;
import org.eclipse.jdt.internal.compiler.classfmt.ClassFileReader;
import org.eclipse.jdt.internal.compiler.env.IBinaryType;
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;

/**
 * @author Teodor Danciu (teodord@users.sourceforge.net)
 * @version $Id: JRJdtCompiler.java,v 1.1 2008/09/29 16:20:08 guehene Exp $
 */
public class JRJdtCompiler extends JRAbstractJavaCompiler {
	private static final String JDT_PROPERTIES_PREFIX = "org.eclipse.jdt.core.";

	/**
	 *  
	 */
	static final Log log = LogFactory.getLog(JRJdtCompiler.class);

	private final ClassLoader classLoader;

	Constructor constrNameEnvAnsBin;
	Constructor constrNameEnvAnsCompUnit;

	boolean is2ArgsConstr;
	Constructor constrNameEnvAnsBin2Args;
	Constructor constrNameEnvAnsCompUnit2Args;

	public JRJdtCompiler() {
		super(false);

		this.classLoader = getClassLoader();

		boolean success;
		try //FIXME remove support for pre 3.1 jdt
		{
			Class classAccessRestriction =
				loadClass("org.eclipse.jdt.internal.compiler.env.AccessRestriction");
			this.constrNameEnvAnsBin2Args =
				NameEnvironmentAnswer.class.getConstructor(new Class[] {
						IBinaryType.class, classAccessRestriction });
			this.constrNameEnvAnsCompUnit2Args =
				NameEnvironmentAnswer.class.getConstructor(new Class[] {
						ICompilationUnit.class, classAccessRestriction });
			this.is2ArgsConstr = true;
			success = true;
		}
		catch (NoSuchMethodException e) {
			success = false;
		}
		catch (ClassNotFoundException ex) {
			success = false;
		}

		if (!success) {
			try {
				this.constrNameEnvAnsBin =
					NameEnvironmentAnswer.class
						.getConstructor(new Class[] { IBinaryType.class });
				this.constrNameEnvAnsCompUnit =
					NameEnvironmentAnswer.class
						.getConstructor(new Class[] { ICompilationUnit.class });
				this.is2ArgsConstr = false;
			}
			catch (NoSuchMethodException ex) {
				throw new JRRuntimeException("Not able to load JDT classes", ex);
			}
		}
	}

	class CompilationUnit implements ICompilationUnit {
		protected String srcCode;
		protected String className;

		public CompilationUnit(String srcCode, String className) {
			this.srcCode = srcCode;
			this.className = className;
		}

		public char[] getFileName() {
			return this.className.toCharArray();
		}

		public char[] getContents() {
			return this.srcCode.toCharArray();
		}

		public char[] getMainTypeName() {
			return this.className.toCharArray();
		}

		public char[][] getPackageName() {
			return new char[0][0];
		}

		public boolean ignoreOptionalProblems() {
			return false;
		}
	}

	/**
	 *
	 */
	protected String compileUnits(
		final JRCompilationUnit[] units,
		String classpath,
		File tempDirFile) {
		final StringBuffer problemBuffer = new StringBuffer();

		INameEnvironment env = getNameEnvironment(units);

		final IErrorHandlingPolicy policy =
			DefaultErrorHandlingPolicies.proceedWithAllProblems();

		final Map settings = getJdtSettings();

		final IProblemFactory problemFactory =
			new DefaultProblemFactory(Locale.getDefault());

		final ICompilerRequestor requestor =
			getCompilerRequestor(units, problemBuffer);

		ICompilationUnit[] compilationUnits =
			new ICompilationUnit[units.length];
		for (int i = 0; i < compilationUnits.length; i++) {
			compilationUnits[i] =
				new CompilationUnit(
					units[i].getSourceCode(),
					units[i].getName());
		}

		Compiler compiler =
			new Compiler(env, policy, settings, requestor, problemFactory);
		compiler.compile(compilationUnits);

		if (problemBuffer.length() > 0) {
			return problemBuffer.toString();
		}

		return null;
	}

	protected INameEnvironment getNameEnvironment(
		final JRCompilationUnit[] units) {
		final INameEnvironment env = new INameEnvironment() {
			public NameEnvironmentAnswer findType(char[][] compoundTypeName) {
				StringBuffer result = new StringBuffer();
				String sep = "";
				for (int i = 0; i < compoundTypeName.length; i++) {
					result.append(sep);
					result.append(compoundTypeName[i]);
					sep = ".";
				}
				return findType(result.toString());
			}

			public NameEnvironmentAnswer findType(
				char[] typeName,
				char[][] packageName) {
				StringBuffer result = new StringBuffer();
				String sep = "";
				for (int i = 0; i < packageName.length; i++) {
					result.append(sep);
					result.append(packageName[i]);
					sep = ".";
				}
				result.append(sep);
				result.append(typeName);
				return findType(result.toString());
			}

			private int getClassIndex(String className) {
				int classIdx;
				for (classIdx = 0; classIdx < units.length; ++classIdx) {
					if (className.equals(units[classIdx].getName())) {
						break;
					}
				}

				if (classIdx >= units.length) {
					classIdx = -1;
				}

				return classIdx;
			}

			private NameEnvironmentAnswer findType(String className) {
				try {
					int classIdx = getClassIndex(className);

					if (classIdx >= 0) {
						ICompilationUnit compilationUnit =
							new CompilationUnit(
								units[classIdx].getSourceCode(),
								className);
						if (JRJdtCompiler.this.is2ArgsConstr) {
							return (NameEnvironmentAnswer) JRJdtCompiler.this.constrNameEnvAnsCompUnit2Args
								.newInstance(new Object[] { compilationUnit,
										null });
						}

						return (NameEnvironmentAnswer) JRJdtCompiler.this.constrNameEnvAnsCompUnit
							.newInstance(new Object[] { compilationUnit });
					}

					String resourceName =
						className.replace('.', '/') + ".class";
					InputStream is = getResource(resourceName);
					if (is != null) {
						byte[] classBytes;
						byte[] buf = new byte[8192];
						ByteArrayOutputStream baos =
							new ByteArrayOutputStream(buf.length);
						int count;
						while ((count = is.read(buf, 0, buf.length)) > 0) {
							baos.write(buf, 0, count);
						}
						baos.flush();
						classBytes = baos.toByteArray();
						char[] fileName = className.toCharArray();
						ClassFileReader classFileReader =
							new ClassFileReader(classBytes, fileName, true);

						if (JRJdtCompiler.this.is2ArgsConstr) {
							return (NameEnvironmentAnswer) JRJdtCompiler.this.constrNameEnvAnsBin2Args
								.newInstance(new Object[] { classFileReader,
										null });
						}

						return (NameEnvironmentAnswer) JRJdtCompiler.this.constrNameEnvAnsBin
							.newInstance(new Object[] { classFileReader });
					}
				}
				catch (IOException exc) {
					log.error("Compilation error", exc);
				}
				catch (org.eclipse.jdt.internal.compiler.classfmt.ClassFormatException exc) {
					log.error("Compilation error", exc);
				}
				catch (InvocationTargetException e) {
					throw new JRRuntimeException(
						"Not able to create NameEnvironmentAnswer",
						e);
				}
				catch (IllegalArgumentException e) {
					throw new JRRuntimeException(
						"Not able to create NameEnvironmentAnswer",
						e);
				}
				catch (InstantiationException e) {
					throw new JRRuntimeException(
						"Not able to create NameEnvironmentAnswer",
						e);
				}
				catch (IllegalAccessException e) {
					throw new JRRuntimeException(
						"Not able to create NameEnvironmentAnswer",
						e);
				}
				return null;
			}

			private boolean isPackage(String result) {
				int classIdx = getClassIndex(result);
				if (classIdx >= 0) {
					return false;
				}

				String resourceName = result.replace('.', '/') + ".class";

				boolean isPackage = true;

				InputStream is = getResource(resourceName);

				if (is != null)// cannot just test for null; need to read from "is" to avoid bug 
				{ // with sun.plugin.cache.EmptyInputStream on JRE 1.5 plugin
					try // http://sourceforge.net/tracker/index.php?func=detail&aid=1478460&group_id=36382&atid=416703
					{
						isPackage = (is.read() > 0);
					}
					catch (IOException e) {
						//ignore
					}
					finally {
						try {
							is.close();
						}
						catch (IOException e) {
							//ignore
						}
					}
				}

				return isPackage;
			}

			public boolean isPackage(
				char[][] parentPackageName,
				char[] packageName) {
				StringBuffer result = new StringBuffer();
				String sep = "";
				if (parentPackageName != null) {
					for (int i = 0; i < parentPackageName.length; i++) {
						result.append(sep);
						result.append(parentPackageName[i]);
						sep = ".";
					}
				}
				if (Character.isUpperCase(packageName[0])) {
					if (!isPackage(result.toString())) {
						return false;
					}
				}
				result.append(sep);
				result.append(packageName);
				return isPackage(result.toString());
			}

			public void cleanup() {
			}

		};

		return env;
	}

	protected ICompilerRequestor getCompilerRequestor(
		final JRCompilationUnit[] units,
		final StringBuffer problemBuffer) {
		final ICompilerRequestor requestor = new ICompilerRequestor() {
			public void acceptResult(CompilationResult result) {
				String className =
					((CompilationUnit) result.getCompilationUnit()).className;

				int classIdx;
				for (classIdx = 0; classIdx < units.length; ++classIdx) {
					if (className.equals(units[classIdx].getName())) {
						break;
					}
				}

				if (result.hasErrors()) {
					String sourceCode = units[classIdx].getSourceCode();

					IProblem[] problems = getJavaCompilationErrors(result);
					for (int i = 0; i < problems.length; i++) {
						IProblem problem = problems[i];
						//if (problem.isError()) 
						{
							problemBuffer.append(i + 1);
							problemBuffer.append(". ");
							problemBuffer.append(problem.getMessage());

							if (problem.getSourceStart() >= 0
									&& problem.getSourceEnd() >= 0) {
								int problemStartIndex =
									sourceCode.lastIndexOf(
										"\n",
										problem.getSourceStart()) + 1;
								int problemEndIndex =
									sourceCode.indexOf(
										"\n",
										problem.getSourceEnd());
								if (problemEndIndex < 0) {
									problemEndIndex = sourceCode.length();
								}

								problemBuffer.append("\n");
								problemBuffer.append(sourceCode.substring(
									problemStartIndex,
									problemEndIndex));
								problemBuffer.append("\n");
								for (int j = problemStartIndex; j < problem
									.getSourceStart(); j++) {
									problemBuffer.append(" ");
								}
								if (problem.getSourceStart() == problem
									.getSourceEnd()) {
									problemBuffer.append("^");
								}
								else {
									problemBuffer.append("<");
									for (int j = problem.getSourceStart() + 1; j < problem
										.getSourceEnd(); j++) {
										problemBuffer.append("-");
									}
									problemBuffer.append(">");
								}
							}

							problemBuffer.append("\n");
						}
					}
					problemBuffer.append(problems.length);
					problemBuffer.append(" errors\n");
				}
				if (problemBuffer.length() == 0) {
					ClassFile[] resultClassFiles = result.getClassFiles();
					for (int i = 0; i < resultClassFiles.length; i++) {
						units[classIdx].setCompileData(resultClassFiles[i]
							.getBytes());
					}
				}
			}
		};

		return requestor;
	}

	protected Map getJdtSettings() {
		final Map settings = new HashMap();
		settings.put(
			CompilerOptions.OPTION_LineNumberAttribute,
			CompilerOptions.GENERATE);
		settings.put(
			CompilerOptions.OPTION_SourceFileAttribute,
			CompilerOptions.GENERATE);
		settings.put(
			CompilerOptions.OPTION_ReportDeprecation,
			CompilerOptions.IGNORE);
		//		if (ctxt.getOptions().getJavaEncoding() != null) 
		//		{
		//			settings.put(CompilerOptions.OPTION_Encoding, ctxt.getOptions().getJavaEncoding());
		//		}
		//		if (ctxt.getOptions().getClassDebugInfo()) 
		//		{
		//			settings.put(CompilerOptions.OPTION_LocalVariableAttribute, CompilerOptions.GENERATE);
		//		}

		List properties = JRProperties.getProperties(JDT_PROPERTIES_PREFIX);
		for (Iterator it = properties.iterator(); it.hasNext();) {
			JRProperties.PropertySuffix property =
				(JRProperties.PropertySuffix) it.next();
			String propVal = property.getValue();
			if (propVal != null && propVal.length() > 0) {
				settings.put(property.getKey(), propVal);
			}
		}

		Properties systemProps = System.getProperties();
		for (Enumeration it = systemProps.propertyNames(); it.hasMoreElements();) {
			String propName = (String) it.nextElement();
			if (propName.startsWith(JDT_PROPERTIES_PREFIX)) {
				String propVal = systemProps.getProperty(propName);
				if (propVal != null && propVal.length() > 0) {
					settings.put(propName, propVal);
				}
			}
		}

		return settings;
	}

	/**
	 *
	 */
	private ClassLoader getClassLoader() {
		ClassLoader clsLoader = Thread.currentThread().getContextClassLoader();

		if (clsLoader != null) {
			try {
				Class.forName(JRJdtCompiler.class.getName(), true, clsLoader);
			}
			catch (ClassNotFoundException e) {
				clsLoader = null;
				//if (log.isWarnEnabled())
				//	log.warn("Failure using Thread.currentThread().getContextClassLoader() in JRJdtCompiler class. Using JRJdtCompiler.class.getClassLoader() instead.");
			}
		}

		if (clsLoader == null) {
			clsLoader = JRClassLoader.class.getClassLoader();
		}

		return clsLoader;
	}

	protected InputStream getResource(String resourceName) {
		if (this.classLoader == null) {
			return JRJdtCompiler.class.getResourceAsStream("/" + resourceName);
		}
		return this.classLoader.getResourceAsStream(resourceName);
	}

	protected Class loadClass(String className) throws ClassNotFoundException {
		if (this.classLoader == null) {
			return Class.forName(className);
		}
		return this.classLoader.loadClass(className);
	}

	protected void checkLanguage(String language) throws JRException {
		if (!JRReport.LANGUAGE_JAVA.equals(language)) {
			throw new JRException("Language \"" + language
					+ "\" not supported by this report compiler.\n"
					+ "Expecting \"java\" instead.");
		}
	}

	protected JRCompilationSourceCode generateSourceCode(
		JRSourceCompileTask sourceTask) throws JRException {
		return JRClassGenerator.generateClass(sourceTask);
	}

	protected String getSourceFileName(String unitName) {
		return unitName + ".java";
	}

	protected String getCompilerClass() {
		return JRJavacCompiler.class.getName();
	}

	protected IProblem[] getJavaCompilationErrors(CompilationResult result) {
		try {
			Method getErrorsMethod =
				result.getClass().getMethod("getErrors", null);
			return (IProblem[]) getErrorsMethod.invoke(result, null);
		}
		catch (SecurityException e) {
			throw new JRRuntimeException("Error resolving JDT methods", e);
		}
		catch (NoSuchMethodException e) {
			throw new JRRuntimeException("Error resolving JDT methods", e);
		}
		catch (IllegalArgumentException e) {
			throw new JRRuntimeException("Error invoking JDT methods", e);
		}
		catch (IllegalAccessException e) {
			throw new JRRuntimeException("Error invoking JDT methods", e);
		}
		catch (InvocationTargetException e) {
			throw new JRRuntimeException("Error invoking JDT methods", e);
		}
	}
}
