Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.raml.ramltopojo.GenerationContextImpl Maven / Gradle / Ivy
package org.raml.ramltopojo;
import com.google.common.base.Optional;
import com.google.common.collect.HashMultimap;
import com.google.common.collect.SetMultimap;
import com.squareup.javapoet.ClassName;
import com.squareup.javapoet.JavaFile;
import com.squareup.javapoet.TypeName;
import com.squareup.javapoet.TypeSpec;
import org.raml.ramltopojo.extensions.*;
import org.raml.ramltopojo.plugin.PluginManager;
import org.raml.v2.api.model.v10.api.Api;
import org.raml.v2.api.model.v10.datamodel.TypeDeclaration;
import java.io.IOException;
import java.nio.file.Paths;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
/**
* Created. There, you have it.
*/
public class GenerationContextImpl implements GenerationContext {
private final PluginManager pluginManager;
private final Api api;
private final TypeFetcher typeFetcher;
private final ConcurrentHashMap knownTypes = new ConcurrentHashMap<>();
private final SetMultimap childTypes = HashMultimap.create();
private final String defaultPackage;
private final List basePlugins;
private Map supportClasses = new HashMap<>();
public GenerationContextImpl(Api api) {
this(PluginManager.NULL, api, TypeFetchers.NULL_FETCHER, "", Collections.emptyList());
}
public GenerationContextImpl(PluginManager pluginManager, Api api, TypeFetcher typeFetcher, String defaultPackage, List basePlugins) {
this.pluginManager = pluginManager;
this.api = api;
this.typeFetcher = typeFetcher;
this.defaultPackage = defaultPackage;
this.basePlugins = basePlugins;
}
@Override
public CreationResult findCreatedType(String typeName, TypeDeclaration ramlType) {
if ( knownTypes.containsKey(typeName) ) {
return knownTypes.get(typeName);
} else {
TypeDeclaration typeDeclaration = typeFetcher.fetchType(api, typeName);
Optional result = TypeDeclarationType.createType(typeDeclaration, this);
// todo fix this.
if ( result.isPresent() ) {
knownTypes.put(typeName, result.get());
return result.get();
} else {
return null;
}
}
}
@Override
public String defaultPackage() {
return defaultPackage;
}
@Override
public Set childClasses(String ramlTypeName) {
return childTypes.get(ramlTypeName);
}
@Override
public ClassName buildDefaultClassName(String name, EventType eventType) {
return ClassName.get(defaultPackage, name);
}
public void setupTypeHierarchy(TypeDeclaration typeDeclaration) {
List parents = typeDeclaration.parentTypes();
for (TypeDeclaration parent : parents) {
setupTypeHierarchy(parent);
childTypes.put(parent.name(), typeDeclaration.name());
}
}
@Override
public void newExpectedType(String name, CreationResult creationResult) {
knownTypes.put(name, creationResult);
}
@Override
public void createTypes(String rootDirectory) throws IOException {
for (CreationResult creationResult : knownTypes.values()) {
creationResult.createType(rootDirectory);
}
}
@Override
public void createSupportTypes(String rootDirectory) throws IOException {
for (TypeSpec typeSpec : supportClasses.values()) {
JavaFile.builder(defaultPackage(), typeSpec).build().writeTo(Paths.get(rootDirectory));
}
}
private void loadBasePlugins(Set plugins, Class pluginType, TypeDeclaration... typeDeclarations) {
for (TypeDeclaration typeDeclaration : typeDeclarations) {
if ( Annotations.CLASS_NAME.get(typeDeclaration) != null) {
pluginManager.getClassesForName("core.rename", Collections.singletonList(Annotations.CLASS_NAME.get(typeDeclaration)), pluginType);
}
if ( Annotations.IMPLEMENTATION_CLASS_NAME.get(typeDeclaration) != null) {
pluginManager.getClassesForName("core.renameImplementation", Collections.singletonList(Annotations.IMPLEMENTATION_CLASS_NAME.get(typeDeclaration)), pluginType);
}
if (!Annotations.USE_PRIMITIVE.get(typeDeclaration)) {
pluginManager.getClassesForName("core.box", Collections.emptyList(), pluginType);
}
if (!Annotations.ABSTRACT.get(typeDeclaration)) {
pluginManager.getClassesForName("core.makeAbstract", Collections.emptyList(), pluginType);
}
}
for (String basePlugin : basePlugins) {
plugins.addAll(pluginManager.getClassesForName(basePlugin, Collections.emptyList(), pluginType));
}
}
@Override
public TypeName createSupportClass(TypeSpec.Builder newSupportType) {
TypeSpec typeSpec = newSupportType.build();
if ( supportClasses.containsKey(typeSpec.name) ) {
TypeSpec builder = supportClasses.get(typeSpec.name);
return ClassName.get(this.defaultPackage, builder.name);
} else {
this.supportClasses.put(typeSpec.name, typeSpec);
return ClassName.get(this.defaultPackage, typeSpec.name);
}
}
@Override
public ObjectTypeHandlerPlugin pluginsForObjects(TypeDeclaration... typeDeclarations) {
List data = Annotations.PLUGINS.get(Collections.emptyList(), api, typeDeclarations);
//System.err.println("annotation defined plugins for " + typeDeclarations[0].name() + "are " + data);
Set plugins = new HashSet<>();
loadBasePlugins(plugins, ObjectTypeHandlerPlugin.class);
for (PluginDef datum : data) {
plugins.addAll(pluginManager.getClassesForName(datum.getPluginName(), datum.getArguments() , ObjectTypeHandlerPlugin.class));
}
//System.err.println("plugin definitions for object type " + plugins + " for " + typeDeclarations[0].name());
return new ObjectTypeHandlerPlugin.Composite(plugins);
}
@Override
public EnumerationTypeHandlerPlugin pluginsForEnumerations(TypeDeclaration... typeDeclarations) {
List data = Annotations.PLUGINS.get(Collections.emptyList(), api, typeDeclarations);
Set plugins = new HashSet<>();
loadBasePlugins(plugins, EnumerationTypeHandlerPlugin.class);
for (PluginDef datum : data) {
plugins.addAll(pluginManager.getClassesForName(datum.getPluginName(), datum.getArguments() , EnumerationTypeHandlerPlugin.class));
}
return new EnumerationTypeHandlerPlugin.Composite(plugins);
}
@Override
public ArrayTypeHandlerPlugin pluginsForArrays(TypeDeclaration... typeDeclarations) {
List data = Annotations.PLUGINS.get(Collections.emptyList(), api, typeDeclarations);
Set plugins = new HashSet<>();
loadBasePlugins(plugins, ArrayTypeHandlerPlugin.class);
for (PluginDef datum : data) {
plugins.addAll(pluginManager.getClassesForName(datum.getPluginName(), datum.getArguments() , ArrayTypeHandlerPlugin.class));
}
return new ArrayTypeHandlerPlugin.Composite(plugins);
}
@Override
public UnionTypeHandlerPlugin pluginsForUnions(TypeDeclaration... typeDeclarations) {
List data = Annotations.PLUGINS.get(Collections.emptyList(), api, typeDeclarations);
Set plugins = new HashSet<>();
loadBasePlugins(plugins, UnionTypeHandlerPlugin.class);
for (PluginDef datum : data) {
plugins.addAll(pluginManager.getClassesForName(datum.getPluginName(), datum.getArguments() , UnionTypeHandlerPlugin.class));
}
return new UnionTypeHandlerPlugin.Composite(plugins);
}
@Override
public ReferenceTypeHandlerPlugin pluginsForReferences(TypeDeclaration... typeDeclarations) {
List data = Annotations.PLUGINS.get(Collections.emptyList(), api, typeDeclarations);
Set plugins = new HashSet<>();
loadBasePlugins(plugins, ReferenceTypeHandlerPlugin.class);
for (PluginDef datum : data) {
plugins.addAll(pluginManager.getClassesForName(datum.getPluginName(), datum.getArguments() , ReferenceTypeHandlerPlugin.class));
}
return new ReferenceTypeHandlerPlugin.Composite(plugins);
}
@Override
public Api api() {
return api;
}
}
