Migrate getDataChildByName() users
[yangtools.git] / yang / yang-model-util / src / main / java / org / opendaylight / yangtools / yang / model / util / SchemaContextUtil.java
index 0dc0c19e6b19d2b7e539052d2c15bab45848a6b7..73bd1765d1d0cc121702ee110b85075647bec6e5 100644 (file)
  */
 package org.opendaylight.yangtools.yang.model.util;
 
-import java.net.URI;
+import static com.google.common.base.Preconditions.checkArgument;
+import static com.google.common.base.Preconditions.checkState;
+import static java.util.Objects.requireNonNull;
+
+import com.google.common.annotations.Beta;
+import com.google.common.annotations.VisibleForTesting;
+import com.google.common.base.Splitter;
+import com.google.common.collect.Iterables;
+import com.google.common.collect.Lists;
+import java.util.ArrayDeque;
 import java.util.ArrayList;
 import java.util.Arrays;
-import java.util.Collection;
-import java.util.Collections;
-import java.util.Date;
-import java.util.LinkedList;
+import java.util.Deque;
+import java.util.HashSet;
+import java.util.Iterator;
 import java.util.List;
+import java.util.Optional;
 import java.util.Set;
-
+import java.util.regex.Pattern;
+import java.util.stream.Collectors;
+import org.eclipse.jdt.annotation.NonNull;
+import org.eclipse.jdt.annotation.Nullable;
+import org.opendaylight.yangtools.yang.common.AbstractQName;
 import org.opendaylight.yangtools.yang.common.QName;
-import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;
-import org.opendaylight.yangtools.yang.model.api.AugmentationTarget;
-import org.opendaylight.yangtools.yang.model.api.ChoiceCaseNode;
-import org.opendaylight.yangtools.yang.model.api.ChoiceNode;
-import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
+import org.opendaylight.yangtools.yang.common.QNameModule;
+import org.opendaylight.yangtools.yang.common.UnqualifiedQName;
+import org.opendaylight.yangtools.yang.model.api.ActionNodeContainer;
+import org.opendaylight.yangtools.yang.model.api.CaseSchemaNode;
+import org.opendaylight.yangtools.yang.model.api.ChoiceSchemaNode;
+import org.opendaylight.yangtools.yang.model.api.ContainerLike;
 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
+import org.opendaylight.yangtools.yang.model.api.DerivableSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.GroupingDefinition;
-import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
+import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.Module;
 import org.opendaylight.yangtools.yang.model.api.ModuleImport;
+import org.opendaylight.yangtools.yang.model.api.ModuleLike;
 import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
-import org.opendaylight.yangtools.yang.model.api.RevisionAwareXPath;
+import org.opendaylight.yangtools.yang.model.api.NotificationNodeContainer;
+import org.opendaylight.yangtools.yang.model.api.OperationDefinition;
+import org.opendaylight.yangtools.yang.model.api.PathExpression;
+import org.opendaylight.yangtools.yang.model.api.PathExpression.DerefSteps;
+import org.opendaylight.yangtools.yang.model.api.PathExpression.LocationPathSteps;
+import org.opendaylight.yangtools.yang.model.api.PathExpression.Steps;
 import org.opendaylight.yangtools.yang.model.api.RpcDefinition;
 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
-import org.opendaylight.yangtools.yang.model.api.UsesNode;
+import org.opendaylight.yangtools.yang.model.api.Submodule;
+import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
+import org.opendaylight.yangtools.yang.model.api.TypedDataSchemaNode;
+import org.opendaylight.yangtools.yang.model.api.type.InstanceIdentifierTypeDefinition;
+import org.opendaylight.yangtools.yang.model.api.type.LeafrefTypeDefinition;
+import org.opendaylight.yangtools.yang.model.repo.api.RevisionSourceIdentifier;
+import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
+import org.opendaylight.yangtools.yang.xpath.api.YangLocationPath;
+import org.opendaylight.yangtools.yang.xpath.api.YangLocationPath.AxisStep;
+import org.opendaylight.yangtools.yang.xpath.api.YangLocationPath.QNameStep;
+import org.opendaylight.yangtools.yang.xpath.api.YangLocationPath.Step;
+import org.opendaylight.yangtools.yang.xpath.api.YangXPathAxis;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-import com.google.common.base.Preconditions;
-
 /**
- * The Schema Context Util contains support methods for searching through Schema
- * Context modules for specified schema nodes via Schema Path or Revision Aware
- * XPath. The Schema Context Util is designed as mixin, so it is not
+ * The Schema Context Util contains support methods for searching through Schema Context modules for specified schema
+ * nodes via Schema Path or Revision Aware XPath. The Schema Context Util is designed as mixin, so it is not
  * instantiable.
- *
  */
 public final class SchemaContextUtil {
     private static final Logger LOG = LoggerFactory.getLogger(SchemaContextUtil.class);
+    private static final Splitter COLON_SPLITTER = Splitter.on(':');
+    private static final Splitter SLASH_SPLITTER = Splitter.on('/').omitEmptyStrings();
+    private static final Pattern GROUPS_PATTERN = Pattern.compile("\\[(.*?)\\]");
 
     private SchemaContextUtil() {
+        // Hidden on purpose
     }
 
     /**
-     * Method attempts to find DataSchemaNode in Schema Context via specified
-     * Schema Path. The returned DataSchemaNode from method will be the node at
-     * the end of the SchemaPath. If the DataSchemaNode is not present in the
-     * Schema Context the method will return <code>null</code>. <br>
-     * In case that Schema Context or Schema Path are not specified correctly
-     * (i.e. contains <code>null</code> values) the method will return
-     * IllegalArgumentException.
+     * Method attempts to find DataSchemaNode in Schema Context via specified Schema Path. The returned DataSchemaNode
+     * from method will be the node at the end of the SchemaPath. If the DataSchemaNode is not present in the Schema
+     * Context the method will return {@code null}.
      *
-     * @throws IllegalArgumentException
+     * <p>
+     * In case that Schema Context or Schema Path are not specified correctly (i.e. contains {@code null} values) the
+     * method will throw IllegalArgumentException.
      *
-     * @param context
-     *            Schema Context
-     * @param schemaPath
-     *            Schema Path to search for
-     * @return SchemaNode from the end of the Schema Path or <code>null</code>
-     *         if the Node is not present.
+     * @param context Schema Context
+     * @param schemaPath Schema Path to search for
+     * @return SchemaNode from the end of the Schema Path or {@code null} if the Node is not present.
+     * @throws NullPointerException if context or schemaPath is null
      */
     public static SchemaNode findDataSchemaNode(final SchemaContext context, final SchemaPath schemaPath) {
-        Preconditions.checkArgument(context != null, "Schema Context reference cannot be NULL");
-        Preconditions.checkArgument(schemaPath != null, "Schema Path reference cannot be NULL");
-
-        final List<QName> prefixedPath = (schemaPath.getPath());
+        final Iterable<QName> prefixedPath = schemaPath.getPathFromRoot();
         if (prefixedPath == null) {
             LOG.debug("Schema path {} has null path", schemaPath);
             return null;
@@ -86,6 +111,35 @@ public final class SchemaContextUtil {
         return findNodeInSchemaContext(context, prefixedPath);
     }
 
+    /**
+     * Attempt to find a DataSchemaNode based on its path from root, similar to
+     * {@link #findDataSchemaNode(SchemaContext, Module, PathExpression)} without requiring an expression.
+     *
+     * @param context Schema Context
+     * @param path Path to search for
+     * @return SchemaNode from the end of the Schema Path or {@code null} if the Node is not present.
+     * @throws NullPointerException if a any argument is null or if the path contains a null element
+     */
+    @Beta
+    public static SchemaNode findDataSchemaNode(final SchemaContext context, final List<QName> path) {
+        return findTargetNode(context, null, YangLocationPath.absolute(
+            path.stream().map(YangXPathAxis.CHILD::asStep).collect(Collectors.toList())));
+    }
+
+    /**
+     * Attempt to find a DataSchemaNode based on its path from root, similar to
+     * {@link #findDataSchemaNode(SchemaContext, Module, PathExpression)} without requiring an expression.
+     *
+     * @param context Schema Context
+     * @param path Path to search for
+     * @return SchemaNode from the end of the Schema Path or {@code null} if the Node is not present.
+     * @throws NullPointerException if a any argument is null or if the path contains a null element
+     */
+    @Beta
+    public static SchemaNode findDataSchemaNode(final SchemaContext context, final QName... path) {
+        return findDataSchemaNode(context, Arrays.asList(path));
+    }
+
     /**
      * Method attempts to find DataSchemaNode inside of provided Schema Context
      * and Yang Module accordingly to Non-conditional Revision Aware XPath. The
@@ -96,16 +150,11 @@ public final class SchemaContextUtil {
      * Context is completely unaware of data state and will be not able to
      * properly resolve XPath. If the XPath contains condition the method will
      * return IllegalArgumentException. <br>
-     * In case that Schema Context or Module or Revision Aware XPath contains
-     * <code>null</code> references the method will throw
-     * IllegalArgumentException <br>
      * If the Revision Aware XPath is correct and desired Data Schema Node is
      * present in Yang module or in depending module in Schema Context the
      * method will return specified Data Schema Node, otherwise the operation
      * will fail and method will return <code>null</code>.
      *
-     * @throws IllegalArgumentException
-     *
      * @param context
      *            Schema Context
      * @param module
@@ -115,25 +164,58 @@ public final class SchemaContextUtil {
      * @return Returns Data Schema Node for specified Schema Context for given
      *         Non-conditional Revision Aware XPath, or <code>null</code> if the
      *         DataSchemaNode is not present in Schema Context.
+     * @throws NullPointerException if any of the arguments is null
+     * @deprecated Use {@link #findDataTreeSchemaNode(SchemaContext, QNameModule, YangLocationPath)} or
+     *             {@link #findDataTreeSchemaNode(SchemaContext, QNameModule, PathExpression)} instead.
      */
-    public static SchemaNode findDataSchemaNode(final SchemaContext context, final Module module, final RevisionAwareXPath nonCondXPath) {
-        Preconditions.checkArgument(context != null, "Schema Context reference cannot be NULL");
-        Preconditions.checkArgument(module != null, "Module reference cannot be NULL");
-        Preconditions.checkArgument(nonCondXPath != null, "Non Conditional Revision Aware XPath cannot be NULL");
-
-        String strXPath = nonCondXPath.toString();
-        if (strXPath != null) {
-            Preconditions.checkArgument(strXPath.indexOf('[') == -1, "Revision Aware XPath may not contain a condition");
-            if (nonCondXPath.isAbsolute()) {
-                List<QName> qnamedPath = xpathToQNamePath(context, module, strXPath);
-                if (qnamedPath != null) {
-                    return findNodeInSchemaContext(context, qnamedPath);
-                }
-            }
+    // FIXME: This entire method is ill-defined, as the resolution process depends on  where the XPath is defined --
+    //        notably RPCs, actions and notifications modify the data tree temporarily. See sections 6.4.1 and 9.9.2
+    //        of RFC7950.
+    //
+    //        Most notably we need to understand whether the XPath is being resolved in the data tree, or as part of
+    //        a notification/action/RPC, as then the SchemaContext grows tentative nodes ... which could be addressed
+    //        via a derived SchemaContext (i.e. this class would have to have a
+    //
+    //            SchemaContext notificationSchemaContext(SchemaContext delegate, NotificationDefinition notif)
+    //
+    //        which would then be passed in to a method similar to this one. In static contexts, like MD-SAL codegen,
+    //        that feels like an overkill.
+    @Deprecated
+    public static SchemaNode findDataSchemaNode(final SchemaContext context, final Module module,
+            final PathExpression nonCondXPath) {
+        requireNonNull(context, "context");
+        requireNonNull(module, "module");
+
+        final String strXPath = nonCondXPath.getOriginalString();
+        checkArgument(strXPath.indexOf('[') == -1, "Revision Aware XPath may not contain a condition");
+        if (nonCondXPath.isAbsolute()) {
+            return findTargetNode(context, xpathToQNamePath(context, module, strXPath));
         }
         return null;
     }
 
+    @Beta
+    public static SchemaNode findDataTreeSchemaNode(final SchemaContext ctx, final QNameModule localModule,
+            final YangLocationPath absPath) {
+        checkArgument(absPath.isAbsolute(), "Unsupported relative path %s", absPath);
+        return findTargetNode(ctx, localModule, absPath);
+    }
+
+    @Beta
+    public static SchemaNode findDataTreeSchemaNode(final SchemaContext ctx, final QNameModule localModule,
+            final PathExpression absPath) {
+        final Steps pathSteps = absPath.getSteps();
+        if (pathSteps instanceof LocationPathSteps) {
+            return findDataTreeSchemaNode(ctx, localModule, ((LocationPathSteps) pathSteps).getLocationPath());
+        }
+
+        // We would need a reference schema node and no, we do not want to use SchemaContext in its SchemaNode capacity
+        checkArgument(!(pathSteps instanceof DerefSteps), "No reference node for steps %s", pathSteps);
+
+        // We are missing proper API alignment, if this ever triggers
+        throw new IllegalStateException("Unsupported path " + pathSteps);
+    }
+
     /**
      * Method attempts to find DataSchemaNode inside of provided Schema Context
      * and Yang Module accordingly to Non-conditional relative Revision Aware
@@ -148,9 +230,6 @@ public final class SchemaContextUtil {
      * Schema Node will search starts. If the Actual Schema Node is not correct
      * the operation will simply fail, because it will be unable to find desired
      * DataSchemaNode. <br>
-     * In case that Schema Context or Module or Actual Schema Node or relative
-     * Revision Aware XPath contains <code>null</code> references the method
-     * will throw IllegalArgumentException <br>
      * If the Revision Aware XPath doesn't have flag
      * <code>isAbsolute == false</code> the method will throw
      * IllegalArgumentException. <br>
@@ -159,8 +238,6 @@ public final class SchemaContextUtil {
      * the method will return specified Data Schema Node, otherwise the
      * operation will fail and method will return <code>null</code>.
      *
-     * @throws IllegalArgumentException
-     *
      * @param context
      *            Schema Context
      * @param module
@@ -172,542 +249,312 @@ public final class SchemaContextUtil {
      * @return DataSchemaNode if is present in specified Schema Context for
      *         given relative Revision Aware XPath, otherwise will return
      *         <code>null</code>.
+     * @throws NullPointerException if any argument is null
      */
+    // FIXME: This entire method is ill-defined, as the resolution process depends on  where the XPath is defined --
+    //        notably RPCs, actions and notifications modify the data tree temporarily. See sections 6.4.1 and 9.9.2
+    //        of RFC7950.
+    //
+    //        Most notably we need to understand whether the XPath is being resolved in the data tree, or as part of
+    //        a notification/action/RPC, as then the SchemaContext grows tentative nodes ... which could be addressed
+    //        via a derived SchemaContext (i.e. this class would have to have a
+    //
+    //            SchemaContext notificationSchemaContext(SchemaContext delegate, NotificationDefinition notif)
+    //
+    //        which would then be passed in to a method similar to this one. In static contexts, like MD-SAL codegen,
+    //        that feels like an overkill.
+    // FIXME: YANGTOOLS-1052: this is a static analysis util, move it to a dedicated class
     public static SchemaNode findDataSchemaNodeForRelativeXPath(final SchemaContext context, final Module module,
-            final SchemaNode actualSchemaNode, final RevisionAwareXPath relativeXPath) {
-        Preconditions.checkArgument(context != null, "Schema Context reference cannot be NULL");
-        Preconditions.checkArgument(module != null, "Module reference cannot be NULL");
-        Preconditions.checkArgument(actualSchemaNode != null, "Actual Schema Node reference cannot be NULL");
-        Preconditions.checkArgument(relativeXPath != null, "Non Conditional Revision Aware XPath cannot be NULL");
-        Preconditions.checkState(!relativeXPath.isAbsolute(),
-                "Revision Aware XPath MUST be relative i.e. MUST contains ../, "
-                        + "for non relative Revision Aware XPath use findDataSchemaNode method");
-
-        SchemaPath actualNodePath = actualSchemaNode.getPath();
-        if (actualNodePath != null) {
-            List<QName> qnamePath = resolveRelativeXPath(context, module, relativeXPath, actualSchemaNode);
-
-            if (qnamePath != null) {
-                return findNodeInSchemaContext(context, qnamePath);
-            }
-        }
-        return null;
+            final SchemaNode actualSchemaNode, final PathExpression relativeXPath) {
+        checkState(!relativeXPath.isAbsolute(), "Revision Aware XPath MUST be relative i.e. MUST contains ../, "
+                + "for non relative Revision Aware XPath use findDataSchemaNode method");
+        return resolveRelativeXPath(context, module, removePredicatesFromXpath(relativeXPath.getOriginalString()),
+                actualSchemaNode);
+    }
+
+    private static String removePredicatesFromXpath(final String xpath) {
+        return GROUPS_PATTERN.matcher(xpath).replaceAll("");
     }
 
     /**
      * Returns parent Yang Module for specified Schema Context in which Schema
      * Node is declared. If the Schema Node is not present in Schema Context the
-     * operation will return <code>null</code>. <br>
-     * If Schema Context or Schema Node contains <code>null</code> references
-     * the method will throw IllegalArgumentException
+     * operation will return <code>null</code>.
      *
-     * @throws IllegalArgumentException
-     *
-     * @param context
-     *            Schema Context
-     * @param schemaNode
-     *            Schema Node
-     * @return Yang Module for specified Schema Context and Schema Node, if
-     *         Schema Node is NOT present, the method will returns
-     *         <code>null</code>
+     * @param context Schema Context
+     * @param schemaNode Schema Node
+     * @return Yang Module for specified Schema Context and Schema Node, if Schema Node is NOT present, the method will
+     *         return <code>null</code>
+     * @throws NullPointerException if any of the arguments is null
      */
     public static Module findParentModule(final SchemaContext context, final SchemaNode schemaNode) {
-        Preconditions.checkArgument(context != null, "Schema Context reference cannot be NULL!");
-        Preconditions.checkArgument(schemaNode != null, "Schema Node cannot be NULL!");
-        Preconditions.checkState(schemaNode.getPath() != null, "Schema Path for Schema Node is not "
-                + "set properly (Schema Path is NULL)");
-
-        List<QName> qnamedPath = schemaNode.getPath().getPath();
-        if (qnamedPath == null || qnamedPath.isEmpty()) {
-            throw new IllegalStateException("Schema Path contains invalid state of path parts."
-                    + "The Schema Path MUST contain at least ONE QName which defines namespace and Local name"
-                    + "of path.");
-        }
-        QName qname = qnamedPath.get(qnamedPath.size() - 1);
-        return context.findModuleByNamespaceAndRevision(qname.getNamespace(), qname.getRevision());
+        final QName qname = schemaNode.getPath().getLastComponent();
+        checkState(qname != null, "Schema Path contains invalid state of path parts. "
+                + "The Schema Path MUST contain at least ONE QName  which defines namespace and Local name of path.");
+        return context.findModule(qname.getModule()).orElse(null);
     }
 
-    public static SchemaNode findNodeInSchemaContext(final SchemaContext context, final List<QName> path) {
-        final QName current = path.get(0);
+    public static SchemaNode findNodeInSchemaContext(final SchemaContext context, final Iterable<QName> path) {
+        final QName current = path.iterator().next();
 
         LOG.trace("Looking up module {} in context {}", current, path);
-        final Module module = context.findModuleByNamespaceAndRevision(current.getNamespace(), current.getRevision());
-        if (module == null) {
+        final Optional<Module> module = context.findModule(current.getModule());
+        if (module.isEmpty()) {
             LOG.debug("Module {} not found", current);
             return null;
         }
 
-        return findNodeInModule(module, path);
+        return findNodeInModule(module.get(), path);
     }
 
-    public static GroupingDefinition findGrouping(final SchemaContext context, final Module module, final List<QName> path) {
-        QName first = path.get(0);
-        Module m = context.findModuleByNamespace(first.getNamespace()).iterator().next();
-        DataNodeContainer currentParent = m;
-        for (QName qname : path) {
-            boolean found = false;
-            DataNodeContainer node = (DataNodeContainer) currentParent.getDataChildByName(qname.getLocalName());
-            if (node == null) {
-                Set<GroupingDefinition> groupings = currentParent.getGroupings();
-                for (GroupingDefinition gr : groupings) {
-                    if (gr.getQName().getLocalName().equals(qname.getLocalName())) {
-                        currentParent = gr;
-                        found = true;
-                    }
-                }
-            } else {
-                found = true;
-                currentParent = node;
+    /**
+     * Returns NotificationDefinition from Schema Context.
+     *
+     * @param schema SchemaContext in which lookup should be performed.
+     * @param path Schema Path of notification
+     * @return Notification schema or null, if notification is not present in schema context.
+     */
+    @Beta
+    public static @Nullable NotificationDefinition getNotificationSchema(final @NonNull SchemaContext schema,
+            final @NonNull SchemaPath path) {
+        requireNonNull(schema, "Schema context must not be null.");
+        requireNonNull(path, "Schema path must not be null.");
+        for (final NotificationDefinition potential : schema.getNotifications()) {
+            if (path.equals(potential.getPath())) {
+                return potential;
             }
-
-            Preconditions.checkArgument(found, "Failed to find referenced grouping: %s(%s)", path, qname.getLocalName());
         }
-
-        return (GroupingDefinition) currentParent;
+        return null;
     }
 
-    private static SchemaNode findNodeInModule(final Module module, final List<QName> path) {
-        final QName current = path.get(0);
-
-        LOG.trace("Looking for data container {} in module {}", current, module);
-        SchemaNode parent = module.getDataChildByName(current);
-        if (parent != null) {
-            final SchemaNode ret = findNode((DataSchemaNode) parent, nextLevel(path));
-            if (ret != null) {
-                return ret;
+    /**
+     * Returns RPC Input or Output Data container from RPC definition.
+     *
+     * @param schema SchemaContext in which lookup should be performed.
+     * @param path Schema path of RPC input/output data container
+     * @return Notification schema or null, if notification is not present in schema context.
+     */
+    @Beta
+    public static @Nullable ContainerLike getRpcDataSchema(final @NonNull SchemaContext schema,
+            final @NonNull SchemaPath path) {
+        requireNonNull(schema, "Schema context must not be null.");
+        requireNonNull(path, "Schema path must not be null.");
+        final Iterator<QName> it = path.getPathFromRoot().iterator();
+        checkArgument(it.hasNext(), "Rpc must have QName.");
+        final QName rpcName = it.next();
+        checkArgument(it.hasNext(), "input or output must be part of path.");
+        final QName inOrOut = it.next();
+        for (final RpcDefinition potential : schema.getOperations()) {
+            if (rpcName.equals(potential.getQName())) {
+                return SchemaNodeUtils.getRpcDataSchema(potential, inOrOut);
             }
         }
+        return null;
+    }
 
-        LOG.trace("Looking for RPC {} in module {}", current, module);
-        parent = getRpcByName(module, current);
-        if (parent != null) {
-            final SchemaNode ret = findNodeInRpc((RpcDefinition) parent, nextLevel(path));
-            if (ret != null) {
-                return ret;
-            }
-        }
+    /**
+     * Extract the identifiers of all modules and submodules which were used to create a particular SchemaContext.
+     *
+     * @param context SchemaContext to be examined
+     * @return Set of ModuleIdentifiers.
+     */
+    public static Set<SourceIdentifier> getConstituentModuleIdentifiers(final SchemaContext context) {
+        final Set<SourceIdentifier> ret = new HashSet<>();
 
-        LOG.trace("Looking for notification {} in module {}", current, module);
-        parent = getNotificationByName(module, current);
-        if (parent != null) {
-            final SchemaNode ret = findNodeInNotification((NotificationDefinition) parent, nextLevel(path));
-            if (ret != null) {
-                return ret;
-            }
-        }
+        for (Module module : context.getModules()) {
+            ret.add(moduleToIdentifier(module));
 
-        LOG.trace("Looking for grouping {} in module {}", current, module);
-        parent = getGroupingByName(module, current);
-        if (parent != null) {
-            final SchemaNode ret = findNodeInGrouping((GroupingDefinition) parent, nextLevel(path));
-            if (ret != null) {
-                return ret;
+            for (Submodule submodule : module.getSubmodules()) {
+                ret.add(moduleToIdentifier(submodule));
             }
         }
 
-        LOG.debug("No node matching {} found in module {}", path, module);
-        return null;
+        return ret;
     }
 
-    private static SchemaNode findNodeInGrouping(final GroupingDefinition grouping, final List<QName> path) {
-        if (path.isEmpty()) {
-            LOG.debug("Found grouping {}", grouping);
-            return grouping;
-        }
+    private static SourceIdentifier moduleToIdentifier(final ModuleLike module) {
+        return RevisionSourceIdentifier.create(module.getName(), module.getRevision());
+    }
 
-        LOG.trace("Looking for path {} in grouping {}", path, grouping);
-        final QName current = path.get(0);
-        final DataSchemaNode node = grouping.getDataChildByName(current);
-        if (node == null) {
-            LOG.debug("No node matching {} found in grouping {}", current, grouping);
+    private static SchemaNode findNodeInModule(final Module module, final Iterable<QName> path) {
+        if (!path.iterator().hasNext()) {
+            LOG.debug("No node matching {} found in node {}", path, module);
             return null;
         }
 
-        return findNode(node, nextLevel(path));
-    }
-
-    private static SchemaNode findNodeInRpc(final RpcDefinition rpc, final List<QName> path) {
-        if (path.isEmpty()) {
-            LOG.debug("Found RPC {}", rpc);
-            return rpc;
-        }
+        final QName current = path.iterator().next();
+        LOG.trace("Looking for node {} in module {}", current, module);
 
-        LOG.trace("Looking for path {} in rpc {}", path, rpc);
-        final QName current = path.get(0);
-        switch (current.getLocalName()) {
-        case "input":
-            return findNode(rpc.getInput(), nextLevel(path));
-        case "output":
-            return findNode(rpc.getOutput(), nextLevel(path));
-        default:
-            LOG.debug("Invalid component {} of path {} in RPC {}", current, path, rpc);
-            return null;
-        }
-    }
+        SchemaNode foundNode = null;
+        final Iterable<QName> nextPath = nextLevel(path);
 
-    private static SchemaNode findNodeInNotification(final NotificationDefinition ntf, final List<QName> path) {
-        if (path.isEmpty()) {
-            LOG.debug("Found notification {}", ntf);
-            return ntf;
+        foundNode = module.dataChildByName(current);
+        if (foundNode != null && nextPath.iterator().hasNext()) {
+            foundNode = findNodeIn(foundNode, nextPath);
         }
 
-        LOG.trace("Looking for path {} in notification {}", path, ntf);
-        final QName current = path.get(0);
-        DataSchemaNode node = ntf.getDataChildByName(current);
-        if (node == null) {
-            LOG.debug("No node matching {} found in notification {}", current, ntf);
-            return null;
+        if (foundNode == null) {
+            foundNode = getGroupingByName(module, current);
+            if (foundNode != null && nextPath.iterator().hasNext()) {
+                foundNode = findNodeIn(foundNode, nextPath);
+            }
         }
 
-        return findNode(node, nextLevel(path));
-    }
-
-    private static SchemaNode findNode(final ChoiceNode parent, final List<QName> path) {
-        if (path.isEmpty()) {
-            return parent;
-        }
-        QName current = path.get(0);
-        ChoiceCaseNode node = parent.getCaseNodeByName(current);
-        if (node != null) {
-            return findNodeInCase(node, nextLevel(path));
+        if (foundNode == null) {
+            foundNode = getRpcByName(module, current);
+            if (foundNode != null && nextPath.iterator().hasNext()) {
+                foundNode = findNodeIn(foundNode, nextPath);
+            }
         }
-        return null;
-    }
 
-    private static SchemaNode findNode(final ContainerSchemaNode parent, final List<QName> path) {
-        if (path.isEmpty()) {
-            return parent;
+        if (foundNode == null) {
+            foundNode = getNotificationByName(module, current);
+            if (foundNode != null && nextPath.iterator().hasNext()) {
+                foundNode = findNodeIn(foundNode, nextPath);
+            }
         }
 
-        final QName current = path.get(0);
-        final DataSchemaNode node = parent.getDataChildByName(current);
-        if (node == null) {
-            LOG.debug("Failed to find {} in parent {}", path, parent);
-            return null;
+        if (foundNode == null) {
+            LOG.debug("No node matching {} found in node {}", path, module);
         }
 
-        return findNode(node, nextLevel(path));
+        return foundNode;
     }
 
-    private static SchemaNode findNode(final ListSchemaNode parent, final List<QName> path) {
-        if (path.isEmpty()) {
-            return parent;
-        }
-
-        QName current = path.get(0);
-        DataSchemaNode node = parent.getDataChildByName(current);
-        if (node == null) {
-            LOG.debug("Failed to find {} in parent {}", path, parent);
+    private static SchemaNode findNodeIn(final SchemaNode parent, final Iterable<QName> path) {
+        if (!path.iterator().hasNext()) {
+            LOG.debug("No node matching {} found in node {}", path, parent);
             return null;
         }
-        return findNode(node, nextLevel(path));
-    }
-
-    private static SchemaNode findNode(final DataSchemaNode parent, final List<QName> path) {
-        final SchemaNode node;
-        if (!path.isEmpty()) {
-            if (parent instanceof ContainerSchemaNode) {
-                node = findNode((ContainerSchemaNode) parent, path);
-            } else if (parent instanceof ListSchemaNode) {
-                node = findNode((ListSchemaNode) parent, path);
-            } else if (parent instanceof ChoiceNode) {
-                node = findNode((ChoiceNode) parent, path);
-            } else {
-                throw new IllegalArgumentException(
-                        String.format("Path nesting violation in parent %s path %s", parent, path));
-            }
-        } else {
-            node = parent;
-        }
 
-        if (node == null) {
-            LOG.debug("Failed to find {} in parent {}", path, parent);
-            return null;
-        }
-        return node;
-    }
+        final QName current = path.iterator().next();
+        LOG.trace("Looking for node {} in node {}", current, parent);
 
-    public static SchemaNode findNodeInCase(final ChoiceCaseNode parent, final List<QName> path) {
-        if (path.isEmpty()) {
-            return parent;
-        }
+        SchemaNode foundNode = null;
+        final Iterable<QName> nextPath = nextLevel(path);
 
-        QName current = path.get(0);
-        DataSchemaNode node = parent.getDataChildByName(current);
-        if (node == null) {
-            LOG.debug("Failed to find {} in parent {}", path, parent);
-            return null;
-        }
-        return findNode(node, nextLevel(path));
-    }
+        if (parent instanceof DataNodeContainer) {
+            final DataNodeContainer parentDataNodeContainer = (DataNodeContainer) parent;
 
-    public static RpcDefinition getRpcByName(final Module module, final QName name) {
-        for (RpcDefinition rpc : module.getRpcs()) {
-            if (rpc.getQName().equals(name)) {
-                return rpc;
+            foundNode = parentDataNodeContainer.dataChildByName(current);
+            if (foundNode != null && nextPath.iterator().hasNext()) {
+                foundNode = findNodeIn(foundNode, nextPath);
             }
-        }
-        return null;
-    }
 
-    private static List<QName> nextLevel(final List<QName> path) {
-        return path.subList(1, path.size());
-    }
-
-    public static NotificationDefinition getNotificationByName(final Module module, final QName name) {
-        for (NotificationDefinition notification : module.getNotifications()) {
-            if (notification.getQName().equals(name)) {
-                return notification;
+            if (foundNode == null) {
+                foundNode = getGroupingByName(parentDataNodeContainer, current);
+                if (foundNode != null && nextPath.iterator().hasNext()) {
+                    foundNode = findNodeIn(foundNode, nextPath);
+                }
             }
         }
-        return null;
-    }
 
-    public static GroupingDefinition getGroupingByName(final Module module, final QName name) {
-        for (GroupingDefinition grouping : module.getGroupings()) {
-            if (grouping.getQName().equals(name)) {
-                return grouping;
+        if (foundNode == null && parent instanceof ActionNodeContainer) {
+            final Optional<? extends SchemaNode> next = ((ActionNodeContainer) parent).getActions().stream()
+                .filter(act -> current.equals(act.getQName())).findFirst();
+            if (next.isPresent() && nextPath.iterator().hasNext()) {
+                foundNode = findNodeIn(next.orElseThrow(), nextPath);
             }
         }
-        return null;
-    }
 
-    /**
-     * Utility method which search for original node defined in grouping.
-     *
-     * @param node
-     * @return
-     */
-    public static DataSchemaNode findOriginal(final DataSchemaNode node, final SchemaContext ctx) {
-        DataSchemaNode result = findCorrectTargetFromGrouping(node, ctx);
-        if (result == null) {
-            result = findCorrectTargetFromAugment(node, ctx);
-            if (result != null) {
-                if (result.isAddedByUses()) {
-                    result = findOriginal(result, ctx);
-                }
+        if (foundNode == null && parent instanceof NotificationNodeContainer) {
+            foundNode = ((NotificationNodeContainer) parent).getNotifications().stream()
+                    .filter(notif -> current.equals(notif.getQName())).findFirst().orElse(null);
+            if (foundNode != null && nextPath.iterator().hasNext()) {
+                foundNode = findNodeIn(foundNode, nextPath);
             }
         }
-        return result;
-    }
 
-    private static DataSchemaNode findCorrectTargetFromGrouping(final DataSchemaNode node, final SchemaContext ctx) {
-        if (node.getPath().getPath().size() == 1) {
-            // uses is under module statement
-            Module m = findParentModule(ctx, node);
-            DataSchemaNode result = null;
-            for (UsesNode u : m.getUses()) {
-                SchemaNode targetGrouping = findNodeInSchemaContext(ctx, u.getGroupingPath().getPath());
-                if (!(targetGrouping instanceof GroupingDefinition)) {
-                    throw new IllegalArgumentException(String.format("Failed to generate code for augment in %s", u));
+        if (foundNode == null && parent instanceof OperationDefinition) {
+            final OperationDefinition parentRpcDefinition = (OperationDefinition) parent;
+
+            if (current.getLocalName().equals("input")) {
+                foundNode = parentRpcDefinition.getInput();
+                if (foundNode != null && nextPath.iterator().hasNext()) {
+                    foundNode = findNodeIn(foundNode, nextPath);
                 }
-                GroupingDefinition gr = (GroupingDefinition) targetGrouping;
-                result = gr.getDataChildByName(node.getQName().getLocalName());
             }
-            if (result == null) {
-                throw new IllegalArgumentException("Failed to generate code for augment");
-            }
-            return result;
-        } else {
-            DataSchemaNode result = null;
-            QName currentName = node.getQName();
-            List<QName> tmpPath = new ArrayList<>();
-            Object parent = null;
-
-            SchemaPath sp = node.getPath();
-            List<QName> names = sp.getPath();
-            List<QName> newNames = new ArrayList<>(names);
-            newNames.remove(newNames.size() - 1);
-            SchemaPath newSp = new SchemaPath(newNames, sp.isAbsolute());
-            parent = findDataSchemaNode(ctx, newSp);
-
-            do {
-                tmpPath.add(currentName);
-                if (parent instanceof DataNodeContainer) {
-                    DataNodeContainer dataNodeParent = (DataNodeContainer) parent;
-                    for (UsesNode u : dataNodeParent.getUses()) {
-                        if (result == null) {
-                            result = getResultFromUses(u, currentName.getLocalName(), ctx);
-                        }
-                    }
-                }
-                if (result == null) {
-                    currentName = ((SchemaNode) parent).getQName();
-                    if (parent instanceof SchemaNode) {
-                        SchemaPath nodeSp = ((SchemaNode) parent).getPath();
-                        List<QName> nodeNames = nodeSp.getPath();
-                        List<QName> nodeNewNames = new ArrayList<>(nodeNames);
-                        nodeNewNames.remove(nodeNewNames.size() - 1);
-                        if (nodeNewNames.isEmpty()) {
-                            parent = getParentModule((SchemaNode) parent, ctx);
-                        } else {
-                            SchemaPath nodeNewSp = new SchemaPath(nodeNewNames, nodeSp.isAbsolute());
-                            parent = findDataSchemaNode(ctx, nodeNewSp);
-                        }
-                    } else {
-                        throw new IllegalArgumentException("Failed to generate code for augment");
-                    }
+
+            if (current.getLocalName().equals("output")) {
+                foundNode = parentRpcDefinition.getOutput();
+                if (foundNode != null && nextPath.iterator().hasNext()) {
+                    foundNode = findNodeIn(foundNode, nextPath);
                 }
-            } while (result == null && !(parent instanceof Module));
+            }
 
-            if (result != null) {
-                result = getTargetNode(tmpPath, result, ctx);
+            if (foundNode == null) {
+                foundNode = getGroupingByName(parentRpcDefinition, current);
+                if (foundNode != null && nextPath.iterator().hasNext()) {
+                    foundNode = findNodeIn(foundNode, nextPath);
+                }
             }
-            return result;
         }
-    }
 
-    private static DataSchemaNode findCorrectTargetFromAugment(final DataSchemaNode node, final SchemaContext ctx) {
-        if (!node.isAugmenting()) {
-            return null;
-        }
+        if (foundNode == null && parent instanceof ChoiceSchemaNode) {
+            foundNode = ((ChoiceSchemaNode) parent).findCase(current).orElse(null);
 
-        QName currentName = node.getQName();
-        Object currentNode = node;
-        Object parent = node;
-        List<QName> tmpPath = new ArrayList<QName>();
-        List<SchemaNode> tmpTree = new ArrayList<SchemaNode>();
-
-        AugmentationSchema augment = null;
-        do {
-            SchemaPath sp = ((SchemaNode) parent).getPath();
-            List<QName> names = sp.getPath();
-            List<QName> newNames = new ArrayList<>(names);
-            newNames.remove(newNames.size() - 1);
-            SchemaPath newSp = new SchemaPath(newNames, sp.isAbsolute());
-            parent = findDataSchemaNode(ctx, newSp);
-            if (parent instanceof AugmentationTarget) {
-                tmpPath.add(currentName);
-                tmpTree.add((SchemaNode) currentNode);
-                augment = findNodeInAugment(((AugmentationTarget) parent).getAvailableAugmentations(), currentName);
-                if (augment == null) {
-                    currentName = ((DataSchemaNode) parent).getQName();
-                    currentNode = parent;
-                }
+            if (foundNode != null && nextPath.iterator().hasNext()) {
+                foundNode = findNodeIn(foundNode, nextPath);
             }
-        } while (((DataSchemaNode) parent).isAugmenting() && augment == null);
 
-        if (augment == null) {
-            return null;
-        } else {
-            Collections.reverse(tmpPath);
-            Collections.reverse(tmpTree);
-            Object actualParent = augment;
-            DataSchemaNode result = null;
-            for (QName name : tmpPath) {
-                if (actualParent instanceof DataNodeContainer) {
-                    result = ((DataNodeContainer) actualParent).getDataChildByName(name.getLocalName());
-                    actualParent = ((DataNodeContainer) actualParent).getDataChildByName(name.getLocalName());
-                } else {
-                    if (actualParent instanceof ChoiceNode) {
-                        result = ((ChoiceNode) actualParent).getCaseNodeByName(name.getLocalName());
-                        actualParent = ((ChoiceNode) actualParent).getCaseNodeByName(name.getLocalName());
+            if (foundNode == null) {
+                // fallback that tries to map into one of the child cases
+                for (final CaseSchemaNode caseNode : ((ChoiceSchemaNode) parent).getCases()) {
+                    final DataSchemaNode maybeChild = caseNode.dataChildByName(current);
+                    if (maybeChild != null) {
+                        foundNode = findNodeIn(maybeChild, nextPath);
+                        break;
                     }
                 }
             }
+        }
 
-            if (result.isAddedByUses()) {
-                result = findCorrectTargetFromAugmentGrouping(result, augment, tmpTree, ctx);
-            }
-
-            return result;
+        if (foundNode == null) {
+            LOG.debug("No node matching {} found in node {}", path, parent);
         }
-    }
 
-    private static DataSchemaNode getResultFromUses(final UsesNode u, final String currentName, final SchemaContext ctx) {
-        SchemaNode targetGrouping = findNodeInSchemaContext(ctx, u.getGroupingPath().getPath());
+        return foundNode;
 
-        Preconditions.checkArgument(targetGrouping instanceof GroupingDefinition,
-                "Failed to generate code for augment in %s", u);
-        GroupingDefinition gr = (GroupingDefinition) targetGrouping;
-        return gr.getDataChildByName(currentName);
     }
 
-    private static Module getParentModule(final SchemaNode node, final SchemaContext ctx) {
-        QName qname = node.getPath().getPath().get(0);
-        URI namespace = qname.getNamespace();
-        Date revision = qname.getRevision();
-        return ctx.findModuleByNamespaceAndRevision(namespace, revision);
+    private static Iterable<QName> nextLevel(final Iterable<QName> path) {
+        return Iterables.skip(path, 1);
     }
 
-    private static DataSchemaNode getTargetNode(final List<QName> tmpPath, final DataSchemaNode node, final SchemaContext ctx) {
-        DataSchemaNode result = node;
-        if (tmpPath.size() == 1) {
-            if (result != null && result.isAddedByUses()) {
-                result = findOriginal(result, ctx);
-            }
-            return result;
-        } else {
-            DataSchemaNode newParent = result;
-            Collections.reverse(tmpPath);
-
-            tmpPath.remove(0);
-            for (QName name : tmpPath) {
-                // searching by local name is must, because node has different
-                // namespace in its original location
-                if (newParent == null) {
-                    break;
-                }
-                if (newParent instanceof DataNodeContainer) {
-                    newParent = ((DataNodeContainer) newParent).getDataChildByName(name.getLocalName());
-                } else {
-                    newParent = ((ChoiceNode) newParent).getCaseNodeByName(name.getLocalName());
-                }
-            }
-            if (newParent != null && newParent.isAddedByUses()) {
-                newParent = findOriginal(newParent, ctx);
+    private static RpcDefinition getRpcByName(final Module module, final QName name) {
+        for (final RpcDefinition rpc : module.getRpcs()) {
+            if (rpc.getQName().equals(name)) {
+                return rpc;
             }
-            return newParent;
         }
+        return null;
     }
 
-    private static AugmentationSchema findNodeInAugment(final Collection<AugmentationSchema> augments, final QName name) {
-        for (AugmentationSchema augment : augments) {
-            DataSchemaNode node = augment.getDataChildByName(name);
-            if (node != null) {
-                return augment;
+    private static NotificationDefinition getNotificationByName(final Module module, final QName name) {
+        for (final NotificationDefinition notification : module.getNotifications()) {
+            if (notification.getQName().equals(name)) {
+                return notification;
             }
         }
         return null;
     }
 
-    private static DataSchemaNode findCorrectTargetFromAugmentGrouping(final DataSchemaNode node,
-            final AugmentationSchema parentNode, final List<SchemaNode> dataTree, final SchemaContext ctx) {
-
-        DataSchemaNode result = null;
-        QName currentName = node.getQName();
-        List<QName> tmpPath = new ArrayList<>();
-        tmpPath.add(currentName);
-        int i = 1;
-        Object parent = null;
-
-        do {
-            if (dataTree.size() < 2 || dataTree.size() == i) {
-                parent = parentNode;
-            } else {
-                parent = dataTree.get(dataTree.size() - (i + 1));
-                tmpPath.add(((SchemaNode) parent).getQName());
-            }
-
-            if (parent instanceof DataNodeContainer) {
-                DataNodeContainer dataNodeParent = (DataNodeContainer) parent;
-                for (UsesNode u : dataNodeParent.getUses()) {
-                    if (result == null) {
-                        result = getResultFromUses(u, currentName.getLocalName(), ctx);
-                    }
-                }
+    private static GroupingDefinition getGroupingByName(final DataNodeContainer dataNodeContainer, final QName name) {
+        for (final GroupingDefinition grouping : dataNodeContainer.getGroupings()) {
+            if (grouping.getQName().equals(name)) {
+                return grouping;
             }
+        }
+        return null;
+    }
 
-            if (result == null) {
-                i = i + 1;
-                currentName = ((SchemaNode) parent).getQName();
+    private static GroupingDefinition getGroupingByName(final OperationDefinition rpc, final QName name) {
+        for (final GroupingDefinition grouping : rpc.getGroupings()) {
+            if (grouping.getQName().equals(name)) {
+                return grouping;
             }
-        } while (result == null);
-
-        if (result != null) {
-            result = getTargetNode(tmpPath, result, ctx);
         }
-        return result;
+        return null;
     }
 
     /**
@@ -717,8 +564,6 @@ public final class SchemaContextUtil {
      * If Schema Context, Parent Module or XPath string contains
      * <code>null</code> values, the method will throws IllegalArgumentException
      *
-     * @throws IllegalArgumentException
-     *
      * @param context
      *            Schema Context
      * @param parentModule
@@ -727,17 +572,11 @@ public final class SchemaContextUtil {
      *            XPath String
      * @return return a list of QName
      */
-    private static List<QName> xpathToQNamePath(final SchemaContext context, final Module parentModule, final String xpath) {
-        Preconditions.checkArgument(context != null, "Schema Context reference cannot be NULL");
-        Preconditions.checkArgument(parentModule != null, "Parent Module reference cannot be NULL");
-        Preconditions.checkArgument(xpath != null, "XPath string reference cannot be NULL");
-
-        List<QName> path = new LinkedList<QName>();
-        String[] prefixedPath = xpath.split("/");
-        for (String pathComponent : prefixedPath) {
-            if (!pathComponent.isEmpty()) {
-                path.add(stringPathPartToQName(context, parentModule, pathComponent));
-            }
+    private static List<QName> xpathToQNamePath(final SchemaContext context, final Module parentModule,
+            final String xpath) {
+        final List<QName> path = new ArrayList<>();
+        for (final String pathComponent : SLASH_SPLITTER.split(xpath)) {
+            path.add(stringPathPartToQName(context, parentModule, pathComponent));
         }
         return path;
     }
@@ -750,122 +589,391 @@ public final class SchemaContextUtil {
      * can be constructed. <br>
      * If the Prefixed Path Part does not contains prefix the Parent's Module
      * namespace is taken for construction of QName. <br>
-     * If Schema Context, Parent Module or Prefixed Path Part refers to
-     * <code>null</code> the method will throw IllegalArgumentException
      *
-     * @throws IllegalArgumentException
-     *
-     * @param context
-     *            Schema Context
-     * @param parentModule
-     *            Parent Module
-     * @param prefixedPathPart
-     *            Prefixed Path Part string
+     * @param context Schema Context
+     * @param parentModule Parent Module
+     * @param prefixedPathPart Prefixed Path Part string
      * @return QName from prefixed Path Part String.
+     * @throws NullPointerException if any arguments are null
      */
-    private static QName stringPathPartToQName(final SchemaContext context, final Module parentModule, final String prefixedPathPart) {
-        Preconditions.checkArgument(context != null, "Schema Context reference cannot be NULL");
-        Preconditions.checkArgument(parentModule != null, "Parent Module reference cannot be NULL");
-        Preconditions.checkArgument(prefixedPathPart != null, "Prefixed Path Part cannot be NULL!");
-
-        if (prefixedPathPart.contains(":")) {
-            String[] prefixedName = prefixedPathPart.split(":");
-            Module module = resolveModuleForPrefix(context, parentModule, prefixedName[0]);
-            Preconditions.checkArgument(module != null, "Failed to resolve xpath: no module found for prefix %s in module %s",
-                    prefixedName[0], parentModule.getName());
-            return new QName(module.getNamespace(), module.getRevision(), prefixedName[1]);
-        } else {
-            return new QName(parentModule.getNamespace(), parentModule.getRevision(), prefixedPathPart);
+    private static QName stringPathPartToQName(final SchemaContext context, final Module parentModule,
+            final String prefixedPathPart) {
+        requireNonNull(context, "context");
+
+        if (prefixedPathPart.indexOf(':') != -1) {
+            final Iterator<String> prefixedName = COLON_SPLITTER.split(prefixedPathPart).iterator();
+            final String modulePrefix = prefixedName.next();
+
+            final Module module = resolveModuleForPrefix(context, parentModule, modulePrefix);
+            checkArgument(module != null, "Failed to resolve xpath: no module found for prefix %s in module %s",
+                    modulePrefix, parentModule.getName());
+
+            return QName.create(module.getQNameModule(), prefixedName.next());
         }
+
+        return QName.create(parentModule.getQNameModule(), prefixedPathPart);
     }
 
     /**
-     * Method will attempt to resolve and provide Module reference for specified
-     * module prefix. Each Yang module could contains multiple imports which
-     * MUST be associated with corresponding module prefix. The method simply
-     * looks into module imports and returns the module that is bounded with
-     * specified prefix. If the prefix is not present in module or the prefixed
-     * module is not present in specified Schema Context, the method will return
-     * <code>null</code>. <br>
-     * If String prefix is the same as prefix of the specified Module the
-     * reference to this module is returned. <br>
-     * If Schema Context, Module or Prefix are referring to <code>null</code>
-     * the method will return IllegalArgumentException
-     *
-     * @throws IllegalArgumentException
+     * Method will attempt to resolve and provide Module reference for specified module prefix. Each Yang module could
+     * contains multiple imports which MUST be associated with corresponding module prefix. The method simply looks into
+     * module imports and returns the module that is bounded with specified prefix. If the prefix is not present
+     * in module or the prefixed module is not present in specified Schema Context, the method will return {@code null}.
+     * <br>
+     * If String prefix is the same as prefix of the specified Module the reference to this module is returned.<br>
      *
-     * @param context
-     *            Schema Context
-     * @param module
-     *            Yang Module
-     * @param prefix
-     *            Module Prefix
-     * @return Module for given prefix in specified Schema Context if is
-     *         present, otherwise returns <code>null</code>
+     * @param context Schema Context
+     * @param module Yang Module
+     * @param prefix Module Prefix
+     * @return Module for given prefix in specified Schema Context if is present, otherwise returns <code>null</code>
+     * @throws NullPointerException if any arguments are null
      */
-    private static Module resolveModuleForPrefix(final SchemaContext context, final Module module, final String prefix) {
-        Preconditions.checkArgument(context != null, "Schema Context reference cannot be NULL");
-        Preconditions.checkArgument(module != null, "Module reference cannot be NULL");
-        Preconditions.checkArgument(prefix != null, "Prefix string cannot be NULL");
+    private static Module resolveModuleForPrefix(final SchemaContext context, final Module module,
+            final String prefix) {
+        requireNonNull(context, "context");
 
         if (prefix.equals(module.getPrefix())) {
             return module;
         }
 
-        Set<ModuleImport> imports = module.getImports();
-        for (ModuleImport mi : imports) {
+        for (final ModuleImport mi : module.getImports()) {
             if (prefix.equals(mi.getPrefix())) {
-                return context.findModuleByName(mi.getModuleName(), mi.getRevision());
+                return context.findModule(mi.getModuleName(), mi.getRevision()).orElse(null);
             }
         }
         return null;
     }
 
     /**
-     * @throws IllegalArgumentException
+     * Resolve a relative XPath into a set of QNames.
      *
      * @param context
      *            Schema Context
      * @param module
      *            Yang Module
-     * @param relativeXPath
-     *            Non conditional Revision Aware Relative XPath
-     * @param leafrefSchemaPath
-     *            Schema Path for Leafref
-     * @return list of QName
+     * @param pathStr
+     *            xPath of leafref
+     * @param actualSchemaNode
+     *            actual schema node
+     * @return target schema node
+     * @throws IllegalArgumentException if any arguments are null
+     */
+    private static @Nullable SchemaNode resolveRelativeXPath(final SchemaContext context, final Module module,
+            final String pathStr, final SchemaNode actualSchemaNode) {
+        checkState(actualSchemaNode.getPath() != null, "Schema Path reference for Leafref cannot be NULL");
+
+        return pathStr.startsWith("deref(") ? resolveDerefPath(context, module, actualSchemaNode, pathStr)
+                : findTargetNode(context, resolveRelativePath(context, module, actualSchemaNode,
+                    doSplitXPath(pathStr)));
+    }
+
+    private static Iterable<QName> resolveRelativePath(final SchemaContext context, final Module module,
+            final SchemaNode actualSchemaNode, final List<String> steps) {
+        // Find out how many "parent" components there are and trim them
+        final int colCount = normalizeXPath(steps);
+        final List<String> xpaths = colCount == 0 ? steps : steps.subList(colCount, steps.size());
+
+        final List<QName> walkablePath = createWalkablePath(actualSchemaNode.getPath().getPathFromRoot(),
+                context, colCount);
+
+        if (walkablePath.size() - colCount >= 0) {
+            return Iterables.concat(Iterables.limit(walkablePath, walkablePath.size() - colCount),
+                    Iterables.transform(xpaths, input -> stringPathPartToQName(context, module, input)));
+        }
+        return Iterables.concat(walkablePath,
+                Iterables.transform(xpaths, input -> stringPathPartToQName(context, module, input)));
+    }
+
+    /**
+     * Return List of qNames that are walkable using xPath. When getting a path from schema node it will return path
+     * with parents like CaseSchemaNode and ChoiceSchemaNode as well if they are parents of the node. We need to get
+     * rid of these in order to find the node that xPath is pointing to. Also we can not remove any node beyond the
+     * amount of "../" because we will not be able to find the correct schema node from schema context
+     *
+     * @param schemaNodePath list of qNames as a path to the leaf of type leafref
+     * @param context        create schema context
+     * @param colCount       amount of "../" in the xPath expression
+     * @return list of QNames as a path where we should be able to find referenced node
      */
-    private static List<QName> resolveRelativeXPath(final SchemaContext context, final Module module,
-            final RevisionAwareXPath relativeXPath, final SchemaNode leafrefParentNode) {
-        Preconditions.checkArgument(context != null, "Schema Context reference cannot be NULL");
-        Preconditions.checkArgument(module != null, "Module reference cannot be NULL");
-        Preconditions.checkArgument(relativeXPath != null, "Non Conditional Revision Aware XPath cannot be NULL");
-        Preconditions.checkState(!relativeXPath.isAbsolute(),
-                "Revision Aware XPath MUST be relative i.e. MUST contains ../, "
-                        + "for non relative Revision Aware XPath use findDataSchemaNode method");
-        Preconditions.checkState(leafrefParentNode.getPath() != null,
-                "Schema Path reference for Leafref cannot be NULL");
-
-        List<QName> absolutePath = new LinkedList<QName>();
-        String strXPath = relativeXPath.toString();
-        String[] xpaths = strXPath.split("/");
-
-        int colCount = 0;
-        while (xpaths[colCount].contains("..")) {
-            colCount = colCount + 1;
-        }
-        List<QName> path = leafrefParentNode.getPath().getPath();
-        if (path != null) {
-            int lenght = path.size() - colCount;
-            absolutePath.addAll(path.subList(0, lenght));
-            List<String> xpathsList = Arrays.asList(xpaths);
-            List<String> sublistedXPath = xpathsList.subList(colCount, xpaths.length);
-            List<QName> sublist = new ArrayList<>();
-            for (String pathPart : sublistedXPath) {
-                sublist.add(stringPathPartToQName(context, module, pathPart));
+    private static List<QName> createWalkablePath(final Iterable<QName> schemaNodePath, final SchemaContext context,
+            final int colCount) {
+        final List<Integer> indexToRemove = new ArrayList<>();
+        List<QName> schemaNodePathRet = Lists.newArrayList(schemaNodePath);
+        for (int j = 0, i = schemaNodePathRet.size() - 1; i >= 0 && j != colCount; i--, j++) {
+            final SchemaNode nodeIn = findTargetNode(context, schemaNodePathRet);
+            if (nodeIn instanceof CaseSchemaNode || nodeIn instanceof ChoiceSchemaNode) {
+                indexToRemove.add(i);
+                j--;
+            }
+            schemaNodePathRet.remove(i);
+        }
+        schemaNodePathRet = Lists.newArrayList(schemaNodePath);
+        for (int i : indexToRemove) {
+            schemaNodePathRet.remove(i);
+        }
+        return schemaNodePathRet;
+    }
+
+    private static SchemaNode resolveDerefPath(final SchemaContext context, final Module module,
+            final SchemaNode actualSchemaNode, final String xpath) {
+        final int paren = xpath.indexOf(')', 6);
+        checkArgument(paren != -1, "Cannot find matching parentheses in %s", xpath);
+
+        final String derefArg = xpath.substring(6, paren).strip();
+        // Look up the node which we need to reference
+        final SchemaNode derefTarget = findTargetNode(context, resolveRelativePath(context, module, actualSchemaNode,
+            doSplitXPath(derefArg)));
+        checkArgument(derefTarget != null, "Cannot find deref(%s) target node %s in context of %s", derefArg,
+                actualSchemaNode);
+        checkArgument(derefTarget instanceof TypedDataSchemaNode, "deref(%s) resolved to non-typed %s", derefArg,
+            derefTarget);
+
+        // We have a deref() target, decide what to do about it
+        final TypeDefinition<?> targetType = ((TypedDataSchemaNode) derefTarget).getType();
+        if (targetType instanceof InstanceIdentifierTypeDefinition) {
+            // Static inference breaks down, we cannot determine where this points to
+            // FIXME: dedicated exception, users can recover from it, derive from IAE
+            throw new UnsupportedOperationException("Cannot infer instance-identifier reference " + targetType);
+        }
+
+        // deref() is define only for instance-identifier and leafref types, handle the latter
+        checkArgument(targetType instanceof LeafrefTypeDefinition, "Illegal target type %s", targetType);
+
+        final PathExpression targetPath = ((LeafrefTypeDefinition) targetType).getPathStatement();
+        LOG.debug("Derefencing path {}", targetPath);
+
+        final SchemaNode deref = targetPath.isAbsolute()
+                ? findTargetNode(context, actualSchemaNode.getQName().getModule(),
+                    ((LocationPathSteps) targetPath.getSteps()).getLocationPath())
+                        : findDataSchemaNodeForRelativeXPath(context, module, actualSchemaNode, targetPath);
+        if (deref == null) {
+            LOG.debug("Path {} could not be derefenced", targetPath);
+            return null;
+        }
+
+        checkArgument(deref instanceof LeafSchemaNode, "Unexpected %s reference in %s", deref, targetPath);
+
+        final List<String> qnames = doSplitXPath(xpath.substring(paren + 1).stripLeading());
+        return findTargetNode(context, resolveRelativePath(context, module, deref, qnames));
+    }
+
+    private static @Nullable SchemaNode findTargetNode(final SchemaContext context, final QNameModule localNamespace,
+            final YangLocationPath path) {
+        final Deque<QName> ret = new ArrayDeque<>();
+        for (Step step : path.getSteps()) {
+            if (step instanceof AxisStep) {
+                // We only support parent axis steps
+                final YangXPathAxis axis = ((AxisStep) step).getAxis();
+                checkState(axis == YangXPathAxis.PARENT, "Unexpected axis %s", axis);
+                ret.removeLast();
+                continue;
             }
-            absolutePath.addAll(sublist);
+
+            // This has to be a QNameStep
+            checkState(step instanceof QNameStep, "Unhandled step %s in %s", step, path);
+            ret.addLast(resolve(((QNameStep) step).getQName(), localNamespace));
         }
 
-        return absolutePath;
+        return findTargetNode(context, ret);
+    }
+
+    private static @Nullable SchemaNode findTargetNode(final SchemaContext context, final Iterable<QName> qnamePath) {
+        // We do not have enough information about resolution context, hence cannot account for actions, RPCs
+        // and notifications. We therefore attempt to make a best estimate, but this can still fail.
+        final Optional<DataSchemaNode> pureData = context.findDataTreeChild(qnamePath);
+        return pureData.isPresent() ? pureData.get() : findNodeInSchemaContext(context, qnamePath);
+    }
+
+    private static QName resolve(final AbstractQName toResolve, final QNameModule localNamespace) {
+        if (toResolve instanceof QName) {
+            return (QName) toResolve;
+        } else if (toResolve instanceof UnqualifiedQName) {
+            return ((UnqualifiedQName) toResolve).bindTo(localNamespace);
+        } else {
+            throw new IllegalStateException("Unhandled step " + toResolve);
+        }
+    }
+
+    @VisibleForTesting
+    static int normalizeXPath(final List<String> xpath) {
+        LOG.trace("Normalize {}", xpath);
+
+        // We need to make multiple passes here, as the leading XPaths as we can have "../abc/../../def", which really
+        // is "../../def"
+        while (true) {
+            // Next up: count leading ".." components
+            int leadingParents = 0;
+            while (true) {
+                if (leadingParents == xpath.size()) {
+                    return leadingParents;
+                }
+                if (!"..".equals(xpath.get(leadingParents))) {
+                    break;
+                }
+
+                ++leadingParents;
+            }
+
+            // Now let's see if there there is a '..' in the rest
+            final int dots = findDots(xpath, leadingParents + 1);
+            if (dots == -1) {
+                return leadingParents;
+            }
+
+            xpath.remove(dots - 1);
+            xpath.remove(dots - 1);
+            LOG.trace("Next iteration {}", xpath);
+        }
+    }
+
+    private static int findDots(final List<String> xpath, final int startIndex) {
+        for (int i = startIndex; i < xpath.size(); ++i) {
+            if ("..".equals(xpath.get(i))) {
+                return i;
+            }
+        }
+
+        return -1;
+    }
+
+    private static List<String> doSplitXPath(final String xpath) {
+        final List<String> ret = new ArrayList<>();
+        for (String str : SLASH_SPLITTER.split(xpath)) {
+            ret.add(str);
+        }
+        return ret;
+    }
+
+    /**
+     * Extracts the base type of node on which schema node points to. If target node is again of type
+     * LeafrefTypeDefinition, methods will be call recursively until it reach concrete type definition.
+     *
+     * @param typeDefinition
+     *            type of node which will be extracted
+     * @param schemaContext
+     *            Schema Context
+     * @param schema
+     *            Schema Node
+     * @return recursively found type definition this leafref is pointing to or null if the xpath is incorrect (null
+     *         is there to preserve backwards compatibility)
+     */
+    public static TypeDefinition<?> getBaseTypeForLeafRef(final LeafrefTypeDefinition typeDefinition,
+            final SchemaContext schemaContext, final SchemaNode schema) {
+        final PathExpression pathStatement = typeDefinition.getPathStatement();
+        final String pathStr = stripConditionsFromXPathString(pathStatement);
+
+        final DataSchemaNode dataSchemaNode;
+        if (pathStatement.isAbsolute()) {
+            SchemaNode baseSchema = schema;
+            while (baseSchema instanceof DerivableSchemaNode) {
+                final Optional<? extends SchemaNode> basePotential = ((DerivableSchemaNode) baseSchema).getOriginal();
+                if (basePotential.isPresent()) {
+                    baseSchema = basePotential.get();
+                } else {
+                    break;
+                }
+            }
+
+            final Module parentModule = findParentModuleOfReferencingType(schemaContext, baseSchema);
+            dataSchemaNode = (DataSchemaNode) findTargetNode(schemaContext,
+                xpathToQNamePath(schemaContext, parentModule, pathStr));
+        } else {
+            Module parentModule = findParentModule(schemaContext, schema);
+            dataSchemaNode = (DataSchemaNode) resolveRelativeXPath(schemaContext, parentModule, pathStr, schema);
+        }
+
+        // FIXME this is just to preserve backwards compatibility since yangtools do not mind wrong leafref xpaths
+        // and current expected behaviour for such cases is to just use pure string
+        // This should throw an exception about incorrect XPath in leafref
+        if (dataSchemaNode == null) {
+            return null;
+        }
+
+        final TypeDefinition<?> targetTypeDefinition = typeDefinition(dataSchemaNode);
+
+        if (targetTypeDefinition instanceof LeafrefTypeDefinition) {
+            return getBaseTypeForLeafRef((LeafrefTypeDefinition) targetTypeDefinition, schemaContext, dataSchemaNode);
+        }
+
+        return targetTypeDefinition;
+    }
+
+    /**
+     * Returns base type for {@code typeDefinition} which belongs to module specified via {@code qname}. This handle
+     * the case when leafref type isn't specified as type substatement of leaf or leaf-list but is defined in other
+     * module as typedef which is then imported to referenced module.
+     *
+     * <p>
+     * Because {@code typeDefinition} is definied via typedef statement, only absolute path is meaningful.
+     */
+    public static TypeDefinition<?> getBaseTypeForLeafRef(final LeafrefTypeDefinition typeDefinition,
+            final SchemaContext schemaContext, final QName qname) {
+        final PathExpression pathStatement = typeDefinition.getPathStatement();
+        if (!pathStatement.isAbsolute()) {
+            return null;
+        }
+
+        final Optional<Module> parentModule = schemaContext.findModule(qname.getModule());
+        checkArgument(parentModule.isPresent(), "Failed to find parent module for %s", qname);
+
+        final DataSchemaNode dataSchemaNode = (DataSchemaNode) findTargetNode(schemaContext,
+            xpathToQNamePath(schemaContext, parentModule.get(), stripConditionsFromXPathString(pathStatement)));
+        final TypeDefinition<?> targetTypeDefinition = typeDefinition(dataSchemaNode);
+        if (targetTypeDefinition instanceof LeafrefTypeDefinition) {
+            return getBaseTypeForLeafRef((LeafrefTypeDefinition) targetTypeDefinition, schemaContext, dataSchemaNode);
+        }
+
+        return targetTypeDefinition;
+    }
+
+    private static Module findParentModuleOfReferencingType(final SchemaContext schemaContext,
+            final SchemaNode schemaNode) {
+        checkArgument(schemaContext != null, "Schema Context reference cannot be NULL!");
+        checkArgument(schemaNode instanceof TypedDataSchemaNode, "Unsupported node %s", schemaNode);
+
+        TypeDefinition<?> nodeType = ((TypedDataSchemaNode) schemaNode).getType();
+        if (nodeType.getBaseType() != null) {
+            while (nodeType.getBaseType() != null) {
+                nodeType = nodeType.getBaseType();
+            }
+
+            return schemaContext.findModule(nodeType.getQName().getModule()).orElse(null);
+        }
+
+        return findParentModule(schemaContext, schemaNode);
+    }
+
+    private static final Pattern STRIP_PATTERN = Pattern.compile("\\[[^\\[\\]]*\\]");
+
+    /**
+     * Removes conditions from xPath pointed to target node.
+     *
+     * @param pathStatement
+     *            xPath to target node
+     * @return string representation of xPath without conditions
+     */
+    @VisibleForTesting
+    static String stripConditionsFromXPathString(final PathExpression pathStatement) {
+        return STRIP_PATTERN.matcher(pathStatement.getOriginalString()).replaceAll("");
+    }
+
+    /**
+     * Gets the base type of DataSchemaNode value.
+     *
+     * @param node
+     *            a node representing DataSchemaNode
+     * @return concrete type definition of node value
+     */
+    private static TypeDefinition<?> typeDefinition(final DataSchemaNode node) {
+        checkArgument(node instanceof TypedDataSchemaNode, "Unhandled parameter type %s", node);
+
+        TypeDefinition<?> current = ((TypedDataSchemaNode) node).getType();
+        // TODO: don't we have a utility for this?
+        TypeDefinition<?> base = current.getBaseType();
+        while (base != null) {
+            current = base;
+            base = current.getBaseType();
+        }
+        return current;
     }
 }