Java tutorial
/* (C) Copyright 2013-2016 The RISCOSS Project Consortium 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. */ /** * @author Alberto Siena **/ package eu.riscoss.server; import java.io.File; import java.io.FileInputStream; import java.util.ArrayList; import java.util.Collection; import java.util.HashMap; import java.util.HashSet; import java.util.List; import java.util.Map; import java.util.Set; import javax.ws.rs.DELETE; import javax.ws.rs.DefaultValue; import javax.ws.rs.GET; import javax.ws.rs.HeaderParam; import javax.ws.rs.POST; import javax.ws.rs.Path; import javax.ws.rs.PathParam; import javax.ws.rs.Produces; import javax.ws.rs.QueryParam; import javax.xml.parsers.DocumentBuilder; import javax.xml.parsers.DocumentBuilderFactory; import org.apache.poi.xssf.usermodel.XSSFRow; import org.apache.poi.xssf.usermodel.XSSFSheet; import org.apache.poi.xssf.usermodel.XSSFWorkbook; import org.w3c.dom.Document; import org.w3c.dom.Element; import org.w3c.dom.NodeList; import com.google.gson.Gson; import com.google.gson.JsonArray; import com.google.gson.JsonElement; import com.google.gson.JsonObject; import com.google.gson.JsonParser; import com.google.gson.JsonPrimitive; import eu.riscoss.dataproviders.RiskData; import eu.riscoss.db.RecordAbstraction; import eu.riscoss.db.RiscossDB; import eu.riscoss.db.RiscossElements; import eu.riscoss.db.RiskAnalysisSession; import eu.riscoss.db.SearchParams; import eu.riscoss.ram.algo.DownwardEntitySearch; import eu.riscoss.ram.algo.DownwardEntitySearch.DIRECTION; import eu.riscoss.ram.algo.TraverseCallback; import eu.riscoss.rdc.RDC; import eu.riscoss.rdc.RDCFactory; import eu.riscoss.rdc.RDCParameter; import eu.riscoss.shared.JEntityData; import eu.riscoss.shared.JEntityNode; import eu.riscoss.shared.JRASInfo; import eu.riscoss.shared.JRiskNativeData; import eu.riscoss.shared.Pair; import eu.riscoss.shared.RiscossUtil; @Path("entities") @Info("Management of entities") public class EntityManager { @GET @Path("/{domain}/list") @Info("Returns a list of the entities sotred in the given domain") public String list(@PathParam("domain") @Info("The selected domain") String domain, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { JsonArray a = new JsonArray(); RiscossDB db = null; try { db = DBConnector.openDB(domain, token); for (String name : db.entities()) { JsonObject o = new JsonObject(); o.addProperty("name", name); o.addProperty("layer", db.layerOf(name)); a.add(o); } } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } return a.toString(); } @GET @Path("/{domain}/list-hierarchy") @Info("Returns a list of the entities stored in the given domain with children information") public String listHierarchy(@PathParam("domain") @Info("The selected domain") String domain, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { JsonArray a = new JsonArray(); RiscossDB db = null; try { db = DBConnector.openDB(domain, token); for (String name : db.entities()) { JsonObject o = new JsonObject(); o.addProperty("name", name); o.addProperty("layer", db.layerOf(name)); { JsonArray ar = new JsonArray(); for (String e : db.getChildren(name)) { ar.add(new JsonPrimitive(e)); } o.add("children", ar); } { JsonArray ar = new JsonArray(); for (String e : db.getParents(name)) { ar.add(new JsonPrimitive(e)); } o.add("parents", ar); } a.add(o); } } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } return a.toString(); } /** * returns the list of entities in a specific layer. * @param domain * @param layer * @param token * @return * @throws Exception */ @GET @Path("/{domain}/{layer}/list") @Info("Returns the list of entities in a specific layer") public String list(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("layer") @Info("The selected layer") String layer, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { //TODO: check if layer exists! JsonArray a = new JsonArray(); RiscossDB db = null; try { db = DBConnector.openDB(domain, token); for (String name : db.entities(layer)) { JsonObject o = new JsonObject(); o.addProperty("name", name); o.addProperty("layer", layer); a.add(o); } } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } return a.toString(); } @GET @Path("/{domain}/search") @Info("Returns a list of entities that match the specified parameters across all layers") public String search(@PathParam("domain") @Info("The selected domain") String domain, @HeaderParam("token") @Info("The authentication token") String token, @QueryParam("query") String query) throws Exception { return searchNew(domain, token, "", query, "0", "0", "false", ""); } @GET @Path("/{domain}/{layer}/search") @Info("Returns a list of entities that match the specified parameters in a specified layer") public String searchNew(@PathParam("domain") @Info("The selected domain") String domain, @HeaderParam("token") @Info("The authentication token") String token, @DefaultValue("") @PathParam("layer") @Info("The layer to search in") String layer, @DefaultValue("") @QueryParam("query") @Info("The actual query (on the entity name") String query, @DefaultValue("0") @QueryParam("from") @Info("Index of the firs entity (for pagination") String strFrom, @DefaultValue("0") @QueryParam("max") @Info("Amount of entities to search") String strMax, @DefaultValue("f") @QueryParam("h") @Info("Tells the server to load the entity's hierarchy") String strHierarchy, @DefaultValue("") @QueryParam("f") @Info("Search flags") String flags) throws Exception { List<JEntityNode> result = new ArrayList<JEntityNode>(); RiscossDB db = null; try { db = DBConnector.openDB(domain, token); SearchParams params = new SearchParams(); params.setMax(strMax); params.setFrom(strFrom); params.setOptLoadHierarchy(flags.indexOf("h") != -1); params.setOptLoadHierarchy(strHierarchy); List<String> layers = new ArrayList<String>(); if (flags.indexOf("s") != -1) { layers = db.getScope(layer); } else { layers.add(layer); } for (String l : layers) { Collection<String> list = db.findEntities(l, query, params); for (String name : list) { JEntityNode jd = new JEntityNode(); jd.name = name; jd.layer = db.layerOf(name); result.add(jd); if (params.loadHierarchy = true) { loadDescendants(jd, db); } } } } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } return new Gson().toJson(result); } Set<String> set = new HashSet<String>(); private void loadDescendants(JEntityNode jparent, RiscossDB db) { for (String name : db.getChildren(jparent.name)) { if (set.contains(name)) { System.err.println("Cycle detected!!!!!!!!!!!!!"); JEntityNode jd = new JEntityNode(); jd.name = "CYCLE DETECTED! (" + name + ")"; jd.layer = db.layerOf(name); jparent.children.add(jd); // FIXME Somehow, somewhere, a cycle has been introduced. Currently we don't know where or how. For now, we just remove it when we find it. db.removeEntity(name, jparent.name); } else { JEntityNode jd = new JEntityNode(); jd.name = name; jd.layer = db.layerOf(name); jparent.children.add(jd); set.add(name); loadDescendants(jd, db); set.remove(name); } } } Map<String, JEntityData> getParents(String entity, RiscossDB db) { Map<String, JEntityData> map = new HashMap<String, JEntityData>(); for (String name : db.getParents(entity)) { JEntityData jd = new JEntityData(); jd.name = name; jd.layer = db.layerOf(name); map.put(name, jd); } return map; } Map<String, JEntityData> getRoots(String entity, RiscossDB db) { List<String> parents = db.getParents(entity); if (parents.size() < 1) return new HashMap<String, JEntityData>(); Map<String, JEntityData> map = new HashMap<String, JEntityData>(); return map; } @POST @Path("/{domain}/create") @Info("Creates a new entity and associates it to the given layer") @Produces("application/json") //TODO: remove parent. Extra call for adding parents. public String createEntity(@PathParam("domain") @Info("The selected domain") String domain, @QueryParam("name") @Info("The selected entity") String name, @QueryParam("layer") @Info("The layer name") String layer, @QueryParam("parent") @Info("A parent entity") String parent, @HeaderParam("token") @Info("The authentication token") String token) { // attention:filename sanitation is not directly notified to the user name = RiscossUtil.sanitize(name); RiscossDB db = null; try { db = DBConnector.openDB(domain, token); JsonObject ret = new JsonObject(); db.addEntity(name, layer); if (parent != null) { if (!"".equals(parent)) { if (db.existsEntity(parent)) { db.assignEntity(name, parent); } } } ret.addProperty("name", name); ret.addProperty("layer", layer); ret.addProperty("parent", parent); System.out.println(ret.toString()); return ret.toString(); // Response.ok(ret, // MediaType.APPLICATION_JSON).build(); } catch (Exception ex) { ex.printStackTrace(); return ""; } finally { DBConnector.closeDB(db); } } @GET @Path("/{domain}/{entity}/description") public String getDescription(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entity, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); String ret = db.getProperty(RiscossElements.ENTITY, entity, "description", ""); return ret; } catch (Exception e) { throw e; } finally { DBConnector.closeDB(db); } } @POST @Path("/{domain}/{entity}/description") public void setDescription(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entity, @HeaderParam("token") @Info("The authentication token") String token, @Info("The description string to be set") String description) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); db.setProperty(RiscossElements.ENTITY, entity, "description", description); } catch (Exception e) { throw e; } finally { DBConnector.closeDB(db); } } @POST @Path("/{domain}/{entity}/edit-layer") @Info("Edit the layer of the specified entity") public void editLayer(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entity, @HeaderParam("token") @Info("The authentication token") String token, @QueryParam("layer") @Info("The new layer") String layer) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); db.editLayer(entity, layer); } catch (Exception e) { throw e; } finally { DBConnector.closeDB(db); } } @DELETE @Path("/{domain}/{entity}/delete") @Info("Deleted the specified entity") public void deleteEntity(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entity, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); db.removeEntity(entity); } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } } @GET @Path("/{domain}/{entity}/data") @Info("Returns detailed information about the specified entity") public String getEntityData(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entity, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); JsonObject json = new JsonObject(); json.addProperty("name", entity); json.addProperty("layer", db.layerOf(entity)); { JsonArray a = new JsonArray(); for (String e : db.getParents(entity)) { a.add(new JsonPrimitive(e)); } json.add("parents", a); } { JsonArray a = new JsonArray(); for (String e : db.getChildren(entity)) { a.add(new JsonPrimitive(e)); } json.add("children", a); } { JsonArray jlist = new JsonArray(); for (RDC rdc : RDCFactory.get().listRDCs()) { String rdcName = rdc.getName(); if (db.isRDCEnabled(entity, rdcName)) { jlist.add(new JsonPrimitive(rdc.getName())); } } json.add("rdcs", jlist); } JsonArray array = new JsonArray(); for (String id : db.listUserData(entity)) { JsonObject o = (JsonObject) new JsonParser().parse(db.readRiskData(entity, id)); array.add(o); } json.add("userdata", array); System.out.println("Returning entity data: " + json.toString()); return json.toString(); } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } } //currently not used! @GET @Path("/{domain}/{entity}/data_new") public String getEntityData_new(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entity, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); Gson gson = new Gson(); JEntityData data = new JEntityData(); data.name = entity; data.layer = db.layerOf(entity); { List<String> list = new ArrayList<>(); for (String e : db.getParents(entity)) { list.add(e); } data.parents = list; } { List<String> list = new ArrayList<>(); for (String e : db.getChildren(entity)) { list.add(e); } data.children = list; } { List<String> list = new ArrayList<>(); for (RDC rdc : RDCFactory.get().listRDCs()) { String rdcName = rdc.getName(); if (db.isRDCEnabled(entity, rdcName)) { list.add(rdcName); } } data.rdcs = list; } { List<JRiskNativeData> list = new ArrayList<>(); for (String id : db.listUserData(entity)) { JRiskNativeData rd = gson.fromJson(db.readRiskData(entity, id), JRiskNativeData.class); list.add(rd); } data.userdata = list; } String json = gson.toJson(data); System.out.println("Returning: " + json); return json; } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } } @POST @Path("/{domain}/{entity}/rename") @Info("Rename the specified entity") public void renameEntity(@PathParam("domain") @Info("The work domain") String domain, @HeaderParam("token") @Info("The authentication token") String token, @PathParam("entity") @Info("The name of an existing entity") String name, @QueryParam("newname") @Info("The new name of the entity") String newName) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); db.renameEntity(name, newName); String layer = db.layerOf(newName); List<String> rcs = db.findCandidateRCs(layer); List<String> ids = new ArrayList<>(); for (String rc : rcs) { for (RecordAbstraction record : db.listRAS(name, rc)) { JRASInfo jras = new JRASInfo(record.getName(), record.getProperty("name", record.getName())); ids.add(jras.getId()); } } for (String id : ids) { RiskAnalysisSession ras = db.openRAS(id); ras.setTarget(newName); } } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } } @POST @Path("/{domain}/{entity}/parents") @Produces("application/json") @Info("Sets the specified entity as the child for all the given parent entities") public void setParents(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entity, @HeaderParam("token") @Info("The authentication token") String token, String parents) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); JsonObject json = (JsonObject) new JsonParser().parse(parents); List<String> old_parents = db.getParents(entity); for (String old_parent : old_parents) { db.removeEntity(entity, old_parent); } JsonArray a = json.get("list").getAsJsonArray(); for (int i = 0; i < a.size(); i++) { String parent = a.get(i).getAsString(); db.assignEntity(entity, parent); } } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } } @GET @Path("/{domain}/{entity}/parents") @Produces("application/json") @Info("Returns a list of parents") public String getParents(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entity, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); JsonObject json = new JsonObject(); JsonArray array = new JsonArray(); for (String e : db.getParents(entity)) { array.add(new JsonPrimitive(e)); } json.add("entities", array); return json.toString(); } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } } @POST @Path("/{domain}/{entity}/children") @Produces("application/json") @Info("Sets the given list of entities as children of the specified parent entity") public void setChildren(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entity, @HeaderParam("token") @Info("The authentication token") String token, String children) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); JsonObject json = (JsonObject) new JsonParser().parse(children); List<String> old_children = db.getChildren(entity); for (String old_child : old_children) { db.removeEntity(old_child, entity); } JsonArray a = json.get("list").getAsJsonArray(); List<String> scope = db.getScope(db.layerOf(entity)); for (int i = 0; i < a.size(); i++) { String child = a.get(i).getAsString(); if (child.equals(entity)) continue; if (hasDescendant(child, entity, db)) continue; if (!scope.contains(db.layerOf(child))) continue; db.assignEntity(child, entity); } } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } } @GET @Path("/{domain}/{entity}/candidatechild") @Info("Checks whether the specified entity could legally be set as child of another given entity (e.g., no cycles)") public String isCandidateChild(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entity, @QueryParam("child") @Info("The candidate child") String child, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); if (child.equals(entity)) throw new Exception("Invalid candidate child: an entity can not be child of itself"); if (hasDescendant(child, entity, db)) throw new Exception("Invalid candidate child: cycle detected"); List<String> scope = db.getScope(db.layerOf(entity)); if (!scope.contains(db.layerOf(child))) throw new Exception( "Invalid candidate child: must be in a layer equal or lower to the parent entity"); return "Ok"; } finally { DBConnector.closeDB(db); } } @GET @Path("/{domain}/{entity}/candidateparent") @Info("Checks whether the specified entity could legally be set as parent of another given entity (e.g., no cycles)") public String isCandidateParent(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The candidate parent") String entity, @QueryParam("child") @Info("The selected entity") String child, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); if (child.equals(entity)) throw new Exception("Invalid candidate child: an entity can not be parent of itself"); if (hasAncestor(child, entity, db)) throw new Exception("Invalid candidate child: cycle detected"); List<String> scope = db.getScope(db.layerOf(child)); if (!scope.contains(db.layerOf(entity))) throw new Exception( "Invalid candidate child: must be in a layer equal or higher to the parent entity"); return "Ok"; } finally { DBConnector.closeDB(db); } } /** * returns direct parents and children of an entity (not the whole hierarchy) * @param domain * @param entity * @param token * @return * @throws Exception */ @GET @Path("/{domain}/{entity}/hierarchy") @Produces("application/json") @Info("Returns direct parents and children of an entity (not the whole hierarchy)") public String getHierarchyInfo(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entity, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); JsonObject json = new JsonObject(); JsonArray array = new JsonArray(); for (String e : db.getParents(entity)) { array.add(new JsonPrimitive(e)); } json.add("parents", array); array = new JsonArray(); for (String e : db.getChildren(entity)) { array.add(new JsonPrimitive(e)); } json.add("children", array); return json.toString(); } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } } @GET @Path("/{domain}/{entity}/rdcs/list") @Info("Returns the list ov available RDC for the given entity; the list includes both enabled and not enabled RDCs") public String listRDCs(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entityName, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { JsonObject o = new JsonObject(); RiscossDB db = null; try { db = DBConnector.openDB(domain, token); for (RDC rdc : RDCFactory.get().listRDCs()) { String rdcName = rdc.getName(); boolean enabled = db.isRDCEnabled(entityName, rdcName); if (enabled) { JsonObject jrdc = new JsonObject(); JsonObject params = new JsonObject(); for (RDCParameter par : rdc.getParameterList()) { params.addProperty(par.getName(), db.getRDCParmeter(entityName, rdcName, par.getName(), "")); } jrdc.addProperty("enabled", enabled); jrdc.add("params", params); o.add(rdc.getName(), jrdc); } } System.out.println("Returning enabled rdcs list: " + o.toString()); return o.toString(); } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } } @POST @Path("/{domain}/{entity}/rdcs/store") @Info("Stores the parameters for each RDC to be associated to a selected entity") public void setRDCs(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The entity to set the paremeters") String entityName, @HeaderParam("token") @Info("The authentication token") String token, @Info("A list, containing the value for each parameter of each RDC, and a flag telling whether to enable the RDC") String rdcmapString) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); System.out.println("Received: " + rdcmapString); JsonObject json = (JsonObject) new JsonParser().parse(rdcmapString); for (Map.Entry<String, JsonElement> entry : json.entrySet()) { String rdcName = entry.getKey(); JsonObject o = entry.getValue().getAsJsonObject(); // Do we need to add this check? if (RDCFactory.get().getRDC(rdcName) == null) continue; boolean enabled = false; try { enabled = "true".equals(entry.getValue().getAsJsonObject().get("enabled").getAsString()); } catch (Exception ex) { } db.setRDCEnabled(entityName, rdcName, enabled); if (enabled) { for (RDCParameter par : RDCFactory.get().getRDC(rdcName).getParameterList()) { String value = o.get("params").getAsJsonObject().get(par.getName()).getAsString() .toString(); db.setRDCParmeter(entityName, rdcName, par.getName(), value); } } } } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } } @GET @Path("/{domain}/{entity}/rdcs/newrun") @Produces("application/json") @Info("Executes all the RDC associated to the given entity") public String runRDCS(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entityName, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); JsonObject json = new JsonObject(); String msg = "Data successfully stored in the data repository"; for (RDC rdc : RDCFactory.get().listRDCs()) { String rdcName = rdc.getName(); boolean enabled = db.isRDCEnabled(entityName, rdcName); if (enabled) { JsonObject o = new JsonObject(); for (RDCParameter par : rdc.getParameterList()) { rdc.setParameter(par.getName(), db.getRDCParmeter(entityName, rdcName, par.getName(), "")); } try { DBConnector.closeDB(db); Map<String, RiskData> values = rdc.getIndicators(entityName); db = DBConnector.openDB(domain, token); if (values == null) { throw new Exception("The RDC '" + rdcName + "' returned an empty map for the entity '" + entityName + "'"); } for (String key : values.keySet()) { RiskData rd = values.get(key); try { db.storeRiskData(rd.toJSON()); } catch (Exception e) { e.printStackTrace(); } } o.addProperty("result", "ok"); json.add(rdcName, o); } catch (Exception ex) { ex.printStackTrace(); msg = "Some data were not gathered and/or stored in the RDR"; o.addProperty("result", "error"); o.addProperty("error-message", ex.getMessage()); json.add(rdcName, o); } } } json.addProperty("msg", msg); System.out.println("Returning newrun: " + json.toString()); return json.toString(); } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } } @GET @Path("/{domain}/import") @Info("Import entities and info from file") public void importEntities(@PathParam("domain") @Info("The selected domain") String domain, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); readImportFile(db); } catch (Exception e) { throw e; } finally { DBConnector.closeDB(db); } } @GET @Path("/{domain}/{entity}/rd") @Info("Returns the RiskData associated to the given entity") public String getRiskData(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entity, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); JsonObject json = new JsonObject(); JsonArray array = new JsonArray(); for (String id : db.listRiskData(entity)) { JsonObject o = (JsonObject) new JsonParser().parse(db.readRiskData(entity, id)); array.add(o); } json.add("list", array); return json.toString(); } catch (Exception e) { throw e; } finally { DBConnector.closeDB(db); } } @GET @Path("/{domain}/{entity}/ras") @Produces("application/json") public String getRAD(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The selected entity") String entity, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); return db.readRASResult(entity); } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } } @GET @Path("/{domain}/{entity}/candidatechildren") @Info("Returns a list of entities that can be set as child of the given entity. Thi is useful to avoid circles.") public String getCandidateChildren(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The candidate parent entity") String entity, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); List<String> scope = db.getScope(db.layerOf(entity)); List<String> entities = new ArrayList<String>(); for (String layer : scope) { for (String candidate : db.entities(layer)) { if (!hasDescendant(candidate, entity, db)) { entities.add(candidate); } } } return new Gson().toJson(entities); } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } // return ""; } @GET @Path("/{domain}/checkimportfiles") @Info("Checks the existence of xlsx and xml files for entity import") public String checkImportFiles(@PathParam("domain") @Info("The selected domain") String domain, @HeaderParam("token") @Info("The authentication token") String token) { boolean entFileLoaded = false; boolean confFileLoaded = false; File ent = new File("resources/entities_info.xlsx"); if (ent.exists()) { entFileLoaded = true; } File conf = new File("resources/importation_config.xml"); if (conf.exists()) { confFileLoaded = true; } JsonObject json = new JsonObject(); json.addProperty("entFile", entFileLoaded); json.addProperty("confFile", confFileLoaded); return json.toString(); } @GET @Path("/{domain}/{entity}/candidateparents") @Info("Returns a list of entities that can be set as parent of the given entity. This is useful to avoid circles.") public String getCandidateParents(@PathParam("domain") @Info("The selected domain") String domain, @PathParam("entity") @Info("The candidate child entity") String entity, @HeaderParam("token") @Info("The authentication token") String token) throws Exception { RiscossDB db = null; try { db = DBConnector.openDB(domain, token); Collection<String> layers = db.layerNames(); List<String> entities = new ArrayList<String>(); String l = db.layerOf(entity); for (String layer : layers) { for (String candidate : db.entities(layer)) { if (!hasAncestor(candidate, entity, db)) { entities.add(candidate); } } if (layer.equals(l)) { break; } } return new Gson().toJson(entities); } catch (Exception ex) { throw ex; } finally { DBConnector.closeDB(db); } } class ConfigItem { public String sheet; public int nameColumn; public List<Pair<String, Pair<Integer, Integer>>> definedIdItem; public List<Pair<String, Integer>> definedValueItem; } private void readImportFile(RiscossDB db) throws Exception { //Load importing config xml file DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance(); DocumentBuilder builder = factory.newDocumentBuilder(); FileInputStream f = new FileInputStream("resources/importation_config.xml"); Document doc = builder.parse(f); Element element = doc.getDocumentElement(); //Get relationships config info NodeList importNodes = element.getElementsByTagName("relationships"); Element e = (Element) importNodes.item(0); String sheet = e.getElementsByTagName("sheet_name").item(0).getTextContent(); List<Integer> entitiesColumns = new ArrayList<>(); NodeList ent = e.getElementsByTagName("layer"); while (ent != null) { Element el = (Element) ent.item(0); if (el != null) { entitiesColumns.add(Integer.valueOf(el.getAttribute("entity_column")) - 1); ent = el.getElementsByTagName("layer"); } else break; } List<ConfigItem> config = new ArrayList<>(); //Get imported entitites config info NodeList cf = element.getElementsByTagName("entities"); Element impEnt = (Element) cf.item(0); NodeList configNodes = impEnt.getElementsByTagName("imported_entity"); for (int i = 0; i < configNodes.getLength(); ++i) { ConfigItem conf = new ConfigItem(); Element entity = (Element) configNodes.item(i); conf.sheet = entity.getElementsByTagName("sheet_name").item(0).getTextContent(); conf.nameColumn = Integer.parseInt(entity.getElementsByTagName("name_column").item(0).getTextContent()) - 1; List<Pair<String, Pair<Integer, Integer>>> definedIdItem = new ArrayList<>(); List<Pair<String, Integer>> definedValueItem = new ArrayList<>(); NodeList p = entity.getElementsByTagName("custom_information"); Element ee = (Element) p.item(0); NodeList prop = ee.getElementsByTagName("custom_field"); for (int j = 0; j < prop.getLength(); ++j) { Element b = (Element) prop.item(j); if (b.getElementsByTagName("id").item(0) != null) { String id = b.getElementsByTagName("id").item(0).getTextContent(); int column = Integer.parseInt(b.getElementsByTagName("value_column").item(0).getTextContent()) - 1; definedValueItem.add(new Pair<>(id, column)); } else { int column = Integer.parseInt(b.getElementsByTagName("id_column").item(0).getTextContent()) - 1; int val = Integer.parseInt(b.getElementsByTagName("value").item(0).getTextContent()); String prefix = b.getElementsByTagName("prefix").item(0).getTextContent(); definedIdItem.add(new Pair<>(prefix, new Pair<>(column, val))); } } conf.definedIdItem = definedIdItem; conf.definedValueItem = definedValueItem; config.add(conf); } File xlsx = new File("resources/entities_info.xlsx"); FileInputStream fis = new FileInputStream(xlsx); XSSFWorkbook wb = new XSSFWorkbook(fis); XSSFSheet ws = wb.getSheet(sheet); Map<String, Pair<String, String>> list = new HashMap<>(); Map<Integer, List<String>> entities = new HashMap<>(); List<Pair<String, String>> relationships = new ArrayList<>(); boolean read = true; int i = 6; while (read) { //For every valid row in xlsx file XSSFRow row = ws.getRow(i); if (row == null || row.getCell(0).toString().equals("")) read = false; else { //For every custom entity defined in i row for (int j = 0; j < config.size(); ++j) { String parent = row.getCell(config.get(j).nameColumn).toString(); if (!parent.equals("")) { int x = 0; while (x < entitiesColumns.size()) { if (entitiesColumns.get(x) == config.get(j).nameColumn) break; else ++x; } if (x > 0) relationships .add(new Pair<>(row.getCell(entitiesColumns.get(x - 1)).toString(), parent)); if (entities.containsKey(config.get(j).nameColumn)) entities.get(config.get(j).nameColumn).add(parent); else { List<String> n = new ArrayList<>(); n.add(parent); entities.put(config.get(j).nameColumn, n); } String prefix = ""; //For every custom info defined for j entity in i row for (int k = 0; k < config.get(j).definedIdItem.size(); ++k) { prefix = config.get(j).definedIdItem.get(k).getLeft(); if (!row.getCell(config.get(j).definedIdItem.get(k).getRight().getLeft()).toString() .equals("")) { checkNewInfo(parent, prefix + row .getCell(config.get(j).definedIdItem.get(k).getRight().getLeft()) .toString(), config.get(j).definedIdItem.get(k).getRight().getRight().toString(), list, db); } } for (int k = 0; k < config.get(j).definedValueItem.size(); ++k) { if (!row.getCell(config.get(j).definedValueItem.get(k).getRight()).toString() .equals("")) { String value = row.getCell(config.get(j).definedValueItem.get(k).getRight()) .toString(); checkNewInfo(parent, config.get(j).definedValueItem.get(k).getLeft(), value, list, db); } } } } ++i; } } Map<String, String> en = importEntities(entitiesColumns, entities, relationships, db); //Delete old imported data for entities Set<String> all = new HashSet<>(); for (List<String> s : entities.values()) all.addAll(s); for (String target : all) { for (String id : db.listRiskData(target)) { JsonObject o = (JsonObject) new JsonParser().parse(db.readRiskData(target, id)); if (o.get("type").toString().equals("\"imported\"")) { JsonObject delete = new JsonObject(); delete.addProperty("id", id); delete.addProperty("target", target); JsonArray array = new JsonArray(); array.add(delete); db.storeRiskData(delete.toString()); } } } for (String child : list.keySet()) { if (!list.get(child).getLeft().equals("") && !list.get(child).getRight().equals("")) storeRDR(child.split("@")[0], en.get(child.split("@")[0]), list.get(child).getLeft(), list.get(child).getRight(), db); } } private Map<String, String> importEntities(List<Integer> entitiesColumns, Map<Integer, List<String>> entities, List<Pair<String, String>> relationships, RiscossDB db) throws Exception { List<String> layers = (List<String>) db.layerNames(); Map<String, String> e = new HashMap<>(); //Load all entities stored in database Collection<String> ents = db.entities(); List<String> ent = new ArrayList<>(); for (String s : ents) { ent.add(s); } //Create those entities not currently created for (int i = 0; i < layers.size(); ++i) { List<String> entList = entities.get(entitiesColumns.get(i)); for (String en : entList) { if (!ent.contains(en)) { if (ent.contains("/")) throw new Error("Not a valid entity name (containing special #@/&% characters"); db.addEntity(en, layers.get(i)); } e.put(en, layers.get(i)); } } //Create relationships (right -> parent, left -> children) for (Pair<String, String> rel : relationships) { db.assignEntity(rel.getRight(), rel.getLeft()); } return e; } private void checkNewInfo(String parent, String license, String value, Map<String, Pair<String, String>> list, RiscossDB db) { if (!list.containsKey(parent + license) && !value.equals("")) { list.put(parent + "@" + license, new Pair<>(license, value)); } } private void storeRDR(String target, String layer, String license, String value, RiscossDB db) throws Exception { String json = db.getLayerData(layer, "ci"); if (json != null) { JsonElement j = new JsonParser().parse(json); if (!j.isJsonNull()) { JsonObject r = (JsonObject) new JsonParser().parse(json); for (JsonElement s : r.get("contextInfoList").getAsJsonArray()) { JsonObject p = s.getAsJsonObject(); if (p.get("id").toString().equals("\"" + license + "\"") || p.get("name").toString().equals("\"" + license + "\"")) { JsonObject o = new JsonObject(); o.addProperty("id", p.get("id").toString().substring(1, p.get("id").toString().length() - 1)); o.addProperty("target", target); o.addProperty("value", value); o.addProperty("datatype", p.get("type").toString().substring(1, p.get("type").toString().length() - 1)); o.addProperty("type", "custom"); o.addProperty("origin", "user"); System.out.println(o.toString()); db.storeRiskData(o.toString()); return; } } } } /*for (String id : db.listRiskData(target)) { JsonObject o = (JsonObject) new JsonParser().parse(db.readRiskData(target, id)); if (id.equals(license) || o.get("name").equals(license)) { JsonObject s = new JsonObject(); s.addProperty("id", id); s.addProperty("target", target); s.addProperty("value", value); s.addProperty("datatype", o.get("datatype").toString()); s.addProperty("type", "custom"); s.addProperty("origin", "user"); db.storeRiskData(s.toString()); return; } }*/ //If layer data did not exist JsonObject o = new JsonObject(); o.addProperty("id", license); o.addProperty("target", target); o.addProperty("value", value); o.addProperty("datatype", "CUSTOM"); o.addProperty("type", "imported"); o.addProperty("origin", "user"); db.storeRiskData(o.toString()); } static class EntityFinder extends TraverseCallback<String> { boolean found = false; public EntityFinder(String storedValue) { super(storedValue); } @Override public boolean onEntityFound(String entity) { found = getValue().equals(entity); return found; } } boolean hasDescendant(String entity, String descendant, RiscossDB db) { DownwardEntitySearch ett = new DownwardEntitySearch(db); EntityFinder finder = new EntityFinder(descendant); ett.analyseEntity(entity, finder); return finder.found; } boolean hasAncestor(String entity, String ancestor, RiscossDB db) { DownwardEntitySearch ett = new DownwardEntitySearch(db, DIRECTION.UP); EntityFinder finder = new EntityFinder(ancestor); ett.analyseEntity(entity, finder); return finder.found; } }