blob: 3b6b48a6edebc7e297cba472e873d70fa40a5007 (
plain) (
tree)
|
|
package com.juick.rss;
import com.juick.Message;
import com.juick.User;
import com.juick.rss.extension.JuickModule;
import com.juick.rss.extension.JuickModuleImpl;
import com.juick.util.MessageUtils;
import com.rometools.modules.atom.modules.AtomLinkModule;
import com.rometools.modules.atom.modules.AtomLinkModuleImpl;
import com.rometools.modules.mediarss.MediaEntryModuleImpl;
import com.rometools.modules.mediarss.MediaModule;
import com.rometools.modules.mediarss.MediaModuleImpl;
import com.rometools.modules.mediarss.types.MediaContent;
import com.rometools.modules.mediarss.types.Metadata;
import com.rometools.modules.mediarss.types.Thumbnail;
import com.rometools.modules.mediarss.types.UrlReference;
import com.rometools.rome.feed.atom.Link;
import com.rometools.rome.feed.rss.*;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.servlet.view.feed.AbstractRssFeedView;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
/**
* Created by vitalyster on 13.12.2016.
*/
public class MessagesView extends AbstractRssFeedView {
private static final Logger logger = LoggerFactory.getLogger(MessagesView.class);
@Override
protected List<Item> buildFeedItems(Map<String, Object> model, HttpServletRequest request, HttpServletResponse response) throws Exception {
List<Message> msgs = (List<Message>)model.get("messages");
return msgs.stream().map(this::createRssItem).collect(Collectors.toList());
}
@Override
protected void buildFeedMetadata(Map<String, Object> model, Channel feed, HttpServletRequest request) {
Object userObj = model.get("user");
if (userObj != null) {
User user = (User) userObj;
feed.setDescription(String.format("The latest messages by @%s at Juick", user.getName()));
String title = String.format("%s - Juick", user.getName());
feed.setTitle(title);
String link = String.format("http://juick.com/%s/", user.getName());
feed.setLink(link);
Image rssImage = new Image();
rssImage.setUrl(String.format("http://juick.com/a/%d.png", user.getUid()));
rssImage.setTitle(title);
rssImage.setLink(link);
feed.setImage(rssImage);
String href = String.format("http://rss.juick.com/%s/blog", user.getName());
AtomLinkModule atomLinkModule = new AtomLinkModuleImpl();
Link atomLink = new Link();
atomLink.setHref(href);
atomLink.setType("application/rss+xml");
atomLink.setRel("self");
atomLinkModule.setLink(atomLink);
feed.getModules().add(atomLinkModule);
} else {
feed.setDescription("The latest messages at Juick");
feed.setLink("http://juick.com/");
feed.setTitle("Juick");
}
MediaModule mediaModule = new MediaModuleImpl();
feed.getModules().add(mediaModule);
}
private Item createRssItem(Message msg) {
Item item = new Item();
String messageUrl = String.format("http://juick.com/%s/%d", msg.getUser().getName(), msg.getMid());
String messageTitle = String.format("@%s: %s", msg.getUser().getName(), msg.getTagsString());
boolean isCode = msg.getTags().stream().anyMatch(t -> t.getName().equals("code"));
String messageDescription = isCode ? MessageUtils.formatMessageCode(msg.getText())
: MessageUtils.formatMessage(msg.getText());
item.setLink(messageUrl);
//item.setGuid(messageUrl);
item.setTitle(messageTitle);
Description description = new Description();
description.setType("text/html");
description.setValue(messageDescription);
item.setDescription(description);
item.setPubDate(msg.getDate());
item.setComments(messageUrl);
msg.getTags().stream().map(t -> {
Category category = new Category();
category.setValue(t.getName());
return category;
}).forEach(c -> item.getCategories().add(c));
JuickModule juickModule = new JuickModuleImpl();
juickModule.setUid(msg.getUser().getUid());
item.getModules().add(juickModule);
if (StringUtils.isNotEmpty(msg.getAttachmentType())) {
String type = msg.getAttachmentType().equals("jpg") ? "image/jpeg" : "image/png";
MediaEntryModuleImpl module = new MediaEntryModuleImpl();
try {
UrlReference reference = new UrlReference(msg.getAttachmentURL());
MediaContent mediaContent = new MediaContent(reference);
mediaContent.setType(type);
Metadata metadata = new Metadata();
metadata.setThumbnail(new Thumbnail[]{new Thumbnail(new URI(msg.getPhoto().getThumbnail()))});
module.setMetadata(metadata);
module.setMediaContents(new MediaContent[]{mediaContent});
item.getModules().add(module);
} catch (URISyntaxException e) {
logger.error("Invalid URI", e);
}
}
return item;
}
}
|