Fix SonarQube violations
[cps.git] / cps-service / src / main / java / org / onap / cps / utils / YangUtils.java
index 8743b7d..edef5cb 100644 (file)
@@ -1,6 +1,8 @@
 /*
  * ============LICENSE_START=======================================================
- *  Copyright (C) 2020 Nordix Foundation
+ *  Copyright (C) 2020-2021 Nordix Foundation
+ *  Modifications Copyright (C) 2021 Bell Canada. All rights reserved.
+ *  Modifications Copyright (C) 2021 Pantheon.tech
  *  ================================================================================
  *  Licensed under the Apache License, Version 2.0 (the "License");
  *  you may not use this file except in compliance with the License.
 
 package org.onap.cps.utils;
 
+import com.google.gson.JsonSyntaxException;
 import com.google.gson.stream.JsonReader;
 import java.io.IOException;
 import java.io.StringReader;
+import java.util.Arrays;
 import java.util.Collection;
 import java.util.Collections;
 import java.util.List;
+import java.util.Optional;
 import java.util.stream.Collectors;
+import lombok.AccessLevel;
+import lombok.NoArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
-import org.onap.cps.api.impl.Fragment;
-import org.opendaylight.yangtools.yang.common.QName;
+import org.onap.cps.spi.exceptions.DataValidationException;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
-import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
-import org.opendaylight.yangtools.yang.data.api.schema.DataContainerNode;
-import org.opendaylight.yangtools.yang.data.api.schema.LeafSetNode;
-import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
-import org.opendaylight.yangtools.yang.data.api.schema.MapNode;
 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
-import org.opendaylight.yangtools.yang.data.api.schema.ValueNode;
 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
 import org.opendaylight.yangtools.yang.data.codec.gson.JSONCodecFactory;
 import org.opendaylight.yangtools.yang.data.codec.gson.JSONCodecFactorySupplier;
 import org.opendaylight.yangtools.yang.data.codec.gson.JsonParserStream;
 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNormalizedNodeStreamWriter;
 import org.opendaylight.yangtools.yang.data.impl.schema.NormalizedNodeResult;
-import org.opendaylight.yangtools.yang.model.api.Module;
+import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
+import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
 
 @Slf4j
+@NoArgsConstructor(access = AccessLevel.PRIVATE)
 public class YangUtils {
 
-    private YangUtils() {
-        throw new IllegalStateException("Utility class");
+    private static final String XPATH_DELIMITER_REGEX = "\\/";
+    private static final String XPATH_NODE_KEY_ATTRIBUTES_REGEX = "\\[.+";
+
+    /**
+     * Parses jsonData into NormalizedNode according to given schema context.
+     *
+     * @param jsonData      json data as string
+     * @param schemaContext schema context describing associated data model
+     * @return the NormalizedNode object
+     */
+    @SuppressWarnings("squid:S1452")  // Generic type <? ,?> is returned by external librray, opendaylight.yangtools
+    public static NormalizedNode<?, ?> parseJsonData(final String jsonData, final SchemaContext schemaContext) {
+        return parseJsonData(jsonData, schemaContext, Optional.empty());
     }
 
     /**
-     * Parse a string containing json data for a certain model (schemaContext).
+     * Parses jsonData into NormalizedNode according to given schema context.
      *
-     * @param jsonData      a string containing json data for the given model
-     * @param schemaContext the SchemaContext for the given data
-     * @return the NormalizedNode representing the json data
+     * @param jsonData        json data fragment as string
+     * @param schemaContext   schema context describing associated data model
+     * @param parentNodeXpath the xpath referencing the parent node current data fragment belong to
+     * @return the NormalizedNode object
      */
-    public static NormalizedNode<?, ?> parseJsonData(final String jsonData, final SchemaContext schemaContext)
-            throws IOException {
+    @SuppressWarnings("squid:S1452")  // Generic type <? ,?> is returned by external librray, opendaylight.yangtools
+    public static NormalizedNode<?, ?> parseJsonData(final String jsonData, final SchemaContext schemaContext,
+        final String parentNodeXpath) {
+        final DataSchemaNode parentSchemaNode = getDataSchemaNodeByXpath(parentNodeXpath, schemaContext);
+        return parseJsonData(jsonData, schemaContext, Optional.of(parentSchemaNode));
+    }
+
+    private static NormalizedNode<?, ?> parseJsonData(final String jsonData, final SchemaContext schemaContext,
+        final Optional<DataSchemaNode> optionalParentSchemaNode) {
         final JSONCodecFactory jsonCodecFactory = JSONCodecFactorySupplier.DRAFT_LHOTKA_NETMOD_YANG_JSON_02
-                .getShared(schemaContext);
+            .getShared(schemaContext);
         final NormalizedNodeResult normalizedNodeResult = new NormalizedNodeResult();
         final NormalizedNodeStreamWriter normalizedNodeStreamWriter = ImmutableNormalizedNodeStreamWriter
-                .from(normalizedNodeResult);
-        try (final JsonParserStream jsonParserStream = JsonParserStream
-                .create(normalizedNodeStreamWriter, jsonCodecFactory)) {
+            .from(normalizedNodeResult);
+
+        try (final JsonParserStream jsonParserStream = optionalParentSchemaNode.isPresent()
+            ? JsonParserStream.create(normalizedNodeStreamWriter, jsonCodecFactory, optionalParentSchemaNode.get())
+            : JsonParserStream.create(normalizedNodeStreamWriter, jsonCodecFactory)
+        ) {
             final JsonReader jsonReader = new JsonReader(new StringReader(jsonData));
             jsonParserStream.parse(jsonReader);
+
+        } catch (final IOException | IllegalStateException | JsonSyntaxException exception) {
+            throw new DataValidationException(
+                    "Failed to parse json data: " + jsonData,  exception.getMessage(), exception);
         }
         return normalizedNodeResult.getResult();
     }
 
     /**
-     * Break a Normalized Node tree into fragments that can be stored by the persistence service.
+     * Create an xpath form a Yang Tools NodeIdentifier (i.e. PathArgument).
      *
-     * @param tree   the normalized node tree
-     * @param module the module applicable for the data in the normalized node
-     * @return the 'root' Fragment for the tree contain all relevant children etc.
+     * @param nodeIdentifier the NodeIdentifier
+     * @return an xpath
      */
-    public static Fragment fragmentNormalizedNode(
-            final NormalizedNode<? extends YangInstanceIdentifier.PathArgument, ?> tree,
-            final Module module) {
-        final QName[] nodeTypes = {tree.getNodeType()};
-        final String xpath = buildXpathId(tree.getIdentifier());
-        final Fragment rootFragment = Fragment.createRootFragment(module, nodeTypes, xpath);
-        fragmentNormalizedNode(rootFragment, tree);
-        return rootFragment;
-    }
-
-    private static void fragmentNormalizedNode(final Fragment currentFragment,
-                                               final NormalizedNode normalizedNode) {
-        if (normalizedNode instanceof DataContainerNode) {
-            inspectContainer(currentFragment, (DataContainerNode) normalizedNode);
-        } else if (normalizedNode instanceof MapNode) {
-            inspectKeyedList(currentFragment, (MapNode) normalizedNode);
-        } else if (normalizedNode instanceof ValueNode) {
-            inspectLeaf(currentFragment, (ValueNode) normalizedNode);
-        } else if (normalizedNode instanceof LeafSetNode) {
-            inspectLeafList(currentFragment, (LeafSetNode) normalizedNode);
-        } else {
-            log.warn("Cannot normalize " + normalizedNode.getClass());
-        }
-    }
-
-    private static void inspectLeaf(final Fragment currentFragment,
-                                    final ValueNode valueNode) {
-        final Object value = valueNode.getValue();
-        currentFragment.addLeafValue(valueNode.getNodeType().getLocalName(), value);
-    }
-
-    private static void inspectLeafList(final Fragment currentFragment,
-                                        final LeafSetNode leafSetNode) {
-        currentFragment.addLeafListName(leafSetNode.getNodeType().getLocalName());
-        for (final NormalizedNode value : (Collection<NormalizedNode>) leafSetNode.getValue()) {
-            fragmentNormalizedNode(currentFragment, value);
-        }
-    }
-
-    private static void inspectContainer(final Fragment currentFragment,
-                                         final DataContainerNode dataContainerNode) {
-        final Collection<NormalizedNode> leaves = (Collection) dataContainerNode.getValue();
-        for (final NormalizedNode leaf : leaves) {
-            fragmentNormalizedNode(currentFragment, leaf);
-        }
-    }
-
-    private static void inspectKeyedList(final Fragment currentFragment,
-                                         final MapNode mapNode) {
-        createNodeForEachListElement(currentFragment, mapNode);
-    }
-
-    private static void createNodeForEachListElement(final Fragment currentFragment, final MapNode mapNode) {
-        final Collection<MapEntryNode> mapEntryNodes = mapNode.getValue();
-        for (final MapEntryNode mapEntryNode : mapEntryNodes) {
-            final String xpathId = buildXpathId(mapEntryNode.getIdentifier());
-            final Fragment listElementFragment =
-                currentFragment.createChildFragment(mapNode.getNodeType(), xpathId);
-            fragmentNormalizedNode(listElementFragment, mapEntryNode);
-        }
-    }
-
-    private static String buildXpathId(final YangInstanceIdentifier.PathArgument nodeIdentifier) {
-        final StringBuilder xpathIdBuilder = new StringBuilder();
-        xpathIdBuilder.append("/").append(nodeIdentifier.getNodeType().getLocalName());
+    public static String buildXpath(final YangInstanceIdentifier.PathArgument nodeIdentifier) {
+        final StringBuilder xpathBuilder = new StringBuilder();
+        xpathBuilder.append("/").append(nodeIdentifier.getNodeType().getLocalName());
 
-        if (nodeIdentifier instanceof NodeIdentifierWithPredicates) {
-            xpathIdBuilder.append(getKeyAttributesStatement((NodeIdentifierWithPredicates) nodeIdentifier));
+        if (nodeIdentifier instanceof YangInstanceIdentifier.NodeIdentifierWithPredicates) {
+            xpathBuilder.append(getKeyAttributesStatement(
+                (YangInstanceIdentifier.NodeIdentifierWithPredicates) nodeIdentifier));
         }
-        return xpathIdBuilder.toString();
+        return xpathBuilder.toString();
     }
 
-    private static String getKeyAttributesStatement(final NodeIdentifierWithPredicates nodeIdentifier) {
+    private static String getKeyAttributesStatement(
+        final YangInstanceIdentifier.NodeIdentifierWithPredicates nodeIdentifier) {
         final List<String> keyAttributes = nodeIdentifier.entrySet().stream().map(
             entry -> {
                 final String name = entry.getKey().getLocalName();
@@ -170,4 +137,49 @@ public class YangUtils {
             return "[" + String.join(" and ", keyAttributes) + "]";
         }
     }
+
+    private static DataSchemaNode getDataSchemaNodeByXpath(final String parentNodeXpath,
+        final SchemaContext schemaContext) {
+        final String[] xpathNodeIdSequence = xpathToNodeIdSequence(parentNodeXpath);
+        return findDataSchemaNodeByXpathNodeIdSequence(xpathNodeIdSequence, schemaContext.getChildNodes());
+    }
+
+    private static String[] xpathToNodeIdSequence(final String xpath) {
+        final String[] xpathNodeIdSequence = Arrays.stream(xpath.split(XPATH_DELIMITER_REGEX))
+            .map(identifier -> identifier.replaceFirst(XPATH_NODE_KEY_ATTRIBUTES_REGEX, ""))
+            .filter(identifier -> !identifier.isEmpty())
+            .toArray(String[]::new);
+        if (xpathNodeIdSequence.length < 1) {
+            throw new DataValidationException("Invalid xpath.", "Xpath contains no node identifiers.");
+        }
+        return xpathNodeIdSequence;
+    }
+
+    private static DataSchemaNode findDataSchemaNodeByXpathNodeIdSequence(final String[] xpathNodeIdSequence,
+        final Collection<? extends DataSchemaNode> dataSchemaNodes) {
+        final String currentXpathNodeId = xpathNodeIdSequence[0];
+        final DataSchemaNode currentDataSchemaNode = dataSchemaNodes.stream()
+            .filter(dataSchemaNode -> currentXpathNodeId.equals(dataSchemaNode.getQName().getLocalName()))
+            .findFirst().orElseThrow(() -> schemaNodeNotFoundException(currentXpathNodeId));
+        if (xpathNodeIdSequence.length <= 1) {
+            return currentDataSchemaNode;
+        }
+        if (currentDataSchemaNode instanceof DataNodeContainer) {
+            return findDataSchemaNodeByXpathNodeIdSequence(
+                getNextLevelXpathNodeIdSequence(xpathNodeIdSequence),
+                ((DataNodeContainer) currentDataSchemaNode).getChildNodes());
+        }
+        throw schemaNodeNotFoundException(xpathNodeIdSequence[1]);
+    }
+
+    private static String[] getNextLevelXpathNodeIdSequence(final String[] xpathNodeIdSequence) {
+        final String[] nextXpathNodeIdSequence = new String[xpathNodeIdSequence.length - 1];
+        System.arraycopy(xpathNodeIdSequence, 1, nextXpathNodeIdSequence, 0, nextXpathNodeIdSequence.length);
+        return nextXpathNodeIdSequence;
+    }
+
+    private static DataValidationException schemaNodeNotFoundException(final String schemaNodeIdentifier) {
+        return new DataValidationException("Invalid xpath.",
+            String.format("No schema node was found for xpath identifier '%s'.", schemaNodeIdentifier));
+    }
 }