6879078635eeb734c9e6b0ec7677f80dff4539e5
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / meta / CaseStatementSupport.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.annotations.Beta;
11 import com.google.common.collect.ImmutableList;
12 import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
13 import java.util.Collection;
14 import java.util.Optional;
15 import org.eclipse.jdt.annotation.NonNull;
16 import org.eclipse.jdt.annotation.Nullable;
17 import org.opendaylight.yangtools.yang.common.QName;
18 import org.opendaylight.yangtools.yang.model.api.CaseSchemaNode;
19 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
20 import org.opendaylight.yangtools.yang.model.api.Status;
21 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
22 import org.opendaylight.yangtools.yang.model.api.meta.DeclarationReference;
23 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
24 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
25 import org.opendaylight.yangtools.yang.model.api.meta.StatementOrigin;
26 import org.opendaylight.yangtools.yang.model.api.stmt.CaseEffectiveStatement;
27 import org.opendaylight.yangtools.yang.model.api.stmt.CaseStatement;
28 import org.opendaylight.yangtools.yang.model.api.stmt.StatusEffectiveStatement;
29 import org.opendaylight.yangtools.yang.model.ri.stmt.DeclaredStatementDecorators;
30 import org.opendaylight.yangtools.yang.model.ri.stmt.DeclaredStatements;
31 import org.opendaylight.yangtools.yang.model.ri.stmt.EffectiveStatements;
32 import org.opendaylight.yangtools.yang.model.ri.stmt.ImplicitStatements;
33 import org.opendaylight.yangtools.yang.model.spi.meta.EffectiveStatementMixins.EffectiveStatementWithFlags.FlagsBuilder;
34 import org.opendaylight.yangtools.yang.model.spi.meta.SubstatementIndexingException;
35 import org.opendaylight.yangtools.yang.parser.api.YangParserConfiguration;
36 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
37 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Parent.EffectiveConfig;
38 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
39 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
40 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
41
42 @Beta
43 public final class CaseStatementSupport
44         extends AbstractImplicitStatementSupport<CaseStatement, CaseEffectiveStatement> {
45     private static final SubstatementValidator RFC6020_VALIDATOR = SubstatementValidator.builder(YangStmtMapping.CASE)
46             .addAny(YangStmtMapping.ANYXML)
47             .addAny(YangStmtMapping.CHOICE)
48             .addAny(YangStmtMapping.CONTAINER)
49             .addOptional(YangStmtMapping.DESCRIPTION)
50             .addAny(YangStmtMapping.IF_FEATURE)
51             .addAny(YangStmtMapping.LEAF)
52             .addAny(YangStmtMapping.LEAF_LIST)
53             .addAny(YangStmtMapping.LIST)
54             .addOptional(YangStmtMapping.REFERENCE)
55             .addOptional(YangStmtMapping.STATUS)
56             .addAny(YangStmtMapping.USES)
57             .addOptional(YangStmtMapping.WHEN)
58             .build();
59     private static final SubstatementValidator RFC7950_VALIDATOR = SubstatementValidator.builder(YangStmtMapping.CASE)
60             .addAny(YangStmtMapping.ANYDATA)
61             .addAny(YangStmtMapping.ANYXML)
62             .addAny(YangStmtMapping.CHOICE)
63             .addAny(YangStmtMapping.CONTAINER)
64             .addOptional(YangStmtMapping.DESCRIPTION)
65             .addAny(YangStmtMapping.IF_FEATURE)
66             .addAny(YangStmtMapping.LEAF)
67             .addAny(YangStmtMapping.LEAF_LIST)
68             .addAny(YangStmtMapping.LIST)
69             .addOptional(YangStmtMapping.REFERENCE)
70             .addOptional(YangStmtMapping.STATUS)
71             .addAny(YangStmtMapping.USES)
72             .addOptional(YangStmtMapping.WHEN)
73             .build();
74
75     private CaseStatementSupport(final YangParserConfiguration config, final SubstatementValidator validator) {
76         super(YangStmtMapping.CASE, instantiatedPolicy(), config, validator);
77     }
78
79     public static @NonNull CaseStatementSupport rfc6020Instance(final YangParserConfiguration config) {
80         return new CaseStatementSupport(config, RFC6020_VALIDATOR);
81     }
82
83     public static @NonNull CaseStatementSupport rfc7950Instance(final YangParserConfiguration config) {
84         return new CaseStatementSupport(config, RFC7950_VALIDATOR);
85     }
86
87     @Override
88     protected CaseStatement createDeclared(final StmtContext<QName, CaseStatement, ?> ctx,
89             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
90         final StatementOrigin origin = ctx.origin();
91         switch (origin) {
92             case CONTEXT:
93                 return ImplicitStatements.createCase(ctx.getArgument(), substatements);
94             case DECLARATION:
95                 return DeclaredStatements.createCase(ctx.getArgument(), substatements);
96             default:
97                 throw new IllegalStateException("Unhandled statement origin " + origin);
98         }
99     }
100
101     @Override
102     protected CaseStatement attachDeclarationReference(final CaseStatement stmt, final DeclarationReference reference) {
103         return DeclaredStatementDecorators.decorateCase(stmt, reference);
104     }
105
106     @Override
107     protected CaseEffectiveStatement copyDeclaredEffective(final Current<QName, CaseStatement> stmt,
108             final CaseEffectiveStatement original) {
109         return EffectiveStatements.copyCase(original, stmt.effectivePath(),
110             computeFlags(stmt, original.effectiveSubstatements()), stmt.original(CaseSchemaNode.class));
111     }
112
113     @Override
114     protected CaseEffectiveStatement copyUndeclaredEffective(final Current<QName, CaseStatement> stmt,
115             final CaseEffectiveStatement original) {
116         return EffectiveStatements.copyCase(original, stmt.effectivePath(),
117             computeFlags(stmt, original.effectiveSubstatements()), stmt.original(CaseSchemaNode.class));
118     }
119
120     @Override
121     protected CaseEffectiveStatement createDeclaredEffective(final Current<QName, CaseStatement> stmt,
122             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
123         try {
124             return EffectiveStatements.createCase(stmt.declared(), stmt.effectivePath(),
125                 computeFlags(stmt, substatements), substatements, stmt.original(CaseSchemaNode.class));
126         } catch (SubstatementIndexingException e) {
127             throw new SourceException(e.getMessage(), stmt, e);
128         }
129     }
130
131     @Override
132     protected CaseEffectiveStatement createUndeclaredEffective(final Current<QName, CaseStatement> stmt,
133             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
134         try {
135             return EffectiveStatements.createCase(stmt.effectivePath(), computeFlags(stmt, substatements),
136                 substatements, stmt.original(CaseSchemaNode.class));
137         } catch (SubstatementIndexingException e) {
138             throw new SourceException(e.getMessage(), stmt, e);
139         }
140     }
141
142     private static int computeFlags(final Current<?, ?> stmt,
143             final Collection<? extends EffectiveStatement<?, ?>> substatements) {
144         final Boolean config;
145         final EffectiveConfig effective = stmt.effectiveConfig();
146         switch (effective) {
147             case FALSE:
148                 config = Boolean.FALSE;
149                 break;
150             case IGNORED:
151                 config = null;
152                 break;
153             case TRUE:
154                 final Boolean sub = substatementEffectiveConfig(substatements);
155                 config = sub != null ? sub : Boolean.TRUE;
156                 break;
157             case UNDETERMINED:
158                 config = substatementEffectiveConfig(substatements);
159                 break;
160             default:
161                 throw new IllegalStateException("Unhandled effective config " + effective);
162         }
163
164         return new FlagsBuilder()
165                 .setHistory(stmt.history())
166                 .setStatus(findFirstArgument(substatements, StatusEffectiveStatement.class, Status.CURRENT))
167                 .setConfiguration(config)
168                 .toFlags();
169     }
170
171     @SuppressFBWarnings(value = "NP_BOOLEAN_RETURN_NULL", justification = "Internal use tagged with @Nullable")
172     private static @Nullable Boolean substatementEffectiveConfig(
173             final Collection<? extends EffectiveStatement<?, ?>> substatements) {
174         for (EffectiveStatement<?, ?> stmt : substatements) {
175             if (stmt instanceof DataSchemaNode) {
176                 final Optional<Boolean> opt = ((DataSchemaNode) stmt).effectiveConfig();
177                 if (opt.isPresent()) {
178                     return opt.orElseThrow();
179                 }
180             }
181         }
182         return null;
183     }
184 }