Teach YANG parser to retain DeclarationReference
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / meta / LeafStatementSupport.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 java.util.Collection;
12 import org.opendaylight.yangtools.yang.common.QName;
13 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
14 import org.opendaylight.yangtools.yang.model.api.Status;
15 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
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.DefaultEffectiveStatement;
19 import org.opendaylight.yangtools.yang.model.api.stmt.LeafEffectiveStatement;
20 import org.opendaylight.yangtools.yang.model.api.stmt.LeafStatement;
21 import org.opendaylight.yangtools.yang.model.api.stmt.MandatoryEffectiveStatement;
22 import org.opendaylight.yangtools.yang.model.api.stmt.StatusEffectiveStatement;
23 import org.opendaylight.yangtools.yang.model.api.stmt.TypeEffectiveStatement;
24 import org.opendaylight.yangtools.yang.model.parser.api.YangParserConfiguration;
25 import org.opendaylight.yangtools.yang.model.ri.stmt.DeclaredStatements;
26 import org.opendaylight.yangtools.yang.model.ri.stmt.EffectiveStatements;
27 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.EffectiveStatementWithFlags.FlagsBuilder;
28 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStmtUtils;
29 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractSchemaTreeStatementSupport;
30 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
31 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
32 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
33 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
34 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
35 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
36
37 public final class LeafStatementSupport
38         extends AbstractSchemaTreeStatementSupport<LeafStatement, LeafEffectiveStatement> {
39     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR =
40         SubstatementValidator.builder(YangStmtMapping.LEAF)
41             .addOptional(YangStmtMapping.CONFIG)
42             .addOptional(YangStmtMapping.DEFAULT)
43             .addOptional(YangStmtMapping.DESCRIPTION)
44             .addAny(YangStmtMapping.IF_FEATURE)
45             .addOptional(YangStmtMapping.MANDATORY)
46             .addAny(YangStmtMapping.MUST)
47             .addOptional(YangStmtMapping.REFERENCE)
48             .addOptional(YangStmtMapping.STATUS)
49             .addMandatory(YangStmtMapping.TYPE)
50             .addOptional(YangStmtMapping.UNITS)
51             .addOptional(YangStmtMapping.WHEN)
52             .build();
53
54     public LeafStatementSupport(final YangParserConfiguration config) {
55         super(YangStmtMapping.LEAF, instantiatedPolicy(), config);
56     }
57
58     @Override
59     public void onFullDefinitionDeclared(final Mutable<QName, LeafStatement, LeafEffectiveStatement> ctx) {
60         super.onFullDefinitionDeclared(ctx);
61         StmtContextUtils.validateIfFeatureAndWhenOnListKeys(ctx);
62     }
63
64     @Override
65     protected SubstatementValidator getSubstatementValidator() {
66         return SUBSTATEMENT_VALIDATOR;
67     }
68
69     @Override
70     protected LeafStatement createDeclared(final StmtContext<QName, LeafStatement, ?> ctx,
71             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
72         return DeclaredStatements.createLeaf(ctx.getArgument(), substatements);
73     }
74
75     @Override
76     public LeafEffectiveStatement copyEffective(final Current<QName, LeafStatement> stmt,
77             final LeafEffectiveStatement original) {
78         return EffectiveStatements.copyLeaf(original, stmt.effectivePath(),
79             computeFlags(stmt, original.effectiveSubstatements()), stmt.original(LeafSchemaNode.class));
80     }
81
82     @Override
83     protected LeafEffectiveStatement createEffective(final Current<QName, LeafStatement> stmt,
84             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
85         final TypeEffectiveStatement<?> typeStmt = SourceException.throwIfNull(
86             findFirstStatement(substatements, TypeEffectiveStatement.class), stmt,
87             "Leaf is missing a 'type' statement");
88         final String dflt = findFirstArgument(substatements, DefaultEffectiveStatement.class, null);
89         SourceException.throwIf(
90             EffectiveStmtUtils.hasDefaultValueMarkedWithIfFeature(stmt.yangVersion(), typeStmt, dflt), stmt,
91             "Leaf '%s' has default value '%s' marked with an if-feature statement.", stmt.argument(), dflt);
92
93         return EffectiveStatements.createLeaf(stmt.declared(), stmt.effectivePath(), computeFlags(stmt, substatements),
94             substatements, stmt.original(LeafSchemaNode.class));
95     }
96
97     private static int computeFlags(final Current<?, ?> stmt,
98         final Collection<? extends EffectiveStatement<?, ?>> substatements) {
99         return new FlagsBuilder()
100             .setHistory(stmt.history())
101             .setStatus(findFirstArgument(substatements, StatusEffectiveStatement.class, Status.CURRENT))
102             .setConfiguration(stmt.effectiveConfig().asNullable())
103             .setMandatory(findFirstArgument(substatements, MandatoryEffectiveStatement.class, Boolean.FALSE))
104             .toFlags();
105     }
106 }