f67c317ee0a93bc3d35878bbb838bf532b4df784
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / case_ / AbstractCaseStatementSupport.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.case_;
9
10 import com.google.common.collect.ImmutableList;
11 import org.eclipse.jdt.annotation.Nullable;
12 import org.opendaylight.yangtools.yang.common.QName;
13 import org.opendaylight.yangtools.yang.model.api.CaseSchemaNode;
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.meta.StatementSource;
19 import org.opendaylight.yangtools.yang.model.api.stmt.CaseEffectiveStatement;
20 import org.opendaylight.yangtools.yang.model.api.stmt.CaseStatement;
21 import org.opendaylight.yangtools.yang.model.api.stmt.StatusEffectiveStatement;
22 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.BaseImplicitStatementSupport;
23 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.EffectiveStatementMixins.EffectiveStatementWithFlags.FlagsBuilder;
24 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
25 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
26
27 abstract class AbstractCaseStatementSupport
28         extends BaseImplicitStatementSupport<CaseStatement, CaseEffectiveStatement> {
29     AbstractCaseStatementSupport() {
30         super(YangStmtMapping.CASE);
31     }
32
33     @Override
34     protected final CaseStatement createDeclared(final StmtContext<QName, CaseStatement, ?> ctx,
35             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
36         final StatementSource source = ctx.source();
37         switch (source) {
38             case CONTEXT:
39                 return new RegularUndeclaredCaseStatement(ctx.getArgument(), substatements);
40             case DECLARATION:
41                 return new RegularCaseStatement(ctx.getArgument(), substatements);
42             default:
43                 throw new IllegalStateException("Unhandled statement source " + source);
44         }
45     }
46
47     @Override
48     protected final CaseStatement createEmptyDeclared(final StmtContext<QName, CaseStatement, ?> ctx) {
49         final StatementSource source = ctx.source();
50         switch (source) {
51             case CONTEXT:
52                 return new EmptyUndeclaredCaseStatement(ctx.getArgument());
53             case DECLARATION:
54                 return new EmptyCaseStatement(ctx.getArgument());
55             default:
56                 throw new IllegalStateException("Unhandled statement source " + source);
57         }
58     }
59
60     @Override
61     protected final CaseEffectiveStatement createDeclaredEffective(final Current<QName, CaseStatement> stmt,
62             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
63         return new DeclaredCaseEffectiveStatement(stmt.declared(), substatements, stmt.sourceReference(),
64             computeFlags(stmt, substatements), stmt.getSchemaPath(), findOriginal(stmt));
65     }
66
67     @Override
68     protected final CaseEffectiveStatement createUndeclaredEffective(final Current<QName, CaseStatement> stmt,
69             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
70         return new UndeclaredCaseEffectiveStatement(substatements, stmt.sourceReference(),
71             computeFlags(stmt, substatements), stmt.getSchemaPath(), findOriginal(stmt));
72     }
73
74     private static @Nullable CaseSchemaNode findOriginal(final Current<?, ?> stmt) {
75         return (CaseSchemaNode) stmt.original();
76     }
77
78     private static int computeFlags(final Current<?, ?> stmt,
79             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
80         return new FlagsBuilder()
81                 .setHistory(stmt.history())
82                 .setStatus(findFirstArgument(substatements, StatusEffectiveStatement.class, Status.CURRENT))
83                 .setConfiguration(stmt.effectiveConfig()
84                     && stmt.caerbannog().allSubstatementsStream().anyMatch(StmtContext::isConfiguration))
85                 .toFlags();
86     }
87 }