/** * Generates assembler classes if the API returns more than one object. * * @param resources * @param templateGroup */ private void generateModelClassesForInput( List<Resource> resources, StringTemplateGroup templateGroup) { List<String> generatedClasses = new ArrayList<String>(); 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 (method.getInputModel() != null) { Model model = method.getInputModel(); if (model != null) { if (!generatedClasses.contains(model.getName())) { List<String> imports = new ArrayList<String>(); imports.addAll(this.config.getDefaultModelImports()); for (ModelField param : model.getFields()) { param.setName(reservedWordMapper.translate(param.getName())); for (String importDef : param .getFieldDefinition( this.getDataTypeMappingProvider(), config, nameGenerator) .getImportDefinitions()) { if (!imports.contains(importDef)) { imports.add(importDef); } } } StringTemplate template = templateGroup.getInstanceOf(MODEL_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(), "Input model class"); generatedClasses.add(model.getName()); } } } } } } } } }
/** * Creates a wrapper model class that contains all model classes as list of objects. This class is * used for storing test data */ private void generateWrapperClassForTestData( List<String> generatedClassNames, StringTemplateGroup templateGroup) { Model model = new Model(); model.setName("TestData"); model.setDescription( "Class used to store all the test data. This should not be used for any development"); List<ModelField> modelFields = new ArrayList<ModelField>(); model.setFields(modelFields); for (String className : generatedClassNames) { ModelField aParam = new ModelField(); aParam.setName(this.getNameGenerator().applyMethodNamingPolicy(className) + "List"); aParam.setParamType(this.getDataTypeMappingProvider().getListReturnTypeSignature(className)); modelFields.add(aParam); } List<String> imports = new ArrayList<String>(); imports.addAll(this.config.getDefaultModelImports()); imports.addAll(this.getDataTypeMappingProvider().getListIncludes()); for (ModelField param : model.getFields()) { for (String importDef : param .getFieldDefinition(this.getDataTypeMappingProvider(), config, nameGenerator) .getImportDefinitions()) { if (!imports.contains(importDef)) { imports.add(importDef); } } } StringTemplate template = templateGroup.getInstanceOf(MODEL_OBJECT_TEMPLATE); template.setAttribute("fields", model.getFields()); template.setAttribute("imports", imports); template.setAttribute("annotationPackageName", languageConfig.getAnnotationPackageName()); template.setAttribute("extends", config.getDefaultModelBaseClass()); template.setAttribute(PACKAGE_NAME, config.getModelPackageName()); template.setAttribute("className", model.getGenratedClassName()); File aFile = new File( languageConfig.getModelClassLocation() + model.getGenratedClassName() + languageConfig.getClassFileExtension()); writeFile(aFile, template.toString(), "Wrapper class for test data file"); }
/** Generates model classes. If the class is already generated then ignores the same. */ private void generateModelClasses(List<Resource> resources, StringTemplateGroup templateGroup) { List<String> generatedClassNames = new ArrayList(); for (Resource resource : resources) { for (Model model : resource.getModels()) { if (!generatedClassNames.contains(model.getName()) && !this.getCodeGenRulesProvider().isModelIgnored(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); } } } StringTemplate template = templateGroup.getInstanceOf(MODEL_OBJECT_TEMPLATE); template.setAttribute("model", model); template.setAttribute("fields", model.getFields()); template.setAttribute("imports", imports); template.setAttribute("annotationPackageName", languageConfig.getAnnotationPackageName()); template.setAttribute("extends", config.getDefaultModelBaseClass()); 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(), "Model class"); generatedClassNames.add(model.getName()); } } } generateWrapperClassForTestData(generatedClassNames, templateGroup); }
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()); } } } } } } } } }