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
  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 = expandJstlSpec(jsltSpec)
  39                 val jslt = Parser.compileString(expandedJstlSpec)
  40                 val output = jslt.apply(input)
  41                 output.toString()
  42             }
  43         }
  44
  45         /**
  46          * Creates an extended JSTL spec by appending the "*: ." wildcard pattern to every inner JSON object, and
  47          * removing the extra quotes added by the standard YAML/JSON converters on fields prefixed by "?".
  48          *
  49          * @param jstlSpec the JSTL spec as a structured JSON object.
  50          * @return the string representation of the extended JSTL spec.
  51          */
  52         private fun expandJstlSpec(jstlSpec: JsonNode): String {
  53             val extendedJstlSpec = updateObjectNodes(jstlSpec, "*", ".")
  54             return extendedJstlSpec.toString()
  55                     // Handle the "?" as a prefix to literal/non-quoted values
  56                     .replace("\"\\?([^\"]+)\"".toRegex(), "$1")
  57                     // Also, remove the quotes added by Jackson for key and value of the wildcard matcher
  58                     .replace("\"([.*])\"".toRegex(), "$1")
  59         }
  60
  61         /**
  62          * Expands a structured JSON object, by adding the given key and value to every nested ObjectNode.
  63          *
  64          * @param jsonNode the root node.
  65          * @param fieldName the fixed field name.
  66          * @param fieldValue the fixed field value.
  67          */
  68         private fun updateObjectNodes(jsonNode: JsonNode, fieldName: String, fieldValue: String): JsonNode {
  69             if (jsonNode is ContainerNode<*>) {
  70                 (jsonNode as? ObjectNode)?.put(fieldName, fieldValue)
  71                 jsonNode.forEach { child ->
  72                     updateObjectNodes(child, fieldName, fieldValue)
  73                 }
  74             }
  75             return jsonNode
  76         }
  77     }
  78 }