code-generator/maven-sal-api-gen-plugin/src/main/java/org/opendaylight/yangtools/maven/sal/api/gen/plugin/CodeGeneratorImpl.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.yangtools.maven.sal.api.gen.plugin;
   9
  10 import java.io.BufferedWriter;
  11 import java.io.File;
  12 import java.io.IOException;
  13 import java.io.OutputStream;
  14 import java.io.OutputStreamWriter;
  15 import java.io.Writer;
  16 import java.util.Collection;
  17 import java.util.HashSet;
  18 import java.util.List;
  19 import java.util.Map;
  20 import java.util.Set;
  21
  22 import org.apache.maven.plugin.logging.Log;
  23 import org.apache.maven.project.MavenProject;
  24 import org.opendaylight.yangtools.binding.generator.util.BindingGeneratorUtil;
  25 import org.opendaylight.yangtools.sal.binding.generator.api.BindingGenerator;
  26 import org.opendaylight.yangtools.sal.binding.generator.impl.BindingGeneratorImpl;
  27 import org.opendaylight.yangtools.sal.binding.model.api.Type;
  28 import org.opendaylight.yangtools.sal.java.api.generator.GeneratorJavaFile;
  29 import org.opendaylight.yangtools.sal.java.api.generator.YangModuleInfoTemplate;
  30 import org.opendaylight.yangtools.yang.model.api.Module;
  31 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  32 import org.opendaylight.yangtools.yang2sources.spi.BuildContextAware;
  33 import org.opendaylight.yangtools.yang2sources.spi.CodeGenerator;
  34 import org.sonatype.plexus.build.incremental.BuildContext;
  35
  36 import com.google.common.base.Preconditions;
  37
  38 public final class CodeGeneratorImpl implements CodeGenerator, BuildContextAware {
  39     private static final String FS = File.separator;
  40     private BuildContext buildContext;
  41     private File projectBaseDir;
  42     private Map<String, String> additionalConfig;
  43
  44     @Override
  45     public Collection<File> generateSources(final SchemaContext context, final File outputDir,
  46             final Set<Module> yangModules) throws IOException {
  47         final File outputBaseDir;
  48         if (outputDir == null) {
  49             outputBaseDir = new File("target" + File.separator + "generated-sources" + File.separator
  50                     + "maven-sal-api-gen");
  51         } else {
  52             outputBaseDir = outputDir;
  53         }
  54
  55         final BindingGenerator bindingGenerator = new BindingGeneratorImpl();
  56         final List<Type> types = bindingGenerator.generateTypes(context, yangModules);
  57         final GeneratorJavaFile generator = new GeneratorJavaFile(buildContext, new HashSet<>(types));
  58
  59         File persistentSourcesDir = null;
  60         if (additionalConfig != null) {
  61             String persistenSourcesPath = additionalConfig.get("persistentSourcesDir");
  62             if (persistenSourcesPath != null) {
  63                 persistentSourcesDir = new File(persistenSourcesPath);
  64             }
  65         }
  66         if (persistentSourcesDir == null) {
  67             persistentSourcesDir = new File(projectBaseDir, "src" + FS + "main" + FS + "java");
  68         }
  69
  70         List<File> result = generator.generateToFile(outputBaseDir, persistentSourcesDir);
  71         for (Module module : yangModules) {
  72             // TODO: add YangModuleInfo class
  73             result.add(generateYangModuleInfo(outputBaseDir, module, context));
  74         }
  75         return result;
  76     }
  77
  78     @Override
  79     public void setLog(Log log) {
  80         // use maven logging if necessary
  81
  82     }
  83
  84     @Override
  85     public void setAdditionalConfig(Map<String, String> additionalConfiguration) {
  86         this.additionalConfig = additionalConfiguration;
  87     }
  88
  89     @Override
  90     public void setResourceBaseDir(File resourceBaseDir) {
  91         // no resource processing necessary
  92     }
  93
  94     @Override
  95     public void setMavenProject(MavenProject project) {
  96         this.projectBaseDir = project.getBasedir();
  97     }
  98
  99     @Override
 100     public void setBuildContext(BuildContext buildContext) {
 101         this.buildContext = Preconditions.checkNotNull(buildContext);
 102     }
 103
 104     private File generateYangModuleInfo(File outputBaseDir, Module module, SchemaContext ctx) {
 105         final YangModuleInfoTemplate template = new YangModuleInfoTemplate(module, ctx);
 106         String generatedCode = template.generate();
 107         if (generatedCode.isEmpty()) {
 108             throw new IllegalStateException("Generated code should not be empty!");
 109         }
 110
 111         final File packageDir = GeneratorJavaFile.packageToDirectory(outputBaseDir, BindingGeneratorUtil.moduleNamespaceToPackageName(module));
 112
 113         final File file = new File(packageDir, "$YangModuleInfoImpl.java");
 114         try (final OutputStream stream = buildContext.newFileOutputStream(file)) {
 115             try (final Writer fw = new OutputStreamWriter(stream)) {
 116                 try (final BufferedWriter bw = new BufferedWriter(fw)) {
 117                     bw.write(generatedCode);
 118                 }
 119             } catch (Exception e) {
 120                 // TODO handle exception
 121             }
 122         } catch (Exception e) {
 123             // TODO handle exception
 124         }
 125         return file;
 126
 127     }
 128
 129 }