Initial favorites export commit

1 files changed, 131 insertions, 0 deletions

diff --git a/core/src/main/java/de/danoeh/antennapod/core/export/favorites/FavoritesWriter.java b/core/src/main/java/de/danoeh/antennapod/core/export/favorites/FavoritesWriter.java
new file mode 100644
index 000000000..a6686a8a4
--- /dev/null
+++ b/core/src/main/java/de/danoeh/antennapod/core/export/favorites/FavoritesWriter.java
@@ -0,0 +1,131 @@
+package de.danoeh.antennapod.core.export.favorites;
+
+import android.content.Context;
+import android.util.Log;
+
+import org.apache.commons.io.IOUtils;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.Writer;
+import java.text.SimpleDateFormat;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+import java.util.TreeMap;
+
+import de.danoeh.antennapod.core.export.ExportWriter;
+import de.danoeh.antennapod.core.feed.Feed;
+import de.danoeh.antennapod.core.feed.FeedItem;
+import de.danoeh.antennapod.core.storage.DBReader;
+
+/** Writes saved favorites to file */
+public class FavoritesWriter implements ExportWriter {
+    private static final String TAG = "FavoritesWriter";
+    private static final SimpleDateFormat DATE_FORMAT = new SimpleDateFormat("yyyy-MM-dd");
+
+    private static final int PAGE_LIMIT = 100;
+
+    @Override
+    public void writeDocument(List<Feed> feeds, Writer writer, Context context) throws IllegalArgumentException, IllegalStateException, IOException {
+        Log.d(TAG, "Starting to write document");
+
+        InputStream templateStream = context.getAssets().open("favorites-export-template.html");
+        String template = IOUtils.toString(templateStream, "UTF-8");
+        String[] templateParts = template.split("\\{FAVORITES\\}");
+
+        Map<Long, List<FeedItem>> favoriteByFeed = getFeedMap(getFavorites(feeds));
+
+        writer.append(templateParts[0]);
+        writer.append("<ul>");
+
+        for (Long feedId : favoriteByFeed.keySet()) {
+            List<FeedItem> favorites = favoriteByFeed.get(feedId);
+
+            writer.append("<li><div>");
+            writeFeed(writer, favorites.get(0).getFeed());
+
+            writer.append("<ul>");
+            for (FeedItem item : favorites) {
+                writeFavoriteItem(writer, item);
+            }
+            writer.append("</ul></div></li>\n");
+        }
+
+        writer.append("</ul>");
+
+        writer.append(templateParts[1]);
+
+        Log.d(TAG, "Finished writing document");
+    }
+
+    private List<FeedItem> getFavorites(List<Feed> feeds) {
+        int page = 0;
+
+        List<FeedItem> favoritesPage = DBReader.getFavoriteItemsList(page, PAGE_LIMIT);
+        List<FeedItem> favoritesList = new ArrayList<>();
+
+        while (!favoritesPage.isEmpty()) {
+            favoritesList.addAll(favoritesPage);
+
+            // save a DB call if there are no more items to fetch
+            if (favoritesPage.size() < PAGE_LIMIT) {
+                break;
+            }
+
+            ++page;
+
+            favoritesPage = DBReader.getFavoriteItemsList(page * PAGE_LIMIT, PAGE_LIMIT);
+        }
+
+        Collections.sort(favoritesList, (lhs, rhs) -> lhs.getPubDate().compareTo(rhs.getPubDate()));
+
+        return favoritesList;
+    }
+
+    private Map<Long, List<FeedItem>> getFeedMap(List<FeedItem> favoritesList) {
+        Map<Long, List<FeedItem>> feedMap = new TreeMap<>();
+
+        for (FeedItem item : favoritesList) {
+            List<FeedItem> feedEpisodes = feedMap.get(item.getFeedId());
+
+            if (feedEpisodes == null) {
+                feedEpisodes = new ArrayList<>();
+                feedMap.put(item.getFeedId(), feedEpisodes);
+            }
+
+            feedEpisodes.add(item);
+        }
+
+        return feedMap;
+    }
+
+    private void writeFeed(Writer writer, Feed feed) throws IOException {
+        writer.append("<img src=\"");
+        writer.append(feed.getImageUrl());
+        writer.append("\" /><p>");
+        writer.append(feed.getTitle());
+        writer.append(" <span><a href=\"");
+        writer.append(feed.getLink());
+        writer.append("\">Website</a> • <a href=\"");
+        writer.append(feed.getDownload_url());
+        writer.append("\">Feed</a></span></p>");
+    }
+
+    private void writeFavoriteItem(Writer writer, FeedItem item) throws IOException {
+        writer.append("<li><span>[");
+        writer.append(DATE_FORMAT.format(item.getPubDate()));
+        writer.append("] <a href=\"");
+        writer.append(item.getLink());
+        writer.append("\">");
+        writer.append(item.getTitle().trim());
+        writer.append("</a>");
+        writer.append("</span></li>\n");
+    }
+
+    @Override
+    public String fileExtension() {
+        return "html";
+    }
+}