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;
10 import java.io.BufferedWriter;
12 import java.io.FileWriter;
13 import java.io.IOException;
14 import java.io.Writer;
15 import java.util.ArrayList;
16 import java.util.HashSet;
17 import java.util.List;
20 import org.opendaylight.controller.sal.binding.model.api.CodeGenerator;
21 import org.opendaylight.controller.sal.binding.model.api.GeneratedTransferObject;
22 import org.opendaylight.controller.sal.binding.model.api.GeneratedType;
23 import org.opendaylight.controller.sal.binding.model.api.Type;
24 import org.slf4j.Logger;
25 import org.slf4j.LoggerFactory;
27 public class GeneratorJavaFile {
29 private static final Logger log = LoggerFactory
30 .getLogger(GeneratorJavaFile.class);
31 private final CodeGenerator interfaceGenerator;
32 private final ClassCodeGenerator classGenerator;
33 private final Set<GeneratedType> genTypes;
34 private final Set<GeneratedTransferObject> genTransferObjects;
36 public GeneratorJavaFile(final CodeGenerator codeGenerator,
37 final Set<GeneratedType> types) {
38 this.interfaceGenerator = codeGenerator;
39 this.genTypes = types;
40 this.genTransferObjects = new HashSet<GeneratedTransferObject>();
41 classGenerator = new ClassCodeGenerator();
44 public GeneratorJavaFile(final Set<GeneratedType> types,
45 final Set<GeneratedTransferObject> genTransferObjects) {
46 this.interfaceGenerator = new InterfaceGenerator();
47 this.classGenerator = new ClassCodeGenerator();
48 this.genTypes = types;
49 this.genTransferObjects = genTransferObjects;
53 public List<File> generateToFile(String path) throws IOException {
54 final List<File> result = new ArrayList<File>();
56 for (GeneratedType genType : genTypes) {
57 final String parentPath = generateParentPath(path,
58 genType.getPackageName());
60 final File directory = new File(parentPath);
61 final File genFile = generateTypeToJavaFile(directory, genType,
64 if (genFile != null) {
69 for (GeneratedTransferObject transferObject : genTransferObjects) {
70 final String parentPath = generateParentPath(path,
71 transferObject.getPackageName());
73 final File directory = new File(parentPath);
74 final File genFile = generateTypeToJavaFile(directory,
75 transferObject, classGenerator);
77 if (genFile != null) {
84 public List<File> generateToFile(final File directory) throws IOException {
85 final List<File> result = new ArrayList<File>();
86 for (GeneratedType type : genTypes) {
87 final File genFile = generateTypeToJavaFile(directory, type,
90 if (genFile != null) {
94 for (GeneratedTransferObject transferObject : genTransferObjects) {
95 final File genFile = generateTypeToJavaFile(directory,
96 transferObject, classGenerator);
98 if (genFile != null) {
105 private File generateTypeToJavaFile(final File directory, final Type type,
106 final CodeGenerator generator) throws IOException {
107 if ((directory != null) && (type != null) && (generator != null)) {
109 if (!directory.exists()) {
113 final File file = new File(directory, type.getName() + ".java");
114 try (final FileWriter fw = new FileWriter(file)) {
115 file.createNewFile();
117 try (final BufferedWriter bw = new BufferedWriter(fw)) {
118 Writer writer = generator.generate(type);
119 bw.write(writer.toString());
121 } catch (IOException e) {
122 log.error(e.getMessage());
123 throw new IOException(e.getMessage());
131 private String generateParentPath(String path, String pkg) {
132 List<String> dirs = new ArrayList<String>();
135 if (pkg.length() > 0) {
136 if (pkg.contains(".")) {
137 String[] split = pkg.split("\\.");
138 for (String dir : split) {
144 for (int i = 0; i < dirs.size(); i++) {
146 pkgPath += dirs.get(i);
148 pkgPath += File.separator + dirs.get(i);
153 String fullPath = "";
155 if (path.endsWith(File.separator)) {
156 fullPath = path + pkgPath;
158 fullPath = path + File.separator + pkgPath;