-/*\r
- * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.\r
- *\r
- * This program and the accompanying materials are made available under the\r
- * terms of the Eclipse Public License v1.0 which accompanies this distribution,\r
- * and is available at http://www.eclipse.org/legal/epl-v10.html\r
- */\r
-package org.opendaylight.controller.sal.java.api.generator;\r
-\r
-import java.io.BufferedWriter;\r
-import java.io.File;\r
-import java.io.FileWriter;\r
-import java.io.IOException;\r
-import java.io.Writer;\r
-import java.util.ArrayList;\r
-import java.util.List;\r
-import java.util.Set;\r
-\r
-import org.opendaylight.controller.sal.binding.model.api.CodeGenerator;\r
-import org.opendaylight.controller.sal.binding.model.api.GeneratedType;\r
-\r
-public class GeneratorJavaFile {\r
-\r
- private final CodeGenerator codeGenerator;\r
- private final Set<GeneratedType> types;\r
-\r
- public GeneratorJavaFile(CodeGenerator codeGenerator,\r
- Set<GeneratedType> types) {\r
- this.codeGenerator = codeGenerator;\r
- this.types = types;\r
- }\r
-\r
- public boolean generateToFile() {\r
- return generateToFile(null);\r
- }\r
-\r
- public boolean generateToFile(String path) {\r
- try {\r
- for (GeneratedType type : types) {\r
- String parentPath = generateParentPath(path,\r
- type.getPackageName());\r
-\r
- File file = new File(parentPath, type.getName() + ".java");\r
- File parent = file.getParentFile();\r
- if (!parent.exists()) {\r
- parent.mkdirs();\r
- }\r
-\r
- if (!file.exists()) {\r
- FileWriter fw = null;\r
- BufferedWriter bw = null;\r
-\r
- file.createNewFile();\r
- fw = new FileWriter(file);\r
- bw = new BufferedWriter(fw);\r
- Writer writer = codeGenerator.generate(type);\r
- bw.write(writer.toString());\r
-\r
- if (bw != null) {\r
- try {\r
- bw.close();\r
- } catch (IOException e) {\r
- // TODO: log?\r
- }\r
- }\r
- }\r
- }\r
- return true;\r
- } catch (IOException e) {\r
- // TODO: log?\r
- return false;\r
- }\r
- }\r
-\r
- private String generateParentPath(String path, String pkg) {\r
- List<String> dirs = new ArrayList<String>();\r
- String pkgPath = "";\r
- if (pkg != null) {\r
- if (pkg.length() > 0) {\r
- if (pkg.contains(".")) {\r
- String[] split = pkg.split("\\.");\r
- for (String dir : split) {\r
- dirs.add(dir);\r
- }\r
- } else {\r
- dirs.add(pkg);\r
- }\r
- for (int i = 0; i < dirs.size(); i++) {\r
- if (i == 0) {\r
- pkgPath += dirs.get(i);\r
- } else {\r
- pkgPath += File.separator + dirs.get(i);\r
- }\r
- }\r
- }\r
- }\r
- String fullPath = "";\r
- if (path != null) {\r
- if (path.endsWith(File.separator)) {\r
- fullPath = path + pkgPath;\r
- } else {\r
- fullPath = path + File.separator + pkgPath;\r
- }\r
- } else {\r
- fullPath = pkgPath;\r
- }\r
- return fullPath;\r
- }\r
-\r
-}\r
+/*
+ * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.controller.sal.java.api.generator;
+
+import java.io.*;
+import java.util.ArrayList;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
+import org.opendaylight.controller.sal.binding.model.api.*;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public final class GeneratorJavaFile {
+
+ private static final Logger log = LoggerFactory.getLogger(GeneratorJavaFile.class);
+ private final CodeGenerator interfaceGenerator;
+ private final ClassCodeGenerator classGenerator;
+ private final EnumGenerator enumGenerator;
+
+ private final Set<GeneratedType> genTypes;
+ private final Set<GeneratedTransferObject> genTransferObjects;
+ private final Set<Enumeration> enumerations;
+
+ public GeneratorJavaFile(final CodeGenerator codeGenerator, final Set<GeneratedType> types) {
+ this.interfaceGenerator = codeGenerator;
+ this.genTypes = types;
+ this.genTransferObjects = new HashSet<>();
+ this.enumerations = new HashSet<>();
+ this.classGenerator = new ClassCodeGenerator();
+ this.enumGenerator = new EnumGenerator();
+ }
+
+ public GeneratorJavaFile(final Set<GeneratedType> types, final Set<GeneratedTransferObject> genTransferObjects,
+ final Set<Enumeration> enumerations) {
+ this.interfaceGenerator = new InterfaceGenerator();
+ this.classGenerator = new ClassCodeGenerator();
+ this.enumGenerator = new EnumGenerator();
+
+ this.genTypes = types;
+ this.genTransferObjects = genTransferObjects;
+ this.enumerations = enumerations;
+ }
+
+ public List<File> generateToFile(final File parentDirectory) throws IOException {
+ final List<File> result = new ArrayList<>();
+ for (GeneratedType type : genTypes) {
+ final File genFile = generateTypeToJavaFile(parentDirectory, type, interfaceGenerator);
+
+ if (genFile != null) {
+ result.add(genFile);
+ }
+ }
+ for (GeneratedTransferObject transferObject : genTransferObjects) {
+ final File genFile = generateTypeToJavaFile(parentDirectory, transferObject, classGenerator);
+
+ if (genFile != null) {
+ result.add(genFile);
+ }
+ }
+
+ for (Enumeration enumeration : enumerations) {
+ final File genFile = generateTypeToJavaFile(parentDirectory, enumeration, enumGenerator);
+
+ if (genFile != null) {
+ result.add(genFile);
+ }
+ }
+
+ return result;
+ }
+
+ private File generateTypeToJavaFile(final File parentDir, final Type type, final CodeGenerator generator)
+ throws IOException {
+ if (parentDir == null) {
+ log.warn("Parent Directory not specified, files will be generated "
+ + "accordingly to generated Type package path.");
+ }
+ if (type == null) {
+ log.error("Cannot generate Type into Java File because " + "Generated Type is NULL!");
+ throw new IllegalArgumentException("Generated Type Cannot be NULL!");
+ }
+ if (generator == null) {
+ log.error("Cannot generate Type into Java File because " + "Code Generator instance is NULL!");
+ throw new IllegalArgumentException("Code Generator Cannot be NULL!");
+ }
+ final File packageDir = packageToDirectory(parentDir, type.getPackageName());
+
+ if (!packageDir.exists()) {
+ packageDir.mkdirs();
+ }
+ final File file = new File(packageDir, type.getName() + ".java");
+ try (final FileWriter fw = new FileWriter(file)) {
+ file.createNewFile();
+
+ try (final BufferedWriter bw = new BufferedWriter(fw)) {
+ Writer writer = generator.generate(type);
+ bw.write(writer.toString());
+ }
+ } catch (IOException e) {
+ log.error(e.getMessage());
+ throw new IOException(e.getMessage());
+ }
+ return file;
+ }
+
+ private File packageToDirectory(final File parentDirectory, final String packageName) {
+ if (packageName == null) {
+ throw new IllegalArgumentException("Package Name cannot be NULL!");
+ }
+
+ final String[] subDirNames = packageName.split("\\.");
+ final StringBuilder dirPathBuilder = new StringBuilder();
+ dirPathBuilder.append(subDirNames[0]);
+ for (int i = 1; i < subDirNames.length; ++i) {
+ dirPathBuilder.append(File.separator);
+ dirPathBuilder.append(subDirNames[i]);
+ }
+ return new File(parentDirectory, dirPathBuilder.toString());
+ }
+}
Code Review / controller.git / blobdiff