Fix StmtContext nullness problems
[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / uses / UsesEffectiveStatementImpl.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.rfc7950.stmt.uses;
9
10 import com.google.common.collect.ImmutableList;
11 import com.google.common.collect.ImmutableMap;
12 import com.google.common.collect.ImmutableSet;
13 import java.util.ArrayList;
14 import java.util.HashMap;
15 import java.util.LinkedHashSet;
16 import java.util.List;
17 import java.util.Map;
18 import java.util.Objects;
19 import java.util.Optional;
20 import java.util.Set;
21 import org.eclipse.jdt.annotation.NonNull;
22 import org.eclipse.jdt.annotation.Nullable;
23 import org.opendaylight.yangtools.yang.common.QName;
24 import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
25 import org.opendaylight.yangtools.yang.model.api.RevisionAwareXPath;
26 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
27 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
28 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
29 import org.opendaylight.yangtools.yang.model.api.UsesNode;
30 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
31 import org.opendaylight.yangtools.yang.model.api.stmt.GroupingStatement;
32 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaNodeIdentifier;
33 import org.opendaylight.yangtools.yang.model.api.stmt.UsesEffectiveStatement;
34 import org.opendaylight.yangtools.yang.model.api.stmt.UsesStatement;
35 import org.opendaylight.yangtools.yang.model.api.stmt.WhenEffectiveStatement;
36 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.AbstractEffectiveDocumentedNode;
37 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.refine.RefineEffectiveStatementImpl;
38 import org.opendaylight.yangtools.yang.parser.spi.GroupingNamespace;
39 import org.opendaylight.yangtools.yang.parser.spi.meta.CopyType;
40 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
41
42 final class UsesEffectiveStatementImpl extends AbstractEffectiveDocumentedNode<QName, UsesStatement>
43         implements UsesEffectiveStatement, UsesNode {
44     private final @NonNull SchemaPath groupingPath;
45     private final boolean addedByUses;
46     private final @NonNull Map<SchemaPath, SchemaNode> refines;
47     private final @NonNull Set<AugmentationSchemaNode> augmentations;
48     private final @NonNull List<UnknownSchemaNode> unknownNodes;
49     private final @Nullable RevisionAwareXPath whenCondition;
50
51     UsesEffectiveStatementImpl(final StmtContext<QName, UsesStatement, EffectiveStatement<QName, UsesStatement>> ctx) {
52         super(ctx);
53
54         // initGroupingPath
55         final StmtContext<?, GroupingStatement, EffectiveStatement<QName, GroupingStatement>> grpCtx =
56                 ctx.getFromNamespace(GroupingNamespace.class, ctx.coerceStatementArgument());
57         this.groupingPath = grpCtx.getSchemaPath().get();
58
59         // initCopyType
60         addedByUses = ctx.getCopyHistory().contains(CopyType.ADDED_BY_USES);
61
62         // initSubstatementCollections
63         final List<UnknownSchemaNode> unknownNodesInit = new ArrayList<>();
64         final Set<AugmentationSchemaNode> augmentationsInit = new LinkedHashSet<>();
65         final Map<SchemaPath, SchemaNode> refinesInit = new HashMap<>();
66         for (final EffectiveStatement<?, ?> effectiveStatement : effectiveSubstatements()) {
67             if (effectiveStatement instanceof UnknownSchemaNode) {
68                 final UnknownSchemaNode unknownNode = (UnknownSchemaNode) effectiveStatement;
69                 unknownNodesInit.add(unknownNode);
70             }
71             if (effectiveStatement instanceof AugmentationSchemaNode) {
72                 final AugmentationSchemaNode augmentationSchema = (AugmentationSchemaNode) effectiveStatement;
73                 augmentationsInit.add(augmentationSchema);
74             }
75             if (effectiveStatement instanceof RefineEffectiveStatementImpl) {
76                 final RefineEffectiveStatementImpl refineStmt = (RefineEffectiveStatementImpl) effectiveStatement;
77                 final SchemaNodeIdentifier identifier = refineStmt.argument();
78                 refinesInit.put(identifier.asSchemaPath(), refineStmt.getRefineTargetNode());
79             }
80         }
81         this.unknownNodes = ImmutableList.copyOf(unknownNodesInit);
82         this.augmentations = ImmutableSet.copyOf(augmentationsInit);
83         this.refines = ImmutableMap.copyOf(refinesInit);
84
85         whenCondition = findFirstEffectiveSubstatementArgument(WhenEffectiveStatement.class).orElse(null);
86     }
87
88     @Override
89     public SchemaPath getGroupingPath() {
90         return groupingPath;
91     }
92
93     @Override
94     public Set<AugmentationSchemaNode> getAugmentations() {
95         return augmentations;
96     }
97
98     @Deprecated
99     @Override
100     public boolean isAugmenting() {
101         return false;
102     }
103
104     @Deprecated
105     @Override
106     public boolean isAddedByUses() {
107         return addedByUses;
108     }
109
110     @Override
111     public Map<SchemaPath, SchemaNode> getRefines() {
112         return refines;
113     }
114
115     @Override
116     public List<UnknownSchemaNode> getUnknownSchemaNodes() {
117         return unknownNodes;
118     }
119
120     @Override
121     public Optional<RevisionAwareXPath> getWhenCondition() {
122         return Optional.ofNullable(whenCondition);
123     }
124
125     @Override
126     public int hashCode() {
127         final int prime = 31;
128         int result = 1;
129         result = prime * result + Objects.hashCode(groupingPath);
130         result = prime * result + Objects.hashCode(augmentations);
131         return result;
132     }
133
134     @Override
135     public boolean equals(final Object obj) {
136         if (this == obj) {
137             return true;
138         }
139         if (obj == null) {
140             return false;
141         }
142         if (getClass() != obj.getClass()) {
143             return false;
144         }
145         final UsesEffectiveStatementImpl other = (UsesEffectiveStatementImpl) obj;
146         return Objects.equals(groupingPath, other.groupingPath) && Objects.equals(augmentations, other.augmentations);
147     }
148
149     @Override
150     public String toString() {
151         return UsesEffectiveStatementImpl.class.getSimpleName() + "[groupingPath=" + groupingPath + "]";
152     }
153 }