/*
* Copyright 2005 JBoss Inc
*
* 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.
*/
package org.drools.compiler.compiler;
import static org.drools.compiler.compiler.DRLFactory.buildLexer;
import static org.drools.compiler.compiler.DRLFactory.buildParser;
import java.io.IOException;
import java.io.InputStream;
import java.io.Reader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import org.drools.core.io.impl.ClassPathResource;
import org.drools.core.io.impl.InputStreamResource;
import org.drools.core.io.impl.ReaderResource;
import org.drools.compiler.lang.DRLLexer;
import org.drools.compiler.lang.DRLParser;
import org.drools.compiler.lang.DroolsSentence;
import org.drools.compiler.lang.Expander;
import org.drools.compiler.lang.ExpanderException;
import org.drools.compiler.lang.Location;
import org.drools.compiler.lang.descr.PackageDescr;
import org.drools.compiler.lang.dsl.DefaultExpanderResolver;
import org.kie.internal.builder.conf.LanguageLevelOption;
import org.kie.api.io.Resource;
/**
* This is a low level parser API. This will return textual AST representations
* of the DRL source, including with DSL expanders if appropriate.
*/
public class DrlParser {
// TODO: REMOVE THIS GENERIC MESSAGE ASAP
private static final String GENERIC_ERROR_MESSAGE = "Unexpected exception raised while parsing. This is a bug. Please contact the Development team :\n";
private final List<DroolsError> results = new ArrayList<DroolsError>();
private List<DroolsSentence> editorSentences = null;
private Location location = new Location( Location.LOCATION_UNKNOWN );
private DRLLexer lexer = null;
private Resource resource = null;
public static final LanguageLevelOption DEFAULT_LANGUAGE_LEVEL = LanguageLevelOption.DRL6;
private final LanguageLevelOption languageLevel;
public DrlParser() {
this(DEFAULT_LANGUAGE_LEVEL);
}
public DrlParser(LanguageLevelOption languageLevel) {
this.languageLevel = languageLevel;
}
/** Parse a rule from text */
public PackageDescr parse(final String text) throws DroolsParserException {
return parse(false, text);
}
public PackageDescr parse(final boolean isEditor,
final String text) throws DroolsParserException {
lexer = buildLexer(text, languageLevel);
DRLParser parser = buildParser(lexer, languageLevel);
return compile(isEditor, parser);
}
public PackageDescr parse(final boolean isEditor,
final Reader reader) throws DroolsParserException {
lexer = buildLexer(reader, languageLevel);
DRLParser parser = buildParser( lexer, languageLevel );
return compile(isEditor, parser);
}
public PackageDescr parse(final Resource resource, final Reader reader) throws DroolsParserException {
this.resource = resource;
return parse(false, reader);
}
public PackageDescr parse(final Reader reader) throws DroolsParserException {
return parse(false, reader);
}
/**
* Parse and build a rule package from a DRL source with a domain specific
* language.
*/
public PackageDescr parse(final Reader drl,
final Reader dsl) throws DroolsParserException,
IOException {
return parse(false,
drl,
dsl);
}
public PackageDescr parse(boolean isEditor,
final Reader drl,
final Reader dsl) throws DroolsParserException,
IOException {
final StringBuilder text = getDRLText(drl);
return parse(text.toString(), dsl);
}
/**
* Parse and build a rule package from a DRL source with a domain specific
* language.
*
* @param source
* As Text.
* @param dsl
* @return
* @throws DroolsParserException
*/
public PackageDescr parse(boolean isEditor,
final String source,
final Reader dsl) throws DroolsParserException {
DefaultExpanderResolver resolver = getDefaultResolver(dsl);
final Expander expander = resolver.get( "*", null );
final String expanded = expander.expand( source );
if ( expander.hasErrors() ) {
this.results.addAll( expander.getErrors() );
}
return this.parse(isEditor, expanded);
}
public PackageDescr parse(final String source,
final Reader dsl) throws DroolsParserException {
return this.parse(false, source, dsl);
}
public PackageDescr parse(final Resource resource) throws DroolsParserException, IOException {
return parse(false, resource);
}
public PackageDescr parse(final boolean isEditor,
final Resource resource) throws DroolsParserException, IOException {
this.resource = resource;
InputStream is = resource.getInputStream();
String encoding = null;
if (resource instanceof ClassPathResource) {
encoding = ((ClassPathResource) resource).getEncoding();
}
if (resource instanceof ReaderResource) {
encoding = ((ReaderResource) resource).getEncoding();
}
if (resource instanceof InputStreamResource) {
encoding = ((InputStreamResource) resource).getEncoding();
}
lexer = buildLexer(is, encoding, languageLevel);
DRLParser parser = buildParser(lexer, languageLevel);
return compile(isEditor, parser);
}
/**
* This will expand the DRL. useful for debugging.
*
* @param source -
* the source which use a DSL
* @param dsl -
* the DSL itself.
* @throws DroolsParserException
* If unable to expand in any way.
*/
public String getExpandedDRL(final String source,
final Reader dsl) throws DroolsParserException {
DefaultExpanderResolver resolver = getDefaultResolver(dsl);
return getExpandedDRL(source,
resolver);
}
/**
* This will expand the DRL using the given expander resolver. useful for
* debugging.
*
* @param source -
* the source which use a DSL
* @param resolver -
* the DSL expander resolver itself.
* @throws DroolsParserException
* If unable to expand in any way.
*/
public String getExpandedDRL(final String source,
final DefaultExpanderResolver resolver) throws DroolsParserException {
final Expander expander = resolver.get("*",
null);
final String expanded = expander.expand( source );
if ( expander.hasErrors() ) {
String err = "";
for ( ExpanderException ex : expander.getErrors() ) {
err = err + "\n Line:[" + ex.getLine() + "] " + ex.getMessage();
}
throw new DroolsParserException( err );
}
return expanded;
}
private StringBuilder getDRLText(final Reader reader) throws IOException {
final StringBuilder text = new StringBuilder();
final char[] buf = new char[1024];
int len;
while ( (len = reader.read( buf )) >= 0 ) {
text.append( buf,
0,
len );
}
return text;
}
/**
* @return true if there were parser errors.
*/
public boolean hasErrors() {
return !this.results.isEmpty();
}
/**
* @return a list of errors found while parsing. DroolsError: either ParserError, or ExpanderException
*/
public List<DroolsError> getErrors() {
return this.results;
}
private PackageDescr compile(boolean isEditor,
final DRLParser parser ) throws DroolsParserException {
PackageDescr pkgDescr = null;
try {
if ( isEditor ) {
parser.enableEditorInterface();
}
pkgDescr = parser.compilationUnit(resource);
editorSentences = parser.getEditorInterface();
makeErrorList( parser );
if ( isEditor || !this.hasErrors() ) {
return pkgDescr;
} else {
return null;
}
} catch ( Exception e ) {
e.printStackTrace();
final ParserError err = new ParserError( resource,
GENERIC_ERROR_MESSAGE + e.toString()+"\n"+ Arrays.toString( e.getStackTrace() ),
-1,
0 );
this.results.add( err );
if ( isEditor ) {
return pkgDescr;
} else {
throw new DroolsParserException( GENERIC_ERROR_MESSAGE + e.getMessage(),
e );
}
}
}
/** Convert the antlr exceptions to drools parser exceptions */
private void makeErrorList( final DRLParser parser ) {
for ( final DroolsParserException recogErr : lexer.getErrors() ) {
final ParserError err = new ParserError( resource,
recogErr.getMessage(),
recogErr.getLineNumber(),
recogErr.getColumn() );
this.results.add( err );
}
for ( final DroolsParserException recogErr : parser.getErrors() ) {
final ParserError err = new ParserError( resource,
recogErr.getMessage(),
recogErr.getLineNumber(),
recogErr.getColumn() );
this.results.add( err );
}
}
public Location getLocation() {
return this.location;
}
public DefaultExpanderResolver getDefaultResolver(final Reader dsl) throws DroolsParserException {
DefaultExpanderResolver resolver;
try {
resolver = new DefaultExpanderResolver( dsl );
} catch ( final IOException e ) {
throw new DroolsParserException( "Error parsing the DSL.",
e );
}
return resolver;
}
public List<DroolsSentence> getEditorSentences() {
return editorSentences;
}
}