org.wallride.web.support.RssFeedView.java Source code

Introduction

Here is the source code for org.wallride.web.support.RssFeedView.java

Source

/*
 * Copyright 2014 Tagbangers, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.wallride.web.support;

import com.rometools.rome.feed.rss.Channel;
import com.rometools.rome.feed.rss.Description;
import com.rometools.rome.feed.rss.Item;
import org.springframework.context.i18n.LocaleContextHolder;
import org.springframework.web.servlet.support.ServletUriComponentsBuilder;
import org.springframework.web.servlet.view.feed.AbstractRssFeedView;
import org.springframework.web.util.UriComponentsBuilder;
import org.wallride.domain.Article;
import org.wallride.domain.Blog;
import org.wallride.service.BlogService;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.time.ZoneId;
import java.util.*;

public class RssFeedView extends AbstractRssFeedView {

    private BlogService blogService;

    public void setBlogService(BlogService blogService) {
        this.blogService = blogService;
    }

    @Override
    protected void buildFeedMetadata(Map<String, Object> model, Channel feed, HttpServletRequest request) {
        Blog blog = blogService.getBlogById(Blog.DEFAULT_ID);
        String language = LocaleContextHolder.getLocale().getLanguage();

        feed.setTitle(blog.getTitle(language));
        feed.setDescription(blog.getTitle(language));

        UriComponentsBuilder builder = ServletUriComponentsBuilder.fromCurrentContextPath();
        feed.setLink(builder.buildAndExpand().toUriString());
    }

    @SuppressWarnings("unchecked")
    @Override
    protected List<Item> buildFeedItems(Map<String, Object> model, HttpServletRequest request,
            HttpServletResponse response) throws Exception {
        Set<Article> articles = (Set<Article>) model.get("articles");
        List<Item> items = new ArrayList<>(articles.size());
        for (Article article : articles) {
            Item item = new Item();
            item.setTitle(article.getTitle());
            item.setPubDate(Date.from(article.getDate().atZone(ZoneId.systemDefault()).toInstant()));
            Description description = new Description();
            description.setType("text/html");
            description.setValue(article.getBody());
            item.setDescription(description);
            item.setLink(link(article));
            items.add(item);
        }
        return items;
    }

    private String link(Article article) {
        UriComponentsBuilder builder = ServletUriComponentsBuilder.fromCurrentContextPath();
        Map<String, Object> params = new HashMap<>();

        Blog blog = blogService.getBlogById(Blog.DEFAULT_ID);
        if (blog.getLanguages().size() > 1) {
            builder.path("/{language}");
            params.put("language", LocaleContextHolder.getLocale().getLanguage());
        }
        builder.path("/{year}/{month}/{day}/{code}");
        params.put("year", String.format("%04d", article.getDate().getYear()));
        params.put("month", String.format("%02d", article.getDate().getMonth().getValue()));
        params.put("day", String.format("%02d", article.getDate().getDayOfMonth()));
        params.put("code", article.getCode());
        return builder.buildAndExpand(params).encode().toUriString();
    }
}