James Moger
2012-01-05 cb285cbfddfc0b633d6b8cdb4dc0d2bd2b8b51ef
src/com/gitblit/utils/SyndicationUtils.java
@@ -26,7 +26,9 @@
import com.gitblit.Constants;
import com.gitblit.GitBlitException;
import com.gitblit.models.SyndicatedEntryModel;
import com.gitblit.models.FeedEntryModel;
import com.sun.syndication.feed.synd.SyndCategory;
import com.sun.syndication.feed.synd.SyndCategoryImpl;
import com.sun.syndication.feed.synd.SyndContent;
import com.sun.syndication.feed.synd.SyndContentImpl;
import com.sun.syndication.feed.synd.SyndEntry;
@@ -51,6 +53,7 @@
    * Outputs an RSS feed of the list of entries to the outputstream.
    * 
    * @param hostUrl
    * @param feedLink
    * @param title
    * @param description
    * @param repository
@@ -59,16 +62,15 @@
    * @throws IOException
    * @throws FeedException
    */
   public static void toRSS(String hostUrl, String title, String description, String repository,
         List<SyndicatedEntryModel> entryModels, OutputStream os) throws IOException,
         FeedException {
   public static void toRSS(String hostUrl, String feedLink, String title, String description,
         String repository, List<FeedEntryModel> entryModels, OutputStream os)
         throws IOException, FeedException {
      SyndFeed feed = new SyndFeedImpl();
      feed.setFeedType("rss_2.0");
      feed.setEncoding("UTF-8");
      feed.setTitle(title);
      feed.setLink(MessageFormat.format("{0}/summary/{1}", hostUrl,
            StringUtils.encodeURL(repository)));
      feed.setLink(feedLink);
      feed.setDescription(description);
      SyndImageImpl image = new SyndImageImpl();
      image.setTitle(Constants.NAME);
@@ -77,17 +79,34 @@
      feed.setImage(image);
      List<SyndEntry> entries = new ArrayList<SyndEntry>();
      for (SyndicatedEntryModel entryModel : entryModels) {
      for (FeedEntryModel entryModel : entryModels) {
         SyndEntry entry = new SyndEntryImpl();
         entry.setTitle(entryModel.title);
         entry.setAuthor(entryModel.author);
         entry.setLink(entryModel.link);
         entry.setPublishedDate(entryModel.published);
         if (entryModel.tags != null && entryModel.tags.size() > 0) {
            List<SyndCategory> tags = new ArrayList<SyndCategory>();
            for (String tag : entryModel.tags) {
               SyndCategoryImpl cat = new SyndCategoryImpl();
               cat.setName(tag);
               tags.add(cat);
            }
            entry.setCategories(tags);
         }
         SyndContent content = new SyndContentImpl();
         content.setType(entryModel.contentType);
         content.setValue(entryModel.content);
         if (StringUtils.isEmpty(entryModel.contentType)
               || entryModel.contentType.equalsIgnoreCase("text/plain")) {
            content.setType("text/html");
            content.setValue(StringUtils.breakLinesForHtml(entryModel.content));
         } else {
            content.setType(entryModel.contentType);
            content.setValue(entryModel.content);
         }
         entry.setDescription(content);
         entries.add(entry);
      }
      feed.setEntries(entries);
@@ -110,34 +129,103 @@
    * @param numberOfEntries
    *            the number of entries to retrieve. if <= 0 the server default
    *            is used.
    * @param page
    *            0-indexed. used to paginate the results.
    * @param username
    * @param password
    * @return a list of SyndicationModel entries
    * @throws {@link IOException}
    */
   public static List<SyndicatedEntryModel> readFeed(String url, String repository, String branch,
         int numberOfEntries, String username, char[] password) throws IOException {
      String feedUrl;
      if (StringUtils.isEmpty(branch)) {
         // no branch specified
         if (numberOfEntries > 0) {
            // fixed number of entries
            feedUrl = MessageFormat.format("{0}/feed/{1}?l={2,number,0}", url, repository);
         } else {
            // server default number of entries
            feedUrl = MessageFormat.format("{0}/feed/{1}", url, repository);
         }
      } else {
         // branch specified
         if (numberOfEntries > 0) {
            // fixed number of entries
            feedUrl = MessageFormat.format("{0}/feed/{1}?h={2}&l={3,number,0}", url,
                  repository, branch, numberOfEntries);
         } else {
            // server default number of entries
            feedUrl = MessageFormat.format("{0}/feed/{1}?h={2}", url, repository, branch);
   public static List<FeedEntryModel> readFeed(String url, String repository, String branch,
         int numberOfEntries, int page, String username, char[] password) throws IOException {
      // build feed url
      List<String> parameters = new ArrayList<String>();
      if (numberOfEntries > 0) {
         parameters.add("l=" + numberOfEntries);
      }
      if (page > 0) {
         parameters.add("pg=" + page);
      }
      if (!StringUtils.isEmpty(branch)) {
         parameters.add("h=" + branch);
      }
      return readFeed(url, parameters, repository, branch, username, password);
   }
   /**
    * Reads a Gitblit RSS search feed.
    *
    * @param url
    *            the url of the Gitblit server
    * @param repository
    *            the repository name
    * @param fragment
    *            the search fragment
    * @param searchType
    *            the search type (optional, defaults to COMMIT)
    * @param numberOfEntries
    *            the number of entries to retrieve. if <= 0 the server default
    *            is used.
    * @param page
    *            0-indexed. used to paginate the results.
    * @param username
    * @param password
    * @return a list of SyndicationModel entries
    * @throws {@link IOException}
    */
   public static List<FeedEntryModel> readSearchFeed(String url, String repository, String branch,
         String fragment, Constants.SearchType searchType, int numberOfEntries, int page,
         String username, char[] password) throws IOException {
      // determine parameters
      List<String> parameters = new ArrayList<String>();
      parameters.add("s=" + StringUtils.encodeURL(fragment));
      if (numberOfEntries > 0) {
         parameters.add("l=" + numberOfEntries);
      }
      if (page > 0) {
         parameters.add("pg=" + page);
      }
      if (!StringUtils.isEmpty(branch)) {
         parameters.add("h=" + branch);
      }
      if (searchType != null) {
         parameters.add("st=" + searchType.name());
      }
      return readFeed(url, parameters, repository, branch, username, password);
   }
   /**
    * Reads a Gitblit RSS feed.
    *
    * @param url
    *            the url of the Gitblit server
    * @param parameters
    *            the list of RSS parameters
    * @param repository
    *            the repository name
    * @param username
    * @param password
    * @return a list of SyndicationModel entries
    * @throws {@link IOException}
    */
   private static List<FeedEntryModel> readFeed(String url, List<String> parameters,
         String repository, String branch, String username, char[] password) throws IOException {
      // build url
      StringBuilder sb = new StringBuilder();
      sb.append(MessageFormat.format("{0}" + Constants.SYNDICATION_PATH + "{1}", url, repository));
      if (parameters.size() > 0) {
         boolean first = true;
         for (String parameter : parameters) {
            if (first) {
               sb.append('?');
               first = false;
            } else {
               sb.append('&');
            }
            sb.append(parameter);
         }
      }
      String feedUrl = sb.toString();
      URLConnection conn = ConnectionUtils.openReadConnection(feedUrl, username, password);
      InputStream is = conn.getInputStream();
      SyndFeedInput input = new SyndFeedInput();
@@ -148,10 +236,10 @@
         throw new GitBlitException(f);
      }
      is.close();
      List<SyndicatedEntryModel> entries = new ArrayList<SyndicatedEntryModel>();
      List<FeedEntryModel> entries = new ArrayList<FeedEntryModel>();
      for (Object o : feed.getEntries()) {
         SyndEntryImpl entry = (SyndEntryImpl) o;
         SyndicatedEntryModel model = new SyndicatedEntryModel();
         FeedEntryModel model = new FeedEntryModel();
         model.repository = repository;
         model.branch = branch;
         model.title = entry.getTitle();
@@ -160,6 +248,14 @@
         model.link = entry.getLink();
         model.content = entry.getDescription().getValue();
         model.contentType = entry.getDescription().getType();
         if (entry.getCategories() != null && entry.getCategories().size() > 0) {
            List<String> tags = new ArrayList<String>();
            for (Object p : entry.getCategories()) {
               SyndCategory cat = (SyndCategory) p;
               tags.add(cat.getName());
            }
            model.tags = tags;
         }
         entries.add(model);
      }
      return entries;