aai-core/src/main/java/org/openecomp/aai/ingestModel/CreateWidgetModels.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * org.openecomp.aai
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8 you may not use this file except in compliance with the License.
   9 You may obtain a copy of the License at
  10
  11      http://www.apache.org/licenses/LICENSE-2.0
  12
  13 Unless required by applicable law or agreed to in writing, software
  14 distributed under the License is distributed on an "AS IS" BASIS,
  15 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16 See the License for the specific language governing permissions and
  17 limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.aai.ingestModel;
  22
  23 import java.io.File;
  24 import java.io.PrintWriter;
  25 import java.io.StringWriter;
  26 import java.util.ArrayList;
  27 import java.util.Map;
  28 import java.util.UUID;
  29
  30 import javax.xml.transform.stream.StreamSource;
  31
  32 import org.eclipse.persistence.dynamic.DynamicEntity;
  33 import org.eclipse.persistence.jaxb.JAXBMarshaller;
  34 import org.eclipse.persistence.jaxb.JAXBUnmarshaller;
  35 import org.eclipse.persistence.jaxb.MarshallerProperties;
  36 import org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContext;
  37
  38 import org.openecomp.aai.domain.model.AAIResource;
  39 import org.openecomp.aai.domain.model.AAIResources;
  40 import org.openecomp.aai.util.AAIConfig;
  41 import org.openecomp.aai.util.AAIConstants;
  42 import com.google.common.base.CaseFormat;
  43
  44 /**
  45  * The Class CreateWidgetModels.
  46  */
  47 public class CreateWidgetModels
  48 {
  49         /**
  50          * The main method.
  51          *
  52          * @param args the arguments
  53          * @throws Exception the exception
  54          */
  55         public static void main(String[] args) throws Exception {
  56
  57                 String _apiVersion = AAIConfig.get(AAIConstants.AAI_DEFAULT_API_VERSION_PROP);
  58                 String widgetJsonDir = null;
  59                 String modelVersion = null;
  60                 if (args.length > 0) {
  61                         if (args[0] != null) {
  62                                 _apiVersion = args[0];
  63                         }
  64                         if (args[1] != null) {
  65                                 widgetJsonDir = args[1];
  66                         }
  67                         if (args[2] != null) {
  68                                 modelVersion = args[2];
  69                         }
  70                 }
  71
  72                 if (widgetJsonDir == null) {
  73                         System.err.println("You must specify a directory for widgetModelJson");
  74                         System.exit(0);
  75                 }
  76                 if (modelVersion == null) {
  77                         System.err.println("You must specify a modelVersion");
  78                         System.exit(0);
  79                 }
  80
  81                 ArrayList<String> apiVersions = new ArrayList<String>();
  82                 apiVersions.add(_apiVersion);
  83                 final IngestModelMoxyOxm m = new IngestModelMoxyOxm();
  84                 m.init(apiVersions, false);
  85
  86                 AAIResources aaiResources = IngestModelMoxyOxm.aaiResourceContainer.get(_apiVersion);
  87
  88                 DynamicJAXBContext jaxbContext = aaiResources.getJaxbContext();
  89
  90                 // iterate the collection of resources
  91
  92                 ArrayList<String> processedWidgets = new ArrayList<String>();
  93                 for (Map.Entry<String, AAIResource> aaiResEnt : aaiResources.getAaiResources().entrySet()) {
  94                         DynamicEntity meObject = jaxbContext.newDynamicEntity("inventory.aai.openecomp.org." + _apiVersion + ".Model");
  95                         // no need for a ModelVers DynamicEntity
  96
  97                         AAIResource aaiRes = aaiResEnt.getValue();
  98
  99                         if (aaiRes.getResourceType().equals("node")) {
 100                                 String resource = aaiRes.getSimpleName();
 101
 102                                 if (processedWidgets.contains(resource)) {
 103                                         continue;
 104                                 }
 105                                 processedWidgets.add(resource);
 106
 107                                 String widgetName = CaseFormat.UPPER_CAMEL.to(CaseFormat.LOWER_HYPHEN, resource);
 108                                 String filePathString = widgetJsonDir + "/" + widgetName + "-" + modelVersion + ".json";
 109                                 File f = new File(filePathString);
 110
 111                                 String filePathString2 = widgetJsonDir + "/../widget-model-json-old/" + widgetName + "-" + modelVersion + ".json";
 112                                 File f2 = new File(filePathString2);
 113
 114                                 if(!f.exists() && !f.isDirectory()) {
 115
 116                                         if (f2.exists()) {
 117                                                 System.out.println("Using old file for " + resource + ".");
 118
 119                                                 JAXBUnmarshaller unmarshaller = jaxbContext.createUnmarshaller();
 120                                                 unmarshaller.setProperty("eclipselink.media-type", "application/json");
 121                                                 unmarshaller.setProperty("eclipselink.json.include-root", false);
 122                                                 Class<? extends DynamicEntity> resultClass = meObject.getClass();
 123                                                 meObject = (DynamicEntity) unmarshaller.unmarshal(new StreamSource(f2), resultClass).getValue();
 124                                                 // override, some of them are wrong
 125                                                 meObject.set("modelVersion", modelVersion);
 126                                         } else {
 127
 128                                                 System.out.println("Making new file for " + resource + ".");
 129                                                 meObject.set("modelInvariantId", UUID.randomUUID().toString());
 130                                                 meObject.set("modelType", "widget");
 131                                                 DynamicEntity mevObject = jaxbContext.newDynamicEntity("inventory.aai.openecomp.org." + _apiVersion + ".ModelVer");
 132                                                 DynamicEntity mevsObject = jaxbContext.newDynamicEntity("inventory.aai.openecomp.org." + _apiVersion + ".ModelVers");
 133                                                 mevObject.set("modelVersionId", UUID.randomUUID().toString());
 134                                                 mevObject.set("modelVersion", modelVersion);
 135                                                 mevObject.set("modelName", widgetName);
 136                                                 // make a list of dynamic Entities
 137                                                 ArrayList<DynamicEntity> mevsList = new ArrayList<DynamicEntity>();
 138                                                 // add this one, it will be the only one in the list in this case
 139                                                 mevsList.add(mevObject);
 140                                                 mevsObject.set("modelVer", mevsList);
 141                                                 // Have to figure out how to add my mev object to the mevsObject,
 142                                                 // the modelVers is a list of dynamic entities so we can just attach the array here
 143                                                 meObject.set("modelVers",mevsObject);
 144                                         }
 145
 146                                         // put it out as JSON
 147
 148                                         JAXBMarshaller marshaller = jaxbContext.createMarshaller();
 149                                         marshaller.setProperty(JAXBMarshaller.JAXB_FORMATTED_OUTPUT, true);
 150
 151                                         marshaller.setProperty("eclipselink.media-type", "application/json");
 152                                         marshaller.setProperty("eclipselink.json.include-root", false);
 153                                         marshaller.setProperty(MarshallerProperties.JSON_MARSHAL_EMPTY_COLLECTIONS, Boolean.FALSE) ;
 154
 155                                         StringWriter writer = new StringWriter();
 156                                         marshaller.marshal(meObject, writer);
 157                                         PrintWriter out = new PrintWriter(f);
 158                                         out.println(writer.toString());
 159                                         out.close();
 160
 161                                 } else {
 162                                         System.out.println("File already exists for " + resource + ".  Skipping.");
 163                                 }
 164                         }
 165                 }
 166                 System.exit(0);
 167         }
 168 }