Clean up argument checking
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / if_feature / IfFeatureStatementSupport.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.if_feature;
9
10 import static com.google.common.base.Verify.verifyNotNull;
11
12 import com.google.common.collect.ImmutableList;
13 import java.util.Collection;
14 import java.util.HashMap;
15 import java.util.HashSet;
16 import java.util.Map;
17 import java.util.Set;
18 import org.opendaylight.yangtools.yang.common.QName;
19 import org.opendaylight.yangtools.yang.common.YangVersion;
20 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
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.IfFeatureEffectiveStatement;
24 import org.opendaylight.yangtools.yang.model.api.stmt.IfFeatureExpr;
25 import org.opendaylight.yangtools.yang.model.api.stmt.IfFeatureStatement;
26 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.BaseStatementSupport;
27 import org.opendaylight.yangtools.yang.parser.spi.FeatureNamespace;
28 import org.opendaylight.yangtools.yang.parser.spi.meta.InferenceException;
29 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder;
30 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder.InferenceAction;
31 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder.InferenceContext;
32 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelActionBuilder.Prerequisite;
33 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
34 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
35 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
36 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
37 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
38 import org.slf4j.Logger;
39 import org.slf4j.LoggerFactory;
40
41 public final class IfFeatureStatementSupport
42         extends BaseStatementSupport<IfFeatureExpr, IfFeatureStatement, IfFeatureEffectiveStatement> {
43     private static final Logger LOG = LoggerFactory.getLogger(IfFeatureStatementSupport.class);
44     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(
45         YangStmtMapping.IF_FEATURE)
46         .build();
47     private static final IfFeatureStatementSupport INSTANCE = new IfFeatureStatementSupport();
48
49     private IfFeatureStatementSupport() {
50         super(YangStmtMapping.IF_FEATURE, CopyPolicy.CONTEXT_INDEPENDENT);
51     }
52
53     public static IfFeatureStatementSupport getInstance() {
54         return INSTANCE;
55     }
56
57     @Override
58     public IfFeatureExpr parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
59         if (YangVersion.VERSION_1_1.equals(ctx.getRootVersion())) {
60             return IfFeaturePredicateVisitor.parseIfFeatureExpression(ctx, value);
61         }
62         return IfFeatureExpr.isPresent(StmtContextUtils.parseNodeIdentifier(ctx, value));
63     }
64
65     @Override
66     public void onFullDefinitionDeclared(
67             final Mutable<IfFeatureExpr, IfFeatureStatement, IfFeatureEffectiveStatement> stmt) {
68         super.onFullDefinitionDeclared(stmt);
69
70         final ModelActionBuilder verifyFeatures = stmt.newInferenceAction(ModelProcessingPhase.EFFECTIVE_MODEL);
71         final Map<Prerequisite<?>, QName> backRef = new HashMap<>();
72         for (QName feature : stmt.coerceStatementArgument().getReferencedFeatures()) {
73             backRef.put(verifyFeatures.requiresCtx(stmt, FeatureNamespace.class, feature,
74                 ModelProcessingPhase.EFFECTIVE_MODEL), feature);
75         }
76
77         verifyFeatures.apply(new InferenceAction() {
78             @Override
79             public void apply(final InferenceContext ctx) {
80                 LOG.debug("Resolved all feature references in {}", backRef.values());
81             }
82
83             @Override
84             public void prerequisiteFailed(final Collection<? extends Prerequisite<?>> failed) {
85                 final Set<QName> unresolvedFeatures = new HashSet<>();
86                 for (Prerequisite<?> prereq : failed) {
87                     unresolvedFeatures.add(verifyNotNull(backRef.get(prereq)));
88                 }
89
90                 throw new InferenceException(stmt.getStatementSourceReference(),
91                     "Failed to resolve feature references %s in \"%s\"", unresolvedFeatures,
92                     stmt.rawStatementArgument());
93             }
94         });
95     }
96
97     @Override
98     protected SubstatementValidator getSubstatementValidator() {
99         return SUBSTATEMENT_VALIDATOR;
100     }
101
102     @Override
103     protected IfFeatureStatement createDeclared(final StmtContext<IfFeatureExpr, IfFeatureStatement, ?> ctx,
104             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
105         return new RegularIfFeatureStatement(ctx.coerceRawStatementArgument(), ctx.coerceStatementArgument(),
106             substatements);
107     }
108
109     @Override
110     protected IfFeatureStatement createEmptyDeclared(final StmtContext<IfFeatureExpr, IfFeatureStatement, ?> ctx) {
111         return new EmptyIfFeatureStatement(ctx.coerceRawStatementArgument(), ctx.coerceStatementArgument());
112     }
113
114     @Override
115     protected IfFeatureEffectiveStatement createEffective(
116             final StmtContext<IfFeatureExpr, IfFeatureStatement, IfFeatureEffectiveStatement> ctx,
117             final IfFeatureStatement declared, final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
118         return new RegularIfFeatureEffectiveStatement(declared, substatements);
119     }
120
121     @Override
122     protected IfFeatureEffectiveStatement createEmptyEffective(
123             final StmtContext<IfFeatureExpr, IfFeatureStatement, IfFeatureEffectiveStatement> ctx,
124             final IfFeatureStatement declared) {
125         return new EmptyIfFeatureEffectiveStatement(declared);
126     }
127 }