b7ff22251d3ce9faae9a4ece1e1de584616a4e14
[appc.git] / appc-directed-graph / dg-loader / provider / src / main / java / org / onap / sdnc / dg / loader / DGXMLGenerator.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * ONAP : APPC
4  * ================================================================================
5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Copyright (C) 2017 Amdocs
8  * =============================================================================
9  * Licensed under the Apache License, Version 2.0 (the "License");
10  * you may not use this file except in compliance with the License.
11  * You may obtain a copy of the License at
12  * 
13  *      http://www.apache.org/licenses/LICENSE-2.0
14  * 
15  * Unless required by applicable law or agreed to in writing, software
16  * distributed under the License is distributed on an "AS IS" BASIS,
17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18  * See the License for the specific language governing permissions and
19  * limitations under the License.
20  * 
21  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
22  * ============LICENSE_END=========================================================
23  */
24
25 package org.onap.sdnc.dg.loader;
26
27 import java.io.File;
28 import java.security.InvalidParameterException;
29 import java.util.ArrayList;
30 import java.util.List;
31 import javax.script.Invocable;
32 import javax.script.ScriptEngine;
33 import javax.script.ScriptEngineManager;
34 import org.apache.commons.io.FileUtils;
35 import org.apache.commons.io.IOUtils;
36 import org.slf4j.Logger;
37 import org.slf4j.LoggerFactory;
38
39
40 public class DGXMLGenerator {
41     private final static Logger logger = LoggerFactory.getLogger(DGXMLGenerator.class);
42
43     public static final String STRING_ENCODING = "utf-8";
44     public static final String JS_INTERFACE_DG_CONVERTOR = "dgconverter";
45     public static final String JS_METHOD_GET_NODE_TO_XML = "getNodeToXml";
46     public static final String GENERATOR_TEMPLATE_FILE = "js/dg_xml2json.js";
47
48     public void generateXMLFromJSON(String jsonPath, String xmlpath, String propertyPath)
49             throws InvalidParameterException {
50         try {
51             ScriptEngineManager manager = new ScriptEngineManager();
52             ScriptEngine engine = manager.getEngineByName("JavaScript");
53             if (!(engine instanceof Invocable)) {
54                 logger.error("Invoking methods is not supported.");
55                 return;
56             }
57             Invocable inv = (Invocable) engine;
58             // engine.eval(new
59             // FileReader(DGXMLGenerator.class.getClassLoader().getResource(GENERATOR_TEMPLATE_FILE).getPath()));
60             String js = IOUtils.toString(DGXMLGenerator.class.getClassLoader()
61                     .getResourceAsStream(GENERATOR_TEMPLATE_FILE), STRING_ENCODING);
62             engine.eval(js);
63
64             Object dgconverter = engine.get(JS_INTERFACE_DG_CONVERTOR);
65
66             List<File> files = new ArrayList<File>();
67             if (dgconverter != null) {
68                 File jsonPathFile = new File(jsonPath);
69                 if (jsonPathFile.isDirectory()) {
70                     String[] extensions = new String[] {"json", "JSON"};
71                     files = (List<File>) FileUtils.listFiles(jsonPathFile, extensions, true);
72                 } else if (jsonPathFile.isFile()) {
73                     files.add(jsonPathFile);
74                 } else {
75                     throw new InvalidParameterException("Failed to get the nature of the JSON path :" + jsonPath);
76                 }
77
78                 logger.info("JSON Files identified " + files.size());
79
80                 if (files.size() > 0) {
81                     boolean isXmlPathDeleted = FileUtils.deleteQuietly(new File(xmlpath));
82                     logger.info("Cleaning old DG XML under : " + xmlpath + ", delete status :"
83                             + isXmlPathDeleted);
84
85                     for (File file : files) {
86                         String dgJson = FileUtils.readFileToString(file, STRING_ENCODING);
87                         logger.info("Generating XML from  :" + file.getName());
88                         String xmlFileName =
89                                 xmlpath + "/" + file.getName().replace(".json", ".xml");
90
91                         Object dgXMl =
92                                 inv.invokeMethod(dgconverter, JS_METHOD_GET_NODE_TO_XML, dgJson);
93                         // Write the XML File
94                         if (dgXMl != null) {
95                             File xmlFile = new File(xmlFileName);
96                             FileUtils.writeStringToFile(xmlFile, dgXMl.toString(), STRING_ENCODING);
97                             logger.info("Generated XML File under  :" + xmlFile.getCanonicalPath());
98                         }
99                     }
100
101                 } else {
102                     logger.info("No JSON Files to generate XML");
103                 }
104             } else {
105                 logger.error("Couldn't get Java Script Engine..");
106             }
107         } catch (Exception e) {
108             logger.error("Failed to generate generateXMLFromJSON :" + e.getMessage());
109         }
110     }
111
112
113     public static void main(String[] args) {
114         try {
115             DGXMLGenerator application = new DGXMLGenerator();
116             String jsonPath = null;
117             String xmlPath = null;
118             String propertyPath = null;
119             // Generate, GenerateLoad, GenerateLoadActivate
120             // args = new String[]{"src/main/resources/json","src/test/resources/xml"};
121             logger.info("DGXML Conversion Started with arguments :" + args[0] + ":" + args[1]);
122             if (args != null && args.length >= 2) {
123                 jsonPath = args[0];
124                 xmlPath = args[1];
125             } else {
126                 throw new InvalidParameterException("Required inputs are missing <jsonPath> <xmlPath>");
127             }
128
129             application.generateXMLFromJSON(jsonPath, xmlPath, propertyPath);
130             logger.info("DGXML Conversion Completed...");
131         } catch (Exception e) {
132             logger.error("Failed in DG XML Generation :" + e.getMessage());
133             e.printStackTrace();
134         }
135
136     }
137
138 }