Use new URL for pirate bay

[rhynodge.git] / src / main / java / net / pterodactylus / rhynodge / triggers / NewTorrentTrigger.java
diff --git a/src/main/java/net/pterodactylus/rhynodge/triggers/NewTorrentTrigger.java b/src/main/java/net/pterodactylus/rhynodge/triggers/NewTorrentTrigger.java

index 872349e..7b28fee 100644 (file)
--- a/src/main/java/net/pterodactylus/rhynodge/triggers/NewTorrentTrigger.java
+++ b/src/main/java/net/pterodactylus/rhynodge/triggers/NewTorrentTrigger.java
@@ -33,6 +33,7 @@ import net.pterodactylus.rhynodge.states.TorrentState.TorrentFile;
  import org.apache.commons.lang3.StringEscapeUtils;
  
  import com.google.common.collect.Lists;
+import com.google.common.collect.Ordering;
  import com.google.common.collect.Sets;
  
  /**
@@ -54,7 +55,7 @@ public class NewTorrentTrigger implements Trigger {
         //
  
         /**
-        * {@inheritDocs}
+        * {@inheritDoc}
          */
         @Override
         public State mergeStates(State previousState, State currentState) {
@@ -95,7 +96,7 @@ public class NewTorrentTrigger implements Trigger {
         }
  
         //
-       // STATIC METHODS
+       // PRIVATE METHODS
         //
  
         /**
@@ -131,24 +132,63 @@ public class NewTorrentTrigger implements Trigger {
          * @return The generated HTML
          */
         private String getHtmlTextList(Reaction reaction) {
-               StringBuilder htmlText = new StringBuilder();
-               htmlText.append("<html><body>\n");
-               htmlText.append("<h1>New Torrents</h1>\n");
-               htmlText.append("<ul>\n");
-               for (TorrentFile torrentFile : torrentFiles) {
-                       htmlText.append("<li><strong>").append(StringEscapeUtils.escapeHtml4(torrentFile.name())).append("</strong></li>");
-                       htmlText.append("<div>Size: <strong>").append(StringEscapeUtils.escapeHtml4(torrentFile.size())).append("</strong> in <strong>").append(torrentFile.fileCount()).append("</strong> file(s)</div>");
-                       htmlText.append("<div><strong>").append(torrentFile.seedCount()).append("</strong> seed(s), <strong>").append(torrentFile.leechCount()).append("</strong> leecher(s)</div>");
-                       if ((torrentFile.magnetUri() != null) && (torrentFile.magnetUri().length() > 0)) {
-                               htmlText.append(String.format("<div><a href=\"%s\">Magnet URI</a></div>", StringEscapeUtils.escapeHtml4(torrentFile.magnetUri())));
-                       }
-                       if ((torrentFile.downloadUri() != null) && (torrentFile.downloadUri().length() > 0)) {
-                               htmlText.append(String.format("<div><a href=\"%s\">Download URI</a></div>", StringEscapeUtils.escapeHtml4(torrentFile.downloadUri())));
+               StringBuilder htmlBuilder = new StringBuilder();
+               htmlBuilder.append("<html><body>\n");
+               htmlBuilder.append("<table>\n<caption>All Known Torrents</caption>\n");
+               htmlBuilder.append("<thead>\n");
+               htmlBuilder.append("<tr>");
+               htmlBuilder.append("<th>Filename</th>");
+               htmlBuilder.append("<th>Size</th>");
+               htmlBuilder.append("<th>File(s)</th>");
+               htmlBuilder.append("<th>Seeds</th>");
+               htmlBuilder.append("<th>Leechers</th>");
+               htmlBuilder.append("<th>Magnet</th>");
+               htmlBuilder.append("<th>Download</th>");
+               htmlBuilder.append("</tr>\n");
+               htmlBuilder.append("</thead>\n");
+               htmlBuilder.append("<tbody>\n");
+               for (TorrentFile torrentFile : sortNewFirst().sortedCopy(allTorrentFiles)) {
+                       if (newTorrentFiles.contains(torrentFile)) {
+                               htmlBuilder.append("<tr style=\"color: #008000; font-weight: bold;\">");
+                       } else {
+                               htmlBuilder.append("<tr>");
                         }
+                       htmlBuilder.append("<td>").append(StringEscapeUtils.escapeHtml4(torrentFile.name())).append("</td>");
+                       htmlBuilder.append("<td>").append(StringEscapeUtils.escapeHtml4(torrentFile.size())).append("</td>");
+                       htmlBuilder.append("<td>").append(torrentFile.fileCount()).append("</td>");
+                       htmlBuilder.append("<td>").append(torrentFile.seedCount()).append("</td>");
+                       htmlBuilder.append("<td>").append(torrentFile.leechCount()).append("</td>");
+                       htmlBuilder.append("<td><a href=\"").append(StringEscapeUtils.escapeHtml4(torrentFile.magnetUri())).append("\">Link</a></td>");
+                       htmlBuilder.append("<td><a href=\"").append(StringEscapeUtils.escapeHtml4(torrentFile.downloadUri())).append("\">Link</a></td>");
+                       htmlBuilder.append("</tr>\n");
                 }
-               htmlText.append("</ul>\n");
-               htmlText.append("</body></html>\n");
-               return htmlText.toString();
+               htmlBuilder.append("</tbody>\n");
+               htmlBuilder.append("</table>\n");
+               htmlBuilder.append("</body></html>\n");
+               return htmlBuilder.toString();
+       }
+
+       /**
+        * Returns an ordering that sorts torrent files by whether they are new
+        * (according to {@link #newTorrentFiles}) or not. New files will be sorted
+        * first.
+        *
+        * @return An ordering for “new files first”
+        */
+       private Ordering<TorrentFile> sortNewFirst() {
+               return new Ordering<TorrentFile>() {
+
+                       @Override
+                       public int compare(TorrentFile leftTorrentFile, TorrentFile rightTorrentFile) {
+                               if (newTorrentFiles.contains(leftTorrentFile) && !newTorrentFiles.contains(rightTorrentFile)) {
+                                       return -1;
+                               }
+                               if (!newTorrentFiles.contains(leftTorrentFile) && newTorrentFiles.contains(rightTorrentFile)) {
+                                       return 1;
+                               }
+                               return 0;
+                       }
+               };
         }
  
  }