private void addGetRepresentation( OJAnnotatedInterface enumLookupInf, OJAnnotatedClass enumLookup) { OJAnnotatedOperation getInf = new OJAnnotatedOperation("get", UmlgRestletGenerationUtil.Representation); enumLookupInf.addToOperations(getInf); getInf.addAnnotationIfNew(new OJAnnotationValue(UmlgRestletGenerationUtil.Get, "json")); enumLookupInf.addToOperations(getInf); OJAnnotatedOperation get = new OJAnnotatedOperation("get", UmlgRestletGenerationUtil.Representation); get.addToThrows(UmlgRestletGenerationUtil.ResourceException); enumLookup.addToImports(UmlgRestletGenerationUtil.ResourceException); UmlgGenerationUtil.addOverrideAnnotation(get); enumLookup.addToOperations(get); get.getBody() .addToStatements( "String enumQualifiedName = getQuery().getFirst(\"enumQualifiedName\").getValue();"); get.getBody() .addToStatements( "Class<?> enumClass = " + UmlgGenerationUtil.QualifiedNameClassMap.getLast() + ".INSTANCE.get(enumQualifiedName)"); enumLookup.addToImports(UmlgGenerationUtil.QualifiedNameClassMap); get.getBody() .addToStatements( UmlgGenerationUtil.UmlgEnum.getLast() + "[] enumConstants = (" + UmlgGenerationUtil.UmlgEnum.getLast() + "[])enumClass.getEnumConstants()"); enumLookup.addToImports(UmlgGenerationUtil.UmlgEnum); get.getBody().addToStatements("StringBuilder json = new StringBuilder()"); get.getBody().addToStatements("json.append(\"{\\\"data\\\": [\")"); get.getBody() .addToStatements( "json.append(" + UmlgGenerationUtil.ToJsonUtil.getLast() + ".enumsToJson(Arrays.asList(enumConstants)))"); enumLookup.addToImports("java.util.Arrays"); enumLookup.addToImports(UmlgGenerationUtil.ToJsonUtil); get.getBody().addToStatements("json.append(\"]}\")"); get.getBody() .addToStatements( "return new " + UmlgRestletGenerationUtil.JsonRepresentation.getLast() + "(json.toString())"); enumLookup.addToImports(UmlgRestletGenerationUtil.JsonRepresentation); }
private void buildToJsonForOption( PropertyWrapper pWrap, OJAnnotatedClass annotatedClass, OJBlock block) { SortedSet<Classifier> concreteImplementations = new TreeSet<Classifier>( new Comparator<Classifier>() { @Override public int compare(Classifier o1, Classifier o2) { return o1.getName().compareTo(o2.getName()); } }); // For derived unions only show tabs for each property that makes up the union. if (pWrap.isDerivedUnion()) { List<Property> subsettingProperties = ModelLoader.INSTANCE.findSubsettingProperties(pWrap.getProperty()); for (Property subsettingProperty : subsettingProperties) { concreteImplementations.addAll( UmlgClassOperations.getConcreteImplementations( (Classifier) subsettingProperty.getType())); } } else { // For non derived union show all concrete implementations concreteImplementations = UmlgClassOperations.getConcreteImplementations((Classifier) pWrap.getType()); } Set<Classifier> concreteImplementationsFrom = UmlgClassOperations.getConcreteImplementations((Classifier) pWrap.getOwningType()); if (!concreteImplementationsFrom.isEmpty()) { annotatedClass.addToImports(UmlgGenerationUtil.ToJsonUtil); // blok get reassigned for the meta data if statement OJBlock returnBlock = block; block.addToStatements("StringBuilder json = new StringBuilder()"); block.addToStatements("json.append(\"[\")"); int count = 1; // For meta data, put where one is navigating to first, then where on is // navigating from // This is consistent with navigating to a entity with a vertex where // there is no navigating from. // i.e. the first meta data in the array is the entity navigating to. for (Classifier concreteClassifierTo : concreteImplementations) { annotatedClass.addToImports(UmlgClassOperations.getPathName(concreteClassifierTo)); for (Classifier concreteClassifierFrom : concreteImplementationsFrom) { block.addToStatements("json.append(\"{\")"); block.addToStatements("json.append(\" \\\"meta\\\" : {\")"); block.addToStatements( "json.append(\"\\\"qualifiedName\\\": \\\"" + pWrap.getQualifiedName() + "\\\"\")"); // The execute ocl query resource is only required if the below // visitor is available if (RestletVisitors.containsVisitorForClass(QueryExecuteResourceBuilder.class) && (pWrap .getType() .getQualifiedName() .equals(UmlgRestletGenerationUtil.instanceQueryQualifiedName) || pWrap .getType() .getQualifiedName() .equals(UmlgRestletGenerationUtil.classQueryQualifiedName))) { block.addToStatements( "json.append(\", \\\"oclExecuteUri\\\": \\\"/" + pWrap.getModel().getName() + "/{contextId}/oclExecuteQuery\\\"\")"); } block.addToStatements("json.append(\", \\\"to\\\": \")"); OJBlock conditionBlockFrom = new OJBlock(); annotatedClass.addToImports(UmlgClassOperations.getPathName(concreteClassifierFrom)); conditionBlockFrom.addToStatements( "json.append(" + UmlgClassOperations.propertyEnumName(concreteClassifierTo) + ".asJson())"); conditionBlockFrom.addToStatements("json.append(\", \\\"from\\\": \")"); conditionBlockFrom.addToStatements( "json.append(" + UmlgClassOperations.propertyEnumName(concreteClassifierFrom) + ".asJson())"); annotatedClass.addToImports( UmlgClassOperations.getPathName(concreteClassifierFrom) .append(UmlgClassOperations.propertyEnumName(concreteClassifierFrom))); block.addToStatements(conditionBlockFrom); annotatedClass.addToImports( UmlgClassOperations.getPathName(pWrap.getOwningType()) .append(UmlgClassOperations.propertyEnumName(pWrap.getOwningType()))); annotatedClass.addToImports( UmlgClassOperations.getPathName(concreteClassifierTo) .append(UmlgClassOperations.propertyEnumName(concreteClassifierTo))); if (count++ < (concreteImplementations.size() * concreteImplementationsFrom.size())) { block.addToStatements("json.append(\"}},\")"); } else { block.addToStatements("json.append(\"}}\")"); } } block = returnBlock; } returnBlock.addToStatements("json.append(\"]\")"); returnBlock.addToStatements( "return new " + UmlgRestletGenerationUtil.JsonRepresentation.getLast() + "(json.toString())"); } else { // TODO not thought through block.addToStatements("return null"); } }
private void buildToJson(PropertyWrapper pWrap, OJAnnotatedClass annotatedClass, OJBlock block) { // This is very important to be a sorted set. The get and options method need to return the meta // data in the same order. // This allows the client to merge them easily SortedSet<Classifier> sortedConcreteImplementations = new TreeSet<Classifier>( new Comparator<Classifier>() { @Override public int compare(Classifier o1, Classifier o2) { return o1.getName().compareTo(o2.getName()); } }); // For derived unions only show tabs for each property that makes up the union. if (pWrap.isDerivedUnion()) { List<Property> subsettingProperties = ModelLoader.INSTANCE.findSubsettingProperties(pWrap.getProperty()); for (Property subsettingProperty : subsettingProperties) { sortedConcreteImplementations.addAll( UmlgClassOperations.getConcreteImplementations( (Classifier) subsettingProperty.getType())); } } else { // For non derived union show all concrete implementations sortedConcreteImplementations = UmlgClassOperations.getConcreteImplementations((Classifier) pWrap.getType()); } Set<Classifier> concreteImplementationsFrom = UmlgClassOperations.getConcreteImplementations((Classifier) pWrap.getOwningType()); if (!concreteImplementationsFrom.isEmpty()) { annotatedClass.addToImports(UmlgGenerationUtil.ToJsonUtil); // block get reassigned for the meta data if statement OJBlock returnBlock = block; block.addToStatements("StringBuilder json = new StringBuilder()"); block.addToStatements("json.append(\"[\")"); int count = 1; // For meta data, put where one is navigating to first, then where on is // navigating from // This is consistent with navigating to a entity with a vertex where // there is no navigating from. // i.e. the first meta data in the array is the entity navigating to. for (Classifier concreteClassifierTo : sortedConcreteImplementations) { annotatedClass.addToImports(UmlgClassOperations.getPathName(concreteClassifierTo)); if (pWrap.isOne()) { block.addToStatements("json.append(\"{\\\"data\\\": \")"); } else { block.addToStatements("json.append(\"{\\\"data\\\": [\")"); } if (pWrap.isOne()) { OJIfStatement ifOneInstanceOf = new OJIfStatement( "parentResource." + pWrap.getter() + "() != null && parentResource." + pWrap.getter() + "().getClass() == " + UmlgClassOperations.getPathName(concreteClassifierTo).getLast() + ".class"); ifOneInstanceOf.addToThenPart( "json.append(" + UmlgGenerationUtil.ToJsonUtil.getLast() + ".toJsonWithoutCompositeParent(parentResource." + pWrap.getter() + "()))"); ifOneInstanceOf.addToElsePart("json.append(\"null\")"); block.addToStatements(ifOneInstanceOf); } else { block.addToStatements( "json.append(" + UmlgGenerationUtil.ToJsonUtil.getLast() + ".toJsonWithoutCompositeParent(parentResource." + pWrap.getter() + "().select(new " + UmlgGenerationUtil.BooleanExpressionEvaluator.getCopy() .addToGenerics(UmlgClassOperations.getPathName(pWrap.getType())) .getLast() + "() {\n @Override\n public Boolean evaluate(" + UmlgClassOperations.getPathName(pWrap.getType()).getLast() + " e) {\n return e.getClass() == " + UmlgClassOperations.getPathName(concreteClassifierTo).getLast() + ".class;\n }\n })))"); annotatedClass.addToImports(UmlgGenerationUtil.BooleanExpressionEvaluator); } annotatedClass.addToImports(UmlgClassOperations.getPathName(pWrap.getType())); if (pWrap.isOne()) { block.addToStatements("json.append(\",\")"); } else { block.addToStatements("json.append(\"],\")"); } block.addToStatements("json.append(\" \\\"meta\\\" : {\")"); block.addToStatements( "json.append(\"\\\"qualifiedName\\\": \\\"" + pWrap.getQualifiedName() + "\\\"\")"); block.addToStatements( "json.append(\",\\\"qualifiedNameFrom\\\": \\\"\" + parentResource.getQualifiedName() + \"\\\"\")"); block.addToStatements( "json.append(\",\\\"qualifiedNameTo\\\": \\\"" + concreteClassifierTo.getQualifiedName() + "\\\"\")"); block.addToStatements("json.append(\"}\")"); if (sortedConcreteImplementations.size() != 1 && count != sortedConcreteImplementations.size()) { block.addToStatements("json.append(\"}, \")"); } block = returnBlock; count++; } returnBlock.addToStatements("json.append(\"}]\")"); returnBlock.addToStatements( "return new " + UmlgRestletGenerationUtil.JsonRepresentation.getLast() + "(json.toString())"); } else { // TODO not thought through block.addToStatements("return null"); } }
private void addPostObjectRepresentation(PropertyWrapper pWrap, OJAnnotatedClass annotatedClass) { OJAnnotatedOperation post = new OJAnnotatedOperation("post", UmlgRestletGenerationUtil.Representation); post.addToParameters(new OJParameter("entity", UmlgRestletGenerationUtil.Representation)); post.addToThrows(UmlgRestletGenerationUtil.ResourceException); annotatedClass.addToImports(UmlgRestletGenerationUtil.ResourceException); UmlgGenerationUtil.addOverrideAnnotation(post); UmlgGenerationUtil.addSuppressWarning(post); PropertyWrapper otherEndPWrap = new PropertyWrapper(pWrap.getOtherEnd()); OJPathName parentPathName = otherEndPWrap.javaBaseTypePath(); post.getBody() .addToStatements( "this." + parentPathName.getLast().toLowerCase() + "Id = " + UmlgRestletGenerationUtil.UmlgURLDecoder.getLast() + ".decode((String)getRequestAttributes().get(\"" + parentPathName.getLast().toLowerCase() + "Id\"))"); annotatedClass.addToImports(UmlgRestletGenerationUtil.UmlgURLDecoder); post.getBody() .addToStatements( parentPathName.getLast() + " parentResource = " + UmlgGenerationUtil.UMLGAccess + "." + UmlgGenerationUtil.getEntity + "(" + parentPathName.getLast().toLowerCase() + "Id" + ")"); OJTryStatement ojTryStatement = new OJTryStatement(); OJField mapper = new OJField("mapper", UmlgGenerationUtil.ObjectMapper); mapper.setInitExp( UmlgGenerationUtil.ObjectMapperFactory.getLast() + ".INSTANCE.getObjectMapper()"); annotatedClass.addToImports(UmlgGenerationUtil.ObjectMapperFactory); ojTryStatement.getTryPart().addToLocals(mapper); OJPathName pathName = new OJPathName("java.util.Map").addToGenerics("String").addToGenerics("Object"); OJAnnotatedField entityText = new OJAnnotatedField("entityText", "String"); entityText.setInitExp("entity.getText()"); ojTryStatement.getTryPart().addToLocals(entityText); OJField resultMap = new OJField( "resultMap", new OJPathName("java.util.Map") .addToGenerics( new OJPathName("Class<? extends " + pWrap.javaBaseTypePath().getLast() + ">")) .addToGenerics( "List<" + UmlgRestletGenerationUtil.UmlgNodeJsonHolder.getLast() + ">")); resultMap.setInitExp( "new HashMap<Class<? extends " + pWrap.javaBaseTypePath().getLast() + ">, List<" + UmlgRestletGenerationUtil.UmlgNodeJsonHolder.getLast() + ">>()"); annotatedClass.addToImports("java.util.HashMap"); annotatedClass.addToImports("java.util.List"); annotatedClass.addToImports(UmlgRestletGenerationUtil.UmlgNodeJsonHolder); ojTryStatement.getTryPart().addToLocals(resultMap); ojTryStatement .getTryPart() .addToStatements( pathName.getLast() + " overloaded = mapper.readValue(" + entityText.getName() + ", Map.class)"); ojTryStatement.getTryPart().addToStatements("Object o = overloaded.get(\"insert\")"); // Insert OJIfStatement ifInsert = new OJIfStatement("o != null"); OJIfStatement ifArrayForInsert = new OJIfStatement("o instanceof ArrayList"); OJPathName genericsForArray = new OJPathName("java.util.Map").addToGenerics("String").addToGenerics("Object"); OJField array = new OJField("array", new OJPathName("java.util.List").addToGenerics(genericsForArray)); array.setInitExp("(ArrayList<Map<String, Object>>)o"); ifArrayForInsert.getThenPart().addToLocals(array); ifInsert.addToThenPart(ifArrayForInsert); ojTryStatement.getTryPart().addToStatements(ifInsert); OJForStatement forArray = new OJForStatement( "overloadedJsonMap", new OJPathName("java.util.Map") .addToGenerics(new OJPathName("String")) .addToGenerics(new OJPathName("Object")), "array"); ifArrayForInsert.addToThenPart(forArray); forArray.getBody().addToStatements("add(resultMap, parentResource, overloadedJsonMap)"); OJField map = new OJField( "overloadedJsonMap", new OJPathName("java.util.Map").addToGenerics("String").addToGenerics("Object")); map.setInitExp("(Map<String, Object>) o"); ifArrayForInsert.setElsePart(new OJBlock()); ifArrayForInsert.getElsePart().addToLocals(map); ifArrayForInsert .getElsePart() .addToStatements("add(resultMap, parentResource, overloadedJsonMap)"); addPostResource(pWrap, annotatedClass, parentPathName); // Delete ojTryStatement.getTryPart().addToStatements("o = overloaded.get(\"delete\")"); OJIfStatement ifDelete = new OJIfStatement("o != null"); OJIfStatement ifArrayForDelete = new OJIfStatement("o instanceof ArrayList"); genericsForArray = new OJPathName("java.util.Map").addToGenerics("String").addToGenerics("Object"); array = new OJField("array", new OJPathName("java.util.ArrayList").addToGenerics(genericsForArray)); array.setInitExp("(ArrayList<Map<String, Object>>)o"); ifArrayForDelete.getThenPart().addToLocals(array); ifDelete.addToThenPart(ifArrayForDelete); ojTryStatement.getTryPart().addToStatements(ifDelete); forArray = new OJForStatement( "map", new OJPathName("java.util.Map") .addToGenerics(new OJPathName("String")) .addToGenerics(new OJPathName("Object")), "array"); ifArrayForDelete.addToThenPart(forArray); if (pWrap.isComposite()) { forArray.getBody().addToStatements("delete(map)"); } else { forArray.getBody().addToStatements("delete(parentResource, map)"); } map = new OJField( "map", new OJPathName("java.util.Map").addToGenerics("String").addToGenerics("Object")); map.setInitExp("(Map<String, Object>) o"); ifArrayForDelete.setElsePart(new OJBlock()); ifArrayForDelete.getElsePart().addToLocals(map); if (pWrap.isComposite()) { ifArrayForDelete.getElsePart().addToStatements("delete(map)"); } else { ifArrayForDelete.getElsePart().addToStatements("delete(parentResource, map)"); } addDeleteResource(pWrap, annotatedClass, parentPathName); // Update ojTryStatement.getTryPart().addToStatements("o = overloaded.get(\"update\")"); OJIfStatement ifUpdate = new OJIfStatement("o != null"); OJIfStatement ifArrayForUpdate = new OJIfStatement("o instanceof ArrayList"); genericsForArray = new OJPathName("java.util.Map").addToGenerics("String").addToGenerics("Object"); array = new OJField("array", new OJPathName("java.util.ArrayList").addToGenerics(genericsForArray)); array.setInitExp("(ArrayList<Map<String, Object>>)o"); ifArrayForUpdate.getThenPart().addToLocals(array); ifUpdate.addToThenPart(ifArrayForUpdate); ojTryStatement.getTryPart().addToStatements(ifUpdate); forArray = new OJForStatement( "overloadedJsonMap", new OJPathName("java.util.Map") .addToGenerics(new OJPathName("String")) .addToGenerics(new OJPathName("Object")), "array"); ifArrayForUpdate.addToThenPart(forArray); if (pWrap.isOrdered()) { forArray.getBody().addToStatements("put(resultMap, parentResource, overloadedJsonMap)"); } else { forArray.getBody().addToStatements("put(resultMap, overloadedJsonMap)"); } map = new OJField( "overloadedJsonMap", new OJPathName("java.util.Map").addToGenerics("String").addToGenerics("Object")); map.setInitExp("(Map<String, Object>) o"); ifArrayForUpdate.setElsePart(new OJBlock()); ifArrayForUpdate.getElsePart().addToLocals(map); if (pWrap.isOrdered()) { // Include the parent as it will be needed to add the child at a particular index ifArrayForUpdate .getElsePart() .addToStatements("put(resultMap, parentResource, overloadedJsonMap)"); } else { ifArrayForUpdate.getElsePart().addToStatements("put(resultMap, overloadedJsonMap)"); } addPutResource(pWrap, annotatedClass, parentPathName); // Check if transaction needs commiting commitOrRollback(ojTryStatement); OJBlock jsonResultBlock = new OJBlock(); ojTryStatement.getTryPart().addToStatements(jsonResultBlock); OJField result = new OJField("result", "java.lang.StringBuilder"); result.setInitExp("new StringBuilder(\"[\")"); jsonResultBlock.addToLocals(result); OJField count = new OJField("count", "int"); count.setInitExp("1"); jsonResultBlock.addToLocals(count); OJForStatement forConcreteClassifiers = new OJForStatement( "baseClass", new OJPathName("Class") .addToGenerics("? extends " + pWrap.javaBaseTypePath().getLast()), "resultMap.keySet()"); jsonResultBlock.addToStatements(forConcreteClassifiers); if (pWrap.isOne()) { forConcreteClassifiers.getBody().addToStatements("result.append(\"{\\\"data\\\": \")"); } else { forConcreteClassifiers.getBody().addToStatements("result.append(\"{\\\"data\\\": [\")"); } OJField objectList = new OJField( "objectList", new OJPathName("java.util.List") .addToGenerics(UmlgRestletGenerationUtil.UmlgNodeJsonHolder)); objectList.setInitExp("resultMap.get(baseClass)"); forConcreteClassifiers.getBody().addToLocals(objectList); OJField objectListCount = new OJField("objectListCount", new OJPathName("int")); objectListCount.setInitExp("1"); forConcreteClassifiers.getBody().addToLocals(objectListCount); OJForStatement forObjectList = new OJForStatement("object", UmlgRestletGenerationUtil.UmlgNodeJsonHolder, "objectList"); forObjectList.getBody().addToStatements("result.append(object.toJson())"); OJIfStatement ifObjectListCountSmallerThanSize = new OJIfStatement("objectListCount++ < objectList.size()", "result.append(\",\")"); forObjectList.getBody().addToStatements(ifObjectListCountSmallerThanSize); forConcreteClassifiers.getBody().addToStatements(forObjectList); if (pWrap.isOne()) { forConcreteClassifiers.getBody().addToStatements("result.append(\",\")"); } else { forConcreteClassifiers.getBody().addToStatements("result.append(\"],\")"); } forConcreteClassifiers.getBody().addToStatements("result.append(\" \\\"meta\\\" : {\")"); forConcreteClassifiers .getBody() .addToStatements( "result.append(\"\\\"qualifiedName\\\": \\\"" + pWrap.getQualifiedName() + "\\\"\")"); forConcreteClassifiers.getBody().addToStatements("result.append(\", \\\"to\\\": \")"); OJIfStatement ifClassInstanceOf = null; Set<Classifier> concreteImplementations = UmlgClassOperations.getConcreteImplementations((Classifier) pWrap.getType()); if (!concreteImplementations.isEmpty()) { ifClassInstanceOf = new OJIfStatement(); forConcreteClassifiers.getBody().addToStatements(ifClassInstanceOf); } boolean first = true; for (Classifier concreteImplementation : concreteImplementations) { OJBlock ojIfBlock; if (first) { first = false; ifClassInstanceOf.setCondition( "baseClass.equals(" + UmlgClassOperations.getPathName(concreteImplementation).getLast() + ".class)"); ojIfBlock = ifClassInstanceOf.getThenPart(); } else { ojIfBlock = ifClassInstanceOf.addToElseIfCondition( "baseClass.equals(" + UmlgClassOperations.getPathName(concreteImplementation).getLast() + ".class)", ""); } ojIfBlock.addToStatements( "result.append(" + UmlgClassOperations.propertyEnumName(concreteImplementation) + ".asJson())"); annotatedClass.addToImports( UmlgClassOperations.getPathName(concreteImplementation) .append(UmlgClassOperations.propertyEnumName(concreteImplementation))); ojIfBlock.addToStatements("result.append(\", \\\"from\\\": \")"); Classifier owningType = (Classifier) pWrap.getOwningType(); ojIfBlock.addToStatements( "result.append(" + UmlgClassOperations.propertyEnumName(owningType) + ".asJson())"); annotatedClass.addToImports( UmlgClassOperations.getPathName(owningType) .append(UmlgClassOperations.propertyEnumName(owningType))); ojIfBlock.addToStatements("result.append(\"}\")"); } OJIfStatement ifLast = new OJIfStatement("count++ == resultMap.size()"); ifLast.addToThenPart("result.append(\"}\")"); ifLast.addToElsePart("result.append(\"},\")"); forConcreteClassifiers.getBody().addToStatements(ifLast); if (!concreteImplementations.isEmpty()) { ifClassInstanceOf.addToElsePart( "throw new IllegalStateException(\"Unknown type \" + baseClass.getName())"); } jsonResultBlock.addToStatements("result.append(\"]\")"); jsonResultBlock.addToStatements( "return new " + UmlgRestletGenerationUtil.JsonRepresentation.getLast() + "(result.toString())"); ojTryStatement.setCatchParam(new OJParameter("e", new OJPathName("java.lang.Exception"))); ojTryStatement.getCatchPart().addToStatements(UmlgGenerationUtil.UMLGAccess + ".rollback()"); ojTryStatement .getCatchPart() .addToStatements( "throw " + UmlgRestletGenerationUtil.UmlgExceptionUtilFactory.getLast() + ".getTumlExceptionUtil().handle(e)"); annotatedClass.addToImports(UmlgRestletGenerationUtil.UmlgExceptionUtilFactory); post.getBody().addToStatements(ojTryStatement); annotatedClass.addToImports(parentPathName); annotatedClass.addToImports(UmlgGenerationUtil.UMLGPathName); annotatedClass.addToImports(UmlgRestletGenerationUtil.JsonRepresentation); annotatedClass.addToOperations(post); }