@Override
protected void doGet(final HttpServletRequest req, final HttpServletResponse resp)
throws ServletException, IOException {
final Session session = createAdminSession();
try {
String statusFilter = req.getParameter("status");
if (statusFilter == null) {
// By default, only approved posts are returned.
statusFilter = RelatedUgcNodeIndexerExtension.FIELD_STATE_APPROVED;
}
final List<Post> mltPosts =
RelatedSearchUtil.getRelatedPosts(
getResourceResolver(session),
req.getParameter("q"),
statusFilter,
req.getParameter("resource-type"),
req.getParameter("component-path"),
MLT_FIELDS,
Integer.valueOf(req.getParameter("max-results")),
MIN_TERM_FREQ,
MIN_DOC_FREQ);
if (mltPosts == null) {
return;
}
final JSONObject json = new JSONObject();
JSONObject match;
final List<JSONObject> matches = new ArrayList<JSONObject>();
for (final Post p : mltPosts) {
match = new JSONObject();
match.put("subject", xssProtectionService.protectFromXSS(p.getSubject()));
match.put("replyCount", p.getRepliesCount());
match.put("url", p.getUrl());
match.put("id", p.getId());
match.put("created", p.getCreated());
match.put("createdBy", xssProtectionService.protectFromXSS(p.getCreatedBy().getFullName()));
matches.add(match);
}
json.put("mlt", matches);
resp.setContentType("application/json");
resp.getWriter().write(json.toString());
} catch (final Exception e) {
throw new ServletException(e);
} finally {
if (session != null) {
session.logout();
}
}
}
public static void extractTopic(
final JSONWriter writer,
final Post post,
final ResourceResolver resolver,
final String resourceType,
final String childResourceType,
final Writer responseWriter)
throws JSONException, IOException {
final ValueMap vm = post.getProperties();
final JSONArray timestampFields = new JSONArray();
for (final Map.Entry<String, Object> prop : vm.entrySet()) {
final Object value = prop.getValue();
if (value instanceof String[]) {
final JSONArray list = new JSONArray();
for (String v : (String[]) value) {
list.put(v);
}
writer.key(prop.getKey());
writer.value(list);
} else if (value instanceof GregorianCalendar) {
timestampFields.put(prop.getKey());
writer.key(prop.getKey());
writer.value(((Calendar) value).getTimeInMillis());
} else if (prop.getKey().equals("sling:resourceType")) {
writer.key(prop.getKey());
writer.value(resourceType);
} else if (prop.getKey().equals("sentiment")) {
writer.key(prop.getKey());
// 1 = 1, 2 = 3, 3 = 5, 4 = 8, 5 = 10
short shortValue = Short.parseShort(value.toString());
switch (shortValue) {
case 1:
writer.value(1);
break;
case 2:
writer.value(3);
break;
case 3:
writer.value(5);
break;
case 4:
writer.value(8);
break;
case 5:
writer.value(10);
break;
default:
writer.value(value);
}
} else {
writer.key(prop.getKey());
try {
writer.value(URLEncoder.encode(prop.getValue().toString(), "UTF-8"));
} catch (final UnsupportedEncodingException e) {
throw new JSONException(
"Unsupported encoding (UTF-8) for resource at " + post.getPath(), e);
}
}
}
if (timestampFields.length() > 0) {
writer.key(ContentTypeDefinitions.LABEL_TIMESTAMP_FIELDS);
writer.value(timestampFields);
}
final Resource thisResource = resolver.getResource(post.getPath());
final Resource attachments = thisResource.getChild("attachments");
if (attachments != null) {
writer.key(ContentTypeDefinitions.LABEL_ATTACHMENTS);
final JSONWriter attachmentsWriter = writer.array();
for (final Resource attachment : attachments.getChildren()) {
UGCExportHelper.extractAttachment(responseWriter, attachmentsWriter.object(), attachment);
attachmentsWriter.endObject();
}
writer.endArray();
}
final Iterable<Resource> children = thisResource.getChildren();
for (final Resource child : children) {
if (child.isResourceType("social/tally/components/hbs/voting")
|| child.isResourceType("social/tally/components/voting")) {
writer.key(ContentTypeDefinitions.LABEL_TALLY);
final JSONWriter voteObjects = writer.array();
UGCExportHelper.extractTally(voteObjects, child, "Voting");
writer.endArray();
} else if (child.getName().equals("translation")) {
extractTranslation(writer, child);
}
}
final Iterator<Post> posts = post.getPosts();
if (posts.hasNext()) {
writer.key(ContentTypeDefinitions.LABEL_REPLIES);
final JSONWriter replyWriter = writer.object();
while (posts.hasNext()) {
Post childPost = posts.next();
replyWriter.key(childPost.getId());
extractTopic(
replyWriter.object(),
childPost,
resolver,
childResourceType,
childResourceType,
responseWriter);
replyWriter.endObject();
}
writer.endObject();
}
}