Retain DeclarationReference in DeclaredStatements
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / meta / ContainerStatementSupport.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 java.util.Objects.requireNonNull;
11
12 import com.google.common.annotations.Beta;
13 import com.google.common.collect.ImmutableList;
14 import java.util.Collection;
15 import org.eclipse.jdt.annotation.NonNull;
16 import org.opendaylight.yangtools.yang.common.QName;
17 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
18 import org.opendaylight.yangtools.yang.model.api.Status;
19 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
20 import org.opendaylight.yangtools.yang.model.api.meta.DeclarationReference;
21 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
22 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
23 import org.opendaylight.yangtools.yang.model.api.stmt.ContainerEffectiveStatement;
24 import org.opendaylight.yangtools.yang.model.api.stmt.ContainerStatement;
25 import org.opendaylight.yangtools.yang.model.api.stmt.PresenceEffectiveStatement;
26 import org.opendaylight.yangtools.yang.model.api.stmt.StatusEffectiveStatement;
27 import org.opendaylight.yangtools.yang.model.parser.api.YangParserConfiguration;
28 import org.opendaylight.yangtools.yang.model.ri.stmt.DeclaredStatementDecorators;
29 import org.opendaylight.yangtools.yang.model.ri.stmt.DeclaredStatements;
30 import org.opendaylight.yangtools.yang.model.ri.stmt.EffectiveStatements;
31 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.EffectiveStatementWithFlags.FlagsBuilder;
32 import org.opendaylight.yangtools.yang.model.spi.meta.SubstatementIndexingException;
33 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStmtUtils;
34 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractSchemaTreeStatementSupport;
35 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
36 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
37 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
38 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
39
40 @Beta
41 public final class ContainerStatementSupport
42         extends AbstractSchemaTreeStatementSupport<ContainerStatement, ContainerEffectiveStatement> {
43     private static final SubstatementValidator RFC6020_VALIDATOR =
44         SubstatementValidator.builder(YangStmtMapping.CONTAINER)
45             .addAny(YangStmtMapping.ANYXML)
46             .addAny(YangStmtMapping.CHOICE)
47             .addOptional(YangStmtMapping.CONFIG)
48             .addAny(YangStmtMapping.CONTAINER)
49             .addOptional(YangStmtMapping.DESCRIPTION)
50             .addAny(YangStmtMapping.GROUPING)
51             .addAny(YangStmtMapping.IF_FEATURE)
52             .addAny(YangStmtMapping.LEAF)
53             .addAny(YangStmtMapping.LEAF_LIST)
54             .addAny(YangStmtMapping.LIST)
55             .addAny(YangStmtMapping.MUST)
56             .addOptional(YangStmtMapping.PRESENCE)
57             .addOptional(YangStmtMapping.REFERENCE)
58             .addOptional(YangStmtMapping.STATUS)
59             .addAny(YangStmtMapping.TYPEDEF)
60             .addAny(YangStmtMapping.USES)
61             .addOptional(YangStmtMapping.WHEN)
62             .build();
63
64     private static final SubstatementValidator RFC7950_VALIDATOR =
65         SubstatementValidator.builder(YangStmtMapping.CONTAINER)
66             .addAny(YangStmtMapping.ACTION)
67             .addAny(YangStmtMapping.ANYDATA)
68             .addAny(YangStmtMapping.ANYXML)
69             .addAny(YangStmtMapping.CHOICE)
70             .addOptional(YangStmtMapping.CONFIG)
71             .addAny(YangStmtMapping.CONTAINER)
72             .addOptional(YangStmtMapping.DESCRIPTION)
73             .addAny(YangStmtMapping.GROUPING)
74             .addAny(YangStmtMapping.IF_FEATURE)
75             .addAny(YangStmtMapping.LEAF)
76             .addAny(YangStmtMapping.LEAF_LIST)
77             .addAny(YangStmtMapping.LIST)
78             .addAny(YangStmtMapping.MUST)
79             .addAny(YangStmtMapping.NOTIFICATION)
80             .addOptional(YangStmtMapping.PRESENCE)
81             .addOptional(YangStmtMapping.REFERENCE)
82             .addOptional(YangStmtMapping.STATUS)
83             .addAny(YangStmtMapping.TYPEDEF)
84             .addAny(YangStmtMapping.USES)
85             .addOptional(YangStmtMapping.WHEN)
86             .build();
87
88     private final SubstatementValidator validator;
89
90     private ContainerStatementSupport(final YangParserConfiguration config, final SubstatementValidator validator) {
91         super(YangStmtMapping.CONTAINER, instantiatedPolicy(), config);
92         this.validator = requireNonNull(validator);
93     }
94
95     public static @NonNull ContainerStatementSupport rfc6020Instance(final YangParserConfiguration config) {
96         return new ContainerStatementSupport(config, RFC6020_VALIDATOR);
97     }
98
99     public static @NonNull ContainerStatementSupport rfc7950Instance(final YangParserConfiguration config) {
100         return new ContainerStatementSupport(config, RFC7950_VALIDATOR);
101     }
102
103     @Override
104     protected SubstatementValidator getSubstatementValidator() {
105         return validator;
106     }
107
108     @Override
109     protected ContainerStatement createDeclared(final StmtContext<QName, ContainerStatement, ?> ctx,
110             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
111         return DeclaredStatements.createContainer(ctx.getArgument(), substatements);
112     }
113
114     @Override
115     protected ContainerStatement attachDeclarationReference(final ContainerStatement stmt,
116             final DeclarationReference reference) {
117         return DeclaredStatementDecorators.decorateContainer(stmt, reference);
118     }
119
120     @Override
121     protected ContainerEffectiveStatement createEffective(final Current<QName, ContainerStatement> stmt,
122             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
123         EffectiveStmtUtils.checkUniqueGroupings(stmt, substatements);
124         EffectiveStmtUtils.checkUniqueTypedefs(stmt, substatements);
125         EffectiveStmtUtils.checkUniqueUses(stmt, substatements);
126
127         try {
128             return EffectiveStatements.createContainer(stmt.declared(), stmt.effectivePath(),
129                 createFlags(stmt, substatements), substatements, stmt.original(ContainerSchemaNode.class));
130         } catch (SubstatementIndexingException e) {
131             throw new SourceException(e.getMessage(), stmt, e);
132         }
133     }
134
135     @Override
136     public ContainerEffectiveStatement copyEffective(final Current<QName, ContainerStatement> stmt,
137             final ContainerEffectiveStatement original) {
138         return EffectiveStatements.copyContainer(original, stmt.effectivePath(),
139             createFlags(stmt, original.effectiveSubstatements()), stmt.original(ContainerSchemaNode.class));
140     }
141
142     private static int createFlags(final Current<?, ?> stmt,
143             final Collection<? extends EffectiveStatement<?, ?>> substatements) {
144         return new FlagsBuilder()
145             .setHistory(stmt.history())
146             .setStatus(findFirstArgument(substatements, StatusEffectiveStatement.class, Status.CURRENT))
147             .setConfiguration(stmt.effectiveConfig().asNullable())
148             .setPresence(findFirstStatement(substatements, PresenceEffectiveStatement.class) != null)
149             .toFlags();
150     }
151 }
152