Fix NPE when parsing deviation from submodule
[yangtools.git] / parser / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / meta / DeviationStatementSupport.java
1 /*
2  * Copyright (c) 2017 Pantheon Technologies, 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.yang.parser.rfc7950.stmt.meta;
9
10 import static com.google.common.base.Verify.verifyNotNull;
11
12 import com.google.common.collect.ImmutableList;
13 import com.google.common.collect.Iterables;
14 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
15 import org.opendaylight.yangtools.yang.model.api.meta.DeclarationReference;
16 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
17 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
18 import org.opendaylight.yangtools.yang.model.api.stmt.DeviationEffectiveStatement;
19 import org.opendaylight.yangtools.yang.model.api.stmt.DeviationStatement;
20 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier.Absolute;
21 import org.opendaylight.yangtools.yang.model.api.stmt.SubmoduleStatement;
22 import org.opendaylight.yangtools.yang.model.ri.stmt.DeclaredStatementDecorators;
23 import org.opendaylight.yangtools.yang.model.ri.stmt.DeclaredStatements;
24 import org.opendaylight.yangtools.yang.model.ri.stmt.EffectiveStatements;
25 import org.opendaylight.yangtools.yang.parser.api.YangParserConfiguration;
26 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.ArgumentUtils;
27 import org.opendaylight.yangtools.yang.parser.spi.ParserNamespaces;
28 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
29 import org.opendaylight.yangtools.yang.parser.spi.meta.BoundStmtCtx;
30 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
31 import org.opendaylight.yangtools.yang.parser.spi.meta.InferenceException;
32 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
33 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
34 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
35
36 public final class DeviationStatementSupport
37         extends AbstractStatementSupport<Absolute, DeviationStatement, DeviationEffectiveStatement> {
38     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR =
39         SubstatementValidator.builder(YangStmtMapping.DEVIATION)
40             .addOptional(YangStmtMapping.DESCRIPTION)
41             .addAny(YangStmtMapping.DEVIATE)
42             .addOptional(YangStmtMapping.REFERENCE)
43             .build();
44
45     public DeviationStatementSupport(final YangParserConfiguration config) {
46         super(YangStmtMapping.DEVIATION, StatementPolicy.reject(), config, SUBSTATEMENT_VALIDATOR);
47     }
48
49     @Override
50     public Absolute parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
51         return ArgumentUtils.parseAbsoluteSchemaNodeIdentifier(ctx, value);
52     }
53
54     @Override
55     public void onFullDefinitionDeclared(final Mutable<Absolute, DeviationStatement, DeviationEffectiveStatement> ctx) {
56         super.onFullDefinitionDeclared(ctx);
57
58         StmtContext<?, ?, ?> root = ctx.getRoot();
59         if (root.producesDeclared(SubmoduleStatement.class)) {
60             // root is submodule, we need to find the module we belong to. We can rely on there being exactly one
61             // belongs-to statement, enforced SubmoduleStatementSupport's validator.
62             root = Iterables.getOnlyElement(
63                 root.getAllFromNamespace(ParserNamespaces.BELONGSTO_PREFIX_TO_MODULECTX).values());
64         }
65
66         final var currentModule = verifyNotNull(ctx.getFromNamespace(ParserNamespaces.MODULECTX_TO_QNAME, root),
67             "Failed to find QName for %s", root);
68         final var targetModule = Iterables.getLast(ctx.getArgument().getNodeIdentifiers()).getModule();
69         if (currentModule.equals(targetModule)) {
70             throw new InferenceException(ctx,
71                     "Deviation must not target the same module as the one it is defined in: %s", currentModule);
72         }
73     }
74
75     @Override
76     protected DeviationStatement createDeclared(final BoundStmtCtx<Absolute> ctx,
77             final ImmutableList<DeclaredStatement<?>> substatements) {
78         return DeclaredStatements.createDeviation(ctx.getRawArgument(), ctx.getArgument(), substatements);
79     }
80
81     @Override
82     protected DeviationStatement attachDeclarationReference(final DeviationStatement stmt,
83             final DeclarationReference reference) {
84         return DeclaredStatementDecorators.decorateDeviation(stmt, reference);
85     }
86
87     @Override
88     protected DeviationEffectiveStatement createEffective(final Current<Absolute, DeviationStatement> stmt,
89             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
90         return EffectiveStatements.createDeviation(stmt.declared(), substatements);
91     }
92 }