|
3 | 3 | import java.io.Serializable;
|
4 | 4 | import java.lang.annotation.Annotation;
|
5 | 5 | import java.lang.reflect.Field;
|
| 6 | +import java.util.ArrayList; |
6 | 7 | import java.util.Arrays;
|
| 8 | +import java.util.Comparator; |
| 9 | +import java.util.HashMap; |
| 10 | +import java.util.LinkedHashMap; |
| 11 | +import java.util.List; |
| 12 | +import java.util.Map; |
| 13 | +import java.util.function.Predicate; |
7 | 14 |
|
8 | 15 | import com.fasterxml.jackson.databind.JsonMappingException;
|
| 16 | +import com.fasterxml.jackson.databind.JsonNode; |
9 | 17 | import com.fasterxml.jackson.databind.ObjectMapper;
|
10 | 18 | import com.fasterxml.jackson.databind.node.ArrayNode;
|
11 | 19 | import com.fasterxml.jackson.databind.node.ObjectNode;
|
12 | 20 | import com.fasterxml.jackson.module.jsonSchema.JsonSchema;
|
13 | 21 | import com.fasterxml.jackson.module.jsonSchema.JsonSchemaGenerator;
|
14 | 22 |
|
15 | 23 | import io.asfjava.ui.core.FormDefinitionGeneratorFactory;
|
| 24 | +import io.asfjava.ui.core.form.Tab; |
16 | 25 | import io.asfjava.ui.dto.UiForm;
|
17 | 26 |
|
18 | 27 | public final class UiFormSchemaGenerator {
|
19 | 28 |
|
20 | 29 | private static UiFormSchemaGenerator INSTANCE;
|
21 | 30 |
|
22 | 31 | public UiForm generate(Class<? extends Serializable> formDto) throws JsonMappingException {
|
| 32 | + Field[] declaredFields = formDto.getDeclaredFields(); |
23 | 33 | ObjectMapper mapper = new ObjectMapper();
|
24 |
| - JsonSchemaGenerator schemaGen = new JsonSchemaGenerator(mapper, new CustomSchemaFactoryWrapper()); |
25 |
| - JsonSchema schema = schemaGen.generateSchema(formDto); |
26 | 34 |
|
27 |
| - ArrayNode formDefinition = mapper.createArrayNode(); |
| 35 | + JsonSchemaGenerator schemaGen = initSchemaGen(mapper); |
| 36 | + JsonSchema schema = generateSchema(formDto, schemaGen); |
| 37 | + Map<Field, JsonNode> nodes = initFieldFormDefinition(mapper, formDto, declaredFields); |
| 38 | + |
| 39 | + handlerGroupedFields(); |
| 40 | + |
| 41 | + ObjectNode tabbedFields = handleTabbedFields(mapper, declaredFields, nodes); |
28 | 42 |
|
29 |
| - Arrays.stream(formDto.getDeclaredFields()).forEach(field -> buildFormDefinition(mapper, formDefinition, field)); |
| 43 | + ArrayNode formDefinition = mapper.createArrayNode(); |
| 44 | + formDefinition.add(tabbedFields); |
30 | 45 |
|
31 | 46 | return new UiForm(schema, formDefinition);
|
32 | 47 | }
|
33 | 48 |
|
34 |
| - private void buildFormDefinition(ObjectMapper mapper, ArrayNode formDefinitions, Field field) { |
| 49 | + private void handlerGroupedFields() { |
| 50 | + // TODO Grouping fieldset must handle it |
| 51 | + |
| 52 | + } |
| 53 | + |
| 54 | + private ObjectNode handleTabbedFields(ObjectMapper mapper, Field[] declaredFields, Map<Field, JsonNode> nodes) { |
| 55 | + Predicate<? super Field> checkTabAnnotation = field -> field.isAnnotationPresent(Tab.class); |
| 56 | + Comparator<? super Field> tabIndexComparator = (field1, field2) -> Integer |
| 57 | + .compare(field1.getAnnotation(Tab.class).index(), field2.getAnnotation(Tab.class).index()); |
| 58 | + |
| 59 | + Map<String, List<JsonNode>> groupedFieldsByTab = new LinkedHashMap<>(); |
| 60 | + |
| 61 | + Arrays.stream(declaredFields).filter(checkTabAnnotation).sorted(tabIndexComparator) |
| 62 | + .forEach(field -> groupFieldsByTab(nodes, field, groupedFieldsByTab)); |
| 63 | + |
| 64 | + ArrayNode tabs = mapper.createArrayNode(); |
| 65 | + |
| 66 | + groupedFieldsByTab.entrySet().stream().forEachOrdered(tabElements -> { |
| 67 | + ObjectNode tabNode = mapper.createObjectNode(); |
| 68 | + tabNode.put("title", tabElements.getKey()); |
| 69 | + ArrayNode tabItems = mapper.createArrayNode(); |
| 70 | + tabElements.getValue().stream().forEach(fieldNode -> tabItems.add(fieldNode)); |
| 71 | + tabNode.put("items", tabItems); |
| 72 | + tabs.add(tabNode); |
| 73 | + }); |
| 74 | + |
| 75 | + ObjectNode tabsNode = mapper.createObjectNode(); |
| 76 | + tabsNode.put("type", "tabs"); |
| 77 | + tabsNode.put("tabs", tabs); |
| 78 | + return tabsNode; |
| 79 | + |
| 80 | + } |
| 81 | + |
| 82 | + private Map<Field, JsonNode> initFieldFormDefinition(ObjectMapper mapper, Class<? extends Serializable> formDto, |
| 83 | + Field[] declaredFields) { |
| 84 | + Map<Field, JsonNode> nodes = new HashMap<>(); |
| 85 | + Arrays.stream(declaredFields).forEach(field -> buildFormDefinition(nodes, mapper, field)); |
| 86 | + return nodes; |
| 87 | + } |
| 88 | + |
| 89 | + private JsonSchema generateSchema(Class<? extends Serializable> formDto, JsonSchemaGenerator schemaGen) |
| 90 | + throws JsonMappingException { |
| 91 | + return schemaGen.generateSchema(formDto); |
| 92 | + } |
| 93 | + |
| 94 | + private JsonSchemaGenerator initSchemaGen(ObjectMapper mapper) { |
| 95 | + return new JsonSchemaGenerator(mapper, new CustomSchemaFactoryWrapper()); |
| 96 | + } |
| 97 | + |
| 98 | + private void groupFieldsByTab(Map<Field, JsonNode> nodes, Field field, Map<String, List<JsonNode>> groupedFields) { |
| 99 | + Tab tab = field.getAnnotation(Tab.class); |
| 100 | + List<JsonNode> fieldsGroupedByTab = groupedFields.get(tab.title()); |
| 101 | + if (fieldsGroupedByTab == null) { |
| 102 | + fieldsGroupedByTab = new ArrayList<>(); |
| 103 | + groupedFields.put(tab.title(), fieldsGroupedByTab); |
| 104 | + } |
| 105 | + fieldsGroupedByTab.add(nodes.get(field)); |
| 106 | + nodes.remove(field); |
| 107 | + } |
| 108 | + |
| 109 | + private void buildFormDefinition(Map<Field, JsonNode> nodes, ObjectMapper mapper, Field field) { |
35 | 110 | Arrays.stream(field.getAnnotations())
|
36 |
| - .forEach(annotation -> buildFieldDefinition(field, annotation, mapper, formDefinitions)); |
| 111 | + .forEach(annotation -> buildFieldDefinition(field, annotation, mapper, nodes)); |
37 | 112 | }
|
38 | 113 |
|
39 | 114 | private void buildFieldDefinition(Field field, Annotation annotation, ObjectMapper mapper,
|
40 |
| - ArrayNode formDefinitions) { |
| 115 | + Map<Field, JsonNode> nodes) { |
41 | 116 | ObjectNode fieldFormDefinition = mapper.createObjectNode();
|
42 | 117 | FormDefinitionGeneratorFactory.getInstance().getGenerator(annotation.annotationType().getName())
|
43 | 118 | .ifPresent(generator -> {
|
44 | 119 | generator.generate(fieldFormDefinition, field);
|
45 |
| - formDefinitions.add(fieldFormDefinition); |
| 120 | + nodes.put(field, fieldFormDefinition); |
46 | 121 | });
|
47 | 122 | }
|
48 | 123 |
|
|
0 commit comments