Bug 6871: [YANG 1.1] Allow "must" in "input", "output" and "notification" statements
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / NotificationStatementImpl.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc. 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.stmt.rfc6020;
9
10 import java.util.Collection;
11 import javax.annotation.Nonnull;
12 import javax.annotation.Nullable;
13 import org.opendaylight.yangtools.yang.common.QName;
14 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
15 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
16 import org.opendaylight.yangtools.yang.model.api.stmt.DataDefinitionStatement;
17 import org.opendaylight.yangtools.yang.model.api.stmt.DescriptionStatement;
18 import org.opendaylight.yangtools.yang.model.api.stmt.GroupingStatement;
19 import org.opendaylight.yangtools.yang.model.api.stmt.MustStatement;
20 import org.opendaylight.yangtools.yang.model.api.stmt.NotificationStatement;
21 import org.opendaylight.yangtools.yang.model.api.stmt.ReferenceStatement;
22 import org.opendaylight.yangtools.yang.model.api.stmt.StatusStatement;
23 import org.opendaylight.yangtools.yang.model.api.stmt.TypedefStatement;
24 import org.opendaylight.yangtools.yang.parser.spi.SubstatementValidator;
25 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractDeclaredStatement;
26 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
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.stmt.rfc6020.effective.NotificationEffectiveStatementImpl;
30
31 public class NotificationStatementImpl extends AbstractDeclaredStatement<QName>
32         implements NotificationStatement {
33     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(YangStmtMapping
34             .NOTIFICATION)
35             .addAny(YangStmtMapping.ANYXML)
36             .addAny(YangStmtMapping.CHOICE)
37             .addAny(YangStmtMapping.CONTAINER)
38             .addOptional(YangStmtMapping.DESCRIPTION)
39             .addAny(YangStmtMapping.GROUPING)
40             .addAny(YangStmtMapping.IF_FEATURE)
41             .addAny(YangStmtMapping.LEAF)
42             .addAny(YangStmtMapping.LEAF_LIST)
43             .addAny(YangStmtMapping.LIST)
44             .addOptional(YangStmtMapping.REFERENCE)
45             .addOptional(YangStmtMapping.STATUS)
46             .addAny(YangStmtMapping.TYPEDEF)
47             .addAny(YangStmtMapping.USES)
48             .build();
49
50     protected NotificationStatementImpl(
51             final StmtContext<QName, NotificationStatement, ?> context) {
52         super(context);
53     }
54
55     public static class Definition
56             extends
57             AbstractStatementSupport<QName, NotificationStatement, EffectiveStatement<QName, NotificationStatement>> {
58
59         public Definition() {
60             super(YangStmtMapping.NOTIFICATION);
61         }
62
63         @Override
64         public QName parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
65             return Utils.qNameFromArgument(ctx, value);
66         }
67
68         @Override
69         public void onStatementAdded(
70                 final Mutable<QName, NotificationStatement, EffectiveStatement<QName, NotificationStatement>> stmt) {
71             stmt.getParentContext().addToNs(ChildSchemaNodes.class, stmt.getStatementArgument(), stmt);
72         }
73
74         @Override
75         public NotificationStatement createDeclared(
76                 final StmtContext<QName, NotificationStatement, ?> ctx) {
77             return new NotificationStatementImpl(ctx);
78         }
79
80         @Override
81         public EffectiveStatement<QName, NotificationStatement> createEffective(
82                 final StmtContext<QName, NotificationStatement, EffectiveStatement<QName, NotificationStatement>> ctx) {
83             return new NotificationEffectiveStatementImpl(ctx);
84         }
85
86         @Override
87         public void onFullDefinitionDeclared(final Mutable<QName, NotificationStatement,
88                 EffectiveStatement<QName, NotificationStatement>> stmt) {
89             super.onFullDefinitionDeclared(stmt);
90             getSubstatementValidator().validate(stmt);
91         }
92
93         @Override
94         protected SubstatementValidator getSubstatementValidator() {
95             return SUBSTATEMENT_VALIDATOR;
96         }
97     }
98
99     @Nonnull
100     @Override
101     public QName getName() {
102         return argument();
103     }
104
105     @Nonnull
106     @Override
107     public Collection<? extends TypedefStatement> getTypedefs() {
108         return allDeclared(TypedefStatement.class);
109     }
110
111     @Nonnull
112     @Override
113     public Collection<? extends GroupingStatement> getGroupings() {
114         return allDeclared(GroupingStatement.class);
115     }
116
117     @Nonnull
118     @Override
119     public Collection<? extends DataDefinitionStatement> getDataDefinitions() {
120         return allDeclared(DataDefinitionStatement.class);
121     }
122
123     @Nonnull
124     @Override
125     public Collection<? extends MustStatement> getMusts() {
126         return allDeclared(MustStatement.class);
127     }
128
129     @Nullable
130     @Override
131     public StatusStatement getStatus() {
132         return firstDeclared(StatusStatement.class);
133     }
134
135     @Nullable
136     @Override
137     public DescriptionStatement getDescription() {
138         return firstDeclared(DescriptionStatement.class);
139     }
140
141     @Nullable
142     @Override
143     public ReferenceStatement getReference() {
144         return firstDeclared(ReferenceStatement.class);
145     }
146 }