use of org.kie.workbench.common.stunner.core.definition.annotation.Definition in project kie-wb-common by kiegroup.
the class MainProcessor method processDefinitions.
private boolean processDefinitions(final Set<? extends TypeElement> set, final Element e, final RoundEnvironment roundEnv) throws Exception {
final boolean isClass = e.getKind() == ElementKind.CLASS;
if (isClass) {
TypeElement classElement = (TypeElement) e;
PackageElement packageElement = (PackageElement) classElement.getEnclosingElement();
String defintionClassName = packageElement.getQualifiedName().toString() + "." + classElement.getSimpleName();
Map<String, String> baseTypes = processingContext.getDefinitionAnnotations().getBaseTypes();
TypeElement parentElement = getDefinitionInheritedType(classElement);
if (null != parentElement && !baseTypes.containsKey(defintionClassName)) {
PackageElement basePackageElement = (PackageElement) parentElement.getEnclosingElement();
String baseClassName = basePackageElement.getQualifiedName().toString() + "." + parentElement.getSimpleName();
baseTypes.put(defintionClassName, baseClassName);
}
// Id field.
processFieldName(classElement, defintionClassName, ANNOTATION_DEFINITION_ID, processingContext.getDefinitionAnnotations().getIdFieldNames(), false);
// Category field.
processFieldName(classElement, defintionClassName, ANNOTATION_DEFINITION_CATEGORY, processingContext.getDefinitionAnnotations().getCategoryFieldNames(), true);
// Title field.
processFieldName(classElement, defintionClassName, ANNOTATION_DEFINITION_TITLE, processingContext.getDefinitionAnnotations().getTitleFieldNames(), false);
// Description field.
processFieldName(classElement, defintionClassName, ANNOTATION_DESCRIPTION, processingContext.getDefinitionAnnotations().getDescriptionFieldNames(), false);
// Labels field.
processFieldName(classElement, defintionClassName, ANNOTATION_DEFINITION_LABELS, processingContext.getDefinitionAnnotations().getLabelsFieldNames(), true);
// Builder class.
processDefinitionModelBuilder(e, defintionClassName, processingContext.getDefinitionAnnotations().getBuilderFieldNames());
// Graph element.
Definition definitionAnn = e.getAnnotation(Definition.class);
TypeMirror mirror = null;
try {
Class<?> graphClass = definitionAnn.graphFactory();
} catch (MirroredTypeException mte) {
mirror = mte.getTypeMirror();
}
if (null == mirror) {
throw new RuntimeException("No graph factory class specified for the @Definition.");
}
String fqcn = mirror.toString();
processingContext.getDefinitionAnnotations().getGraphFactoryFieldNames().put(defintionClassName, fqcn);
// PropertySets fields.
Map<String, Element> propertySetElements = getFieldNames(classElement, ANNOTATION_PROPERTY_SET);
if (null != propertySetElements && !propertySetElements.isEmpty()) {
processingContext.getPropertySetElements().addAll(propertySetElements.values());
processingContext.getDefinitionAnnotations().getPropertySetFieldNames().put(defintionClassName, new LinkedHashSet<>(propertySetElements.keySet()));
} else {
note("Definition for tye [" + defintionClassName + "] have no Property Set members.");
}
// Properties fields.
Map<String, Element> propertyElements = getFieldNames(classElement, ANNOTATION_PROPERTY);
if (null != propertyElements && !propertyElements.isEmpty()) {
processingContext.getPropertyElements().addAll(propertyElements.values());
processingContext.getDefinitionAnnotations().getPropertyFieldNames().put(defintionClassName, new LinkedHashSet<>(propertyElements.keySet()));
} else {
note("Definition for tye [" + defintionClassName + "] have no Property members.");
}
// -- Morphing annotations --
MorphBase morphBaseAnn = e.getAnnotation(MorphBase.class);
Morph morphAnn = e.getAnnotation(Morph.class);
if (null != morphBaseAnn && null != morphAnn) {
TypeElement superElement = getAnnotationInTypeInheritance(classElement, MorphBase.class.getName());
final String packageName = packageElement.getQualifiedName().toString();
String morphBaseClassName = packageName + "." + superElement.getSimpleName().toString();
Map<String, String> defaultTypesMap = processingContext.getMorphingAnnotations().getBaseDefaultTypes();
if (null == defaultTypesMap.get(morphBaseClassName)) {
TypeMirror morphDefaultTypeMirror = null;
try {
Class<?> defaultTypeClass = morphBaseAnn.defaultType();
} catch (MirroredTypeException mte) {
morphDefaultTypeMirror = mte.getTypeMirror();
}
if (null == morphDefaultTypeMirror) {
throw new RuntimeException("No default type class specifyed for the @MorphBase.");
}
String morphDefaultTypeClassName = morphDefaultTypeMirror.toString();
processingContext.getMorphingAnnotations().getBaseDefaultTypes().put(morphBaseClassName, morphDefaultTypeClassName);
// MorphBase - targets
List<? extends TypeMirror> morphTargetMirrors = null;
try {
Class<?>[] defsClasses = morphBaseAnn.targets();
} catch (MirroredTypesException mte) {
morphTargetMirrors = mte.getTypeMirrors();
}
if (null != morphTargetMirrors) {
Set<String> morphTargetMirrorClasses = new LinkedHashSet<>();
for (TypeMirror morphTargetMirror : morphTargetMirrors) {
String morphTargetMirrorClassName = morphTargetMirror.toString();
morphTargetMirrorClasses.add(morphTargetMirrorClassName);
}
processingContext.getMorphingAnnotations().getBaseTargets().put(morphBaseClassName, morphTargetMirrorClasses);
}
// Morph Properties.
processMorphProperties(superElement, morphBaseClassName);
}
TypeMirror morphBaseTypeMirror = null;
try {
Class<?> defaultTypeClass = morphAnn.base();
} catch (MirroredTypeException mte) {
morphBaseTypeMirror = mte.getTypeMirror();
}
if (null == morphBaseTypeMirror) {
throw new RuntimeException("No base type class specifyed for the @MorphBase.");
}
String morphBaseTypeClassName = morphBaseTypeMirror.toString();
Set<String> currentTargets = processingContext.getMorphingAnnotations().getBaseTargets().get(morphBaseTypeClassName);
if (null == currentTargets) {
currentTargets = new LinkedHashSet<>();
processingContext.getMorphingAnnotations().getBaseTargets().put(morphBaseTypeClassName, currentTargets);
}
currentTargets.add(defintionClassName);
}
}
return false;
}
use of org.kie.workbench.common.stunner.core.definition.annotation.Definition in project kie-wb-common by kiegroup.
the class MainProcessor method processDefinitionModelBuilder.
private void processDefinitionModelBuilder(final Element e, final String className, final Map<String, String> processingContextMap) {
Definition definitionAnn = e.getAnnotation(Definition.class);
TypeMirror bMirror = null;
try {
Class<?> builderClass = definitionAnn.builder();
} catch (MirroredTypeException mte) {
bMirror = mte.getTypeMirror();
}
if (null != bMirror && !VoidBuilder.class.getName().equals(bMirror.toString())) {
String fqcn = bMirror.toString();
processingContextMap.put(className, fqcn);
}
}
Aggregations