Generation of java files from templates
[controller.git] / opendaylight / sal / yang-prototype / code-generator / binding-java-api-generator / src / main / java / org / opendaylight / controller / sal / java / api / generator / InterfaceGenerator.java
index 27570ca77d7ede88b8c72cb110dfdcbfced1a634..381b1575781d2991f4da11280a507475070b4c42 100644 (file)
  */
 package org.opendaylight.controller.sal.java.api.generator;
 
-import static org.opendaylight.controller.sal.java.api.generator.Constants.*;
-
 import java.io.IOException;
 import java.io.StringWriter;
 import java.io.Writer;
-import java.util.List;
-import java.util.Map;
 
-import org.opendaylight.controller.sal.binding.model.api.*;
+import org.opendaylight.controller.sal.binding.model.api.CodeGenerator;
+import org.opendaylight.controller.sal.binding.model.api.GeneratedTransferObject;
+import org.opendaylight.controller.sal.binding.model.api.GeneratedType;
+import org.opendaylight.controller.sal.binding.model.api.Type;
 
 public final class InterfaceGenerator implements CodeGenerator {
 
-    private Map<String, String> imports;
-    private Map<String, String> innerTypeImports;
-
-    private String generateEnums(List<Enumeration> enums) {
-        String result = "";
-        if (enums != null) {
-            EnumGenerator enumGenerator = new EnumGenerator();
-            for (Enumeration en : enums) {
-                try {
-                    result = result + (enumGenerator.generateInnerEnumeration(en, TAB).toString() + NL);
-                } catch (IOException e) {
-                    e.printStackTrace();
-                }
-            }
-        }
-        return result;
-    }
-
-    private String generateConstants(List<Constant> constants, String pkgName) {
-        String result = "";
-        if (constants != null) {
-            for (Constant c : constants) {
-                result = result + GeneratorUtil.createConstant(c, TAB, imports, pkgName) + NL;
-            }
-            result.concat(NL);
-        }
-        return result;
-    }
-
-    public String generateMethods(List<MethodSignature> methods, String pkgName) {
-        String result = "";
-
-        if (methods != null) {
-            for (MethodSignature m : methods) {
-                result = result + GeneratorUtil.createMethodDeclaration(m, TAB, imports, pkgName) + NL;
-            }
-            result = result + NL;
-        }
-        return result;
-    }
-
-    public String generateInnerClasses(final List<GeneratedType> generatedTypes) throws IOException {
-        String result = "";
-
-        if (generatedTypes != null) {
-            ClassCodeGenerator classCodeGenerator = new ClassCodeGenerator();
-            for (GeneratedType genType : generatedTypes) {
-                if (genType instanceof GeneratedTransferObject) {
-                    result = result + classCodeGenerator.generateOnlyClass(genType, imports).toString();
-                    result = result + NL + NL;
-                }
-            }
-        }
-
-        return result;
-    }
-
+    @Override
     public Writer generate(Type type) throws IOException {
-        Writer writer = new StringWriter();
+        final Writer writer = new StringWriter();
         if (type instanceof GeneratedType && !(type instanceof GeneratedTransferObject)) {
             final GeneratedType genType = (GeneratedType) type;
-            imports = GeneratorUtil.createImports(genType);
-            innerTypeImports = GeneratorUtil.createChildImports(genType);
-
-            final String currentPkg = genType.getPackageName();
-            final List<Constant> constants = genType.getConstantDefinitions();
-            final List<MethodSignature> methods = genType.getMethodDefinitions();
-            final List<Enumeration> enums = genType.getEnumerations();
-            final List<GeneratedType> enclosedGeneratedTypes = genType.getEnclosedTypes();
-
-            writer.write(GeneratorUtil.createPackageDeclaration(genType.getPackageName()));
-            writer.write(NL);
-
-            List<String> importLines = GeneratorUtil.createImportLines(imports, innerTypeImports);
-
-            for (String line : importLines) {
-                writer.write(line + NL);
-            }
-            writer.write(NL);
-            writer.write(GeneratorUtil.createIfcDeclaration(genType, "", imports));
-            writer.write(NL);
-
-            writer.write(generateInnerClasses(enclosedGeneratedTypes));
-            writer.write(generateEnums(enums));
-            writer.write(generateConstants(constants, currentPkg));
-            writer.write(generateMethods(methods, currentPkg));
-
-            writer.write(RCB);
+            final InterfaceTemplate interfaceTemplate = new InterfaceTemplate(genType);
+            writer.write(interfaceTemplate.generate().toString());
         }
         return writer;
     }
+    
 }