1 files changed, 81 insertions, 0 deletions
diff --git a/netconf/restconf/restconf-nb-bierman02/src/main/java/org/opendaylight/netconf/sal/restconf/impl/DataNormalizer.java b/netconf/restconf/restconf-nb-bierman02/src/main/java/org/opendaylight/netconf/sal/restconf/impl/DataNormalizer.java
new file mode 100644
index 0000000..334a6c7
--- /dev/null
+++ b/netconf/restconf/restconf-nb-bierman02/src/main/java/org/opendaylight/netconf/sal/restconf/impl/DataNormalizer.java
@@ -0,0 +1,81 @@
+/*
+ * Copyright (c) 2014 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.netconf.sal.restconf.impl;
+
+import static com.google.common.base.Preconditions.checkArgument;
+import static java.util.Objects.requireNonNull;
+
+import com.google.common.collect.ImmutableList;
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
+import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
+import org.opendaylight.yangtools.yang.model.util.SchemaInferenceStack;
+
+class DataNormalizer {
+    private final DataNormalizationOperation<?> operation;
+    private final EffectiveModelContext context;
+
+    DataNormalizer(final EffectiveModelContext ctx) {
+        context = requireNonNull(ctx);
+        operation = DataNormalizationOperation.from(ctx);
+    }
+
+    Entry<YangInstanceIdentifier, SchemaInferenceStack> toNormalized(final YangInstanceIdentifier legacy) {
+        List<PathArgument> normalizedArgs = new ArrayList<>();
+
+        DataNormalizationOperation<?> currentOp = operation;
+        Iterator<PathArgument> arguments = legacy.getPathArguments().iterator();
+        SchemaInferenceStack stack = SchemaInferenceStack.of(context);
+
+        try {
+            while (arguments.hasNext()) {
+                PathArgument legacyArg = arguments.next();
+                currentOp = currentOp.enterChild(legacyArg, stack);
+                checkArgument(currentOp != null,
+                        "Legacy Instance Identifier %s is not correct. Normalized Instance Identifier so far %s",
+                        legacy, normalizedArgs);
+                while (currentOp.isMixin()) {
+                    normalizedArgs.add(currentOp.getIdentifier());
+                    currentOp = currentOp.enterChild(legacyArg.getNodeType(), stack);
+                }
+                normalizedArgs.add(legacyArg);
+            }
+        } catch (DataNormalizationException e) {
+            throw new IllegalArgumentException("Failed to normalize path " + legacy, e);
+        }
+
+        return Map.entry(YangInstanceIdentifier.create(normalizedArgs), stack);
+    }
+
+    DataNormalizationOperation<?> getOperation(final YangInstanceIdentifier legacy)
+            throws DataNormalizationException {
+        DataNormalizationOperation<?> currentOp = operation;
+
+        for (PathArgument pathArgument : legacy.getPathArguments()) {
+            currentOp = currentOp.getChild(pathArgument);
+        }
+        return currentOp;
+    }
+
+    YangInstanceIdentifier toLegacy(final YangInstanceIdentifier normalized) throws DataNormalizationException {
+        ImmutableList.Builder<PathArgument> legacyArgs = ImmutableList.builder();
+        DataNormalizationOperation<?> currentOp = operation;
+        for (PathArgument normalizedArg : normalized.getPathArguments()) {
+            currentOp = currentOp.getChild(normalizedArg);
+            if (!currentOp.isMixin()) {
+                legacyArgs.add(normalizedArg);
+            }
+        }
+        return YangInstanceIdentifier.create(legacyArgs.build());
+    }
+}