Introduce UnresolvedQName
[yangtools.git] / parser / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / stmt / module / ModuleEffectiveStatementImpl.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.module;
9
10 import static java.util.Objects.requireNonNull;
11
12 import com.google.common.collect.ImmutableList;
13 import com.google.common.collect.ImmutableMap;
14 import com.google.common.collect.ImmutableMap.Builder;
15 import com.google.common.collect.Maps;
16 import java.util.Collection;
17 import java.util.Map;
18 import java.util.Map.Entry;
19 import java.util.Optional;
20 import java.util.function.Function;
21 import org.eclipse.jdt.annotation.NonNull;
22 import org.opendaylight.yangtools.yang.common.QName;
23 import org.opendaylight.yangtools.yang.common.QNameModule;
24 import org.opendaylight.yangtools.yang.common.UnresolvedQName.Unqualified;
25 import org.opendaylight.yangtools.yang.model.api.Module;
26 import org.opendaylight.yangtools.yang.model.api.Submodule;
27 import org.opendaylight.yangtools.yang.model.api.meta.EffectiveStatement;
28 import org.opendaylight.yangtools.yang.model.api.meta.IdentifierNamespace;
29 import org.opendaylight.yangtools.yang.model.api.stmt.ExtensionEffectiveStatement;
30 import org.opendaylight.yangtools.yang.model.api.stmt.ExtensionEffectiveStatementNamespace;
31 import org.opendaylight.yangtools.yang.model.api.stmt.FeatureEffectiveStatement;
32 import org.opendaylight.yangtools.yang.model.api.stmt.FeatureEffectiveStatementNamespace;
33 import org.opendaylight.yangtools.yang.model.api.stmt.IdentityEffectiveStatement;
34 import org.opendaylight.yangtools.yang.model.api.stmt.IdentityEffectiveStatementNamespace;
35 import org.opendaylight.yangtools.yang.model.api.stmt.ModuleEffectiveStatement;
36 import org.opendaylight.yangtools.yang.model.api.stmt.ModuleStatement;
37 import org.opendaylight.yangtools.yang.model.api.stmt.PrefixEffectiveStatement;
38 import org.opendaylight.yangtools.yang.model.api.stmt.SubmoduleEffectiveStatement;
39 import org.opendaylight.yangtools.yang.parser.rfc7950.stmt.AbstractEffectiveModule;
40 import org.opendaylight.yangtools.yang.parser.spi.meta.EffectiveStmtCtx.Current;
41 import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
42 import org.opendaylight.yangtools.yang.parser.spi.source.IncludedSubmoduleNameToModuleCtx;
43
44 final class ModuleEffectiveStatementImpl extends AbstractEffectiveModule<ModuleStatement, ModuleEffectiveStatement>
45         implements Module, ModuleEffectiveStatement {
46     private final ImmutableMap<String, SubmoduleEffectiveStatement> nameToSubmodule;
47     private final ImmutableMap<QName, ExtensionEffectiveStatement> qnameToExtension;
48     private final ImmutableMap<QName, FeatureEffectiveStatement> qnameToFeature;
49     private final ImmutableMap<QName, IdentityEffectiveStatement> qnameToIdentity;
50     private final ImmutableMap<String, ModuleEffectiveStatement> prefixToModule;
51     private final ImmutableMap<QNameModule, String> namespaceToPrefix;
52     private final @NonNull QNameModule qnameModule;
53     private final ImmutableList<Submodule> submodules;
54
55     ModuleEffectiveStatementImpl(final Current<Unqualified, ModuleStatement> stmt,
56             final ImmutableList<? extends EffectiveStatement<?, ?>> substatements,
57             final Collection<? extends Submodule> submodules, final QNameModule qnameModule) {
58         super(stmt, substatements, findPrefix(stmt, substatements, "module", stmt.getRawArgument()));
59         this.qnameModule = requireNonNull(qnameModule);
60         this.submodules = ImmutableList.copyOf(submodules);
61
62         final String localPrefix = findFirstEffectiveSubstatementArgument(PrefixEffectiveStatement.class).get();
63         final Builder<String, ModuleEffectiveStatement> prefixToModuleBuilder = ImmutableMap.builder();
64         prefixToModuleBuilder.put(localPrefix, this);
65         appendPrefixes(stmt, prefixToModuleBuilder);
66         prefixToModule = prefixToModuleBuilder.build();
67
68         final Map<QNameModule, String> tmp = Maps.newLinkedHashMapWithExpectedSize(prefixToModule.size() + 1);
69         tmp.put(qnameModule, localPrefix);
70         for (Entry<String, ModuleEffectiveStatement> e : prefixToModule.entrySet()) {
71             tmp.putIfAbsent(e.getValue().localQNameModule(), e.getKey());
72         }
73         namespaceToPrefix = ImmutableMap.copyOf(tmp);
74
75         final Map<String, StmtContext<?, ?, ?>> includedSubmodules =
76                 stmt.localNamespacePortion(IncludedSubmoduleNameToModuleCtx.class);
77         nameToSubmodule = includedSubmodules == null ? ImmutableMap.of()
78                 : ImmutableMap.copyOf(Maps.transformValues(includedSubmodules,
79                     submodule -> (SubmoduleEffectiveStatement) submodule.buildEffective()));
80
81         qnameToExtension = substatements.stream()
82             .filter(ExtensionEffectiveStatement.class::isInstance)
83             .map(ExtensionEffectiveStatement.class::cast)
84             .collect(ImmutableMap.toImmutableMap(ExtensionEffectiveStatement::argument, Function.identity()));
85         qnameToFeature = substatements.stream()
86             .filter(FeatureEffectiveStatement.class::isInstance)
87             .map(FeatureEffectiveStatement.class::cast)
88             .collect(ImmutableMap.toImmutableMap(FeatureEffectiveStatement::argument, Function.identity()));
89         qnameToIdentity = substatements.stream()
90             .filter(IdentityEffectiveStatement.class::isInstance)
91             .map(IdentityEffectiveStatement.class::cast)
92             .collect(ImmutableMap.toImmutableMap(IdentityEffectiveStatement::argument, Function.identity()));
93     }
94
95     @Override
96     public QNameModule localQNameModule() {
97         return qnameModule;
98     }
99
100     @Override
101     public QNameModule getQNameModule() {
102         return qnameModule;
103     }
104
105     @Override
106     public Collection<? extends @NonNull Submodule> getSubmodules() {
107         return submodules;
108     }
109
110     @Override
111     public ModuleEffectiveStatement asEffectiveStatement() {
112         return this;
113     }
114
115     @Override
116     @SuppressWarnings("unchecked")
117     public <K, V, N extends IdentifierNamespace<K, V>> Optional<? extends Map<K, V>> getNamespaceContents(
118             final @NonNull Class<N> namespace) {
119         if (PrefixToEffectiveModuleNamespace.class.equals(namespace)) {
120             return Optional.of((Map<K, V>) prefixToModule);
121         }
122         if (QNameModuleToPrefixNamespace.class.equals(namespace)) {
123             return Optional.of((Map<K, V>) namespaceToPrefix);
124         }
125         if (NameToEffectiveSubmoduleNamespace.class.equals(namespace)) {
126             return Optional.of((Map<K, V>) nameToSubmodule);
127         }
128         if (ExtensionEffectiveStatementNamespace.class.equals(namespace)) {
129             return Optional.of((Map<K, V>) qnameToExtension);
130         }
131         if (FeatureEffectiveStatementNamespace.class.equals(namespace)) {
132             return Optional.of((Map<K, V>) qnameToFeature);
133         }
134         if (IdentityEffectiveStatementNamespace.class.equals(namespace)) {
135             return Optional.of((Map<K, V>) qnameToIdentity);
136         }
137         return super.getNamespaceContents(namespace);
138     }
139 }