package org.drools.compiler.builder.impl;
import org.drools.compiler.compiler.BoundIdentifiers;
import org.drools.compiler.compiler.PackageRegistry;
import org.drools.compiler.compiler.TypeDeclarationError;
import org.drools.compiler.lang.descr.AbstractClassTypeDeclarationDescr;
import org.drools.compiler.rule.builder.PackageBuildContext;
import org.drools.compiler.rule.builder.dialect.mvel.MVELAnalysisResult;
import org.drools.compiler.rule.builder.dialect.mvel.MVELDialect;
import org.drools.core.base.ClassFieldAccessor;
import org.drools.core.base.ClassFieldAccessorStore;
import org.drools.core.base.evaluators.TimeIntervalParser;
import org.drools.core.base.mvel.MVELCompileable;
import org.drools.core.definitions.InternalKnowledgePackage;
import org.drools.core.factmodel.ClassDefinition;
import org.drools.core.factmodel.FieldDefinition;
import org.drools.core.rule.MVELDialectRuntimeData;
import org.drools.core.rule.TypeDeclaration;
import org.drools.core.spi.InternalReadAccessor;
import org.drools.core.util.ClassUtils;
import org.kie.api.definition.type.ClassReactive;
import org.kie.api.definition.type.Duration;
import org.kie.api.definition.type.Expires;
import org.kie.api.definition.type.PropertyReactive;
import org.kie.api.definition.type.Role;
import org.kie.api.definition.type.Timestamp;
import org.kie.internal.builder.conf.PropertySpecificOption;
import java.beans.IntrospectionException;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.util.Collections;
import java.util.Map;
public class TypeDeclarationConfigurator {
protected TimeIntervalParser timeParser;
protected KnowledgeBuilderImpl kbuilder;
public TypeDeclarationConfigurator( KnowledgeBuilderImpl kbuilder ) {
this.kbuilder = kbuilder;
}
public void finalize( TypeDeclaration type, AbstractClassTypeDeclarationDescr typeDescr, PackageRegistry pkgRegistry, Map<String, PackageRegistry> pkgRegistryMap, ClassHierarchyManager hierarchyManager ) {
// prefer definitions where possible
if ( type.getNature() == TypeDeclaration.Nature.DEFINITION ) {
hierarchyManager.addDeclarationToPackagePreservingOrder( type, typeDescr, pkgRegistry.getPackage(), pkgRegistryMap );
} else {
TypeDeclaration oldType = pkgRegistry.getPackage().getTypeDeclaration( type.getTypeName() );
if ( oldType == null ) {
pkgRegistry.getPackage().addTypeDeclaration( type );
} else {
if (type.getRole() == Role.Type.EVENT) {
oldType.setRole(Role.Type.EVENT);
if ( type.getDurationAttribute() != null ) {
oldType.setDurationAttribute( type.getDurationAttribute() );
oldType.setDurationExtractor( type.getDurationExtractor() );
}
if ( type.getTimestampAttribute() != null ) {
oldType.setTimestampAttribute( type.getTimestampAttribute() );
oldType.setTimestampExtractor( type.getTimestampExtractor() );
}
if ( type.getExpirationOffset() >= 0 ) {
oldType.setExpirationOffset( type.getExpirationOffset() );
}
}
if (type.isPropertyReactive()) {
oldType.setPropertyReactive(true);
}
}
}
}
public boolean wireFieldAccessors( PackageRegistry pkgRegistry,
AbstractClassTypeDeclarationDescr typeDescr,
TypeDeclaration type ) {
if ( type.getTypeClassDef() != null ) {
try {
buildFieldAccessors( type, pkgRegistry );
} catch ( Throwable e ) {
kbuilder.addBuilderResult(new TypeDeclarationError(typeDescr,
"Error creating field accessors for TypeDeclaration '" + type.getTypeName() +
"' for type '" +
type.getTypeName() +
" : " + e.getMessage() +
"'"));
return false;
}
}
wireTimestampAccessor( typeDescr, type, pkgRegistry );
wireDurationAccessor( typeDescr, type, pkgRegistry );
configureExpirationOffset( typeDescr, type );
configurePropertyReactivity( typeDescr, type );
return true;
}
protected void buildFieldAccessors(final TypeDeclaration type,
final PackageRegistry pkgRegistry) throws SecurityException,
IllegalArgumentException,
InstantiationException,
IllegalAccessException,
IOException,
IntrospectionException,
ClassNotFoundException,
NoSuchMethodException,
InvocationTargetException,
NoSuchFieldException {
ClassDefinition cd = type.getTypeClassDef();
ClassFieldAccessorStore store = pkgRegistry.getPackage().getClassFieldAccessorStore();
for ( FieldDefinition attrDef : cd.getFieldsDefinitions() ) {
ClassFieldAccessor accessor = store.getAccessor( cd.getDefinedClass().getName(),
attrDef.getName() );
attrDef.setReadWriteAccessor( accessor );
}
}
protected void wireTimestampAccessor( AbstractClassTypeDeclarationDescr typeDescr, TypeDeclaration type, PackageRegistry pkgRegistry ) {
Timestamp timestamp = typeDescr.getTypedAnnotation(Timestamp.class);
if ( timestamp != null ) {
String timestampField = null;
try {
timestampField = timestamp.value();
} catch (Exception e) {
kbuilder.addBuilderResult(new TypeDeclarationError(typeDescr, e.getMessage()));
return;
}
type.setTimestampAttribute( timestampField );
InternalKnowledgePackage pkg = pkgRegistry.getPackage();
MVELDialect dialect = (MVELDialect) pkgRegistry.getDialectCompiletimeRegistry().getDialect( "mvel" );
PackageBuildContext context = new PackageBuildContext();
context.init( kbuilder, pkg, typeDescr, pkgRegistry.getDialectCompiletimeRegistry(), dialect, null );
if ( ! type.isTypesafe() ) {
context.setTypesafe( false );
}
MVELAnalysisResult results = (MVELAnalysisResult)
context.getDialect().analyzeExpression(context,
typeDescr,
timestampField,
new BoundIdentifiers( Collections.EMPTY_MAP,
Collections.EMPTY_MAP,
Collections.EMPTY_MAP,
type.getTypeClass()));
if (results != null) {
InternalReadAccessor reader = pkg.getClassFieldAccessorStore().getMVELReader( ClassUtils.getPackage( type.getTypeClass() ),
type.getTypeClass().getName(),
timestampField,
type.isTypesafe(),
results.getReturnType());
MVELDialectRuntimeData data = (MVELDialectRuntimeData) pkg.getDialectRuntimeRegistry().getDialectData("mvel");
data.addCompileable((MVELCompileable) reader);
((MVELCompileable) reader).compile(data);
type.setTimestampExtractor(reader);
} else {
kbuilder.addBuilderResult(new TypeDeclarationError(typeDescr,
"Error creating field accessors for timestamp field '" + timestamp +
"' for type '" +
type.getTypeName() +
"'"));
}
}
}
protected void wireDurationAccessor( AbstractClassTypeDeclarationDescr typeDescr, TypeDeclaration type, PackageRegistry pkgRegistry ) {
Duration duration = typeDescr.getTypedAnnotation(Duration.class);
if (duration != null) {
String durationField = null;
try {
durationField = duration.value();
} catch (Exception e) {
kbuilder.addBuilderResult(new TypeDeclarationError(typeDescr, e.getMessage()));
return;
}
type.setDurationAttribute(durationField);
InternalKnowledgePackage pkg = pkgRegistry.getPackage();
MVELDialect dialect = (MVELDialect) pkgRegistry.getDialectCompiletimeRegistry().getDialect("mvel");
PackageBuildContext context = new PackageBuildContext();
context.init(kbuilder, pkg, typeDescr, pkgRegistry.getDialectCompiletimeRegistry(), dialect, null);
if (!type.isTypesafe()) {
context.setTypesafe(false);
}
MVELAnalysisResult results = (MVELAnalysisResult)
context.getDialect().analyzeExpression(context,
typeDescr,
durationField,
new BoundIdentifiers(Collections.EMPTY_MAP,
Collections.EMPTY_MAP,
Collections.EMPTY_MAP,
type.getTypeClass()));
if (results != null) {
InternalReadAccessor reader = pkg.getClassFieldAccessorStore().getMVELReader(ClassUtils.getPackage(type.getTypeClass()),
type.getTypeClass().getName(),
durationField,
type.isTypesafe(),
results.getReturnType());
MVELDialectRuntimeData data = (MVELDialectRuntimeData) pkg.getDialectRuntimeRegistry().getDialectData("mvel");
data.addCompileable((MVELCompileable) reader);
((MVELCompileable) reader).compile(data);
type.setDurationExtractor(reader);
} else {
kbuilder.addBuilderResult(new TypeDeclarationError(typeDescr,
"Error processing @duration for TypeDeclaration '" + type.getFullName() +
"': cannot access the field '" + durationField + "'"));
}
}
}
protected void configureExpirationOffset( AbstractClassTypeDeclarationDescr typeDescr, TypeDeclaration type ) {
Expires expires = typeDescr.getTypedAnnotation(Expires.class);
if (expires != null) {
String expiration = null;
try {
expiration = expires.value();
} catch (Exception e) {
kbuilder.addBuilderResult(new TypeDeclarationError(typeDescr, e.getMessage()));
return;
}
if (timeParser == null) {
timeParser = new TimeIntervalParser();
}
type.setExpirationOffset(timeParser.parse(expiration)[0]);
}
}
protected void configurePropertyReactivity( AbstractClassTypeDeclarationDescr typeDescr, TypeDeclaration type ) {
PropertySpecificOption propertySpecificOption = kbuilder.getBuilderConfiguration().getPropertySpecificOption();
boolean propertyReactive = propertySpecificOption.isPropSpecific( typeDescr.hasAnnotation(PropertyReactive.class),
typeDescr.hasAnnotation(ClassReactive.class) );
type.setPropertyReactive(propertyReactive);
}
}