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.yangtools.maven.sal.api.gen.plugin;
11 import java.io.IOException;
12 import java.util.Collection;
13 import java.util.HashSet;
14 import java.util.List;
18 import org.apache.maven.plugin.logging.Log;
19 import org.apache.maven.project.MavenProject;
20 import org.opendaylight.yangtools.sal.binding.generator.api.BindingGenerator;
21 import org.opendaylight.yangtools.sal.binding.generator.impl.BindingGeneratorImpl;
22 import org.opendaylight.yangtools.sal.binding.model.api.Type;
23 import org.opendaylight.yangtools.sal.java.api.generator.GeneratorJavaFile;
24 import org.opendaylight.yangtools.yang.model.api.Module;
25 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
26 import org.opendaylight.yangtools.yang2sources.spi.BuildContextAware;
27 import org.opendaylight.yangtools.yang2sources.spi.CodeGenerator;
28 import org.sonatype.plexus.build.incremental.BuildContext;
30 import com.google.common.base.Preconditions;
32 public final class CodeGeneratorImpl implements CodeGenerator, BuildContextAware {
33 private static final String FS = File.separator;
34 private BuildContext buildContext;
35 private File projectBaseDir;
36 private Map<String, String> additionalConfig;
39 public Collection<File> generateSources(final SchemaContext context, final File outputDir,
40 final Set<Module> yangModules) throws IOException {
41 final File outputBaseDir;
42 if (outputDir == null) {
43 outputBaseDir = new File("target" + File.separator + "generated-sources" + File.separator
44 + "maven-sal-api-gen");
46 outputBaseDir = outputDir;
49 final BindingGenerator bindingGenerator = new BindingGeneratorImpl();
50 final List<Type> types = bindingGenerator.generateTypes(context, yangModules);
51 final GeneratorJavaFile generator = new GeneratorJavaFile(buildContext, new HashSet<>(types));
53 File persistentSourcesDir = null;
54 if (additionalConfig != null) {
55 String persistenSourcesPath = additionalConfig.get("persistentSourcesDir");
56 if (persistenSourcesPath != null) {
57 persistentSourcesDir = new File(persistenSourcesPath);
60 if (persistentSourcesDir == null) {
61 persistentSourcesDir = new File(projectBaseDir, "src" + FS + "main" + FS + "java");
63 return generator.generateToFile(outputBaseDir, persistentSourcesDir);
67 public void setLog(Log log) {
68 // use maven logging if necessary
73 public void setAdditionalConfig(Map<String, String> additionalConfiguration) {
74 this.additionalConfig = additionalConfiguration;
78 public void setResourceBaseDir(File resourceBaseDir) {
79 // no resource processing necessary
83 public void setMavenProject(MavenProject project) {
84 this.projectBaseDir = project.getBasedir();
88 public void setBuildContext(BuildContext buildContext) {
89 this.buildContext = Preconditions.checkNotNull(buildContext);