aboutsummaryrefslogtreecommitdiffstats
path: root/appc-client/client-kit/src/test/java/org/onap/appc/generator/JsonHelper.java
blob: e75a44e6525411ec47cb56cf915c90242748e076 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
/*-
 * ============LICENSE_START=======================================================
 * ONAP : APPC
 * ================================================================================
 * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
 * ================================================================================
 * Copyright (C) 2017 Amdocs
 * =============================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * 
 * ECOMP is a trademark and service mark of AT&T Intellectual Property.
 * ============LICENSE_END=========================================================
 */

package org.onap.appc.generator;

import com.fasterxml.jackson.databind.JsonNode;

import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
import java.util.regex.Pattern;

public class JsonHelper {

    private enum PopulateType{
        PARAMETER,
        SCHEMA_REF,
        ;
    }

    public static void populateRpcInputOutputParamsFromJson(final JsonNode jsonNode, Map<String, Set<String>> jsonBodyInputParams, Map<String, Set<String>> jsonBodyOutputParams) {
        populateRpcInputOutputParamsFromJson(jsonNode,jsonBodyInputParams, jsonBodyOutputParams,PopulateType.PARAMETER,false);
    }
    public static void populateRpcInputOutputSchemaRefFromJson(final JsonNode jsonNode, Map<String, Set<String>> jsonBodyInputParams, Map<String, Set<String>> jsonBodyOutputParams) {
        populateRpcInputOutputParamsFromJson(jsonNode,jsonBodyInputParams, jsonBodyOutputParams,PopulateType.SCHEMA_REF,true);
    }
    public static void populateRpcInputOutputParamsFromJson(final JsonNode jsonNode, Map<String, Set<String>> jsonBodyInputParams, Map<String, Set<String>> jsonBodyOutputParams,PopulateType populateType,boolean normalizeRpcOperation) {
        //get all yang2json Input and output Params and populate jsonBodyInputParams & jsonBodyOutputParams
        for(Iterator<JsonNode> pathsElements = jsonNode.get("paths").elements(); pathsElements.hasNext(); ){
            JsonNode pathNode = pathsElements.next();
            String rpcOperation = pathNode.get("post").get("operationId").textValue();
            String bodyInputSchemaRef = pathNode.get("post").get("parameters").get(0).get("schema").get("properties").get("input").get("$ref").textValue();
            String bodyInputSchemaRefSuffix = JsonHelper.getStringSuffix(bodyInputSchemaRef,"/");
            rpcOperation = normalizeRpcOperation ? rpcOperation.replaceAll(Pattern.quote("-"),"").toLowerCase() : rpcOperation;

            String bodyOutputSchemaRef = pathNode.get("post").get("responses").get("200").get("schema").get("properties").get("output").get("$ref").textValue();
            String bodyOutputSchemaRefSuffix = JsonHelper.getStringSuffix(bodyOutputSchemaRef,"/");

            Set<String> inputParametersOrSchemaRef;
            Set<String> outputParametersOrSchemaRef;
            if(populateType == PopulateType.SCHEMA_REF){
                inputParametersOrSchemaRef = new HashSet();
                inputParametersOrSchemaRef.add(bodyInputSchemaRefSuffix);

                outputParametersOrSchemaRef = new HashSet();
                outputParametersOrSchemaRef.add(bodyOutputSchemaRefSuffix);
            }else{
                inputParametersOrSchemaRef = extractParams(jsonNode,bodyInputSchemaRefSuffix);
                outputParametersOrSchemaRef = extractParams(jsonNode,bodyOutputSchemaRefSuffix);
            }
            jsonBodyInputParams.put(rpcOperation,inputParametersOrSchemaRef);
            jsonBodyOutputParams.put(rpcOperation,outputParametersOrSchemaRef);
        }
    }

    private static Set<String> extractParams(final JsonNode jsonNode,String schemaRefName) {
        Iterator<Map.Entry<String, JsonNode>> fields = jsonNode.get("definitions").get(schemaRefName).get("properties").fields();
        Set<String> params = new HashSet();
        for( ;fields.hasNext(); ){
            Map.Entry<String, JsonNode> fieldEntry = fields.next();
            params.add(fieldEntry.getKey());
        }
        return params;
    }

    public static String getStringSuffix(String input, String afterString){
        int indexOf = input.lastIndexOf(afterString);
        String stringSuffix = (indexOf > -1) ? input.substring(indexOf+1) : input;
        return stringSuffix;
    }
}