private void generateOutputWrappers(List<Resource> resources, StringTemplateGroup templateGroup) {
List<String> generatedClasses = new ArrayList<String>();
StringTemplate template = templateGroup.getInstanceOf(WRAPPER_OBJECT_TEMPLATE);
if (template == null) {
System.out.println("WrapperObject template not found to generate output wrappers");
return;
}
for (Resource resource : resources) {
if (resource.getEndPoints() != null) {
for (Endpoint endpoint : resource.getEndPoints()) {
if (endpoint.getOperations() != null) {
for (EndpointOperation operation : endpoint.getOperations()) {
ResourceMethod method =
operation.generateMethod(
endpoint, resource, dataTypeMappingProvider, nameGenerator);
if (codeGenRulesProvider.isModelIgnored(
nameGenerator.applyMethodNamingPolicy(method.getReturnClassName()))) {
continue;
}
if (method.getListWrapperModel() != null) {
Model model = method.getListWrapperModel();
method.setReturnClassName(model.getName());
if (model != null) {
if (!generatedClasses.contains(model.getName())) {
List<String> imports = new ArrayList<String>();
imports.addAll(this.config.getDefaultModelImports());
for (ModelField param : model.getFields()) {
for (String importDef :
param
.getFieldDefinition(
this.getDataTypeMappingProvider(), config, nameGenerator)
.getImportDefinitions()) {
if (!imports.contains(importDef)) {
imports.add(importDef);
}
}
}
template = templateGroup.getInstanceOf(WRAPPER_OBJECT_TEMPLATE);
template.setAttribute("fields", model.getFields());
template.setAttribute("imports", imports);
template.setAttribute("extends", config.getDefaultModelBaseClass());
template.setAttribute(
"annotationPackageName", languageConfig.getAnnotationPackageName());
template.setAttribute("className", model.getGenratedClassName());
template.setAttribute(PACKAGE_NAME, config.getModelPackageName());
File aFile =
new File(
languageConfig.getModelClassLocation()
+ model.getGenratedClassName()
+ languageConfig.getClassFileExtension());
writeFile(aFile, template.toString(), "List wrapper model class");
generatedClasses.add(model.getName());
}
}
}
}
}
}
}
}
}
