Retain DeclarationReference in DeclaredStatements
[yangtools.git] / yang / 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 com.google.common.collect.ImmutableList;
11 import com.google.common.collect.Iterables;
12 import org.opendaylight.yangtools.yang.common.QNameModule;
13 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
14 import org.opendaylight.yangtools.yang.model.api.meta.DeclarationReference;
15 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
16 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
17 import org.opendaylight.yangtools.yang.model.api.stmt.DeviationEffectiveStatement;
18 import org.opendaylight.yangtools.yang.model.api.stmt.DeviationStatement;
19 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier.Absolute;
20 import org.opendaylight.yangtools.yang.model.parser.api.YangParserConfiguration;
21 import org.opendaylight.yangtools.yang.model.ri.stmt.DeclaredStatementDecorators;
22 import org.opendaylight.yangtools.yang.model.ri.stmt.DeclaredStatements;
23 import org.opendaylight.yangtools.yang.model.ri.stmt.EffectiveStatements;
24 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.ArgumentUtils;
25 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
26 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
27 import org.opendaylight.yangtools.yang.parser.spi.meta.InferenceException;
28 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
29 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
30 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
31 import org.opendaylight.yangtools.yang.parser.spi.source.ModuleCtxToModuleQName;
32
33 public final class DeviationStatementSupport
34         extends AbstractStatementSupport<Absolute, DeviationStatement, DeviationEffectiveStatement> {
35     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(YangStmtMapping
36         .DEVIATION)
37         .addOptional(YangStmtMapping.DESCRIPTION)
38         .addAny(YangStmtMapping.DEVIATE)
39         .addOptional(YangStmtMapping.REFERENCE)
40         .build();
41
42     public DeviationStatementSupport(final YangParserConfiguration config) {
43         super(YangStmtMapping.DEVIATION, StatementPolicy.reject(), config);
44     }
45
46     @Override
47     public Absolute parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
48         return ArgumentUtils.parseAbsoluteSchemaNodeIdentifier(ctx, value);
49     }
50
51     @Override
52     public void onFullDefinitionDeclared(
53             final Mutable<Absolute, DeviationStatement, DeviationEffectiveStatement> ctx) {
54         final QNameModule currentModule = ctx.getFromNamespace(ModuleCtxToModuleQName.class, ctx.getRoot());
55         final QNameModule targetModule = Iterables.getLast(ctx.getArgument().getNodeIdentifiers()).getModule();
56
57         if (currentModule.equals(targetModule)) {
58             throw new InferenceException(ctx,
59                     "Deviation must not target the same module as the one it is defined in: %s", currentModule);
60         }
61     }
62
63     @Override
64     protected SubstatementValidator getSubstatementValidator() {
65         return SUBSTATEMENT_VALIDATOR;
66     }
67
68     @Override
69     protected DeviationStatement createDeclared(final StmtContext<Absolute, DeviationStatement, ?> ctx,
70             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
71         return DeclaredStatements.createDeviation(ctx.getRawArgument(), ctx.getArgument(), substatements);
72     }
73
74     @Override
75     protected DeviationStatement attachDeclarationReference(final DeviationStatement stmt,
76             final DeclarationReference reference) {
77         return DeclaredStatementDecorators.decorateDeviation(stmt, reference);
78     }
79
80     @Override
81     protected DeviationEffectiveStatement createEffective(final Current<Absolute, DeviationStatement> stmt,
82             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
83         return EffectiveStatements.createDeviation(stmt.declared(), substatements);
84     }
85 }