JsonListener.java
4.03 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package com.ishchuk.antlr.json;
import com.ishchuk.antlr.json.model.*;
import com.ishchuk.antlr4.JSONgramBaseListener;
import com.ishchuk.antlr4.JSONgramParser;
import java.util.*;
public class JsonListener extends JSONgramBaseListener {
private JsonModel jsonFile;
private Map<String, JsonElement> json;
private List<String> var;
private boolean read;
@Override
public void enterJson(JSONgramParser.JsonContext ctx) {
this.jsonFile = new JsonModel();
this.json = new HashMap<>();
this.var = new ArrayList<>();
this.read = true;
}
public JsonModel getJsonFile() {
return this.jsonFile;
}
@Override
public void enterContent(JSONgramParser.ContentContext ctx) {
if (read && Objects.nonNull(ctx.object())) {
ctx.object().keyValue().forEach(kv -> {
JsonElement elem = new JsonElement();
elem.setContent(kv.content());
json.put(kv.CHAINE().getSymbol().getText(), elem);
});
read = false;
} else {
System.out.println("content: " + ctx.getText());
String parent = getParent(ctx);
System.out.println("parent: " + parent);
if (json.containsKey(parent)) {
System.out.println("parent in json");
searchMap(json, parent);
} else {
System.out.println("NOT in json");
json.values().forEach(kv -> {
if (kv instanceof JsonObject) {
searchMap(((JsonObject) kv).getMap(), parent);
} else if (kv instanceof JsonArray) {
((JsonArray) kv).getArray().forEach(obj -> {
searchMap(obj.getMap(), parent);
});
}
});
}
}
}
private void searchMap(Map<String, JsonElement> json, final String parent) {
if (json.containsKey(parent)) {
System.out.println("parent in json");
if (json.get(parent).getContent() instanceof JSONgramParser.ContentContext) {
JSONgramParser.ContentContext content =
(JSONgramParser.ContentContext) json.get(parent).getContent();
if (Objects.nonNull(content.object())) {
json.replace(parent, prepareJsonObject(content));
} else if (Objects.nonNull(content.table())) {
JsonArray arr = new JsonArray();
content.table().content().forEach(cont -> {
JsonObject obj = prepareJsonObject(cont);
arr.getArray().add(obj);
});
arr.getArray().forEach(temp -> {
searchMap(temp.getMap(), parent);
});
json.replace(parent, arr);
} else {
JsonPrimitive prim = new JsonPrimitive();
prim.setValue(content.getText());
json.replace(parent, prim);
}
}
}
}
private JsonObject prepareJsonObject(JSONgramParser.ContentContext cont) {
JsonObject obj = new JsonObject();
cont.object().keyValue().forEach(kv -> {
JsonElement elem = new JsonElement();
elem.setContent(kv.content());
obj.getMap().put(kv.CHAINE().getSymbol().getText(), elem);
});
return obj;
}
private String getParent(JSONgramParser.ContentContext ctx) {
String parent;
if (Objects.nonNull(ctx.table()) || Objects.nonNull(ctx.object())) {
if (ctx.getParent().children.get(0).getText().equals("[")) {
parent = ctx.getParent().getParent().getParent().children.get(0).getText();
} else {
parent = ctx.getParent().children.get(0).getText();
}
} else {
parent = ctx.getParent().children.get(0).getText();
}
return parent;
}
}