2 * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.controller.sal.java.api.generator;
11 import java.util.ArrayList;
12 import java.util.HashSet;
13 import java.util.List;
16 import org.opendaylight.controller.sal.binding.model.api.*;
17 import org.slf4j.Logger;
18 import org.slf4j.LoggerFactory;
20 public final class GeneratorJavaFile {
22 private static final Logger log = LoggerFactory.getLogger(GeneratorJavaFile.class);
23 private final CodeGenerator interfaceGenerator;
24 private final ClassCodeGenerator classGenerator;
25 private final EnumGenerator enumGenerator;
27 private final Set<GeneratedType> genTypes;
28 private final Set<GeneratedTransferObject> genTransferObjects;
29 private final Set<Enumeration> enumerations;
31 public GeneratorJavaFile(final CodeGenerator codeGenerator, final Set<GeneratedType> types) {
32 this.interfaceGenerator = codeGenerator;
33 this.genTypes = types;
34 this.genTransferObjects = new HashSet<>();
35 this.enumerations = new HashSet<>();
36 this.classGenerator = new ClassCodeGenerator();
37 this.enumGenerator = new EnumGenerator();
40 public GeneratorJavaFile(final Set<GeneratedType> types, final Set<GeneratedTransferObject> genTransferObjects,
41 final Set<Enumeration> enumerations) {
42 this.interfaceGenerator = new InterfaceGenerator();
43 this.classGenerator = new ClassCodeGenerator();
44 this.enumGenerator = new EnumGenerator();
46 this.genTypes = types;
47 this.genTransferObjects = genTransferObjects;
48 this.enumerations = enumerations;
51 public List<File> generateToFile(final File parentDirectory) throws IOException {
52 final List<File> result = new ArrayList<>();
53 for (GeneratedType type : genTypes) {
54 final File genFile = generateTypeToJavaFile(parentDirectory, type, interfaceGenerator);
56 if (genFile != null) {
60 for (GeneratedTransferObject transferObject : genTransferObjects) {
61 final File genFile = generateTypeToJavaFile(parentDirectory, transferObject, classGenerator);
63 if (genFile != null) {
68 for (Enumeration enumeration : enumerations) {
69 final File genFile = generateTypeToJavaFile(parentDirectory, enumeration, enumGenerator);
71 if (genFile != null) {
79 private File generateTypeToJavaFile(final File parentDir, final Type type, final CodeGenerator generator)
81 if (parentDir == null) {
82 log.warn("Parent Directory not specified, files will be generated "
83 + "accordingly to generated Type package path.");
86 log.error("Cannot generate Type into Java File because " + "Generated Type is NULL!");
87 throw new IllegalArgumentException("Generated Type Cannot be NULL!");
89 if (generator == null) {
90 log.error("Cannot generate Type into Java File because " + "Code Generator instance is NULL!");
91 throw new IllegalArgumentException("Code Generator Cannot be NULL!");
93 final File packageDir = packageToDirectory(parentDir, type.getPackageName());
95 if (!packageDir.exists()) {
98 final File file = new File(packageDir, type.getName() + ".java");
99 try (final FileWriter fw = new FileWriter(file)) {
100 file.createNewFile();
102 try (final BufferedWriter bw = new BufferedWriter(fw)) {
103 Writer writer = generator.generate(type);
104 bw.write(writer.toString());
106 } catch (IOException e) {
107 log.error(e.getMessage());
108 throw new IOException(e.getMessage());
113 private File packageToDirectory(final File parentDirectory, final String packageName) {
114 if (packageName == null) {
115 throw new IllegalArgumentException("Package Name cannot be NULL!");
118 final String[] subDirNames = packageName.split("\\.");
119 final StringBuilder dirPathBuilder = new StringBuilder();
120 dirPathBuilder.append(subDirNames[0]);
121 for (int i = 1; i < subDirNames.length; ++i) {
122 dirPathBuilder.append(File.separator);
123 dirPathBuilder.append(subDirNames[i]);
125 return new File(parentDirectory, dirPathBuilder.toString());