Package net.sf.jasperreports.engine.design

Source Code of net.sf.jasperreports.engine.design.JRJdtCompiler$CompilationUnit

/*
* JasperReports - Free Java Reporting Library.
* Copyright (C) 2001 - 2009 Jaspersoft Corporation. All rights reserved.
* http://www.jaspersoft.com
*
* Unless you have purchased a commercial license agreement from Jaspersoft,
* the following license terms apply:
*
* This program is part of JasperReports.
*
* JasperReports 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 3 of the License, or
* (at your option) any later version.
*
* JasperReports 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 JasperReports. If not, see <http://www.gnu.org/licenses/>.
*/
package net.sf.jasperreports.engine.design;

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.JRLoader;
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 3420 2010-02-18 09:17:47Z teodord $
*/
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);
   
    classLoader = getClassLoader();

    boolean success;
    try //FIXME remove support for pre 3.1 jdt
    {
      Class classAccessRestriction = loadClass("org.eclipse.jdt.internal.compiler.env.AccessRestriction");
      constrNameEnvAnsBin2Args = NameEnvironmentAnswer.class.getConstructor(new Class[]{IBinaryType.class, classAccessRestriction});
      constrNameEnvAnsCompUnit2Args = NameEnvironmentAnswer.class.getConstructor(new Class[]{ICompilationUnit.class, classAccessRestriction});
      is2ArgsConstr = true;
      success = true;
    }
    catch (NoSuchMethodException e)
    {
      success = false;
    }
    catch (ClassNotFoundException ex)
    {
      success = false;
    }
   
    if (!success)
    {
      try
      {
        constrNameEnvAnsBin = NameEnvironmentAnswer.class.getConstructor(new Class[]{IBinaryType.class});
        constrNameEnvAnsCompUnit = NameEnvironmentAnswer.class.getConstructor(new Class[]{ICompilationUnit.class});
        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 className.toCharArray();
    }

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

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

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

 
  /**
   *
   */
  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 (is2ArgsConstr)
            {
              return (NameEnvironmentAnswer) constrNameEnvAnsCompUnit2Args.newInstance(new Object[] { compilationUnit, null });
            }

            return (NameEnvironmentAnswer) constrNameEnvAnsCompUnit.newInstance(new Object[] { compilationUnit });
          }
         
          String resourceName = className.replace('.', '/') + ".class";
          InputStream is = getResource(resourceName);
          if (is != null)
          {
            try
            {
              byte[] classBytes = JRLoader.loadBytes(is);
              char[] fileName = className.toCharArray();
              ClassFileReader classFileReader =
                new ClassFileReader(classBytes, fileName, true);
             
              if (is2ArgsConstr)
              {
                return (NameEnvironmentAnswer) constrNameEnvAnsBin2Args.newInstance(new Object[] { classFileReader, null });
              }

              return (NameEnvironmentAnswer) constrNameEnvAnsBin.newInstance(new Object[] { classFileReader });
            }
            finally
            {
              try
              {
                is.close();
              }
              catch (IOException e)
              {
                // ignore
              }
            }
          }
        }
        catch (JRException e)
        {
          log.error("Compilation error", e);
        }
        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 (classLoader == null)
    {
      return JRJdtCompiler.class.getResourceAsStream("/" + resourceName);
    }
    return classLoader.getResourceAsStream(resourceName);
  }
 
 
  protected Class loadClass (String className) throws ClassNotFoundException
  {
    if (classLoader == null)
    {
      return Class.forName(className);
    }
    return 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", (Class[])null);
      return (IProblem[]) getErrorsMethod.invoke(result, (Object[])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);
    }
  }
}
TOP

Related Classes of net.sf.jasperreports.engine.design.JRJdtCompiler$CompilationUnit

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.