org.drools.ide.common.server.util.SuggestionCompletionEngineBuilder Maven / Gradle / Ivy
/**
* Copyright 2010 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.ide.common.server.util;
import java.lang.reflect.Field;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.drools.ide.common.client.modeldriven.FieldAccessorsAndMutators;
import org.drools.ide.common.client.modeldriven.ModelField;
import org.drools.ide.common.client.modeldriven.SuggestionCompletionEngine;
import org.drools.ide.common.client.modeldriven.ModelField.FIELD_CLASS_TYPE;
import org.drools.ide.common.client.modeldriven.brl.DSLSentence;
import org.drools.lang.dsl.DSLMappingEntry;
/**
* A builder to incrementally populate a SuggestionCompletionEngine
*
* @author etirelli
*/
public class SuggestionCompletionEngineBuilder {
private SuggestionCompletionEngine instance = new SuggestionCompletionEngine();
private Map factTypes = new HashMap();
private Map fieldsForType = new HashMap();
private Map modifiersForType = new HashMap();
private Map fieldTypes = new HashMap();
private Map> fieldClasses = new HashMap>();
private Map fieldTypesField = new HashMap();
private Map globalTypes = new HashMap();
private List actionDSLSentences = new ArrayList();
private List conditionDSLSentences = new ArrayList();
private List keywordDSLItems = new ArrayList();
private List anyScopeDSLItems = new ArrayList();
private List globalCollections = new ArrayList();
private Map accessorsAndMutators = new HashMap();
public SuggestionCompletionEngineBuilder() {
}
/**
* Start the creation of a new SuggestionCompletionEngine
*/
public void newCompletionEngine() {
this.instance = new SuggestionCompletionEngine();
this.factTypes = new HashMap();
this.fieldsForType = new HashMap();
this.modifiersForType = new HashMap();
this.fieldTypes = new HashMap();
this.fieldTypesField = new HashMap();
this.globalTypes = new HashMap();
this.actionDSLSentences = new ArrayList();
this.conditionDSLSentences = new ArrayList();
this.keywordDSLItems = new ArrayList();
this.anyScopeDSLItems = new ArrayList();
this.globalCollections = new ArrayList();
this.accessorsAndMutators = new HashMap();
}
/**
* Adds a fact type to the engine
*
* @param factType
*/
public void addFactType(final String factType,final FIELD_CLASS_TYPE type) {
this.factTypes.put(factType,type);
}
/**
* Adds the list of fields for a given type
*
* @param type
* @param fields
*/
public void addFieldsForType(final String type,
final String[] fields) {
this.fieldsForType.put(type,
fields);
}
/**
* Adds the list of modifiers for a given type
*
* @param type
* @param fields
*/
public void addModifiersForType(final String type,
final String[] fields) {
this.modifiersForType.put(type,
fields);
}
/**
* @return true if this has the type already registered (field information).
*/
public boolean hasFieldsForType(final String type) {
return this.fieldsForType.containsKey(type);
}
/**
* Adds a type declaration for a field
*
* @param field format: class.field
* @param type parametrized type of clazz
* @param clazz the class of field
*/
public void addFieldType(final String field, final String type,
final Class clazz) {
this.fieldTypes.put(field, type);
this.fieldClasses.put(field, clazz);
}
/**
* Adds a type declaration for a field
*
* @param field format: class.field
* @param type
*/
public void addFieldTypeField(final String field, final Field type) {
this.fieldTypesField.put(field, type);
}
/**
* Adds a global and its corresponding type to the engine
*
* @param global
* @param type
*/
public void addGlobalType(final String global,
final String type) {
this.globalTypes.put(global, type);
}
public void addGlobalCollection(String global) {
this.globalCollections.add(global);
}
/**
* Add a DSL sentence for an action.
*/
public void addDSLActionSentence(final String sentence) {
final DSLSentence sen = new DSLSentence();
sen.sentence = sentence;
this.actionDSLSentences.add(sen);
}
/**
* Add a DSL sentence for a condition.
*/
public void addDSLConditionSentence(final String sentence) {
final DSLSentence sen = new DSLSentence();
sen.sentence = sentence;
this.conditionDSLSentences.add(sen);
}
static public String obtainGenericType(Type type) {
if (type instanceof ParameterizedType) {
ParameterizedType pt = (ParameterizedType) type;
Type goodType = null;
for (Type t : pt.getActualTypeArguments()) {
goodType = t;
}
int index = goodType.toString().lastIndexOf(".");
return goodType.toString().substring(index + 1);
}
return null;
}
public void putParametricFieldType(String fieldName, String genericType) {
this.instance.putParametricFieldType(fieldName, genericType);
}
/**
* Returns a SuggestionCompletionEngine instance populated with
* all the data since last call to newCompletionEngine() method
*
* @return
*/
public SuggestionCompletionEngine getInstance() {
this.instance.setFactTypes( this.factTypes.keySet().toArray( new String[this.factTypes.keySet().size()] ) );
this.instance.setModifiers(this.modifiersForType);
//convert this.fieldsForType, this.fieldClasses and this.fieldTypes into Map.
Map modelMap = new HashMap();
for (Map.Entry typeEntry : this.fieldsForType.entrySet()) {
List fields = new ArrayList();
for (String field : typeEntry.getValue()) {
String fieldName = field;
String fieldType = this.fieldTypes.get(typeEntry.getKey() + "." + field);
Class fieldClazz = this.fieldClasses.get(typeEntry.getKey() + "." + field);
fields.add(new ModelField(
fieldName, fieldClazz == null ? null : fieldClazz.getName(), this.factTypes.get( typeEntry.getKey() ), fieldType));
}
modelMap.put(typeEntry.getKey(), fields.toArray(new ModelField[fields.size()]));
}
this.instance.setFieldsForTypes(modelMap);
for (String fieldName : this.fieldTypesField.keySet()) {
Field field = this.fieldTypesField.get(fieldName);
if (field != null) {
String genericType = obtainGenericType(field.getGenericType());
if (genericType != null) {
this.instance.putParametricFieldType(fieldName, genericType);
}
Class fieldClass = field.getType();
if (fieldClass.isEnum()) {
Field[] flds = fieldClass.getDeclaredFields();
List listEnum = new ArrayList();
int i = 0;
for (Field f : flds) {
if (f.isEnumConstant()) {
listEnum.add(i + "=" + f.getName());
i++;
}
}
String a[] = new String[listEnum.size()];
i = 0;
for (String value : listEnum) {
a[i] = value;
i++;
}
this.instance.putDataEnumList(fieldName, a);
}
}
}
this.instance.setGlobalVariables(this.globalTypes);
this.instance.actionDSLSentences = makeArray(this.actionDSLSentences);
this.instance.conditionDSLSentences = makeArray(this.conditionDSLSentences);
this.instance.keywordDSLItems = makeArray(this.keywordDSLItems);
this.instance.anyScopeDSLItems = makeArray(this.anyScopeDSLItems);
this.instance.setGlobalCollections(this.globalCollections.toArray(new String[globalCollections.size()]));
this.instance.setAccessorsAndMutators( accessorsAndMutators );
return this.instance;
}
private DSLSentence[] makeArray(List ls) {
return ls.toArray(new DSLSentence[ls.size()]);
}
public void addDSLMapping(DSLMappingEntry entry) {
DSLSentence sen = new DSLSentence();
sen.sentence = entry.getMappingKey();
if (entry.getSection() == DSLMappingEntry.CONDITION) {
this.conditionDSLSentences.add(sen);
} else if (entry.getSection() == DSLMappingEntry.CONSEQUENCE) {
this.actionDSLSentences.add(sen);
} else if (entry.getSection() == DSLMappingEntry.KEYWORD) {
this.keywordDSLItems.add(sen);
} else if (entry.getSection() == DSLMappingEntry.ANY) {
this.anyScopeDSLItems.add(sen);
}
}
public void addFieldAccessorsAndMutatorsForField(Map accessorsAndMutators) {
this.accessorsAndMutators.putAll( accessorsAndMutators );
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy