All StatementSupports must support DeclarationReference
[yangtools.git] / yang / rfc7952-parser-support / src / main / java / org / opendaylight / yangtools / rfc7952 / parser / AnnotationStatementSupport.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.rfc7952.parser;
9
10 import com.google.common.collect.ImmutableList;
11 import org.opendaylight.yangtools.rfc7952.model.api.AnnotationEffectiveStatement;
12 import org.opendaylight.yangtools.rfc7952.model.api.AnnotationStatement;
13 import org.opendaylight.yangtools.rfc7952.model.api.MetadataStatements;
14 import org.opendaylight.yangtools.yang.common.QName;
15 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
16 import org.opendaylight.yangtools.yang.model.api.meta.DeclarationReference;
17 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
18 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
19 import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;
20 import org.opendaylight.yangtools.yang.parser.api.YangParserConfiguration;
21 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
22 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
23 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
24 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
25 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContextUtils;
26 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
27 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
28
29 public final class AnnotationStatementSupport
30          extends AbstractStatementSupport<QName, AnnotationStatement, AnnotationEffectiveStatement> {
31     private static final SubstatementValidator VALIDATOR = SubstatementValidator.builder(MetadataStatements.ANNOTATION)
32         .addMandatory(YangStmtMapping.TYPE)
33         .addOptional(YangStmtMapping.DESCRIPTION)
34         .addAny(YangStmtMapping.IF_FEATURE)
35         .addOptional(YangStmtMapping.REFERENCE)
36         .addOptional(YangStmtMapping.STATUS)
37         .addOptional(YangStmtMapping.UNITS)
38         .build();
39
40     public AnnotationStatementSupport(final YangParserConfiguration config) {
41         super(MetadataStatements.ANNOTATION, StatementPolicy.reject(), config);
42     }
43
44     @Override
45     public QName parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
46         return StmtContextUtils.parseIdentifier(ctx, value);
47     }
48
49     @Override
50     public void onStatementAdded(final Mutable<QName, AnnotationStatement, AnnotationEffectiveStatement> stmt) {
51         final StatementDefinition parentDef = stmt.coerceParentContext().publicDefinition();
52         SourceException.throwIf(YangStmtMapping.MODULE != parentDef && YangStmtMapping.SUBMODULE != parentDef,
53                 stmt, "Annotations may only be defined at root of either a module or a submodule");
54     }
55
56     @Override
57     protected SubstatementValidator getSubstatementValidator() {
58         return VALIDATOR;
59     }
60
61     @Override
62     protected AnnotationStatement createDeclared(final StmtContext<QName, AnnotationStatement, ?> ctx,
63             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
64         return new AnnotationStatementImpl(ctx.getArgument(), substatements);
65     }
66
67     @Override
68     protected AnnotationStatement attachDeclarationReference(final AnnotationStatement stmt,
69             final DeclarationReference reference) {
70         return new RefAnnotationStatement(stmt, reference);
71     }
72
73     @Override
74     protected AnnotationEffectiveStatement createEffective(final Current<QName, AnnotationStatement> stmt,
75             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
76         return new AnnotationEffectiveStatementImpl(stmt, substatements);
77     }
78 }