/*
* Copyright 2015 Red Hat, Inc. and/or its affiliates.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
*
* 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.kproject.models;
import com.thoughtworks.xstream.converters.MarshallingContext;
import com.thoughtworks.xstream.converters.UnmarshallingContext;
import com.thoughtworks.xstream.io.HierarchicalStreamReader;
import com.thoughtworks.xstream.io.HierarchicalStreamWriter;
import org.drools.core.util.AbstractXStreamConverter;
import org.drools.core.util.StringUtils;
import org.kie.api.builder.model.KieBaseModel;
import org.kie.api.builder.model.KieModuleModel;
import org.kie.api.builder.model.KieSessionModel;
import org.kie.api.builder.model.RuleTemplateModel;
import org.kie.api.conf.DeclarativeAgendaOption;
import org.kie.api.conf.EqualityBehaviorOption;
import org.kie.api.conf.EventProcessingOption;
import org.kie.api.io.ResourceType;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import static org.drools.core.util.IoUtils.recursiveListFile;
public class KieBaseModelImpl
implements
KieBaseModel {
private String name;
private Set<String> includes= new HashSet<String>();
private List<String> packages;
private EqualityBehaviorOption equalsBehavior = EqualityBehaviorOption.IDENTITY;
private EventProcessingOption eventProcessingMode = EventProcessingOption.CLOUD;
private DeclarativeAgendaOption declarativeAgenda = DeclarativeAgendaOption.DISABLED;
private Map<String, KieSessionModel> kSessions = new HashMap<String, KieSessionModel>();
private KieModuleModel kModule;
private String scope = "javax.enterprise.context.ApplicationScoped";
private List<RuleTemplateModel> ruleTemplates = new ArrayList<RuleTemplateModel>();
private boolean isDefault = false;
public KieBaseModelImpl() {
}
public KieBaseModelImpl(KieModuleModel kModule,
String name) {
this.kModule = kModule;
this.includes = new HashSet<String>();
this.name = name;
this.kSessions = Collections.emptyMap();
}
public boolean isDefault() {
return isDefault;
}
public KieBaseModel setDefault(boolean isDefault) {
this.isDefault = isDefault;
return this;
}
public List<String> getPackages() {
return packages != null ? packages : Collections.<String>emptyList();
}
public KieBaseModel addPackage(String pkg) {
if ( packages == null ) {
packages = new ArrayList<String>();
}
packages.add( pkg );
return this;
}
public KieBaseModel removePackage(String pkg) {
if ( packages != null ) {
packages.remove( pkg );
}
return this;
}
public KieModuleModel getKModule() {
return kModule;
}
public void setKModule(KieModuleModel kieProject) {
this.kModule = kieProject;
}
/* (non-Javadoc)
* @see org.kie.kproject.KieBaseModel#getKieSessionModels()
*/
public Map<String, KieSessionModel> getKieSessionModels() {
return Collections.unmodifiableMap( kSessions );
}
public Map<String, KieSessionModel> getRawKieSessionModels() {
return kSessions;
}
/* (non-Javadoc)
* @see org.kie.kproject.KieBaseModel#setKSessions(java.util.Map)
*/
private void setKSessions(Map<String, KieSessionModel> kSessions) {
this.kSessions = kSessions;
}
/* (non-Javadoc)
* @see org.kie.kproject.KieBaseModel#addKSession(org.kie.kproject.KieSessionModelImpl)
*/
public KieSessionModel newKieSessionModel(String name) {
KieSessionModel kieSessionModel = new KieSessionModelImpl( this, name );
Map<String, KieSessionModel> newMap = new HashMap<String, KieSessionModel>();
newMap.putAll( this.kSessions );
newMap.put( kieSessionModel.getName(), kieSessionModel );
setKSessions( newMap );
return kieSessionModel;
}
/* (non-Javadoc)
* @see org.kie.kproject.KieBaseModel#removeKieSessionModel(org.kie.kproject.KieSessionModelImpl)
*/
public KieBaseModel removeKieSessionModel(String qName) {
Map<String, KieSessionModel> newMap = new HashMap<String, KieSessionModel>();
newMap.putAll( this.kSessions );
newMap.remove(qName);
setKSessions(newMap);
return this;
}
public void moveKSession(String oldQName,
String newQName) {
Map<String, KieSessionModel> newMap = new HashMap<String, KieSessionModel>();
newMap.putAll( this.kSessions );
KieSessionModel kieSessionModel = newMap.remove( oldQName );
newMap.put(newQName, kieSessionModel);
setKSessions( newMap );
}
public List<RuleTemplateModel> getRuleTemplates() {
return Collections.unmodifiableList( ruleTemplates );
}
public List<RuleTemplateModel> getRawRuleTemplates() {
return ruleTemplates;
}
public KieBaseModel addRuleTemplate(String dtable, String template, int row, int col) {
ruleTemplates.add( new RuleTemplateModelImpl( this, dtable, template, row, col ) );
return this;
}
/* (non-Javadoc)
* @see org.kie.kproject.KieBaseModel#getName()
*/
public String getName() {
return name;
}
public KieBaseModel setName(String name) {
((KieModuleModelImpl)kModule).changeKBaseName(this, this.name, name);
this.name = name;
return this;
}
void changeKSessionName(KieSessionModel kieSession, String oldName, String newName) {
kSessions.remove(oldName);
kSessions.put(newName, kieSession);
}
public Set<String> getIncludes() {
return Collections.unmodifiableSet( includes );
}
public KieBaseModel addInclude(String kBaseQName) {
this.includes.add( kBaseQName );
return this;
}
public KieBaseModel removeInclude(String kBaseQName) {
this.includes.remove( kBaseQName );
return this;
}
/* (non-Javadoc)
* @see org.kie.kproject.KieBaseModel#getEqualsBehavior()
*/
public EqualityBehaviorOption getEqualsBehavior() {
return equalsBehavior;
}
/* (non-Javadoc)
* @see org.kie.kproject.KieBaseModel#setEqualsBehavior(org.kie.api.conf.EqualityBehaviorOption)
*/
public KieBaseModel setEqualsBehavior(EqualityBehaviorOption equalsBehaviour) {
this.equalsBehavior = equalsBehaviour;
return this;
}
/* (non-Javadoc)
* @see org.kie.kproject.KieBaseModel#getEventProcessingMode()
*/
public EventProcessingOption getEventProcessingMode() {
return eventProcessingMode;
}
/* (non-Javadoc)
* @see org.kie.kproject.KieBaseModel#setEventProcessingMode(org.kie.api.conf.EventProcessingOption)
*/
public KieBaseModel setEventProcessingMode(EventProcessingOption eventProcessingMode) {
this.eventProcessingMode = eventProcessingMode;
return this;
}
public DeclarativeAgendaOption getDeclarativeAgenda() {
return declarativeAgenda;
}
public KieBaseModel setDeclarativeAgenda(DeclarativeAgendaOption declarativeAgenda) {
this.declarativeAgenda = declarativeAgenda;
return this;
}
@Override
public KieBaseModel setScope(String scope) {
this.scope = scope;
return this;
}
@Override
public String getScope() {
return this.scope;
}
public static List<String> getFiles(String kBaseName,
ZipFile zipFile) {
List<String> files = new ArrayList<String>();
Enumeration< ? extends ZipEntry> zipEntries = zipFile.entries();
while ( zipEntries.hasMoreElements() ) {
ZipEntry zipEntry = zipEntries.nextElement();
String fileName = zipEntry.getName();
if ( filterFileInKBase( kBaseName.replace( '.', '/' ), fileName ) ) {
files.add( fileName );
}
}
return files;
}
public static List<String> getFiles(java.io.File rootFolder) {
return recursiveListFile( rootFolder, "", file -> {
String fileName = file.getName();
return fileName.endsWith( ResourceType.DRL.getDefaultExtension() ) ||
fileName.endsWith( ResourceType.GDRL.getDefaultExtension() ) ||
fileName.endsWith( ResourceType.RDRL.getDefaultExtension() ) ||
fileName.endsWith( ResourceType.BPMN2.getDefaultExtension() ) ||
fileName.endsWith( ResourceType.TDRL.getDefaultExtension() );
} );
}
private static boolean filterFileInKBase(String rootPath,
String fileName) {
return fileName.startsWith( rootPath ) &&
(fileName.endsWith( ResourceType.DRL.getDefaultExtension() ) ||
fileName.endsWith( ResourceType.GDRL.getDefaultExtension() ) ||
fileName.endsWith( ResourceType.RDRL.getDefaultExtension() ) ||
fileName.endsWith( ResourceType.BPMN2.getDefaultExtension() ) ||
fileName.endsWith( ResourceType.TDRL.getDefaultExtension() ) );
}
public static class KBaseConverter extends AbstractXStreamConverter {
public KBaseConverter() {
super( KieBaseModelImpl.class );
}
public void marshal(Object value,
HierarchicalStreamWriter writer,
MarshallingContext context) {
KieBaseModelImpl kBase = (KieBaseModelImpl) value;
writer.addAttribute( "name", kBase.getName() );
writer.addAttribute( "default", Boolean.toString(kBase.isDefault()) );
if ( kBase.getEventProcessingMode() != null ) {
writer.addAttribute( "eventProcessingMode", kBase.getEventProcessingMode().getMode() );
}
if ( kBase.getEqualsBehavior() != null ) {
writer.addAttribute( "equalsBehavior", kBase.getEqualsBehavior().toString().toLowerCase() );
}
if ( kBase.getDeclarativeAgenda() != null ) {
writer.addAttribute( "declarativeAgenda", kBase.getDeclarativeAgenda().toString().toLowerCase() );
}
if ( kBase.getScope() != null ) {
writer.addAttribute( "scope", kBase.getScope() );
}
if ( ! kBase.getPackages().isEmpty() ) {
StringBuilder buf = new StringBuilder();
boolean first = true;
for( String pkg : kBase.getPackages() ) {
if( first ) {
first = false;
} else {
buf.append( ", " );
}
buf.append( pkg );
}
writer.addAttribute( "packages", buf.toString() );
}
if ( !kBase.getIncludes().isEmpty() ) {
StringBuilder sb = new StringBuilder();
boolean insertComma = false;
for ( String include : kBase.getIncludes() ) {
if ( insertComma ) {
sb.append( ", " );
}
sb.append( include );
if ( !insertComma ) {
insertComma = true;
}
}
writer.addAttribute( "includes", sb.toString() );
}
for ( RuleTemplateModel ruleTemplateModel : kBase.getRuleTemplates()) {
writeObject( writer, context, "ruleTemplate", ruleTemplateModel);
}
for ( KieSessionModel kSessionModel : kBase.getKieSessionModels().values()) {
writeObject( writer, context, "ksession", kSessionModel);
}
}
public Object unmarshal(HierarchicalStreamReader reader,
final UnmarshallingContext context) {
final KieBaseModelImpl kBase = new KieBaseModelImpl();
String kbaseName = reader.getAttribute( "name" );
kBase.name = kbaseName != null ? kbaseName : StringUtils.uuid();
kBase.setDefault( "true".equals(reader.getAttribute( "default" )) );
String eventMode = reader.getAttribute( "eventProcessingMode" );
if ( eventMode != null ) {
kBase.setEventProcessingMode( EventProcessingOption.determineEventProcessingMode( eventMode ) );
}
String equalsBehavior = reader.getAttribute( "equalsBehavior" );
if ( equalsBehavior != null ) {
kBase.setEqualsBehavior( EqualityBehaviorOption.determineEqualityBehavior( equalsBehavior ) );
}
String declarativeAgenda = reader.getAttribute( "declarativeAgenda" );
if ( declarativeAgenda != null ) {
kBase.setDeclarativeAgenda( DeclarativeAgendaOption.determineDeclarativeAgenda( declarativeAgenda ) );
}
String scope = reader.getAttribute( "scope" );
if ( scope != null ) {
kBase.setScope( scope.trim() );
}
String pkgs = reader.getAttribute( "packages" );
if( pkgs != null ) {
for( String pkg : pkgs.split( "," ) ) {
kBase.addPackage( pkg.trim() );
}
}
String includes = reader.getAttribute( "includes" );
if( includes != null ) {
for( String include : includes.split( "," ) ) {
kBase.addInclude( include.trim() );
}
}
readNodes( reader, new AbstractXStreamConverter.NodeReader() {
public void onNode(HierarchicalStreamReader reader,
String name,
String value) {
if ( "ksession".equals( name ) ) {
KieSessionModelImpl kSession = readObject( reader, context, KieSessionModelImpl.class );
kBase.getRawKieSessionModels().put( kSession.getName(), kSession );
kSession.setKBase(kBase);
} else if ( "ruleTemplate".equals( name ) ) {
RuleTemplateModelImpl ruleTemplate = readObject( reader, context, RuleTemplateModelImpl.class );
kBase.getRawRuleTemplates().add( ruleTemplate );
ruleTemplate.setKBase( kBase );
}
// @TODO we don't use support nested includes
// if ( "includes".equals( name ) ) {
// for ( String include : readList( reader ) ) {
// kBase.addInclude( include );
// }
// }
}
} );
return kBase;
}
}
@Override
public String toString() {
return "KieBaseModelImpl [name=" + name + ", includes=" + includes + ", packages=" + getPackages() + ", equalsBehavior=" + equalsBehavior + ", eventProcessingMode=" + eventProcessingMode + ", kSessions=" + kSessions + "]";
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
KieBaseModelImpl that = (KieBaseModelImpl) o;
return name.equals(that.name);
}
@Override
public int hashCode() {
return name.hashCode();
}
}