2 * Rhynodge - NewEpisodeTrigger.java - Copyright © 2013 David Roden
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 package net.pterodactylus.rhynodge.triggers;
20 import static com.google.common.base.Preconditions.checkState;
22 import java.util.Collection;
24 import java.util.Map.Entry;
26 import net.pterodactylus.rhynodge.Reaction;
27 import net.pterodactylus.rhynodge.State;
28 import net.pterodactylus.rhynodge.Trigger;
29 import net.pterodactylus.rhynodge.output.DefaultOutput;
30 import net.pterodactylus.rhynodge.output.Output;
31 import net.pterodactylus.rhynodge.states.EpisodeState;
32 import net.pterodactylus.rhynodge.states.EpisodeState.Episode;
33 import net.pterodactylus.rhynodge.states.TorrentState.TorrentFile;
35 import org.apache.commons.lang3.StringEscapeUtils;
37 import com.google.common.base.Function;
38 import com.google.common.collect.FluentIterable;
39 import com.google.common.collect.ImmutableMap;
40 import com.google.common.collect.Lists;
41 import com.google.common.collect.Maps;
42 import com.google.common.collect.Ordering;
43 import com.google.common.collect.Sets;
46 * {@link Trigger} implementation that compares two {@link EpisodeState}s for
47 * new and changed {@link Episode}s.
49 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
51 public class NewEpisodeTrigger implements Trigger {
54 private final Collection<Episode> allEpisodes = Sets.newHashSet();
56 /** All new episodes. */
57 private final Collection<Episode> newEpisodes = Sets.newHashSet();
59 /** All changed episodes. */
60 private final Collection<Episode> changedEpisodes = Sets.newHashSet();
71 public State mergeStates(State previousState, State currentState) {
72 checkState(currentState instanceof EpisodeState, "currentState is not a EpisodeState but a %s", currentState.getClass().getName());
73 checkState(previousState instanceof EpisodeState, "previousState is not a EpisodeState but a %s", currentState.getClass().getName());
75 changedEpisodes.clear();
76 this.allEpisodes.clear();
77 Map<Episode, Episode> allEpisodes = Maps.newHashMap(FluentIterable.from(((EpisodeState) previousState).episodes()).toMap(new Function<Episode, Episode>() {
80 public Episode apply(Episode episode) {
84 for (Episode episode : ((EpisodeState) currentState).episodes()) {
85 if (!allEpisodes.containsKey(episode)) {
86 allEpisodes.put(episode, episode);
87 newEpisodes.add(episode);
89 for (TorrentFile torrentFile : Lists.newArrayList(episode.torrentFiles())) {
90 int oldSize = allEpisodes.get(episode).torrentFiles().size();
91 allEpisodes.get(episode).addTorrentFile(torrentFile);
92 int newSize = allEpisodes.get(episode).torrentFiles().size();
93 if (!newEpisodes.contains(episode) && (oldSize != newSize)) {
94 changedEpisodes.add(episode);
98 this.allEpisodes.addAll(allEpisodes.values());
99 return new EpisodeState(this.allEpisodes);
106 public boolean triggers() {
107 return !newEpisodes.isEmpty() || !changedEpisodes.isEmpty();
114 public Output output(Reaction reaction) {
116 if (!newEpisodes.isEmpty()) {
117 if (!changedEpisodes.isEmpty()) {
118 summary = String.format("%d new and %d changed Torrent(s) for “%s!”", newEpisodes.size(), changedEpisodes.size(), reaction.name());
120 summary = String.format("%d new Torrent(s) for “%s!”", newEpisodes.size(), reaction.name());
123 summary = String.format("%d changed Torrent(s) for “%s!”", changedEpisodes.size(), reaction.name());
125 DefaultOutput output = new DefaultOutput(summary);
126 output.addText("text/plain", generatePlainText(reaction, newEpisodes, changedEpisodes, allEpisodes));
127 output.addText("text/html", generateHtmlText(reaction, newEpisodes, changedEpisodes, allEpisodes));
136 * Generates the plain text trigger output.
139 * The reaction that was triggered
142 * @param changedEpisodes
143 * The changed episodes
146 * @return The plain text output
148 private static String generatePlainText(Reaction reaction, Collection<Episode> newEpisodes, Collection<Episode> changedEpisodes, Collection<Episode> allEpisodes) {
149 StringBuilder stringBuilder = new StringBuilder();
150 if (!newEpisodes.isEmpty()) {
151 stringBuilder.append(reaction.name()).append(" - New Episodes\n\n");
152 for (Episode episode : newEpisodes) {
153 stringBuilder.append("- ").append(episode.identifier()).append("\n");
154 for (TorrentFile torrentFile : episode) {
155 stringBuilder.append(" - ").append(torrentFile.name()).append(", ").append(torrentFile.size()).append("\n");
156 if ((torrentFile.magnetUri() != null) && (torrentFile.magnetUri().length() > 0)) {
157 stringBuilder.append(" Magnet: ").append(torrentFile.magnetUri()).append("\n");
159 if ((torrentFile.downloadUri() != null) && (torrentFile.downloadUri().length() > 0)) {
160 stringBuilder.append(" Download: ").append(torrentFile.downloadUri()).append("\n");
165 if (!changedEpisodes.isEmpty()) {
166 stringBuilder.append(reaction.name()).append(" - Changed Episodes\n\n");
167 for (Episode episode : changedEpisodes) {
168 stringBuilder.append("- ").append(episode.identifier()).append("\n");
169 for (TorrentFile torrentFile : episode) {
170 stringBuilder.append(" - ").append(torrentFile.name()).append(", ").append(torrentFile.size()).append("\n");
171 if ((torrentFile.magnetUri() != null) && (torrentFile.magnetUri().length() > 0)) {
172 stringBuilder.append(" Magnet: ").append(torrentFile.magnetUri()).append("\n");
174 if ((torrentFile.downloadUri() != null) && (torrentFile.downloadUri().length() > 0)) {
175 stringBuilder.append(" Download: ").append(torrentFile.downloadUri()).append("\n");
180 /* list all known episodes. */
181 stringBuilder.append(reaction.name()).append(" - All Known Episodes\n\n");
182 ImmutableMap<Integer, Collection<Episode>> episodesBySeason = FluentIterable.from(allEpisodes).index(new Function<Episode, Integer>() {
185 public Integer apply(Episode episode) {
186 return episode.season();
189 for (Entry<Integer, Collection<Episode>> seasonEntry : episodesBySeason.entrySet()) {
190 stringBuilder.append(" Season ").append(seasonEntry.getKey()).append("\n\n");
191 for (Episode episode : Ordering.natural().sortedCopy(seasonEntry.getValue())) {
192 stringBuilder.append(" Episode ").append(episode.episode()).append("\n");
193 for (TorrentFile torrentFile : episode) {
194 stringBuilder.append(" Size: ").append(torrentFile.size());
195 stringBuilder.append(" in ").append(torrentFile.fileCount()).append(" file(s): ");
196 stringBuilder.append(torrentFile.magnetUri());
201 return stringBuilder.toString();
205 * Generates the HTML trigger output.
208 * The reaction that was triggered
211 * @param changedEpisodes
212 * The changed episodes
215 * @return The HTML output
217 private static String generateHtmlText(Reaction reaction, Collection<Episode> newEpisodes, Collection<Episode> changedEpisodes, Collection<Episode> allEpisodes) {
218 StringBuilder htmlBuilder = new StringBuilder();
219 htmlBuilder.append("<html><body>\n");
220 htmlBuilder.append("<h1>").append(StringEscapeUtils.escapeHtml4(reaction.name())).append("</h1>\n");
221 if (!newEpisodes.isEmpty()) {
222 htmlBuilder.append("<h2>New Episodes</h2>\n");
223 htmlBuilder.append("<ul>\n");
224 for (Episode episode : newEpisodes) {
225 htmlBuilder.append("<li>Season ").append(episode.season()).append(", Episode ").append(episode.episode()).append("</li>\n");
226 htmlBuilder.append("<ul>\n");
227 for (TorrentFile torrentFile : episode) {
228 htmlBuilder.append("<li>").append(StringEscapeUtils.escapeHtml4(torrentFile.name())).append("</li>\n");
229 htmlBuilder.append("<div>");
230 htmlBuilder.append("<strong>").append(StringEscapeUtils.escapeHtml4(torrentFile.size())).append("</strong>, ");
231 htmlBuilder.append("<strong>").append(torrentFile.fileCount()).append("</strong> file(s), ");
232 htmlBuilder.append("<strong>").append(torrentFile.seedCount()).append("</strong> seed(s), ");
233 htmlBuilder.append("<strong>").append(torrentFile.leechCount()).append("</strong> leecher(s)</div>\n");
234 htmlBuilder.append("<div>");
235 if ((torrentFile.magnetUri() != null) && (torrentFile.magnetUri().length() > 0)) {
236 htmlBuilder.append("<a href=\"").append(StringEscapeUtils.escapeHtml4(torrentFile.magnetUri())).append("\">Magnet</a> ");
238 if ((torrentFile.downloadUri() != null) && (torrentFile.downloadUri().length() > 0)) {
239 htmlBuilder.append("<a href=\"").append(StringEscapeUtils.escapeHtml4(torrentFile.downloadUri())).append("\">Download</a>");
241 htmlBuilder.append("</div>\n");
243 htmlBuilder.append("</ul>\n");
245 htmlBuilder.append("</ul>\n");
247 if (!changedEpisodes.isEmpty()) {
248 htmlBuilder.append("<h2>Changed Episodes</h2>\n");
249 htmlBuilder.append("<ul>\n");
250 for (Episode episode : changedEpisodes) {
251 htmlBuilder.append("<li>Season ").append(episode.season()).append(", Episode ").append(episode.episode()).append("</li>\n");
252 htmlBuilder.append("<ul>\n");
253 for (TorrentFile torrentFile : episode) {
254 htmlBuilder.append("<li>").append(StringEscapeUtils.escapeHtml4(torrentFile.name())).append("</li>\n");
255 htmlBuilder.append("<div>");
256 htmlBuilder.append("<strong>").append(StringEscapeUtils.escapeHtml4(torrentFile.size())).append("</strong>, ");
257 htmlBuilder.append("<strong>").append(torrentFile.fileCount()).append("</strong> file(s), ");
258 htmlBuilder.append("<strong>").append(torrentFile.seedCount()).append("</strong> seed(s), ");
259 htmlBuilder.append("<strong>").append(torrentFile.leechCount()).append("</strong> leecher(s)</div>\n");
260 htmlBuilder.append("<div>");
261 if ((torrentFile.magnetUri() != null) && (torrentFile.magnetUri().length() > 0)) {
262 htmlBuilder.append("<a href=\"").append(StringEscapeUtils.escapeHtml4(torrentFile.magnetUri())).append("\">Magnet</a> ");
264 if ((torrentFile.downloadUri() != null) && (torrentFile.downloadUri().length() > 0)) {
265 htmlBuilder.append("<a href=\"").append(StringEscapeUtils.escapeHtml4(torrentFile.downloadUri())).append("\">Download</a>");
267 htmlBuilder.append("</div>\n");
269 htmlBuilder.append("</ul>\n");
271 htmlBuilder.append("</ul>\n");
273 htmlBuilder.append("</body></html>\n");
274 return htmlBuilder.toString();