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.Collections; 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 buildFeedItems(Map model, HttpServletRequest request, HttpServletResponse response) throws Exception { List msgs = (List)model.get("messages"); return msgs.stream().map(this::createRssItem).collect(Collectors.toList()); } @Override protected void buildFeedMetadata(Map 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.setLinks(Collections.singletonList(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; } }