Merge Input/OutputStatementSupport
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / output / OutputStatementSupport.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.output;
9
10 import static java.util.Objects.requireNonNull;
11
12 import com.google.common.annotations.Beta;
13 import com.google.common.collect.ImmutableList;
14 import org.eclipse.jdt.annotation.NonNull;
15 import org.opendaylight.yangtools.yang.common.QName;
16 import org.opendaylight.yangtools.yang.common.YangConstants;
17 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
18 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
19 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
20 import org.opendaylight.yangtools.yang.model.api.meta.StatementSource;
21 import org.opendaylight.yangtools.yang.model.api.stmt.OutputEffectiveStatement;
22 import org.opendaylight.yangtools.yang.model.api.stmt.OutputStatement;
23 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.BaseOperationContainerStatementSupport;
24 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.SubstatementIndexingException;
25 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
26 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
27 import org.opendaylight.yangtools.yang.parser.spi.meta.SubstatementValidator;
28 import org.opendaylight.yangtools.yang.parser.spi.source.SourceException;
29
30 @Beta
31 public final class OutputStatementSupport
32         extends BaseOperationContainerStatementSupport<OutputStatement, OutputEffectiveStatement> {
33     private static final @NonNull OutputStatementSupport RFC6020_INSTANCE = new OutputStatementSupport(
34         SubstatementValidator.builder(YangStmtMapping.OUTPUT)
35             .addAny(YangStmtMapping.ANYXML)
36             .addAny(YangStmtMapping.CHOICE)
37             .addAny(YangStmtMapping.CONTAINER)
38             .addAny(YangStmtMapping.GROUPING)
39             .addAny(YangStmtMapping.LEAF)
40             .addAny(YangStmtMapping.LEAF_LIST)
41             .addAny(YangStmtMapping.LIST)
42             .addAny(YangStmtMapping.TYPEDEF)
43             .addAny(YangStmtMapping.USES)
44             .build());
45     private static final @NonNull OutputStatementSupport RFC7950_INSTANCE = new OutputStatementSupport(
46         SubstatementValidator.builder(YangStmtMapping.OUTPUT)
47             .addAny(YangStmtMapping.ANYDATA)
48             .addAny(YangStmtMapping.ANYXML)
49             .addAny(YangStmtMapping.CHOICE)
50             .addAny(YangStmtMapping.CONTAINER)
51             .addAny(YangStmtMapping.GROUPING)
52             .addAny(YangStmtMapping.LEAF)
53             .addAny(YangStmtMapping.LEAF_LIST)
54             .addAny(YangStmtMapping.LIST)
55             .addAny(YangStmtMapping.MUST)
56             .addAny(YangStmtMapping.TYPEDEF)
57             .addAny(YangStmtMapping.USES)
58             .build());
59
60     private final SubstatementValidator validator;
61
62     private OutputStatementSupport(final SubstatementValidator validator) {
63         super(YangStmtMapping.OUTPUT, YangConstants::operationOutputQName, CopyPolicy.DECLARED_COPY);
64         this.validator = requireNonNull(validator);
65     }
66
67     public static @NonNull OutputStatementSupport rfc6020Instance() {
68         return RFC6020_INSTANCE;
69     }
70
71     public static @NonNull OutputStatementSupport rfc7950Instance() {
72         return RFC7950_INSTANCE;
73     }
74
75     @Override
76     protected SubstatementValidator getSubstatementValidator() {
77         return validator;
78     }
79
80     @Override
81     protected OutputStatement createDeclared(final StmtContext<QName, OutputStatement, ?> ctx,
82             final ImmutableList<? extends DeclaredStatement<?>> substatements) {
83         final StatementSource source = ctx.source();
84         switch (source) {
85             case CONTEXT:
86                 return new RegularUndeclaredOutputStatement(ctx.getArgument(), substatements);
87             case DECLARATION:
88                 return new RegularOutputStatement(ctx.getArgument(), substatements);
89             default:
90                 throw new IllegalStateException("Unhandled statement source " + source);
91         }
92     }
93
94     @Override
95     protected OutputStatement createEmptyDeclared(final StmtContext<QName, OutputStatement, ?> ctx) {
96         final StatementSource source = ctx.source();
97         switch (source) {
98             case CONTEXT:
99                 return new EmptyUndeclaredOutputStatement(ctx.getArgument());
100             case DECLARATION:
101                 return new EmptyOutputStatement(ctx.getArgument());
102             default:
103                 throw new IllegalStateException("Unhandled statement source " + source);
104         }
105     }
106
107     @Override
108     protected OutputEffectiveStatement createDeclaredEffective(final int flags,
109             final Current<QName, OutputStatement> stmt,
110             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
111         return new DeclaredOutputEffectiveStatement(flags, stmt.declared(), substatements, stmt.wrapSchemaPath());
112     }
113
114     @Override
115     protected OutputEffectiveStatement createUndeclaredEffective(final int flags,
116             final Current<QName, OutputStatement> stmt,
117             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements) {
118         try {
119             return new UndeclaredOutputEffectiveStatement(flags, substatements, stmt.wrapSchemaPath());
120         } catch (SubstatementIndexingException e) {
121             throw new SourceException(e.getMessage(), stmt, e);
122         }
123     }
124 }