public static Object gsonToPrimitive(JsonElement element) {
if (element.isJsonPrimitive()) {
JsonPrimitive prim = element.getAsJsonPrimitive();
if (prim.isString()) {
return prim.getAsString();
} else if (prim.isBoolean()) {
return prim.getAsBoolean();
} else if (prim.isNumber()) {
return prim.getAsNumber();
} else {
throw new IllegalArgumentException("Unknown Gson primitive: " + prim);
}
} else if (element.isJsonArray()) {
JsonArray array = element.getAsJsonArray();
List<Object> list = new ArrayList<Object>();
for (int i = 0; i < array.size(); i++) {
list.add(gsonToPrimitive(array.get(i)));
}
return list;
} else if (element.isJsonNull()) {
return null;
} else if (element.isJsonObject()) {
Map<String, Object> map = new HashMap<String, Object>();
for (Map.Entry<String, JsonElement> entry : element.getAsJsonObject().entrySet()) {
map.put(entry.getKey(), gsonToPrimitive(entry.getValue()));
}
return map;
} else {
throw new IllegalArgumentException("Unknown Gson value: " + element);
}
}
public WikidataEntity parse(String json) throws WpParseException {
JacksonTermedStatementDocument mwDoc;
try {
mwDoc = mapper.readValue(json, JacksonTermedStatementDocument.class);
} catch (IOException e) {
LOG.info("Error parsing: " + json);
throw new WpParseException(e);
}
WikidataEntity record = new WikidataEntity(mwDoc.getEntityId().getId());
// Aliases (multiple per language)
for (List<MonolingualTextValue> vlist : mwDoc.getAliases().values()) {
if (vlist.isEmpty()) continue;
if (!validLanguage(vlist.get(0).getLanguageCode())) continue;
Language lang = Language.getByLangCodeLenient(vlist.get(0).getLanguageCode());
record.getAliases().put(lang, new ArrayList<String>());
for (MonolingualTextValue v : vlist) {
record.getAliases().get(lang).add(v.getText());
}
}
// Descriptions (one per language)
for (MonolingualTextValue v : mwDoc.getDescriptions().values()) {
if (validLanguage(v.getLanguageCode())) {
Language lang = Language.getByLangCodeLenient(v.getLanguageCode());
record.getDescriptions().put(lang, v.getText());
}
}
// Labels (one per language)
for (MonolingualTextValue v : mwDoc.getLabels().values()) {
if (validLanguage(v.getLanguageCode())) {
Language lang = Language.getByLangCodeLenient(v.getLanguageCode());
record.getLabels().put(lang, v.getText());
}
}
// Claims (only for Item entities)
if (mwDoc instanceof JacksonItemDocument) {
for (List<JacksonStatement> statements :
((JacksonItemDocument) mwDoc).getJsonClaims().values()) {
for (JacksonStatement s : statements) {
record.getStatements().add(parseOneClaim(record, s));
}
}
}
return record;
}