package net.pterodactylus.rhynodge.triggers;
-import static com.google.common.base.Preconditions.checkState;
-
+import java.util.ArrayList;
import java.util.Collection;
+import java.util.HashSet;
import java.util.Map;
-import java.util.Map.Entry;
-import net.pterodactylus.rhynodge.Reaction;
import net.pterodactylus.rhynodge.State;
import net.pterodactylus.rhynodge.Trigger;
-import net.pterodactylus.rhynodge.output.DefaultOutput;
-import net.pterodactylus.rhynodge.output.Output;
import net.pterodactylus.rhynodge.states.EpisodeState;
import net.pterodactylus.rhynodge.states.EpisodeState.Episode;
import net.pterodactylus.rhynodge.states.TorrentState.TorrentFile;
-import org.apache.commons.lang3.StringEscapeUtils;
-
-import com.google.common.base.Function;
-import com.google.common.collect.FluentIterable;
-import com.google.common.collect.ImmutableMap;
-import com.google.common.collect.Ordering;
-import com.google.common.collect.Sets;
+import static com.google.common.base.Preconditions.checkState;
+import static java.util.function.Function.identity;
+import static java.util.stream.Collectors.toMap;
/**
* {@link Trigger} implementation that compares two {@link EpisodeState}s for
*/
public class NewEpisodeTrigger implements Trigger {
- /** All new episodes. */
- private final Collection<Episode> newEpisodes = Sets.newHashSet();
-
- /** All changed episodes. */
- private final Collection<Episode> changedEpisodes = Sets.newHashSet();
-
- /** All episodes. */
- private final Collection<Episode> allEpisodes = Sets.newHashSet();
-
- //
- // TRIGGER METHODS
- //
+ private boolean triggered = false;
/**
- * {@inheritDocs}
+ * {@inheritDoc}
*/
@Override
public State mergeStates(State previousState, State currentState) {
checkState(currentState instanceof EpisodeState, "currentState is not a EpisodeState but a %s", currentState.getClass().getName());
checkState(previousState instanceof EpisodeState, "previousState is not a EpisodeState but a %s", currentState.getClass().getName());
- newEpisodes.clear();
- changedEpisodes.clear();
- this.allEpisodes.clear();
- Map<Episode, Episode> allEpisodes = FluentIterable.from(((EpisodeState) previousState).episodes()).toMap(new Function<Episode, Episode>() {
- @Override
- public Episode apply(Episode episode) {
- return episode;
- }
- });
+ Collection<Episode> newEpisodes = new HashSet<>();
+ Collection<Episode> changedEpisodes = new HashSet<>();
+ Collection<TorrentFile> newTorrentFiles = new HashSet<>();
+ Map<Episode, Episode> allEpisodes = ((EpisodeState) previousState).episodes().stream().collect(toMap(identity(), identity()));
for (Episode episode : ((EpisodeState) currentState).episodes()) {
if (!allEpisodes.containsKey(episode)) {
allEpisodes.put(episode, episode);
newEpisodes.add(episode);
+ triggered = true;
}
- for (TorrentFile torrentFile : episode.torrentFiles()) {
- int oldSize = allEpisodes.get(episode).torrentFiles().size();
- allEpisodes.get(episode).addTorrentFile(torrentFile);
- int newSize = allEpisodes.get(episode).torrentFiles().size();
- if (!newEpisodes.contains(episode) && (oldSize != newSize)) {
- changedEpisodes.add(episode);
+ Episode existingEpisode = allEpisodes.get(episode);
+ for (TorrentFile torrentFile : new ArrayList<>(episode.torrentFiles())) {
+ int oldSize = existingEpisode.torrentFiles().size();
+ existingEpisode.addTorrentFile(torrentFile);
+ int newSize = existingEpisode.torrentFiles().size();
+ if (oldSize != newSize) {
+ newTorrentFiles.add(torrentFile);
+ }
+ if (!newEpisodes.contains(existingEpisode) && (oldSize != newSize)) {
+ changedEpisodes.add(existingEpisode);
}
}
}
- this.allEpisodes.addAll(allEpisodes.values());
- return new EpisodeState(this.allEpisodes);
+ return new EpisodeState(allEpisodes.values(), newEpisodes, changedEpisodes, newTorrentFiles);
}
/**
*/
@Override
public boolean triggers() {
- return !newEpisodes.isEmpty() || !changedEpisodes.isEmpty();
- }
-
- /**
- * {@inheritDoc}
- */
- @Override
- public Output output(Reaction reaction) {
- String summary;
- if (!newEpisodes.isEmpty()) {
- if (!changedEpisodes.isEmpty()) {
- summary = String.format("%d new and %d changed Torrent(s) for “%s!”", newEpisodes.size(), changedEpisodes.size(), reaction.name());
- } else {
- summary = String.format("%d new Torrent(s) for “%s!”", newEpisodes.size(), reaction.name());
- }
- } else {
- summary = String.format("%d changed Torrent(s) for “%s!”", changedEpisodes.size(), reaction.name());
- }
- DefaultOutput output = new DefaultOutput(summary);
- output.addText("text/plain", generatePlainText(reaction, newEpisodes, changedEpisodes, allEpisodes));
- output.addText("text/html", generateHtmlText(reaction, newEpisodes, changedEpisodes, allEpisodes));
- return output;
- }
-
- //
- // STATIC METHODS
- //
-
- /**
- * Generates the plain text trigger output.
- *
- * @param reaction
- * The reaction that was triggered
- * @param newEpisodes
- * The new episodes
- * @param changedEpisodes
- * The changed episodes
- * @param allEpisodes
- * All episodes
- * @return The plain text output
- */
- private static String generatePlainText(Reaction reaction, Collection<Episode> newEpisodes, Collection<Episode> changedEpisodes, Collection<Episode> allEpisodes) {
- StringBuilder stringBuilder = new StringBuilder();
- if (!newEpisodes.isEmpty()) {
- stringBuilder.append(reaction.name()).append(" - New Episodes\n\n");
- for (Episode episode : newEpisodes) {
- stringBuilder.append("- ").append(episode.identifier()).append("\n");
- for (TorrentFile torrentFile : episode) {
- stringBuilder.append(" - ").append(torrentFile.name()).append(", ").append(torrentFile.size()).append("\n");
- if ((torrentFile.magnetUri() != null) && (torrentFile.magnetUri().length() > 0)) {
- stringBuilder.append(" Magnet: ").append(torrentFile.magnetUri()).append("\n");
- }
- if ((torrentFile.downloadUri() != null) && (torrentFile.downloadUri().length() > 0)) {
- stringBuilder.append(" Download: ").append(torrentFile.downloadUri()).append("\n");
- }
- }
- }
- }
- if (!changedEpisodes.isEmpty()) {
- stringBuilder.append(reaction.name()).append(" - Changed Episodes\n\n");
- for (Episode episode : changedEpisodes) {
- stringBuilder.append("- ").append(episode.identifier()).append("\n");
- for (TorrentFile torrentFile : episode) {
- stringBuilder.append(" - ").append(torrentFile.name()).append(", ").append(torrentFile.size()).append("\n");
- if ((torrentFile.magnetUri() != null) && (torrentFile.magnetUri().length() > 0)) {
- stringBuilder.append(" Magnet: ").append(torrentFile.magnetUri()).append("\n");
- }
- if ((torrentFile.downloadUri() != null) && (torrentFile.downloadUri().length() > 0)) {
- stringBuilder.append(" Download: ").append(torrentFile.downloadUri()).append("\n");
- }
- }
- }
- }
- /* list all known episodes. */
- stringBuilder.append(reaction.name()).append(" - All Known Episodes\n\n");
- ImmutableMap<Integer, Collection<Episode>> episodesBySeason = FluentIterable.from(allEpisodes).index(new Function<Episode, Integer>() {
-
- @Override
- public Integer apply(Episode episode) {
- return episode.season();
- }
- }).asMap();
- for (Entry<Integer, Collection<Episode>> seasonEntry : episodesBySeason.entrySet()) {
- stringBuilder.append(" Season ").append(seasonEntry.getKey()).append("\n\n");
- for (Episode episode : Ordering.natural().sortedCopy(seasonEntry.getValue())) {
- stringBuilder.append(" Episode ").append(episode.episode()).append("\n");
- for (TorrentFile torrentFile : episode) {
- stringBuilder.append(" Size: ").append(torrentFile.size());
- stringBuilder.append(" in ").append(torrentFile.fileCount()).append(" file(s): ");
- stringBuilder.append(torrentFile.magnetUri());
- }
- }
- }
-
- return stringBuilder.toString();
- }
-
- /**
- * Generates the HTML trigger output.
- *
- * @param reaction
- * The reaction that was triggered
- * @param newEpisodes
- * The new episodes
- * @param changedEpisodes
- * The changed episodes
- * @param allEpisodes
- * All episodes
- * @return The HTML output
- */
- private static String generateHtmlText(Reaction reaction, Collection<Episode> newEpisodes, Collection<Episode> changedEpisodes, Collection<Episode> allEpisodes) {
- StringBuilder htmlBuilder = new StringBuilder();
- htmlBuilder.append("<html><body>\n");
- htmlBuilder.append("<h1>").append(StringEscapeUtils.escapeHtml4(reaction.name())).append("</h1>\n");
- if (!newEpisodes.isEmpty()) {
- htmlBuilder.append("<h2>New Episodes</h2>\n");
- htmlBuilder.append("<ul>\n");
- for (Episode episode : newEpisodes) {
- htmlBuilder.append("<li>Season ").append(episode.season()).append(", Episode ").append(episode.episode()).append("</li>\n");
- htmlBuilder.append("<ul>\n");
- for (TorrentFile torrentFile : episode) {
- htmlBuilder.append("<li>").append(StringEscapeUtils.escapeHtml4(torrentFile.name())).append("</li>\n");
- htmlBuilder.append("<div>");
- htmlBuilder.append("<strong>").append(StringEscapeUtils.escapeHtml4(torrentFile.size())).append("</strong>, ");
- htmlBuilder.append("<strong>").append(torrentFile.fileCount()).append("</strong> file(s), ");
- htmlBuilder.append("<strong>").append(torrentFile.seedCount()).append("</strong> seed(s), ");
- htmlBuilder.append("<strong>").append(torrentFile.leechCount()).append("</strong> leecher(s)</div>\n");
- htmlBuilder.append("<div>");
- if ((torrentFile.magnetUri() != null) && (torrentFile.magnetUri().length() > 0)) {
- htmlBuilder.append("<a href=\"").append(StringEscapeUtils.escapeHtml4(torrentFile.magnetUri())).append("\">Magnet</a> ");
- }
- if ((torrentFile.downloadUri() != null) && (torrentFile.downloadUri().length() > 0)) {
- htmlBuilder.append("<a href=\"").append(StringEscapeUtils.escapeHtml4(torrentFile.downloadUri())).append("\">Download</a>");
- }
- htmlBuilder.append("</div>\n");
- }
- htmlBuilder.append("</ul>\n");
- }
- htmlBuilder.append("</ul>\n");
- }
- if (!changedEpisodes.isEmpty()) {
- htmlBuilder.append("<h2>Changed Episodes</h2>\n");
- htmlBuilder.append("<ul>\n");
- for (Episode episode : changedEpisodes) {
- htmlBuilder.append("<li>Season ").append(episode.season()).append(", Episode ").append(episode.episode()).append("</li>\n");
- htmlBuilder.append("<ul>\n");
- for (TorrentFile torrentFile : episode) {
- htmlBuilder.append("<li>").append(StringEscapeUtils.escapeHtml4(torrentFile.name())).append("</li>\n");
- htmlBuilder.append("<div>");
- htmlBuilder.append("<strong>").append(StringEscapeUtils.escapeHtml4(torrentFile.size())).append("</strong>, ");
- htmlBuilder.append("<strong>").append(torrentFile.fileCount()).append("</strong> file(s), ");
- htmlBuilder.append("<strong>").append(torrentFile.seedCount()).append("</strong> seed(s), ");
- htmlBuilder.append("<strong>").append(torrentFile.leechCount()).append("</strong> leecher(s)</div>\n");
- htmlBuilder.append("<div>");
- if ((torrentFile.magnetUri() != null) && (torrentFile.magnetUri().length() > 0)) {
- htmlBuilder.append("<a href=\"").append(StringEscapeUtils.escapeHtml4(torrentFile.magnetUri())).append("\">Magnet</a> ");
- }
- if ((torrentFile.downloadUri() != null) && (torrentFile.downloadUri().length() > 0)) {
- htmlBuilder.append("<a href=\"").append(StringEscapeUtils.escapeHtml4(torrentFile.downloadUri())).append("\">Download</a>");
- }
- htmlBuilder.append("</div>\n");
- }
- htmlBuilder.append("</ul>\n");
- }
- htmlBuilder.append("</ul>\n");
- }
- htmlBuilder.append("</body></html>\n");
- return htmlBuilder.toString();
+ return triggered;
}
}