BUG-7052: reduce StatementContextBase proliferation
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / rfc6020 / AnyxmlStatementImpl.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 com.google.common.base.Optional;
11 import java.util.Collection;
12 import java.util.Map;
13 import javax.annotation.Nonnull;
14 import org.opendaylight.yangtools.yang.common.QName;
15 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
16 import org.opendaylight.yangtools.yang.model.api.YangStmtMapping;
17 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
18 import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;
19 import org.opendaylight.yangtools.yang.model.api.stmt.AnyxmlStatement;
20 import org.opendaylight.yangtools.yang.model.api.stmt.ConfigStatement;
21 import org.opendaylight.yangtools.yang.model.api.stmt.DescriptionStatement;
22 import org.opendaylight.yangtools.yang.model.api.stmt.IfFeatureStatement;
23 import org.opendaylight.yangtools.yang.model.api.stmt.MandatoryStatement;
24 import org.opendaylight.yangtools.yang.model.api.stmt.MustStatement;
25 import org.opendaylight.yangtools.yang.model.api.stmt.ReferenceStatement;
26 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier;
27 import org.opendaylight.yangtools.yang.model.api.stmt.StatusStatement;
28 import org.opendaylight.yangtools.yang.model.api.stmt.UnknownStatement;
29 import org.opendaylight.yangtools.yang.model.api.stmt.WhenStatement;
30 import org.opendaylight.yangtools.yang.parser.spi.SubstatementValidator;
31 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractDeclaredStatement;
32 import org.opendaylight.yangtools.yang.parser.spi.meta.AbstractStatementSupport;
33 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
34 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
35 import org.opendaylight.yangtools.yang.parser.spi.source.AnyxmlSchemaLocationNamespace;
36 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.AnyXmlEffectiveStatementImpl;
37 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.YangModeledAnyXmlEffectiveStatementImpl;
38
39 public class AnyxmlStatementImpl extends AbstractDeclaredStatement<QName> implements AnyxmlStatement {
40     private static final SubstatementValidator SUBSTATEMENT_VALIDATOR = SubstatementValidator.builder(YangStmtMapping
41             .ANYXML)
42             .addOptional(YangStmtMapping.CONFIG)
43             .addOptional(YangStmtMapping.DESCRIPTION)
44             .addAny(YangStmtMapping.IF_FEATURE)
45             .addOptional(YangStmtMapping.MANDATORY)
46             .addAny(YangStmtMapping.MUST)
47             .addOptional(YangStmtMapping.REFERENCE)
48             .addOptional(YangStmtMapping.STATUS)
49             .addOptional(YangStmtMapping.WHEN)
50             .addOptional(SupportedExtensionsMapping.ANYXML_SCHEMA_LOCATION)
51             .build();
52
53     protected AnyxmlStatementImpl(final StmtContext<QName, AnyxmlStatement, ?> context) {
54         super(context);
55     }
56
57     public static class Definition extends
58             AbstractStatementSupport<QName, AnyxmlStatement, EffectiveStatement<QName, AnyxmlStatement>> {
59
60         public Definition() {
61             super(YangStmtMapping.ANYXML);
62         }
63
64         @Override
65         public QName parseArgumentValue(final StmtContext<?, ?, ?> ctx, final String value) {
66             return Utils.qNameFromArgument(ctx, value);
67         }
68
69         @Override
70         public void onStatementAdded(final Mutable<QName, AnyxmlStatement, EffectiveStatement<QName, AnyxmlStatement>> stmt) {
71             stmt.getParentContext().addToNs(ChildSchemaNodes.class, stmt.getStatementArgument(), stmt);
72         }
73
74         @Override
75         public AnyxmlStatement createDeclared(final StmtContext<QName, AnyxmlStatement, ?> ctx) {
76             return new AnyxmlStatementImpl(ctx);
77         }
78
79         @Override
80         public EffectiveStatement<QName, AnyxmlStatement> createEffective(
81                 final StmtContext<QName, AnyxmlStatement, EffectiveStatement<QName, AnyxmlStatement>> ctx) {
82             final Map<StatementDefinition, Mutable<SchemaNodeIdentifier, UnknownStatement<SchemaNodeIdentifier>, EffectiveStatement<SchemaNodeIdentifier, UnknownStatement<SchemaNodeIdentifier>>>> schemaLocations = ctx
83                     .getAllFromCurrentStmtCtxNamespace(AnyxmlSchemaLocationNamespace.class);
84             if (schemaLocations != null && !schemaLocations.isEmpty()) {
85                 final SchemaNodeIdentifier anyXmlSchemaNodeIdentifier = schemaLocations.values().iterator().next()
86                         .getStatementArgument();
87                 final Optional<ContainerSchemaNode> anyXmlSchema = getAnyXmlSchema(ctx, anyXmlSchemaNodeIdentifier);
88                 if (anyXmlSchema.isPresent()) {
89                     return new YangModeledAnyXmlEffectiveStatementImpl(ctx, anyXmlSchema.get());
90                 }
91             }
92             return new AnyXmlEffectiveStatementImpl(ctx);
93         }
94
95         private static Optional<ContainerSchemaNode> getAnyXmlSchema(
96                 final StmtContext<QName, AnyxmlStatement, EffectiveStatement<QName, AnyxmlStatement>> ctx,
97                 final SchemaNodeIdentifier contentSchemaPath) {
98             final StmtContext<?, ?, ?> findNode = Utils.findNode(ctx.getRoot(), contentSchemaPath);
99             if (findNode != null) {
100                 final EffectiveStatement<?, ?> anyXmlSchemaNode = findNode.buildEffective();
101                 if (anyXmlSchemaNode instanceof ContainerSchemaNode) {
102                     return Optional.of((ContainerSchemaNode) anyXmlSchemaNode);
103                 }
104             }
105             return Optional.absent();
106         }
107
108         @Override
109         protected SubstatementValidator getSubstatementValidator() {
110             return SUBSTATEMENT_VALIDATOR;
111         }
112     }
113
114     @Nonnull
115     @Override
116     public QName getName() {
117         return argument();
118     }
119
120     @Override
121     public WhenStatement getWhenStatement() {
122         return firstDeclared(WhenStatement.class);
123     }
124
125     @Nonnull
126     @Override
127     public Collection<? extends IfFeatureStatement> getIfFeatures() {
128         return allDeclared(IfFeatureStatement.class);
129     }
130
131     @Nonnull
132     @Override
133     public Collection<? extends MustStatement> getMusts() {
134         return allDeclared(MustStatement.class);
135     }
136
137     @Override
138     public ConfigStatement getConfig() {
139         return firstDeclared(ConfigStatement.class);
140     }
141
142     @Override
143     public StatusStatement getStatus() {
144         return firstDeclared(StatusStatement.class);
145     }
146
147     @Override
148     public DescriptionStatement getDescription() {
149         return firstDeclared(DescriptionStatement.class);
150     }
151
152     @Override
153     public ReferenceStatement getReference() {
154         return firstDeclared(ReferenceStatement.class);
155     }
156
157     @Override
158     public MandatoryStatement getMandatory() {
159         return firstDeclared(MandatoryStatement.class);
160     }
161
162 }