Update DataTreeFactory
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / stmt / reactor / StatementDefinitionContext.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.reactor;
9
10 import com.google.common.base.MoreObjects;
11 import com.google.common.base.MoreObjects.ToStringHelper;
12 import com.google.common.base.Preconditions;
13 import java.util.HashMap;
14 import java.util.Map;
15 import java.util.Optional;
16 import javax.annotation.Nonnull;
17 import org.opendaylight.yangtools.yang.common.QName;
18 import org.opendaylight.yangtools.yang.common.QNameModule;
19 import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
20 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
21 import org.opendaylight.yangtools.yang.model.api.meta.IdentifierNamespace;
22 import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;
23 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
24 import org.opendaylight.yangtools.yang.parser.spi.meta.StatementFactory;
25 import org.opendaylight.yangtools.yang.parser.spi.meta.StatementSupport;
26 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
27 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext.Mutable;
28
29 public class StatementDefinitionContext<A, D extends DeclaredStatement<A>, E extends EffectiveStatement<A, D>> {
30     private final StatementSupport<A, D, E> support;
31     private final Map<String, StatementDefinitionContext<?, ?, ?>> argumentSpecificSubDefinitions;
32     private Map<StatementDefinitionContext<?,?,?>, StatementDefinitionContext<?,?,?>> unknownStmtDefsOfYangStmts;
33
34     public StatementDefinitionContext(final StatementSupport<A, D, E> support) {
35         this.support = Preconditions.checkNotNull(support);
36         this.argumentSpecificSubDefinitions = support.hasArgumentSpecificSupports() ? new HashMap<>() : null;
37     }
38
39     public StatementFactory<A,D,E> getFactory() {
40         return support;
41     }
42
43     public A parseArgumentValue(final StmtContext<A, D, E> context, final String value) {
44         return support.parseArgumentValue(context, value);
45     }
46
47     public A adaptArgumentValue(final StmtContext<A, D, E> context, final QNameModule targetModule) {
48         return support.adaptArgumentValue(context, targetModule);
49     }
50
51     public void checkNamespaceAllowed(final Class<? extends IdentifierNamespace<?,?>> namespace) {
52         // Noop
53     }
54
55     public StatementDefinition getPublicView() {
56         return support.getPublicView();
57     }
58
59     public Optional<StatementSupport<?, ?, ?>> getImplicitParentFor(final StatementDefinition stmtDef) {
60         return support.getImplicitParentFor(stmtDef);
61     }
62
63     public boolean onStatementAdded(final Mutable<A, D, E> stmt) {
64         support.onStatementAdded(stmt);
65         return false;
66     }
67
68     public void onDeclarationFinished(final Mutable<A, D, E> statement, final ModelProcessingPhase phase) {
69         switch (phase) {
70             case SOURCE_PRE_LINKAGE:
71                 support.onPreLinkageDeclared(statement);
72                 break;
73             case SOURCE_LINKAGE:
74                 support.onLinkageDeclared(statement);
75                 break;
76             case STATEMENT_DEFINITION:
77                 support.onStatementDefinitionDeclared(statement);
78                 break;
79             case FULL_DECLARATION:
80                 support.onFullDefinitionDeclared(statement);
81                 break;
82             default:
83                 break;
84         }
85     }
86
87     public Class<?> getRepresentingClass() {
88         return support.getDeclaredRepresentationClass();
89     }
90
91     public boolean hasArgument() {
92         return support.getArgumentName() != null;
93     }
94
95     public boolean isArgumentYinElement() {
96         return support.isArgumentYinElement();
97     }
98
99     public QName getStatementName() {
100         return support.getStatementName();
101     }
102
103     public QName getArgumentName() {
104         return support.getArgumentName();
105     }
106
107     @Override
108     public final String toString() {
109         return addToStringAttributes(MoreObjects.toStringHelper(this).omitNullValues()).toString();
110     }
111
112     protected ToStringHelper addToStringAttributes(final ToStringHelper toStringHelper) {
113         return toStringHelper.add("statement", getStatementName());
114     }
115
116     @Nonnull
117     StatementDefinitionContext<?, ?, ?> getSubDefinitionSpecificForArgument(final String argument) {
118         if (!hasArgumentSpecificSubDefinitions()) {
119             return this;
120         }
121
122         StatementDefinitionContext<?, ?, ?> potential = argumentSpecificSubDefinitions.get(argument);
123         if (potential == null) {
124             final StatementSupport<?, ?, ?> argumentSpecificSupport = support.getSupportSpecificForArgument(argument);
125             potential = argumentSpecificSupport != null ? new StatementDefinitionContext<>(argumentSpecificSupport)
126                     : this;
127             argumentSpecificSubDefinitions.put(argument, potential);
128         }
129
130         return potential;
131     }
132
133     boolean hasArgumentSpecificSubDefinitions() {
134         return support.hasArgumentSpecificSupports();
135     }
136
137     String internArgument(final String rawArgument) {
138         return support.internArgument(rawArgument);
139     }
140
141     StatementDefinitionContext<?, ?, ?> getAsUnknownStatementDefinition(
142             final StatementDefinitionContext<?, ?, ?> yangStmtDef) {
143         if (unknownStmtDefsOfYangStmts != null) {
144             final StatementDefinitionContext<?, ?, ?> existing = unknownStmtDefsOfYangStmts.get(yangStmtDef);
145             if (existing != null) {
146                 return existing;
147             }
148         } else {
149             unknownStmtDefsOfYangStmts = new HashMap<>();
150         }
151
152         @SuppressWarnings("unchecked")
153         final StatementDefinitionContext<?, ?, ?> ret = support.getUnknownStatementDefinitionOf(
154             yangStmtDef.getPublicView()).map(StatementDefinitionContext::new).orElse(null);
155         if (ret != null) {
156             unknownStmtDefsOfYangStmts.put(yangStmtDef, ret);
157         }
158         return ret;
159     }
160
161     boolean isIgnoringIfFeatures() {
162         return support.isIgnoringIfFeatures();
163     }
164
165     boolean isIgnoringConfig() {
166         return support.isIgnoringConfig();
167     }
168 }