Clean up argument checking
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / base / BaseStatementSupport.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.base;
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.YangStmtMapping;
14 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
15 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
16 import org.opendaylight.yangtools.yang.model.api.stmt.BaseEffectiveStatement;
17 import org.opendaylight.yangtools.yang.model.api.stmt.BaseStatement;
18 import org.opendaylight.yangtools.yang.model.api.stmt.IdentityStatement;
19 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.BaseQNameStatementSupport;
20 import org.opendaylight.yangtools.yang.parser.spi.IdentityNamespace;
21 import org.opendaylight.yangtools.yang.parser.spi.meta.InferenceException;
22 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder;
23 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder.InferenceAction;
24 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder.InferenceContext;
25 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder.Prerequisite;
26 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
27 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
28 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
29 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
30 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
31
32 public final class BaseStatementSupport extends BaseQNameStatementSupport<BaseStatement, BaseEffectiveStatement> {
33     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR =
34             SubstatementValidator.builder(YangStmtMapping.BASE).build();
35     private static final BaseStatementSupport INSTANCE = new BaseStatementSupport();
36
37     private BaseStatementSupport() {
38         super(YangStmtMapping.BASE);
39     }
40
41     public static BaseStatementSupport getInstance() {
42         return INSTANCE;
43     }
44
45     @Override
46     public QName parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
47         return StmtContextUtils.parseNodeIdentifier(ctx, value);
48     }
49
50     @Override
51     public void onStatementDefinitionDeclared(final Mutable<QName, BaseStatement, BaseEffectiveStatement> baseStmtCtx) {
52         final Mutable<?, ?, ?> baseParentCtx = baseStmtCtx.coerceParentContext();
53         if (baseParentCtx.producesDeclared(IdentityStatement.class)) {
54             final QName baseIdentityQName = baseStmtCtx.coerceStatementArgument();
55             final ModelActionBuilder baseIdentityAction = baseStmtCtx.newInferenceAction(
56                 ModelProcessingPhase.STATEMENT_DEFINITION);
57             baseIdentityAction.requiresCtx(baseStmtCtx, IdentityNamespace.class, baseIdentityQName,
58                 ModelProcessingPhase.STATEMENT_DEFINITION);
59             baseIdentityAction.mutatesCtx(baseParentCtx, ModelProcessingPhase.STATEMENT_DEFINITION);
60
61             baseIdentityAction.apply(new InferenceAction() {
62                 @Override
63                 public void apply(final InferenceContext ctx) {
64                     // No-op, we just want to ensure the statement is specified
65                 }
66
67                 @Override
68                 public void prerequisiteFailed(final Collection<? extends Prerequisite<?>> failed) {
69                     throw new InferenceException(baseStmtCtx.getStatementSourceReference(),
70                         "Unable to resolve identity %s and base identity %s",
71                         baseParentCtx.getStatementArgument(), baseStmtCtx.getStatementArgument());
72                 }
73             });
74         }
75     }
76
77     @Override
78     protected SubstatementValidator getSubstatementValidator() {
79         return SUBSTATEMENT_VALIDATOR;
80     }
81
82     @Override
83     protected BaseStatement createDeclared(final StmtContext<QName, BaseStatement, ?> ctx,
84             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
85         return new RegularBaseStatement(ctx.coerceStatementArgument(), substatements);
86     }
87
88     @Override
89     protected BaseStatement createEmptyDeclared(final StmtContext<QName, BaseStatement, ?> ctx) {
90         return new EmptyBaseStatement(ctx.coerceStatementArgument());
91     }
92
93     @Override
94     protected BaseEffectiveStatement createEffective(
95             final StmtContext<QName, BaseStatement, BaseEffectiveStatement> ctx, final BaseStatement declared,
96             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
97         return new RegularBaseEffectiveStatement(declared, substatements);
98     }
99
100     @Override
101     protected BaseEffectiveStatement createEmptyEffective(
102             final StmtContext<QName, BaseStatement, BaseEffectiveStatement> ctx, final BaseStatement declared) {
103         return new EmptyBaseEffectiveStatement(declared);
104     }
105 }