1a625c279b752f99655fb6a2c03161c2327ee0a4
[ccsdk/cds.git] / ms / blueprintsprocessor / application / src / main / kotlin / org / onap / ccsdk / cds / blueprintsprocessor / uat / JsonNormalizer.kt
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 }