/*
 * 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);
    }
  }
}