648b79826930d71ece0043a5b8d09709bfe3dbe0
[yangtools.git] / plugin / yang-maven-plugin / src / main / java / org / opendaylight / yangtools / yang2sources / plugin / GeneratorTaskFactory.java
1 /*
2  * Copyright (c) 2020 PANTHEON.tech, s.r.o. 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.yang2sources.plugin;
9
10 import com.google.common.base.MoreObjects;
11 import com.google.common.base.MoreObjects.ToStringHelper;
12 import org.apache.maven.project.MavenProject;
13 import org.eclipse.jdt.annotation.NonNullByDefault;
14 import org.opendaylight.yangtools.plugin.generator.api.FileGenerator.ImportResolutionMode;
15 import org.opendaylight.yangtools.yang.parser.api.YangParserConfiguration;
16
17 @NonNullByDefault
18 abstract class GeneratorTaskFactory extends ParserConfigAware {
19     private static final YangParserConfiguration SEMVER_CONFIG = YangParserConfiguration.builder()
20         .importResolutionMode(org.opendaylight.yangtools.yang.parser.api.ImportResolutionMode.OPENCONFIG_SEMVER)
21         .build();
22
23     private final YangParserConfiguration parserConfig;
24
25     GeneratorTaskFactory(final ImportResolutionMode importMode) {
26         switch (importMode) {
27             case REVISION_EXACT_OR_LATEST:
28                 parserConfig = YangParserConfiguration.DEFAULT;
29                 break;
30             case SEMVER_LATEST:
31                 parserConfig = SEMVER_CONFIG;
32                 break;
33             default:
34                 throw new LinkageError("Unhandled import mode " + importMode);
35         }
36     }
37
38     @Override
39     final YangParserConfiguration parserConfig() {
40         return parserConfig;
41     }
42
43     final String generatorName() {
44         return generator().getClass().getName();
45     }
46
47     /**
48      * Create a new {@link GeneratorTask} which will work in scope of specified {@link MavenProject} with the effective
49      * model held in specified {@link ContextHolder}.
50      *
51      * @param project current Maven Project
52      * @param context model generation context
53      */
54     abstract GeneratorTask<?> createTask(MavenProject project, ContextHolder context);
55
56     abstract Object generator();
57
58     ToStringHelper addToStringProperties(final ToStringHelper helper) {
59         return helper.add("generator", generatorName());
60     }
61
62     @Override
63     public final String toString() {
64         return addToStringProperties(MoreObjects.toStringHelper(this).omitNullValues()).toString();
65     }
66 }