opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/GeneratorJavaFile.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.controller.sal.java.api.generator;
   9
  10 import java.io.*;
  11 import java.util.ArrayList;
  12 import java.util.HashSet;
  13 import java.util.List;
  14 import java.util.Set;
  15
  16 import org.opendaylight.controller.sal.binding.model.api.*;
  17 import org.slf4j.Logger;
  18 import org.slf4j.LoggerFactory;
  19
  20 public final class GeneratorJavaFile {
  21
  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;
  26
  27     private final Set<GeneratedType> genTypes;
  28     private final Set<GeneratedTransferObject> genTransferObjects;
  29     private final Set<Enumeration> enumerations;
  30
  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();
  38     }
  39
  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();
  45
  46         this.genTypes = types;
  47         this.genTransferObjects = genTransferObjects;
  48         this.enumerations = enumerations;
  49     }
  50
  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);
  55
  56             if (genFile != null) {
  57                 result.add(genFile);
  58             }
  59         }
  60         for (GeneratedTransferObject transferObject : genTransferObjects) {
  61             final File genFile = generateTypeToJavaFile(parentDirectory, transferObject, classGenerator);
  62
  63             if (genFile != null) {
  64                 result.add(genFile);
  65             }
  66         }
  67
  68         for (Enumeration enumeration : enumerations) {
  69             final File genFile = generateTypeToJavaFile(parentDirectory, enumeration, enumGenerator);
  70
  71             if (genFile != null) {
  72                 result.add(genFile);
  73             }
  74         }
  75
  76         return result;
  77     }
  78
  79     private File generateTypeToJavaFile(final File parentDir, final Type type, final CodeGenerator generator)
  80             throws IOException {
  81         if (parentDir == null) {
  82             log.warn("Parent Directory not specified, files will be generated "
  83                     + "accordingly to generated Type package path.");
  84         }
  85         if (type == null) {
  86             log.error("Cannot generate Type into Java File because " + "Generated Type is NULL!");
  87             throw new IllegalArgumentException("Generated Type Cannot be NULL!");
  88         }
  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!");
  92         }
  93         final File packageDir = packageToDirectory(parentDir, type.getPackageName());
  94
  95         if (!packageDir.exists()) {
  96             packageDir.mkdirs();
  97         }
  98         final File file = new File(packageDir, type.getName() + ".java");
  99         try (final FileWriter fw = new FileWriter(file)) {
 100             file.createNewFile();
 101
 102             try (final BufferedWriter bw = new BufferedWriter(fw)) {
 103                 Writer writer = generator.generate(type);
 104                 bw.write(writer.toString());
 105             }
 106         } catch (IOException e) {
 107             log.error(e.getMessage());
 108             throw new IOException(e.getMessage());
 109         }
 110         return file;
 111     }
 112
 113     private File packageToDirectory(final File parentDirectory, final String packageName) {
 114         if (packageName == null) {
 115             throw new IllegalArgumentException("Package Name cannot be NULL!");
 116         }
 117
 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]);
 124         }
 125         return new File(parentDirectory, dirPathBuilder.toString());
 126     }
 127 }