Split out yang-data-tree-impl
[yangtools.git] / data / yang-data-tree-ri / src / main / java / org / opendaylight / yangtools / yang / data / tree / leafref / LeafRefContextBuilder.java
diff --git a/data/yang-data-tree-ri/src/main/java/org/opendaylight/yangtools/yang/data/tree/leafref/LeafRefContextBuilder.java b/data/yang-data-tree-ri/src/main/java/org/opendaylight/yangtools/yang/data/tree/leafref/LeafRefContextBuilder.java
new file mode 100644 (file)
index 0000000..06aada4
--- /dev/null
@@ -0,0 +1,148 @@
+/*
+ * 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.data.tree.leafref;
+
+import static java.util.Objects.requireNonNull;
+
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.Map;
+import org.opendaylight.yangtools.concepts.Builder;
+import org.opendaylight.yangtools.yang.common.QName;
+import org.opendaylight.yangtools.yang.common.QNameModule;
+import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
+import org.opendaylight.yangtools.yang.model.api.Module;
+import org.opendaylight.yangtools.yang.model.api.SchemaPath;
+
+final class LeafRefContextBuilder implements Builder<LeafRefContext> {
+
+    private final Map<QName, LeafRefContext> referencingChildren = new HashMap<>();
+    private final Map<QName, LeafRefContext> referencedByChildren = new HashMap<>();
+    private final Map<QName, LeafRefContext> referencedByLeafRefCtx = new HashMap<>();
+
+    private final QName currentNodeQName;
+    private final SchemaPath currentNodePath;
+    private final EffectiveModelContext schemaContext;
+
+    private LeafRefPath leafRefTargetPath = null;
+    private LeafRefPath absoluteLeafRefTargetPath = null;
+    private String leafRefTargetPathString = "";
+
+    private boolean isReferencedBy = false;
+    private boolean isReferencing = false;
+
+    LeafRefContextBuilder(final QName currentNodeQName, final SchemaPath currentNodePath,
+            final EffectiveModelContext schemaContext) {
+        this.currentNodeQName = requireNonNull(currentNodeQName);
+        this.currentNodePath = requireNonNull(currentNodePath);
+        this.schemaContext = requireNonNull(schemaContext);
+    }
+
+    @Override
+    public LeafRefContext build() {
+        final LeafRefContext leafRefContext = new LeafRefContext(this);
+
+        // LeafRefContext has made a copy of these
+        referencingChildren.clear();
+        referencedByChildren.clear();
+        referencedByLeafRefCtx.clear();
+
+        return leafRefContext;
+    }
+
+    boolean isReferencedBy() {
+        return isReferencedBy;
+    }
+
+    void setReferencedBy(final boolean referencedBy) {
+        this.isReferencedBy = referencedBy;
+    }
+
+    boolean isReferencing() {
+        return isReferencing;
+    }
+
+    void setReferencing(final boolean referencing) {
+        this.isReferencing = referencing;
+    }
+
+    void addReferencingChild(final LeafRefContext child, final QName childQName) {
+        referencingChildren.put(childQName, child);
+    }
+
+    Map<QName, LeafRefContext> getReferencingChilds() {
+        return referencingChildren;
+    }
+
+    void addReferencedByChild(final LeafRefContext child, final QName childQName) {
+        referencedByChildren.put(childQName, child);
+    }
+
+    Map<QName, LeafRefContext> getReferencedByChilds() {
+        return referencedByChildren;
+    }
+
+    SchemaPath getCurrentNodePath() {
+        return currentNodePath;
+    }
+
+    LeafRefPath getLeafRefTargetPath() {
+        return leafRefTargetPath;
+    }
+
+    void setLeafRefTargetPath(final LeafRefPath leafRefPath) {
+        this.leafRefTargetPath = requireNonNull(leafRefPath);
+    }
+
+    String getLeafRefTargetPathString() {
+        return leafRefTargetPathString;
+    }
+
+    void setLeafRefTargetPathString(final String leafRefPathString) {
+        this.leafRefTargetPathString = requireNonNull(leafRefPathString);
+    }
+
+    QName getCurrentNodeQName() {
+        return currentNodeQName;
+    }
+
+    EffectiveModelContext getSchemaContext() {
+        return schemaContext;
+    }
+
+    LeafRefPath getAbsoluteLeafRefTargetPath() {
+        if (isReferencing && absoluteLeafRefTargetPath == null) {
+            if (leafRefTargetPath.isAbsolute()) {
+                absoluteLeafRefTargetPath = leafRefTargetPath;
+            } else {
+                absoluteLeafRefTargetPath = LeafRefUtils.createAbsoluteLeafRefPath(leafRefTargetPath,
+                    currentNodePath, getLeafRefContextModule());
+            }
+        }
+
+        return absoluteLeafRefTargetPath;
+    }
+
+    Module getLeafRefContextModule() {
+        final Iterator<QName> it = currentNodePath.getPathFromRoot().iterator();
+        final QNameModule qnameModule = it.hasNext() ? it.next().getModule() : currentNodeQName.getModule();
+        return schemaContext.findModule(qnameModule).orElse(null);
+    }
+
+    void addReferencedByLeafRefCtx(final QName qname, final LeafRefContext leafRef) {
+        referencedByLeafRefCtx.put(qname, leafRef);
+    }
+
+    LeafRefContext getReferencedByLeafRefCtxByName(final QName qname) {
+        return referencedByLeafRefCtx.get(qname);
+    }
+
+    Map<QName, LeafRefContext> getAllReferencedByLeafRefCtxs() {
+        return referencedByLeafRefCtx;
+    }
+}