/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.
*/
package org.apache.commons.jci.compilers;
import groovy.lang.GroovyClassLoader;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import org.apache.commons.jci.problems.CompilationProblem;
import org.apache.commons.jci.readers.ResourceReader;
import org.apache.commons.jci.stores.ResourceStore;
import org.apache.commons.jci.utils.ConversionUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.codehaus.groovy.control.CompilationFailedException;
import org.codehaus.groovy.control.CompilationUnit;
import org.codehaus.groovy.control.CompilerConfiguration;
import org.codehaus.groovy.control.ErrorCollector;
import org.codehaus.groovy.control.MultipleCompilationErrorsException;
import org.codehaus.groovy.control.Phases;
import org.codehaus.groovy.control.SourceUnit;
import org.codehaus.groovy.control.messages.Message;
import org.codehaus.groovy.control.messages.WarningMessage;
import org.codehaus.groovy.tools.GroovyClass;
/**
* Groovy implementation of the JavaCompiler interface
*
* @author tcurdt
*/
public final class GroovyJavaCompiler extends AbstractJavaCompiler {
private final Log log = LogFactory.getLog(GroovyJavaCompiler.class);
private final GroovyJavaCompilerSettings defaultSettings;
public GroovyJavaCompiler() {
defaultSettings = new GroovyJavaCompilerSettings(new CompilerConfiguration());
}
public CompilationResult compile(
final String[] pResourceNames,
final ResourceReader pReader,
final ResourceStore pStore,
final ClassLoader pClassLoader,
final JavaCompilerSettings pSettings
) {
final CompilerConfiguration configuration = ((GroovyJavaCompilerSettings) pSettings).getCompilerConfiguration();
final ErrorCollector collector = new ErrorCollector(configuration);
final GroovyClassLoader groovyClassLoader = new GroovyClassLoader(pClassLoader);
final CompilationUnit unit = new CompilationUnit(configuration, null, groovyClassLoader);
final SourceUnit[] source = new SourceUnit[pResourceNames.length];
for (int i = 0; i < source.length; i++) {
final String resourceName = pResourceNames[i];
source[i] = new SourceUnit(
ConversionUtils.convertResourceToClassName(resourceName),
new String(pReader.getBytes(resourceName)), // FIXME delay the read
configuration,
groovyClassLoader,
collector
);
unit.addSource(source[i]);
}
final Collection<CompilationProblem> problems = new ArrayList<CompilationProblem>();
try {
log.debug("compiling");
unit.compile(Phases.CLASS_GENERATION);
@SuppressWarnings("unchecked") // Groovy library is not yet generic
final List<GroovyClass> classes = unit.getClasses();
for (GroovyClass clazz : classes) {
final byte[] bytes = clazz.getBytes();
pStore.write(ConversionUtils.convertClassToResourcePath(clazz.getName()), bytes);
}
} catch (final MultipleCompilationErrorsException e) {
final ErrorCollector col = e.getErrorCollector();
@SuppressWarnings("unchecked") // Groovy library is not yet generic
final Collection<WarningMessage> warnings = col.getWarnings();
if (warnings != null) {
for (WarningMessage warning : warnings) {
final CompilationProblem problem = new GroovyCompilationProblem(warning);
if (problemHandler != null) {
problemHandler.handle(problem);
}
problems.add(problem);
}
}
@SuppressWarnings("unchecked") // Groovy library is not yet generic
final Collection<Message> errors = col.getErrors();
if (errors != null) {
for (Message message : errors) {
final CompilationProblem problem = new GroovyCompilationProblem(message);
if (problemHandler != null) {
problemHandler.handle(problem);
}
problems.add(problem);
}
}
} catch (CompilationFailedException e) {
throw new RuntimeException("no expected");
}
final CompilationProblem[] result = new CompilationProblem[problems.size()];
problems.toArray(result);
return new CompilationResult(result);
}
public JavaCompilerSettings createDefaultSettings() {
return defaultSettings;
}
}