Disconnect StmtNamespaceContext from statement

[yangtools.git] / yang / yang-parser-rfc7950 / src / main / java / org / opendaylight / yangtools / yang / parser / rfc7950 / namespace / StmtNamespaceContext.java

diff --git a/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/namespace/StmtNamespaceContext.java b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/namespace/StmtNamespaceContext.java
new file mode 100644 (file)
index 0000000..38bd6c1
--- /dev/null
+++ b/yang/yang-parser-rfc7950/src/main/java/org/opendaylight/yangtools/yang/parser/rfc7950/namespace/StmtNamespaceContext.java
@@ -0,0 +1,75 @@
+/*
+ * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.yangtools.yang.parser.rfc7950.namespace;
+
+import com.google.common.collect.ImmutableBiMap;
+import com.google.common.collect.ImmutableMap;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Optional;
+import javax.xml.namespace.NamespaceContext;
+import org.opendaylight.yangtools.yang.common.QNameModule;
+import org.opendaylight.yangtools.yang.common.YangNamespaceContext;
+import org.opendaylight.yangtools.yang.model.api.stmt.SubmoduleStatement;
+import org.opendaylight.yangtools.yang.parser.spi.meta.StmtContext;
+import org.opendaylight.yangtools.yang.parser.spi.source.BelongsToPrefixToModuleName;
+import org.opendaylight.yangtools.yang.parser.spi.source.ImportPrefixToModuleCtx;
+import org.opendaylight.yangtools.yang.parser.spi.source.ModuleCtxToModuleQName;
+import org.opendaylight.yangtools.yang.parser.spi.source.ModuleNameToModuleQName;
+
+/**
+ * A {@link NamespaceContext} implementation based on the set of imports and local module namespace.
+ */
+// TODO: this is a useful utility, so it may be useful to expose it either in this package, or yang.parser.spi.source
+final class StmtNamespaceContext implements YangNamespaceContext {
+    private static final long serialVersionUID = 1L;
+
+    private final ImmutableBiMap<QNameModule, String> moduleToPrefix;
+    private final ImmutableMap<String, QNameModule> prefixToModule;
+
+    StmtNamespaceContext(final StmtContext<?, ?, ?> ctx) {
+        // QNameModule -> prefix mappings
+        final Map<QNameModule, String> qnameToPrefix = ctx.getAllFromNamespace(ModuleQNameToPrefix.class);
+        this.moduleToPrefix = qnameToPrefix == null ? ImmutableBiMap.of() : ImmutableBiMap.copyOf(qnameToPrefix);
+
+        // Additional mappings
+        final Map<String, StmtContext<?, ?, ?>> imports = ctx.getAllFromNamespace(ImportPrefixToModuleCtx.class);
+        if (imports != null) {
+            final Map<String, QNameModule> additional = new HashMap<>();
+            for (Entry<String, StmtContext<?, ?, ?>> entry : imports.entrySet()) {
+                if (!moduleToPrefix.containsValue(entry.getKey())) {
+                    QNameModule qnameModule = ctx.getFromNamespace(ModuleCtxToModuleQName.class, entry.getValue());
+                    if (qnameModule == null && ctx.producesDeclared(SubmoduleStatement.class)) {
+                        final String moduleName = ctx.getFromNamespace(BelongsToPrefixToModuleName.class,
+                            entry.getKey());
+                        qnameModule = ctx.getFromNamespace(ModuleNameToModuleQName.class, moduleName);
+                    }
+
+                    if (qnameModule != null) {
+                        additional.put(entry.getKey(), qnameModule);
+                    }
+                }
+            }
+            this.prefixToModule = ImmutableMap.copyOf(additional);
+        } else {
+            this.prefixToModule = ImmutableMap.of();
+        }
+    }
+
+    @Override
+    public Optional<String> findPrefixForNamespace(final QNameModule namespace) {
+        return Optional.ofNullable(moduleToPrefix.get(namespace));
+    }
+
+    @Override
+    public Optional<QNameModule> findNamespaceForPrefix(final String prefix) {
+        final QNameModule normal = moduleToPrefix.inverse().get(prefix);
+        return normal != null ? Optional.of(normal) : Optional.ofNullable(prefixToModule.get(prefix));
+    }
+}