James Moger
2011-11-03 9bdb91ea80119121791271819f2c1fbf07bf5591
src/com/gitblit/SyndicationServlet.java
@@ -16,20 +16,26 @@
package com.gitblit;
import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import javax.servlet.http.HttpServlet;
import org.eclipse.jgit.lib.ObjectId;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.revwalk.RevCommit;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.gitblit.models.RefModel;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.SyndicatedEntryModel;
import com.gitblit.utils.HttpUtils;
import com.gitblit.utils.JGitUtils;
import com.gitblit.utils.JGitUtils.SearchType;
import com.gitblit.utils.StringUtils;
import com.gitblit.utils.SyndicationUtils;
import com.gitblit.wicket.WicketUtils;
/**
 * SyndicationServlet generates RSS 2.0 feeds and feed links.
@@ -116,14 +122,22 @@
         javax.servlet.http.HttpServletResponse response) throws javax.servlet.ServletException,
         java.io.IOException {
      String hostURL = WicketUtils.getHostURL(request);
      String url = request.getRequestURI().substring(request.getServletPath().length());
      String servletUrl = request.getContextPath() + request.getServletPath();
      String url = request.getRequestURI().substring(servletUrl.length());
      if (url.charAt(0) == '/' && url.length() > 1) {
         url = url.substring(1);
      }
      String repositoryName = url;
      String objectId = request.getParameter("h");
      String l = request.getParameter("l");
      String searchString = request.getParameter("s");
      SearchType searchType = SearchType.COMMIT;
      if (!StringUtils.isEmpty(request.getParameter("st"))) {
         SearchType type = SearchType.forName(request.getParameter("st"));
         if (type != null) {
            searchType = type;
         }
      }
      int length = GitBlit.getInteger(Keys.web.syndicationEntries, 25);
      if (StringUtils.isEmpty(objectId)) {
         objectId = org.eclipse.jgit.lib.Constants.HEAD;
@@ -135,12 +149,67 @@
         }
      }
      response.setContentType("application/rss+xml; charset=UTF-8");
      Repository repository = GitBlit.self().getRepository(repositoryName);
      RepositoryModel model = GitBlit.self().getRepositoryModel(repositoryName);
      List<RevCommit> commits = JGitUtils.getRevLog(repository, objectId, 0, length);
      List<RevCommit> commits;
      if (StringUtils.isEmpty(searchString)) {
         // standard log/history lookup
         commits = JGitUtils.getRevLog(repository, objectId, 0, length);
      } else {
         // repository search
         commits = JGitUtils.searchRevlogs(repository, objectId, searchString, searchType, 0,
               length);
      }
      Map<ObjectId, List<RefModel>> allRefs = JGitUtils.getAllRefs(repository);
      List<SyndicatedEntryModel> entries = new ArrayList<SyndicatedEntryModel>();
      boolean mountParameters = GitBlit.getBoolean(Keys.web.mountParameters, true);
      String urlPattern;
      if (mountParameters) {
         // mounted parameters
         urlPattern = "{0}/commit/{1}/{2}";
      } else {
         // parameterized parameters
         urlPattern = "{0}/commit/?r={1}&h={2}";
      }
      String gitblitUrl = HttpUtils.getGitblitURL(request);
      // convert RevCommit to SyndicatedEntryModel
      for (RevCommit commit : commits) {
         SyndicatedEntryModel entry = new SyndicatedEntryModel();
         entry.title = commit.getShortMessage();
         entry.author = commit.getAuthorIdent().getName();
         entry.link = MessageFormat.format(urlPattern, gitblitUrl,
               StringUtils.encodeURL(model.name), commit.getName());
         entry.published = commit.getCommitterIdent().getWhen();
         entry.contentType = "text/plain";
         entry.content = commit.getFullMessage();
         entry.repository = model.name;
         entry.branch = objectId;
         List<RefModel> refs = allRefs.get(commit.getId());
         if (refs != null && refs.size() > 0) {
            List<String> tags = new ArrayList<String>();
            for (RefModel ref : refs) {
               tags.add(ref.getName());
            }
            entry.tags = tags;
         }
         entries.add(entry);
      }
      String feedLink;
      if (mountParameters) {
         // mounted url
         feedLink = MessageFormat.format("{0}/summary/{1}", gitblitUrl,
               StringUtils.encodeURL(model.name));
      } else {
         // parameterized url
         feedLink = MessageFormat.format("{0}/summary/?r={1}", gitblitUrl,
               StringUtils.encodeURL(model.name));
      }
      try {
         SyndicationUtils.toRSS(hostURL, getTitle(model.name, objectId), model.description,
               model.name, commits, response.getOutputStream());
         SyndicationUtils.toRSS(gitblitUrl, feedLink, getTitle(model.name, objectId),
               model.description, model.name, entries, response.getOutputStream());
      } catch (Exception e) {
         logger.error("An error occurred during feed generation", e);
      }