gerrit.onap Code Review - ccsdk/cds.git/blob

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2019 Nordix Foundation.
   4  * ================================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  * SPDX-License-Identifier: Apache-2.0
  18  * ============LICENSE_END=========================================================
  19  */
  20 package org.onap.ccsdk.cds.blueprintsprocessor.uat.utils
  21
  22 import com.fasterxml.jackson.databind.JsonNode
  23 import com.fasterxml.jackson.databind.ObjectMapper
  24 import com.fasterxml.jackson.databind.node.ContainerNode
  25 import com.fasterxml.jackson.databind.node.ObjectNode
  26 import com.schibsted.spt.data.jslt.Parser
  27
  28 internal class JsonNormalizer {
  29
  30     companion object {
  31
  32         fun getNormalizer(mapper: ObjectMapper, jsltSpec: JsonNode?): (String) -> String {
  33             if (jsltSpec == null) {
  34                 return { it }
  35             }
  36             return { s: String ->
  37                 val input = mapper.readTree(s)
  38                 val expandedJstlSpec =
  39                     expandJstlSpec(jsltSpec)
  40                 val jslt = Parser.compileString(expandedJstlSpec)
  41                 val output = jslt.apply(input)
  42                 output.toString()
  43             }
  44         }
  45
  46         /**
  47          * Creates an extended JSTL spec by appending the "*: ." wildcard pattern to every inner JSON object, and
  48          * removing the extra quotes added by the standard YAML/JSON converters on fields prefixed by "?".
  49          *
  50          * @param jstlSpec the JSTL spec as a structured JSON object.
  51          * @return the string representation of the extended JSTL spec.
  52          */
  53         private fun expandJstlSpec(jstlSpec: JsonNode): String {
  54             val extendedJstlSpec =
  55                 updateObjectNodes(jstlSpec, "*", ".")
  56             return extendedJstlSpec.toString()
  57                 // Handle the "?" as a prefix to literal/non-quoted values
  58                 .replace("\"\\?([^\"]+)\"".toRegex(), "$1")
  59                 // Also, remove the quotes added by Jackson for key and value of the wildcard matcher
  60                 .replace("\"([.*])\"".toRegex(), "$1")
  61         }
  62
  63         /**
  64          * Expands a structured JSON object, by adding the given key and value to every nested ObjectNode.
  65          *
  66          * @param jsonNode the root node.
  67          * @param fieldName the fixed field name.
  68          * @param fieldValue the fixed field value.
  69          */
  70         private fun updateObjectNodes(jsonNode: JsonNode, fieldName: String, fieldValue: String): JsonNode {
  71             if (jsonNode is ContainerNode<*>) {
  72                 (jsonNode as? ObjectNode)?.put(fieldName, fieldValue)
  73                 jsonNode.forEach { child ->
  74                     updateObjectNodes(
  75                         child,
  76                         fieldName,
  77                         fieldValue
  78                     )
  79                 }
  80             }
  81             return jsonNode
  82         }
  83     }
  84 }